swift Swift中的通用分段控件

Posted 2021-05-19

import UIKit

class SegmentedControl: UIControl {
    private let action = Selector("selectSegment:")

    @IBOutlet
    private var segments: [UIButton] = [] {
        didSet {
            segments.sortInPlace { $0.tag < $1.tag }
            subscribe(segments)
        }
    }

    var selectedSegmentIndex: Int = 0 {
        didSet { selectSegmentAtIndex(selectedSegmentIndex) }
    }

    override var enabled: Bool {
        didSet { segments.forEach { $0.enabled = enabled } }
    }

    private func subscribe(segments: [UIButton]) {
        segments.forEach {
            $0.addTarget(self, action: action, forControlEvents: .TouchDown)
        }
    }

    private dynamic func selectSegment(sender: UIButton) {
        if let index = segments.indexOf(sender) where index != selectedSegmentIndex {
            selectedSegmentIndex = index
            sendActionsForControlEvents(.ValueChanged)
        }
    }

    private func selectSegmentAtIndex(index: Int) {
        segments.forEach { $0.selected = false }
        if index != UISegmentedControlNoSegment {
            segments[index].selected = true
        }
    }
}