From ac2e8b095b96056a2fa50854fd36c123424bf886 Mon Sep 17 00:00:00 2001
From: Daniel Vancura <daniel_vancura@me.com>
Date: Sun, 27 Dec 2015 14:40:25 +0100
Subject: [PATCH] Replaced PIDatePicker with an extension for UIDatePicker and
 changed the example accordingly.

---
 Example/PIDatePicker/Main.storyboard        |  32 +-
 Example/PIDatePicker/PIViewController.swift |  33 +-
 Pod/Classes/PIDatePicker.swift              | 642 +++++---------------
 Pod/Classes/PIDatePickerComponents.swift    |  14 -
 Pod/Classes/PIDatePickerDelegate.swift      |  13 -
 5 files changed, 169 insertions(+), 565 deletions(-)
 delete mode 100644 Pod/Classes/PIDatePickerComponents.swift
 delete mode 100644 Pod/Classes/PIDatePickerDelegate.swift

diff --git a/Example/PIDatePicker/Main.storyboard b/Example/PIDatePicker/Main.storyboard
index 0a4bc0b..36748a4 100644
--- a/Example/PIDatePicker/Main.storyboard
+++ b/Example/PIDatePicker/Main.storyboard
@@ -1,8 +1,8 @@
 <?xml version="1.0" encoding="UTF-8" standalone="no"?>
-<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="7531" systemVersion="14D131" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" initialViewController="42N-tQ-fvD">
+<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="9531" systemVersion="15D9c" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" initialViewController="42N-tQ-fvD">
     <dependencies>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="7520"/>
-        <capability name="Constraints to layout margins" minToolsVersion="6.0"/>
+        <deployment identifier="iOS"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="9529"/>
     </dependencies>
     <scenes>
         <!--View Controller-->
@@ -17,13 +17,6 @@
                         <rect key="frame" x="0.0" y="0.0" width="600" height="600"/>
                         <autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
                         <subviews>
-                            <view contentMode="scaleToFill" translatesAutoresizingMaskIntoConstraints="NO" id="XUL-K8-ur5" customClass="PIDatePicker" customModule="PIDatePicker">
-                                <rect key="frame" x="0.0" y="0.0" width="600" height="212"/>
-                                <color key="backgroundColor" red="0.40000000596046448" green="0.80000001192092896" blue="1" alpha="0.5" colorSpace="calibratedRGB"/>
-                                <constraints>
-                                    <constraint firstAttribute="height" constant="212" placeholder="YES" id="EaD-3N-3zS"/>
-                                </constraints>
-                            </view>
                             <button opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="roundedRect" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="xh8-4R-V3E">
                                 <rect key="frame" x="141" y="275" width="155" height="50"/>
                                 <constraints>
@@ -60,15 +53,24 @@
                                 <color key="textColor" cocoaTouchSystemColor="darkTextColor"/>
                                 <nil key="highlightedColor"/>
                             </label>
+                            <datePicker contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" datePickerMode="dateAndTime" minuteInterval="1" translatesAutoresizingMaskIntoConstraints="NO" id="rb7-VE-DC0">
+                                <rect key="frame" x="0.0" y="28" width="600" height="216"/>
+                                <date key="date" timeIntervalSinceReferenceDate="472915293.63234299">
+                                    <!--2015-12-27 13:21:33 +0000-->
+                                </date>
+                                <connections>
+                                    <action selector="dateChanged:" destination="42N-tQ-fvD" eventType="valueChanged" id="rFf-zV-rP5"/>
+                                </connections>
+                            </datePicker>
                         </subviews>
                         <color key="backgroundColor" red="0.80000001192092896" green="0.80000001192092896" blue="0.80000001192092896" alpha="1" colorSpace="calibratedRGB"/>
                         <constraints>
+                            <constraint firstItem="rb7-VE-DC0" firstAttribute="top" secondItem="ZxE-uC-JeZ" secondAttribute="bottom" constant="8" symbolic="YES" id="0Ff-JN-Svw"/>
+                            <constraint firstItem="rb7-VE-DC0" firstAttribute="leading" secondItem="r7S-n0-yCC" secondAttribute="leading" id="6x8-Ob-Q91"/>
                             <constraint firstAttribute="centerX" secondItem="xh8-4R-V3E" secondAttribute="centerX" constant="146.5" id="8Ap-pJ-ZRh"/>
