這是一個(gè)擴(kuò)展, 不知道怎么編輯簡(jiǎn)書(shū), 復(fù)制過(guò)來(lái)的不好看, 大家直接復(fù)制出去就行了, 主要看方法
@implementation UIViewController (IMYPublic)
- (UINavigationController*)imy_navigationController
{
UINavigationController* nav = nil;
if ([self isKindOfClass:[UINavigationController class]]) {
nav = (id)self;
}? else {
if ([self isKindOfClass:[UITabBarController class]]) {
nav = [((UITabBarController*)self).selectedViewController imy_navigationController];
} else {
nav = self.navigationController;
}
}
return nav;
}
@end
