just use the rootVC instead of specifying which vc is going to use.

Author: michaelhenry

Example/Demo/GalleryViewController.swift

@@ -82,7 +82,6 @@ extension GalleryViewController:UICollectionViewDataSource {
 
         // Setup Image Viewer
         cell.imageView.setupImageViewer(
-            with: self,
             images: images,
             initialIndex: indexPath.item)
         return cell

Sources/ImageViewerController.swift

@@ -271,7 +271,6 @@ class ImageViewerController : UIViewController, UIGestureRecognizerDelegate {
     }
 }
 
-
 extension ImageViewerController:UIScrollViewDelegate {
 
     func viewForZooming(in scrollView: UIScrollView) -> UIView? {

Sources/UIImageView_Extensions.swift

@@ -4,20 +4,23 @@ extension UIImageView {
 
     // Data holder tap recognizer
     private class TapWithDataRecognizer:UITapGestureRecognizer {
-        unowned var viewController:UIViewController!
         var imageDatasource:ImageDataSource?
         var initialIndex:Int = 0
         var options:[ImageViewerOption] = []
     }
 
+    private var vc:UIViewController? {
+        return UIApplication.shared.keyWindow?.rootViewController
+    }
+    
     public func setupImageViewer(
-        with vc:UIViewController, options:[ImageViewerOption] = []) {
-         setup(with: vc, datasource: nil, options: options)
+        options:[ImageViewerOption] = []) {
+        setup(datasource: nil, options: options)
     }
 
     public func setupImageViewer(
-        with vc:UIViewController,
-        url:URL, initialIndex:Int = 0,
+        url:URL,
+        initialIndex:Int = 0,
         placeholder: UIImage? = nil,
         options:[ImageViewerOption] = []) {
 
@@ -26,31 +29,29 @@ extension UIImageView {
                 ImageItem.url($0, placeholder: placeholder ?? image)
         })
         setup(
-            with: vc,
             datasource: datasource,
             initialIndex: initialIndex,
             options: options)
     }
 
     public func setupImageViewer(
-        with vc:UIViewController,
-        images:[UIImage], initialIndex:Int = 0,
+        images:[UIImage],
+        initialIndex:Int = 0,
         options:[ImageViewerOption] = []) {
 
         let datasource = SimpleImageDatasource(
             imageItems: images.compactMap {
                 ImageItem.image($0)
         })
         setup(
-            with: vc,
             datasource: datasource,
             initialIndex: initialIndex,
             options: options)
     }
 
     public func setupImageViewer(
-        with vc:UIViewController,
-        urls:[URL], initialIndex:Int = 0,
+        urls:[URL],
+        initialIndex:Int = 0,
         options:[ImageViewerOption] = [],
         placeholder: UIImage? = nil) {
 
@@ -59,18 +60,23 @@ extension UIImageView {
                 ImageItem.url($0, placeholder: placeholder)
         })
         setup(
-            with: vc,
             datasource: datasource,
             initialIndex: initialIndex,
             options: options)
     }
 
-    public func setupImageViewer(with vc:UIViewController, datasource:ImageDataSource, initialIndex:Int = 0, options:[ImageViewerOption] = []) {
-        setup(with: vc, datasource: datasource, initialIndex: initialIndex, options: options)
+    public func setupImageViewer(
+        datasource:ImageDataSource,
+        initialIndex:Int = 0,
+        options:[ImageViewerOption] = []) {
+        
+        setup(
+            datasource: datasource,
+            initialIndex: initialIndex,
+            options: options)
     }
 
     private func setup(
-        with vc:UIViewController,
         datasource:ImageDataSource?,
         initialIndex:Int = 0,
         options:[ImageViewerOption] = []) {
@@ -90,7 +96,6 @@ extension UIImageView {
         _tapRecognizer.numberOfTapsRequired = 1
 
         // Pass the Data
-        _tapRecognizer.viewController = vc
         _tapRecognizer.imageDatasource = datasource
         _tapRecognizer.initialIndex = initialIndex
         _tapRecognizer.options = options
@@ -111,6 +116,6 @@ extension UIImageView {
         imageCarousel.options = sender.options
         imageCarousel.modalPresentationStyle = .overFullScreen
         imageCarousel.modalPresentationCapturesStatusBarAppearance = true
-        sender.viewController.present(imageCarousel, animated: false, completion: nil)
+        vc?.present(imageCarousel, animated: false, completion: nil)
     }
 }