swift實現自定義datePicker，只查詢年份和月份

//

//  SDDatePickerView.swift

//  OA

//

//  Created by HelloMac on 16/6/2.

//  Copyright © 2016年 HelloMac. All rights reserved.

//

import UIKit

class SDDatePickerView: UIPickerView,UIPickerViewDelegate,UIPickerViewDataSource {

    var todayIndexPath:NSIndexPath?

    var months:NSArray?

    var years:NSArray?

    var bigRowCount = 1000

    var minYear = 2008

    var maxYear = 2030

    var rowHeight = 44

    var componentNumber:Int = 2

    var MONTH:Int = 0

    var YEAR:Int = 1

    var LABEL_TAG:Int = 43

    //var numberOfComponents: Int

    override func awakeFromNib() {

        super.awakeFromNib()

        self.months = self.nameOfMonths()

        self.years = self.nameOfYears()

        self.todayIndexPath = self.todayPath()

        //設置代理

        self.delegate = self

        self.dataSource = self

        self.selectToday()

    }

    func date() -> NSDate {

        let monthCount:Int = (self.months?.count)!

        let month = self.months?.objectAtIndex((self.selectedRowInComponent(MONTH)) % monthCount) as! String

        let yearCount:Int = (self.years?.count)!

        let year = self.years?.objectAtIndex((self.selectedRowInComponent(YEAR)) % yearCount) as! String

        let str = "\(year)-\(month)"

        let formatter = NSDateFormatter()

        formatter.dateFormat = "yyyy-MMMM"

        let date = formatter.dateFromString(str)

        return date!

    }

    /**

        建立年和月的數據源

     */

    func nameOfMonths() -> NSArray {

        let dateFormatter = NSDateFormatter()

        return dateFormatter.standaloneMonthSymbols

    }

    func nameOfYears() -> NSArray {

        let years = NSMutableArray()

        for year in self.minYear..<self.maxYear {

            let yeatStr = "\(year)"

            years.addObject(yeatStr)

        }

        return years

    }

    /**

        代理方法

    */

    func pickerView(pickerView: UIPickerView, widthForComponent component: Int) -> CGFloat {

        return self.componentWidth()

    }

    func pickerView(pickerView: UIPickerView, viewForRow row: Int, forComponent component: Int, reusingView view: UIView?) -> UIView {

        var selected:Bool = false

        if component == MONTH{

            let monthCount:Int = (self.months?.count)!

            let monthName:String = self.months?.objectAtIndex(row % monthCount) as! String

            let currentMonthName:String = self.currentMonthName()

            if monthName == currentMonthName {

                selected = true

            }

        }else{

            let yearCount:Int = (self.years?.count)!

            let yearName:String = self.years?.objectAtIndex(row % yearCount) as! String

            let currentYearName:String = self.currentYearName()

            if yearName == currentYearName {

                selected = true

            }

        }

        var returnView:UILabel?

        if view?.tag == LABEL_TAG {

            returnView = view as? UILabel

        }else{

            returnView = self.labelForComponent(component, Selected: selected)

        }

        returnView?.textColor = selected ? UIColor.blueColor() : UIColor.blackColor()

        returnView?.text = self.titleForRow(row, forComponent: component)

        return returnView!

    }

    func pickerView(pickerView: UIPickerView, rowHeightForComponent component: Int) -> CGFloat {

        return (CGFloat)(rowHeight)

    }

    func numberOfComponentsInPickerView(pickerView: UIPickerView) -> Int {

        return componentNumber

    }

    func pickerView(pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {

        if component == MONTH {

            return self.bigRowMonthCount()

        }

        return self.bigRowYearCount()

    }

    func selectToday() {

        self.selectRow(self.todayPath().row, inComponent: MONTH, animated: false)

        self.selectRow(self.todayPath().section, inComponent: YEAR, animated: false)

    }

    func todayPath() -> NSIndexPath {

        //row ==month  section ==year

        var row:Int = 0

        var section:Int = 0

        let currentmonth = self.currentMonthName()

        let currentYear = self.currentYearName()

        let count:Int = (self.months?.count)!

        for month in 0..<count{

            let cellMonth = months![month] as! String

            if cellMonth == currentmonth {

                let currentRow:Int = (self.months?.indexOfObject(cellMonth))!

                row = currentRow

                //let rowCount:Int = self.bigRowMonthCount()/2

                row = row + self.bigRowMonthCount()/2

                break

            }

        }

        let Ycount:Int = (self.years?.count)!

        for year in 0..<Ycount{

            let cellYear = years![year] as! String

            if cellYear == currentYear {

                let currentRow:Int = (self.years?.indexOfObject(cellYear))!

                section = currentRow

               // let rowCount:Int = self.bigRowYearCount()/2

                section = section + self.bigRowYearCount()/2

                break

            }

        }

        return NSIndexPath.init(forRow: row, inSection: section)

    }

    func bigRowMonthCount() -> NSInteger {

        return (self.months?.count)! * bigRowCount

    }

    func bigRowYearCount() -> NSInteger {

        return (self.years?.count)! * bigRowCount

    }

    func componentWidth() -> CGFloat {

        let numberOfComponent:CGFloat = (CGFloat)(self.componentNumber)

        return self.bounds.size.width / numberOfComponent

    }

    func titleForRow(row:Int ,forComponent component:Int) -> String {

        if component == MONTH {

            let monthCount:Int = (self.months?.count)!

            return (self.months?.objectAtIndex(row % monthCount))! as! String

        }

        let  yearCount:Int = (self.years?.count)!

        return (self.years?.objectAtIndex(row % yearCount))! as! String

    }

    func labelForComponent(component:Int , Selected selected:Bool) -> UILabel {

        let frame = CGRectMake(0, 0, self.componentWidth(), (CGFloat)(rowHeight))

        let label = UILabel.init(frame: frame)

        label.textAlignment = NSTextAlignment.Center

        label.backgroundColor = UIColor.clearColor()

        label.textColor = selected ? UIColor.blueColor() : UIColor.blackColor()

        label.font = UIFont.boldSystemFontOfSize(18)

        label.userInteractionEnabled = false

        label.tag = LABEL_TAG

        return label

    }

    func currentMonthName() -> String {

        let date = NSDate()

        let formatter = NSDateFormatter()

        formatter.dateFormat = "MMMM"

        return formatter.stringFromDate(date)

    }

    func currentYearName() -> String {

        let date = NSDate()

        let formatter = NSDateFormatter()

        formatter.dateFormat = "yyyy"

        return formatter.stringFromDate(date)

    }

}