nv12 yuv420p的相互转换

//初始化输入,输出的类型
-(void)initCapureSession{
    //创建AVCaptureDevice的视频设备对象
    AVCaptureDevice* videoDevice = [AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeVideo];
    
    NSError* error;
    //创建视频输入端对象
    AVCaptureDeviceInput* input = [AVCaptureDeviceInput deviceInputWithDevice:videoDevice error:&error];
    if (error) {
        NSLog(@"创建输入端失败,%@",error);
        return;
    }
    
    //创建功能会话对象
    self.captureSession = [[AVCaptureSession alloc] init];
    //设置会话输出的视频分辨率
    [self.captureSession setSessionPreset:AVCaptureSessionPreset1280x720];
    
    //添加输入端
    if (![self.captureSession canAddInput:input]) {
        NSLog(@"输入端添加失败");
        return;
    }
    [self.captureSession addInput:input];
    
    //显示摄像头捕捉到的数据
    AVCaptureVideoPreviewLayer* layer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.captureSession];
    layer.frame = CGRectMake(0, 0, self.view.frame.size.width, self.view.frame.size.height - 100);
    [self.view.layer addSublayer:layer];
    
    //创建输出端
    
    AVCaptureVideoDataOutput *videoDataOutput = [[AVCaptureVideoDataOutput alloc] init];
    NSString* key = (NSString*)kCVPixelBufferPixelFormatTypeKey;
    NSNumber* value = [NSNumber
                       numberWithUnsignedInt:kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange];
    NSDictionary* videoSettings = [NSDictionary
                                   dictionaryWithObject:value forKey:key];
    [videoDataOutput setVideoSettings:videoSettings];
    //会话对象添加输出端
    if ([self.captureSession canAddOutput:videoDataOutput]) {
        [self.captureSession addOutput:videoDataOutput];
        self.videoDataOutput = videoDataOutput;
        //创建输出调用的队列
        dispatch_queue_t videoDataOutputQueue = dispatch_queue_create("videoDataOutputQueue", DISPATCH_QUEUE_SERIAL);
        self.videoDataOutputQueue = videoDataOutputQueue;
        //设置代理和调用的队列
        [self.videoDataOutput setSampleBufferDelegate:self queue:videoDataOutputQueue];
        //设置延时丢帧
        self.videoDataOutput.alwaysDiscardsLateVideoFrames = NO;
    }
}


