[svn] / trunk / xvidcore / dshow / src / CXvidDecoder.cpp Repository:
ViewVC logotype

Diff of /trunk/xvidcore/dshow/src/CXvidDecoder.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1437, Sun Apr 18 07:55:11 2004 UTC revision 2133, Fri Jan 8 18:01:44 2016 UTC
# Line 1  Line 1 
1  /*****************************************************************************  /*****************************************************************************
2   *   *
3   *  XVID MPEG-4 VIDEO CODEC   *  XVID MPEG-4 VIDEO CODEC
4   *  - XviD Decoder part of the DShow Filter  -   *  - Xvid Decoder part of the DShow Filter  -
5   *   *
6   *  Copyright(C) 2002-2004 Peter Ross <pross@xvid.org>   *  Copyright(C) 2002-2011 Peter Ross <pross@xvid.org>
7     *               2003-2012 Michael Militzer <michael@xvid.org>
8   *   *
9   *  This program is free software ; you can redistribute it and/or modify   *  This program is free software ; you can redistribute it and/or modify
10   *  it under the terms of the GNU General Public License as published by   *  it under the terms of the GNU General Public License as published by
# Line 19  Line 20 
20   *  along with this program ; if not, write to the Free Software   *  along with this program ; if not, write to the Free Software
21   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
22   *   *
23   * $Id: CXvidDecoder.cpp,v 1.7 2004-04-18 07:55:11 syskin Exp $   * $Id$
  *  
  ****************************************************************************/  
   
 /****************************************************************************  
  *  
  * 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  
24   *   *
25   ****************************************************************************/   ****************************************************************************/
26    
# Line 36  Line 29 
29          place these paths at the top of the Tools|Options|Directories list          place these paths at the top of the Tools|Options|Directories list
30    
31          headers:          headers:
32          C:\DXVCSDK\include          C:\DX90SDK\Include
33          C:\DXVCSDK\samples\Multimedia\DirectShow\BaseClasses          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses
34    
35          libraries (optional):          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Release
36          C:\DXVCSDK\samples\Multimedia\DirectShow\BaseClasses\Release          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Debug
37  */  */
38    
39    #ifdef ENABLE_MFT
40    #define XVID_USE_MFT
41    #endif
42    
43  #include <windows.h>  #include <windows.h>
44    
# Line 55  Line 50 
50  #endif  #endif
51  #include <dvdmedia.h>   // VIDEOINFOHEADER2  #include <dvdmedia.h>   // VIDEOINFOHEADER2
52    
53  #include <xvid.h>               // XviD API  #if defined(XVID_USE_MFT)
54    #define MFT_UNIQUE_METHOD_NAMES
55    #include <mftransform.h>
56    #include <mfapi.h>
57    #include <mferror.h>
58    #include <shlwapi.h>
59    #endif
60    
61    #include <shellapi.h>
62    
63    #include <xvid.h>               // Xvid API
64    
65    #include "resource.h"
66    
67  #include "IXvidDecoder.h"  #include "IXvidDecoder.h"
68  #include "CXvidDecoder.h"  #include "CXvidDecoder.h"
# Line 81  Line 88 
88          { &MEDIATYPE_Video, &CLSID_DIVX_UC },          { &MEDIATYPE_Video, &CLSID_DIVX_UC },
89          { &MEDIATYPE_Video, &CLSID_DX50 },          { &MEDIATYPE_Video, &CLSID_DX50 },
90          { &MEDIATYPE_Video, &CLSID_DX50_UC },          { &MEDIATYPE_Video, &CLSID_DX50_UC },
91            { &MEDIATYPE_Video, &CLSID_3IVX },
92            { &MEDIATYPE_Video, &CLSID_3IVX_UC },
93            { &MEDIATYPE_Video, &CLSID_3IV0 },
94            { &MEDIATYPE_Video, &CLSID_3IV0_UC },
95            { &MEDIATYPE_Video, &CLSID_3IV1 },
96            { &MEDIATYPE_Video, &CLSID_3IV1_UC },
97            { &MEDIATYPE_Video, &CLSID_3IV2 },
98            { &MEDIATYPE_Video, &CLSID_3IV2_UC },
99            { &MEDIATYPE_Video, &CLSID_LMP4 },
100            { &MEDIATYPE_Video, &CLSID_LMP4_UC },
101            { &MEDIATYPE_Video, &CLSID_RMP4 },
102            { &MEDIATYPE_Video, &CLSID_RMP4_UC },
103            { &MEDIATYPE_Video, &CLSID_SMP4 },
104            { &MEDIATYPE_Video, &CLSID_SMP4_UC },
105            { &MEDIATYPE_Video, &CLSID_HDX4 },
106            { &MEDIATYPE_Video, &CLSID_HDX4_UC },
107          { &MEDIATYPE_Video, &CLSID_MP4V },          { &MEDIATYPE_Video, &CLSID_MP4V },
108      { &MEDIATYPE_Video, &CLSID_MP4V_UC },
109  };  };
110    
111  const AMOVIESETUP_MEDIATYPE sudOutputPinTypes[] =  const AMOVIESETUP_MEDIATYPE sudOutputPinTypes[] =
# Line 121  Line 145 
145  {  {
146          &CLSID_XVID,                    // Filter CLSID          &CLSID_XVID,                    // Filter CLSID
147          XVID_NAME_L,                    // Filter name          XVID_NAME_L,                    // Filter name
148          MERIT_PREFERRED,                // Its merit          MERIT_PREFERRED+2,              // Its merit
149          sizeof(psudPins) / sizeof(AMOVIESETUP_PIN),     // Number of pins          sizeof(psudPins) / sizeof(AMOVIESETUP_PIN),     // Number of pins
150          psudPins                                // Pin details          psudPins                                // Pin details
151  };  };
# Line 148  Line 172 
172    
173  };  };
174    
   
175  /* note: g_cTemplates must be global; used by strmbase.lib(dllentry.cpp,dllsetup.cpp) */  /* note: g_cTemplates must be global; used by strmbase.lib(dllentry.cpp,dllsetup.cpp) */
176  int g_cTemplates = sizeof(g_Templates) / sizeof(CFactoryTemplate);  int g_cTemplates = sizeof(g_Templates) / sizeof(CFactoryTemplate);
177    
178    extern HINSTANCE g_xvid_hInst;
179    
180    static int GUI_Page = 0;
181    static HWND MSG_hwnd = NULL; /* message handler window */
182    
183    extern "C" void CALLBACK Configure(HWND hWndParent, HINSTANCE hInstParent, LPSTR lpCmdLine, int nCmdShow );
184    
185    LRESULT CALLBACK msg_proc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
186    {
187            switch ( uMsg )
188            {
189            case WM_ICONMESSAGE:
190                    switch(lParam)
191                    {
192                    case WM_LBUTTONUP:
193                    case WM_LBUTTONDBLCLK:
194                            if (!GUI_Page) {
195                                    GUI_Page = 1;
196                                    Configure(hwnd, g_xvid_hInst, "", 1);
197                                    GUI_Page = 0;
198                            }
199                            break;
200                    default:
201                            return DefWindowProc(hwnd, uMsg, wParam, lParam);
202                    };
203                    break;
204    
205            case WM_DESTROY:
206                    NOTIFYICONDATA nid;
207                    ZeroMemory(&nid, sizeof(NOTIFYICONDATA));
208    
209                    nid.cbSize = sizeof(NOTIFYICONDATA);
210                    nid.hWnd = hwnd;
211                    nid.uID = 1456;
212                    Shell_NotifyIcon(NIM_DELETE, &nid);
213            default:
214                    return DefWindowProc(hwnd, uMsg, wParam, lParam);
215            }
216    
217            return TRUE; /* ok */
218    }
219    
220  STDAPI DllRegisterServer()  STDAPI DllRegisterServer()
221  {  {
222    #if defined(XVID_USE_MFT)
223            int inputs_num = sizeof(sudInputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE);
224            int outputs_num = sizeof(sudOutputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE);
225            MFT_REGISTER_TYPE_INFO * mft_bs = new MFT_REGISTER_TYPE_INFO[inputs_num];
226            MFT_REGISTER_TYPE_INFO * mft_csp = new MFT_REGISTER_TYPE_INFO[outputs_num];
227    
228            {
229                    int i;
230                    for(i=0;i<inputs_num;i++) {
231                            mft_bs[i].guidMajorType = *sudInputPinTypes[i].clsMajorType;
232                            mft_bs[i].guidSubtype = *sudInputPinTypes[i].clsMinorType;
233                    }
234                    for(i=0;i<outputs_num;i++) {
235                            mft_csp[i].guidMajorType = *sudOutputPinTypes[i].clsMajorType;
236                            mft_csp[i].guidSubtype = *sudOutputPinTypes[i].clsMinorType; // MFT and AM GUIDs really the same?
237                    }
238            }
239    
240            /* Register the MFT decoder */
241            MFTRegister(CLSID_XVID,                          // CLSID
242                            MFT_CATEGORY_VIDEO_DECODER,          // Category
243                                    const_cast<LPWSTR>(XVID_NAME_MFT_L), // Friendly name
244                            0,                                   // Flags
245                            inputs_num,                          // Number of input types
246                            mft_bs,                              // Input types
247                            outputs_num,                         // Number of output types
248                            mft_csp,                             // Output types
249                            NULL                                 // Attributes (optional)
250                            );
251    
252            delete[] mft_bs;
253            delete[] mft_csp;
254    #endif /* XVID_USE_MFT */
255    
256      return AMovieDllRegisterServer2( TRUE );      return AMovieDllRegisterServer2( TRUE );
257  }  }
258    
259    
260  STDAPI DllUnregisterServer()  STDAPI DllUnregisterServer()
261  {  {
262    #if defined(XVID_USE_MFT)
263            MFTUnregister(CLSID_XVID);
264    #endif
265      return AMovieDllRegisterServer2( FALSE );      return AMovieDllRegisterServer2( FALSE );
266  }  }
267    
# Line 194  Line 295 
295          return GetInterface((ISpecifyPropertyPages *) this, ppv);          return GetInterface((ISpecifyPropertyPages *) this, ppv);
296          }          }
297    
298    #if defined(XVID_USE_MFT)
299            if (riid == IID_IMFTransform)
300            {
301                    return GetInterface((IMFTransform *) this, ppv);
302            }
303    #endif
304    
305          return CVideoTransformFilter::NonDelegatingQueryInterface(riid, ppv);          return CVideoTransformFilter::NonDelegatingQueryInterface(riid, ppv);
306  }  }
307    
# Line 201  Line 309 
309    
310  /* constructor */  /* constructor */
311    
 #define XVID_DLL_NAME "xvidcore.dll"  
   
