Set default width of window or screen

Author: kitasuke

Pod/Classes/MenuItemView.swift

@@ -229,7 +229,11 @@ open class MenuItemView: UIView {
         let width: CGFloat
         switch menuOptions.displayMode {
         case .segmentedControl:
-            width = UIApplication.shared.keyWindow!.bounds.size.width / CGFloat(menuOptions.itemsOptions.count)
+            if let windowWidth = UIApplication.shared.keyWindow?.bounds.size.width {
+                width = windowWidth / CGFloat(menuOptions.itemsOptions.count)
+            } else {
+                width = UIScreen.main.bounds.width / CGFloat(menuOptions.itemsOptions.count)
+            }
         default:
             width = image.size.width + horizontalMargin * 2
         }

Pod/Classes/MenuView.swift

@@ -56,7 +56,13 @@ open class MenuView: UIScrollView {
         }
     }
     fileprivate var centerOfScreenWidth: CGFloat {
-        return menuItemViews[currentPage].frame.midX - UIApplication.shared.keyWindow!.bounds.width / 2
+        let screenWidth: CGFloat
+        if let width = UIApplication.shared.keyWindow?.bounds.width {
+            screenWidth = width
+        } else {
+            screenWidth = UIScreen.main.bounds.width
+        }
+        return menuItemViews[currentPage].frame.midX - screenWidth / 2
     }
     fileprivate var contentOffsetXForCurrentPage: CGFloat {
         guard menuItemCount > MinimumSupportedViewCount else { return 0.0 }