--- branches/dev-api-4/xvidcore/dshow/src/CXvidDecoder.cpp 2003/11/15 02:51:41 1208 +++ branches/dev-api-4/xvidcore/dshow/src/CXvidDecoder.cpp 2004/01/09 04:37:59 1310 @@ -19,7 +19,15 @@ * along with this program ; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * - * $Id: CXvidDecoder.cpp,v 1.1.2.3 2003-11-15 02:51:41 suxen_drol Exp $ + * $Id: CXvidDecoder.cpp,v 1.1.2.10 2004-01-09 04:37:59 syskin Exp $ + * + ****************************************************************************/ + +/**************************************************************************** + * + * 2003/12/11 - added some additional options, mainly to make the deblocking + * code from xvidcore available. Most of the new code is taken + * from Nic's dshow filter, (C) Nic, http://nic.dnsalias.com * ****************************************************************************/ @@ -53,7 +61,20 @@ #include "CXvidDecoder.h" #include "CAbout.h" +// Externs defined here +PostProcessing_Settings PPSettings; + +int rgb_flip; +bool USE_IYUV; +bool USE_YV12; +bool USE_YUY2; +bool USE_YVYU; +bool USE_UYVY; +bool USE_RGB32; +bool USE_RGB24; +bool USE_RG555; +bool USE_RG565; const AMOVIESETUP_MEDIATYPE sudInputPinTypes[] = { @@ -132,8 +153,6 @@ int g_cTemplates = sizeof(g_Templates) / sizeof(CFactoryTemplate); - - STDAPI DllRegisterServer() { return AMovieDllRegisterServer2( TRUE ); @@ -180,9 +199,10 @@ - /* constructor */ +#define XVID_DLL_NAME "xvidcore.dll" + CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) : CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID) { @@ -191,7 +211,29 @@ xvid_gbl_init_t init; memset(&init, 0, sizeof(init)); init.version = XVID_VERSION; - if (xvid_global(0, XVID_GBL_INIT, &init, NULL) < 0) + + ar_x = ar_y = 0; + + m_hdll = LoadLibrary(XVID_DLL_NAME); + if (m_hdll == NULL) { + DPRINTF("dll load failed"); + MessageBox(0, XVID_DLL_NAME " not found","Error", 0); + return; + } + + xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_global"); + if (xvid_global_func == NULL) { + MessageBox(0, "xvid_global() not found", "Error", 0); + return; + } + + xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_decore"); + if (xvid_decore_func == NULL) { + MessageBox(0, "xvid_decore() not found", "Error", 0); + return; + } + + if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0) { MessageBox(0, "xvid_global() failed", "Error", 0); return; @@ -203,6 +245,58 @@ memset(&m_frame, 0, sizeof(m_frame)); m_frame.version = XVID_VERSION; + + HKEY hKey; + DWORD size; + RegOpenKeyEx(XVID_REG_KEY, XVID_REG_SUBKEY, 0, KEY_READ, &hKey); + + // Set the default post-processing settings + REG_GET_N("Brightness", PPSettings.nBrightness, 25) + REG_GET_N("Deblock_Y", PPSettings.bDeblock_Y, 1) + REG_GET_N("Deblock_UV", PPSettings.bDeblock_UV, 1) + REG_GET_N("Dering", PPSettings.bDering, 1) + REG_GET_N("FilmEffect", PPSettings.bFilmEffect, 1) + REG_GET_N("ForceColorspace", PPSettings.nForceColorspace, 0) + + RegCloseKey(hKey); + + USE_IYUV = false; + USE_YV12 = false; + USE_YUY2 = false; + USE_YVYU = false; + USE_UYVY = false; + USE_RGB32 = false; + USE_RGB24 = false; + USE_RG555 = false; + USE_RG565 = false; + + switch ( PPSettings.nForceColorspace ) + { + case FORCE_NONE: + USE_IYUV = true; + USE_YV12 = true; + USE_YUY2 = true; + USE_YVYU = true; + USE_UYVY = true; + USE_RGB32 = true; + USE_RGB24 = true; + USE_RG555 = true; + USE_RG565 = true; + break; + case FORCE_YV12: + USE_IYUV = true; + USE_YV12 = true; + break; + case FORCE_YUY2: + USE_YUY2 = true; + break; + case FORCE_RGB24: + USE_RGB24 = true; + break; + case FORCE_RGB32: + USE_RGB32 = true; + break; + } } @@ -215,9 +309,15 @@ if (m_create.handle != NULL) { - xvid_decore(m_create.handle, XVID_DEC_DESTROY, 0, 0); + xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0); m_create.handle = NULL; } + + if (m_hdll != NULL) + { + FreeLibrary(m_hdll); + m_hdll = NULL; + } } @@ -239,11 +339,17 @@ { VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format(); hdr = &vih->bmiHeader; + /* PAR (x:y) is (1/ppm_X):(1/ppm_Y) where ppm is pixels-per-meter + which is equal to ppm_Y:ppm_X */ + ar_x = vih->bmiHeader.biYPelsPerMeter*hdr->biWidth; + ar_y = vih->bmiHeader.biXPelsPerMeter*hdr->biHeight; } else if (*mtIn->FormatType() == FORMAT_VideoInfo2) { VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format(); hdr = &vih2->bmiHeader; + ar_x = vih2->dwPictAspectRatioX; + ar_y = vih2->dwPictAspectRatioY; } else { @@ -280,16 +386,6 @@ } -#define USE_IYUV -#define USE_YV12 -#define USE_YUY2 -#define USE_YVYU -#define USE_UYVY -#define USE_RGB32 -#define USE_RGB24 -#define USE_RG555 -#define USE_RG565 - /* get list of supported output colorspaces */ @@ -302,7 +398,7 @@ return E_UNEXPECTED; } - VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER)); + VIDEOINFOHEADER2 * vih = (VIDEOINFOHEADER2 *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER2)); if (vih == NULL) { return E_OUTOFMEMORY; @@ -321,77 +417,95 @@ switch(iPosition) { - case 0 : -#ifdef USE_IYUV - vih->bmiHeader.biCompression = CLSID_MEDIASUBTYPE_IYUV.Data1; - vih->bmiHeader.biBitCount = 12; - mtOut->SetSubtype(&CLSID_MEDIASUBTYPE_IYUV); - break; -#endif - case 1 : -#ifdef USE_YV12 - vih->bmiHeader.biCompression = MEDIASUBTYPE_YV12.Data1; - vih->bmiHeader.biBitCount = 12; - mtOut->SetSubtype(&MEDIASUBTYPE_YV12); - break; -#endif - case 2: -#ifdef USE_YUY2 + + case 0: +if ( USE_YUY2 ) +{ vih->bmiHeader.biCompression = MEDIASUBTYPE_YUY2.Data1; vih->bmiHeader.biBitCount = 16; mtOut->SetSubtype(&MEDIASUBTYPE_YUY2); break; -#endif - case 3 : -#ifdef USE_YVYU +} + case 1 : +if ( USE_YVYU ) +{ vih->bmiHeader.biCompression = MEDIASUBTYPE_YVYU.Data1; vih->bmiHeader.biBitCount = 16; mtOut->SetSubtype(&MEDIASUBTYPE_YVYU); break; -#endif - case 4 : -#ifdef USE_UYVY +} + case 2 : +if ( USE_UYVY ) +{ vih->bmiHeader.biCompression = MEDIASUBTYPE_UYVY.Data1; vih->bmiHeader.biBitCount = 16; mtOut->SetSubtype(&MEDIASUBTYPE_UYVY); break; -#endif +} + case 3 : + if ( USE_IYUV ) +{ + vih->bmiHeader.biCompression = CLSID_MEDIASUBTYPE_IYUV.Data1; + vih->bmiHeader.biBitCount = 12; + mtOut->SetSubtype(&CLSID_MEDIASUBTYPE_IYUV); + break; +} + case 4 : +if ( USE_YV12 ) +{ + vih->bmiHeader.biCompression = MEDIASUBTYPE_YV12.Data1; + vih->bmiHeader.biBitCount = 12; + mtOut->SetSubtype(&MEDIASUBTYPE_YV12); + break; +} case 5 : -#ifdef USE_RGB32 +if ( USE_RGB32 ) +{ vih->bmiHeader.biCompression = BI_RGB; vih->bmiHeader.biBitCount = 32; mtOut->SetSubtype(&MEDIASUBTYPE_RGB32); break; -#endif +} case 6 : -#ifdef USE_RGB24 +if ( USE_RGB24 ) +{ vih->bmiHeader.biCompression = BI_RGB; vih->bmiHeader.biBitCount = 24; mtOut->SetSubtype(&MEDIASUBTYPE_RGB24); break; -#endif +} case 7 : -#ifdef USE_RG555 +if ( USE_RG555 ) +{ vih->bmiHeader.biCompression = BI_RGB; vih->bmiHeader.biBitCount = 16; mtOut->SetSubtype(&MEDIASUBTYPE_RGB555); break; -#endif +} case 8 : -#ifdef USE_RG565 +if ( USE_RG565 ) +{ vih->bmiHeader.biCompression = BI_RGB; vih->bmiHeader.biBitCount = 16; mtOut->SetSubtype(&MEDIASUBTYPE_RGB565); break; -#endif +} default : return VFW_S_NO_MORE_ITEMS; } vih->bmiHeader.biSizeImage = GetBitmapSize(&vih->bmiHeader); + if (ar_x != 0 && ar_y != 0) { + vih->dwPictAspectRatioX = ar_x; + vih->dwPictAspectRatioY = ar_y; + } else { // just to be safe + vih->dwPictAspectRatioX = m_create.width; + vih->dwPictAspectRatioY = m_create.height; + } + mtOut->SetType(&MEDIATYPE_Video); - mtOut->SetFormatType(&FORMAT_VideoInfo); + mtOut->SetFormatType(&FORMAT_VideoInfo2); mtOut->SetTemporalCompression(FALSE); mtOut->SetSampleSize(vih->bmiHeader.biSizeImage); @@ -403,8 +517,6 @@ HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format) { - int rgb_flip; - if (formattype == FORMAT_VideoInfo) { VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format; @@ -553,7 +665,7 @@ if (m_create.handle == NULL) { - if (xvid_decore(0, XVID_DEC_CREATE, &m_create, 0) < 0) + if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) { DPRINTF("*** XVID_DEC_CREATE error"); return S_FALSE; @@ -588,14 +700,31 @@ } m_frame.general = XVID_LOWDELAY; + if (pIn->IsDiscontinuity() == S_OK) m_frame.general = XVID_DISCONTINUITY; + if (PPSettings.bDeblock_Y) + m_frame.general |= XVID_DEBLOCKY; + + if (PPSettings.bDeblock_UV) + m_frame.general |= XVID_DEBLOCKUV; +/* + if (PPSettings.bDering) + m_frame.general |= XVID_DERING; +*/ + if (PPSettings.bFilmEffect) + m_frame.general |= XVID_FILMEFFECT; + + m_frame.output.csp &= ~XVID_CSP_VFLIP; + m_frame.output.csp |= rgb_flip^(PPSettings.bFlipVideo ? XVID_CSP_VFLIP : 0); + repeat : if (pIn->IsPreroll() != S_OK) { - length = xvid_decore(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats); + length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats); + if (length < 0) { DPRINTF("*** XVID_DEC_DECODE"); @@ -603,11 +732,19 @@ } } else - { + { /* Preroll frame - won't be displayed */ int tmp = m_frame.output.csp; + int tmp_gen = m_frame.general; + m_frame.output.csp = XVID_CSP_NULL; - length = xvid_decore(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats); + /* Disable postprocessing to speed-up seeking */ + m_frame.general &= ~XVID_DEBLOCKY; + m_frame.general &= ~XVID_DEBLOCKUV; +/* m_frame.general &= ~XVID_DERING; */ + m_frame.general &= ~XVID_FILMEFFECT; + + length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats); if (length < 0) { DPRINTF("*** XVID_DEC_DECODE"); @@ -615,8 +752,15 @@ } m_frame.output.csp = tmp; + m_frame.general = tmp_gen; + } + + if (stats.type == XVID_TYPE_NOTHING) { + DPRINTF("B-Frame decoder lag"); + return S_FALSE; } + if (stats.type == XVID_TYPE_VOL) { if (stats.data.vol.width != m_create.width || @@ -626,11 +770,18 @@ return S_FALSE; } + pOut->SetDiscontinuity(TRUE); + pOut->SetSyncPoint(TRUE); + m_frame.bitstream = (BYTE*)m_frame.bitstream + length; m_frame.length -= length; goto repeat; } + if (pIn->IsPreroll() == S_OK) { + return S_FALSE; + } + return S_OK; }