312  CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) :  CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) :
313      CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID)      CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID), m_hdll (NULL)
314  {  {
315          DPRINTF("Constructor");          DPRINTF("Constructor");
316    
317        xvid_decore_func = NULL; // Hmm, some strange errors appearing if I try to initialize...
318        xvid_global_func = NULL; // ...this in constructor's init-list. So, they assigned here.
319    
320            m_tray_icon = 0;
321            m_startClock = clock();
322            interlaced = 0;
323    
324    #if defined(XVID_USE_MFT)
325            InitializeCriticalSection(&m_mft_lock);
326            m_pInputType = NULL;
327            m_pOutputType = NULL;
328            m_pOutputTypeBPP = 32;
329            m_rtFrame = 0;
330            m_duration = 0;
331            m_discont = 0;
332            m_frameRate.Denominator = 1;
333            m_frameRate.Numerator = 1;
334            m_thread_handle = NULL;
335    #endif
336    
337        LoadRegistryInfo();
338    
339        *phr = OpenLib();
340    
341            {
342                    TCHAR lpFilename[MAX_PATH];
343                    int sLen = GetModuleFileName(NULL, lpFilename, MAX_PATH);
344    #ifdef _UNICODE
345                    if ((sLen >= 11) && (_wcsnicmp(&(lpFilename[sLen - 11]), TEXT("dllhost.exe"), 11) == 0)) {
346    #else
347                    if ((sLen >= 11) && (_strnicmp(&(lpFilename[sLen - 11]), TEXT("dllhost.exe"), 11) == 0)) {
348    #endif
349                            if (m_tray_icon == 0) m_tray_icon = -1; // create no tray icon upon thumbnail generation
350                    }
351            }
352    
353    }
354    
355    HRESULT CXvidDecoder::OpenLib()
356    {
357        DPRINTF("OpenLib");
358    
359        if (m_hdll != NULL)
360                    return E_UNEXPECTED; // Seems, that library already opened.
361    
362          xvid_gbl_init_t init;          xvid_gbl_init_t init;
363          memset(&init, 0, sizeof(init));          memset(&init, 0, sizeof(init));
364          init.version = XVID_VERSION;          init.version = XVID_VERSION;
365            init.cpu_flags = g_config.cpu;
366    
367          ar_x = ar_y = 0;          xvid_gbl_info_t info;
368            memset(&info, 0, sizeof(info));
369            info.version = XVID_VERSION;
370    
371          m_hdll = LoadLibrary(XVID_DLL_NAME);          m_hdll = LoadLibrary(XVID_DLL_NAME);
372          if (m_hdll == NULL) {          if (m_hdll == NULL) {
373                  DPRINTF("dll load failed");                  DPRINTF("dll load failed");
374                  MessageBox(0, XVID_DLL_NAME " not found","Error", 0);                  MessageBox(0, XVID_DLL_NAME " not found","Error", MB_TOPMOST);
375                  return;                  return E_FAIL;
376          }          }
377    
378          xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_global");          xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_global");
379          if (xvid_global_func == NULL) {          if (xvid_global_func == NULL) {
380                  MessageBox(0, "xvid_global() not found", "Error", 0);          FreeLibrary(m_hdll);
381                  return;          m_hdll = NULL;
382                    MessageBox(0, "xvid_global() not found", "Error", MB_TOPMOST);
383                    return E_FAIL;
384          }          }
385    
386          xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_decore");          xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_decore");
387          if (xvid_decore_func == NULL) {          if (xvid_decore_func == NULL) {
388                  MessageBox(0, "xvid_decore() not found", "Error", 0);          xvid_global_func = NULL;
389                  return;          FreeLibrary(m_hdll);
390            m_hdll = NULL;
391                    MessageBox(0, "xvid_decore() not found", "Error", MB_TOPMOST);
392                    return E_FAIL;
393          }          }
394    
395          if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0)          if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0)
396          {          {
397                  MessageBox(0, "xvid_global() failed", "Error", 0);          xvid_global_func = NULL;
398                  return;          xvid_decore_func = NULL;
399            FreeLibrary(m_hdll);
400            m_hdll = NULL;
401                    MessageBox(0, "xvid_global() failed", "Error", MB_TOPMOST);
402                    return E_FAIL;
403            }
404    
405            if (xvid_global_func(0, XVID_GBL_INFO, &info, NULL) < 0)
406            {
407            xvid_global_func = NULL;
408            xvid_decore_func = NULL;
409            FreeLibrary(m_hdll);
410            m_hdll = NULL;
411                    MessageBox(0, "xvid_global() failed", "Error", MB_TOPMOST);
412                    return E_FAIL;
413          }          }
414    
415          memset(&m_create, 0, sizeof(m_create));          memset(&m_create, 0, sizeof(m_create));
416          m_create.version = XVID_VERSION;          m_create.version = XVID_VERSION;
417          m_create.handle = NULL;          m_create.handle = NULL;
418        /* Decoder threads */
419        if (g_config.cpu & XVID_CPU_FORCE) {
420                    m_create.num_threads = g_config.num_threads;
421            }
422            else {
423            m_create.num_threads = info.num_threads; /* Autodetect */
424                    g_config.num_threads = info.num_threads;
425            }
426    
427          memset(&m_frame, 0, sizeof(m_frame));          memset(&m_frame, 0, sizeof(m_frame));
428          m_frame.version = XVID_VERSION;          m_frame.version = XVID_VERSION;
429    
         LoadRegistryInfo();  
   
430          USE_IYUV = false;          USE_IYUV = false;
431          USE_YV12 = false;          USE_YV12 = false;
432          USE_YUY2 = false;          USE_YUY2 = false;
# Line 285  Line 464 
464                  USE_RGB32 = true;                  USE_RGB32 = true;
465                  break;                  break;
466          }          }
467    
468            switch (g_config.aspect_ratio)
469            {
470            case 0:
471            case 1:
472                    break;
473            case 2:
474                    ar_x = 4;
475                    ar_y = 3;
476                    break;
477            case 3:
478                    ar_x = 16;
479                    ar_y = 9;
480                    break;
481            case 4:
482                    ar_x = 47;
483                    ar_y = 20;
484                    break;
485            }
486    
487            return S_OK;
488  }  }
489    
490  void CXvidDecoder::CloseLib()  void CXvidDecoder::CloseLib()
491  {  {
492          DPRINTF("Destructor");          DPRINTF("CloseLib");
493    
494          if (m_create.handle != NULL) {          if ((m_create.handle != NULL) && (xvid_decore_func != NULL))
495            {
496                  xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0);                  xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0);
497                  m_create.handle = NULL;                  m_create.handle = NULL;
498          }          }
# Line 300  Line 501 
501                  FreeLibrary(m_hdll);                  FreeLibrary(m_hdll);
502                  m_hdll = NULL;                  m_hdll = NULL;
503          }          }
504        xvid_decore_func = NULL;
505        xvid_global_func = NULL;
506  }  }
507    
508  /* destructor */  /* destructor */
509    
510  CXvidDecoder::~CXvidDecoder()  CXvidDecoder::~CXvidDecoder()
511  {  {
512        DPRINTF("Destructor");
513    
514            if ((MSG_hwnd != NULL) && (m_tray_icon == 1)) { /* Destroy tray icon */
515                    SendMessage(MSG_hwnd, WM_CLOSE, 0, 0);
516            }
517    
518            /* Close xvidcore library */
519          CloseLib();          CloseLib();
520    
521            clock_t endClock = clock();
522            if (((endClock - m_startClock) / CLOCKS_PER_SEC) > 3) {
523                    SaveRegistryInfo((endClock - m_startClock) / (CLOCKS_PER_SEC / 10));
524            }
525    
526            if ((MSG_hwnd != 0) && (m_tray_icon == 1)) { /* Final clean-up */
527              MSG_hwnd = 0;
528              Sleep(200);
529              m_tray_icon = 0;
530    #if defined(XVID_USE_MFT)
531              if (m_thread_handle) {
532                      TerminateThread(m_thread_handle, 0);
533                      CloseHandle(m_thread_handle);
534                      m_thread_handle = NULL;
535              }
536    #endif
537            }
538    
539    #if defined(XVID_USE_MFT)
540            DeleteCriticalSection(&m_mft_lock);
541    #endif
542  }  }
543    
544    
# Line 318  Line 550 
550          DPRINTF("CheckInputType");          DPRINTF("CheckInputType");
551          BITMAPINFOHEADER * hdr;          BITMAPINFOHEADER * hdr;
552    
553            ar_x = ar_y = 0;
554    
555          if (*mtIn->Type() != MEDIATYPE_Video)          if (*mtIn->Type() != MEDIATYPE_Video)
556          {          {
557                  DPRINTF("Error: Unknown Type");                  DPRINTF("Error: Unknown Type");
# Line 325  Line 559 
559                  return VFW_E_TYPE_NOT_ACCEPTED;                  return VFW_E_TYPE_NOT_ACCEPTED;
560          }          }
561    
562        if (m_hdll == NULL)
563        {
564                    HRESULT hr = OpenLib();
565    
566            if (FAILED(hr) || (m_hdll == NULL)) // Paranoid checks.
567                            return VFW_E_TYPE_NOT_ACCEPTED;
568        }
569    
570          if (*mtIn->FormatType() == FORMAT_VideoInfo)          if (*mtIn->FormatType() == FORMAT_VideoInfo)
571          {          {
572                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format();                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format();
573                  hdr = &vih->bmiHeader;                  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 * abs(hdr->biWidth);  
                 ar_y = vih->bmiHeader.biXPelsPerMeter * abs(hdr->biHeight);  
                 DPRINTF("VIDEOINFOHEADER PAR: %d:%d -> AR %d:%d",  
                         vih->bmiHeader.biYPelsPerMeter,vih->bmiHeader.biXPelsPerMeter, ar_x, ar_y);  
574          }          }
575          else if (*mtIn->FormatType() == FORMAT_VideoInfo2)          else if (*mtIn->FormatType() == FORMAT_VideoInfo2)
576          {          {
577                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format();                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format();
578                  hdr = &vih2->bmiHeader;                  hdr = &vih2->bmiHeader;
579                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
580                            ar_x = vih2->dwPictAspectRatioX;
581                            ar_y = vih2->dwPictAspectRatioY;
582                    }
583                    DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
584            }
585      else if (*mtIn->FormatType() == FORMAT_MPEG2Video) {
586        MPEG2VIDEOINFO * mpgvi = (MPEG2VIDEOINFO*)mtIn->Format();
587        VIDEOINFOHEADER2 * vih2 = &mpgvi->hdr;
588                    hdr = &vih2->bmiHeader;
589                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
590                  ar_x = vih2->dwPictAspectRatioX;                  ar_x = vih2->dwPictAspectRatioX;
591                  ar_y = vih2->dwPictAspectRatioY;                  ar_y = vih2->dwPictAspectRatioY;
592                    }
593                  DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);                  DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
594    
595        /* haali media splitter reports VOL information in the format header */
596    
597        if (mpgvi->cbSequenceHeader>0) {
598    
599          xvid_dec_stats_t stats;
600                memset(&stats, 0, sizeof(stats));
601                stats.version = XVID_VERSION;
602    
603                if (m_create.handle == NULL) {
604                        if (xvid_decore_func == NULL)
605                                return E_FAIL;
606                        if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) {
607              DPRINTF("*** XVID_DEC_CREATE error");
608                                return E_FAIL;
609                        }
610                }
611    
612          m_frame.general = 0;
613          m_frame.bitstream = (void*)mpgvi->dwSequenceHeader;
614          m_frame.length = mpgvi->cbSequenceHeader;
615          m_frame.output.csp = XVID_CSP_NULL;
616    
617          int ret = 0;
618          if ((ret=xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats)) >= 0) {
619            /* honour video dimensions reported in VOL header */
620                  if (stats.type == XVID_TYPE_VOL) {
621              hdr->biWidth = stats.data.vol.width;
622              hdr->biHeight = stats.data.vol.height;
623                      }
624              }
625          if (ret == XVID_ERR_MEMORY) return E_FAIL;
626            }
627          }          }
628          else          else
629          {          {
# Line 350  Line 631 
631                  CloseLib();                  CloseLib();
632                  return VFW_E_TYPE_NOT_ACCEPTED;                  return VFW_E_TYPE_NOT_ACCEPTED;
633          }          }
   
