[iOS]利用libyuv 转化视频分辨率(nv12格式)

libyuv下载地址
链接: https://pan.baidu.com/s/1DTInjVivZHQW0PnzJYQK5g 提取码: cy4b

我也是在网上查找了好多东西,自己试出来可以正常转化将nv12数据分辨率的方法,有的地方也并不能完全解释,但确实可行,仅供参考吧,也是自己记录下,因为这个过程真的好辛苦,从找到nv12转化分辨率到将yuv转化成CVPixelBufferRef,再到CMSampleBufferRef.

过程:得到CMSampleBufferRef(此处得到的是420f,也是nv12,也是kCVPixelFormatType_420YpCbCr8BiPlanarFullRange),解析得到nv12格式yuv数据.利用libyuv ,先将nv12转化成I420,通过I420转化分辨率,再将I420转化成nv12,用到的libyuv的方法有三个:NV12ToI420,I420Scale,I420ToNV12.
首先我运用的场景是通过系统的录屏方法获取到的CMSampleBufferRef,得到的视频流分辨率是1080*1920,但是我需要的是720 *1280.

[[RPScreenRecorder sharedRecorder] startCaptureWithHandler:^(CMSampleBufferRef  _Nonnull sampleBuffer, RPSampleBufferType bufferType, NSError * _Nullable error) {
            
            if (bufferType == RPSampleBufferTypeVideo && sampleBuffer != nil) {
                
                CVPixelBufferRef pixel =[self convertVideoSmapleBufferToYuvData:sampleBuffer];
                CMSampleBufferRef sample = [self pixelBufferToSampleBuffer:pixel];
                [self setwangyiBuffer:sample];
            }
        } completionHandler:^(NSError * _Nullable error) {}];

其中 [[RPScreenRecorder sharedRecorder] startCaptureWithHandler:^(CMSampleBufferRef _Nonnull sampleBuffer, RPSampleBufferType bufferType, NSError * _Nullable error) 是系统的录屏方法,可以直接得到录屏数据的视频流,CVPixelBufferRef pixel =[self convertVideoSmapleBufferToYuvData:sampleBuffer];
是我自己转化调用的方法其中shareSWidth = 720;
shareSHeight = 1280;

//转化
-(CVPixelBufferRef)convertVideoSmapleBufferToYuvData:(CMSampleBufferRef) videoSample{
    
//    1.
    //CVPixelBufferRef是CVImageBufferRef的别名,两者操作几乎一致。
    //获取CMSampleBuffer的图像地址
    CVImageBufferRef pixelBuffer = CMSampleBufferGetImageBuffer(videoSample);
    //表示开始操作数据
    CVPixelBufferLockBaseAddress(pixelBuffer, 0);
    //图像宽度(像素)
    size_t pixelWidth = CVPixelBufferGetWidth(pixelBuffer);
    //图像高度(像素)
    size_t pixelHeight = CVPixelBufferGetHeight(pixelBuffer);
    //获取CVImageBufferRef中的y数据
    uint8_t *y_frame = (unsigned char *)CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 0);
    //获取CMVImageBufferRef中的uv数据
    uint8_t *uv_frame =(unsigned char *) CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 1);
    
    //y stride
    size_t plane1_stride = CVPixelBufferGetBytesPerRowOfPlane (pixelBuffer, 0);
    //uv stride
    size_t plane2_stride = CVPixelBufferGetBytesPerRowOfPlane (pixelBuffer, 1);
    //y_size
    size_t plane1_size = plane1_stride * CVPixelBufferGetHeightOfPlane(pixelBuffer, 0);
    //uv_size
    size_t plane2_size = CVPixelBufferGetBytesPerRowOfPlane (pixelBuffer, 1) * CVPixelBufferGetHeightOfPlane(pixelBuffer, 1);
    //yuv_size(内存空间)
    size_t frame_size = plane1_size + plane2_size;
    
    //开辟frame_size大小的内存空间用于存放转换好的i420数据
    uint8* buffer = (unsigned char *)malloc(frame_size);
    //buffer为这段内存的首地址,plane1_size代表这一帧中y数据的长度
    uint8* dst_u = buffer + plane1_size;
    //dst_u为u数据的首地,plane1_size/4为u数据的长度
    uint8* dst_v = dst_u + plane1_size/4;
    
    
    // Let libyuv convert
    libyuv::NV12ToI420(y_frame,plane1_stride,
             uv_frame, plane2_stride,
               buffer, plane1_stride,
            dst_u,plane2_stride/2,
              dst_v, plane2_stride/2,
               pixelWidth, pixelHeight);
    
    
