support rectForSegment public function

teambition · Nov 3, 2016 · fa24aae · fa24aae
1 parent 2fa12f7
commit fa24aae
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/SegmentedControl/SegmentedControl.swift b/SegmentedControl/SegmentedControl.swift
@@ -90,6 +90,15 @@ open class SegmentedControl: UIControl {
         }
     }
     open fileprivate(set) var isLongPressActivated = false
+    open var scrollContentInset: UIEdgeInsets {
+        return scrollView.contentInset
+    }
+    open var scrollContentSize: CGSize {
+        return scrollView.contentSize
+    }
+    open var scrollContentOffset: CGPoint {
+        return scrollView.contentOffset
+    }
 
     fileprivate lazy var scrollView: SCScrollView = {
         let scrollView = SCScrollView()
@@ -153,6 +162,25 @@ open class SegmentedControl: UIControl {
         }
     }
 
+    open func rectForSegment(at index: Int) -> CGRect? {
+        guard 0..<segmentsCount() ~= index else {
+            return nil
+        }
+        switch layoutPolicy {
+        case .fixed:
+            return CGRect(x: singleSegmentWidth() * CGFloat(index),
+                          y: 0,
+                          width: singleSegmentWidth(),
+                          height: frame.height)
+        case .dynamic:
+            let frontWidths = totalSegmentsWidths(before: index)
+            return CGRect(x: contentInset.left + frontWidths.reduce(0, +) + segmentSpacing * CGFloat(frontWidths.count),
+                          y: 0,
+                          width: singleSegmentWidth(at: index),
+                          height: frame.height)
+        }
+    }
+
     // MARK: - Initialization
     public override init(frame: CGRect) {
         super.init(frame: frame)