一行代碼實現(xiàn)三級地區(qū)選擇器，可使用自定義方法創(chuàng)建，返回選中地區(qū)，實現(xiàn)地區(qū)選擇的三級聯(lián)動。
感謝兔斯拉老哥跟我一起探討地區(qū)選擇器的一些問題，學(xué)到了一些我曾經(jīng)不懂的東西~
以下更新內(nèi)容為探討出的實現(xiàn)。
三級地區(qū)選擇器
1. SHPlacePickerView內(nèi)部代碼
SHPlacePickerView.h
//
// SHPlacePickerView.h
// SHPlacePickViewDemo
//
// Created by HarrySun on 2017/3/31.
// Copyright ? 2017年 Mobby. All rights reserved.
//
#import <UIKit/UIKit.h>
typedef void(^SendPlaceArray)(NSArray *placeArray);
@interface SHPlacePickerView : UIView
/**
傳出選中數(shù)組
*/
@property (nonatomic,strong) SendPlaceArray sendPlaceArray;
/**
是否定位到上次選中位置
*/
@property (nonatomic, assign) BOOL isRecordLocation;
/**
創(chuàng)建SHPlacePickerView
@param isrecordLocation 是否定位到上次選中地區(qū)
@param sendPlaceArray 傳出選中的地區(qū)數(shù)組
@return SHPlacePickerView
*/
- (instancetype)initWithIsRecordLocation:(BOOL)isrecordLocation SendPlaceArray:(SendPlaceArray)sendPlaceArray;
@end
SHPlacePickerView.m
//
// SHPlacePickerView.m
// SHPlacePickViewDemo
//
// Created by HarrySun on 2017/3/31.
// Copyright ? 2017年 Mobby. All rights reserved.
//
#import "SHPlacePickerView.h"
#import "PlaceModel.h"
@interface SHPlacePickerView ()<UIPickerViewDataSource, UIPickerViewDelegate>
@property (nonatomic, strong) NSMutableArray *dataArray; // 存放model的數(shù)據(jù)數(shù)組
@property (nonatomic, strong) UIPickerView *pickerView; // pickerView
@property (nonatomic, strong) UIView *toolView; // 工具視圖
// 存儲索引數(shù)據(jù)
@property (nonatomic, strong) NSArray *selectedProvinceArray;
@property (nonatomic, strong) NSMutableArray *selectedCityArray;
@property (nonatomic, strong) NSMutableArray *selectedDistrictArray;
@property (nonatomic, strong) NSArray *saveArray; // 存儲選中地區(qū)索引數(shù)組
@property (nonatomic, strong) NSArray *selectArray; // 存儲選中地區(qū)數(shù)組
@property (nonatomic, strong) NSString *selectedProvince;
@property (nonatomic, strong) NSString *selectedCity;
@property (nonatomic, strong) NSString *selectedDistrict;
@end
@implementation SHPlacePickerView
- (instancetype)initWithIsRecordLocation:(BOOL)isrecordLocation SendPlaceArray:(SendPlaceArray)sendPlaceArray{
self.sendPlaceArray = sendPlaceArray;
self.isRecordLocation = isrecordLocation;
self = [self init];
if (self.isRecordLocation) {
// 跳到上次選中位置
if ([[NSUserDefaults standardUserDefaults] objectForKey:@"saveArray"]) {
NSArray *array = [[NSUserDefaults standardUserDefaults] objectForKey:@"saveArray"];
[_pickerView selectRow:[array[0] integerValue] inComponent:0 animated:YES];
[_pickerView selectRow:[array[1] integerValue] inComponent:1 animated:YES];
[_pickerView selectRow:[array[2] integerValue] inComponent:2 animated:YES];
}else{
[_pickerView selectRow:0 inComponent:0 animated:YES];
}
}
return self;
}
- (instancetype)init{
self = [super initWithFrame:CGRectMake(0, [UIScreen mainScreen].bounds.size.height - 250, [UIScreen mainScreen].bounds.size.width, 250)];
if (self) {
[self loadData];
[self drawView];
}
return self;
}
#pragma mark - UIPickerViewDataSource
- (NSInteger)numberOfComponentsInPickerView:(UIPickerView *)pickerView{
return 3;
}
- (NSInteger)pickerView:(UIPickerView *)pickerView numberOfRowsInComponent:(NSInteger)component{
[self getSelectedArray];
switch (component) {
case 0:
return self.dataArray.count;
break;
case 1:
return self.selectedCityArray.count;
break;
case 2:
return self.selectedDistrictArray.count;
break;
default:
return 0;
break;
}
}
#pragma mark - UIPickerViewDelegate
// 自定義行
- (UIView *)pickerView:(UIPickerView *)pickerView viewForRow:(NSInteger)row forComponent:(NSInteger)component reusingView:(UIView *)view{
[self getSelectedArray];
UILabel* pickerLabel = (UILabel*)view;
if (!pickerLabel){
pickerLabel = [[UILabel alloc] init];
// pickerLabel.adjustsFontSizeToFitWidth = YES; // 是否根據(jù)寬度適應(yīng)文字大小
[pickerLabel setTextAlignment:NSTextAlignmentCenter];
[pickerLabel setBackgroundColor:[UIColor clearColor]];
[pickerLabel setFont:[UIFont boldSystemFontOfSize:17]];
}
NSString *text;
switch (component) {
case 0:
text = self.selectedProvinceArray[row];
break;
case 1:
text = self.selectedCityArray[row];
break;
case 2:
text = self.selectedDistrictArray[row];
break;
default:
text = @"";
break;
}
pickerLabel.text = text;
return pickerLabel;
}
- (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component{
switch (component) {
case 0:{
self.selectedProvince = self.selectedProvinceArray[row];
self.selectedCity = [self.dataArray[row] cityArray][0];
self.selectedDistrict = [self.dataArray[row] districtArray][0][0];
[self getSelectedArray];
[pickerView reloadComponent:1];
[pickerView reloadComponent:2];
}
break;
case 1:{
NSInteger provinceIndex = [self.selectedProvinceArray indexOfObject:self.selectedProvince];
self.selectedCity = [self.dataArray[provinceIndex] cityArray][row];
self.selectedDistrict = [self.dataArray[provinceIndex] districtArray][row][0];
[self getSelectedArray];
[pickerView reloadComponent:2];
}
break;
case 2:{
self.selectedDistrict = self.selectedDistrictArray[row];
}
break;
default:
break;
}
// 存下當(dāng)前選擇的地區(qū)
if(self.isRecordLocation){
NSInteger provinceIndex = [self.selectedProvinceArray indexOfObject:_selectedProvince];
NSArray *selectedCityArray = [self.dataArray[provinceIndex] cityArray];
NSInteger cityIndex = [selectedCityArray indexOfObject:_selectedCity];
NSArray *districtArray = [self.dataArray[provinceIndex] districtArray][cityIndex];
NSInteger districtIndex = [districtArray indexOfObject:_selectedDistrict];
self.saveArray = @[[NSNumber numberWithInteger:provinceIndex],[NSNumber numberWithInteger:cityIndex],[NSNumber numberWithInteger:districtIndex]];
}
if (self.selectedProvince && self.selectedCity && self.selectedDistrict) {
self.selectArray = @[self.selectedProvince,self.selectedCity,self.selectedDistrict];
// NSLog(@"省:%@ 市:%@ 區(qū):%@",self.selectedProvince,self.selectedCity,self.selectedDistrict);
}
}
#pragma mark - getSelectedArray
- (void)getSelectedArray{
NSInteger provinceIndex = [self.selectedProvinceArray indexOfObject:self.selectedProvince];
self.selectedCityArray = [self.dataArray[provinceIndex] cityArray];
NSInteger cityIndex = [self.selectedCityArray indexOfObject:self.selectedCity];
self.selectedDistrictArray = [self.dataArray[provinceIndex] districtArray][cityIndex];
}
#pragma mark - 繪制pickerView
- (void)drawView{
_pickerView = [[UIPickerView alloc]initWithFrame:CGRectMake(0, 34, self.frame.size.width, 216)];
[self.pickerView setBackgroundColor:[UIColor whiteColor]];
[self.pickerView setDataSource:self];
[self.pickerView setDelegate:self];
[self addSubview:self.pickerView];
self.toolView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, self.frame.size.width, 34)];
[self addSubview:self.toolView];
UIView *belowLineView = [[UIView alloc] initWithFrame:CGRectMake(0, 33, self.frame.size.width, 1)];
belowLineView.backgroundColor = [UIColor grayColor];
[self.toolView addSubview:belowLineView];
UIView *aboveLineView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, self.frame.size.width, 1)];
aboveLineView.backgroundColor = [UIColor grayColor];
[self.toolView addSubview:aboveLineView];
UIButton *cancelButton = [UIButton buttonWithType:UIButtonTypeCustom];
cancelButton.frame = CGRectMake(10, 2, 40, 28);
[cancelButton setTitle:@"取消" forState:UIControlStateNormal];
[cancelButton setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
[cancelButton addTarget:self action:@selector(cancelAction) forControlEvents:UIControlEventTouchUpInside];
[self.toolView addSubview:cancelButton];
UIButton *sureButton = [UIButton buttonWithType:UIButtonTypeCustom];
sureButton.frame = CGRectMake(self.bounds.size.width - 50, 2, 40, 28);
[sureButton setTitle:@"確認(rèn)" forState:UIControlStateNormal];
[sureButton setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
[sureButton addTarget:self action:@selector(sureAction) forControlEvents:UIControlEventTouchUpInside];
[self.toolView addSubview:sureButton];
}
#pragma mark - 請求地區(qū)Plist文件
- (void)loadData{
self.dataArray = [NSMutableArray arrayWithCapacity:34];
NSDictionary *dict = [NSDictionary dictionaryWithContentsOfFile:[[NSBundle mainBundle] pathForResource:@"Place" ofType:@"plist"]];
NSArray *provinceArray = [dict allKeys];
self.selectedProvinceArray = provinceArray;
for (int i = 0; i < provinceArray.count; i ++) {
PlaceModel *placeModel = [[PlaceModel alloc] init];
placeModel.provinceName = provinceArray[i];
NSDictionary *cityDict = [[dict objectForKey:provinceArray[i]] firstObject];
[cityDict enumerateKeysAndObjectsUsingBlock:^(id _Nonnull key, id _Nonnull obj, BOOL * _Nonnull stop) {
[placeModel.cityArray addObject:key];
[placeModel.districtArray addObject:obj];
}];
[self.dataArray addObject:placeModel];
}
if (self.isRecordLocation && [[NSUserDefaults standardUserDefaults] objectForKey:@"saveArray"]) {
NSArray *array = [[NSUserDefaults standardUserDefaults] objectForKey:@"saveArray"];
PlaceModel *placeModel = self.dataArray[[array[0] integerValue]];
self.selectedProvince = placeModel.provinceName;
self.selectedCity = placeModel.cityArray[[array[1] integerValue]];
self.selectedDistrict = placeModel.districtArray[[array[1] integerValue]][[array[2] integerValue]];
}else{
PlaceModel *placeModel = self.dataArray[0];
self.selectedProvince = placeModel.provinceName;
self.selectedCity = placeModel.cityArray[0];
self.selectedDistrict = placeModel.districtArray[0][0];
}
[self getSelectedArray];
}
#pragma mark - ToolViewAction
- (void)cancelAction{
if (!_isRecordLocation) {
self.saveArray = nil;
}
[self removeFromSuperview];
}
- (void)sureAction{
// 存下當(dāng)前選擇的地區(qū)
if(self.isRecordLocation){
[[NSUserDefaults standardUserDefaults] setObject:self.saveArray forKey:@"saveArray"];
}
if(self.selectArray){
_sendPlaceArray(self.selectArray);
[self removeFromSuperview];
}else{
NSString *title = NSLocalizedString(@"未選擇地區(qū)", nil);
NSString *message = NSLocalizedString(@"請您選擇或者更改地區(qū)后再點(diǎn)擊確定。", nil);
NSString *OKButtonTitle = NSLocalizedString(@"OK", nil);
UIAlertController *alertVC = [UIAlertController alertControllerWithTitle:title message:message preferredStyle:UIAlertControllerStyleAlert];
UIAlertAction *OKAction = [UIAlertAction actionWithTitle:OKButtonTitle style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
}];
[alertVC addAction:OKAction];
[self.window.rootViewController presentViewController:alertVC animated:YES completion:nil];
}
}
@end
2.Model類代碼
PlaceModel.h
//
// PlaceModel.h
// SHPlacePickViewDemo
//
// Created by HarrySun on 2017/3/31.
// Copyright ? 2017年 Mobby. All rights reserved.
//
#import <Foundation/Foundation.h>
@interface PlaceModel : NSObject
@property (nonatomic, strong) NSString *provinceName;
@property (nonatomic, strong) NSMutableArray *cityArray;
@property (nonatomic, strong) NSMutableArray *districtArray;
@end
PlaceModel.m
//
// PlaceModel.m
// SHPlacePickViewDemo
//
// Created by HarrySun on 2017/3/31.
// Copyright ? 2017年 Mobby. All rights reserved.
//
#import "PlaceModel.h"
@implementation PlaceModel
- (instancetype)init{
self = [super init];
if (self) {
self.provinceName = @"";
self.cityArray = [NSMutableArray array];
self.districtArray = [NSMutableArray array];
}
return self;
}
@end
3. SHPlacePickerView使用
使用自定義方法創(chuàng)建：
__weak __typeof(self)weakSelf = self;
self.shplacePicker = [[SHPlacePickerView alloc] initWithIsRecordLocation:YES SendPlaceArray:^(NSArray *placeArray) {
[weakSelf.selectButton setTitle:[NSString stringWithFormat:@"省:%@ 市:%@ 區(qū):%@",placeArray[0],placeArray[1],placeArray[2]] forState:UIControlStateNormal];
}];
[self.view addSubview:self.shplacePicker];
不使用自定義方法創(chuàng)建：
self.shplacePicker = [[SHPlacePickerView alloc] init];
self.shplacePicker.isRecordLocation = YES;
__weak __typeof(self)weakSelf = self;
self.shplacePicker.sendPlaceArray = ^(NSArray *placeArray){
NSLog(@"省:%@ 市:%@ 區(qū):%@",placeArray[0],placeArray[1],placeArray[2]);
[weakSelf.selectButton setTitle:[NSString stringWithFormat:@"省:%@ 市:%@ 區(qū):%@",placeArray[0],placeArray[1],placeArray[2]] forState:UIControlStateNormal];
};
[self.view addSubview:self.shplacePicker];
