Get UIImage with proper orientation from AVCaptureVideoDataOutputSampleBufferDelegate

UIImage+SampleBuffer.swift

func imageFromSampleBuffer(
        sampleBuffer: CMSampleBuffer,
        videoOrientation: AVCaptureVideoOrientation) -> UIImage? {
        if let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) {
            let context = CIContext()
            var ciImage = CIImage(cvPixelBuffer: imageBuffer)

            // FIXME: - change to Switch
            if videoOrientation == .landscapeLeft {
                ciImage = ciImage.applyingOrientation(3)
            } else if videoOrientation == .landscapeRight {
                ciImage = ciImage.applyingOrientation(1)
            } else if videoOrientation == .portrait {
                ciImage = ciImage.applyingOrientation(6)
            } else if videoOrientation == .portraitUpsideDown {
                ciImage = ciImage.applyingOrientation(8)
            }

            if let cgImage = context.createCGImage(ciImage, from: ciImage.extent) {
                return UIImage(cgImage: cgImage)
            }
        }

        return nil
    }