iOS 获取当前显示的ViewController
程序员文章站
2024-01-15 08:15:40
...
我们在非视图类中想要随时展示一个view时,需要将被展示的view加到当前view的子视图,或需要 presentViewController,或pushViewContrller,这些操作都需要获取当前正在显示的ViewController。
发觉百度搜出来好多不科学的方法,有时候还是自己动脑子想好。。。
下面提供了一个公用的方法去获取当前ViewController,也建议直接改成静态方法,放到工具类里直接调用更加科学。
具体代码如下:
//获取当前屏幕显示的viewcontroller
- (UIViewController *)getCurrentVC
{
UIViewController *rootViewController = [UIApplication sharedApplication].keyWindow.rootViewController;
UIViewController *currentVC = [self getCurrentVCFrom:rootViewController];
return currentVC;
}
- (UIViewController *)getCurrentVCFrom:(UIViewController *)rootVC
{
UIViewController *currentVC;
if ([rootVC presentedViewController]) {
// 视图是被presented出来的
rootVC = [rootVC presentedViewController];
}
if ([rootVC isKindOfClass:[UITabBarController class]]) {
// 根视图为UITabBarController
currentVC = [self getCurrentVCFrom:[(UITabBarController *)rootVC selectedViewController]];
} else if ([rootVC isKindOfClass:[UINavigationController class]]){
// 根视图为UINavigationController
currentVC = [self getCurrentVCFrom:[(UINavigationController *)rootVC visibleViewController]];
} else {
// 根视图为非导航类
currentVC = rootVC;
}
return currentVC;
}
解析:代码主要使用了递归的思想(哈哈哈,毕业工作半年,发觉第一次写iOS用到递归,突然觉得高大上)。
[UIApplication sharedApplication].keyWindow.rootViewController获取到的是项目的根视图,结合可能用到UITabBarController或者UINavigationController作为导航结构,以及可能present出新的VC,其实如果用storyboard的方式写UI的话就很清晰,类似树的结构,再利用递归找到当前视图。
ps:
如果是需要push新的视图,就很简单啦:
用上面的方法获取到顶层的视图,判断currentVC.navigationController是否为nil。(为nil,则新建UINavigationController在push;否则直接用currentVC.navigationController去push)
ps2:
补充评论里好的建议,如果用到的场景主要是vc里,可以弄成类别如下:
#import "UIViewController+Helper.h"
@property (nonatomic, strong ,readonly) UIViewController * _Nullable currentVC;
//当前屏幕显示的viewcontroller
-(UIViewController *)currentVC{
UIViewController *rootViewController = [UIApplication sharedApplication].keyWindow.rootViewController;
UIViewController *controller = [self getCurrentVCFrom:rootViewController];
return controller;
}
//getCurrentVCFrom参考上文
还有什么不懂的就留言问问哈。。。。