//
//  UISegmentedControlExt.swift
//  ExtensionKit
//
//  Created by FFIB on 2017/9/14.
//  Copyright © 2017年 FFIB. All rights reserved.
//

import UIKit

extension UISegmentedControl {
    public var segmentTitles: [String] {
        get {
            let range = 0..<numberOfSegments
            return range.flatMap { titleForSegment(at: $0) }
        }
        set {
            removeAllSegments()
            for (index, title) in newValue.enumerated() {
                insertSegment(withTitle: title, at: index, animated: false)
            }
        }
    }
    
    public var segmentImages: [UIImage] {
        get {
            let range = 0..<numberOfSegments
            return range.flatMap { imageForSegment(at: $0) }
        }
        set {
            removeAllSegments()
            for (index, image) in newValue.enumerated() {
                insertSegment(with: image, at: index, animated: false)
            }
        }
    }
}