Optimize masking #42

lbrndnr · Aug 29, 2015 · bcc6891 · bcc6891
1 parent 49ae18e
commit bcc6891
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/ImagePickerSheetController/ImagePickerSheetController/Sheet/SheetCollectionViewCell.swift b/ImagePickerSheetController/ImagePickerSheetController/Sheet/SheetCollectionViewCell.swift
@@ -68,6 +68,19 @@ class SheetCollectionViewCell: UICollectionViewCell {
         }
     }
 
+    private var needsMasking: Bool {
+        guard backgroundInsets == UIEdgeInsets() else {
+            return true
+        }
+
+        switch roundedCorners {
+        case .None:
+            return false
+        default:
+            return true
+        }
+    }
+
     // MARK: - Initialization
 
     override init(frame: CGRect) {
@@ -97,7 +110,7 @@ class SheetCollectionViewCell: UICollectionViewCell {
     // MARK: - Mask
 
     private func reloadMask() {
-        if layer.mask == nil {
+        if needsMasking && layer.mask == nil {
             let maskLayer = CAShapeLayer()
             maskLayer.frame = bounds
             maskLayer.lineWidth = 0