fixup

PagingKit/MenuView.swift

@@ -159,21 +159,13 @@ public class PagingMenuView: UIScrollView {
         guard let dataSource = dataSource else {
             return
         }
-        
+
         visibleCells.forEach { $0.removeFromSuperview() }
         visibleCells = []
 
         numberOfItem = dataSource.numberOfItemForPagingMenuView()
 
-        frameQueue = []
-        var containerWidth: CGFloat = 0
-        (0..<numberOfItem).forEach { (index) in
-            let width = dataSource.pagingMenuView(pagingMenuView: self, widthForItemAt: index)
-            frameQueue.append(CGRect(x: containerWidth, y: 0, width: width, height: bounds.height))
-            containerWidth += width
-        }
-        contentSize = CGSize(width: containerWidth, height: bounds.height)
-        containerView.frame = CGRect(origin: .zero, size: contentSize)
+        invalidateLayout()
 
         setNeedsLayout()
         layoutIfNeeded()
@@ -229,6 +221,23 @@ public class PagingMenuView: UIScrollView {
         return CGRect(x: x, y: 0, width: frameQueue[index].width, height: bounds.height)
     }
 
+    public func invalidateLayout() {
+        guard let dataSource = dataSource else {
+            return
+        }
+
+        frameQueue = []
+        var containerWidth: CGFloat = 0
+        (0..<numberOfItem).forEach { (index) in
+            let width = dataSource.pagingMenuView(pagingMenuView: self, widthForItemAt: index)
+            frameQueue.append(CGRect(x: containerWidth, y: 0, width: width, height: bounds.height))
+            containerWidth += width
+        }
+        contentSize = CGSize(width: containerWidth, height: bounds.height)
+        containerView.frame = CGRect(origin: .zero, size: contentSize)
+        align()
+    }
+
     private func recenterIfNeeded() {
         let currentOffset = contentOffset
         let contentWidth = contentSize.width

PagingKit/PagingContentViewController.swift

@@ -49,6 +49,7 @@ public class PagingContentViewController: UIViewController {
     public func reloadData(with page: Int = 0) {
         initialLoad(with: page)
         scroll(to: page, animated: false)
+        leftSidePageIndex = page
     }
 
     public func scroll(to page: Int, animated: Bool) {
@@ -65,7 +66,6 @@ public class PagingContentViewController: UIViewController {
 
     fileprivate var numberOfPages: Int = 0
 
-    fileprivate var lastContentOffset = CGPoint.zero
     fileprivate var leftSidePageIndex = 0
     fileprivate var isExplicityScrolling = false
 
@@ -157,10 +157,8 @@ extension PagingContentViewController: UIScrollViewDelegate {
     }
 
     public func scrollViewDidScroll(_ scrollView: UIScrollView) {
-        lastContentOffset = scrollView.contentOffset
-        leftSidePageIndex = Int(scrollView.contentOffset.x / scrollView.bounds.width)
-
         if isExplicityScrolling {
+            leftSidePageIndex = Int(scrollView.contentOffset.x / scrollView.bounds.width)
             let leftSideContentOffset = CGFloat(leftSidePageIndex) * scrollView.bounds.width
             let percent = (scrollView.contentOffset.x - leftSideContentOffset) / scrollView.bounds.width
             let normalizedPercent = min(max(0, percent), 1)

PagingKit/PagingMenuViewController.swift

@@ -174,6 +174,7 @@ public class PagingMenuViewController: UIViewController {
 
         let index = menuView.indexForItem(at: focusView.center)
         layoutHandler = { [weak self] in
+            self?.menuView.invalidateLayout()
             self?.scroll(index: index ?? 0, percent: 0, animated: false)
             self?.layoutHandler = nil
         }

iOS Sample/iOS Sample/ViewController.swift

@@ -30,11 +30,6 @@ class ViewController: UIViewController {
         menuViewController?.reloadData(startingOn: dataSource.count - 1)
         contentViewController?.reloadData(with: dataSource.count - 1)
     }
-
-    override func viewDidLayoutSubviews() {
-        super.viewDidLayoutSubviews()
-        menuViewController?.reloadData()
-    }
 
     override func didReceiveMemoryWarning() {
         super.didReceiveMemoryWarning()