634          if (hdr->biHeight < 0)          if (hdr->biHeight < 0)
635          {          {
636                  DPRINTF("colorspace: inverted input format not supported");                  DPRINTF("colorspace: inverted input format not supported");
# Line 361  Line 641 
641    
642          switch(hdr->biCompression)          switch(hdr->biCompression)
643          {          {
644            case FOURCC_mp4v :
645          case FOURCC_MP4V:          case FOURCC_MP4V:
646            case FOURCC_lmp4 :
647            case FOURCC_LMP4 :
648            case FOURCC_rmp4 :
649            case FOURCC_RMP4 :
650            case FOURCC_smp4 :
651            case FOURCC_SMP4 :
652            case FOURCC_hdx4 :
653            case FOURCC_HDX4 :
654                  if (!(g_config.supported_4cc & SUPPORT_MP4V)) {                  if (!(g_config.supported_4cc & SUPPORT_MP4V)) {
655                          CloseLib();                          CloseLib();
656                          return VFW_E_TYPE_NOT_ACCEPTED;                          return VFW_E_TYPE_NOT_ACCEPTED;
657                  }                  }
658                  break;                  break;
659            case FOURCC_divx :
660          case FOURCC_DIVX :          case FOURCC_DIVX :
661            case FOURCC_dx50 :
662            case FOURCC_DX50 :
663                  if (!(g_config.supported_4cc & SUPPORT_DIVX)) {                  if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
664                          CloseLib();                          CloseLib();
665                          return VFW_E_TYPE_NOT_ACCEPTED;                          return VFW_E_TYPE_NOT_ACCEPTED;
666                  }                  }
667                  break;                  break;
668          case FOURCC_DX50 :          case FOURCC_3ivx :
669                  if (!(g_config.supported_4cc & SUPPORT_DX50)) {          case FOURCC_3IVX :
670            case FOURCC_3iv0 :
671            case FOURCC_3IV0 :
672            case FOURCC_3iv1 :
673            case FOURCC_3IV1 :
674            case FOURCC_3iv2 :
675            case FOURCC_3IV2 :
676                    if (!(g_config.supported_4cc & SUPPORT_3IVX)) {
677                          CloseLib();                          CloseLib();
678                          return VFW_E_TYPE_NOT_ACCEPTED;                          return VFW_E_TYPE_NOT_ACCEPTED;
679                  }                  }
680            case FOURCC_xvid :
681          case FOURCC_XVID :          case FOURCC_XVID :
682                  break;                  break;
683    
# Line 393  Line 692 
692                  CloseLib();                  CloseLib();
693                  return VFW_E_TYPE_NOT_ACCEPTED;                  return VFW_E_TYPE_NOT_ACCEPTED;
694          }          }
695    
696            m_create.fourcc = hdr->biCompression;
697    
698          return S_OK;          return S_OK;
699  }  }
700    
# Line 421  Line 723 
723                  if (ar_x != 0 && ar_y != 0) {                  if (ar_x != 0 && ar_y != 0) {
724                          vih->dwPictAspectRatioX = ar_x;                          vih->dwPictAspectRatioX = ar_x;
725                          vih->dwPictAspectRatioY = ar_y;                          vih->dwPictAspectRatioY = ar_y;
726                            forced_ar = true;
727                  } else { // just to be safe                  } else { // just to be safe
728                          vih->dwPictAspectRatioX = m_create.width;                          vih->dwPictAspectRatioX = m_create.width;
729                          vih->dwPictAspectRatioY = abs(m_create.height);                          vih->dwPictAspectRatioY = abs(m_create.height);
730                            forced_ar = false;
731                    }
732                    if (interlaced) {
733                            vih->dwInterlaceFlags = AMINTERLACE_IsInterlaced;
734                            if (interlaced > 1) {
735                                    vih->dwInterlaceFlags |= AMINTERLACE_Field1First;
736                            }
737                    }
738                    else {
739                            vih->dwInterlaceFlags = 0;
740                  }                  }
   