-                            <constraint firstItem="XUL-K8-ur5" firstAttribute="top" secondItem="ZxE-uC-JeZ" secondAttribute="bottom" constant="-20" id="G03-nO-TrM"/>
+                            <constraint firstItem="G44-H8-jKp" firstAttribute="top" secondItem="rb7-VE-DC0" secondAttribute="bottom" constant="31" id="LjU-3l-DO8"/>
+                            <constraint firstItem="rb7-VE-DC0" firstAttribute="centerX" secondItem="AKU-dB-Qgb" secondAttribute="centerX" id="bXz-YF-Muz"/>
                             <constraint firstItem="AKU-dB-Qgb" firstAttribute="top" secondItem="G44-H8-jKp" secondAttribute="bottom" constant="20" id="eM5-Zn-8zX"/>
-                            <constraint firstItem="xh8-4R-V3E" firstAttribute="top" secondItem="XUL-K8-ur5" secondAttribute="bottom" constant="63" id="f8P-7u-8fk"/>
-                            <constraint firstItem="XUL-K8-ur5" firstAttribute="leading" secondItem="r7S-n0-yCC" secondAttribute="leadingMargin" constant="-16" id="hbh-Xe-MSm"/>
-                            <constraint firstAttribute="trailingMargin" secondItem="XUL-K8-ur5" secondAttribute="trailing" constant="-16" id="jBj-W6-2Jc"/>
                             <constraint firstItem="G44-H8-jKp" firstAttribute="centerY" secondItem="xh8-4R-V3E" secondAttribute="centerY" id="k3k-ce-ElU"/>
                             <constraint firstAttribute="centerX" secondItem="AKU-dB-Qgb" secondAttribute="centerX" constant="-0.5" id="yYb-l8-afJ"/>
                             <constraint firstAttribute="centerX" secondItem="xh8-4R-V3E" secondAttribute="centerX" constant="81.5" id="yoe-we-2cb"/>
@@ -81,7 +83,7 @@
                         </variation>
                     </view>
                     <connections>
-                        <outlet property="datePicker" destination="XUL-K8-ur5" id="SMl-I9-g5s"/>
+                        <outlet property="datePicker" destination="rb7-VE-DC0" id="jHd-pq-z2F"/>
                         <outlet property="label" destination="AKU-dB-Qgb" id="PIz-gG-8HH"/>
                     </connections>
                 </viewController>
