iOS-視頻硬編/解碼(對幀圖片編輯)

開發(fā)遇到這樣一個需求,需要對一個視頻進行解碼,解碼生成的每一幀圖片,都要對其進行編輯,生成一張新的圖片,在實時編碼成一個新的視頻。

這里我們分為三步走

1. 對視頻進行解碼

我這邊采用的是iOS利用VideoToolbox實現(xiàn)視頻硬解碼中的方式,demo中提供了兩種解碼方式,一種是第三方庫ffmpeg,另外一種是系統(tǒng)的videotoolbox庫,效率基本上都差不多,作者封裝后都會有一個代理回調(diào),

// XDXVideoDecoderDelegate
- (void)getVideoDecodeDataCallback:(CMSampleBufferRef)sampleBuffer isFirstFrame:(BOOL)isFirstFrame;
// XDXFFmpegVideoDecoderDelegate
- (void)getDecodeVideoDataByFFmpeg:(CMSampleBufferRef)sampleBuffer;

兩種回調(diào)都會有一個CMSampleBufferRef的對象,它是像素緩沖,其實就是我們解碼后的每一幀的圖片。

2. 合成圖片

在第一步中我們已經(jīng)拿到了每一幀的圖片,現(xiàn)在我們在其基礎上合成本地圖片,合成圖片我首先想到的是下面的方法,

- (UIImage *)composeImg:(UIImage *)image1 coverImage:(UIImage *)image2 coverRect:(CGRect)rect {
    CGImageRef imgRef1 = image1.CGImage;
    CGFloat w1 = CGImageGetWidth(imgRef1);
    CGFloat h1 = CGImageGetHeight(imgRef1);
    
    //以1的圖大小為畫布創(chuàng)建上下文
    UIGraphicsBeginImageContext(CGSizeMake(w1, h1));
    [image1 drawInRect:CGRectMake(0, 0, w1, h1)];//先把1.png 畫到上下文中
    [image2 drawInRect:rect];//再把小圖放在上下文中
    UIImage *resultImg = UIGraphicsGetImageFromCurrentImageContext();//從當前上下文中獲得最終圖片
    UIGraphicsEndImageContext();//關閉上下文
    CGImageRelease(imgRef1);
    
    return resultImg;
}

這個方法是通過兩個image的對象,最終合成一個新的image,效果可以實現(xiàn),但是在我們這個場景中,我們在對視頻解碼時,是一個連續(xù)異步的操作,代理會不斷的回調(diào),此時如果采用這個方式,會多次創(chuàng)建局部對象,不能及時釋放,導致內(nèi)存爆增,引起閃退。所以我采用以下方式

-(void)getDecodeVideoDataByFFmpeg:(CMSampleBufferRef)sampleBuffer {
    CVPixelBufferRef pix = CMSampleBufferGetImageBuffer(sampleBuffer);
    [self.previewView displayPixelBuffer:pix];
    count ++;
    NSLog(@"count ==== %ld" , (long)count);
    
    UIImage *coverImage = [UIImage imageWithContentsOfFile:[[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"%ld", (long)count] ofType:@"png"]];

    [[VideoCreator manager] appendPixelBuffer2:pix overlayImage:coverImage overlayFrame:[self getCoverRect:count] frameCounter:1];
    
}

這里我用的是CIImage,它提供了一個系統(tǒng)的api

// CIImage提供的api,把一個image放在buffer上面,bounds可以截取image的區(qū)域,并不能改變他的位置,colorSpace是渲染格式一般是CGColorSpaceCreateDeviceRGB()
- (void)render:(CIImage *)image
toCVPixelBuffer:(CVPixelBufferRef)buffer
        bounds:(CGRect)bounds
    colorSpace:(nullable CGColorSpaceRef)colorSpace NS_AVAILABLE(10_11,5_0);
- (void)appendPixelBuffer2:(CVPixelBufferRef)pixelBuffer overlayImage:(UIImage *)overlayImage overlayFrame:(CGRect)overlayFrame frameCounter:(int)frameCounter {
    int bufferW = (int)CVPixelBufferGetWidth(pixelBuffer);
    int bufferH = (int)CVPixelBufferGetHeight(pixelBuffer);
    CVPixelBufferLockBaseAddress( pixelBuffer, 0 );

    CGColorSpaceRef cSpace = CGColorSpaceCreateDeviceRGB();

    CIImage *ciimage = [self imageWithBuffer:pixelBuffer Size:CGSizeMake(bufferW, bufferH) waterImg:overlayImage waterFrame:overlayFrame];

    [_ciContext render:ciimage toCVPixelBuffer:pixelBuffer bounds:ciimage.extent colorSpace:cSpace];

    [self appendPixelBuffer:pixelBuffer frameCounter:frameCounter];

    CVPixelBufferUnlockBaseAddress(pixelBuffer, 0);
    CGColorSpaceRelease(cSpace);
}
/// 特別要注意這里的autoreleasepool,因為在arc的狀態(tài)下,oc對象系統(tǒng)會及時釋放,但是UIGraphicsBeginImageContext這樣的創(chuàng)建的c的對象,并不能及時釋放,所以要把此類對象放到緩存池里面,不使用時能夠釋放掉。
- (CIImage *)imageWithBuffer:(CVPixelBufferRef)buffer Size:(CGSize)layerSize  waterImg:(UIImage *)waterImg waterFrame:(CGRect)rect{
    @autoreleasepool{
        UIImage *bufferImage = [UIImage imageFromPixelBuffer:buffer];
        UIGraphicsBeginImageContext(layerSize);

        [bufferImage drawInRect:CGRectMake(0, 0, layerSize.width, layerSize.height)];
        
        [waterImg drawInRect:rect];
        
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        
        UIGraphicsEndImageContext();
        
        CIImage *ciimage = [CIImage imageWithCGImage:image.CGImage];
        
        return  ciimage;
    }
}

3. 合成視頻

- (void)appendPixelBuffer:(CVPixelBufferRef)pixelBuffer frameCounter:(int)frameCounter {
    CMTime frameTime = CMTimeMake(_frameCount, self.fps);
    
    if (!_adaptor.assetWriterInput.readyForMoreMediaData) {
        return;
    }
    if (pixelBuffer == NULL) {
        return;
    }
    
    if (![_adaptor appendPixelBuffer:pixelBuffer withPresentationTime:frameTime]) {
        NSLog(@"保存視頻失敗");
        if (self.inputFieldBlcok) {
            self.inputFieldBlcok();
        }
    }else {
        NSLog(@"保存視頻成功");
        if (self.inputSuccessBlcok) {
            self.inputSuccessBlcok();
        }
    }
    
    _frameCount += frameCounter;
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容