add segment handler

kazuhiro4949 · Jul 17, 2017 · c9614d4 · c9614d4
1 parent d6d5492
commit c9614d4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/iOS Sample/iOS Sample/ViewController.swift b/iOS Sample/iOS Sample/ViewController.swift
@@ -31,6 +31,11 @@ class ViewController: UIViewController {
         contentViewController?.reloadData(with: 2)
     }
 
+    override func viewDidLayoutSubviews() {
+        super.viewDidLayoutSubviews()
+        menuViewController?.reloadDate()
+    }
+
     override func didReceiveMemoryWarning() {
         super.didReceiveMemoryWarning()
     }
@@ -60,7 +65,7 @@ extension ViewController: PagingMenuViewControllerDataSource {
     }
 
     func menuViewController(viewController: PagingMenuViewController, areaForItemAt index: Int) -> CGFloat {
-        return viewController.view.bounds.size.width / CGFloat(dataSource.count)
+        return UIScreen.main.bounds.width / CGFloat(dataSource.count)
     }