項目中，我們往往需要用到選擇器選擇身高、性別等。這篇文章教大家做一個炫酷的pickerViewController~
使用方法：
#import "PickerViewController.h"
[UIApplication sharedApplication].statusBarStyle = UIStatusBarStyleLightContent;
//選中傳值
PickerViewController *picker = [[PickerViewController alloc]init];
picker.myblock = ^(id data1){
NSLog(@"%@",data1);
};
//自定義數(shù)組
NSMutableArray *arr = [NSMutableArray array];
for (int i = 140; i<200; i++) {
[arr addObject:[NSString stringWithFormat:@"%dcm",i]];
}
picker.dataArray = [NSArray arrayWithArray:arr];
[picker createPickerWithTitile:@"身高" locText:@"165cm"];
[self presentViewController:picker animated:YES completion:nil];
實現(xiàn)
pickerViewController.h
@interface PickerViewController : UIViewController
/** 外界傳數(shù)組等*/
@property (nonatomic, strong) NSArray *dataArray;
/** 設置頂部顏色*/
@property (nonatomic, strong) UIColor *titleColor;
@property (nonatomic, strong)MyBlock myblock;
/**
* title:標題 type:數(shù)組類型 locText:默認顯示數(shù)據(jù)，如178cm
*/
- (void)createPickerWithTitile:(NSString *)title
locText:(NSString *)locText;
PickerViewController.m
#import "PickerViewController.h"
#import "PresentOneTransition.h"
#import "UIView+Extension.h"
#define KButtonH 30
#define KButtonW 50
@interface PickerViewController ()<UIViewControllerTransitioningDelegate,UIPickerViewDelegate,UIPickerViewDataSource>
{
//默認行
NSUInteger placeHolderRow;
//選中的
NSString *chooseStr;
}
/** 選擇器*/
@property (nonatomic, strong) UIPickerView *pickerView;
/** 取消按鈕*/
@property (nonatomic, strong) UIButton *leftButton;
/** 確定*/
@property (nonatomic, strong) UIButton *rightButton;
/** 標題View*/
@property (nonatomic, strong) UIView *titleView;
/** 標題lable*/
@property (nonatomic, strong) UILabel *titileLabel;
@end
@implementation PickerViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.view.layer.cornerRadius = 10;
self.view.layer.masksToBounds = YES;
self.view.backgroundColor = [UIColor whiteColor];
}
- (instancetype)init
{
self = [super init];
if (self) {
self.transitioningDelegate = self;
self.modalPresentationStyle = UIModalPresentationCustom;
}
return self;
}
- (UIButton *)rightButton {
if (!_rightButton) {
UIButton *rightButton = [self setBtnWithTitle:@"確定"];
_rightButton = rightButton;
}
return _rightButton;
}
- (UIButton *)leftButton {
if (!_leftButton) {
UIButton *leftButton = [self setBtnWithTitle:@"取消"];
_leftButton = leftButton;
}
return _leftButton;
}
- (UILabel *)titileLabel {
if (!_titileLabel) {
UILabel *titleLabel = [[UILabel alloc]init];
titleLabel.textAlignment = NSTextAlignmentCenter;
titleLabel.font = [UIFont systemFontOfSize:12];
titleLabel.textColor = [UIColor blackColor];
titleLabel.width = self.view.width - 2*KButtonW;
titleLabel.height = KButtonH;
titleLabel.x = KButtonW;
_titileLabel = titleLabel;
}
return _titileLabel;
}
- (UIView *)titleView {
if (!_titleView) {
UIView *titleView = [[UIView alloc]init];
titleView.x = 0;
titleView.y = 0;
titleView.width = self.view.width;
titleView.height = KButtonH;
titleView.backgroundColor = self.titleColor;
[titleView addSubview:self.leftButton];
[titleView addSubview:self.rightButton];
[titleView addSubview:self.titileLabel];
self.leftButton.x = 0;
self.leftButton.y = 0;
self.rightButton.x = titleView.width - KButtonW;
self.rightButton.y = 0;
_titleView = titleView;
}
return _titleView;
}
- (UIPickerView *)pickerView {
if (!_pickerView) {
UIPickerView *pickerView = [[UIPickerView alloc]init];
[[UIPickerView appearance] setBackgroundColor:[UIColor whiteColor]];
pickerView.x = 0;
pickerView.width = self.view.width;
pickerView.y = CGRectGetMaxY(self.titleView.frame);
// pickerView.height = self.view.height - KButtonH;
pickerView.dataSource = self;
pickerView.delegate = self;
[pickerView reloadAllComponents];
_pickerView = pickerView;
}
return _pickerView;
}
- (UIButton *)setBtnWithTitle:(NSString *)title {
UIButton *btn = [[UIButton alloc]init];
btn.height = KButtonH;
btn.width = KButtonW;
[btn setTitle:title forState:UIControlStateNormal];
btn.titleLabel.font = [UIFont systemFontOfSize:12];
[btn setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
[btn addTarget:self action:@selector(dismiss:) forControlEvents:UIControlEventTouchUpInside];
return btn;
}
/** 可以自己定義二維數(shù)組、三維數(shù)組。根據(jù)type傳數(shù)據(jù)，pickerView 的Component根據(jù)數(shù)組維度來變*/
- (void)createPickerWithTitile:(NSString *)title
locText:(NSString *)locText
{
//subViews
[self.view addSubview:self.titleView];
[self.view addSubview:self.pickerView];
//賦值
self.titileLabel.text = title;
//獲得默認顯示列
NSUInteger row = [self returnRowwithArr:self.dataArray withStr:locText];
//默認
chooseStr = title;
//選中
[self.pickerView selectRow:row inComponent:0 animated:YES];
}
//確定，傳值
- (void)dismiss:(UIButton *)button
{
[self presentedControllerPressedDissmiss];
if ([button isEqual:self.leftButton]) {
}else if ([button isEqual:self.rightButton]){
if (_myblock) {
_myblock(chooseStr);
}
}
}
- (void)presentedControllerPressedDissmiss
{
[self dismissViewControllerAnimated:YES completion:^{
[UIApplication sharedApplication].statusBarStyle = UIStatusBarStyleDefault;
}];
}
//記住相應字符串在數(shù)組中得位置
- (NSUInteger)returnRowwithArr:(NSArray *)arr withStr:(NSString *)str
{
NSUInteger index = 0;
for (int i = 0; i<arr.count; i++) {
if ([str isEqual:arr[i]]) {
index = i;
}
}
return index;
}
- (UIColor *)titleColor {
if (!_titleColor) {
_titleColor = Color(250, 230, 0);
}
return _titleColor;
}
- (NSInteger)pickerView:(UIPickerView *)pickerView numberOfRowsInComponent:(NSInteger)component {
return self.dataArray.count;
}
- (NSInteger)numberOfComponentsInPickerView:(UIPickerView *)pickerView;
{
return 1;
}
- (NSString *)pickerView:(UIPickerView *)pickerView titleForRow:(NSInteger)row forComponent:(NSInteger)component {
return self.dataArray[row];
}
- (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component {
chooseStr = self.dataArray[row];
}
- (id<UIViewControllerAnimatedTransitioning>)animationControllerForPresentedController:(UIViewController *)presented presentingController:(UIViewController *)presenting sourceController:(UIViewController *)source{
return [PresentOneTransition transitionWithTransitionType:PresentOneTransitionTypePresent];
}
- (id<UIViewControllerAnimatedTransitioning>)animationControllerForDismissedController:(UIViewController *)dismissed{
return [PresentOneTransition transitionWithTransitionType:PresentOneTransitionTypeDismiss];
}
@end
test.gif
** 可以自行擴充數(shù)組維度，比如說兩列picker,但是pickerView的dataSource和代理要做相應的改變**
代碼地址：我刪了...之后再重新上傳
