NVENCSTATUS NVEncFilterDenoiseGauss::run_filter(const FrameInfo *pInputFrame, FrameInfo **ppOutputFrames, int *pOutputFrameNum) {
NVENCSTATUS sts = NV_ENC_SUCCESS;
if (pInputFrame->ptr == nullptr) {
return sts;
}
*pOutputFrameNum = 1;
if (ppOutputFrames[0] == nullptr) {
auto pOutFrame = m_pFrameBuf[m_nFrameIdx].get();
ppOutputFrames[0] = &pOutFrame->frame;
m_nFrameIdx = (m_nFrameIdx + 1) % m_pFrameBuf.size();
}
ppOutputFrames[0]->picstruct = pInputFrame->picstruct;
if (interlaced(*pInputFrame)) {
return filter_as_interlaced_pair(pInputFrame, ppOutputFrames[0], cudaStreamDefault);
}
const auto memcpyKind = getCudaMemcpyKind(pInputFrame->deivce_mem, ppOutputFrames[0]->deivce_mem);
if (memcpyKind != cudaMemcpyDeviceToDevice) {
AddMessage(RGY_LOG_ERROR, _T("only supported on device memory.\n"));
return NV_ENC_ERR_UNSUPPORTED_PARAM;
}
if (m_pParam->frameOut.csp != m_pParam->frameIn.csp) {
AddMessage(RGY_LOG_ERROR, _T("csp does not match.\n"));
return NV_ENC_ERR_UNSUPPORTED_PARAM;
}
const auto supportedCspYV12 = make_array<RGY_CSP>(RGY_CSP_YV12, RGY_CSP_YV12_09, RGY_CSP_YV12_10, RGY_CSP_YV12_12, RGY_CSP_YV12_14, RGY_CSP_YV12_16);
const auto supportedCspYUV444 = make_array<RGY_CSP>(RGY_CSP_YUV444, RGY_CSP_YUV444_09, RGY_CSP_YUV444_10, RGY_CSP_YUV444_12, RGY_CSP_YUV444_14, RGY_CSP_YUV444_16);
if (std::find(supportedCspYV12.begin(), supportedCspYV12.end(), m_pParam->frameIn.csp) != supportedCspYV12.end()) {
sts = denoiseYV12(ppOutputFrames[0], pInputFrame);
} else if (std::find(supportedCspYUV444.begin(), supportedCspYUV444.end(), m_pParam->frameIn.csp) != supportedCspYUV444.end()) {
sts = denoiseYUV444(ppOutputFrames[0], pInputFrame);
} else {
AddMessage(RGY_LOG_ERROR, _T("unsupported csp.\n"));
sts = NV_ENC_ERR_UNIMPLEMENTED;
}
return sts;
}
bool ffMpeg2Configure(void)
{
diaMenuEntry arE[]=
{
{0,QT_TRANSLATE_NOOP("ffmpeg2","Normal (4:3)")},
{1,QT_TRANSLATE_NOOP("ffmpeg2","Wide (16:9)")}
};
diaMenuEntry matrixE[]=
{
{MPEG2_MATRIX_DEFAULT,QT_TRANSLATE_NOOP("ffmpeg2","Default")},
{MPEG2_MATRIX_TMPGENC,QT_TRANSLATE_NOOP("ffmpeg2","Tmpgenc")},
{MPEG2_MATRIX_ANIME,QT_TRANSLATE_NOOP("ffmpeg2","Animes")},
{MPEG2_MATRIX_KVCD,QT_TRANSLATE_NOOP("ffmpeg2","KVCD")},
};
diaMenuEntry rdE[]={
{0,QT_TRANSLATE_NOOP("ffmpeg2","MB comparison")},
{1,QT_TRANSLATE_NOOP("ffmpeg2","Fewest bits (vhq)")},
{2,QT_TRANSLATE_NOOP("ffmpeg2","Rate distortion")}
};
diaMenuEntry threads[]={
{0,QT_TRANSLATE_NOOP("ffmpeg2","One thread")},
{2,QT_TRANSLATE_NOOP("ffmpeg2","Two threads)")},
{3,QT_TRANSLATE_NOOP("ffmpeg2","Three threads")},
{99,QT_TRANSLATE_NOOP("ffmpeg2","Auto (#cpu)")}
};
diaMenuEntry interE[]={
{0,QT_TRANSLATE_NOOP("ffmpeg2","Progressive")},
{1,QT_TRANSLATE_NOOP("ffmpeg2","Interlaced")},
};
diaMenuEntry foE[]={
{0,QT_TRANSLATE_NOOP("ffmpeg2","Top Field First")},
{1,QT_TRANSLATE_NOOP("ffmpeg2","Bottom Field First")},
};
mpeg2_encoder *conf=&Mp2Settings;
uint32_t me=(uint32_t)conf->lavcSettings.me_method;
#define PX(x) &(conf->lavcSettings.x)
diaElemBitrate bitrate(&(Mp2Settings.params),NULL);
diaElemMenu threadM(PX(MultiThreaded),QT_TRANSLATE_NOOP("ffmpeg2","Threading"),4,threads);
diaElemUInteger qminM(PX(qmin),QT_TRANSLATE_NOOP("ffmpeg2","Mi_n. quantizer:"),1,31);
diaElemUInteger qmaxM(PX(qmax),QT_TRANSLATE_NOOP("ffmpeg2","Ma_x. quantizer:"),1,31);
diaElemUInteger qdiffM(PX(max_qdiff),QT_TRANSLATE_NOOP("ffmpeg2","Max. quantizer _difference:"),1,31);
diaElemUInteger bufferS(PX(bufferSize),QT_TRANSLATE_NOOP("ffmpeg2","VBV Buffer Size:"),1,1024);
diaElemUInteger maxBitrate(PX(maxBitrate),QT_TRANSLATE_NOOP("ffmpeg2","Max bitrate (kb/s):"),1,50000);
diaElemToggle trellis(PX(_TRELLIS_QUANT),QT_TRANSLATE_NOOP("ffmpeg2","_Trellis quantization"));
diaElemUInteger max_b_frames(PX(max_b_frames),QT_TRANSLATE_NOOP("ffmpeg2","_Number of B frames:"),0,32);
uint32_t widescreen= conf->lavcSettings.widescreen;
uint32_t iinterlaced= conf->lavcSettings.interlaced;
uint32_t bff= conf->lavcSettings.bff;
diaElemMenu rdM(PX(mb_eval),QT_TRANSLATE_NOOP("ffmpeg2","_Macroblock decision:"),3,rdE);
diaElemMenu arM(&(widescreen),QT_TRANSLATE_NOOP("ffmpeg2","Aspect ratio:"),2,arE);
diaElemMenu matrixM(&(Mp2Settings.matrix),QT_TRANSLATE_NOOP("ffmpeg2","Matrices:"),MPEG2_MATRIX_LAST,matrixE);
diaElemUInteger filetol(PX(vratetol),QT_TRANSLATE_NOOP("ffmpeg2","_Filesize tolerance (kb):"),0,100000);
diaElemFloat qzComp(PX(qcompress),QT_TRANSLATE_NOOP("ffmpeg2","_Quantizer compression:"),0,1);
diaElemFloat qzBlur(PX(qblur),QT_TRANSLATE_NOOP("ffmpeg2","Quantizer _blur:"),0,1);
diaElemUInteger GopSize(PX(gop_size),QT_TRANSLATE_NOOP("ffmpeg2","_Gop Size:"),1,30);
diaElemMenu interlaced(&(iinterlaced),QT_TRANSLATE_NOOP("ffmpeg2","_Interlaced:"),2,interE);
diaElemMenu fieldOrder(&(bff),QT_TRANSLATE_NOOP("ffmpeg2","Field Order:"),2,foE);
/* First Tab : encoding mode */
diaElem *diamode[]={&arM,&threadM,&bitrate};
diaElemTabs tabMode(QT_TRANSLATE_NOOP("ffmpeg2","Basic Settings"),3,diamode);
/* 2nd Tab : advanced*/
diaElem *diaAdv[]={&bufferS,&matrixM,&max_b_frames,&GopSize,&maxBitrate};
diaElemTabs tabAdv(QT_TRANSLATE_NOOP("ffmpeg2","Adv. Settings"),5,diaAdv);
/* 2ndb Tab : interlacing*/
diaElem *diaInter[]={&interlaced,&fieldOrder};
diaElemTabs tabInter(QT_TRANSLATE_NOOP("ffmpeg2","Interlacing"),2,diaInter);
/* 3nd Tab : Qz */
diaElem *diaQze[]={&rdM,&qminM,&qmaxM,&qdiffM,&trellis};
diaElemTabs tabQz(QT_TRANSLATE_NOOP("ffmpeg2","Quantization"),5,diaQze);
/* 4th Tab : RControl */
diaElem *diaRC[]={&filetol,&qzComp,&qzBlur};
diaElemTabs tabRC(QT_TRANSLATE_NOOP("ffmpeg2","Rate Control"),3,diaRC);
diaElemTabs *tabs[]={&tabMode,&tabAdv,&tabInter,&tabQz,&tabRC};
if( diaFactoryRunTabs(QT_TRANSLATE_NOOP("ffmpeg2","libavcodec MPEG-2 configuration"),5,tabs))
{
conf->lavcSettings.me_method=(Motion_Est_ID)me;
conf->lavcSettings.widescreen= widescreen;
conf->lavcSettings.interlaced= iinterlaced;
conf->lavcSettings.bff= bff;
return true;
}
return false;
}