Cacher is the code snippet organizer for pro developers

import UIKit

enum ScreenSize {
    case size3_5Inches
    case size4Inches
    case size4_7Inches
    case size5_5Inches
}

extension UIScreen {
    static var inchSize: ScreenSize {
        switch self.main.bounds.height {
        case 480: return .size3_5Inches
        case 568: return .size4Inches
        case 667: return .size4_7Inches
        default: return .size5_5Inches
        }
    }
}