ihealth_ios/Inhealth/THScrollChooseView.m

//
//  THScrollChooseView.m
//  THChooseTool


#define THScreenW [UIScreen mainScreen].bounds.size.width

#define THScreenH [UIScreen mainScreen].bounds.size.height

#define THWParam [UIScreen mainScreen].bounds.size.width/375.0f

#define THfloat(a) a*THWParam

#import "THScrollChooseView.h"


@interface THScrollChooseView()<UIPickerViewDataSource, UIPickerViewDelegate>


@property (strong, nonatomic) UIPickerView *pickerView;

@property (strong, nonatomic) UIView *bottomView;

/**
 取消按钮
 */
@property (strong, nonatomic) UIButton *cancelButton;

/**
 确定按钮
 */
@property (strong, nonatomic) UIButton *confirmButton;

/**
 选中数据是第几条
 */
@property (assign, nonatomic) NSInteger selectedValue;

/**
 数组
 */
@property (strong, nonatomic) NSArray *questionArray;
/**
 默认的值
 */
@property (copy, nonatomic) NSString *defaultDesc;

@end

@implementation THScrollChooseView


static NSInteger recordRowOfQuestion;


- (instancetype)initWithQuestionArray:(NSArray *)questionArray withDefaultDesc:(NSString *)defaultDesc {
    
    if (self = [super init]) {
    
        self.backgroundColor = [[UIColor blackColor]colorWithAlphaComponent:0.3];
        
        self.frame = CGRectMake(0, 0, THScreenW, THScreenH);
        
        UIView *whiteView = [[UIView alloc]initWithFrame:CGRectMake(0, CGRectGetHeight(self.frame)-260, CGRectGetWidth(self.frame), 260)];
        whiteView.backgroundColor = [UIColor whiteColor];
        [self addSubview:whiteView];
        
        //按钮所在区域
        UIView *viewBg = [[UIView alloc] initWithFrame:CGRectMake(0,0,THScreenW,THfloat(45))];
        [viewBg setBackgroundColor:[UIColor whiteColor]];
        [whiteView addSubview:viewBg];
        
        //创建取消 确定按钮
        UIButton *cannel = [UIButton buttonWithType:UIButtonTypeCustom];
        cannel.frame = CGRectMake(THfloat(20), 0, THfloat(50), THfloat(45));
        [cannel setTitle:NSLocalizedString(@"取消",  nil) forState:0];
        cannel.titleLabel.font = [UIFont fontWithName:@"PingFangSC-Semibold" size:13];
        [cannel setTitleColor:[UIColor darkGrayColor] forState:UIControlStateNormal];
        cannel.tag = 1;
        [cannel addTarget:self action:@selector(cancelButtonAction:) forControlEvents:UIControlEventTouchUpInside];
        [viewBg addSubview:cannel];
        
        UIButton *confirm = [UIButton buttonWithType:UIButtonTypeCustom];
        confirm.frame = CGRectMake(THScreenW - THfloat(70), 0, THfloat(50), THfloat(45));
        [confirm setTitle:NSLocalizedString(@"确定",  nil) forState:0];
        confirm.titleLabel.font = [UIFont fontWithName:@"PingFangSC-Semibold" size:13];
        [confirm setTitleColor:[UIColor darkGrayColor] forState:UIControlStateNormal];
        confirm.tag = 2;
        [confirm addTarget:self action:@selector(confirmButtonAction:) forControlEvents:UIControlEventTouchUpInside];
        [viewBg addSubview:confirm];
        
        
        self.questionArray = questionArray;
        self.defaultDesc = defaultDesc;
        [whiteView addSubview:self.pickerView];
        //[self.pickerView reloadAllComponents];
        
    }
    
    return self;
}


#pragma mark - action

- (void)cancelButtonAction:(UIButton *)button {
    
    [self dismissView];
    
}

- (void)confirmButtonAction:(UIButton *)button {
    if (self.selectedValue >= self.questionArray.count) { //防止切换的时候 会出现问题
        self.selectedValue = 0; //啥都不选，就点击OK //有默认值的话，一定是小雨self.questionArray.count
    }
    self.confirmBlock(self.selectedValue);
    [self dismissView];
}

- (void)showView{
    
    [[UIApplication sharedApplication].keyWindow addSubview:self];
    self.alpha = 0;
    [UIView animateWithDuration:0.3 animations:^{
        self.alpha = 1;
        
    } completion:^(BOOL finished) {
        
    }];
    
}

- (void)dismissView{
    [UIView animateWithDuration:0.3 animations:^{
        self.alpha = 0;
    } completion:^(BOOL finished) {
        [self removeFromSuperview];
    }];
}

- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
    [self dismissView];
}


#pragma mark - pickerView 代理方法

// 列数
- (NSInteger)numberOfComponentsInPickerView:(UIPickerView *)pickerView {
    
    return 1;
}

// 行数
- (NSInteger)pickerView:(UIPickerView *)pickerView numberOfRowsInComponent:(NSInteger)component {
    
    return self.questionArray.count;
}

-(CGFloat)pickerView:(UIPickerView*)pickerView rowHeightForComponent:(NSInteger)component {
    return THfloat(35);
}

// 显示什么
- (NSString*)pickerView:(UIPickerView *)pickerView titleForRow:(NSInteger)row forComponent:(NSInteger)component{
    
    return self.questionArray[row];
}

// 选中时
- (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component {
    
    self.selectedValue = row;
}


- (UIPickerView *)pickerView {
    
    if (!_pickerView) {
        _pickerView = [[UIPickerView alloc] initWithFrame:CGRectMake(0, THfloat(45), THScreenW, THfloat(180))];
        _pickerView.backgroundColor = [UIColor clearColor];
        _pickerView.dataSource = self;
        _pickerView.delegate = self;
        
        // _pickerView 初始化显示的值
        if (self.defaultDesc.length == 0) {
            //无默认值时显示第一个
            [_pickerView selectRow:0 inComponent:0 animated:YES];
            
        }else {
            //有默认值时显示默认值
            recordRowOfQuestion = [self rowOfQuestionWithName:self.defaultDesc];
            [_pickerView selectRow:recordRowOfQuestion inComponent:0 animated:YES];
            self.selectedValue = recordRowOfQuestion;
            
        }
    }
    return _pickerView;
}


- (NSInteger)rowOfQuestionWithName:(NSString *)questionName{
    
    NSInteger row = 0;
    for (NSString *str in self.questionArray) {
        if ([str containsString:questionName]) {
            NSLog(@"row = %ld",row);
            return row;
        }
        row++;
    }
    return row;
}

@end