ffmpeg 内存读写相关

需要的解码的视频数据在一段内存中。例如,通过其他系统送来的视频数据。同样,有的时候编码后的视频数据也未必要保存成一个文件。例如,要求将编码后的视频数据送给其他的系统进行下一步的处理。以上两种情况就要求FFmpeg不仅仅是对文件进行“读,写”操作,而是要对内存进行“读,写”操作。

从内存中读取数据

经过 分析ffmpeg的源代码 ,发现其是可以从内存中读取数据的:

  1. AVFormatContext *ic = NULL;  
  2. ic = avformat_alloc_context();  

  1. unsigned char * iobuffer=(unsigned char *)av_malloc(32768);  
  2. AVIOContext *avio =avio_alloc_context(iobuffer, 32768,0,NULL,fill_iobuffer,NULL,NULL);  
  3. ic->pb=avio;  
  4. err = avformat_open_input(&ic, "nothing", NULL, NULL);  


关键要在avformat_open_input() 之前 初始化一个AVIOContext, 而且将原本的AVFormatContext的指针pb(AVIOContext类型)指向这个自行初始化的AVIOContext。当自行指定了AVIOContext之后,avformat_open_input()里面的 URL参数就不起作用了 。示例代码开辟了一块空间iobuffer作为AVIOContext的 缓存

fill_iobuffer是将数据读取至iobuffer的回调函数。fill_iobuffer()形式(参数,返回值) 是固定的,是一个回调函数, 如下所示(只是个例子,具体怎么读取数据可以自行设计)。示例中回调函数将文件中的内容通过fread() 读入内存

  1. //读取数据的回调函数-------------------------  
  2. //AVIOContext使用的回调函数!  
  3. //注意:返回值是读取的字节数  
  4. //手动初始化AVIOContext只需要两个东西:内容来源的buffer,和读取这个Buffer到FFmpeg中的函数  
  5. //回调函数,功能就是:把buf_size字节数据送入buf即可  
  6. //第一个参数(void *opaque)一般情况下可以不用  
  7. int fill_iobuffer(void * opaque,uint8_t *buf, int bufsize){  
  8.     if(!feof(fp_open)){  
  9.         int true_size=fread(buf,1,buf_size,fp_open);  
  10.         return true_size;  
  11.     }else{  
  12.         return -1;  
  13.     }  
  14. }  




整体结构大致如下:

  1. FILE *fp_open;  
  2.   
  3. int fill_iobuffer(void *opaque, uint8_t *buf, int buf_size){  
  4. ...  
  5. }  
  6.   
  7. int main(){  
  8.     ...  
  9.     fp_open=fopen("test.h264","rb+");  
  10.     AVFormatContext *ic = NULL;  
  11.     ic = avformat_alloc_context();  
  12.     unsigned char * iobuffer=(unsigned char *)av_malloc(32768);  
  13.     AVIOContext *avio =avio_alloc_context(iobuffer, 32768,0,NULL,fill_iobuffer,NULL,NULL);  
  14.     ic->pb=avio;  
  15.     err = avformat_open_input(&ic, "nothing", NULL, NULL);  
  16.     ...//解码  
  17. }  




将数据输出到内存
也可以将处理后的数据输出到内存。
回调函数如下示例,可以将输出到内存的数据写入到文件中。

  1. //写文件的回调函数  
  2. int write_buffer(void *opaque, uint8_t *buf, int buf_size){  
  3.     if(!feof(fp_write)){  
  4.         int true_size=fwrite(buf,1,buf_size,fp_write);  
  5.         return true_size;  
  6.     }else{  
  7.         return -1;  
  8.     }  
  9. }  


主函数如下所示。

  1. FILE *fp_write;  
  2.   
  3. int write_buffer(void *opaque, uint8_t *buf, int buf_size){  
  4. ...  
  5. }  
  6.   
  7. main(){  
  8.     ...  
  9.     fp_write=fopen("src01.h264","wb+"); //输出文件  
  10.     ...  
  11.     AVFormatContext* ofmt_ctx=NULL;  
  12.     avformat_alloc_output_context2(&ofmt_ctx, NULL, "h264", NULL);  
  13.     unsigned char* outbuffer=(unsigned char*)av_malloc(32768);  
  14.   
  15.     AVIOContext *avio_out =avio_alloc_context(outbuffer, 32768,0,NULL,NULL,write_buffer,NULL);    
  16.   
  17.     ofmt_ctx->pb=avio_out;   
  18.     ofmt_ctx->flags=AVFMT_FLAG_CUSTOM_IO;  
  19.     ...  
  20. }  




关键点
关键点就两个:
1.      初始化自定义的AVIOContext,指定自定义的回调函数。示例代码如下:

  1. //AVIOContext中的缓存  
  2. unsigned char *aviobuffer=(unsigned char*)av_malloc(32768);  
  3. AVIOContext *avio=avio_alloc_context(aviobuffer, 32768,0,NULL,read_buffer,NULL,NULL);  
  4. pFormatCtx->pb=avio;  
  5.    
  6. if(avformat_open_input(&pFormatCtx,NULL,NULL,NULL)!=0){  
  7.            printf("Couldn't open inputstream.(无法打开输入流)\n");  
  8.            return -1;  
  9. }  


上述代码中,自定义了回调函数read_buffer()。在使用avformat_open_input()打开媒体数据的时候,就可以不指定文件的URL了,即其 第2个参数为NULL (因为数据不是靠文件读取,而是由read_buffer()提供)
2.      自己写回调函数。示例代码如下:

  1. //Callback  
  2. int read_buffer(void *opaque, uint8_t *buf, int buf_size){  
  3.     if(!feof(fp_open)){  
  4.         int true_size=fread(buf,1,buf_size,fp_open);  
  5.         return true_size;  
  6.     }else{  
  7.         return -1;  
  8.     }  
  9. }  


当系统需要数据的时候, 会自动调用 该回调函数以获取数据。 这个例子为了简单,直接使用fread()读取数据至内存。回调函数需要 格外注意 它的参数和返回值。



avio_alloc_context第3个参数 Set to 1 if the buffer should be writable, 0 otherwise. 为0时表示ffmpeg要从回调函数里 读取 数据。为1表示ffmpeg会将数据输出到回调函数中。



参考:
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值