FFmpeg_03_视音频流分离

/******************* 程序功能： 获取多节目码流中的所有节目号，然后选定节目号，将该节目号对应的节目的视频流和音频流从码流中分离出来 *******************/ extern "C" { #include "libavformat\avformat.h" }; int PrintfMediaProgram(char* MediaName) { AVFormatContext *pFormatContext = NULL; pFormatContext = avformat_alloc_context(); if(avformat_open_input(&pFormatContext,MediaName,NULL,NULL)!= 0) { avformat_free_context(pFormatContext); return -1; } if(avformat_find_stream_info(pFormatContext,NULL) < 0) { avformat_close_input(&pFormatContext); return -2; } av_dump_format(pFormatContext,NULL,MediaName,0); for(int i=0;i<pFormatContext->nb_programs;i++) { printf("第%d个节目的节目号：%d\n",i+1,pFormatContext->programs[i]->id); } avformat_close_input(&pFormatContext); return 0; } int MediaSeparate(char *MediaName,int ProgramNumber,char *VideoName,char *AudioName) { AVFormatContext *pFormatContext = NULL; pFormatContext = avformat_alloc_context(); if(avformat_open_input(&pFormatContext,MediaName,NULL,NULL)!= 0) { avformat_free_context(pFormatContext); return -1; } if(avformat_find_stream_info(pFormatContext,NULL) < 0) { avformat_close_input(&pFormatContext); return -2; } int VideoIndex = -1,AudioIndex = -1; for(int i=0;i<pFormatContext->nb_programs;i++) { if(pFormatContext->programs[i]->id == ProgramNumber) { for(int j=0;j<pFormatContext->programs[i]->nb_stream_indexes;j++) { if(pFormatContext->streams[pFormatContext->programs[i]->stream_index[j]]->codec->codec_type == AVMEDIA_TYPE_VIDEO) VideoIndex = pFormatContext->programs[i]->stream_index[j]; if(pFormatContext->streams[pFormatContext->programs[i]->stream_index[j]]->codec->codec_type == AVMEDIA_TYPE_AUDIO) AudioIndex = pFormatContext->programs[i]->stream_index[j]; } } } FILE *fw_v = fopen(VideoName,"wb"); FILE *fw_a = fopen(AudioName,"wb"); if((fw_v == NULL) || (fw_a == NULL)) { avformat_close_input(&pFormatContext); return -3; } AVPacket pPacket; while(av_read_frame(pFormatContext,&pPacket) >= 0) { if(pPacket.stream_index == VideoIndex) { fwrite(pPacket.data,1,pPacket.size,fw_v); } if(pPacket.stream_index == AudioIndex) fwrite(pPacket.data,1,pPacket.size,fw_a); av_free_packet(&pPacket); } fclose(fw_a); fclose(fw_v); avformat_close_input(&pFormatContext); return 0; } int main() { av_register_all(); char *FileName = "Media_2.ts"; PrintfMediaProgram(FileName); printf("请输入节目号："); int ProgramNumber = 0; scanf("%d",&ProgramNumber); MediaSeparate(FileName,ProgramNumber,"Video.h264","Audio.aac"); return 0; }