UIDeviceOrientation to Image Orientation

imageOrientation.swift

extension UIDeviceOrientation {

    var imageOrientation: UIImage.Orientation {
        switch UIDevice.current.orientation {
        case .landscapeLeft:
            return UIImage.Orientation.right
        case .landscapeRight:
            return UIImage.Orientation.left
        case .portrait:
            return UIImage.Orientation.up
        case .portraitUpsideDown:
            return UIImage.Orientation.down
        default:
            return UIImage.Orientation.up
        }
    }
}