call layoutIfNeeded before doing batch updates (#13)

FluidGroup · Sep 9, 2020 · 4a33bb5 · 4a33bb5
1 parent c2920a2
commit 4a33bb5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/StackScrollView/StackScrollView.swift b/StackScrollView/StackScrollView.swift
@@ -115,6 +115,8 @@ open class StackScrollView: UICollectionView, UICollectionViewDataSource, UIColl
 
   open func insert(views _views: [UIView], at index: Int, animated: Bool) {
 
+    layoutIfNeeded()
+
     var _views = _views
     _views.removeAll(where: views.contains(_:))
     views.insert(contentsOf: _views, at: index)
@@ -164,7 +166,9 @@ open class StackScrollView: UICollectionView, UICollectionViewDataSource, UIColl
   }
 
   open func remove(view: UIView, animated: Bool) {
-
+
+    layoutIfNeeded()
+
     if let index = views.firstIndex(of: view) {
       views.remove(at: index)
       if animated {
@@ -200,6 +204,8 @@ open class StackScrollView: UICollectionView, UICollectionViewDataSource, UIColl
 
   open func remove(views: [UIView], animated: Bool) {
 
+    layoutIfNeeded()
+
     var indicesForRemove: [Int] = []
 
     for view in views {