When star points is less than 3, use 5 as default.

IBAnimatable · Feb 14, 2016 · c1bd8a7 · c1bd8a7
1 parent 5f0bcd8
commit c1bd8a7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/IBAnimatable/MaskDesignable.swift b/IBAnimatable/MaskDesignable.swift
@@ -53,13 +53,17 @@ public extension MaskDesignable where Self: UIView {
   public func maskStar(points: Int = 5) {
     // FIXME: Do not mask the shadow.
 
-    assert(points >= 2, "Stars must has at least 2 points.")
+    // Stars must has at least 3 points.
+    var starPoints = points
+    if points <= 2 {
+      starPoints = 5
+    }
     layer.mask?.removeFromSuperlayer()
     layer.sublayers?
       .filter  { $0.name == "borderSideLayer" || $0.name == "borderLayer" }
       .forEach { $0.removeFromSuperlayer() }
 
-    let path = starPath(points)
+    let path = starPath(starPoints)
     drawPath(path)
   }