741          } else {          } else {
742    
743                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER));                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER));
# Line 533  Line 845 
845    
846    
847  /* (internal function) change colorspace */  /* (internal function) change colorspace */
848    #define CALC_BI_STRIDE(width, bitcount)  ((((width * bitcount) + 31) & ~31) >> 3)
849    
850  HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format)  HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void *format, int *bitdepth, int noflip)
851  {  {
852            DWORD biWidth;
853            *bitdepth = 32;
854    
855          if (formattype == FORMAT_VideoInfo)          if (formattype == FORMAT_VideoInfo)
856          {          {
857                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;
858                  m_frame.output.stride[0] = (((vih->bmiHeader.biWidth * vih->bmiHeader.biBitCount) + 31) & ~31) >> 3;                  biWidth = vih->bmiHeader.biWidth;
859                    *bitdepth = vih->bmiHeader.biBitCount;
860                    out_stride = CALC_BI_STRIDE(vih->bmiHeader.biWidth, vih->bmiHeader.biBitCount);
861                  rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);                  rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
862          }          }
863          else if (formattype == FORMAT_VideoInfo2)          else if (formattype == FORMAT_VideoInfo2)
864          {          {
865                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;
866                  m_frame.output.stride[0] = (((vih2->bmiHeader.biWidth * vih2->bmiHeader.biBitCount) + 31) & ~31) >> 3;                  biWidth = vih2->bmiHeader.biWidth;
867                    *bitdepth = vih2->bmiHeader.biBitCount;
868                    out_stride = CALC_BI_STRIDE(vih2->bmiHeader.biWidth, vih2->bmiHeader.biBitCount);
869                  rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);                  rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
870          }          }
871          else          else
# Line 553  Line 873 
873                  return S_FALSE;                  return S_FALSE;
874          }          }
875    
876            if (noflip) rgb_flip = 0;
877    
878          if (subtype == CLSID_MEDIASUBTYPE_IYUV)          if (subtype == CLSID_MEDIASUBTYPE_IYUV)
879          {          {
880                  DPRINTF("IYUV");                  DPRINTF("IYUV");
881                  rgb_flip = 0;                  rgb_flip = 0;
882                  m_frame.output.csp = XVID_CSP_I420;                  m_frame.output.csp = XVID_CSP_I420;
883                  m_frame.output.stride[0] = (m_frame.output.stride[0] * 2) / 3;  /* planar format fix */                  out_stride = CALC_BI_STRIDE(biWidth, 8);        /* planar format fix */
884          }          }
885          else if (subtype == MEDIASUBTYPE_YV12)          else if (subtype == MEDIASUBTYPE_YV12)
886          {          {
887                  DPRINTF("YV12");                  DPRINTF("YV12");
888                  rgb_flip = 0;                  rgb_flip = 0;
889                  m_frame.output.csp = XVID_CSP_YV12;                  m_frame.output.csp = XVID_CSP_YV12;
890                  m_frame.output.stride[0] = (m_frame.output.stride[0] * 2) / 3;  /* planar format fix */                  out_stride = CALC_BI_STRIDE(biWidth, 8);        /* planar format fix */
891          }          }
892          else if (subtype == MEDIASUBTYPE_YUY2)          else if (subtype == MEDIASUBTYPE_YUY2)
893          {          {
# Line 622  Line 944 
944    
945  HRESULT CXvidDecoder::SetMediaType(PIN_DIRECTION direction, const CMediaType *pmt)  HRESULT CXvidDecoder::SetMediaType(PIN_DIRECTION direction, const CMediaType *pmt)
946  {  {
947            int bitdepth;
948          DPRINTF("SetMediaType");          DPRINTF("SetMediaType");
949    
950          if (direction == PINDIR_OUTPUT)          if (direction == PINDIR_OUTPUT)
951          {          {
952                  return ChangeColorspace(*pmt->Subtype(), *pmt->FormatType(), pmt->Format());                  return ChangeColorspace(*pmt->Subtype(), *pmt->FormatType(), pmt->Format(), &bitdepth, 0);
953          }          }
954    
955          return S_OK;          return S_OK;
# Line 638  Line 961 
961  HRESULT CXvidDecoder::CheckTransform(const CMediaType *mtIn, const CMediaType *mtOut)  HRESULT CXvidDecoder::CheckTransform(const CMediaType *mtIn, const CMediaType *mtOut)
962  {  {
963          DPRINTF("CheckTransform");          DPRINTF("CheckTransform");
964    
965            return S_OK;
966    }
967    
968    /* input/output pin connection complete */
969    
970    HRESULT CXvidDecoder::CompleteConnect(PIN_DIRECTION direction, IPin *pReceivePin)
971    {
972            DPRINTF("CompleteConnect");
973    
974            if ((direction == PINDIR_OUTPUT) && (MSG_hwnd == 0) && (m_tray_icon == 0) && (g_config.bTrayIcon != 0))
975            {
976                    WNDCLASSEX wc;
977    
978                    wc.cbSize = sizeof(WNDCLASSEX);
979                    wc.lpfnWndProc = msg_proc;
980                    wc.style = CS_HREDRAW | CS_VREDRAW;
981                    wc.cbWndExtra = 0;
982                    wc.cbClsExtra = 0;
983                    wc.hInstance = (HINSTANCE) g_xvid_hInst;
984                    wc.hbrBackground = (HBRUSH) GetStockObject(NULL_BRUSH);
985                    wc.lpszMenuName = NULL;
986                    wc.lpszClassName = "XVID_MSG_WINDOW";
987                    wc.hIcon = NULL;
988                    wc.hIconSm = NULL;
989                    wc.hCursor = NULL;
990                    RegisterClassEx(&wc);
991    
992                    MSG_hwnd = CreateWindowEx(0, "XVID_MSG_WINDOW", NULL, 0, CW_USEDEFAULT,
993                                      CW_USEDEFAULT, 0, 0, HWND_MESSAGE, NULL, (HINSTANCE) g_xvid_hInst, NULL);
994    
995                    /* display the tray icon */
996                    NOTIFYICONDATA nid;
997                    ZeroMemory(&nid,sizeof(NOTIFYICONDATA));
998    
999                    nid.cbSize = sizeof(NOTIFYICONDATA);
1000                    nid.hWnd = MSG_hwnd;
1001                    nid.uID = 1456;
1002                    nid.uVersion = NOTIFYICON_VERSION;
1003                    nid.uCallbackMessage = WM_ICONMESSAGE;
1004                    nid.hIcon = LoadIcon(g_xvid_hInst, MAKEINTRESOURCE(IDI_ICON));
1005                    strcpy_s(nid.szTip, 19, "Xvid Video Decoder");
1006                    nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP | NIF_SHOWTIP;
1007    
1008                    Shell_NotifyIcon(NIM_ADD, &nid);
1009                    Shell_NotifyIcon(NIM_SETVERSION, &nid);
1010    
1011                    DestroyIcon(nid.hIcon);
1012                    m_tray_icon = 1;
1013            }
1014    
1015          return S_OK;          return S_OK;
1016  }  }
1017    
1018    /* input/output pin disconnected */
1019    HRESULT CXvidDecoder::BreakConnect(PIN_DIRECTION direction)
1020    {
1021            DPRINTF("BreakConnect");
1022    
1023            return S_OK;
1024    }
1025    
1026  /* alloc output buffer */  /* alloc output buffer */
1027    
# Line 689  Line 1070 
1070    
1071          if (m_create.handle == NULL)          if (m_create.handle == NULL)
1072          {          {
1073                    if (xvid_decore_func == NULL)
1074                            return E_FAIL;
1075    
1076                  if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)                  if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)
1077                  {                  {
1078              DPRINTF("*** XVID_DEC_CREATE error");              DPRINTF("*** XVID_DEC_CREATE error");
1079                          return S_FALSE;                          return E_FAIL;
1080                  }                  }
1081          }          }
1082    
# Line 700  Line 1084 
1084          pOut->GetMediaType(&mtOut);          pOut->GetMediaType(&mtOut);
1085          if (mtOut != NULL)          if (mtOut != NULL)
1086          {          {
1087                    int bitdepth;
1088                  HRESULT result;                  HRESULT result;
1089    
1090                  result = ChangeColorspace(mtOut->subtype, mtOut->formattype, mtOut->pbFormat);                  result = ChangeColorspace(mtOut->subtype, mtOut->formattype, mtOut->pbFormat, &bitdepth, 0);
1091                  DeleteMediaType(mtOut);                  DeleteMediaType(mtOut);
1092    
1093                  if (result != S_OK)                  if (result != S_OK)
# Line 747  Line 1132 
1132    
1133          m_frame.output.csp &= ~XVID_CSP_VFLIP;          m_frame.output.csp &= ~XVID_CSP_VFLIP;
1134          m_frame.output.csp |= rgb_flip^(g_config.nFlipVideo ? XVID_CSP_VFLIP : 0);          m_frame.output.csp |= rgb_flip^(g_config.nFlipVideo ? XVID_CSP_VFLIP : 0);
1135            m_frame.output.stride[0] = out_stride;
1136    
1137        // Paranoid check.
1138        if (xvid_decore_func == NULL)
1139                    return E_FAIL;
1140    
1141    
1142  repeat :  repeat :
1143    
# Line 754  Line 1145 
1145          {          {
1146                  length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);                  length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1147    
1148                  if (length < 0)                  if (length == XVID_ERR_MEMORY)
1149                            return E_FAIL;
1150                    else if (length < 0)
1151                  {                  {
1152              DPRINTF("*** XVID_DEC_DECODE");              DPRINTF("*** XVID_DEC_DECODE");
1153                          return S_FALSE;                          return S_FALSE;
1154                    } else
1155                            if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1 && forced_ar == false) {
1156    
1157          if (stats.type != XVID_TYPE_NOTHING) {  /* dont attempt to set vmr aspect ratio if no frame was returned by decoder */
1158                            // inspired by minolta! works for VMR 7 + 9
1159                            IMediaSample2 *pOut2 = NULL;
1160                            AM_SAMPLE2_PROPERTIES outProp2;
1161                            if (SUCCEEDED(pOut->QueryInterface(IID_IMediaSample2, (void **)&pOut2)) &&
1162                                    SUCCEEDED(pOut2->GetProperties(FIELD_OFFSET(AM_SAMPLE2_PROPERTIES, tStart), (PBYTE)&outProp2)))
1163                            {
1164                                    CMediaType mtOut2 = m_pOutput->CurrentMediaType();
1165                                    VIDEOINFOHEADER2* vihOut2 = (VIDEOINFOHEADER2*)mtOut2.Format();
1166    
1167                                    if (*mtOut2.FormatType() == FORMAT_VideoInfo2 &&
1168                                            vihOut2->dwPictAspectRatioX != ar_x && vihOut2->dwPictAspectRatioY != ar_y)
1169                                    {
1170                                            vihOut2->dwPictAspectRatioX = ar_x;
1171                                            vihOut2->dwPictAspectRatioY = ar_y;
1172                                            if (interlaced) {
1173                                                    vihOut2->dwInterlaceFlags = AMINTERLACE_IsInterlaced;
1174                                                    if (interlaced > 2) {
1175                                                            vihOut2->dwInterlaceFlags |= AMINTERLACE_Field1First;
1176                                                    }
1177                                            }
1178                                            else {
1179                                                    vihOut2->dwInterlaceFlags = 0;
1180                                            }
1181                                            pOut2->SetMediaType(&mtOut2);
1182                                            m_pOutput->SetMediaType(&mtOut2);
1183                                    }
1184                                    pOut2->Release();
1185                            }
1186          }
1187                  }                  }
1188          }          }
1189          else          else
# Line 774  Line 1200 
1200                  m_frame.general &= ~XVID_FILMEFFECT;                  m_frame.general &= ~XVID_FILMEFFECT;
1201    
1202                  length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);                  length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1203                  if (length < 0)                  if (length == XVID_ERR_MEMORY)
1204                            return E_FAIL;
1205                    else if (length < 0)
1206                  {                  {
1207              DPRINTF("*** XVID_DEC_DECODE");              DPRINTF("*** XVID_DEC_DECODE");
1208                          return S_FALSE;                          return S_FALSE;
# Line 799  Line 1227 
1227                          return S_FALSE;                          return S_FALSE;
1228                  }                  }
1229    
 //              pOut->SetDiscontinuity(TRUE);  
1230                  pOut->SetSyncPoint(TRUE);                  pOut->SetSyncPoint(TRUE);
1231    
1232                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) { /* auto */
1233                            int par_x, par_y;
1234                            if (stats.data.vol.par == XVID_PAR_EXT) {
1235                                    par_x = stats.data.vol.par_width;
1236                                    par_y = stats.data.vol.par_height;
1237                            } else {
1238                                    par_x = PARS[stats.data.vol.par-1][0];
1239                                    par_y = PARS[stats.data.vol.par-1][1];
1240                            }
1241    
1242                            ar_x = par_x * stats.data.vol.width;
1243                            ar_y = par_y * stats.data.vol.height;
1244                    }
1245    
1246                  m_frame.bitstream = (BYTE*)m_frame.bitstream + length;                  m_frame.bitstream = (BYTE*)m_frame.bitstream + length;
1247                  m_frame.length -= length;                  m_frame.length -= length;
1248                  goto repeat;                  goto repeat;
# Line 841  Line 1282 
1282          CoTaskMemFree(pPages->pElems);          CoTaskMemFree(pPages->pElems);
1283          return S_OK;          return S_OK;
1284  }  }
1285    
1286    /*===============================================================================
1287    // MFT Interface
1288    //=============================================================================*/
1289    #if defined(XVID_USE_MFT)
1290    #include <limits.h> // _I64_MAX
1291    #define INVALID_TIME  _I64_MAX
1292    
1293    HRESULT CXvidDecoder::MFTGetStreamLimits(DWORD *pdwInputMinimum, DWORD *pdwInputMaximum, DWORD *pdwOutputMinimum, DWORD *pdwOutputMaximum)
1294    {
1295            DPRINTF("(MFT)GetStreamLimits");
1296    
1297            if ((pdwInputMinimum == NULL) || (pdwInputMaximum == NULL) || (pdwOutputMinimum == NULL) || (pdwOutputMaximum == NULL))
1298                    return E_POINTER;
1299    
1300            /* Just a fixed number of streams allowed */
1301            *pdwInputMinimum = *pdwInputMaximum = 1;
1302            *pdwOutputMinimum = *pdwOutputMaximum = 1;
1303    
1304            return S_OK;
1305    }
1306    
1307    HRESULT CXvidDecoder::MFTGetStreamCount(DWORD *pcInputStreams, DWORD *pcOutputStreams)
1308    {
1309            DPRINTF("(MFT)GetStreamCount");
1310    
1311            if ((pcInputStreams == NULL) || (pcOutputStreams == NULL))
1312                    return E_POINTER;
1313    
1314            /* We have a fixed number of streams */
1315            *pcInputStreams = 1;
1316            *pcOutputStreams = 1;
1317    
1318            return S_OK;
1319    }
1320    
1321    HRESULT CXvidDecoder::MFTGetStreamIDs(DWORD dwInputIDArraySize, DWORD *pdwInputIDs, DWORD dwOutputIDArraySize, DWORD *pdwOutputIDs)
1322    {
1323            DPRINTF("(MFT)GetStreamIDs");
1324            return E_NOTIMPL; /* We have fixed number of streams, so stream ID match stream index */
1325    }
1326    
1327    HRESULT CXvidDecoder::MFTGetInputStreamInfo(DWORD dwInputStreamID, MFT_INPUT_STREAM_INFO *pStreamInfo)
1328    {
1329            DPRINTF("(MFT)GetInputStreamInfo");
1330    
1331            if (pStreamInfo == NULL)
1332                    return E_POINTER;
1333    
1334            if (dwInputStreamID != 0)
1335                    return MF_E_INVALIDSTREAMNUMBER;
1336    
1337            EnterCriticalSection(&m_mft_lock);
1338    
1339            pStreamInfo->dwFlags = MFT_INPUT_STREAM_WHOLE_SAMPLES | MFT_INPUT_STREAM_SINGLE_SAMPLE_PER_BUFFER;
1340            pStreamInfo->hnsMaxLatency = 0;
1341    
1342            pStreamInfo->cbSize = 1; /* Need atleast 1 byte input */
1343            pStreamInfo->cbMaxLookahead = 0;
1344            pStreamInfo->cbAlignment = 1;
1345    
1346            LeaveCriticalSection(&m_mft_lock);
1347            return S_OK;
1348    }
1349    
1350    HRESULT CXvidDecoder::MFTGetOutputStreamInfo(DWORD dwOutputStreamID, MFT_OUTPUT_STREAM_INFO *pStreamInfo)
1351    {
1352            DPRINTF("(MFT)GetOutputStreamInfo");
1353    
1354            if (pStreamInfo == NULL)
1355                    return E_POINTER;
1356    
1357            if (dwOutputStreamID != 0)
1358                    return MF_E_INVALIDSTREAMNUMBER;
1359    
1360            EnterCriticalSection(&m_mft_lock);
1361    
1362            pStreamInfo->dwFlags = MFT_OUTPUT_STREAM_WHOLE_SAMPLES | MFT_OUTPUT_STREAM_SINGLE_SAMPLE_PER_BUFFER | MFT_OUTPUT_STREAM_FIXED_SAMPLE_SIZE | MFT_OUTPUT_STREAM_DISCARDABLE;
1363    
1364            if (m_pOutputType == NULL) {
1365                    pStreamInfo->cbSize = 0;
1366                    pStreamInfo->cbAlignment = 0;
1367            }
1368            else {
1369                    pStreamInfo->cbSize = (m_create.width * abs(m_create.height) * m_pOutputTypeBPP) >> 3;
1370                    pStreamInfo->cbAlignment = 1;
1371            }
1372    
1373            LeaveCriticalSection(&m_mft_lock);
1374            return S_OK;
1375    }
1376    
1377    HRESULT CXvidDecoder::GetAttributes(IMFAttributes** pAttributes)
1378    {
1379            DPRINTF("(MFT)GetAttributes");
1380            return E_NOTIMPL; /* We don't support any attributes */
1381    }
1382    
1383    HRESULT CXvidDecoder::GetInputStreamAttributes(DWORD dwInputStreamID, IMFAttributes **ppAttributes)
1384    {
1385            DPRINTF("(MFT)GetInputStreamAttributes");
1386            return E_NOTIMPL; /* We don't support any attributes */
1387    }
1388    
1389    HRESULT CXvidDecoder::GetOutputStreamAttributes(DWORD dwOutputStreamID, IMFAttributes **ppAttributes)
1390    {
1391            DPRINTF("(MFT)GetOutputStreamAttributes");
1392            return E_NOTIMPL; /* We don't support any attributes */
1393    }
1394    
1395    HRESULT CXvidDecoder::MFTDeleteInputStream(DWORD dwStreamID)
1396    {
1397            DPRINTF("(MFT)DeleteInputStream");
1398            return E_NOTIMPL; /* We have a fixed number of streams */
1399    }
1400    
1401    HRESULT CXvidDecoder::MFTAddInputStreams(DWORD cStreams, DWORD *adwStreamIDs)
1402    {
1403            DPRINTF("(MFT)AddInputStreams");
1404            return E_NOTIMPL; /* We have a fixed number of streams */
1405    }
1406    
1407    HRESULT CXvidDecoder::MFTGetInputAvailableType(DWORD dwInputStreamID, DWORD dwTypeIndex, IMFMediaType **ppType)
1408    {
1409            DPRINTF("(MFT)GetInputAvailableType");
1410    
1411            if (dwInputStreamID != 0)
1412                    return MF_E_INVALIDSTREAMNUMBER;
1413    
1414            DWORD i = 0;
1415            GUID *bs_guid_table[32];
1416    
1417            bs_guid_table[i++] = (GUID *)&CLSID_XVID;
1418            bs_guid_table[i++] = (GUID *)&CLSID_XVID_UC;
1419    
1420            if (g_config.supported_4cc & SUPPORT_3IVX) {
1421                    bs_guid_table[i++] = (GUID *)&CLSID_3IVX;
1422                    bs_guid_table[i++] = (GUID *)&CLSID_3IVX_UC;
1423                    bs_guid_table[i++] = (GUID *)&CLSID_3IV0;
1424                    bs_guid_table[i++] = (GUID *)&CLSID_3IV0_UC;
1425                    bs_guid_table[i++] = (GUID *)&CLSID_3IV1;
1426                    bs_guid_table[i++] = (GUID *)&CLSID_3IV1_UC;
1427                    bs_guid_table[i++] = (GUID *)&CLSID_3IV2;
1428                    bs_guid_table[i++] = (GUID *)&CLSID_3IV2_UC;
1429            }
1430            if (g_config.supported_4cc & SUPPORT_DIVX) {
1431                    bs_guid_table[i++] = (GUID *)&CLSID_DIVX;
1432                    bs_guid_table[i++] = (GUID *)&CLSID_DIVX_UC;
1433                    bs_guid_table[i++] = (GUID *)&CLSID_DX50;
1434                    bs_guid_table[i++] = (GUID *)&CLSID_DX50_UC;
1435            }
1436            if (g_config.supported_4cc & SUPPORT_MP4V) {
1437                    bs_guid_table[i++] = (GUID *)&CLSID_MP4V;
1438                    bs_guid_table[i++] = (GUID *)&CLSID_MP4V_UC;
1439                    bs_guid_table[i++] = (GUID *)&CLSID_LMP4;
1440                    bs_guid_table[i++] = (GUID *)&CLSID_LMP4_UC;
1441                    bs_guid_table[i++] = (GUID *)&CLSID_RMP4;
1442                    bs_guid_table[i++] = (GUID *)&CLSID_RMP4_UC;
1443                    bs_guid_table[i++] = (GUID *)&CLSID_SMP4;
1444                    bs_guid_table[i++] = (GUID *)&CLSID_SMP4_UC;
1445                    bs_guid_table[i++] = (GUID *)&CLSID_HDX4;
1446                    bs_guid_table[i++] = (GUID *)&CLSID_HDX4_UC;
1447            }
1448    
1449            const GUID *subtype;
1450            if (dwTypeIndex < i) {
1451                    subtype = bs_guid_table[dwTypeIndex];
1452            }
1453            else {
1454                    return MF_E_NO_MORE_TYPES;
1455            }
1456    
1457            EnterCriticalSection(&m_mft_lock);
1458    
1459            HRESULT hr = S_OK;
1460    
1461            if (ppType) {
1462                    IMFMediaType *pInputType = NULL;
1463                    hr = MFCreateMediaType(&pInputType);
1464    
1465                    if (SUCCEEDED(hr))
1466                            hr = pInputType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video);
1467    
1468                    if (SUCCEEDED(hr))
1469                            hr = pInputType->SetGUID(MF_MT_SUBTYPE, *subtype);
1470    
1471                    if (SUCCEEDED(hr)) {
1472                            *ppType = pInputType;
1473                            (*ppType)->AddRef();
1474                    }
1475                    if (pInputType) pInputType->Release();
1476            }
1477    
1478            LeaveCriticalSection(&m_mft_lock);
1479    
1480            return hr;
1481    }
1482    
1483    HRESULT CXvidDecoder::MFTGetOutputAvailableType(DWORD dwOutputStreamID, DWORD dwTypeIndex, IMFMediaType **ppType)
1484    {
1485            DPRINTF("(MFT)GetOutputAvailableType");
1486    
1487            if (ppType == NULL)
1488                    return E_INVALIDARG;
1489    
1490            if (dwOutputStreamID != 0)
1491                    return MF_E_INVALIDSTREAMNUMBER;
1492    
1493            if (dwTypeIndex < 0) return E_INVALIDARG;
1494    
1495            GUID csp;
1496            int bitdepth = 8;
1497            switch(dwTypeIndex)
1498            {
1499            case 0:
1500    if ( USE_YUY2 )
1501    {
1502                    csp = MFVideoFormat_YUY2;
1503                    bitdepth = 4;
1504                    break;
1505    }
1506            case 1:
1507    if ( USE_YVYU )
1508    {
1509                csp = MFVideoFormat_YVYU;
1510                bitdepth = 4;
1511                break;
1512    }
1513            case 2 :
1514    if ( USE_UYVY )
1515    {
1516                    csp = MFVideoFormat_UYVY;
1517                    bitdepth = 4;
1518                    break;
1519    }
1520            case 3  :
1521                    if ( USE_IYUV )
1522    {
1523                    csp = MFVideoFormat_IYUV;
1524                    bitdepth = 3;
1525                    break;
1526    }
1527            case 4  :
1528    if ( USE_YV12 )
1529    {
1530                    csp = MFVideoFormat_YV12;
1531                    bitdepth = 3;
1532                    break;
1533    }
1534            case 5 :
1535    if ( USE_RGB32 )
1536    {
1537                    csp = MFVideoFormat_RGB32;
1538                    bitdepth = 8;
1539                    break;
1540    }
1541            case 6 :
1542    if ( USE_RGB24 )
1543    {
1544                    csp = MFVideoFormat_RGB24;
1545                    bitdepth = 6;
1546                    break;
1547    }
1548            case 7 :
1549    if ( USE_RG555 )
1550    {
1551                    csp = MFVideoFormat_RGB555;
1552                    bitdepth = 4;
1553                    break;
1554    }
1555            case 8 :
1556    if ( USE_RG565 )
1557    {
1558                    csp = MFVideoFormat_RGB565;
1559                    bitdepth = 4;
1560                    break;
1561    }
1562            default :
1563                    return MF_E_NO_MORE_TYPES;
1564            }
1565    
1566            if (m_pInputType == NULL)
1567                    return MF_E_TRANSFORM_TYPE_NOT_SET;
1568    
1569            EnterCriticalSection(&m_mft_lock);
1570    
1571            HRESULT hr = S_OK;
1572    
1573            IMFMediaType *pOutputType = NULL;
1574            hr = MFCreateMediaType(&pOutputType);
1575    
1576            if (SUCCEEDED(hr)) {
1577                    hr = pOutputType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video);
1578            }
1579    
1580            if (SUCCEEDED(hr)) {
1581                    hr = pOutputType->SetGUID(MF_MT_SUBTYPE, csp);
1582        }
1583    
1584            if (SUCCEEDED(hr)) {
1585                    hr = pOutputType->SetUINT32(MF_MT_FIXED_SIZE_SAMPLES, TRUE);
1586            }
1587    
1588            if (SUCCEEDED(hr)) {
1589                    hr = pOutputType->SetUINT32(MF_MT_ALL_SAMPLES_INDEPENDENT, TRUE);
1590            }
1591    
1592            if (SUCCEEDED(hr)) {
1593                    hr = pOutputType->SetUINT32(MF_MT_SAMPLE_SIZE, (abs(m_create.height) * m_create.width * bitdepth) >> 1);
1594            }
1595    
1596            if (SUCCEEDED(hr)) {
1597                    hr = MFSetAttributeSize(pOutputType, MF_MT_FRAME_SIZE, m_create.width, abs(m_create.height));
1598            }
1599    
1600            if (SUCCEEDED(hr)) {
1601                    hr = MFSetAttributeRatio(pOutputType, MF_MT_FRAME_RATE, m_frameRate.Numerator, m_frameRate.Denominator);
1602            }
1603    
1604            if (SUCCEEDED(hr)) {
1605                    if (interlaced > 1) {
1606                            hr = pOutputType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_FieldInterleavedUpperFirst);
1607                    }
1608                    else if (interlaced) {
1609                            hr = pOutputType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_FieldInterleavedLowerFirst);
1610                    }
1611                    else {
1612                            hr = pOutputType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_Progressive);
1613                    }
1614            }
1615    
1616            if (SUCCEEDED(hr)) {
1617                    hr = MFSetAttributeRatio(pOutputType, MF_MT_PIXEL_ASPECT_RATIO, ar_x, ar_y);
1618            }
1619    
1620            if (SUCCEEDED(hr)) {
1621                    *ppType = pOutputType;
1622                    (*ppType)->AddRef();
1623            }
1624    
1625            if (pOutputType) pOutputType->Release();
1626    
1627            LeaveCriticalSection(&m_mft_lock);
1628            return hr;
1629    }
1630    
1631    HRESULT CXvidDecoder::MFTSetInputType(DWORD dwInputStreamID, IMFMediaType *pType, DWORD dwFlags)
1632    {
1633            DPRINTF("(MFT)SetInputType");
1634    
1635            if (dwInputStreamID != 0)
1636                    return MF_E_INVALIDSTREAMNUMBER;
1637    
1638            if (dwFlags & ~MFT_SET_TYPE_TEST_ONLY)
1639                    return E_INVALIDARG;
1640    
1641            EnterCriticalSection(&m_mft_lock);
1642    
1643            HRESULT hr = S_OK;
1644    
1645            /* Actually set the type or just test it? */
1646            BOOL bReallySet = ((dwFlags & MFT_SET_TYPE_TEST_ONLY) == 0);
1647    
1648            /* If we have samples pending the type can't be changed right now */
1649            if (HasPendingOutput())
1650                    hr = MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
1651    
1652            if (SUCCEEDED(hr)) {
1653            if (pType) { /* Check the type */
1654                hr = OnCheckInputType(pType);
1655            }
1656            }
1657    
1658            if (SUCCEEDED(hr)) {
1659                    if (bReallySet) { /* Set the type if needed */
1660                            hr = OnSetInputType(pType);
1661                    }
1662            }
1663    
1664            LeaveCriticalSection(&m_mft_lock);
1665            return hr;
1666    }
1667    
1668    DWORD WINAPI CreateTrayIcon(LPVOID lpParameter)
1669    {
1670            WNDCLASSEX wc;
1671    
1672            wc.cbSize = sizeof(WNDCLASSEX);
1673            wc.lpfnWndProc = msg_proc;
1674            wc.style = CS_HREDRAW | CS_VREDRAW;
1675            wc.cbWndExtra = 0;
1676            wc.cbClsExtra = 0;
1677            wc.hInstance = (HINSTANCE)g_xvid_hInst;
1678            wc.hbrBackground = (HBRUSH)GetStockObject(NULL_BRUSH);
1679            wc.lpszMenuName = NULL;
1680            wc.lpszClassName = "XVID_MSG_WINDOW";
1681            wc.hIcon = NULL;
1682            wc.hIconSm = NULL;
1683            wc.hCursor = NULL;
1684            RegisterClassEx(&wc);
1685    
1686            MSG_hwnd = CreateWindowEx(0, "XVID_MSG_WINDOW", NULL, 0, CW_USEDEFAULT,
1687                    CW_USEDEFAULT, 0, 0, HWND_MESSAGE, NULL, (HINSTANCE)g_xvid_hInst, NULL);
1688    
1689            /* display the tray icon */
1690            NOTIFYICONDATA nid;
1691            ZeroMemory(&nid, sizeof(NOTIFYICONDATA));
1692    
1693            nid.cbSize = sizeof(NOTIFYICONDATA);
1694            nid.hWnd = MSG_hwnd;
1695            nid.uID = 1456;
1696            nid.uVersion = NOTIFYICON_VERSION;
1697            nid.uCallbackMessage = WM_ICONMESSAGE;
1698            nid.hIcon = LoadIcon(g_xvid_hInst, MAKEINTRESOURCE(IDI_ICON));
1699            strcpy_s(nid.szTip, 19, "Xvid Video Decoder");
1700            nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP | NIF_SHOWTIP;
1701    
1702            Shell_NotifyIcon(NIM_ADD, &nid);
1703            Shell_NotifyIcon(NIM_SETVERSION, &nid);
1704    
1705            DestroyIcon(nid.hIcon);
1706    
1707            MSG msg;
1708            while (MSG_hwnd && GetMessage(&msg, MSG_hwnd, 0, 0)) {
1709                    TranslateMessage(&msg);
1710                    DispatchMessage(&msg);
1711            }
1712    
1713            return 0;
1714    }
1715    
1716    HRESULT CXvidDecoder::MFTSetOutputType(DWORD dwOutputStreamID, IMFMediaType *pType, DWORD dwFlags)
1717    {
1718            DPRINTF("(MFT)SetOutputType");
1719    
1720            if (dwOutputStreamID != 0)
1721                    return MF_E_INVALIDSTREAMNUMBER;
1722    
1723            if (dwFlags & ~MFT_SET_TYPE_TEST_ONLY)
1724                    return E_INVALIDARG;
1725    
1726            HRESULT hr = S_OK;
1727    
1728            EnterCriticalSection(&m_mft_lock);
1729    
1730            /* Actually set the type or just test it? */
1731            BOOL bReallySet = ((dwFlags & MFT_SET_TYPE_TEST_ONLY) == 0);
1732    
1733            /* If we have samples pending the type can't be changed right now */
1734            if (HasPendingOutput())
1735                    hr = MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
1736    
1737            int bitdepth;
1738            if (SUCCEEDED(hr)) {
1739                    if (pType) { /* Check the type */
1740                            AM_MEDIA_TYPE *am;
1741                            hr = MFCreateAMMediaTypeFromMFMediaType(pType, GUID_NULL, &am);
1742    
1743                            if (SUCCEEDED(hr)) {
1744                                    if (FAILED(ChangeColorspace(am->subtype, am->formattype, am->pbFormat, &bitdepth, 1))) {
1745                                            DPRINTF("(MFT)InternalCheckOutputType (MF_E_INVALIDTYPE)");
1746                                            return MF_E_INVALIDTYPE;
1747                                    }
1748    
1749                            CoTaskMemFree(am->pbFormat);
1750                            CoTaskMemFree(am);
1751                            }
1752                    }
1753            }
1754    
1755            if (SUCCEEDED(hr)) {
1756                    if (bReallySet) { /* Set the type if needed */
1757                            hr = OnSetOutputType(pType, bitdepth);
1758                    }
1759            }
1760    
1761            if (SUCCEEDED(hr) && (MSG_hwnd == 0) && (m_tray_icon == 0) && (g_config.bTrayIcon != 0))
1762            {
1763                    m_thread_handle = CreateThread(NULL, 0, CreateTrayIcon, NULL, 0, NULL);  /* Create message passing window */
1764    
1765                    if (m_thread_handle)
1766                        m_tray_icon = 1;
1767            }
1768    
1769            LeaveCriticalSection(&m_mft_lock);
1770    
1771            return hr;
1772    }
1773    
1774    HRESULT CXvidDecoder::MFTGetInputCurrentType(DWORD dwInputStreamID, IMFMediaType **ppType)
1775    {
1776            DPRINTF("(MFT)GetInputCurrentType");
1777    
1778            if (ppType == NULL)
1779                    return E_POINTER;
1780    
1781            if (dwInputStreamID != 0)
1782                    return MF_E_INVALIDSTREAMNUMBER;
1783    
1784            EnterCriticalSection(&m_mft_lock);
1785    
1786            HRESULT hr = S_OK;
1787    
1788            if (!m_pInputType)
1789                    hr = MF_E_TRANSFORM_TYPE_NOT_SET;
1790    
1791            if (SUCCEEDED(hr)) {
1792                    *ppType = m_pInputType;
1793                    (*ppType)->AddRef();
1794            }
1795    
1796            LeaveCriticalSection(&m_mft_lock);
1797            return hr;
1798    }
1799    
1800    HRESULT CXvidDecoder::MFTGetOutputCurrentType(DWORD dwOutputStreamID, IMFMediaType **ppType)
1801    {
1802            DPRINTF("(MFT)GetOutputCurrentType");
1803    
1804            if (ppType == NULL)
1805                    return E_POINTER;
1806    
1807            if (dwOutputStreamID != 0)
1808                    return MF_E_INVALIDSTREAMNUMBER;
1809    
1810            EnterCriticalSection(&m_mft_lock);
1811    
1812            HRESULT hr = S_OK;
1813    
1814            if (!m_pOutputType)
1815                    hr = MF_E_TRANSFORM_TYPE_NOT_SET;
1816    
1817            if (SUCCEEDED(hr)) {
1818                    *ppType = m_pOutputType;
1819                    (*ppType)->AddRef();
1820            }
1821    
1822            LeaveCriticalSection(&m_mft_lock);
1823            return hr;
1824    }
1825    
1826    HRESULT CXvidDecoder::MFTGetInputStatus(DWORD dwInputStreamID, DWORD *pdwFlags)
1827    {
1828            DPRINTF("(MFT)GetInputStatus");
1829    
1830            if (pdwFlags == NULL)
1831                    return E_POINTER;
1832    
1833            if (dwInputStreamID != 0)
1834                    return MF_E_INVALIDSTREAMNUMBER;
1835    
1836            EnterCriticalSection(&m_mft_lock);
1837    
1838            /* If there's pending output samples we don't accept new
1839               input data until ProcessOutput() or Flush() was called */
1840            if (!HasPendingOutput()) {
1841                    *pdwFlags = MFT_INPUT_STATUS_ACCEPT_DATA;
1842            }
1843            else {
1844                    *pdwFlags = 0;
1845            }
1846    
1847            LeaveCriticalSection(&m_mft_lock);
1848    
1849            return S_OK;
1850    }
1851    
1852    HRESULT CXvidDecoder::MFTGetOutputStatus(DWORD *pdwFlags)
1853    {
1854            DPRINTF("(MFT)GetOutputStatus");
1855    
1856            if (pdwFlags == NULL)
1857                    return E_POINTER;
1858    
1859            EnterCriticalSection(&m_mft_lock);
1860    
1861            /* We can render an output sample only after we
1862               have decoded one */
1863            if (HasPendingOutput()) {
1864                    *pdwFlags = MFT_OUTPUT_STATUS_SAMPLE_READY;
1865            }
1866            else {
1867                    *pdwFlags = 0;
1868            }
1869    
1870            LeaveCriticalSection(&m_mft_lock);
1871    
1872            return S_OK;
1873    }
1874    
1875    HRESULT CXvidDecoder::MFTSetOutputBounds(LONGLONG hnsLowerBound, LONGLONG hnsUpperBound)
1876    {
1877            DPRINTF("(MFT)SetOutputBounds");
1878            return E_NOTIMPL;
1879    }
1880    
1881    HRESULT CXvidDecoder::MFTProcessEvent(DWORD dwInputStreamID, IMFMediaEvent *pEvent)
1882    {
1883            DPRINTF("(MFT)ProcessEvent");
1884            return E_NOTIMPL; /* We don't handle any stream events */
1885    }
1886    
1887    HRESULT CXvidDecoder::MFTProcessMessage(MFT_MESSAGE_TYPE eMessage, ULONG_PTR ulParam)
1888    {
1889            DPRINTF("(MFT)ProcessMessage");
1890            HRESULT hr = S_OK;
1891    
1892            EnterCriticalSection(&m_mft_lock);
1893    
1894            switch (eMessage)
1895            {
1896            case MFT_MESSAGE_COMMAND_FLUSH:
1897                    if (m_create.handle != NULL) {
1898                            DPRINTF("(MFT)CommandFlush");
1899    
1900                            xvid_dec_stats_t stats;
1901                            int used_bytes;
1902    
1903                            memset(&stats, 0, sizeof(stats));
1904                            stats.version = XVID_VERSION;
1905    
1906                            int csp = m_frame.output.csp;
1907    
1908                            m_frame.output.csp = XVID_CSP_INTERNAL;
1909                            m_frame.bitstream = NULL;
1910                            m_frame.length = -1;
1911                            m_frame.general = XVID_LOWDELAY;
1912    
1913                            do {
1914                                    used_bytes = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1915                            } while(used_bytes>=0 && stats.type <= 0);
1916    
1917                            m_frame.output.csp = csp;
1918                            m_frame.output.plane[1] = NULL; /* Don't display flushed samples */
1919    
1920                            //m_timestamp = INVALID_TIME;
1921                            //m_timelength = INVALID_TIME;
1922                            //m_rtFrame = 0;
1923            }
1924        break;
1925    
1926            case MFT_MESSAGE_COMMAND_DRAIN:
1927                    m_discont = 1; /* Set discontinuity flag */
1928                    m_rtFrame = 0;
1929                    break;
1930    
1931            case MFT_MESSAGE_SET_D3D_MANAGER:
1932                    hr = E_NOTIMPL;
1933                    break;
1934    
1935            case MFT_MESSAGE_NOTIFY_BEGIN_STREAMING:
1936            case MFT_MESSAGE_NOTIFY_END_STREAMING:
1937                    break;
1938    
1939            case MFT_MESSAGE_NOTIFY_START_OF_STREAM:
1940            case MFT_MESSAGE_NOTIFY_END_OF_STREAM:
1941                    break;
1942            }
1943    
1944            LeaveCriticalSection(&m_mft_lock);
1945    
1946            return hr;
1947    }
1948    
1949    HRESULT CXvidDecoder::MFTProcessInput(DWORD dwInputStreamID, IMFSample *pSample, DWORD dwFlags)
1950    {
1951            DPRINTF("(MFT)ProcessInput");
1952    
1953            if (pSample == NULL)
1954                    return E_POINTER;
1955    
1956            if (dwInputStreamID != 0)
1957                    return MF_E_INVALIDSTREAMNUMBER;
1958    
1959            if (dwFlags != 0)
1960                    return E_INVALIDARG;
1961    
1962            if (!m_pInputType || !m_pOutputType) {
1963                    return MF_E_NOTACCEPTING;   /* Must have set input and output types */
1964            }
1965            else if (HasPendingOutput()) {
1966                    return MF_E_NOTACCEPTING;   /* We still have output samples to render */
1967            }
1968    
1969            xvid_dec_stats_t stats;
1970            int length;
1971    
1972            memset(&stats, 0, sizeof(stats));
1973            stats.version = XVID_VERSION;
1974    
1975            if (m_create.handle == NULL)
1976            {
1977                    if (xvid_decore_func == NULL)
1978                            return E_FAIL;
1979                    if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)
1980                    {
1981                            DPRINTF("*** XVID_DEC_CREATE error");
1982                            return E_FAIL;
1983                    }
1984            }
1985    
1986            EnterCriticalSection(&m_mft_lock);
1987    
1988            HRESULT hr = S_OK;
1989            IMFMediaBuffer *pBuffer;
1990    
1991            if (SUCCEEDED(hr)) {
1992                    hr = pSample->ConvertToContiguousBuffer(&pBuffer);
1993            }
1994    
1995            if (SUCCEEDED(hr)) {
1996                    hr = pBuffer->Lock((BYTE**)&m_frame.bitstream, NULL, (DWORD *)&m_frame.length);
1997            }
1998    
1999            m_frame.general = XVID_LOWDELAY;
2000    
2001            if (m_discont == 1) {
2002                    m_frame.general |= XVID_DISCONTINUITY;
2003                    m_discont = 0;
2004            }
2005    
2006            if (g_config.nDeblock_Y)
2007                    m_frame.general |= XVID_DEBLOCKY;
2008    
2009            if (g_config.nDeblock_UV)
2010                    m_frame.general |= XVID_DEBLOCKUV;
2011    
2012            if (g_config.nDering_Y)
2013                    m_frame.general |= XVID_DERINGY;
2014    
2015            if (g_config.nDering_UV)
2016                    m_frame.general |= XVID_DERINGUV;
2017    
2018            if (g_config.nFilmEffect)
2019                    m_frame.general |= XVID_FILMEFFECT;
2020    
2021            m_frame.brightness = g_config.nBrightness;
2022    
2023            m_frame.output.csp &= ~XVID_CSP_VFLIP;
2024            m_frame.output.csp |= rgb_flip^(g_config.nFlipVideo ? XVID_CSP_VFLIP : 0);
2025    
2026            int csp = m_frame.output.csp;
2027            m_frame.output.csp = XVID_CSP_INTERNAL;
2028    
2029        // Paranoid check.
2030            if (xvid_decore_func == NULL) {
2031                    hr = E_FAIL;
2032                    goto END_LOOP;
2033            }
2034    
2035    repeat :
2036            length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
2037    
2038            if (length == XVID_ERR_MEMORY) {
2039                    hr = E_FAIL;
2040                    goto END_LOOP;
2041            }
2042            else if (length < 0)
2043            {
2044                    DPRINTF("*** XVID_DEC_DECODE");
2045                    goto END_LOOP;
2046            }
2047    
2048            if (stats.type == XVID_TYPE_NOTHING && length > 0) {
2049                    DPRINTF(" B-Frame decoder lag");
2050                    m_frame.output.plane[1] = NULL;
2051                    goto END_LOOP;
2052            }
2053    
2054            if (stats.type == XVID_TYPE_VOL)
2055            {
2056                    if (stats.data.vol.width != m_create.width ||
2057                            stats.data.vol.height != m_create.height)
2058                    {
2059                            DPRINTF("TODO: auto-resize");
2060                            m_frame.output.plane[1] = NULL;
2061                            hr = E_FAIL;
2062                    }
2063    
2064                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) { /* auto */
2065                            int par_x, par_y;
2066                            if (stats.data.vol.par == XVID_PAR_EXT) {
2067                                    par_x = stats.data.vol.par_width;
2068                                    par_y = stats.data.vol.par_height;
2069                            } else {
2070                                    par_x = PARS[stats.data.vol.par-1][0];
2071                                    par_y = PARS[stats.data.vol.par-1][1];
2072                            }
2073    
2074                            ar_x = par_x * stats.data.vol.width;
2075                            ar_y = par_y * stats.data.vol.height; /* TODO: Actually set the newly determined AR on the output sample type
2076                                                                                                               or it'll have no effect at all... */
2077                    }
2078    
2079                    if (stats.data.vol.general & XVID_VOL_INTERLACING) {
2080                            interlaced = (stats.data.vop.general & XVID_VOP_TOPFIELDFIRST) ? 2 : 1;
2081                    }
2082                    else {
2083                            interlaced = 0;
2084                    }
2085    
2086                    m_frame.bitstream = (BYTE*)m_frame.bitstream + length;
2087                    m_frame.length -= length;
2088                    goto repeat;
2089            }
2090    
2091    END_LOOP:
2092            m_frame.output.csp = csp;
2093    
2094            if (pBuffer) {
2095                    pBuffer->Unlock();
2096                    pBuffer->Release();
2097            }
2098    
2099            if (SUCCEEDED(hr)) {
2100                    /* Try to get a timestamp */
2101                    if (FAILED(pSample->GetSampleTime(&m_timestamp)))
2102                            m_timestamp = INVALID_TIME;
2103    
2104                    if (FAILED(pSample->GetSampleDuration(&m_timelength))) {
2105                            m_timelength = INVALID_TIME;
2106                    }
2107                    if (m_timestamp != INVALID_TIME && stats.type == XVID_TYPE_IVOP) {
2108                            m_rtFrame = m_timestamp;
2109                    }
2110            }
2111    
2112            LeaveCriticalSection(&m_mft_lock);
2113    
2114            return hr;
2115    }
2116    
2117    HRESULT CXvidDecoder::MFTProcessOutput(DWORD dwFlags, DWORD cOutputBufferCount, MFT_OUTPUT_DATA_BUFFER *pOutputSamples, DWORD *pdwStatus)
2118    {
2119            DPRINTF("(MFT)ProcessOutput");
2120    
2121            /* Preroll in MFT ??
2122               Flags ?? -> TODO... */
2123            if (dwFlags != 0)
2124                    return E_INVALIDARG;
2125    
2126            if (pOutputSamples == NULL || pdwStatus == NULL)
2127                    return E_POINTER;
2128    
2129            if (cOutputBufferCount != 1) /* Must be exactly one output buffer */
2130                    return E_INVALIDARG;
2131    
2132            if (pOutputSamples[0].pSample == NULL) /* Must have a sample */
2133                    return E_INVALIDARG;
2134    
2135            if (!HasPendingOutput()) { /* If there's no sample we need to decode one first */
2136                    return MF_E_TRANSFORM_NEED_MORE_INPUT;
2137            }
2138    
2139            EnterCriticalSection(&m_mft_lock);
2140    
2141            HRESULT hr = S_OK;
2142    
2143            BYTE *Dst = NULL;
2144            DWORD buffer_size;
2145            LONG stride = m_create.width;
2146            IMFMediaBuffer *pOutput = NULL;
2147            IMF2DBuffer *pOutput2D = NULL;
2148    
2149            if (SUCCEEDED(hr)) {
2150                    hr = pOutputSamples[0].pSample->GetBufferByIndex(0, &pOutput); /* Get output buffer */
2151            }
2152    
2153            if (SUCCEEDED(hr)) {
2154                    hr = pOutput->GetMaxLength(&buffer_size);
2155            }
2156    
2157            if (SUCCEEDED(hr)) {
2158                    hr = pOutput->QueryInterface(IID_IMF2DBuffer, (void **)&pOutput2D);
2159            }
2160    
2161            if (SUCCEEDED(hr)) {
2162                    hr = pOutput2D->Lock2D(&Dst, &stride);
2163            }
2164            else {
2165                    hr = pOutput->Lock(&Dst, NULL, NULL);
2166            }
2167    
2168            if (SUCCEEDED(hr)) {
2169                    xvid_gbl_convert_t convert;
2170    
2171                    memset(&convert, 0, sizeof(convert));
2172                    convert.version = XVID_VERSION;
2173    
2174                    convert.input.csp = XVID_CSP_INTERNAL;
2175                    convert.input.plane[0] = m_frame.output.plane[0];
2176                    convert.input.plane[1] = m_frame.output.plane[1];
2177                    convert.input.plane[2] = m_frame.output.plane[2];
2178                    convert.input.stride[0] = m_frame.output.stride[0];
2179                    convert.input.stride[1] = m_frame.output.stride[1];
2180                    convert.input.stride[2] = m_frame.output.stride[2];
2181    
2182                    convert.output.csp = m_frame.output.csp;
2183                    convert.output.plane[0] = Dst;
2184                    convert.output.stride[0] = stride;
2185    
2186                    convert.width = m_create.width;
2187                    convert.height = m_create.height;
2188                    convert.interlacing = (interlaced > 0) ? 1 : 0;
2189    
2190                    if (m_frame.output.plane[1] != NULL && Dst != NULL && xvid_global_func != NULL)
2191                            if (xvid_global_func(0, XVID_GBL_CONVERT, &convert, NULL) < 0) /* CSP convert into output buffer */
2192                                    hr = E_FAIL;
2193                    m_frame.output.plane[1] = NULL;
2194            }
2195    
2196            *pdwStatus = 0;
2197    
2198            if (SUCCEEDED(hr)) {
2199                    if (SUCCEEDED(hr))
2200                            hr = pOutputSamples[0].pSample->SetUINT32(MFSampleExtension_CleanPoint, TRUE); // key frame
2201    
2202                    if (SUCCEEDED(hr)) { /* Set timestamp of output sample */
2203                            if (m_timestamp != INVALID_TIME)
2204                                    hr = pOutputSamples[0].pSample->SetSampleTime(m_timestamp);
2205                            else
2206                                    hr = pOutputSamples[0].pSample->SetSampleTime(m_rtFrame);
2207    
2208                            if (m_timelength != INVALID_TIME)
2209                                    hr = pOutputSamples[0].pSample->SetSampleDuration(m_timelength);
2210                            else
2211                                    hr = pOutputSamples[0].pSample->SetSampleDuration(m_duration);
2212    
2213                            m_rtFrame += m_duration;
2214                    }
2215    
2216                    if (SUCCEEDED(hr))
2217                            hr = pOutput->SetCurrentLength((m_create.width * abs(m_create.height) * m_pOutputTypeBPP) >> 3);
2218            }
2219    
2220            if (pOutput2D) {
2221                    pOutput2D->Unlock2D();
2222                    pOutput2D->Release();
2223                    if (pOutput)
2224                        pOutput->Release();
2225            }
2226            else if (pOutput) {
2227                    pOutput->Unlock();
2228                    pOutput->Release();
2229            }
2230    
2231            LeaveCriticalSection(&m_mft_lock);
2232    
2233            return hr;
2234    }
2235    
2236    HRESULT CXvidDecoder::OnCheckInputType(IMFMediaType *pmt)
2237    {
2238            DPRINTF("(MFT)CheckInputType");
2239    
2240            HRESULT hr = S_OK;
2241    
2242            /*  Check if input type is already set. Reject any type that is not identical */
2243            if (m_pInputType) {
2244                    DWORD dwFlags = 0;
2245                    if (S_OK == m_pInputType->IsEqual(pmt, &dwFlags)) {
2246                            return S_OK;
2247                    }
2248                    else {
2249                            return MF_E_INVALIDTYPE;
2250                    }
2251            }
2252    
2253            GUID majortype = {0}, subtype = {0};
2254            UINT32 width = 0, height = 0;
2255    
2256            hr = pmt->GetMajorType(&majortype);
2257    
2258            if (SUCCEEDED(hr)) {
2259                    if (majortype != MFMediaType_Video) { /* Must be Video */
2260                            hr = MF_E_INVALIDTYPE;
2261                    }
2262            }
2263    
2264            if (m_hdll == NULL) {
2265                    HRESULT hr = OpenLib();
2266    
2267                    if (FAILED(hr) || (m_hdll == NULL)) // Paranoid checks.
2268                            hr = MF_E_INVALIDTYPE;
2269            }
2270    
2271            if (SUCCEEDED(hr)) {
2272                    hr = MFGetAttributeSize(pmt, MF_MT_FRAME_SIZE, &width, &height);
2273            }
2274    
2275            /* Check the frame size */
2276            if (SUCCEEDED(hr)) {
2277                    if (width > 4096 || height > 4096) {
2278                            hr = MF_E_INVALIDTYPE;
2279                    }
2280            }
2281            m_create.width = width;
2282            m_create.height = height;
2283    
2284            if (SUCCEEDED(hr)) {
2285                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
2286                            hr = MFGetAttributeRatio(pmt, MF_MT_PIXEL_ASPECT_RATIO, (UINT32*)&ar_x, (UINT32*)&ar_y);
2287                    }
2288            }
2289    
2290            /* TODO1: Make sure there really is a frame rate after all!
2291               TODO2: Use the framerate for something! */
2292            MFRatio fps = {0};
2293            if (SUCCEEDED(hr)) {
2294                    hr = MFGetAttributeRatio(pmt, MF_MT_FRAME_RATE, (UINT32*)&fps.Numerator, (UINT32*)&fps.Denominator);
2295            }
2296    
2297            if (SUCCEEDED(hr)) {
2298                    hr = pmt->GetGUID(MF_MT_SUBTYPE, &subtype);
2299            }
2300    
2301            if (subtype == CLSID_MP4V || subtype == CLSID_MP4V_UC ||
2302                subtype == CLSID_LMP4 || subtype == CLSID_LMP4_UC ||
2303                subtype == CLSID_RMP4 || subtype == CLSID_RMP4_UC ||
2304                subtype == CLSID_SMP4 || subtype == CLSID_SMP4_UC ||
2305                subtype == CLSID_HDX4 || subtype == CLSID_HDX4_UC) {
2306                    if (!(g_config.supported_4cc & SUPPORT_MP4V)) {
2307                            CloseLib();
2308                            hr = MF_E_INVALIDTYPE;
2309                    }
2310                    else m_create.fourcc = FOURCC_MP4V;
2311            }
2312            else if (subtype == CLSID_DIVX || subtype == CLSID_DIVX_UC) {
2313                    if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
2314                            CloseLib();
2315                            hr = MF_E_INVALIDTYPE;
2316                    }
2317                    else m_create.fourcc = FOURCC_DIVX;
2318            }
2319            else if (subtype == CLSID_DX50 || subtype == CLSID_DX50_UC) {
2320                    if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
2321                            CloseLib();
2322                            hr = MF_E_INVALIDTYPE;
2323                    }
2324                    else m_create.fourcc = FOURCC_DX50;
2325            }
2326            else if (subtype == CLSID_3IVX || subtype == CLSID_3IVX_UC ||
2327                     subtype == CLSID_3IV0 || subtype == CLSID_3IV0_UC ||
2328                     subtype == CLSID_3IV1 || subtype == CLSID_3IV1_UC ||
2329                     subtype == CLSID_3IV2 || subtype == CLSID_3IV2_UC) {
2330                    if (!(g_config.supported_4cc & SUPPORT_3IVX)) {
2331                            CloseLib();
2332                            hr = MF_E_INVALIDTYPE;
2333                    }
2334                    else m_create.fourcc = FOURCC_3IVX;
2335            }
2336            else if (subtype == CLSID_XVID || subtype == CLSID_XVID_UC) {
2337                    m_create.fourcc = FOURCC_XVID;
2338            }
2339            else {
2340                    DPRINTF("Unknown subtype!");
2341                    CloseLib();
2342                    hr = MF_E_INVALIDTYPE;
2343            }
2344    
2345            /* haali media splitter reports VOL information in the format header */
2346            if (SUCCEEDED(hr))
2347            {
2348                    UINT32 cbSeqHeader = 0;
2349    
2350                    (void)pmt->GetBlobSize(MF_MT_MPEG_SEQUENCE_HEADER, &cbSeqHeader);
2351    
2352                    if (cbSeqHeader>0) {
2353                            xvid_dec_stats_t stats;
2354                            memset(&stats, 0, sizeof(stats));
2355                            stats.version = XVID_VERSION;
2356    
2357                            if (m_create.handle == NULL) {
2358                                    if (xvid_decore_func == NULL)
2359                                            hr = E_FAIL;
2360                                    if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) {
2361                                            DPRINTF("*** XVID_DEC_CREATE error");
2362                                            hr = E_FAIL;
2363                                    }
2364                            }
2365    
2366                            if (SUCCEEDED(hr)) {
2367                                    (void)pmt->GetAllocatedBlob(MF_MT_MPEG_SEQUENCE_HEADER, (UINT8 **)&m_frame.bitstream, (UINT32 *)&m_frame.length);
2368                                    m_frame.general = 0;
2369                                    m_frame.output.csp = XVID_CSP_NULL;
2370    
2371                                    int ret = 0;
2372                                    if ((ret=xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats)) >= 0) {
2373                                            /* honour video dimensions reported in VOL header */
2374                                            if (stats.type == XVID_TYPE_VOL) {
2375                                                    m_create.width = stats.data.vol.width;
2376                                                    m_create.height = stats.data.vol.height;
2377                                            }
2378                                    }
2379    
2380                                    if (ret == XVID_ERR_MEMORY) hr = E_FAIL;
2381                                    CoTaskMemFree(m_frame.bitstream);
2382                            }
2383                    }
2384            }
2385    
2386            return hr;
2387    }
2388    
2389    HRESULT CXvidDecoder::OnSetInputType(IMFMediaType *pmt)
2390    {
2391            HRESULT hr = S_OK;
2392            UINT32 w, h;
2393    
2394            if (m_pInputType) m_pInputType->Release();
2395    
2396            hr = MFGetAttributeSize(pmt, MF_MT_FRAME_SIZE, &w, &h);
2397            m_create.width = w; m_create.height = h;
2398    
2399            if (SUCCEEDED(hr))
2400                    hr = MFGetAttributeRatio(pmt, MF_MT_FRAME_RATE, (UINT32*)&m_frameRate.Numerator, (UINT32*)&m_frameRate.Denominator);
2401    
2402            if (SUCCEEDED(hr)) { /* Store frame duration, derived from the frame rate */
2403                    hr = MFFrameRateToAverageTimePerFrame(m_frameRate.Numerator, m_frameRate.Denominator, &m_duration);
2404            }
2405    
2406            if (SUCCEEDED(hr)) {
2407                    m_pInputType = pmt;
2408                    m_pInputType->AddRef();
2409            }
2410    
2411            return hr;
2412    }
2413    
2414    HRESULT CXvidDecoder::OnSetOutputType(IMFMediaType *pmt, int bitdepth)
2415    {
2416            if (m_pOutputType) m_pOutputType->Release();
2417    
2418            m_pOutputType = pmt;
2419            m_pOutputType->AddRef();
2420            m_pOutputTypeBPP = bitdepth;
2421    
2422            return S_OK;
2423    }
2424    
2425    #endif /* XVID_USE_MFT */
Legend:
Removed from v.1437  
changed lines
  Added in v.2133
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4