FFmpeg_屏幕录制为TS流（可拼接版）

extern "C" { #include "libavformat\avformat.h" #include "libavdevice\avdevice.h" #include "libswscale\swscale.h" }; //录制屏幕为TS视频流（分辨率调整为352*288） int ScreenToTS() { //依当前本地时间生出输出文件路径 time_t LocalTime = time(0); char LocalTimeBuf[32] = ""; strftime(LocalTimeBuf,sizeof(LocalTimeBuf),"%Y_%m_%d_%H_%M_%S",localtime(&LocalTime)); char ts_out_path[32] = ""; sprintf(ts_out_path,"%s.ts",LocalTimeBuf); //锁定输出文件格式 AVFormatContext* pFormatContext_Out = NULL; avformat_alloc_output_context2(&pFormatContext_Out,NULL,NULL,ts_out_path); if(!pFormatContext_Out) return -1; AVOutputFormat* pOutputFormat = NULL; pOutputFormat = pFormatContext_Out->oformat; //找到输出文件格式对应的编码器 AVCodec* pCodec_Out = NULL; pCodec_Out = avcodec_find_encoder(pOutputFormat->video_codec); //依输出文件编码器创建输出流 AVStream* pStream_Out = NULL; pStream_Out = avformat_new_stream(pFormatContext_Out,pCodec_Out); if(!pStream_Out) { avformat_free_context(pFormatContext_Out); return -2; } //设定输出文件编解码器上下文环境 AVCodecContext* pCodecContext_Out = NULL; pCodecContext_Out = pStream_Out->codec; pCodecContext_Out->codec_id = pOutputFormat->video_codec; pCodecContext_Out->bit_rate = 200000; pCodecContext_Out->width = 352; pCodecContext_Out->height = 288; pStream_Out->time_base.num = 1; pStream_Out->time_base.den = 10; pCodecContext_Out->time_base = pStream_Out->time_base; pCodecContext_Out->gop_size = 250; pCodecContext_Out->pix_fmt = AV_PIX_FMT_YUV420P; pCodecContext_Out->max_b_frames = 2; pCodecContext_Out->codec_type = AVMEDIA_TYPE_VIDEO; if(pFormatContext_Out->oformat->flags & AVFMT_GLOBALHEADER) pCodecContext_Out->flags |= CODEC_FLAG_GLOBAL_HEADER; //打开输出文件对应的编码器 AVDictionary *pDictionary_Out = NULL; av_dict_copy(&pDictionary_Out,NULL,0); if(avcodec_open2(pCodecContext_Out,pCodec_Out,&pDictionary_Out) < 0) { av_dict_free(&pDictionary_Out); avformat_free_context(pFormatContext_Out); return -3; } av_dict_free(&pDictionary_Out); //截屏方式作为输入并打开该输入 AVFormatContext *pFormatContext_In = NULL; AVDictionary* pDictionary_In = NULL; av_dict_set(&pDictionary_In,"framerate","10",0); AVInputFormat* pInputFormat = av_find_input_format("gdigrab"); if(avformat_open_input(&pFormatContext_In,"desktop",pInputFormat,&pDictionary_In) != 0) { avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -4; } if(avformat_find_stream_info(pFormatContext_In,NULL) < 0) { avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -5; } //打开输入文件对应解码器 AVCodecContext* pCodecContext_In = NULL; pCodecContext_In = pFormatContext_In->streams[0]->codec; AVCodec* pCodec_In = NULL; pCodec_In = avcodec_find_decoder(pCodecContext_In->codec_id); if(avcodec_open2(pCodecContext_In,pCodec_In,nullptr)!=0) { avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -6; } //申请帧缓存（bmp帧、调整分辨率之后的yuv420p帧） AVFrame *pFrame,*pFrameYuv; pFrame = av_frame_alloc(); pFrameYuv = av_frame_alloc(); if((pFrame == NULL) || (pFrameYuv == NULL)) { avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -7; } pFrameYuv->format = pCodecContext_Out->pix_fmt; pFrameYuv->width = pCodecContext_Out->width; pFrameYuv->height = pCodecContext_Out->height; int Yuv420pSize = avpicture_get_size(PIX_FMT_YUV420P,pCodecContext_Out->width,pCodecContext_Out->height); uint8_t* Yuv420pBuf = (uint8_t* )av_malloc(Yuv420pSize); avpicture_fill((AVPicture* )pFrameYuv,Yuv420pBuf,PIX_FMT_YUV420P,pCodecContext_Out->width,pCodecContext_Out->height); //创建数据缓存并初始化 AVPacket pPacket; av_init_packet(&pPacket); pPacket.data = NULL; pPacket.size = 0; //打开输出流 if(avio_open(&pFormatContext_Out->pb,ts_out_path,AVIO_FLAG_WRITE) < 0) { av_free(Yuv420pBuf); av_frame_free(&pFrameYuv); av_frame_free(&pFrame); avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -8; } //写输出流的头部数据 if(avformat_write_header(pFormatContext_Out,&pDictionary_Out) < 0) { avio_closep(&pFormatContext_Out->pb); av_free(Yuv420pBuf); av_frame_free(&pFrameYuv); av_frame_free(&pFrame); avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -9; } //对输入视频流解码，编码视频帧序列到输出流 int ReturnValue = 0,got_picture_ptr = 0; while(pStream_Out->codec->frame_number < 600) { if(av_read_frame(pFormatContext_In,&pPacket) >= 0) { if(pPacket.stream_index == 0) { if((ReturnValue = avcodec_decode_video2(pCodecContext_In,pFrame,&got_picture_ptr,&pPacket)) < 0) { avio_closep(&pFormatContext_Out->pb); av_free(Yuv420pBuf); av_frame_free(&pFrameYuv); av_frame_free(&pFrame); avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -10; } if(got_picture_ptr) { struct SwsContext* pSwsContext = sws_getContext(pCodecContext_In->width,pCodecContext_In->height,pCodecContext_In->pix_fmt,pCodecContext_Out->width,pCodecContext_Out->height,PIX_FMT_YUV420P,SWS_SINC,NULL,NULL,NULL); sws_scale(pSwsContext,(const uint8_t* const*)pFrame->data,pFrame->linesize,0,pCodecContext_In->height,pFrameYuv->data,pFrameYuv->linesize); sws_freeContext(pSwsContext); if(avcodec_encode_video2(pCodecContext_Out,&pPacket,pFrameYuv,&got_picture_ptr) < 0) { avio_closep(&pFormatContext_Out->pb); av_free(Yuv420pBuf); av_frame_free(&pFrameYuv); av_frame_free(&pFrame); avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return -11; } if(got_picture_ptr) { av_packet_rescale_ts(&pPacket,pCodecContext_Out->time_base,pStream_Out->time_base); pPacket.stream_index = pStream_Out->index; av_interleaved_write_frame(pFormatContext_Out,&pPacket); } } } av_free_packet(&pPacket); } else break; } //写输出流的尾部数据 av_write_trailer(pFormatContext_Out); //资源释放 avio_closep(&pFormatContext_Out->pb); av_free(Yuv420pBuf); av_frame_free(&pFrameYuv); av_frame_free(&pFrame); avcodec_close(pCodecContext_In); avformat_close_input(&pFormatContext_In); avcodec_close(pCodecContext_Out); avformat_free_context(pFormatContext_Out); return 0; } int main() { av_register_all(); avdevice_register_all(); if(ScreenToTS() == 0) system("PAUSE"); return 0; }