/**
\fn ADM_codecIdFindByFourcc
\brief get lav codec if from fourcc (used by muxer)
*/
AVCodecID ADM_codecIdFindByFourcc(const char *fcc)
{
uint32_t fid=fourCC::get((uint8_t *)fcc);
// Special cases
if (isMSMpeg4Compatible (fid) == 1)
{
return AV_CODEC_ID_MSMPEG4V3;
}
if (isDVCompatible(fid))//"CDVC"))
{
return AV_CODEC_ID_DVVIDEO;
}
if (isH264Compatible (fid))
{
return AV_CODEC_ID_H264;
}
if (isH265Compatible (fid))
{
return AV_CODEC_ID_HEVC;
}
if (isMpeg4Compatible (fid) == 1)
{
return AV_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 AV_CODEC_ID_NONE;
}
/**
\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;
}
bool muxerMP4::open(const char *file, ADM_videoStream *s,uint32_t nbAudioTrack,ADM_audioStream **a)
{
if(!isMpeg4Compatible(s->getFCC()) && !isH264Compatible(s->getFCC()) && !isH265Compatible(s->getFCC()))
{
GUI_Error_HIG("Unsupported","Only MP4Video, H264, and H265 supported for video");
return false;
}
if(nbAudioTrack)
for(int i=0;i<nbAudioTrack;i++)
{
uint32_t acc=a[i]->getInfo()->encoding;
if(acc!=WAV_MP2 && acc!=WAV_MP3 && acc!=WAV_AAC && acc!=WAV_AC3)
{
GUI_Error_HIG("Unsupported","Only AAC, AC3, and mpegaudio supported for audio");
return false;
}
}
/* All seems fine, open stuff */
const char *f="mp4";
if(muxerConfig.muxerType==MP4_MUXER_PSP) f="psp";
if(false==setupMuxer(f,file))
{
printf("[MP4] Failed to open muxer\n");
return false;
}
if(initVideo(s)==false)
{
printf("[MP4] Failed to init video\n");
return false;
}
AVCodecContext *c;
AVRational myTimeBase;
c = video_st->codec;
rescaleFps(s->getAvgFps1000(),&(c->time_base));
myTimeBase=video_st->time_base=c->time_base;
ADM_info("Video stream time base :%d,%d\n",video_st->time_base.num,video_st->time_base.den);
c->gop_size=15;
if(initAudio(nbAudioTrack,a)==false)
{
printf("[MP4] Failed to init audio\n");
return false;
}
// /audio
int er = avio_open(&(oc->pb), file, AVIO_FLAG_WRITE);
ADM_info("Timebase In = %d/%d\n",myTimeBase.num,myTimeBase.den);
if (er)
{
ADM_error("[Mp4]: Failed to open file :%s, er=%d\n",file,er);
return false;
}
AVDictionary *dict = NULL;
char buf[64];
snprintf(buf, sizeof(buf), "%d", AV_TIME_BASE / 10);
av_dict_set(&dict, "preload", buf, 0);
av_dict_set(&dict, "max_delay", "200000", 0);
av_dict_set(&dict, "muxrate", "10080000", 0);
#ifndef _WIN32 // does not work on windows as the file must be opened twice at the same time
av_dict_set(&dict, "movflags","faststart",0);
#endif
ADM_assert(avformat_write_header(oc, &dict) >= 0);
ADM_info("Timebase codec = %d/%d\n",c->time_base.num,c->time_base.den);
ADM_info("Timebase stream = %d/%d\n",video_st->time_base.num,video_st->time_base.den);
if(myTimeBase.den==video_st->time_base.den && video_st->time_base.num==1)
{
roundup=myTimeBase.num;
ADM_warning("Timebase roundup = %d\n",roundup);
}
av_dict_free(&dict);
vStream=s;
aStreams=a;
nbAStreams=nbAudioTrack;
initialized=true;
return true;
}
