Merge pull request #37 from nghialv/feature/hotfix

Cache calculatedHeight
nghialv · Apr 4, 2016 · dda569a · dda569a
2 parents 4021495 + 8b2ab03
commit dda569a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Hakuba/CellModel.swift b/Hakuba/CellModel.swift
@@ -95,6 +95,9 @@ private extension CellModel {
         cell.layoutIfNeeded()
 
         let size = cell.contentView.systemLayoutSizeFittingSize(UILayoutFittingCompressedSize)
-        return size.height + 1
+        let height = size.height + 1
+        calculatedHeight = height
+
+        return height
     }
 }