MY_Swift 之封装三级连版

效果图

三级联动效果图.png

Demo地址

https://github.com/VodkaMY/MYLinkagePickView

Demo 封装代码

//
//  SdLinkagePickViewCtr.swift
//  MYCandy
//
//  Created by MaoYan on 2017/9/15.
//  Copyright © 2017年 MYCandy.com. All rights reserved.
//

import UIKit

class SdLinkagePickViewCtr: UIView, UIPickerViewDelegate ,UIPickerViewDataSource {

    //选择器
    fileprivate var pickerView:UIPickerView!
    
    //所以地址数据集合
    fileprivate var addressArray = [[String: AnyObject]]()

    //选择的省索引
    fileprivate var provinceIndex = 0
    //选择的市索引
    fileprivate var cityIndex = 0
    //选择的县索引
    fileprivate var areaIndex = 0
    
    // 列数
    fileprivate var rowCount:NSInteger?
    
    /// 初始化三集连扳控制器
    ///
    /// - Parameters:
    ///   - frame: 控制器大小
    ///   - addressArray: 城市列表数据源
    ///   - rowCount: 列数
    init(frame: CGRect, addressArr:NSArray, rowC:NSInteger) {
        super.init(frame: frame)
        addressArray = addressArr as! Array
        rowCount = rowC
        
        //创建选择器
        pickerView=UIPickerView()
        
        pickerView.frame = frame
        //将dataSource设置成自己
        pickerView.dataSource=self
        //将delegate设置成自己
        pickerView.delegate=self
        self.addSubview(pickerView)
        
        //建立一个按钮，触摸按钮时获得选择框被选择的索引
        let button = UIButton(frame:CGRect(x:0, y:0, width:100, height:30))
        button.backgroundColor = UIColor.blue
        button.setTitle("获取信息",for:.normal)
        button.addTarget(self, action:#selector(getPickerViewValue),
                         for: .touchUpInside)
        addSubview(button)
    }
    
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    //设置选择框的列数为3列,继承于UIPickerViewDataSource协议
    func numberOfComponents(in pickerView: UIPickerView) -> Int {
        return rowCount ?? 0
    }
    
    //设置选择框的行数，继承于UIPickerViewDataSource协议
    func pickerView(_ pickerView: UIPickerView,
                    numberOfRowsInComponent component: Int) -> Int {
        if component == 0 {
            return self.addressArray.count
        } else if component == 1 {
            let province = self.addressArray[provinceIndex]
            return province["cities"]!.count
        } else {
            let province = self.addressArray[provinceIndex]
            if let city = province["cities"]![cityIndex] as? [String: AnyObject] {
                return city["areas"]!.count
            } else {
                return 0
            }
        }
    }

    //设置选择框各选项的内容，继承于UIPickerViewDelegate协议
    func pickerView(_ pickerView: UIPickerView, titleForRow row: Int,
                    forComponent component: Int) -> String? {
        if component == 0 {
            return self.addressArray[row]["state"] as? String
        }else if component == 1 {
            let province = self.addressArray[provinceIndex]
            let city = province["cities"]![row] as! [String: AnyObject]
            return city["city"] as? String
        }else {
            let province = self.addressArray[provinceIndex]
            let city = province["cities"]![cityIndex] as! [String: AnyObject]
            return city["areas"]![row] as? String
        }
    }

    //选中项改变事件（将在滑动停止后触发）
    func pickerView(_ pickerView: UIPickerView, didSelectRow row: Int,
                    inComponent component: Int) {
        //根据列、行索引判断需要改变数据的区域
        switch (component) {
        case 0:
            provinceIndex = row;
            cityIndex = 0;
            areaIndex = 0;
            pickerView.reloadComponent(1);
            pickerView.reloadComponent(2);
            pickerView.selectRow(0, inComponent: 1, animated: false);
            pickerView.selectRow(0, inComponent: 2, animated: false);
        case 1:
            cityIndex = row;
            areaIndex = 0;
            pickerView.reloadComponent(2);
            pickerView.selectRow(0, inComponent: 2, animated: false);
        case 2:
            areaIndex = row;
        default:
            break;
        }
    }

    //触摸按钮时，获得被选中的索引
    @objc public func getPickerViewValue(){
        //获取选中的省
        let p = self.addressArray[provinceIndex]
        let province = p["state"]!
        
        //获取选中的市
        let c = p["cities"]![cityIndex] as! [String: AnyObject]
        let city = c["city"] as! String
        
        //获取选中的县（地区）
        var area = ""
        if (c["areas"] as! [String]).count > 0 {
            area = (c["areas"] as! [String])[areaIndex]
        }
        
        //拼接输出消息
        let message = "索引：\(provinceIndex)-\(cityIndex)-\(areaIndex)\n"
            + "值：\(province) - \(city) - \(area)"
        
        //消息显示
        let alertController = UIAlertController(title: "您选择了",
                                                message: message, preferredStyle: .alert)
        let cancelAction = UIAlertAction(title: "确定", style: .cancel, handler: nil)
        alertController.addAction(cancelAction)
        self.parentViewCtr()?.present(alertController, animated: true, completion: nil)
    }


    func parentViewCtr() -> UIViewController? {
        
        let n = next
        
        while n != nil {
            
            let controller = next?.next
            
            if (controller is UIViewController) {
                
                return controller as? UIViewController
            }
        }
        
        return nil
    }
}

使用步骤

import UIKit

class ViewController:UIViewController{
    
    //所以地址数据集合
    fileprivate var addressArr = [[String: AnyObject]]()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        //初始化数据
        let path = Bundle.main.path(forResource: "address", ofType:"plist")
        self.addressArr = NSArray(contentsOfFile: path!) as! Array
        
        let pickV = SdLinkagePickViewCtr(frame: CGRect(x: 0, y: 50, width: self.view.frame.width, height: 300), addressArr: self.addressArr as NSArray, rowC: 3)
        self.view.addSubview(pickV)
    }
}