Optimize & Release 0.2.1;

cuzv · Mar 29, 2017 · e2c77f3 · e2c77f3
1 parent 39c59db
commit e2c77f3
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Sources/UINavigationController+RRNavigationBar.m b/Sources/UINavigationController+RRNavigationBar.m
@@ -54,10 +54,10 @@ - (void)_rr_nvc_viewWillLayoutSubviews {
     // when present a nvc, vc:viewDidLoad -> nvc:viewWillLayoutSubviews.
     // So, in vc:viewDidLoad vc.rr_navigationBar isn't the correct styles,
     // therefore recored setups, recover when nvc.rr_navigationBar initialized.
-    NSDictionary *info = self.topViewController.rr_navigationBar._tmpInfo;
-    if (info) {
-        UINavigationBar *this = self.rr_navigationBar;
-        UINavigationBar *bar = self.topViewController.rr_navigationBar;
+    UINavigationBar *this = self.rr_navigationBar;
+    UINavigationBar *bar = self.topViewController.rr_navigationBar;
+    NSDictionary *info = bar._tmpInfo;
+    if (this && info) {
         if (info[@"barStyle"]) {
             bar.barStyle = [info[@"barStyle"] integerValue];
         } else {
@@ -113,7 +113,7 @@ - (void)_rr_nvc_viewWillLayoutSubviews {
         } else {
             bar.rr_forceShadowImageHidden = this.rr_forceShadowImageHidden;
         }
-        self.topViewController.rr_navigationBar._tmpInfo = nil;
+        bar._tmpInfo = nil;
     }
 }