Added update edgeInsets logic (#189)

wayfair-archive · Sep 28, 2017 · 99a323c · 99a323c
1 parent c4922c5
commit 99a323c
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/Source/Cells/BrickCell.swift b/Source/Cells/BrickCell.swift
@@ -181,12 +181,17 @@ open class BrickCell: BaseBrickCell {
         return UIEdgeInsetsMake(defaultTopConstraintConstant, defaultLeftConstraintConstant, defaultBottomConstraintConstant, defaultRightConstraintConstant)
     }
 
+    private var didUpdateEdgeInsets: Bool = false
     @objc open dynamic var edgeInsets: UIEdgeInsets = UIEdgeInsets.zero {
         didSet {
+            if edgeInsets == oldValue {
+                return
+            }
             self.topSpaceConstraint?.constant = edgeInsets.top
             self.bottomSpaceConstraint?.constant = edgeInsets.bottom
             self.leftSpaceConstraint?.constant = edgeInsets.left
             self.rightSpaceConstraint?.constant = edgeInsets.right
+            didUpdateEdgeInsets = true
         }
     }
 
@@ -226,7 +231,14 @@ open class BrickCell: BaseBrickCell {
         guard self._brick.height.isEstimate(withValue: nil) else {
             return layoutAttributes
         }
-
+
+        if !didUpdateEdgeInsets {
+            guard let brickAttributes = layoutAttributes as? BrickLayoutAttributes, brickAttributes.isEstimateSize else {
+                return layoutAttributes
+            }
+        }
+        didUpdateEdgeInsets = false
+
         let preferred = layoutAttributes
 
         // We're inverting the frame because the given frame is already transformed