//    2
    //scale-size
    int scale_yuvBufSize = shareSWidth * shareSHeight * 3 / 2;
    //uint8_t* scale_yuvBuf= new uint8_t[scale_yuvBufSize];
    uint8* scale_yuvBuf = (unsigned char *)malloc(scale_yuvBufSize);

    //scale-stride
    const int32 scale_uv_stride = (shareSWidth + 1) / 2;

    //scale-length
    const int scale_y_length = shareSWidth * shareSHeight;
    int scale_uv_length = scale_uv_stride * ((shareSWidth+1) / 2);

    unsigned char *scale_Y_data_Dst = scale_yuvBuf;
    unsigned char *scale_U_data_Dst = scale_yuvBuf + scale_y_length;
    unsigned char *scale_V_data_Dst = scale_U_data_Dst + scale_y_length/4;


    libyuv::I420Scale(buffer, plane1_stride, dst_u, plane2_stride/2, dst_v, plane2_stride/2, pixelWidth, pixelHeight, scale_Y_data_Dst, shareSWidth,
                      scale_U_data_Dst, scale_uv_stride,
                      scale_V_data_Dst, scale_uv_stride,
                      shareSWidth, shareSHeight,
                      libyuv::kFilterNone);


//    3.
    uint8 *dst_y = (uint8 *)malloc((shareSWidth * shareSHeight * 3) >> 1);
    int dst_Stride_Y = shareSWidth;
    uint8 *dst_uv = dst_y + shareSWidth*shareSHeight;
    int dst_Stride_uv = shareSWidth/2;

    libyuv::I420ToNV12(scale_Y_data_Dst, shareSWidth,
                       scale_U_data_Dst, scale_uv_stride,
                       scale_V_data_Dst, scale_uv_stride,dst_y, dst_Stride_Y, dst_uv, dst_Stride_Y,shareSWidth, shareSHeight);

    CVPixelBufferUnlockBaseAddress(pixelBuffer, 0);
    free(buffer);
    free(scale_yuvBuf);

    //转化
    NSDictionary *pixelAttributes = @{(id)kCVPixelBufferIOSurfacePropertiesKey : @{}};
    CVPixelBufferRef pixelBuffer1 = NULL;
    CVReturn result = CVPixelBufferCreate(kCFAllocatorDefault,
        shareSWidth,shareSHeight,kCVPixelFormatType_420YpCbCr8BiPlanarFullRange,
                                          (__bridge CFDictionaryRef)pixelAttributes,&pixelBuffer1);

    CVPixelBufferLockBaseAddress(pixelBuffer1, 0);
    uint8_t *yDestPlane = (uint8*)CVPixelBufferGetBaseAddressOfPlane(pixelBuffer1, 0);
    memcpy(yDestPlane, dst_y, shareSWidth * shareSHeight);
    uint8_t *uvDestPlane = (uint8*)CVPixelBufferGetBaseAddressOfPlane(pixelBuffer1, 1);
    memcpy(uvDestPlane, dst_uv, shareSWidth * shareSHeight/2);
    if (result != kCVReturnSuccess) {
        NSLog(@"Unable to create cvpixelbuffer %d", result);
    }
    CVPixelBufferUnlockBaseAddress(pixelBuffer1, 0);
    free(dst_y);
//    CVPixelBufferRelease(pixelBuffer1);

    return pixelBuffer1;
}

时间戳
-(CMSampleBufferRef)pixelBufferToSampleBuffer:(CVPixelBufferRef)pixelBuffer
{

    CMSampleBufferRef sampleBuffer;
    CMTime frameTime = CMTimeMakeWithSeconds([[NSDate date] timeIntervalSince1970], 1000000000);
    CMSampleTimingInfo timing = {frameTime, frameTime, kCMTimeInvalid};
    CMVideoFormatDescriptionRef videoInfo = NULL;
    CMVideoFormatDescriptionCreateForImageBuffer(NULL, pixelBuffer, &videoInfo);
    
    OSStatus status = CMSampleBufferCreateForImageBuffer(kCFAllocatorDefault, pixelBuffer, true, NULL, NULL, videoInfo, &timing, &sampleBuffer);
    if (status != noErr) {
        NSLog(@"Failed to create sample buffer with error %zd.", status);
    }
    CVPixelBufferRelease(pixelBuffer);
    if(videoInfo)
    CFRelease(videoInfo);
    
    return sampleBuffer;
}

参考链接
https://blog.csdn.net/sinat_36684217/article/details/75117920
https://www.jianshu.com/p/050234c5fff2
https://www.jianshu.com/p/68e05ad85490
https://www.jianshu.com/p/dac9857b34d0

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,558评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,002评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,024评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,144评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,255评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,295评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,068评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,478评论 1 305
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,789评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,965评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,649评论 4 336
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,267评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,982评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,800评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,847评论 2 351