diff --git a/Example/PIDatePicker/PIViewController.swift b/Example/PIDatePicker/PIViewController.swift
index 994a91a..b80c6d8 100644
--- a/Example/PIDatePicker/PIViewController.swift
+++ b/Example/PIDatePicker/PIViewController.swift
@@ -11,16 +11,20 @@ import PIDatePicker
 
 class PIViewController : UIViewController {
     
-    @IBOutlet weak var datePicker: PIDatePicker!
+    @IBOutlet weak var datePicker: UIDatePicker!
     @IBOutlet weak var label: UILabel!
     
-    let validPast: NSTimeInterval = -10000000000
-    
     override func viewWillAppear(animated: Bool) {
         super.viewWillAppear(animated)
         
-        self.datePicker.minimumDate = NSDate().dateByAddingTimeInterval(validPast)
-        self.datePicker.delegate = self
+        self.datePicker.minimumDate = NSDate.distantPast()
+        self.datePicker.maximumDate = NSDate.distantFuture()
+    }
+    
+    @IBAction func dateChanged(sender: UIDatePicker) {
+        let dateFormatter = NSDateFormatter()
+        dateFormatter.dateStyle = .ShortStyle
+        self.label.text = dateFormatter.stringFromDate(datePicker.date)
     }
 
     @IBAction func randomizeColor(sender: AnyObject) {
@@ -30,23 +34,18 @@ class PIViewController : UIViewController {
         let blue = CGFloat(arc4random_uniform(255))
         
         self.datePicker.textColor = UIColor(red: red/255, green: green/255, blue: blue/255, alpha: 1)
-        self.datePicker.reloadAllComponents()
     }
     
     @IBAction func randomizeFont(sender: AnyObject) {
         let familyNames = UIFont.familyNames()
         let randomNumber = Int(arc4random_uniform(UInt32(familyNames.count)))
         let familyName: String = familyNames[randomNumber]
-        let fontName: String = UIFont.fontNamesForFamilyName(familyName)[0]
-        self.datePicker.font = UIFont(name: fontName, size: 14)!
-        self.datePicker.reloadAllComponents()
-    }
-}
-
-extension PIViewController: PIDatePickerDelegate {
-    func pickerView(pickerView: PIDatePicker, didSelectRow row: Int, inComponent component: Int) {
-        let dateFormatter = NSDateFormatter()
-        dateFormatter.dateStyle = .ShortStyle
-        self.label.text = dateFormatter.stringFromDate(pickerView.date)
+        guard let fontName: String = UIFont.fontNamesForFamilyName(familyName).first else {
+            return
+        }
+        guard let font = UIFont(name: fontName, size: 14) else {
+            return
+        }
+        self.datePicker.font = font
     }
 }
diff --git a/Pod/Classes/PIDatePicker.swift b/Pod/Classes/PIDatePicker.swift
index cd4fd80..fe2e022 100644
--- a/Pod/Classes/PIDatePicker.swift
+++ b/Pod/Classes/PIDatePicker.swift
@@ -1,557 +1,187 @@
 //
 //  PIDatePicker.swift
-//  Pods
-//
-//  Created by Christopher Jones on 3/30/15.
+//  PIDatePicker
 //
+//  Created by Daniel Vancura on 26/12/15.
+//  Copyright © 2015 Daniel Vancura. All rights reserved.
 //
 
 import UIKit
 
-public class PIDatePicker: UIControl, UIPickerViewDataSource, UIPickerViewDelegate {
-
-    // MARK: -
-    // MARK: Public Properties
-    public var delegate: PIDatePickerDelegate?
-    
-    /// The font for the date picker.
-    public var font = UIFont.systemFontOfSize(15.0)
-
-    /// The text color for the date picker components.
-    public var textColor = UIColor.blackColor()
+// MARK: - Static variables
+private var fontKey = "UIDatePicker.font"
+private var textColorKey = "UIDatePicker.textColor"
+private var onceToken: dispatch_once_t = 0
 
-    /// The minimum date to show for the date picker. Set to NSDate.distantPast() by default
-    public var minimumDate = NSDate.distantPast() {
-        didSet {
-            self.validateMinimumAndMaximumDate()
-        }
-    }
+// MARK: - Helper functions
 
-    /// The maximum date to show for the date picker. Set to NSDate.distantFuture() by default
-    public var maximumDate = NSDate.distantFuture() {
-        didSet {
-            self.validateMinimumAndMaximumDate()
-        }
-    }
+/**
+Helper function that replaces a given method in a certain class with the implementation of another method in another class.
+*/
+private func replaceMethod(destMethod: Selector, ofClass destClass: AnyClass, withMethod sourceMethod: Selector, ofClass sourceClass: AnyClass) {
+    let src = class_getInstanceMethod(sourceClass, sourceMethod)
+    let dst = class_getInstanceMethod(destClass, destMethod)
+    
+    method_exchangeImplementations(src, dst)
+}
 
-     /// The current locale to use for formatting the date picker. By default, set to the device's current locale
-    public var locale : NSLocale = NSLocale.currentLocale() {
-        didSet {
-            self.calendar.locale = self.locale
-        }
+/**
+ Function to determine, if a view is a subview of a date picker.
+ 
+ - parameter view: Any view in the view hierarchy.
+ - returns: A date picker within which the current label is contained.
+ */
+func datePicker(view: UIView) -> UIDatePicker? {
+    if view.superview == nil {
+        return nil
+    } else if let datePicker = view.superview as? UIDatePicker {
+        return datePicker
+    } else {
+        return datePicker(view.superview!)
     }
+}
 
-    /// The current date value of the date picker.
-    public private(set) var date = NSDate()
-
-    // MARK: -
-    // MARK: Private Variables
-    
-    private let maximumNumberOfRows = Int(INT16_MAX)
-    
-     /// The internal picker view used for laying out the date components.
-    private let pickerView = UIPickerView()
-
-     /// The calendar used for formatting dates.
-    private var calendar = NSCalendar(calendarIdentifier: NSCalendarIdentifierGregorian)!
+// MARK: - Extension for UIDatePicker to provide additional properties
 
-     /// Calculates the current calendar components for the current date.
-    private var currentCalendarComponents : NSDateComponents {
+public extension UIDatePicker {
+    /**
+     The font name for text in the date picker.
+     
+     - note: This variable is tied to the 'font' property. Changes to one of them also changes the other.
+     */
+    @IBInspectable
+    public var fontName: String {
         get {
-            return self.calendar.components([.Year, .Month, .Day], fromDate: self.date)
+            return self.font.fontName
         }
-    }
-    
-     /// Gets the text color to be used for the label in a disabled state
-    private var disabledTextColor : UIColor {
-        get {
-            var r : CGFloat = 0
-            var g : CGFloat = 0
-            var b : CGFloat = 0
-            
-            self.textColor.getRed(&r, green: &g, blue: &b, alpha: nil)
-            
-            return UIColor(red: r, green: g, blue: b, alpha: 0.35)
+        set {
+            if let font = UIFont(name: newValue, size: CGFloat(self.fontSize)) {
+                self.font = font
+                self.setNeedsDisplay()
+            }
         }
     }
-
-     /// The order in which each component should be ordered in.
-    private var datePickerComponentOrdering = [PIDatePickerComponents]()
-
-    // MARK: -
-    // MARK: LifeCycle
-
-    required public init?(coder aDecoder: NSCoder) {
-        super.init(coder: aDecoder)
-        self.commonInit()
-    }
-    
-    override public init(frame: CGRect) {
-        super.init(frame: frame)
-        self.commonInit()
-    }
-
-    /**
-    Handles the common initialization amongst all init()
-    */
-    func commonInit() {
-        self.translatesAutoresizingMaskIntoConstraints = false
-
-        self.pickerView.dataSource = self
-        self.pickerView.delegate = self
-        
-        self.addSubview(self.pickerView)
-        
-        let topConstraint = NSLayoutConstraint(item: self.pickerView, attribute: .Top, relatedBy: .Equal, toItem: self, attribute: .Top, multiplier: 1, constant: 0)
-        let bottomConstraint = NSLayoutConstraint(item: self.pickerView, attribute: .Bottom, relatedBy: .Equal, toItem: self, attribute: .Bottom, multiplier: 1, constant: 0)
-        let leftConstraint = NSLayoutConstraint(item: self.pickerView, attribute: .Leading, relatedBy: .Equal, toItem: self, attribute: .Leading, multiplier: 1, constant: 0)
-        let rightConstraint = NSLayoutConstraint(item: self.pickerView, attribute: .Trailing, relatedBy: .Equal, toItem: self, attribute: .Trailing, multiplier: 1, constant: 0)
-        
-        self.addConstraints([topConstraint, bottomConstraint, leftConstraint, rightConstraint])
-    }
-    
-    // MARK: -
-    // MARK: Override
-    public override func intrinsicContentSize() -> CGSize {
-        return self.pickerView.intrinsicContentSize()
-    }
     
-    public override func willMoveToSuperview(newSuperview: UIView?) {
-        super.willMoveToSuperview(newSuperview)
-        self.reloadAllComponents()
-        
-        self.setDate(self.date)
-    }
-
-    // MARK: - 
-    // MARK: Public
-
     /**
-        Reloads all of the components in the date picker.
-    */
-    public func reloadAllComponents() {
-        self.refreshComponentOrdering()
-        self.pickerView.reloadAllComponents()
-    }
-    
-    /**
-    Sets the current date value for the date picker.
-    
-    :param: date     The date to set the picker to.
-    :param: animated True if the date picker should changed with an animation; otherwise false,
-    */
-    public func setDate(date : NSDate, animated : Bool) {
-        self.date = date
-        self.updatePickerViewComponentValuesAnimated(animated)
-    }
-
-    // MARK: -
-    // MARK: Private
-    
-    /**
-    Sets the current date with no animation.
-    
-    :param: date The date to be set.
-    */
-    private func setDate(date : NSDate) {
-        self.setDate(date, animated: false)
-    }
-
-    /**
-        Creates a new date formatter with the locale and calendar
-    
-        :returns: A new instance of NSDateFormatter
-    */
-    private func dateFormatter() -> NSDateFormatter {
-        let dateFormatter = NSDateFormatter()
-        dateFormatter.calendar = self.calendar
-        dateFormatter.locale = self.locale
-        
-        return dateFormatter
-    }
-
-    /**
-        Refreshes the ordering of components based on the current locale. Calling this function will not refresh the picker view.
-    */
-    private func refreshComponentOrdering() {
-        var componentOrdering = NSDateFormatter.dateFormatFromTemplate("yMMMMd", options: 0, locale: self.locale)!
-
-        let firstComponentOrderingString = componentOrdering[componentOrdering.startIndex.advancedBy(0)]
-        let lastComponentOrderingString = componentOrdering[componentOrdering.startIndex.advancedBy(componentOrdering.characters.count - 1)]
-
-        let characterSet = NSMutableCharacterSet()
-        
-        characterSet.formUnionWithCharacterSet(NSCharacterSet(charactersInString: String(firstComponentOrderingString) +
-            String(lastComponentOrderingString)))
-        characterSet.formUnionWithCharacterSet(NSCharacterSet.whitespaceAndNewlineCharacterSet())
-        characterSet.formUnionWithCharacterSet(NSCharacterSet.punctuationCharacterSet())
-        
-        componentOrdering = componentOrdering.stringByTrimmingCharactersInSet(characterSet)
-
-        let remainingValue = componentOrdering[componentOrdering.startIndex.advancedBy(0)]
-
-        let firstComponent = PIDatePickerComponents(rawValue: firstComponentOrderingString)!
-        let secondComponent = PIDatePickerComponents(rawValue: remainingValue)!
-        let lastComponent = PIDatePickerComponents(rawValue: lastComponentOrderingString)!
-
-        self.datePickerComponentOrdering = [firstComponent, secondComponent, lastComponent]
-    }
-    
-    /**
-    Validates that the set minimum and maximum dates are valid.
-    */
-    private func validateMinimumAndMaximumDate() {
-        let ordering = self.minimumDate.compare(self.maximumDate)
-        if (ordering != .OrderedAscending ){
-            fatalError("Cannot set a maximum date that is equal or less than the minimum date.")
+     The font size for text in the date picker.
+     
+     - note: This variable is tied to the 'font' property. Changes to one of them also changes the other.
+     */
+    @IBInspectable
+    public var fontSize: Double {
+        get {
+            return Double(self.font.pointSize)
         }
-    }
-
-    /**
-        Gets the value of the current component at the specified row.
-    
-    :param: row            The row index whose value is required
-    :param: componentIndex The component index for the row.
-    
-    :returns: A string containing the value of the current row at the component index.
-    */
-    private func titleForRow(row : Int, inComponentIndex componentIndex: Int) -> String {
-        let dateComponent = self.componentAtIndex(componentIndex)
-
-        let value = self.rawValueForRow(row, inComponent: dateComponent)
-
-        if dateComponent == PIDatePickerComponents.Month {
-            let dateFormatter = self.dateFormatter()
-            return dateFormatter.monthSymbols[value - 1]
-        } else {
-            return String(value)
+        set {
+            self.font = self.font.fontWithSize(CGFloat(newValue))
+            self.setNeedsDisplay()
         }
     }
     
     /**
-    Gets the value of the input component using the current date.
-    
-    :param: component The component whose value is needed.
-    
-    :returns: The value of the component.
-    */
-    private func valueForDateComponent(component : PIDatePickerComponents) -> Int{
-        if component == .Year {
-            return self.currentCalendarComponents.year
-        } else if component == .Day {
-            return self.currentCalendarComponents.day
-        } else {
-            return self.currentCalendarComponents.month
+     The font that is used in the date picker. (Unfortunately, @Inspectable doesn't work with fonts - use fontName and fontSize instead)
+     
+     - note: This variable is tied to the 'fontSize' and 'fontName' properties. Changes to one of them also changes the other.
+     */
+    public var font: UIFont {
+        get {
+            if let font = objc_getAssociatedObject(self, &fontKey) as? UIFont {
+                return font
+            } else {
+                return UIFont(name: "Helvetica", size: 20.0)!
+            }
         }
-    }
-    
-    /**
-    Gets the maximum range for the specified date picker component.
-    
-    :param: component The component to get the range for.
-    
-    :returns: The maximum date range for that component.
-    */
-    private func maximumRangeForComponent(component : PIDatePickerComponents) -> NSRange {
-        var calendarUnit : NSCalendarUnit
-        if component == .Year {
-            calendarUnit = .Year
-        } else if component == .Day {
-            calendarUnit = .Day
-        } else {
-            calendarUnit = .Month
+        set {
+            objc_setAssociatedObject(self, &fontKey, newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_COPY)
+            self.subviews.filter({$0 is UIPickerView}).forEach({($0 as? UIPickerView)?.reloadAllComponents()})
+            self.setNeedsDisplay()
         }
-        
-        return self.calendar.maximumRangeOfUnit(calendarUnit)
-    }
-    
-    /**
-    Calculates the raw value of the row at the current index.
-    
-    :param: row       The row to get.
-    :param: component The component which the row belongs to.
-    
-    :returns: The raw value of the row, in integer. Use NSDateComponents to convert to a usable date object.
-    */
-    private func rawValueForRow(row : Int, inComponent component : PIDatePickerComponents) -> Int {
-        let calendarUnitRange = self.maximumRangeForComponent(component)
-        return calendarUnitRange.location + (row % calendarUnitRange.length)
     }
     
     /**
-    Checks if the specified row should be enabled or not.
-    
-    :param: row       The row to check.
-    :param: component The component to check the row in.
-    
-    :returns: YES if the row should be enabled; otherwise NO.
-    */
-    private func isRowEnabled(row: Int, forComponent component : PIDatePickerComponents) -> Bool {
-        
-        let rawValue = self.rawValueForRow(row, inComponent: component)
-        
-        let components = NSDateComponents()
-        components.year = self.currentCalendarComponents.year
-        components.month = self.currentCalendarComponents.month
-        components.day = self.currentCalendarComponents.day
-        
-        if component == .Year {
-            components.year = rawValue
-        } else if component == .Day {
-            components.day = rawValue
-        } else if component == .Month {
-            components.month = rawValue
+     The color that is used for text in the date picker.
+     */
+    @IBInspectable
+    public var textColor: UIColor? {
+        get {
+            return objc_getAssociatedObject(self, &textColorKey) as? UIColor
         }
-        
-        let dateForRow = self.calendar.dateFromComponents(components)!
-        
-        return self.dateIsInRange(dateForRow)
-    }
-    
-    /**
-    Checks if the input date falls within the date picker's minimum and maximum date ranges.
-    
-    :param: date The date to be checked.
-    
-    :returns: True if the input date is within range of the minimum and maximum; otherwise false.
-    */
-    private func dateIsInRange(date : NSDate) -> Bool {
-        return self.minimumDate.compare(date) != NSComparisonResult.OrderedDescending &&
-            self.maximumDate.compare(date) != NSComparisonResult.OrderedAscending
-    }
-    
-    /**
-    Updates all of the date picker components to the value of the current date.
-    
-    :param: animated True if the update should be animated; otherwise false.
-    */
-    private func updatePickerViewComponentValuesAnimated(animated : Bool) {
-        for (_, dateComponent) in self.datePickerComponentOrdering.enumerate() {
-            self.setIndexOfComponent(dateComponent, animated: animated)
+        set {
+            objc_setAssociatedObject(self, &textColorKey, newValue, objc_AssociationPolicy.OBJC_ASSOCIATION_COPY)
+            self.subviews.filter({$0 is UIPickerView}).forEach({($0 as? UIPickerView)?.reloadAllComponents()})
+            self.setNeedsDisplay()
         }
     }
     
-    /**
-    Updates the index of the specified component to its relevant value in the current date.
-    
-    :param: component The component to be updated.
-    :param: animated  True if the update should be animated; otherwise false.
-    */
-    private func setIndexOfComponent(component : PIDatePickerComponents, animated: Bool) {
-        self.setIndexOfComponent(component, toValue: self.valueForDateComponent(component), animated: animated)
-    }
-    
-    /**
-    Updates the index of the specified component to the input value.
-    
-    :param: component The component to be updated.
-    :param: value     The value the component should be updated ot.
-    :param: animated  True if the update should be animated; otherwise false.
-    */
-    private func setIndexOfComponent(component : PIDatePickerComponents, toValue value : Int, animated: Bool) {
-        let componentRange = self.maximumRangeForComponent(component)
-        
-        let idx = (value - componentRange.location)
-        let middleIndex = (self.maximumNumberOfRows / 2) - (maximumNumberOfRows / 2) % componentRange.length + idx
-        
-        var componentIndex = 0
-        
-        for (index, dateComponent) in self.datePickerComponentOrdering.enumerate() {
-            if (dateComponent == component) {
-                componentIndex = index
-            }
+    // Override as "inspectable variable".
+    @IBInspectable
+    public override var backgroundColor: UIColor? {
+        get {
+            return super.backgroundColor
+        }
+        set {
+            super.backgroundColor = newValue
+            self.setNeedsDisplay()
         }
-        
-        self.pickerView.selectRow(middleIndex, inComponent: componentIndex, animated: animated)
     }
+}
 
-    /**
-        Gets the component type at the current component index.
-    
-    :param: index The component index
-    
-    :returns: The date picker component type at the index.
-    */
-    private func componentAtIndex(index: Int) -> PIDatePickerComponents {
-        return self.datePickerComponentOrdering[index]
-    }
-    
-    /**
-    Gets the number of days of the specified month in the specified year.
-    
-    :param: month The month whose maximum date value is requested.
-    :param: year  The year for which the maximum date value is required.
-    
-    :returns: The number of days in the month.
-    */
-    private func numberOfDaysForMonth(month : Int, inYear year : Int) -> Int {
-        let components = NSDateComponents()
-        components.month = month
-        components.day = 1
-        components.year = year
-        
-        let calendarRange = self.calendar.rangeOfUnit(.Day, inUnit: .Month, forDate: self.calendar.dateFromComponents(components)!)
-        let numberOfDaysInMonth = calendarRange.length
+// MARK: - Extension for UILabel to use text color and font as specified by the date picker (if there is one as a superview)
+
+public extension UILabel {
+    // Enables custom appearance for this view after loading this class.
+    public override class func initialize() {
         
-        return numberOfDaysInMonth
-    }
-    
-    /**
-    Determines if updating the specified component to the input value would evaluate to a valid date using the current date values.
-    
-    :param: value     The value to be updated to.
-    :param: component The component whose value should be updated.
-    
-    :returns: True if updating the component to the specified value would result in a valid date; otherwise false.
-    */
-    private func isValidValue(value : Int, forComponent component: PIDatePickerComponents) -> Bool {
-        if (component == .Year) {
-            let numberOfDaysInMonth = self.numberOfDaysForMonth(self.currentCalendarComponents.month, inYear: value)
-            return self.currentCalendarComponents.day <= numberOfDaysInMonth
-        } else if (component == .Day) {
-            let numberOfDaysInMonth = self.numberOfDaysForMonth(self.currentCalendarComponents.month, inYear: self.currentCalendarComponents.year)
-            return value <= numberOfDaysInMonth
-        } else if (component == .Month) {
-            let numberOfDaysInMonth = self.numberOfDaysForMonth(value, inYear: self.currentCalendarComponents.year)
-            return self.currentCalendarComponents.day <= numberOfDaysInMonth
-        }
+        super.initialize()
         
-        return true
+        // Replace the setters for text color and fonts with some that opt for a date picker's variables instead.
+        dispatch_once(&onceToken, {
+            replaceMethod(Selector("setTextColor:"), ofClass: UILabel.self, withMethod: Selector("pi_setTextColor:"), ofClass: UILabel.self)
+            replaceMethod(Selector("setFont:"), ofClass: UILabel.self, withMethod: Selector("pi_setFont:"), ofClass: UILabel.self)
+        })
     }
     
-    /**
-    Creates date components by updating the specified component to the input value. This does not do any date validation.
-    
-    :param: component The component to be updated.
-    :param: value     The value the component should be updated to.
-    
-    :returns: The components by updating the current date's components to the specified value.
-    */
-    private func currentCalendarComponentsByUpdatingComponent(component : PIDatePickerComponents, toValue value : Int) -> NSDateComponents {
-        let components = self.currentCalendarComponents
+    // Layouts the label to use the proper font and text color.
+    public override func layoutSubviews() {
+        super.layoutSubviews()
         
-        if (component == .Month) {
-            components.month = value
-        } else if (component == .Day) {
-            components.day = value
-        } else {
-            components.year = value
+        // Apply the visual modifications when contained in a date picker.
+        if let datePicker = datePicker(self) {
+            // Check if this class actually responds to 'textColor' (otherwise, there are errors with Apple's private iOS SDK).
+            if self.respondsToSelector(Selector("textColor")) {
+                self.textColor = datePicker.textColor
+            }
+            if self.respondsToSelector("font") {
+                self.font = datePicker.font
+            }
         }
-        
-        return components
     }
     
     /**
-    Creates date components by updating the specified component to the input value. If the resulting value is not a valid date object, the components will be updated to the closest best value.
-    
-    :param: component The component to be updated.
-    :param: value     The value the component should be updated to.
-    
-    :returns: The components by updating the specified value; the components will be a valid date object.
-    */
-    private func validDateValueByUpdatingComponent(component : PIDatePickerComponents, toValue value : Int) -> NSDateComponents {
-        let components = self.currentCalendarComponentsByUpdatingComponent(component, toValue: value)
-        
-        if (!self.isValidValue(value, forComponent: component)) {
-            if (component == .Month) {
-                components.day = self.numberOfDaysForMonth(value, inYear: components.year)
-            } else if (component == .Day) {
-                components.day = self.numberOfDaysForMonth(components.month, inYear:components.year)
-            } else {
-                components.day = self.numberOfDaysForMonth(components.month, inYear: value)
-            }
+     Replacement for UILabel.setTextColor.
+     
+     - parameter color: The new text color for this label. Will only be set, if this label is not contained within a UIDatePicker.
+     */
+    private func pi_setTextColor(color: UIColor) {
+        // If part of a date picker, use the date picker's text color instead.
+        if let datePicker = datePicker(self), let pickerColor = datePicker.textColor {
+            self.pi_setTextColor(pickerColor)
+            return
         }
-        
-        return components
+        self.pi_setTextColor(color)
     }
     
-    // MARK: -
-    // MARK: Protocols
-    // MARK: UIPickerViewDelegate
-    
-    public func pickerView(pickerView: UIPickerView, didSelectRow row: Int, inComponent component: Int) {
-        let datePickerComponent = self.componentAtIndex(component)
-        let value = self.rawValueForRow(row, inComponent: datePickerComponent)
-        
-        // Create the newest valid date components.
-        let components = self.validDateValueByUpdatingComponent(datePickerComponent, toValue: value)
-        
-        // If the resulting components are not in the date range ...
-        if (!self.dateIsInRange(self.calendar.dateFromComponents(components)!)) {
-            // ... go back to original date
-            self.setDate(self.date, animated: true)
-        } else {
-            // Get the components that would result by just force-updating the current components.
-            let rawComponents = self.currentCalendarComponentsByUpdatingComponent(datePickerComponent, toValue: value)
-            
-            if (rawComponents.day != components.day) {
-                // Only animate the change if the day value is not a valid date.
-                self.setIndexOfComponent(.Day, toValue: components.day, animated: self.isValidValue(components.day, forComponent: .Day))
-            }
-            
-            if (rawComponents.month != components.month) {
-                self.setIndexOfComponent(.Month, toValue: components.day, animated: datePickerComponent != .Month)
-            }
-            
-            if (rawComponents.year != components.year) {
-                self.setIndexOfComponent(.Year, toValue: components.day, animated: datePickerComponent != .Year)
-            }
-            
-            self.date = self.calendar.dateFromComponents(components)!
-            self.sendActionsForControlEvents(.ValueChanged)
-        }
-        
-        self.delegate?.pickerView(self, didSelectRow: row, inComponent: component)
-    }
-
-    public func pickerView(pickerView: UIPickerView, viewForRow row: Int, forComponent component: Int, reusingView view: UIView?) -> UIView {
-        let label = view as? UILabel == nil ? UILabel() : view as! UILabel
-        
-        label.font = self.font
-        label.textColor = self.textColor
-        label.text = self.titleForRow(row, inComponentIndex: component)
-        label.textAlignment = self.componentAtIndex(component) == .Month ? NSTextAlignment.Left : NSTextAlignment.Right
-        label.textColor = self.isRowEnabled(row, forComponent: self.componentAtIndex(component)) ? self.textColor : self.disabledTextColor
-
-        return label
-    }
-
-    public func pickerView(pickerView: UIPickerView, widthForComponent component: Int) -> CGFloat {
-        let widthBuffer = 25.0
-        
-        let calendarComponent = self.componentAtIndex(component)
-        let stringSizingAttributes = [NSFontAttributeName : self.font]
-        var size = 0.01
-        
-        if calendarComponent == .Month {
-            let dateFormatter = self.dateFormatter()
-            
-            // Get the length of the longest month string and set the size to it.
-            for symbol in dateFormatter.monthSymbols as [String] {
-                let monthSize = NSString(string: symbol).sizeWithAttributes(stringSizingAttributes)
-                size = max(size, Double(monthSize.width))
-            }
-        } else if calendarComponent == .Day{
-            // Pad the day string to two digits
-            let dayComponentSizingString = NSString(string: "00")
-            size = Double(dayComponentSizingString.sizeWithAttributes(stringSizingAttributes).width)
-        } else if calendarComponent == .Year  {
-            // Pad the year string to four digits.
-            let yearComponentSizingString = NSString(string: "00")
-            size = Double(yearComponentSizingString.sizeWithAttributes(stringSizingAttributes).width)
+    /**
+     Replacement for UILabel.setFont.
+     
+     - parameter font: The new text font for this label. Will only be set, if this label is not contained within a UIDatePicker.
+     */
+    private func pi_setFont(font: UIFont) {
+        // If part of a date picker, use the date picker's font instead.
+        if let datePicker = datePicker(self) {
+            self.pi_setFont(datePicker.font)
+            return
         }
-        
-        // Add the width buffer in order to allow the picker components not to run up against the edges
-        return CGFloat(size + widthBuffer)
-    }
-
-
-    // MARK: UIPickerViewDataSource
-    public func pickerView(pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
-        return self.maximumNumberOfRows
-    }
-
-    public func numberOfComponentsInPickerView(pickerView: UIPickerView) -> Int {
-        return 3
+        self.pi_setFont(font)
     }
 }
diff --git a/Pod/Classes/PIDatePickerComponents.swift b/Pod/Classes/PIDatePickerComponents.swift
deleted file mode 100644
index bc30fcc..0000000
--- a/Pod/Classes/PIDatePickerComponents.swift
+++ /dev/null
@@ -1,14 +0,0 @@
-//
-//  PIDatePickerComponents.swift
-//  Pods
-//
-//  Created by Christopher Jones on 3/30/15.
-//
-//
-
-enum PIDatePickerComponents : Character  {
-    case Invalid = "!",
-            Year = "y",
-            Month = "M",
-            Day = "d"
-}
diff --git a/Pod/Classes/PIDatePickerDelegate.swift b/Pod/Classes/PIDatePickerDelegate.swift
deleted file mode 100644
index b0e9093..0000000
--- a/Pod/Classes/PIDatePickerDelegate.swift
+++ /dev/null
@@ -1,13 +0,0 @@
-//
-//  PIDatePickerDelegate.swift
-//  Pods
-//
-//  Created by Matt Luedke on 5/8/15.
-//
-//
-
-import Foundation
-
-public protocol PIDatePickerDelegate {
-    func pickerView(pickerView: PIDatePicker, didSelectRow row: Int, inComponent component: Int)
-}