AVAssetWriter повернуть буфер для ориентации видео - PullRequest
Новая
       65

AVAssetWriter повернуть буфер для ориентации видео

0 голосов
/ 08 марта 2019

Я работаю над приложением для записи в реальном времени в Swift, используя AVFoundation, и у меня возникла проблема с ориентацией видео. Я использую AVAssetWriter, а не AVCaptureMovieFileOutput, потому что мне нужно записывать в квадратном формате (поправьте меня, если я ошибаюсь).

Я пытался использовать videoInput.transform, но слышал, что он поддерживается не во всех видеопроигрывателях.

Я не могу использовать avcaptureconnection.videoOrientation в зависимости от ориентации устройства, так как имеется некоторая «остановка потока основного интерфейса».

Я прочитал, что лучшим решением является поворот CMSampleBuffer в функции делегата AVCaptureVideoDataOutputSampleBufferDelegate captureOutput(...). Это выглядит немного сложно, документация Apple не очень помогает, и многие посты находятся в Objective-C.

Прежде чем идти по этому пути, я хотел бы знать, есть ли какие-то решения, которые я мог бы пропустить. Спасибо

1 Ответ

0 голосов
/ 21 мая 2019

Так как вы используете AVAssetWriter.Попробуйте, 

  private let assetWriter: AVAssetWriter
  private var adaptor: AVAssetWriterInputPixelBufferAdaptor

И инициализируйте AVAssetWriter следующим образом: 

adaptor = AVAssetWriterInputPixelBufferAdaptor(rotationAngle: AVCaptureDevice.correctOrientation)
assetWriter = AVAssetWriter(input: adaptor.assetWriterInput)

Создайте extension для AVCaptureDevice, Измените углы соответственно для поворота. 

// The angle by which to rotate captured media, based on the current orientation, so that it looks correctly oriented to the user.
    var correctOrientation: CGFloat {
        let angle: CGFloat
        switch(UIDevice.current.orientation) {
         case .portraitUpsideDown: angle = -CGFloat.pi / 2 // Play around with these values
         case .landscapeLeft: angle = -CGFloat.pi
         case .landscapeRight: angle = 0
         case .portrait: angle = CGFloat.pi / 2
         default: angle = CGFloat.pi / 2
       }
       return angle
    }

Создать еще один extension для AVAssetWriterInputPixelBufferAdaptor

extension AVAssetWriterInputPixelBufferAdaptor {
  convenience init(rotationAngle: CGFloat) {
    let input = AVAssetWriterInput(width: UIDevice.activeFormat.width, height: UIDevice.activeFormat.height)
    input.transform = CGAffineTransform(rotationAngle: rotationAngle)

    self.init(
      assetWriterInput: input,
      sourcePixelBufferAttributes: [
        kCVPixelBufferPixelFormatTypeKey as String: kCVPixelFormatType_32BGRA, // use whatever format you used
        kCVPixelBufferWidthKey as String: UIDevice.activeFormat.width,
        kCVPixelBufferHeightKey as String: UIDevice.activeFormat.height])
  }
...