/**
\fn ADM_codecIdFindByFourcc
\brief get lav codec if from fourcc (used by muxer)
*/
CodecID ADM_codecIdFindByFourcc(const char *fcc)
{
uint32_t fid=fourCC::get((uint8_t *)fcc);
// Special cases
if (isMSMpeg4Compatible (fid) == 1)
{
return CODEC_ID_MSMPEG4V3;
}
if (isDVCompatible(fid))//"CDVC"))
{
return CODEC_ID_DVVIDEO;
}
if (isH264Compatible (fid))
{
return CODEC_ID_H264;
}
if (isMpeg4Compatible (fid) == 1)
{
return CODEC_ID_MPEG4;
}
uint32_t nb=sizeof(ffCodec)/sizeof(ffVideoCodec);
for(int i=0;i<nb;i++)
{
if(!strcmp(fcc,ffCodec[i].string)) return ffCodec[i].codecId;
}
return CODEC_ID_NONE;
}
uint8_t lavMuxer::open(const char *filename,uint32_t inbitrate, ADM_MUXER_TYPE type, aviInfo *info,
uint32_t videoExtraDataSize, uint8_t *videoExtraData, WAVHeader *audioheader,
uint32_t audioextraSize,uint8_t *audioextraData)
{
AVCodecContext *c;
_type=type;
_fps1000=info->fps1000;
switch(_type)
{
case MUXER_TS:
fmt=guess_format("mpegts", NULL, NULL);
break;
case MUXER_DVD:
fmt = guess_format("dvd", NULL, NULL);
break;
case MUXER_VCD:
fmt = guess_format("vcd", NULL, NULL);
break;
case MUXER_SVCD:
fmt = guess_format("svcd", NULL, NULL);
break;
case MUXER_MP4:
fmt = guess_format("mp4", NULL, NULL);
break;
case MUXER_PSP:
fmt = guess_format("psp", NULL, NULL);
break;
case MUXER_FLV:
fmt = guess_format("flv", NULL, NULL);
break;
case MUXER_MATROSKA:
fmt = guess_format("matroska", NULL, NULL);
break;
default:
fmt=NULL;
}
if (!fmt)
{
printf("Lav:Cannot guess format\n");
ADM_assert(0);
return 0;
}
oc = av_alloc_format_context();
if (!oc)
{
printf("Lav:Cannot allocate context\n");
return 0;
}
oc->oformat = fmt;
snprintf(oc->filename,1000,"file://%s",filename);
// Video
//________
video_st = av_new_stream(oc, 0);
if (!video_st)
{
printf("Lav: new stream failed\n");
return 0;
}
c = video_st->codec;
switch(_type)
{
case MUXER_FLV:
c->codec=new AVCodec;
memset(c->codec,0,sizeof(AVCodec));
if(fourCC::check(info->fcc,(uint8_t *)"FLV1"))
{
c->codec_id=CODEC_ID_FLV1;
c->codec->name=ADM_strdup("FLV1");
}else
{
if(fourCC::check(info->fcc,(uint8_t *)"VP6F"))
{
c->codec_id=CODEC_ID_VP6F;
c->codec->name=ADM_strdup("VP6F");
}
else
ADM_assert(0);
}
break;
case MUXER_MATROSKA:
strcpy(oc->title,"Avidemux");
strcpy(oc->author,"Avidemux");
c->sample_aspect_ratio.num=1;
c->sample_aspect_ratio.den=1;
if(isMpeg4Compatible(info->fcc))
{
c->codec_id = CODEC_ID_MPEG4;
c->has_b_frames=1; // in doubt...
}else
{
if(isH264Compatible(info->fcc))
{
c->has_b_frames=1; // in doubt...
c->codec_id = CODEC_ID_H264;
c->codec=new AVCodec;
memset(c->codec,0,sizeof(AVCodec));
c->codec->name=ADM_strdup("H264");
}
else
{
if(!ADM_4cc_to_lavcodec((const char *)&(info->fcc),&(c->codec_id)))
{
printf("[lavFormat] Cannot map this\n");
return 0;
}
}
}
if(videoExtraDataSize)
{
c->extradata=videoExtraData;
c->extradata_size= videoExtraDataSize;
}
break;
case MUXER_MP4:
case MUXER_PSP:
{
// probably a memeleak here
char *foo=ADM_strdup(filename);
strcpy(oc->title,ADM_GetFileName(foo));
strcpy(oc->author,"Avidemux");
c->sample_aspect_ratio.num=1;
c->sample_aspect_ratio.den=1;
if(isMpeg4Compatible(info->fcc))
{
c->codec_id = CODEC_ID_MPEG4;
c->has_b_frames=1; // in doubt...
}else
{
if(isH264Compatible(info->fcc))
{
c->has_b_frames=1; // in doubt...
c->codec_id = CODEC_ID_H264;
c->codec=new AVCodec;
memset(c->codec,0,sizeof(AVCodec));
c->codec->name=ADM_strdup("H264");
}
else
{
if(isDVCompatible(info->fcc))
{
c->codec_id = CODEC_ID_DVVIDEO;
}else
{
if(fourCC::check(info->fcc,(uint8_t *)"H263"))
{
c->codec_id=CODEC_ID_H263;
}else{
c->codec_id = CODEC_ID_MPEG4; // Default value
printf("Ooops, cant mux that...\n");
printf("Ooops, cant mux that...\n");
printf("Ooops, cant mux that...\n");
}
}
}
}
if(videoExtraDataSize)
{
c->extradata=videoExtraData;
c->extradata_size= videoExtraDataSize;
}
if(MUXER_PSP==_type)
{
c->rc_buffer_size=0; //8*1024*224;
c->rc_max_rate=0; //768*1000;
c->rc_min_rate=0;
c->bit_rate=768*1000;
}
else
{
c->rc_buffer_size=8*1024*224;
c->rc_max_rate=9500*1000;
c->rc_min_rate=0;
if(!inbitrate)
c->bit_rate=9000*1000;
else
c->bit_rate=inbitrate;
}
}
break;
case MUXER_TS:
c->codec_id = CODEC_ID_MPEG2VIDEO;
c->rc_buffer_size=8*1024*224;
c->rc_max_rate=9500*1000;
c->rc_min_rate=0;
if(!inbitrate)
c->bit_rate=9000*1000;
else
c->bit_rate=inbitrate;
break;
case MUXER_DVD:
c->codec_id = CODEC_ID_MPEG2VIDEO;
c->rc_buffer_size=8*1024*224;
c->rc_max_rate=9500*1000;
c->rc_min_rate=0;
if(!inbitrate)
c->bit_rate=9000*1000;
else
c->bit_rate=inbitrate;
break;
case MUXER_VCD:
c->codec_id = CODEC_ID_MPEG1VIDEO;
c->rc_buffer_size=8*1024*40;
c->rc_max_rate=1152*1000;
c->rc_min_rate=1152*1000;
c->bit_rate=1152*1000;
break;
case MUXER_SVCD:
c->codec_id = CODEC_ID_MPEG2VIDEO;
c->rc_buffer_size=8*1024*112;
c->rc_max_rate=2500*1000;
c->rc_min_rate=0*1000;
if(!inbitrate)
c->bit_rate=2040*1000;
else
c->bit_rate=inbitrate;
break;
default:
ADM_assert(0);
}
c->codec_type = CODEC_TYPE_VIDEO;
c->flags=CODEC_FLAG_QSCALE;
c->width = info->width;
c->height = info->height;
AVRational fps25=(AVRational){1001,25025};
AVRational fps24=(AVRational){1001,24000};
AVRational fps30= (AVRational){1001,30000};
AVRational fpsfree= (AVRational){1000,_fps1000};
switch(_fps1000)
{
case 25000:
{
c->time_base= fps25;
break;
}
case 23976:
if(_type==MUXER_MP4 || _type==MUXER_PSP || _type==MUXER_FLV || _type==MUXER_MATROSKA)
{
c->time_base= fps24; //(AVRational){1001,24000};
break;
}
case 29970:
c->time_base=fps30;
break;
default:
{
if(_type==MUXER_MP4 || _type==MUXER_PSP || _type==MUXER_FLV || _type==MUXER_MATROSKA)
{
c->time_base=fpsfree;// (AVRational){1000,_fps1000};
break;
}
else
{
GUI_Error_HIG(QT_TR_NOOP("Incompatible frame rate"), NULL);
return 0;
}
}
break;
}
c->gop_size=15;
c->max_b_frames=2;
c->has_b_frames=1;
// Audio
//________
if(audioheader)
{
audio_st = av_new_stream(oc, 1);
if (!audio_st)
{
printf("Lav: new stream failed\n");
return 0;
}
c = audio_st->codec;
c->frame_size=1024; //For AAC mainly, sample per frame
printf("[LavFormat] Bitrate %u\n",(audioheader->byterate*8)/1000);
_audioFq=c->sample_rate = audioheader->frequency;
#if 0
if(_type== MUXER_PSP && audioheader->encoding==WAV_AAC)
{
_audioFq=c->sample_rate = audioheader->frequency/2; //_audioFq*=2; // SBR
}
#endif
switch(audioheader->encoding)
{
case WAV_AC3: c->codec_id = CODEC_ID_AC3;break;
case WAV_MP2: c->codec_id = CODEC_ID_MP2;break;
case WAV_MP3:
#warning FIXME : Probe deeper
c->frame_size=1152;
c->codec_id = CODEC_ID_MP3;
break;
case WAV_PCM:
// One chunk is 10 ms (1/100 of fq)
c->frame_size=4;
c->codec_id = CODEC_ID_PCM_S16LE;break;
case WAV_AAC:
c->extradata=audioextraData;
c->extradata_size= audioextraSize;
c->codec_id = CODEC_ID_AAC;
break;
default:
if(_type==MUXER_MATROSKA)
{
if(ADM_WaveTag_to_lavcodec(audioheader->encoding, &(c->codec_id)))
{
if(audioextraData)
{
c->extradata=audioextraData;
c->extradata_size= audioextraSize;
}
// Put a dummy time increment
c->time_base= fps25;
break;
}
}
printf("Cant mux that ! audio\n");
printf("Cant mux that ! audio\n");
c->codec_id = CODEC_ID_MP2;
return 0;
break;
}
c->codec_type = CODEC_TYPE_AUDIO;
c->bit_rate = audioheader->byterate*8;
c->rc_buffer_size=(c->bit_rate/(2*8)); // 500 ms worth
c->channels = audioheader->channels;
_audioByterate=audioheader->byterate;
}
// /audio
//----------------------
switch(_type)
{
case MUXER_FLV:
case MUXER_PSP:
case MUXER_MP4:
case MUXER_MATROSKA:
oc->mux_rate=10080*1000; // Needed ?
break;
case MUXER_TS:
oc->mux_rate=10080*1000;
break;
case MUXER_DVD:
oc->packet_size=2048;
oc->mux_rate=10080*1000;
break;
case MUXER_VCD:
oc->packet_size=2324;
oc->mux_rate=2352 * 75 * 8;
break;
case MUXER_SVCD:
oc->packet_size=2324;
oc->mux_rate=2*2352 * 75 * 8; // ?
break;
default:
ADM_assert(0);
}
oc->preload=AV_TIME_BASE/10; // 100 ms preloading
oc->max_delay=200*1000; // 500 ms
if (av_set_parameters(oc, NULL) < 0)
{
printf("Lav: set param failed \n");
return 0;
}
if (url_fopen(&(oc->pb), filename, URL_WRONLY) < 0)
{
printf("Lav: Failed to open file :%s\n",filename);
return 0;
}
ADM_assert(av_write_header(oc)>=0);
dump_format(oc, 0, filename, 1);
printf("lavformat mpeg muxer initialized\n");
_running=1;
one=(1000*1000*1000)/_fps1000;
_curDTS=one;
return 1;
}
/**
\fn initVideo
\brief setup video part of muxer
*/
bool muxerFFmpeg::initVideo(ADM_videoStream *stream)
{
audioDelay=stream->getVideoDelay();
video_st = avformat_new_stream(oc, NULL);
if (!video_st)
{
printf("[FF] new stream failed\n");
return false;
}
AVCodecContext *c;
c = video_st->codec;
c->sample_aspect_ratio.num=1;
c->sample_aspect_ratio.den=1;
video_st->sample_aspect_ratio=c->sample_aspect_ratio;
uint32_t videoExtraDataSize=0;
uint8_t *videoExtraData;
stream->getExtraData(&videoExtraDataSize,&videoExtraData);
printf("[FF] Using %d bytes for video extradata\n",(int)videoExtraDataSize);
ffmpuxerSetExtradata(c,videoExtraDataSize,videoExtraData);
c->rc_buffer_size=8*1024*224;
c->rc_max_rate=9500*1000;
c->rc_min_rate=0;
c->bit_rate=9000*1000;
c->codec_type = AVMEDIA_TYPE_VIDEO;
c->flags=CODEC_FLAG_QSCALE;
c->width = stream->getWidth();
c->height =stream->getHeight();
uint32_t fcc=stream->getFCC();
if(isMpeg4Compatible(fcc))
{
c->codec_id = CODEC_ID_MPEG4;
if(stream->providePts()==true)
{
c->has_b_frames=1; // in doubt...
c->max_b_frames=2;
}else
{
ADM_warning("Incoming stream does not provide PTS \n");
c->has_b_frames=0; // No PTS=cannot handle CTS...
c->max_b_frames=0;
}
}else
{
if(isH264Compatible(fcc) || isH265Compatible(fcc))
{
if(stream->providePts()==true)
{
c->has_b_frames=1; // in doubt...
c->max_b_frames=2;
}else
{
printf("[MP4] Source video has no PTS information, assuming no b frames\n");
c->has_b_frames=0; // No PTS=cannot handle CTS...
c->max_b_frames=0;
}
if(isH265Compatible(fcc)) {
c->codec_id = AV_CODEC_ID_HEVC;
setAvCodec(c,AV_CODEC_ID_HEVC);
} else {
c->codec_id = CODEC_ID_H264;
setAvCodec(c,CODEC_ID_H264);
}
}
else
{
if(isDVCompatible(fcc))
{
c->codec_id = CODEC_ID_DVVIDEO;
}else
{
if(fourCC::check(fcc,(uint8_t *)"H263"))
{
c->codec_id=CODEC_ID_H263;
}else
if(isVP6Compatible(stream->getFCC()))
{
c->codec_id=CODEC_ID_VP6F;
setAvCodec(c,CODEC_ID_VP6F);
c->has_b_frames=0; // No PTS=cannot handle CTS...
c->max_b_frames=0;
}else
if(fourCC::check(stream->getFCC(),(uint8_t *)"FLV1"))
{
c->has_b_frames=0; // No PTS=cannot handle CTS...
c->max_b_frames=0;
c->codec_id=CODEC_ID_FLV1;
setAvCodec(c,CODEC_ID_FLV1);
}else
{
if(fourCC::check(stream->getFCC(),(uint8_t *)"MPEG1"))
{
c->has_b_frames=1; // No PTS=cannot handle CTS...
c->max_b_frames=2;
c->codec_id=CODEC_ID_MPEG1VIDEO;
}
else if(fourCC::check(stream->getFCC(),(uint8_t *)"MPEG2"))
{
c->has_b_frames=1; // No PTS=cannot handle CTS...
c->max_b_frames=2;
c->codec_id=CODEC_ID_MPEG2VIDEO;
}else
{
uint32_t id=stream->getFCC();
AVCodecID cid=ADM_codecIdFindByFourcc(fourCC::tostring(id));
if(cid==CODEC_ID_NONE)
{
printf("[FF] Unknown video codec\n");
return false;
}
c->codec_id=cid;
}
}
}
}
}
if(useGlobalHeader()==true)
{
if(videoExtraDataSize)
{
ADM_info("Video has extradata and muxer requires globalHeader, assuming it is done so.\n");
c->flags|=CODEC_FLAG_GLOBAL_HEADER;
}else
{
ADM_warning("Video has no extradata but muxer requires globalHeader.\n");
}
}
printf("[FF] Video initialized\n");
return true;
}
/**
\fn getDecoder
\brief returns the correct decoder for a stream w,h,fcc,extraLen,extraData,bpp
*/
decoders *ADM_coreCodecGetDecoder (uint32_t fcc, uint32_t w, uint32_t h, uint32_t extraLen, uint8_t * extraData,uint32_t bpp)
{
ADM_info("Searching decoder in coreVideoCodec(%d x %d, extradataSize:%d)...\n",w,h,extraLen);
if (isMSMpeg4Compatible (fcc) == 1)
{
return (decoders *) (new decoderFFDiv3 (w,h,fcc,extraLen,extraData,bpp));
}
if (isDVCompatible(fcc))//"CDVC"))
{
return (decoders *) (new decoderFFDV (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "HFYU"))
{
return (decoders *) (new decoderFFhuff (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "PNG "))
{
return (decoders *) (new decoderFFPng (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "FFVH"))
{
return (decoders *) (new decoderFF_ffhuff (w,h,fcc,extraLen,extraData,bpp));
}
if (isH264Compatible (fcc))
{
#if defined(USE_VDPAU) && 0
if(vdpauUsable()==true)
return (decoders *) (new decoderFFVDPAU (w,h,fcc,extraLen,extraData,bpp));
else
#endif
return (decoders *) (new decoderFFH264 (w,h,fcc,extraLen,extraData,bpp));
}
/*
Could be either divx5 packed crap or xvid or ffmpeg
For now we return FFmpeg and later will switch to divx5 if available
(ugly hack for ugly hack....)
*/
if (isMpeg4Compatible (fcc) == 1)
{
return (decoders *) (new decoderFFMpeg4 (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "YV12")
|| fourCC::check (fcc, (uint8_t *) "yv12")
|| fourCC::check (fcc, (uint8_t *) "I420"))
{
printf ("\n using null codec\n");
return (decoders *) (new decoderNull (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "UYVY"))
{
printf ("\n using uyvy codec\n");
return (decoders *) (new decoderUYVY (w,h,fcc,extraLen,extraData,bpp));
}
if (fourCC::check (fcc, (uint8_t *) "YUY2"))
{
printf ("\n using YUY2 codec\n");
return (decoders *) (new decoderYUY2 (w,h,fcc,extraLen,extraData,bpp));
}
if ((fcc == 0) || fourCC::check (fcc, (uint8_t *) "RGB "))
{
// RGB 16 Codecs
printf ("\n using RGB codec\n");
return (decoders *) (new decoderRGB16 (w,h,fcc,extraLen,extraData,bpp)); //1
}
if ((fcc == 0) || fourCC::check (fcc, (uint8_t *) "DIB "))
{
// RGB 16 Codecs
printf ("\n using DIB codec (%d bpp)\n",(int)bpp);
return (decoders *) (new decoderRGB16 (w,h,fcc,extraLen,extraData,bpp)); //0
}
if (isMpeg12Compatible (fcc))
return (decoders *) (new decoderFFMpeg12 (w,h,fcc,extraLen,extraData,bpp));
// Search ffsimple
decoders *dec=admCreateFFSimple(w,h,fcc,extraLen,extraData,bpp);
if(dec)
{
printf("using ffSimple\n");
return dec;
}
// default : null decoder
printf ("\n using invalid codec for \n");
fourCC::print (fcc);
return (decoders *) (new decoderEmpty(w,h,fcc,extraLen,extraData,bpp));
}
