ffmpeg不用filter添加水印

/*add watermark into move by operate pic Data. audio has been ignored *缪国凯 MK *821486004@qq.com */ #include "Common.h" #include "ConvertImg.h" #include "MixYUV.h" #define ERRLOG printf #define ERR_OK 0 //成功 #define ERR_OPEN_INPUT -1 //打开输入错误 #define ERR_NO_VIDEO_INPUT -2 //打开视频输入错误 #define ERR_NO_AUDIO_INPUT -3 //打开音频输入错误 #define ERR_OPEN_VIDEO_OUTPUT -4 //打开视频输出错误 #define ERR_OPEN_FILE_HANLDE -5 //打开输出文件错误 #define ERR_WRITE_HEAD -6 //输出文件写头错误 #define ERR_OCTX_ALLOC -7 //无法初始化输出格式错误 #define ERR_MARK_NOPIC -8 //水印图片错误 #define ERR_ADD_SUBTITLE -9 //添加字幕出错 DWORD WINAPI DecodeWork(LPVOID lpParam ); void CodecWork(); int flush_video_encoder(AVFormatContext *fmt_ctx,unsigned int stream_index); void free_output_fmtCtx(AVFormatContext **fmtCtx); int NewVideoStream(AVFormatContext *pOutContext, AVCodecContext *pVideoCodecContext, int &outVedioStramIndex, int w, int h, AVCodecID codecID, float frameRate, AVStream* inPutStream) { int ret = 0; if (pVideoCodecContext && (!(w && h && 0 != codecID))) { outVedioStramIndex = pOutContext->nb_streams; AVStream* out_video_stream = avformat_new_stream(pOutContext, NULL); if (!out_video_stream) { ERRLOG("Fail to allocating output video stream!"); return -1; } if ((ret = avcodec_copy_context(out_video_stream->codec, pVideoCodecContext)) < 0) { ERRLOG("can not copy the video codec context! ret=%d", ret); return ret; } if (inPutStream) { AVDictionaryEntry *tag = NULL; tag = av_dict_get(inPutStream->metadata, "rotate", tag, 0); if (tag != NULL) { av_dict_set(&out_video_stream->metadata, "rotate", tag->value, 0); } } AVRational time_base={1, frameRate}; out_video_stream->time_base = time_base; out_video_stream->codec->codec_tag = 0; if(pOutContext->oformat->flags & AVFMT_GLOBALHEADER) { out_video_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER; } } else if (w && h && 0 != codecID) { //to be add byMK AVCodecContext *videoCodecCtx; outVedioStramIndex = pOutContext->nb_streams; AVStream* pVideoStream = avformat_new_stream(pOutContext, NULL); if (!pVideoStream) { ERRLOG("Fail to allocating output video stream!"); return -1; } AVDictionaryEntry *tag = NULL; if (inPutStream) { tag = av_dict_get(inPutStream->metadata, "rotate", tag, 0); if (tag != NULL) { //av_dict_set(&pVideoStream->metadata, "rotate", tag->value, 0); int tmpRotate = atoi(tag->value); if (tmpRotate == 90 || tmpRotate == 270) { w = w + h; h = w - h; w = w - h; } } } //set codec context param pVideoStream->codec->codec = avcodec_find_encoder(codecID); pVideoStream->codec->height = h; pVideoStream->codec->width = w; pVideoStream->codec->codec_id = codecID; pVideoStream->codec->codec_type = AVMEDIA_TYPE_VIDEO; pVideoStream->codec->qmin = 30; pVideoStream->codec->qmax = 51; AVRational time_base={1, frameRate}; pVideoStream->codec->time_base = time_base; pVideoStream->time_base = time_base; // take first format from list of supported formats AVRational sample_aspect_ratio={0, 1}; pVideoStream->codec->sample_aspect_ratio = sample_aspect_ratio; pVideoStream->codec->pix_fmt = AV_PIX_FMT_YUV420P;//pOutContext->streams[outVedioStramIndex]->codec->codec->pix_fmts[0]; //pVideoStream->codec->bit_rate = (w * h); pVideoStream->codec->gop_size = frameRate; pVideoStream->metadata; //open encoder if (!pVideoStream->codec->codec) { ERRLOG("fail to find the encoder for video stream codedID:%d", codecID); return -1; } if (pOutContext->oformat->flags & AVFMT_GLOBALHEADER) pVideoStream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER; AVDictionary *param = NULL; if (pVideoStream->codec->codec_id == AV_CODEC_ID_H264) { av_dict_set(&param, "preset", "slow", 0); av_dict_set(&param, "tune", "film", 0); } if ((ret = avcodec_open2(pVideoStream->codec, pVideoStream->codec->codec, &param)) < 0) { ERRLOG("fail to open the encoder for video stream, codecID:%d ret:%d", codecID, ret); return ret; } ret = 0; //将ret的值重新置为0 } else { return -1; } return ret; } int OpenInput(const char * file_name, AVFormatContext **in_fmtctx, int &videoIndex, int &audioIndex) { int ret = -1; if ((ret = avformat_open_input(in_fmtctx, file_name, NULL, NULL)) < 0) { ERRLOG("can not open the input file:%s!", file_name); return ret; } if ((ret = avformat_find_stream_info((*in_fmtctx), NULL)) < 0) { ERRLOG("can not find the input stream for file:%s!", file_name); avformat_close_input(in_fmtctx); return ret; } AVFormatContext *pTempInputCtx = (*in_fmtctx); for (int i = 0; i < pTempInputCtx->nb_streams; i++) { if (pTempInputCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) { if (-1 == videoIndex) { videoIndex = i; } ret = avcodec_open2(pTempInputCtx->streams[i]->codec, avcodec_find_decoder(pTempInputCtx->streams[i]->codec->codec_id), NULL); if(0 > ret) { ERRLOG("can not find or open video decoder for file:%s!", file_name); avformat_close_input(in_fmtctx); return ret; } } else if (pTempInputCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO) { if (-1 == audioIndex) { audioIndex = i; } ret = avcodec_open2(pTempInputCtx->streams[i]->codec, avcodec_find_decoder(pTempInputCtx->streams[i]->codec->codec_id), NULL); if(0 > ret) { ERRLOG("can not find or open audio decoder for file:%s!", file_name); avformat_close_input(in_fmtctx); return ret; } } } return 0; } const char *pVideoFileName = "SourceVideo2.mp4"; const char* pOutVideoFileName = "TestAddWaterMark.mp4"; const char *pMarkPicName = "mark.png"; AVFormatContext *pInputContext = NULL; int inputVideoIndex = -1; int inputAudioIndex = -1; //output AVFormatContext *pOutContext = NULL; int outPutVideoIndex = -1; bool bIsFileEnd = false; bool bWork = true; int main() { av_register_all(); if ((NULL == pVideoFileName || NULL == pOutVideoFileName)) { return -1; } int ret_code = 0; int m_x = 0; int m_y = 0; int srcW = 0; int srcH = 0; OpenInput(pVideoFileName, &pInputContext, inputVideoIndex, inputAudioIndex); if (-1 == inputVideoIndex) { printf("open input error\n"); return ERR_OPEN_INPUT; } srcW = pInputContext->streams[inputVideoIndex]->codec->width; srcH = pInputContext->streams[inputVideoIndex]->codec->height; int markW = 0; int markH = 0; //read mark pic AVFrame *pMarkFrame = NULL; { AVFormatContext *pTmpContext = NULL; int tmpVideoIndex = -1; int tmpAudioIndex = -1; OpenInput(pMarkPicName, &pTmpContext, tmpVideoIndex, tmpAudioIndex); if (-1 == tmpVideoIndex) { printf("open input error\n"); return ERR_OPEN_INPUT; } markW = pTmpContext->streams[tmpVideoIndex]->codec->width; markH = pTmpContext->streams[tmpVideoIndex]->codec->height; while(true) { AVPacket* packet = new AVPacket; av_init_packet(packet); packet->data = NULL; packet->size = 0; if (av_read_frame(pTmpContext, packet) < 0) { break; } //video stream int getFrame = 0; if (packet->stream_index == tmpVideoIndex) { AVFrame* pFrameBuf = av_frame_alloc(); if (avcodec_decode_video2(pTmpContext->streams[tmpVideoIndex]->codec, pFrameBuf, &getFrame, packet) < 0) { av_packet_unref(packet); delete packet; continue; } if (!getFrame) { AVPacket tmpPacke; av_init_pac