Delegate view creation back to SMRotaryProtocol

Author: msbit

RotatingWheelController/ViewController.swift

@@ -10,10 +10,18 @@ import os
 import UIKit
 
 class ViewController: UIViewController, SMRotaryProtocol {
+    func viewFor(tag: Int) -> UIView {
+        let view = UILabel(frame: CGRect(x: 0, y: 0, width: 100, height: 40))
+        view.backgroundColor = .red
+        view.tag = tag
+        view.text = "\(tag)"
+        return view
+    }
+
     func wheelDidChangeValue(to: Int) {
         os_log(OSLogType.default, "%d", to);
     }
-    
+
     override func viewDidLoad() {
         super.viewDidLoad()
         let wheel = SMRotaryWheel(frame: view.frame, delegate: self, numberOfSections: 3)

SMRotaryProtocol.swift

@@ -6,6 +6,9 @@
 //  Copyright © 2018 tom. All rights reserved.
 //
 
+import UIKit
+
 protocol SMRotaryProtocol {
     func wheelDidChangeValue(to: Int)
+    func viewFor(tag: Int) -> UIView
 }

SMRotaryWheel.swift

@@ -87,18 +87,10 @@ class SMRotaryWheel: UIControl {
         let dy = point.y - centre.y
         return (dx * dx + dy * dy).squareRoot()
     }
-    
-    func createView(index: Int) -> UIView {
-        let view = UILabel(frame: CGRect(x: 0, y: 0, width: 100, height: 40))
-        view.backgroundColor = .red
-        view.tag = index
-        view.text = "\(index)"
-        return view
-    }
 
     func setup() {
         container = UIView(frame: frame)
-        guard let container = container, let numberOfSections = numberOfSections else {
+        guard let container = container, let delegate = delegate, let numberOfSections = numberOfSections else {
             return
         }
 
@@ -107,7 +99,7 @@ class SMRotaryWheel: UIControl {
 
         for i in 0..<numberOfSections {
             // Create and add view to container
-            let view = createView(index: i)
+            let view = delegate.viewFor(tag: i)
             view.layer.anchorPoint = CGPoint(x: 1.0, y: 0.5)
             view.layer.position = CGPoint(x: container.bounds.size.width / 2.0, y: container.bounds.size.height / 2.0)
             view.transform = CGAffineTransform(rotationAngle: CGFloat(i) * sectorAngle)
@@ -132,6 +124,6 @@ class SMRotaryWheel: UIControl {
         container.isUserInteractionEnabled = false
         addSubview(container)
 
-        delegate?.wheelDidChangeValue(to: currentSector)
+        delegate.wheelDidChangeValue(to: currentSector)
     }
 }