adding closures for callbacks on segment selection

malekwasim · Nov 1, 2019 · c462abe · c462abe
1 parent 7df774a
commit c462abe
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/WMSegmentControl/SegmentViewController.swift b/WMSegmentControl/SegmentViewController.swift
@@ -25,11 +25,16 @@ class SegmentViewController: UITableViewController {
         sgTextOnlyBar.selectorType = .bottomBar
         sgTextOnlyBar.SelectedFont = UIFont(name: "ChalkboardSE-Bold", size: 15)!
         sgTextOnlyBar.normalFont = UIFont(name: "ChalkboardSE-Regular", size: 15)!
-        // Do any additional setup after loading the view.
+
+        //Using callbacks
+        sgTextOnly.onValueChanged = { index in
+            print("I have selected index \(index) from WMSegment!")
+        }
     }
 
     @IBAction func segmentValueChange(_ sender: WMSegment) {
-        switch sender.selectedSegmentIndex {
+        print("selected index = \(sender.selectedSegmentIndex)")
+        /*switch sender.selectedSegmentIndex {
         case 0:
             print("first item")
         case 1:
@@ -38,7 +43,7 @@ class SegmentViewController: UITableViewController {
             print("Third item")
         default:
             print("default item")
-        }
+        }*/
     }
 
 }

diff --git a/WMSegmentControl/Source/WMSegment.swift b/WMSegmentControl/Source/WMSegment.swift
@@ -9,7 +9,7 @@
 import UIKit
 @IBDesignable
 open class WMSegment: UIControl {
-
+    var onValueChanged: ((_ index: Int)->())?
     var buttons = [UIButton]()
     var selector: UIView!
     public var selectedSegmentIndex: Int = 0
@@ -246,7 +246,7 @@ open class WMSegment: UIControl {
                 btn.setTitleColor(selectorTextColor, for: .normal)
             }
         }
-
+        onValueChanged?(selectedSegmentIndex)
         sendActions(for: .valueChanged)
     }
     //MARK: set Selected Index