如何使用AVFoundation设置逐帧生成的视频的方向？

import AVFoundation

func generateVideo(withFrames frames: [UIImage], outputURL: URL, orientation: AVCaptureVideoOrientation) {
    let videoSettings: [String: Any] = [
        AVVideoCodecKey: AVVideoCodecType.h264,
        AVVideoWidthKey: 720,
        AVVideoHeightKey: 1280,
        AVVideoCompressionPropertiesKey: [
            AVVideoAverageBitRateKey: 2300000
        ]
    ]

    let assetWriter = try! AVAssetWriter(outputURL: outputURL, fileType: .mp4)
    let input = AVAssetWriterInput(mediaType: .video, outputSettings: videoSettings)
    input.expectsMediaDataInRealTime = true

    assetWriter.add(input)
    assetWriter.startWriting()
    assetWriter.startSession(atSourceTime: CMTime.zero)

    let pixelBufferAdaptor = AVAssetWriterInputPixelBufferAdaptor(assetWriterInput: input, sourcePixelBufferAttributes: nil)

    for (index, frame) in frames.enumerated() {
        var pixelBuffer: CVPixelBuffer?
        let options = [kCVPixelBufferCGImageCompatibilityKey as String: true, kCVPixelBufferCGBitmapContextCompatibilityKey as String: true]
        let status = CVPixelBufferCreate(kCFAllocatorDefault, Int(frame.size.width), Int(frame.size.height), kCVPixelFormatType_32ARGB, options as CFDictionary, &pixelBuffer)

        if status == kCVReturnSuccess {
            let managedBuffer = pixelBuffer!
            CVPixelBufferLockBaseAddress(managedBuffer, CVPixelBufferLockFlags(rawValue: 0))
            let pixelData = CVPixelBufferGetBaseAddress(managedBuffer)
            let rgbColorSpace = CGColorSpaceCreateDeviceRGB()
            let context = CGContext(data: pixelData, width: Int(frame.size.width), height: Int(frame.size.height), bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(managedBuffer), space: rgbColorSpace, bitmapInfo: CGImageAlphaInfo.noneSkipFirst.rawValue)

            context?.translateBy(x: 0, y: frame.size.height)
            context?.scaleBy(x: 1.0, y: -1.0)
            context?.concatenate(CGAffineTransform(rotationAngle: orientation.rawValue))
            context?.draw(frame.cgImage!, in: CGRect(x: 0, y: 0, width: frame.size.width, height: frame.size.height))

            CVPixelBufferUnlockBaseAddress(managedBuffer, CVPixelBufferLockFlags(rawValue: 0))
            pixelBufferAdaptor.append(managedBuffer, withPresentationTime: CMTimeMake(value: index, timescale: 30))
        }
    }

    input.markAsFinished()
    assetWriter.finishWriting {
        print("Video generated successfully")
    }
}