Improve framework

Friend-LGA · Dec 20, 2016 · 82a4532 · 82a4532
1 parent d95df93
commit 82a4532
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/LGSideMenuController/LGSideMenuController.m b/LGSideMenuController/LGSideMenuController.m
@@ -1372,7 +1372,7 @@ - (void)rootViewsValidate {
 
     // -----
 
-    if (self.rootViewController && !self.isLeftViewVisible && !self.isRightViewVisible) {
+    if (self.rootViewController && !self.isLeftViewGoingToShow && !self.isRightViewGoingToShow) {
         [self addChildViewController:self.rootViewController];
     }
 
@@ -2286,13 +2286,13 @@ - (void)showLeftViewAnimatedActions:(BOOL)animated completionHandler:(LGSideMenu
 - (void)showLeftViewDoneWithGesture:(BOOL)withGesture {
     self.leftViewShowing = YES;
 
-    if (self.rootViewController) {
-        [self addChildViewController:self.rootViewController];
-    }
-
     if (self.isLeftViewGoingToShow) {
         self.leftViewGoingToShow = NO;
 
+        if (self.rootViewController) {
+            [self addChildViewController:self.rootViewController];
+        }
+
         [self didShowLeftViewCallbacks];
     }
 }
@@ -2541,12 +2541,12 @@ - (void)showRightViewAnimatedActions:(BOOL)animated completionHandler:(LGSideMen
 - (void)showRightViewDoneWithGesture:(BOOL)withGesture {
     self.rightViewShowing = YES;
 
-    if (self.rootViewController) {
-        [self addChildViewController:self.rootViewController];
-    }
-
     if (self.isRightViewGoingToShow) {
         self.rightViewGoingToShow = NO;
+
+        if (self.rootViewController) {
+            [self addChildViewController:self.rootViewController];
+        }
 
         [self didShowRightViewCallbacks];
     }