#pragma mark - AVCaptureVideoDataOutputSampleBufferDelegate
- (void)captureOutput:(AVCaptureOutput *)output didOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection {
    NSLog(@"did get %@",output);
    NSData *data22 = [self convertVideoSmapleBufferToYuvData:sampleBuffer];
    [self.h264Tool addFrame:sampleBuffer];
}
// CMSampleBufferRef代表一帧图像,调用苹果提供的api可以获取具体的yuv的数据
-(NSData *) convertVideoSmapleBufferToYuvData:(CMSampleBufferRef) videoSample{
    // 获取yuv数据
    // 通过CMSampleBufferGetImageBuffer方法,获得CVImageBufferRef。
    // 这里面就包含了yuv420(NV12)数据的指针
    CVImageBufferRef pixelBuffer = CMSampleBufferGetImageBuffer(videoSample);
    
    //表示开始操作数据
    CVPixelBufferLockBaseAddress(pixelBuffer, 0);
    
    //图像宽度(像素)
    size_t pixelWidth = CVPixelBufferGetWidth(pixelBuffer);
    //图像高度(像素)
    size_t pixelHeight = CVPixelBufferGetHeight(pixelBuffer);
    //yuv中的y所占字节数
    size_t y_size = pixelWidth * pixelHeight;
    //yuv中的uv所占的字节数
    size_t uv_size = y_size / 2;
    
    uint8_t *yuv_frame = malloc(uv_size + y_size);
    
    //获取CVImageBufferRef中的y数据
    uint8_t *y_frame = CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 0);
    memcpy(yuv_frame, y_frame, y_size);
    
    //获取CMVImageBufferRef中的uv数据
    uint8_t *uv_frame = CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 1);
    memcpy(yuv_frame + y_size, uv_frame, uv_size);
    
    CVPixelBufferUnlockBaseAddress(pixelBuffer, 0);
    uint8_t *I420 = malloc(uv_size + y_size);
    NSString *path = [self getHomePath];
    const char *resultCString = NULL;
    if ([path canBeConvertedToEncoding:NSUTF8StringEncoding]) {
        resultCString = [path cStringUsingEncoding:NSUTF8StringEncoding];
    }
    NSString *path2 = [self getHome2Path];
    const char *resultCString2 = NULL;
    if ([path2 canBeConvertedToEncoding:NSUTF8StringEncoding]) {
        resultCString2 = [path2 cStringUsingEncoding:NSUTF8StringEncoding];
    }
    dispatch_sync(dispatch_get_main_queue(), ^{
        func(yuv_frame, I420,pixelWidth,pixelHeight,resultCString,resultCString2);
        func5(yuv_frame, I420,pixelWidth,pixelHeight,resultCString,resultCString2);
    });
    
    //返回数据
    return [NSData dataWithBytesNoCopy:yuv_frame length:y_size + uv_size];
}
// nv12 写入本地
void func0(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    NSLog(@"start");
    int len = (int)pixelHeight * pixelWidth + pixelHeight * pixelWidth / 2;
    FILE* fpyuv = fopen(resultCString, "wb");
    for (int i = 0; i < len; i ++) {
        fwrite(NV12, 1, 1, fpyuv);
        NV12 ++;
    }
    fclose(fpyuv);
    free(NV12);
    NSLog(@"end");
}
// nv12 转 yuv 并写入本地
void func(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    unsigned char *ybuf = (unsigned char *)malloc(pixelHeight * pixelWidth);
    unsigned char *ubuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *vbuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *y = ybuf;
    unsigned char *u = ubuf;
    unsigned char *v = vbuf;
    int len = (int)pixelHeight * pixelWidth + pixelHeight * pixelWidth / 2;
    FILE* fpyuv = fopen(resultCString, "wb");
    for (int i = 0; i < len; i ++) {
        if (i < (pixelWidth * pixelHeight)) {  //yyyyy
            *y = *NV12;
            y++;
        }
        else{//uvuvuv
            if (i % 2 == 0) {//uuuuu
                *u = *NV12;
                u ++;
            }else{//vvvvv
                *v = *NV12;
                v ++;
            }
        }
        NV12 ++;
    }
    fwrite(ybuf, 1, pixelWidth*pixelHeight, fpyuv);
    fwrite(ubuf, 1, pixelWidth*pixelHeight/4, fpyuv);
    fwrite(vbuf, 1, pixelWidth*pixelHeight/4, fpyuv);
    fclose(fpyuv);
    free(ybuf);
    free(ubuf);
    free(vbuf);
}
// 从本地文件获取yuv的数据 并实现180度的翻转 并写入本地
void func2(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    unsigned char *ybuf = (unsigned char *)malloc(pixelHeight * pixelWidth);
    unsigned char *ubuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *vbuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *y = ybuf;
    unsigned char *u = ubuf;
    unsigned char *v = vbuf;
    FILE* fpyuv = fopen(resultCString, "rb");
    FILE* fpyuv2 = fopen(resultCString2, "wb");
    uint8_t* yuvbuf = malloc(pixelHeight * pixelWidth * 3 / 2);
    fread(yuvbuf, pixelWidth*pixelHeight * 3 / 2, 1, fpyuv);
    fclose(fpyuv);
    // y
    for (int i = pixelHeight -1; i >= 0; i --) {
        for (int j = pixelWidth - 1; j >= 0; j--) {
            *(y++) = *(yuvbuf +(i*pixelWidth + j));
        }
    }
    //U
    uint8_t* uheader = yuvbuf + pixelWidth*pixelHeight;
    for (int i = pixelHeight/2 - 1; i >= 0; i--){
        for (int j = pixelWidth/2-1 ; j >= 0; j--){
            *(u++) = *(uheader +(i*pixelWidth/2 + j));
        }
    }
    //v
    uint8_t* vheader = uheader + pixelWidth*pixelHeight/4;
    for (int i = pixelHeight/2 - 1; i >= 0; i--){
        for (int j = pixelWidth/2-1 ; j >= 0; j--){
            *(v++) = *(vheader +(i*pixelWidth/2 + j));
        }
    }
    fwrite(ybuf, 1, pixelWidth*pixelHeight, fpyuv2);
    fwrite(ubuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fwrite(vbuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fclose(fpyuv2);
    free(ybuf);
    free(ubuf);
    free(vbuf);
}
// 从本地文件获取yuv的数据 并实现逆时针90度的翻转 并写入本地
void func3(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    unsigned char *ybuf = (unsigned char *)malloc(pixelHeight * pixelWidth);
    unsigned char *ubuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *vbuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *y = ybuf;
    unsigned char *u = ubuf;
    unsigned char *v = vbuf;
    FILE* fpyuv = fopen(resultCString, "rb");
    FILE* fpyuv2 = fopen(resultCString2, "wb");
    uint8_t* yuvbuf = malloc(pixelHeight * pixelWidth * 3 / 2);
    fread(yuvbuf, pixelWidth*pixelHeight * 3 / 2, 1, fpyuv);
    fclose(fpyuv);
    for (int i = 0; i < pixelWidth; i ++) {
        for (int j = pixelHeight - 1; j >= 0; j --) {
            *(y++) = *(yuvbuf +(j*pixelWidth +i));
        }
    }
    uint8_t* uheader = yuvbuf + pixelWidth*pixelHeight;
    for (int i = 0; i < pixelWidth/2; i ++) {
        for (int j = pixelHeight/2 - 1; j >= 0; j --) {
            *(u++) = *(uheader +(j*pixelWidth/2 + i));
        }
    }
    uint8_t* vheader = uheader + pixelWidth*pixelHeight/4;
    for (int i = 0; i < pixelWidth/2; i ++) {
        for (int j = pixelHeight/2 - 1; j >= 0; j --) {
            *(v++) = *(vheader +(j*pixelWidth/2 + i));
        }
    }
    fwrite(ybuf, 1, pixelWidth*pixelHeight, fpyuv2);
    fwrite(ubuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fwrite(vbuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fclose(fpyuv2);
    free(ybuf);
    free(ubuf);
    free(vbuf);
}
// 从本地文件获取yuv的数据 并实现顺时针90度的翻转 并写入本地
void func4(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    unsigned char *ybuf = (unsigned char *)malloc(pixelHeight * pixelWidth);
    unsigned char *ubuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *vbuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *y = ybuf;
    unsigned char *u = ubuf;
    unsigned char *v = vbuf;
    FILE* fpyuv = fopen(resultCString, "rb");
    FILE* fpyuv2 = fopen(resultCString2, "wb");
    uint8_t* yuvbuf = malloc(pixelHeight * pixelWidth * 3 / 2);
    fread(yuvbuf, pixelWidth*pixelHeight * 3 / 2, 1, fpyuv);
    fclose(fpyuv);
    for (int i = pixelWidth - 1; i >= 0; i --) {
        for (int j = 0; j < pixelHeight; j ++) {
            *(y++) = *(yuvbuf +(j*pixelWidth +i));
        }
    }
    uint8_t* uheader = yuvbuf + pixelWidth*pixelHeight;
    for (int i = pixelWidth/2 - 1; i >= 0; i --) {
        for (int j = 0; j < pixelHeight/2; j ++) {
            *(u++) = *(uheader +(j*pixelWidth/2 +i));
        }
    }
    uint8_t* vheader = uheader + pixelWidth*pixelHeight/4;
    for (int i = pixelWidth/2 - 1; i >= 0; i --) {
        for (int j = 0; j < pixelHeight/2; j ++) {
            *(v++) = *(vheader +(j*pixelWidth/2 +i));
        }
    }
    fwrite(ybuf, 1, pixelWidth*pixelHeight, fpyuv2);
    fwrite(ubuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fwrite(vbuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fclose(fpyuv2);
    free(ybuf);
    free(ubuf);
    free(vbuf);
}
// 从本地文件获取yuv的数据 制作水平方向的镜像 并写入本地
void func5(uint8_t *NV12, uint8_t *I420,size_t pixelWidth,size_t pixelHeight,const char *resultCString,const char *resultCString2){
    unsigned char *ybuf = (unsigned char *)malloc(pixelHeight * pixelWidth);
    unsigned char *ubuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *vbuf = (unsigned char *)malloc((pixelHeight/2) * (pixelWidth/2));
    unsigned char *y = ybuf;
    unsigned char *u = ubuf;
    unsigned char *v = vbuf;
    FILE* fpyuv = fopen(resultCString, "rb");
    FILE* fpyuv2 = fopen(resultCString2, "wb");
    uint8_t* yuvbuf = malloc(pixelHeight * pixelWidth * 3 / 2);
    fread(yuvbuf, pixelWidth*pixelHeight * 3 / 2, 1, fpyuv);
    fclose(fpyuv);
    for (int i = 0; i < pixelHeight - 1; i ++) {
        for (int j = pixelWidth - 1; j >= 0; j --) {
            *(y++) = *(yuvbuf +(i*pixelWidth +j));
        }
    }
    uint8_t* uheader = yuvbuf + pixelWidth*pixelHeight;
    for (int i = 0; i < pixelHeight/2 - 1; i ++) {
        for (int j = pixelWidth/2 - 1; j >= 0; j --) {
            *(u++) = *(uheader +(i*pixelWidth/2 +j));
        }
    }
    uint8_t* vheader = uheader + pixelWidth*pixelHeight/4;
    for (int i = 0; i < pixelHeight/2 - 1; i ++) {
        for (int j = pixelWidth/2 - 1; j >= 0; j --) {
            *(v++) = *(vheader +(i*pixelWidth/2 +j));
        }
    }
    fwrite(ybuf, 1, pixelWidth*pixelHeight, fpyuv2);
    fwrite(ubuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fwrite(vbuf, 1, pixelWidth*pixelHeight/4, fpyuv2);
    fclose(fpyuv2);
    free(ybuf);
    free(ubuf);
    free(vbuf);
}
- (NSString *)getHomePath{
    NSString *path = NSHomeDirectory();
    
    NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
    NSString *docDir = [paths objectAtIndex:0];
    return [NSString stringWithFormat:@"%@/outforyuv420p_new_new.yuv",docDir];
}
- (NSString *)getHome2Path{
    NSString *path = NSHomeDirectory();
    
    NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
    NSString *docDir = [paths objectAtIndex:0];
    return [NSString stringWithFormat:@"%@/outforyuv420p_new_new_rala.yuv",docDir];
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 151,511评论 1 330
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 64,495评论 1 273
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 101,595评论 0 225
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 42,558评论 0 190
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 50,715评论 3 270
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 39,672评论 1 192
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,112评论 2 291
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 29,837评论 0 181
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 33,417评论 0 228
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 29,928评论 2 232
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,316评论 1 242
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 27,773评论 2 234
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,253评论 3 220
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 25,827评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,440评论 0 180
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 34,523评论 2 249
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 34,583评论 2 249

推荐阅读更多精彩内容