执行CMSampleBuffer OpenGL Swift的平移/旋转

import AVFoundation
import OpenGLES

// 提取CMSampleBuffer中的图像数据
func extractImageFromSampleBuffer(sampleBuffer: CMSampleBuffer) -> UIImage? {
    guard let pixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else {
        return nil
    }
    
    CVPixelBufferLockBaseAddress(pixelBuffer, .readOnly)
    
    let baseAddress = CVPixelBufferGetBaseAddress(pixelBuffer)
    let bytesPerRow = CVPixelBufferGetBytesPerRow(pixelBuffer)
    let width = CVPixelBufferGetWidth(pixelBuffer)
    let height = CVPixelBufferGetHeight(pixelBuffer)
    
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let bitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedFirst.rawValue | CGBitmapInfo.byteOrder32Little.rawValue)
    let context = CGContext(data: baseAddress, width: width, height: height, bitsPerComponent: 8, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo.rawValue)
    
    guard let cgImage = context?.makeImage() else {
        CVPixelBufferUnlockBaseAddress(pixelBuffer, .readOnly)
        return nil
    }
    
    let image = UIImage(cgImage: cgImage)
    
    CVPixelBufferUnlockBaseAddress(pixelBuffer, .readOnly)
    
    return image
}

// 执行平移/旋转操作
func performTransform(image: UIImage, translation: CGPoint, rotation: CGFloat) -> UIImage? {
    guard let cgImage = image.cgImage else {
        return nil
    }
    
    let width = cgImage.width
    let height = cgImage.height
    
    let bytesPerPixel = 4
    let bytesPerRow = bytesPerPixel * width
    let bitsPerComponent = 8
    
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let bitmapInfo = CGBitmapInfo.byteOrder32Little.rawValue | CGImageAlphaInfo.premultipliedFirst.rawValue
    
    guard let context = CGContext(data: nil, width: width, height: height, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo) else {
        return nil
    }
    
    context.translateBy(x: translation.x, y: translation.y)
    context.rotate(by: rotation)
    
    context.draw(cgImage, in: CGRect(x: 0, y: 0, width: width, height: height))
    
    guard let outputCGImage = context.makeImage() else {
        return nil
    }
    
    let outputImage = UIImage(cgImage: outputCGImage)
    
    return outputImage
}

// 示例用法
if let sampleBuffer = // 从音视频源获取CMSampleBuffer
   let image = extractImageFromSampleBuffer(sampleBuffer: sampleBuffer) {
    let translation = CGPoint(x: 100, y: 100)
    let rotation = CGFloat.pi / 4
    
    if let transformedImage = performTransform(image: image, translation: translation, rotation: rotation) {
        // 处理后的图像数据
        // ...
    }
}