Skip to content

Commit 0051b0a

Browse files
author
Erik Malyak
committed
Cleaned up view controller loading functions
1 parent 4d1b354 commit 0051b0a

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

EMPageViewController/EMPageViewController.swift

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -398,20 +398,12 @@ public class EMPageViewController: UIViewController, UIScrollViewDelegate {
398398

399399
private func loadBeforeViewControllerForSelectedViewController(selectedViewController:UIViewController) {
400400
// Retreive the new before controller from the data source if available, otherwise set as nil
401-
if let beforeViewController = self.dataSource?.em_pageViewController(self, viewControllerBeforeViewController: selectedViewController) {
402-
self.beforeViewController = beforeViewController
403-
} else {
404-
self.beforeViewController = nil
405-
}
401+
self.beforeViewController = self.dataSource?.em_pageViewController(self, viewControllerBeforeViewController: selectedViewController)
406402
}
407403

408404
private func loadAfterViewControllerForSelectedViewController(selectedViewController:UIViewController) {
409405
// Retreive the new after controller from the data source if available, otherwise set as nil
410-
if let afterViewController = self.dataSource?.em_pageViewController(self, viewControllerAfterViewController: selectedViewController) {
411-
self.afterViewController = afterViewController
412-
} else {
413-
self.afterViewController = nil
414-
}
406+
self.afterViewController = self.dataSource?.em_pageViewController(self, viewControllerAfterViewController: selectedViewController)
415407
}
416408

417409

0 commit comments

Comments
 (0)