CMSampleBuffer с фронтальной камеры на UIImage

Я пытаюсь создать UIImage из CMSampleBuffer. Следующий код отлично работает для буферов семплов с задней камеры, но не с передней камеры. Когда используется фронтальная камера, CGContext не инициализируется, т. е. конструктор CGContext возвращает nil. Я подозреваю, что мне нужно использовать правильную информацию о растровом изображении, но есть так много комбинаций.

func convert(buffer: CMSampleBuffer) -> UIImage? {

   guard let imageBuffer = CMSampleBufferGetImageBuffer(buffer) else { return nil }

   CVPixelBufferLockBaseAddress(imageBuffer, CVPixelBufferLockFlags.readOnly)



   let baseAddress = CVPixelBufferGetBaseAddress(imageBuffer)

   let bytesPerRow = CVPixelBufferGetBytesPerRow(imageBuffer)

   let width = CVPixelBufferGetWidth(imageBuffer)

   let height = CVPixelBufferGetHeight(imageBuffer)



   let colorSpace = CGColorSpaceCreateDeviceRGB()

   var bitmapInfo: UInt32 = CGBitmapInfo.byteOrder32Little.rawValue

   bitmapInfo |= CGImageAlphaInfo.premultipliedFirst.rawValue & CGBitmapInfo.alphaInfoMask.rawValue

   let context = CGContext(data: baseAddress, width: width, height: height, bitsPerComponent: 8, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo)



   guard let quartzImage = context?.makeImage() else { return nil }

   CVPixelBufferUnlockBaseAddress(imageBuffer, CVPixelBufferLockFlags.readOnly)

   let image = UIImage(cgImage: quartzImage)



   return image

}

Это сопровождается следующей ошибкой: «CGBitmapContextCreate: недопустимые байты данных/строка: должно быть не менее 2560 для 8 целочисленных бит/компонент, 3 компонента, kCGImageAlphaPremultipliedFirst».