Swift － 滚动选择器

import UIKit
class LHHScrollView: UIView,UIScrollViewDelegate {
    /*
    // Only override drawRect: if you perform custom drawing.
    // An empty implementation adversely affects performance during animation.
    override func drawRect(rect: CGRect) {
        // Drawing code
    }
    */
    var mainScrollView = UIScrollView()
    var titleScrollView = UIScrollView()
    var underLine = UIView()
    var _titleArray = NSArray()
    var selectIndex:Int = 0
    var cacheVC = CacheVC()
    /**
     初始化方法
     - parameter frame:      定义大小范围
     - parameter titleArray: 盛放标题
     - returns: nil
     */
    init(frame:CGRect , titleArray:NSArray) {
        super.init(frame: frame)
        _titleArray = NSArray(array: titleArray)
        self.backgroundColor = UIColor.lightGrayColor()
        self.creatMainScrollView(UIColor.whiteColor())
        self.creatSelectBtn(_titleArray)
        self.creatTitleScrollView()
    }
    /**
     创建滚动界面
     - parameter bgColor: 设置滚动条背景颜色
     */
    func creatMainScrollView(bgColor:UIColor) {
        mainScrollView.bounces = false
        mainScrollView.frame = CGRectMake(0, 0, WIDTH, HEIGHT-64)
        mainScrollView.showsHorizontalScrollIndicator = false
        mainScrollView.showsVerticalScrollIndicator = false
        mainScrollView.userInteractionEnabled = true
        mainScrollView.delegate = self
        mainScrollView.contentSize = CGSizeMake(CGFloat(_titleArray.count) * WIDTH, HEIGHT - 64)
        mainScrollView.scrollEnabled = true
        mainScrollView.pagingEnabled = true
        mainScrollView.backgroundColor = bgColor
        self.addSubview(mainScrollView)
    }
    /**
     创建滚动条
     */
    func creatTitleScrollView() {
        titleScrollView.bounces = false
        titleScrollView.frame = CGRectMake(0, 0, WIDTH, 42)
        titleScrollView.showsHorizontalScrollIndicator = false
        titleScrollView.showsVerticalScrollIndicator = false
        titleScrollView.userInteractionEnabled = true
        titleScrollView.scrollEnabled = true
        titleScrollView.backgroundColor = UIColor.lightGrayColor()
        self.addSubview(titleScrollView)
    }
    /**
     创建按钮和滑块
     - parameter titleArray: 标题
     */
    func creatSelectBtn(titleArray:NSArray) {
        var width:CGFloat
        width = 5
        for index in 0..<titleArray.count {
            let btn = UIButton(type: .Custom)
            btn.frame = CGRectMake(width, 5, self.legthOfTitle(titleArray[index] as! String)+30, 30)
            btn.setTitle(titleArray[index] as? String, forState: .Normal)
            btn.backgroundColor = UIColor.orangeColor()
            btn.layer.cornerRadius = 5
            btn.titleLabel?.font = UIFont.systemFontOfSize(13)
            btn.setTitleColor(UIColor.blackColor(), forState: .Normal)
            btn.tag = index + 50
            btn.addTarget(self, action: #selector(self.btnAction(_:)), forControlEvents: .TouchUpInside)
            titleScrollView.addSubview(btn)
            let lineView = UIView(frame: CGRectMake(width, 39, self.legthOfTitle(titleArray[index] as! String)+30, 3))
            lineView.tag = 100 + index
            lineView.backgroundColor = UIColor.redColor()
            lineView.hidden = true
            titleScrollView.addSubview(lineView)
            if index == 50 {
                btn.setTitleColor(UIColor.redColor(), forState: .Selected)
            }
            if lineView.tag == 100 {
                lineView.hidden = false
            }
            width = width + self.legthOfTitle(titleArray[index] as! String) + 30 + 10
        }
        selectIndex = 0
        titleScrollView.contentSize = CGSizeMake(width, 40)
        let viewVC = ViewController1()
        viewVC.contentID =  "\(selectIndex)"
        viewVC.view.frame = CGRectMake(0, 42, WIDTH, HEIGHT - 64 - 42)
        mainScrollView.addSubview(viewVC.view)
        cacheVC.addCurrentVC(selectIndex)
    }
    /**
     计算标题长度
     - parameter title: 标题
     - returns: 标题长度
     */
    func legthOfTitle(title:String) -> CGFloat {
        let attribute = [NSFontAttributeName: UIFont.systemFontOfSize(13)]
//        let text: NSString = NSString(CString: title.cStringUsingEncoding(NSUTF8StringEncoding)!,encoding: NSUTF8StringEncoding)!
        let option = NSStringDrawingOptions.UsesFontLeading
        let size = title.boundingRectWithSize(CGSizeMake(1000, 30), options:option, attributes: attribute, context: nil)
        return size.width
    }
    /**
     点击按钮响应事件
     - parameter btn: tag从50开始
     */
    func btnAction(btn:UIButton) {
        selectIndex = btn.tag - 50
        print(btn.tag)
        /// 按钮和滑块变为未选状态
        for var view in titleScrollView.subviews {
            if view.isKindOfClass(UIButton) {
                let button = view as! UIButton
                button.setTitleColor(UIColor.blackColor(), forState: .Normal)
            }
            else
            {
                let lineView = view 
                lineView.hidden = true
            }
        }
        btn.setTitleColor(UIColor.redColor(), forState: .Normal)
        let lineView = titleScrollView.viewWithTag(selectIndex + 100)
        lineView?.hidden = false
        /**
         *  对当前控制器做处理
         *
         *  @param selectIndex 当前控制器标号
         *
         *  @return nil
         */
        if !cacheVC.vcArray.containsObject(selectIndex) {
            let viewVC = ViewController1()
            viewVC.contentID =  "\(selectIndex)"
            viewVC.view.frame = CGRectMake(CGFloat(selectIndex) * WIDTH, 42, WIDTH, HEIGHT - 64 - 42)
            mainScrollView.addSubview(viewVC.view)
            cacheVC.addCurrentVC(selectIndex)
        }
        /**
         *  设置偏移量
         *
         *  @param self.selectIndex 当前位置
         *
         *  @return nil
         */
        self.mainScrollView.contentOffset = CGPointMake(CGFloat(self.selectIndex) * WIDTH, 0)
        /// 获取当前按钮左右按钮
        let btnRight = titleScrollView.viewWithTag(btn.tag + 1)
        let btnLeft = titleScrollView.viewWithTag(btn.tag - 1)
        /**
         *  通过位置判断偏移量
         */
        if (btn.frame.origin.x + btn.frame.size.width) > WIDTH - 10{
            if btn.tag - 50 != _titleArray.count - 1 {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(btnRight!.frame.origin.x + btnRight!.frame.size.width - WIDTH + 10, 0);
                })
            }
        }
        else
        {
            if selectIndex == 0 {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(0, 0)
                })
            }
            else
            {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(btnLeft!.frame.origin.x, 0)
                })
            }
        }
    }
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    /**
     scrollView代理方法
     - parameter scrollView: 滑动界面相应事件
     */
    func scrollViewDidEndDecelerating(scrollView: UIScrollView) {
        selectIndex = Int(mainScrollView.contentOffset.x / WIDTH)
        /// 按钮和滑块变为未选状态
        for var view in titleScrollView.subviews {
            if view.isKindOfClass(UIButton) {
                let button = view as! UIButton
                button.setTitleColor(UIColor.blackColor(), forState: .Normal)
            }
            else
            {
                let lineView = view
                lineView.hidden = true
            }
        }
        let btn = titleScrollView.viewWithTag(selectIndex + 50) as! UIButton
        btn.setTitleColor(UIColor.redColor(), forState: .Normal)
        let lineView = titleScrollView.viewWithTag(selectIndex + 100)
        lineView?.hidden = false
        /**
         *  对当前控制器做处理
         *
         *  @param selectIndex 当前控制器标号
         *
         *  @return nil
         */
        if !cacheVC.vcArray.containsObject(selectIndex) {
            let viewVC = ViewController1()
            viewVC.contentID =  "\(selectIndex)"
            viewVC.view.frame = CGRectMake(CGFloat(selectIndex) * WIDTH, 42, WIDTH, HEIGHT - 64 - 42)
            mainScrollView.addSubview(viewVC.view)
            cacheVC.addCurrentVC(selectIndex)
        }
        /**
         *  设置偏移量
         *
         *  @param self.selectIndex 当前位置
         *
         *  @return nil
         */
        self.mainScrollView.contentOffset = CGPointMake(CGFloat(selectIndex) * WIDTH, 0)
        /// 获取当前按钮左右按钮
        let btnRight = titleScrollView.viewWithTag(selectIndex + 50 + 1)
        let btnLeft = titleScrollView.viewWithTag(selectIndex + 50 - 1)
        /**
         *  通过位置判断偏移量
         */
        if (btn.frame.origin.x + btn.frame.size.width) > WIDTH - 10{
            if btn.tag - 50  != _titleArray.count - 1 {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(btnRight!.frame.origin.x + btnRight!.frame.size.width - WIDTH + 10, 0);
                })
            }
        }
        else
        {
            if selectIndex == 0 {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(0, 0)
                })
            }
            else
            {
                UIView.animateWithDuration(0.3, animations: {
                    self.titleScrollView.contentOffset = CGPointMake(btnLeft!.frame.origin.x, 0)
                })
            }
        }
    }
}