[svn] / branches / release-1_3-branch / xvidcore / dshow / src / CXvidDecoder.cpp Repository:
ViewVC logotype

Annotation of /branches/release-1_3-branch/xvidcore/dshow/src/CXvidDecoder.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1949 - (view) (download)

1 : edgomez 1384 /*****************************************************************************
2 :     *
3 :     * XVID MPEG-4 VIDEO CODEC
4 : Isibaar 1896 * - Xvid Decoder part of the DShow Filter -
5 : edgomez 1384 *
6 : Isibaar 1896 * Copyright(C) 2002-2010 Peter Ross <pross@xvid.org>
7 :     * 2003-2010 Michael Militzer <michael@xvid.org>
8 : edgomez 1384 *
9 :     * 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
11 :     * the Free Software Foundation ; either version 2 of the License, or
12 :     * (at your option) any later version.
13 :     *
14 :     * This program is distributed in the hope that it will be useful,
15 :     * but WITHOUT ANY WARRANTY ; without even the implied warranty of
16 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 :     * GNU General Public License for more details.
18 :     *
19 :     * You should have received a copy of the GNU General Public License
20 :     * along with this program ; if not, write to the Free Software
21 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 :     *
23 : Isibaar 1949 * $Id: CXvidDecoder.cpp,v 1.25.2.3 2011-02-14 16:58:54 Isibaar Exp $
24 : edgomez 1384 *
25 :     ****************************************************************************/
26 :    
27 :     /*
28 :     this requires the directx sdk
29 :     place these paths at the top of the Tools|Options|Directories list
30 :    
31 :     headers:
32 : suxen_drol 1521 C:\DX90SDK\Include
33 :     C:\DX90SDK\Samples\C++\DirectShow\BaseClasses
34 : edgomez 1384
35 : suxen_drol 1521 C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Release
36 :     C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Debug
37 : edgomez 1384 */
38 :    
39 : Isibaar 1928 /*
40 : Isibaar 1916 #define XVID_USE_MFT
41 :     #define XVID_USE_TRAYICON
42 : Isibaar 1928 */
43 : edgomez 1384
44 :     #include <windows.h>
45 :    
46 :     #include <streams.h>
47 :     #include <initguid.h>
48 :     #include <olectl.h>
49 :     #if (1100 > _MSC_VER)
50 :     #include <olectlid.h>
51 :     #endif
52 :     #include <dvdmedia.h> // VIDEOINFOHEADER2
53 :    
54 : Isibaar 1898 #if defined(XVID_USE_MFT)
55 :     #define MFT_UNIQUE_METHOD_NAMES
56 :     #include <mftransform.h>
57 :     #include <mfapi.h>
58 :     #include <mferror.h>
59 :     #include <shlwapi.h>
60 :     #endif
61 :    
62 : Isibaar 1896 #include <shellapi.h>
63 : edgomez 1384
64 : Isibaar 1896 #include <xvid.h> // Xvid API
65 :    
66 :     #include "resource.h"
67 :    
68 : edgomez 1384 #include "IXvidDecoder.h"
69 :     #include "CXvidDecoder.h"
70 :     #include "CAbout.h"
71 :     #include "config.h"
72 :     #include "debug.h"
73 :    
74 :     static bool USE_IYUV;
75 :     static bool USE_YV12;
76 :     static bool USE_YUY2;
77 :     static bool USE_YVYU;
78 :     static bool USE_UYVY;
79 :     static bool USE_RGB32;
80 :     static bool USE_RGB24;
81 :     static bool USE_RG555;
82 :     static bool USE_RG565;
83 :    
84 :     const AMOVIESETUP_MEDIATYPE sudInputPinTypes[] =
85 :     {
86 :     { &MEDIATYPE_Video, &CLSID_XVID },
87 :     { &MEDIATYPE_Video, &CLSID_XVID_UC },
88 :     { &MEDIATYPE_Video, &CLSID_DIVX },
89 :     { &MEDIATYPE_Video, &CLSID_DIVX_UC },
90 :     { &MEDIATYPE_Video, &CLSID_DX50 },
91 :     { &MEDIATYPE_Video, &CLSID_DX50_UC },
92 : Isibaar 1949 { &MEDIATYPE_Video, &CLSID_3IVX },
93 :     { &MEDIATYPE_Video, &CLSID_3IVX_UC },
94 :     { &MEDIATYPE_Video, &CLSID_3IV0 },
95 :     { &MEDIATYPE_Video, &CLSID_3IV0_UC },
96 :     { &MEDIATYPE_Video, &CLSID_3IV1 },
97 :     { &MEDIATYPE_Video, &CLSID_3IV1_UC },
98 :     { &MEDIATYPE_Video, &CLSID_3IV2 },
99 :     { &MEDIATYPE_Video, &CLSID_3IV2_UC },
100 :     { &MEDIATYPE_Video, &CLSID_LMP4 },
101 :     { &MEDIATYPE_Video, &CLSID_LMP4_UC },
102 :     { &MEDIATYPE_Video, &CLSID_RMP4 },
103 :     { &MEDIATYPE_Video, &CLSID_RMP4_UC },
104 :     { &MEDIATYPE_Video, &CLSID_SMP4 },
105 :     { &MEDIATYPE_Video, &CLSID_SMP4_UC },
106 :     { &MEDIATYPE_Video, &CLSID_HDX4 },
107 :     { &MEDIATYPE_Video, &CLSID_HDX4_UC },
108 : edgomez 1384 { &MEDIATYPE_Video, &CLSID_MP4V },
109 : suxen_drol 1647 { &MEDIATYPE_Video, &CLSID_MP4V_UC },
110 : edgomez 1384 };
111 :    
112 :     const AMOVIESETUP_MEDIATYPE sudOutputPinTypes[] =
113 :     {
114 :     { &MEDIATYPE_Video, &MEDIASUBTYPE_NULL }
115 :     };
116 :    
117 :    
118 :     const AMOVIESETUP_PIN psudPins[] =
119 :     {
120 :     {
121 :     L"Input", // String pin name
122 :     FALSE, // Is it rendered
123 :     FALSE, // Is it an output
124 :     FALSE, // Allowed none
125 :     FALSE, // Allowed many
126 :     &CLSID_NULL, // Connects to filter
127 :     L"Output", // Connects to pin
128 :     sizeof(sudInputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE), // Number of types
129 :     &sudInputPinTypes[0] // The pin details
130 :     },
131 :     {
132 :     L"Output", // String pin name
133 :     FALSE, // Is it rendered
134 :     TRUE, // Is it an output
135 :     FALSE, // Allowed none
136 :     FALSE, // Allowed many
137 :     &CLSID_NULL, // Connects to filter
138 :     L"Input", // Connects to pin
139 :     sizeof(sudOutputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE), // Number of types
140 :     sudOutputPinTypes // The pin details
141 :     }
142 :     };
143 :    
144 :    
145 :     const AMOVIESETUP_FILTER sudXvidDecoder =
146 :     {
147 :     &CLSID_XVID, // Filter CLSID
148 :     XVID_NAME_L, // Filter name
149 : Isibaar 1946 MERIT_PREFERRED+2, // Its merit
150 : edgomez 1384 sizeof(psudPins) / sizeof(AMOVIESETUP_PIN), // Number of pins
151 :     psudPins // Pin details
152 :     };
153 :    
154 :    
155 :     // List of class IDs and creator functions for the class factory. This
156 :     // provides the link between the OLE entry point in the DLL and an object
157 :     // being created. The class factory will call the static CreateInstance
158 :    
159 :     CFactoryTemplate g_Templates[] =
160 :     {
161 :     {
162 :     XVID_NAME_L,
163 :     &CLSID_XVID,
164 :     CXvidDecoder::CreateInstance,
165 :     NULL,
166 :     &sudXvidDecoder
167 :     },
168 :     {
169 :     XVID_NAME_L L"About",
170 :     &CLSID_CABOUT,
171 :     CAbout::CreateInstance
172 :     }
173 :    
174 :     };
175 :    
176 :     /* note: g_cTemplates must be global; used by strmbase.lib(dllentry.cpp,dllsetup.cpp) */
177 :     int g_cTemplates = sizeof(g_Templates) / sizeof(CFactoryTemplate);
178 :    
179 : Isibaar 1896 #ifdef XVID_USE_TRAYICON
180 :     extern HINSTANCE g_xvid_hInst;
181 : edgomez 1384
182 : Isibaar 1896 static int GUI_Page = 0;
183 : Isibaar 1901 static int Tray_Icon = 0;
184 : Isibaar 1896 extern "C" void CALLBACK Configure(HWND hWndParent, HINSTANCE hInstParent, LPSTR lpCmdLine, int nCmdShow );
185 :    
186 :     LRESULT CALLBACK msg_proc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
187 :     {
188 :     switch ( uMsg )
189 :     {
190 :     case WM_ICONMESSAGE:
191 :     switch(lParam)
192 :     {
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 : Isibaar 1901
205 :     case WM_DESTROY:
206 :     NOTIFYICONDATA nid;
207 :     ZeroMemory(&nid,sizeof(NOTIFYICONDATA));
208 :    
209 :     nid.cbSize = NOTIFYICONDATA_V1_SIZE;
210 :     nid.hWnd = hwnd;
211 :     nid.uID = 1456;
212 : Isibaar 1896
213 : Isibaar 1901 Shell_NotifyIcon(NIM_DELETE, &nid);
214 :     Tray_Icon = 0;
215 : Isibaar 1896 default:
216 :     return DefWindowProc(hwnd, uMsg, wParam, lParam);
217 :     }
218 :    
219 :     return TRUE; /* ok */
220 :     }
221 :     #endif
222 :    
223 : edgomez 1384 STDAPI DllRegisterServer()
224 :     {
225 : Isibaar 1898 #if defined(XVID_USE_MFT)
226 :     int inputs_num = sizeof(sudInputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE);
227 :     int outputs_num = sizeof(sudOutputPinTypes) / sizeof(AMOVIESETUP_MEDIATYPE);
228 :     MFT_REGISTER_TYPE_INFO * mft_bs = new MFT_REGISTER_TYPE_INFO[inputs_num];
229 :     MFT_REGISTER_TYPE_INFO * mft_csp = new MFT_REGISTER_TYPE_INFO[outputs_num];
230 :    
231 :     {
232 :     int i;
233 :     for(i=0;i<inputs_num;i++) {
234 :     mft_bs[i].guidMajorType = *sudInputPinTypes[i].clsMajorType;
235 :     mft_bs[i].guidSubtype = *sudInputPinTypes[i].clsMinorType;
236 :     }
237 :     for(i=0;i<outputs_num;i++) {
238 :     mft_csp[i].guidMajorType = *sudOutputPinTypes[i].clsMajorType;
239 :     mft_csp[i].guidSubtype = *sudOutputPinTypes[i].clsMinorType; // MFT and AM GUIDs really the same?
240 :     }
241 :     }
242 :    
243 :     /* Register the MFT decoder */
244 :     MFTRegister(CLSID_XVID, // CLSID
245 :     MFT_CATEGORY_VIDEO_DECODER, // Category
246 :     const_cast<LPWSTR>(XVID_NAME_L), // Friendly name
247 :     0, // Flags
248 :     inputs_num, // Number of input types
249 :     mft_bs, // Input types
250 :     outputs_num, // Number of output types
251 :     mft_csp, // Output types
252 :     NULL // Attributes (optional)
253 :     );
254 :    
255 :     delete[] mft_bs;
256 :     delete[] mft_csp;
257 :     #endif /* XVID_USE_MFT */
258 :    
259 : edgomez 1384 return AMovieDllRegisterServer2( TRUE );
260 :     }
261 :    
262 :    
263 :     STDAPI DllUnregisterServer()
264 :     {
265 : Isibaar 1898 #if defined(XVID_USE_MFT)
266 :     MFTUnregister(CLSID_XVID);
267 :     #endif
268 : edgomez 1384 return AMovieDllRegisterServer2( FALSE );
269 :     }
270 :    
271 :    
272 :     /* create instance */
273 :    
274 :     CUnknown * WINAPI CXvidDecoder::CreateInstance(LPUNKNOWN punk, HRESULT *phr)
275 :     {
276 :     CXvidDecoder * pNewObject = new CXvidDecoder(punk, phr);
277 :     if (pNewObject == NULL)
278 :     {
279 :     *phr = E_OUTOFMEMORY;
280 :     }
281 :     return pNewObject;
282 :     }
283 :    
284 :    
285 :     /* query interfaces */
286 :    
287 :     STDMETHODIMP CXvidDecoder::NonDelegatingQueryInterface(REFIID riid, void **ppv)
288 :     {
289 :     CheckPointer(ppv, E_POINTER);
290 :    
291 :     if (riid == IID_IXvidDecoder)
292 :     {
293 :     return GetInterface((IXvidDecoder *) this, ppv);
294 :     }
295 :    
296 :     if (riid == IID_ISpecifyPropertyPages)
297 :     {
298 :     return GetInterface((ISpecifyPropertyPages *) this, ppv);
299 :     }
300 :    
301 : Isibaar 1898 #if defined(XVID_USE_MFT)
302 :     if (riid == IID_IMFTransform)
303 :     {
304 :     return GetInterface((IMFTransform *) this, ppv);
305 :     }
306 :     #endif
307 :    
308 : edgomez 1384 return CVideoTransformFilter::NonDelegatingQueryInterface(riid, ppv);
309 :     }
310 :    
311 :    
312 :    
313 :     /* constructor */
314 :    
315 :     #define XVID_DLL_NAME "xvidcore.dll"
316 :    
317 :     CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) :
318 : Isibaar 1605 CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID), m_hdll (NULL)
319 : edgomez 1384 {
320 :     DPRINTF("Constructor");
321 :    
322 : Isibaar 1605 xvid_decore_func = NULL; // Hmm, some strange errors appearing if I try to initialize...
323 :     xvid_global_func = NULL; // ...this in constructor's init-list. So, they assigned here.
324 :    
325 : Isibaar 1898 #if defined(XVID_USE_MFT)
326 :     InitializeCriticalSection(&m_mft_lock);
327 :     m_pInputType = NULL;
328 :     m_pOutputType = NULL;
329 :     m_rtFrame = 0;
330 :     m_duration = 0;
331 :     m_discont = 0;
332 :     m_frameRate.Denominator = 1;
333 :     m_frameRate.Numerator = 1;
334 :     #endif
335 :    
336 : Isibaar 1605 LoadRegistryInfo();
337 :    
338 :     *phr = OpenLib();
339 :     }
340 :    
341 :     HRESULT CXvidDecoder::OpenLib()
342 :     {
343 :     DPRINTF("OpenLib");
344 :    
345 :     if (m_hdll != NULL)
346 :     return E_UNEXPECTED; // Seems, that library already opened.
347 :    
348 : edgomez 1384 xvid_gbl_init_t init;
349 :     memset(&init, 0, sizeof(init));
350 :     init.version = XVID_VERSION;
351 : Isibaar 1916 init.cpu_flags = g_config.cpu;
352 : edgomez 1384
353 : Isibaar 1912 xvid_gbl_info_t info;
354 :     memset(&info, 0, sizeof(info));
355 :     info.version = XVID_VERSION;
356 :    
357 : edgomez 1384 m_hdll = LoadLibrary(XVID_DLL_NAME);
358 :     if (m_hdll == NULL) {
359 :     DPRINTF("dll load failed");
360 : syskin 1498 MessageBox(0, XVID_DLL_NAME " not found","Error", MB_TOPMOST);
361 : Isibaar 1605 return E_FAIL;
362 : edgomez 1384 }
363 :    
364 :     xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_global");
365 :     if (xvid_global_func == NULL) {
366 : Isibaar 1605 FreeLibrary(m_hdll);
367 :     m_hdll = NULL;
368 : syskin 1498 MessageBox(0, "xvid_global() not found", "Error", MB_TOPMOST);
369 : Isibaar 1605 return E_FAIL;
370 : edgomez 1384 }
371 :    
372 :     xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(m_hdll, "xvid_decore");
373 :     if (xvid_decore_func == NULL) {
374 : Isibaar 1605 xvid_global_func = NULL;
375 :     FreeLibrary(m_hdll);
376 :     m_hdll = NULL;
377 : syskin 1498 MessageBox(0, "xvid_decore() not found", "Error", MB_TOPMOST);
378 : Isibaar 1605 return E_FAIL;
379 : edgomez 1384 }
380 :    
381 :     if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0)
382 :     {
383 : Isibaar 1605 xvid_global_func = NULL;
384 :     xvid_decore_func = NULL;
385 :     FreeLibrary(m_hdll);
386 :     m_hdll = NULL;
387 : syskin 1498 MessageBox(0, "xvid_global() failed", "Error", MB_TOPMOST);
388 : Isibaar 1605 return E_FAIL;
389 : edgomez 1384 }
390 :    
391 : Isibaar 1912 if (xvid_global_func(0, XVID_GBL_INFO, &info, NULL) < 0)
392 :     {
393 :     xvid_global_func = NULL;
394 :     xvid_decore_func = NULL;
395 :     FreeLibrary(m_hdll);
396 :     m_hdll = NULL;
397 :     MessageBox(0, "xvid_global() failed", "Error", MB_TOPMOST);
398 :     return E_FAIL;
399 :     }
400 :    
401 : edgomez 1384 memset(&m_create, 0, sizeof(m_create));
402 :     m_create.version = XVID_VERSION;
403 :     m_create.handle = NULL;
404 : Isibaar 1916 /* Decoder threads */
405 :     if (g_config.cpu & XVID_CPU_FORCE) {
406 :     m_create.num_threads = g_config.num_threads;
407 :     }
408 :     else {
409 :     m_create.num_threads = info.num_threads; /* Autodetect */
410 :     g_config.num_threads = info.num_threads;
411 :     }
412 : edgomez 1384
413 :     memset(&m_frame, 0, sizeof(m_frame));
414 :     m_frame.version = XVID_VERSION;
415 :    
416 :     USE_IYUV = false;
417 :     USE_YV12 = false;
418 :     USE_YUY2 = false;
419 :     USE_YVYU = false;
420 :     USE_UYVY = false;
421 :     USE_RGB32 = false;
422 :     USE_RGB24 = false;
423 :     USE_RG555 = false;
424 :     USE_RG565 = false;
425 :    
426 :     switch ( g_config.nForceColorspace )
427 :     {
428 :     case FORCE_NONE:
429 :     USE_IYUV = true;
430 :     USE_YV12 = true;
431 :     USE_YUY2 = true;
432 :     USE_YVYU = true;
433 :     USE_UYVY = true;
434 :     USE_RGB32 = true;
435 :     USE_RGB24 = true;
436 :     USE_RG555 = true;
437 :     USE_RG565 = true;
438 :     break;
439 :     case FORCE_YV12:
440 :     USE_IYUV = true;
441 :     USE_YV12 = true;
442 :     break;
443 :     case FORCE_YUY2:
444 :     USE_YUY2 = true;
445 :     break;
446 :     case FORCE_RGB24:
447 :     USE_RGB24 = true;
448 :     break;
449 :     case FORCE_RGB32:
450 :     USE_RGB32 = true;
451 :     break;
452 :     }
453 : syskin 1488
454 :     switch (g_config.aspect_ratio)
455 :     {
456 : syskin 1498 case 0:
457 :     case 1:
458 : syskin 1488 break;
459 : syskin 1498 case 2:
460 : syskin 1488 ar_x = 4;
461 :     ar_y = 3;
462 :     break;
463 : syskin 1498 case 3:
464 : syskin 1488 ar_x = 16;
465 :     ar_y = 9;
466 :     break;
467 : syskin 1498 case 4:
468 : syskin 1488 ar_x = 47;
469 :     ar_y = 20;
470 :     break;
471 :     }
472 : Isibaar 1605
473 :     return S_OK;
474 : edgomez 1384 }
475 :    
476 : syskin 1428 void CXvidDecoder::CloseLib()
477 : edgomez 1384 {
478 : Isibaar 1605 DPRINTF("CloseLib");
479 : edgomez 1384
480 : Isibaar 1605 if ((m_create.handle != NULL) && (xvid_decore_func != NULL))
481 :     {
482 : edgomez 1384 xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0);
483 :     m_create.handle = NULL;
484 :     }
485 :    
486 : syskin 1428 if (m_hdll != NULL) {
487 : edgomez 1384 FreeLibrary(m_hdll);
488 :     m_hdll = NULL;
489 :     }
490 : Isibaar 1605 xvid_decore_func = NULL;
491 :     xvid_global_func = NULL;
492 : edgomez 1384 }
493 :    
494 : syskin 1428 /* destructor */
495 : edgomez 1384
496 : syskin 1428 CXvidDecoder::~CXvidDecoder()
497 :     {
498 : Isibaar 1605 DPRINTF("Destructor");
499 : Isibaar 1896
500 :     #ifdef XVID_USE_TRAYICON
501 : Isibaar 1901 if (Tray_Icon) { /* Destroy tray icon */
502 : Isibaar 1896 NOTIFYICONDATA nid;
503 : Isibaar 1901 ZeroMemory(&nid,sizeof(NOTIFYICONDATA));
504 :    
505 :     nid.cbSize = NOTIFYICONDATA_V1_SIZE;
506 : Isibaar 1896 nid.hWnd = MSG_hwnd;
507 : Isibaar 1901 nid.uID = 1456;
508 : Isibaar 1896
509 : Isibaar 1901 Shell_NotifyIcon(NIM_DELETE, &nid);
510 :     Tray_Icon = 0;
511 : Isibaar 1896 }
512 :     #endif
513 : Isibaar 1898
514 : Isibaar 1901 /* Close xvidcore library */
515 :     CloseLib();
516 :    
517 : Isibaar 1898 #if defined(XVID_USE_MFT)
518 :     DeleteCriticalSection(&m_mft_lock);
519 :     #endif
520 : syskin 1428 }
521 : edgomez 1384
522 : syskin 1428
523 :    
524 : edgomez 1384 /* check input type */
525 :    
526 :     HRESULT CXvidDecoder::CheckInputType(const CMediaType * mtIn)
527 :     {
528 :     DPRINTF("CheckInputType");
529 :     BITMAPINFOHEADER * hdr;
530 : syskin 1498
531 :     ar_x = ar_y = 0;
532 : edgomez 1384
533 :     if (*mtIn->Type() != MEDIATYPE_Video)
534 :     {
535 :     DPRINTF("Error: Unknown Type");
536 : syskin 1428 CloseLib();
537 : edgomez 1384 return VFW_E_TYPE_NOT_ACCEPTED;
538 :     }
539 :    
540 : Isibaar 1605 if (m_hdll == NULL)
541 :     {
542 :     HRESULT hr = OpenLib();
543 :    
544 :     if (FAILED(hr) || (m_hdll == NULL)) // Paranoid checks.
545 :     return VFW_E_TYPE_NOT_ACCEPTED;
546 :     }
547 :    
548 : edgomez 1384 if (*mtIn->FormatType() == FORMAT_VideoInfo)
549 :     {
550 :     VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format();
551 :     hdr = &vih->bmiHeader;
552 :     }
553 :     else if (*mtIn->FormatType() == FORMAT_VideoInfo2)
554 :     {
555 :     VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format();
556 :     hdr = &vih2->bmiHeader;
557 : syskin 1498 if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
558 : syskin 1488 ar_x = vih2->dwPictAspectRatioX;
559 :     ar_y = vih2->dwPictAspectRatioY;
560 :     }
561 : edgomez 1384 DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
562 :     }
563 : suxen_drol 1647 else if (*mtIn->FormatType() == FORMAT_MPEG2Video) {
564 :     MPEG2VIDEOINFO * mpgvi = (MPEG2VIDEOINFO*)mtIn->Format();
565 :     VIDEOINFOHEADER2 * vih2 = &mpgvi->hdr;
566 :     hdr = &vih2->bmiHeader;
567 :     if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
568 :     ar_x = vih2->dwPictAspectRatioX;
569 :     ar_y = vih2->dwPictAspectRatioY;
570 :     }
571 :     DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
572 :    
573 :     /* haali media splitter reports VOL information in the format header */
574 :    
575 :     if (mpgvi->cbSequenceHeader>0) {
576 :    
577 :     xvid_dec_stats_t stats;
578 :     memset(&stats, 0, sizeof(stats));
579 :     stats.version = XVID_VERSION;
580 :    
581 :     if (m_create.handle == NULL) {
582 :     if (xvid_decore_func == NULL)
583 :     return E_FAIL;
584 :     if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) {
585 :     DPRINTF("*** XVID_DEC_CREATE error");
586 : Isibaar 1868 return E_FAIL;
587 : suxen_drol 1647 }
588 :     }
589 :    
590 :     m_frame.general = 0;
591 :     m_frame.bitstream = (void*)mpgvi->dwSequenceHeader;
592 :     m_frame.length = mpgvi->cbSequenceHeader;
593 :     m_frame.output.csp = XVID_CSP_NULL;
594 :    
595 : Isibaar 1868 int ret = 0;
596 :     if ((ret=xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats)) >= 0) {
597 : suxen_drol 1647 /* honour video dimensions reported in VOL header */
598 :     if (stats.type == XVID_TYPE_VOL) {
599 :     hdr->biWidth = stats.data.vol.width;
600 :     hdr->biHeight = stats.data.vol.height;
601 : Isibaar 1898 }
602 :     }
603 : Isibaar 1868 if (ret == XVID_ERR_MEMORY) return E_FAIL;
604 : Isibaar 1898 }
605 : suxen_drol 1647 }
606 : Isibaar 1898 else
607 :     {
608 : edgomez 1384 DPRINTF("Error: Unknown FormatType");
609 : syskin 1428 CloseLib();
610 : edgomez 1384 return VFW_E_TYPE_NOT_ACCEPTED;
611 : Isibaar 1898 }
612 :     if (hdr->biHeight < 0)
613 :     {
614 :     DPRINTF("colorspace: inverted input format not supported");
615 :     }
616 :    
617 :     m_create.width = hdr->biWidth;
618 :     m_create.height = hdr->biHeight;
619 :    
620 :     switch(hdr->biCompression)
621 :     {
622 : Isibaar 1949 case FOURCC_mp4v :
623 :     case FOURCC_MP4V :
624 :     case FOURCC_lmp4 :
625 :     case FOURCC_LMP4 :
626 :     case FOURCC_rmp4 :
627 :     case FOURCC_RMP4 :
628 :     case FOURCC_smp4 :
629 :     case FOURCC_SMP4 :
630 :     case FOURCC_hdx4 :
631 :     case FOURCC_HDX4 :
632 : syskin 1428 if (!(g_config.supported_4cc & SUPPORT_MP4V)) {
633 :     CloseLib();
634 :     return VFW_E_TYPE_NOT_ACCEPTED;
635 :     }
636 : edgomez 1384 break;
637 : Isibaar 1949 case FOURCC_divx :
638 : edgomez 1384 case FOURCC_DIVX :
639 : Isibaar 1949 case FOURCC_dx50 :
640 :     case FOURCC_DX50 :
641 : syskin 1428 if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
642 :     CloseLib();
643 :     return VFW_E_TYPE_NOT_ACCEPTED;
644 :     }
645 : edgomez 1384 break;
646 : Isibaar 1949 case FOURCC_3ivx :
647 :     case FOURCC_3IVX :
648 :     case FOURCC_3iv0 :
649 :     case FOURCC_3IV0 :
650 :     case FOURCC_3iv1 :
651 :     case FOURCC_3IV1 :
652 :     case FOURCC_3iv2 :
653 :     case FOURCC_3IV2 :
654 :     if (!(g_config.supported_4cc & SUPPORT_3IVX)) {
655 : syskin 1428 CloseLib();
656 :     return VFW_E_TYPE_NOT_ACCEPTED;
657 :     }
658 : Isibaar 1949 case FOURCC_xvid :
659 : edgomez 1384 case FOURCC_XVID :
660 :     break;
661 :    
662 :    
663 :     default :
664 :     DPRINTF("Unknown fourcc: 0x%08x (%c%c%c%c)",
665 :     hdr->biCompression,
666 :     (hdr->biCompression)&0xff,
667 :     (hdr->biCompression>>8)&0xff,
668 :     (hdr->biCompression>>16)&0xff,
669 :     (hdr->biCompression>>24)&0xff);
670 : syskin 1428 CloseLib();
671 : edgomez 1384 return VFW_E_TYPE_NOT_ACCEPTED;
672 :     }
673 : Isibaar 1890
674 :     m_create.fourcc = hdr->biCompression;
675 :    
676 : edgomez 1384 return S_OK;
677 :     }
678 :    
679 :    
680 :     /* get list of supported output colorspaces */
681 :    
682 :    
683 :     HRESULT CXvidDecoder::GetMediaType(int iPosition, CMediaType *mtOut)
684 :     {
685 :     BITMAPINFOHEADER * bmih;
686 :     DPRINTF("GetMediaType");
687 :    
688 :     if (m_pInput->IsConnected() == FALSE)
689 :     {
690 :     return E_UNEXPECTED;
691 :     }
692 :    
693 :     if (!g_config.videoinfo_compat) {
694 :     VIDEOINFOHEADER2 * vih = (VIDEOINFOHEADER2 *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER2));
695 :     if (vih == NULL) return E_OUTOFMEMORY;
696 :    
697 :     ZeroMemory(vih, sizeof (VIDEOINFOHEADER2));
698 :     bmih = &(vih->bmiHeader);
699 :     mtOut->SetFormatType(&FORMAT_VideoInfo2);
700 :    
701 :     if (ar_x != 0 && ar_y != 0) {
702 :     vih->dwPictAspectRatioX = ar_x;
703 :     vih->dwPictAspectRatioY = ar_y;
704 : syskin 1498 forced_ar = true;
705 : edgomez 1384 } else { // just to be safe
706 :     vih->dwPictAspectRatioX = m_create.width;
707 :     vih->dwPictAspectRatioY = abs(m_create.height);
708 : syskin 1498 forced_ar = false;
709 :     }
710 : edgomez 1384 } else {
711 :    
712 :     VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER));
713 :     if (vih == NULL) return E_OUTOFMEMORY;
714 :    
715 :     ZeroMemory(vih, sizeof (VIDEOINFOHEADER));
716 :     bmih = &(vih->bmiHeader);
717 :     mtOut->SetFormatType(&FORMAT_VideoInfo);
718 :     }
719 :    
720 :     bmih->biSize = sizeof(BITMAPINFOHEADER);
721 :     bmih->biWidth = m_create.width;
722 :     bmih->biHeight = m_create.height;
723 :     bmih->biPlanes = 1;
724 :    
725 :     if (iPosition < 0) return E_INVALIDARG;
726 :    
727 :     switch(iPosition)
728 :     {
729 :    
730 :     case 0:
731 :     if ( USE_YUY2 )
732 :     {
733 :     bmih->biCompression = MEDIASUBTYPE_YUY2.Data1;
734 :     bmih->biBitCount = 16;
735 :     mtOut->SetSubtype(&MEDIASUBTYPE_YUY2);
736 :     break;
737 :     }
738 :     case 1 :
739 :     if ( USE_YVYU )
740 :     {
741 :     bmih->biCompression = MEDIASUBTYPE_YVYU.Data1;
742 :     bmih->biBitCount = 16;
743 :     mtOut->SetSubtype(&MEDIASUBTYPE_YVYU);
744 :     break;
745 :     }
746 :     case 2 :
747 :     if ( USE_UYVY )
748 :     {
749 :     bmih->biCompression = MEDIASUBTYPE_UYVY.Data1;
750 :     bmih->biBitCount = 16;
751 :     mtOut->SetSubtype(&MEDIASUBTYPE_UYVY);
752 :     break;
753 :     }
754 :     case 3 :
755 :     if ( USE_IYUV )
756 :     {
757 :     bmih->biCompression = CLSID_MEDIASUBTYPE_IYUV.Data1;
758 :     bmih->biBitCount = 12;
759 :     mtOut->SetSubtype(&CLSID_MEDIASUBTYPE_IYUV);
760 :     break;
761 :     }
762 :     case 4 :
763 :     if ( USE_YV12 )
764 :     {
765 :     bmih->biCompression = MEDIASUBTYPE_YV12.Data1;
766 :     bmih->biBitCount = 12;
767 :     mtOut->SetSubtype(&MEDIASUBTYPE_YV12);
768 :     break;
769 :     }
770 :     case 5 :
771 :     if ( USE_RGB32 )
772 :     {
773 :     bmih->biCompression = BI_RGB;
774 :     bmih->biBitCount = 32;
775 :     mtOut->SetSubtype(&MEDIASUBTYPE_RGB32);
776 :     break;
777 :     }
778 :     case 6 :
779 :     if ( USE_RGB24 )
780 :     {
781 :     bmih->biCompression = BI_RGB;
782 :     bmih->biBitCount = 24;
783 :     mtOut->SetSubtype(&MEDIASUBTYPE_RGB24);
784 :     break;
785 :     }
786 :     case 7 :
787 :     if ( USE_RG555 )
788 :     {
789 :     bmih->biCompression = BI_RGB;
790 :     bmih->biBitCount = 16;
791 :     mtOut->SetSubtype(&MEDIASUBTYPE_RGB555);
792 :     break;
793 :     }
794 :     case 8 :
795 :     if ( USE_RG565 )
796 :     {
797 :     bmih->biCompression = BI_RGB;
798 :     bmih->biBitCount = 16;
799 :     mtOut->SetSubtype(&MEDIASUBTYPE_RGB565);
800 :     break;
801 :     }
802 :     default :
803 :     return VFW_S_NO_MORE_ITEMS;
804 :     }
805 :    
806 :     bmih->biSizeImage = GetBitmapSize(bmih);
807 :    
808 :     mtOut->SetType(&MEDIATYPE_Video);
809 :     mtOut->SetTemporalCompression(FALSE);
810 :     mtOut->SetSampleSize(bmih->biSizeImage);
811 :    
812 :     return S_OK;
813 :     }
814 :    
815 :    
816 :     /* (internal function) change colorspace */
817 : suxen_drol 1558 #define CALC_BI_STRIDE(width,bitcount) ((((width * bitcount) + 31) & ~31) >> 3)
818 : edgomez 1384
819 :     HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format)
820 :     {
821 : suxen_drol 1558 DWORD biWidth;
822 :    
823 : edgomez 1384 if (formattype == FORMAT_VideoInfo)
824 :     {
825 :     VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;
826 : suxen_drol 1558 biWidth = vih->bmiHeader.biWidth;
827 : Isibaar 1898 out_stride = CALC_BI_STRIDE(vih->bmiHeader.biWidth, vih->bmiHeader.biBitCount);
828 : edgomez 1384 rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
829 :     }
830 :     else if (formattype == FORMAT_VideoInfo2)
831 :     {
832 :     VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;
833 : suxen_drol 1558 biWidth = vih2->bmiHeader.biWidth;
834 : Isibaar 1898 out_stride = CALC_BI_STRIDE(vih2->bmiHeader.biWidth, vih2->bmiHeader.biBitCount);
835 : edgomez 1384 rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
836 :     }
837 :     else
838 :     {
839 :     return S_FALSE;
840 :     }
841 :    
842 :     if (subtype == CLSID_MEDIASUBTYPE_IYUV)
843 :     {
844 :     DPRINTF("IYUV");
845 :     rgb_flip = 0;
846 :     m_frame.output.csp = XVID_CSP_I420;
847 : Isibaar 1898 out_stride = CALC_BI_STRIDE(biWidth, 8); /* planar format fix */
848 : edgomez 1384 }
849 :     else if (subtype == MEDIASUBTYPE_YV12)
850 :     {
851 :     DPRINTF("YV12");
852 :     rgb_flip = 0;
853 :     m_frame.output.csp = XVID_CSP_YV12;
854 : Isibaar 1898 out_stride = CALC_BI_STRIDE(biWidth, 8); /* planar format fix */
855 : edgomez 1384 }
856 :     else if (subtype == MEDIASUBTYPE_YUY2)
857 :     {
858 :     DPRINTF("YUY2");
859 :     rgb_flip = 0;
860 :     m_frame.output.csp = XVID_CSP_YUY2;
861 :     }
862 :     else if (subtype == MEDIASUBTYPE_YVYU)
863 :     {
864 :     DPRINTF("YVYU");
865 :     rgb_flip = 0;
866 :     m_frame.output.csp = XVID_CSP_YVYU;
867 :     }
868 :     else if (subtype == MEDIASUBTYPE_UYVY)
869 :     {
870 :     DPRINTF("UYVY");
871 :     rgb_flip = 0;
872 :     m_frame.output.csp = XVID_CSP_UYVY;
873 :     }
874 :     else if (subtype == MEDIASUBTYPE_RGB32)
875 :     {
876 :     DPRINTF("RGB32");
877 :     m_frame.output.csp = rgb_flip | XVID_CSP_BGRA;
878 :     }
879 :     else if (subtype == MEDIASUBTYPE_RGB24)
880 :     {
881 :     DPRINTF("RGB24");
882 :     m_frame.output.csp = rgb_flip | XVID_CSP_BGR;
883 :     }
884 :     else if (subtype == MEDIASUBTYPE_RGB555)
885 :     {
886 :     DPRINTF("RGB555");
887 :     m_frame.output.csp = rgb_flip | XVID_CSP_RGB555;
888 :     }
889 :     else if (subtype == MEDIASUBTYPE_RGB565)
890 :     {
891 :     DPRINTF("RGB565");
892 :     m_frame.output.csp = rgb_flip | XVID_CSP_RGB565;
893 :     }
894 :     else if (subtype == GUID_NULL)
895 :     {
896 :     m_frame.output.csp = XVID_CSP_NULL;
897 :     }
898 :     else
899 :     {
900 :     return S_FALSE;
901 :     }
902 :    
903 :     return S_OK;
904 :     }
905 :    
906 :    
907 :     /* set output colorspace */
908 :    
909 :     HRESULT CXvidDecoder::SetMediaType(PIN_DIRECTION direction, const CMediaType *pmt)
910 :     {
911 :     DPRINTF("SetMediaType");
912 :    
913 :     if (direction == PINDIR_OUTPUT)
914 :     {
915 :     return ChangeColorspace(*pmt->Subtype(), *pmt->FormatType(), pmt->Format());
916 :     }
917 :    
918 :     return S_OK;
919 :     }
920 :    
921 :    
922 :     /* check input<->output compatiblity */
923 :    
924 :     HRESULT CXvidDecoder::CheckTransform(const CMediaType *mtIn, const CMediaType *mtOut)
925 :     {
926 :     DPRINTF("CheckTransform");
927 : Isibaar 1896
928 : edgomez 1384 return S_OK;
929 :     }
930 :    
931 : Isibaar 1896 /* input/output pin connection complete */
932 : edgomez 1384
933 : Isibaar 1896 HRESULT CXvidDecoder::CompleteConnect(PIN_DIRECTION direction, IPin *pReceivePin)
934 :     {
935 :     DPRINTF("CompleteConnect");
936 :    
937 :     #ifdef XVID_USE_TRAYICON
938 : Isibaar 1901 if ((direction == PINDIR_OUTPUT) && (Tray_Icon == 0))
939 : Isibaar 1896 {
940 :     WNDCLASSEX wc;
941 :    
942 :     wc.cbSize = sizeof(WNDCLASSEX);
943 :     wc.lpfnWndProc = msg_proc;
944 :     wc.style = CS_HREDRAW | CS_VREDRAW;
945 :     wc.cbWndExtra = 0;
946 :     wc.cbClsExtra = 0;
947 :     wc.hInstance = (HINSTANCE) g_xvid_hInst;
948 :     wc.hbrBackground = (HBRUSH) GetStockObject(NULL_BRUSH);
949 :     wc.lpszMenuName = NULL;
950 :     wc.lpszClassName = "XVID_MSG_WINDOW";
951 :     wc.hIcon = NULL;
952 :     wc.hIconSm = NULL;
953 :     wc.hCursor = NULL;
954 :     RegisterClassEx(&wc);
955 :    
956 :     MSG_hwnd = CreateWindowEx(0, "XVID_MSG_WINDOW", NULL, 0, CW_USEDEFAULT,
957 :     CW_USEDEFAULT, 0, 0, HWND_MESSAGE, NULL, (HINSTANCE) g_xvid_hInst, NULL);
958 :    
959 :     /* display the tray icon */
960 :     NOTIFYICONDATA nid;
961 : Isibaar 1901 ZeroMemory(&nid,sizeof(NOTIFYICONDATA));
962 :    
963 :     nid.cbSize = NOTIFYICONDATA_V1_SIZE;
964 : Isibaar 1896 nid.hWnd = MSG_hwnd;
965 : Isibaar 1901 nid.uID = 1456;
966 : Isibaar 1896 nid.uCallbackMessage = WM_ICONMESSAGE;
967 :     nid.hIcon = LoadIcon(g_xvid_hInst, MAKEINTRESOURCE(IDI_ICON));
968 :     strcpy_s(nid.szTip, 19, "Xvid Video Decoder");
969 :     nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
970 :    
971 :     Shell_NotifyIcon(NIM_ADD, &nid);
972 : Isibaar 1901
973 :     DestroyIcon(nid.hIcon);
974 :     Tray_Icon = 1;
975 : Isibaar 1896 }
976 :     #endif
977 :    
978 :     return S_OK;
979 :     }
980 :    
981 :     /* input/output pin disconnected */
982 :     HRESULT CXvidDecoder::BreakConnect(PIN_DIRECTION direction)
983 :     {
984 :     DPRINTF("BreakConnect");
985 :    
986 :     return S_OK;
987 :     }
988 :    
989 : edgomez 1384 /* alloc output buffer */
990 :    
991 :     HRESULT CXvidDecoder::DecideBufferSize(IMemAllocator *pAlloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
992 :     {
993 :     DPRINTF("DecideBufferSize");
994 :     HRESULT result;
995 :     ALLOCATOR_PROPERTIES ppropActual;
996 :    
997 :     if (m_pInput->IsConnected() == FALSE)
998 :     {
999 :     return E_UNEXPECTED;
1000 :     }
1001 :    
1002 :     ppropInputRequest->cBuffers = 1;
1003 :     ppropInputRequest->cbBuffer = m_create.width * m_create.height * 4;
1004 :     // cbAlign causes problems with the resize filter */
1005 :     // ppropInputRequest->cbAlign = 16;
1006 :     ppropInputRequest->cbPrefix = 0;
1007 :    
1008 :     result = pAlloc->SetProperties(ppropInputRequest, &ppropActual);
1009 :     if (result != S_OK)
1010 :     {
1011 :     return result;
1012 :     }
1013 :    
1014 :     if (ppropActual.cbBuffer < ppropInputRequest->cbBuffer)
1015 :     {
1016 :     return E_FAIL;
1017 :     }
1018 :    
1019 :     return S_OK;
1020 :     }
1021 :    
1022 :    
1023 :     /* decode frame */
1024 :    
1025 :     HRESULT CXvidDecoder::Transform(IMediaSample *pIn, IMediaSample *pOut)
1026 :     {
1027 :     DPRINTF("Transform");
1028 :     xvid_dec_stats_t stats;
1029 :     int length;
1030 :    
1031 :     memset(&stats, 0, sizeof(stats));
1032 :     stats.version = XVID_VERSION;
1033 :    
1034 :     if (m_create.handle == NULL)
1035 :     {
1036 : Isibaar 1605 if (xvid_decore_func == NULL)
1037 :     return E_FAIL;
1038 :    
1039 : edgomez 1384 if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)
1040 :     {
1041 :     DPRINTF("*** XVID_DEC_CREATE error");
1042 : Isibaar 1868 return E_FAIL;
1043 : edgomez 1384 }
1044 :     }
1045 :    
1046 :     AM_MEDIA_TYPE * mtOut;
1047 :     pOut->GetMediaType(&mtOut);
1048 :     if (mtOut != NULL)
1049 :     {
1050 :     HRESULT result;
1051 :    
1052 :     result = ChangeColorspace(mtOut->subtype, mtOut->formattype, mtOut->pbFormat);
1053 :     DeleteMediaType(mtOut);
1054 :    
1055 :     if (result != S_OK)
1056 :     {
1057 :     DPRINTF("*** ChangeColorspace error");
1058 :     return result;
1059 :     }
1060 :     }
1061 :    
1062 :     m_frame.length = pIn->GetActualDataLength();
1063 :     if (pIn->GetPointer((BYTE**)&m_frame.bitstream) != S_OK)
1064 :     {
1065 :     return S_FALSE;
1066 :     }
1067 :    
1068 :     if (pOut->GetPointer((BYTE**)&m_frame.output.plane[0]) != S_OK)
1069 :     {
1070 :     return S_FALSE;
1071 :     }
1072 :    
1073 :     m_frame.general = XVID_LOWDELAY;
1074 :    
1075 :     if (pIn->IsDiscontinuity() == S_OK)
1076 : syskin 1427 m_frame.general |= XVID_DISCONTINUITY;
1077 : edgomez 1384
1078 :     if (g_config.nDeblock_Y)
1079 :     m_frame.general |= XVID_DEBLOCKY;
1080 :    
1081 :     if (g_config.nDeblock_UV)
1082 :     m_frame.general |= XVID_DEBLOCKUV;
1083 : syskin 1437
1084 :     if (g_config.nDering_Y)
1085 :     m_frame.general |= XVID_DERINGY;
1086 :    
1087 :     if (g_config.nDering_UV)
1088 :     m_frame.general |= XVID_DERINGUV;
1089 :    
1090 : edgomez 1384 if (g_config.nFilmEffect)
1091 :     m_frame.general |= XVID_FILMEFFECT;
1092 :    
1093 : suxen_drol 1397 m_frame.brightness = g_config.nBrightness;
1094 :    
1095 : edgomez 1384 m_frame.output.csp &= ~XVID_CSP_VFLIP;
1096 :     m_frame.output.csp |= rgb_flip^(g_config.nFlipVideo ? XVID_CSP_VFLIP : 0);
1097 : Isibaar 1898 m_frame.output.stride[0] = out_stride;
1098 : edgomez 1384
1099 : Isibaar 1605 // Paranoid check.
1100 :     if (xvid_decore_func == NULL)
1101 :     return E_FAIL;
1102 : syskin 1498
1103 :    
1104 : edgomez 1384 repeat :
1105 :    
1106 :     if (pIn->IsPreroll() != S_OK)
1107 :     {
1108 :     length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1109 : Isibaar 1868
1110 :     if (length == XVID_ERR_MEMORY)
1111 :     return E_FAIL;
1112 :     else if (length < 0)
1113 : edgomez 1384 {
1114 :     DPRINTF("*** XVID_DEC_DECODE");
1115 :     return S_FALSE;
1116 : syskin 1498 } else
1117 :     if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1 && forced_ar == false) {
1118 : suxen_drol 1633
1119 :     if (stats.type != XVID_TYPE_NOTHING) { /* dont attempt to set vmr aspect ratio if no frame was returned by decoder */
1120 : syskin 1498 // inspired by minolta! works for VMR 7 + 9
1121 :     IMediaSample2 *pOut2 = NULL;
1122 :     AM_SAMPLE2_PROPERTIES outProp2;
1123 :     if (SUCCEEDED(pOut->QueryInterface(IID_IMediaSample2, (void **)&pOut2)) &&
1124 :     SUCCEEDED(pOut2->GetProperties(FIELD_OFFSET(AM_SAMPLE2_PROPERTIES, tStart), (PBYTE)&outProp2)))
1125 :     {
1126 :     CMediaType mtOut2 = m_pOutput->CurrentMediaType();
1127 :     VIDEOINFOHEADER2* vihOut2 = (VIDEOINFOHEADER2*)mtOut2.Format();
1128 : suxen_drol 1502
1129 :     if (*mtOut2.FormatType() == FORMAT_VideoInfo2 &&
1130 :     vihOut2->dwPictAspectRatioX != ar_x && vihOut2->dwPictAspectRatioY != ar_y)
1131 : syskin 1498 {
1132 :     vihOut2->dwPictAspectRatioX = ar_x;
1133 :     vihOut2->dwPictAspectRatioY = ar_y;
1134 :     pOut2->SetMediaType(&mtOut2);
1135 :     m_pOutput->SetMediaType(&mtOut2);
1136 :     }
1137 :     pOut2->Release();
1138 :     }
1139 : suxen_drol 1633 }
1140 : edgomez 1384 }
1141 :     }
1142 :     else
1143 :     { /* Preroll frame - won't be displayed */
1144 :     int tmp = m_frame.output.csp;
1145 :     int tmp_gen = m_frame.general;
1146 :    
1147 :     m_frame.output.csp = XVID_CSP_NULL;
1148 :    
1149 :     /* Disable postprocessing to speed-up seeking */
1150 :     m_frame.general &= ~XVID_DEBLOCKY;
1151 :     m_frame.general &= ~XVID_DEBLOCKUV;
1152 : Isibaar 1605 /*m_frame.general &= ~XVID_DERING;*/
1153 : edgomez 1384 m_frame.general &= ~XVID_FILMEFFECT;
1154 :    
1155 :     length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1156 : Isibaar 1868 if (length == XVID_ERR_MEMORY)
1157 :     return E_FAIL;
1158 :     else if (length < 0)
1159 : edgomez 1384 {
1160 :     DPRINTF("*** XVID_DEC_DECODE");
1161 :     return S_FALSE;
1162 :     }
1163 :    
1164 :     m_frame.output.csp = tmp;
1165 :     m_frame.general = tmp_gen;
1166 :     }
1167 :    
1168 :     if (stats.type == XVID_TYPE_NOTHING && length > 0) {
1169 : edgomez 1432 DPRINTF(" B-Frame decoder lag");
1170 : edgomez 1384 return S_FALSE;
1171 :     }
1172 :    
1173 :    
1174 :     if (stats.type == XVID_TYPE_VOL)
1175 :     {
1176 :     if (stats.data.vol.width != m_create.width ||
1177 :     stats.data.vol.height != m_create.height)
1178 :     {
1179 :     DPRINTF("TODO: auto-resize");
1180 :     return S_FALSE;
1181 :     }
1182 : syskin 1498
1183 : edgomez 1384 pOut->SetSyncPoint(TRUE);
1184 :    
1185 : syskin 1498 if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) { /* auto */
1186 :     int par_x, par_y;
1187 :     if (stats.data.vol.par == XVID_PAR_EXT) {
1188 :     par_x = stats.data.vol.par_width;
1189 :     par_y = stats.data.vol.par_height;
1190 :     } else {
1191 : syskin 1529 par_x = PARS[stats.data.vol.par-1][0];
1192 :     par_y = PARS[stats.data.vol.par-1][1];
1193 : syskin 1498 }
1194 :    
1195 :     ar_x = par_x * stats.data.vol.width;
1196 :     ar_y = par_y * stats.data.vol.height;
1197 :     }
1198 :    
1199 : edgomez 1384 m_frame.bitstream = (BYTE*)m_frame.bitstream + length;
1200 :     m_frame.length -= length;
1201 :     goto repeat;
1202 :     }
1203 :    
1204 :     if (pIn->IsPreroll() == S_OK) {
1205 :     return S_FALSE;
1206 :     }
1207 :    
1208 :     return S_OK;
1209 :     }
1210 :    
1211 :    
1212 :     /* get property page list */
1213 :    
1214 :     STDMETHODIMP CXvidDecoder::GetPages(CAUUID * pPages)
1215 :     {
1216 :     DPRINTF("GetPages");
1217 :    
1218 :     pPages->cElems = 1;
1219 :     pPages->pElems = (GUID *)CoTaskMemAlloc(pPages->cElems * sizeof(GUID));
1220 :     if (pPages->pElems == NULL)
1221 :     {
1222 :     return E_OUTOFMEMORY;
1223 :     }
1224 :     pPages->pElems[0] = CLSID_CABOUT;
1225 :    
1226 :     return S_OK;
1227 :     }
1228 :    
1229 :    
1230 :     /* cleanup pages */
1231 :    
1232 :     STDMETHODIMP CXvidDecoder::FreePages(CAUUID * pPages)
1233 :     {
1234 :     DPRINTF("FreePages");
1235 :     CoTaskMemFree(pPages->pElems);
1236 :     return S_OK;
1237 :     }
1238 : Isibaar 1898
1239 :     /*===============================================================================
1240 :     // MFT Interface
1241 :     //=============================================================================*/
1242 :     #if defined(XVID_USE_MFT)
1243 :     #include <limits.h> // _I64_MAX
1244 :     #define INVALID_TIME _I64_MAX
1245 :    
1246 :     HRESULT CXvidDecoder::MFTGetStreamLimits(DWORD *pdwInputMinimum, DWORD *pdwInputMaximum, DWORD *pdwOutputMinimum, DWORD *pdwOutputMaximum)
1247 :     {
1248 :     DPRINTF("(MFT)GetStreamLimits");
1249 :    
1250 :     if ((pdwInputMinimum == NULL) || (pdwInputMaximum == NULL) || (pdwOutputMinimum == NULL) || (pdwOutputMaximum == NULL))
1251 :     return E_POINTER;
1252 :    
1253 :     /* Just a fixed number of streams allowed */
1254 :     *pdwInputMinimum = *pdwInputMaximum = 1;
1255 :     *pdwOutputMinimum = *pdwOutputMaximum = 1;
1256 :    
1257 :     return S_OK;
1258 :     }
1259 :    
1260 :     HRESULT CXvidDecoder::MFTGetStreamCount(DWORD *pcInputStreams, DWORD *pcOutputStreams)
1261 :     {
1262 :     DPRINTF("(MFT)GetStreamCount");
1263 :    
1264 :     if ((pcInputStreams == NULL) || (pcOutputStreams == NULL))
1265 :     return E_POINTER;
1266 :    
1267 :     /* We have a fixed number of streams */
1268 :     *pcInputStreams = 1;
1269 :     *pcOutputStreams = 1;
1270 :    
1271 :     return S_OK;
1272 :     }
1273 :    
1274 :     HRESULT CXvidDecoder::MFTGetStreamIDs(DWORD dwInputIDArraySize, DWORD *pdwInputIDs, DWORD dwOutputIDArraySize, DWORD *pdwOutputIDs)
1275 :     {
1276 :     DPRINTF("(MFT)GetStreamIDs");
1277 :     return E_NOTIMPL; /* We have fixed number of streams, so stream ID match stream index */
1278 :     }
1279 :    
1280 :     HRESULT CXvidDecoder::MFTGetInputStreamInfo(DWORD dwInputStreamID, MFT_INPUT_STREAM_INFO *pStreamInfo)
1281 :     {
1282 :     DPRINTF("(MFT)GetInputStreamInfo");
1283 :    
1284 :     if (pStreamInfo == NULL)
1285 :     return E_POINTER;
1286 :    
1287 :     if (dwInputStreamID != 0)
1288 :     return MF_E_INVALIDSTREAMNUMBER;
1289 :    
1290 :     EnterCriticalSection(&m_mft_lock);
1291 :    
1292 :     pStreamInfo->dwFlags = MFT_INPUT_STREAM_WHOLE_SAMPLES | MFT_INPUT_STREAM_SINGLE_SAMPLE_PER_BUFFER;
1293 :     pStreamInfo->hnsMaxLatency = 0;
1294 :    
1295 :     pStreamInfo->cbSize = 1; /* Need atleast 1 byte input */
1296 :     pStreamInfo->cbMaxLookahead = 0;
1297 :     pStreamInfo->cbAlignment = 1;
1298 :    
1299 :     LeaveCriticalSection(&m_mft_lock);
1300 :     return S_OK;
1301 :     }
1302 :    
1303 :     HRESULT CXvidDecoder::MFTGetOutputStreamInfo(DWORD dwOutputStreamID, MFT_OUTPUT_STREAM_INFO *pStreamInfo)
1304 :     {
1305 :     DPRINTF("(MFT)GetOutputStreamInfo");
1306 :    
1307 :     if (pStreamInfo == NULL)
1308 :     return E_POINTER;
1309 :    
1310 :     if (dwOutputStreamID != 0)
1311 :     return MF_E_INVALIDSTREAMNUMBER;
1312 :    
1313 :     EnterCriticalSection(&m_mft_lock);
1314 :    
1315 :     pStreamInfo->dwFlags = MFT_OUTPUT_STREAM_WHOLE_SAMPLES | MFT_OUTPUT_STREAM_SINGLE_SAMPLE_PER_BUFFER | MFT_OUTPUT_STREAM_FIXED_SAMPLE_SIZE | MFT_OUTPUT_STREAM_DISCARDABLE;
1316 :    
1317 :     if (m_pOutputType == NULL) {
1318 :     pStreamInfo->cbSize = 0;
1319 :     pStreamInfo->cbAlignment = 0;
1320 :     }
1321 :     else {
1322 :     pStreamInfo->cbSize = m_create.width * abs(m_create.height) * 4; // XXX
1323 :     pStreamInfo->cbAlignment = 1;
1324 :     }
1325 :    
1326 :     LeaveCriticalSection(&m_mft_lock);
1327 :     return S_OK;
1328 :     }
1329 :    
1330 :     HRESULT CXvidDecoder::GetAttributes(IMFAttributes** pAttributes)
1331 :     {
1332 :     DPRINTF("(MFT)GetAttributes");
1333 :     return E_NOTIMPL; /* We don't support any attributes */
1334 :     }
1335 :    
1336 :     HRESULT CXvidDecoder::GetInputStreamAttributes(DWORD dwInputStreamID, IMFAttributes **ppAttributes)
1337 :     {
1338 :     DPRINTF("(MFT)GetInputStreamAttributes");
1339 :     return E_NOTIMPL; /* We don't support any attributes */
1340 :     }
1341 :    
1342 :     HRESULT CXvidDecoder::GetOutputStreamAttributes(DWORD dwOutputStreamID, IMFAttributes **ppAttributes)
1343 :     {
1344 :     DPRINTF("(MFT)GetOutputStreamAttributes");
1345 :     return E_NOTIMPL; /* We don't support any attributes */
1346 :     }
1347 :    
1348 :     HRESULT CXvidDecoder::MFTDeleteInputStream(DWORD dwStreamID)
1349 :     {
1350 :     DPRINTF("(MFT)DeleteInputStream");
1351 :     return E_NOTIMPL; /* We have a fixed number of streams */
1352 :     }
1353 :    
1354 :     HRESULT CXvidDecoder::MFTAddInputStreams(DWORD cStreams, DWORD *adwStreamIDs)
1355 :     {
1356 :     DPRINTF("(MFT)AddInputStreams");
1357 :     return E_NOTIMPL; /* We have a fixed number of streams */
1358 :     }
1359 :    
1360 :     HRESULT CXvidDecoder::MFTGetInputAvailableType(DWORD dwInputStreamID, DWORD dwTypeIndex, IMFMediaType **ppType)
1361 :     {
1362 :     DPRINTF("(MFT)GetInputAvailableType");
1363 :    
1364 :     if (dwInputStreamID != 0)
1365 :     return MF_E_INVALIDSTREAMNUMBER;
1366 :    
1367 :     DWORD i = 0;
1368 :     GUID *bs_guid_table[8];
1369 :    
1370 :     bs_guid_table[i++] = (GUID *)&CLSID_XVID;
1371 :     bs_guid_table[i++] = (GUID *)&CLSID_XVID_UC;
1372 :    
1373 : Isibaar 1949 if (g_config.supported_4cc & SUPPORT_3IVX) {
1374 :     bs_guid_table[i++] = (GUID *)&CLSID_3IVX;
1375 :     bs_guid_table[i++] = (GUID *)&CLSID_3IVX_UC;
1376 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV0;
1377 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV0_UC;
1378 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV1;
1379 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV1_UC;
1380 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV2;
1381 :     bs_guid_table[i++] = (GUID *)&CLSID_3IV2_UC;
1382 : Isibaar 1898 }
1383 :     if (g_config.supported_4cc & SUPPORT_DIVX) {
1384 :     bs_guid_table[i++] = (GUID *)&CLSID_DIVX;
1385 :     bs_guid_table[i++] = (GUID *)&CLSID_DIVX_UC;
1386 : Isibaar 1949 bs_guid_table[i++] = (GUID *)&CLSID_DX50;
1387 :     bs_guid_table[i++] = (GUID *)&CLSID_DX50_UC;
1388 : Isibaar 1898 }
1389 :     if (g_config.supported_4cc & SUPPORT_MP4V) {
1390 :     bs_guid_table[i++] = (GUID *)&CLSID_MP4V;
1391 :     bs_guid_table[i++] = (GUID *)&CLSID_MP4V_UC;
1392 : Isibaar 1949 bs_guid_table[i++] = (GUID *)&CLSID_LMP4;
1393 :     bs_guid_table[i++] = (GUID *)&CLSID_LMP4_UC;
1394 :     bs_guid_table[i++] = (GUID *)&CLSID_RMP4;
1395 :     bs_guid_table[i++] = (GUID *)&CLSID_RMP4_UC;
1396 :     bs_guid_table[i++] = (GUID *)&CLSID_SMP4;
1397 :     bs_guid_table[i++] = (GUID *)&CLSID_SMP4_UC;
1398 :     bs_guid_table[i++] = (GUID *)&CLSID_HDX4;
1399 :     bs_guid_table[i++] = (GUID *)&CLSID_HDX4_UC;
1400 : Isibaar 1898 }
1401 :    
1402 :     const GUID *subtype;
1403 :     if (dwTypeIndex < i) {
1404 :     subtype = bs_guid_table[dwTypeIndex];
1405 :     }
1406 :     else {
1407 :     return MF_E_NO_MORE_TYPES;
1408 :     }
1409 :    
1410 :     EnterCriticalSection(&m_mft_lock);
1411 :    
1412 :     HRESULT hr = S_OK;
1413 :    
1414 :     if (ppType) {
1415 :     IMFMediaType *pInputType = NULL;
1416 :     hr = MFCreateMediaType(&pInputType);
1417 :    
1418 :     if (SUCCEEDED(hr))
1419 :     hr = pInputType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video);
1420 :    
1421 :     if (SUCCEEDED(hr))
1422 :     hr = pInputType->SetGUID(MF_MT_SUBTYPE, *subtype);
1423 :    
1424 :     if (SUCCEEDED(hr)) {
1425 :     *ppType = pInputType;
1426 :     (*ppType)->AddRef();
1427 :     }
1428 :     if (pInputType) pInputType->Release();
1429 :     }
1430 :    
1431 :     LeaveCriticalSection(&m_mft_lock);
1432 :    
1433 :     return hr;
1434 :     }
1435 :    
1436 :     HRESULT CXvidDecoder::MFTGetOutputAvailableType(DWORD dwOutputStreamID, DWORD dwTypeIndex, IMFMediaType **ppType)
1437 :     {
1438 :     DPRINTF("(MFT)GetOutputAvailableType");
1439 :    
1440 :     if (ppType == NULL)
1441 :     return E_INVALIDARG;
1442 :    
1443 :     if (dwOutputStreamID != 0)
1444 :     return MF_E_INVALIDSTREAMNUMBER;
1445 :    
1446 :     if (dwTypeIndex < 0) return E_INVALIDARG;
1447 :    
1448 :     GUID csp;
1449 :     int bitdepth = 8;
1450 :     switch(dwTypeIndex)
1451 :     {
1452 :     case 0:
1453 :     if ( USE_YUY2 )
1454 :     {
1455 :     csp = MFVideoFormat_YUY2;
1456 :     bitdepth = 4;
1457 :     break;
1458 :     }
1459 :     case 1 :
1460 :     if ( USE_UYVY )
1461 :     {
1462 :     csp = MFVideoFormat_UYVY;
1463 :     bitdepth = 4;
1464 :     break;
1465 :     }
1466 :     case 2 :
1467 :     if ( USE_IYUV )
1468 :     {
1469 :     csp = MFVideoFormat_IYUV;
1470 :     bitdepth = 3;
1471 :     break;
1472 :     }
1473 :     case 3 :
1474 :     if ( USE_YV12 )
1475 :     {
1476 :     csp = MFVideoFormat_YV12;
1477 :     bitdepth = 3;
1478 :     break;
1479 :     }
1480 :     case 4 :
1481 :     if ( USE_RGB32 )
1482 :     {
1483 :     csp = MFVideoFormat_RGB32;
1484 :     bitdepth = 8;
1485 :     break;
1486 :     }
1487 :     case 5 :
1488 :     if ( USE_RGB24 )
1489 :     {
1490 :     csp = MFVideoFormat_RGB24;
1491 :     bitdepth = 6;
1492 :     break;
1493 :     }
1494 :     case 6 :
1495 :     if ( USE_RG555 )
1496 :     {
1497 :     csp = MFVideoFormat_RGB555;
1498 :     bitdepth = 4;
1499 :     break;
1500 :     }
1501 :     case 7 :
1502 :     if ( USE_RG565 )
1503 :     {
1504 :     csp = MFVideoFormat_RGB565;
1505 :     bitdepth = 4;
1506 :     break;
1507 :     }
1508 :     default :
1509 :     return MF_E_NO_MORE_TYPES;
1510 :     }
1511 :    
1512 :     if (m_pInputType == NULL)
1513 :     return MF_E_TRANSFORM_TYPE_NOT_SET;
1514 :    
1515 :     EnterCriticalSection(&m_mft_lock);
1516 :    
1517 :     HRESULT hr = S_OK;
1518 :    
1519 :     IMFMediaType *pOutputType = NULL;
1520 :     hr = MFCreateMediaType(&pOutputType);
1521 :    
1522 :     if (SUCCEEDED(hr)) {
1523 :     hr = pOutputType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video);
1524 :     }
1525 :    
1526 :     if (SUCCEEDED(hr)) {
1527 :     hr = pOutputType->SetGUID(MF_MT_SUBTYPE, csp);
1528 :     }
1529 :    
1530 :     if (SUCCEEDED(hr)) {
1531 :     hr = pOutputType->SetUINT32(MF_MT_FIXED_SIZE_SAMPLES, TRUE);
1532 :     }
1533 :    
1534 :     if (SUCCEEDED(hr)) {
1535 :     hr = pOutputType->SetUINT32(MF_MT_ALL_SAMPLES_INDEPENDENT, TRUE);
1536 :     }
1537 :    
1538 :     if (SUCCEEDED(hr)) {
1539 :     hr = pOutputType->SetUINT32(MF_MT_SAMPLE_SIZE, (m_create.height * m_create.width * bitdepth)>>1);
1540 :     }
1541 :    
1542 :     if (SUCCEEDED(hr)) {
1543 :     hr = MFSetAttributeSize(pOutputType, MF_MT_FRAME_SIZE, m_create.width, m_create.height);
1544 :     }
1545 :    
1546 :     if (SUCCEEDED(hr)) {
1547 :     hr = MFSetAttributeRatio(pOutputType, MF_MT_FRAME_RATE, m_frameRate.Numerator, m_frameRate.Denominator);
1548 :     }
1549 :    
1550 :     if (SUCCEEDED(hr)) {
1551 :     hr = pOutputType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_Progressive);
1552 :     }
1553 :    
1554 :     if (SUCCEEDED(hr)) {
1555 :     hr = MFSetAttributeRatio(pOutputType, MF_MT_PIXEL_ASPECT_RATIO, ar_x, ar_y);
1556 :     }
1557 :    
1558 :     if (SUCCEEDED(hr)) {
1559 :     *ppType = pOutputType;
1560 :     (*ppType)->AddRef();
1561 :     }
1562 :    
1563 :     if (pOutputType) pOutputType->Release();
1564 :    
1565 :     LeaveCriticalSection(&m_mft_lock);
1566 :     return hr;
1567 :     }
1568 :    
1569 :     HRESULT CXvidDecoder::MFTSetInputType(DWORD dwInputStreamID, IMFMediaType *pType, DWORD dwFlags)
1570 :     {
1571 :     DPRINTF("(MFT)SetInputType");
1572 :    
1573 :     if (dwInputStreamID != 0)
1574 :     return MF_E_INVALIDSTREAMNUMBER;
1575 :    
1576 :     if (dwFlags & ~MFT_SET_TYPE_TEST_ONLY)
1577 :     return E_INVALIDARG;
1578 :    
1579 :     EnterCriticalSection(&m_mft_lock);
1580 :    
1581 :     HRESULT hr = S_OK;
1582 :    
1583 :     /* Actually set the type or just test it? */
1584 :     BOOL bReallySet = ((dwFlags & MFT_SET_TYPE_TEST_ONLY) == 0);
1585 :    
1586 :     /* If we have samples pending the type can't be changed right now */
1587 :     if (HasPendingOutput())
1588 :     hr = MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
1589 :    
1590 :     if (SUCCEEDED(hr)) {
1591 :     if (pType) { // /* Check the type */
1592 :     hr = OnCheckInputType(pType);
1593 :     }
1594 :     }
1595 :    
1596 :     if (SUCCEEDED(hr)) {
1597 :     if (bReallySet) { /* Set the type if needed */
1598 :     hr = OnSetInputType(pType);
1599 :     }
1600 :     }
1601 :    
1602 :     LeaveCriticalSection(&m_mft_lock);
1603 :     return hr;
1604 :     }
1605 :    
1606 :     HRESULT CXvidDecoder::MFTSetOutputType(DWORD dwOutputStreamID, IMFMediaType *pType, DWORD dwFlags)
1607 :     {
1608 :     DPRINTF("(MFT)SetOutputType");
1609 :    
1610 :     if (dwOutputStreamID != 0)
1611 :     return MF_E_INVALIDSTREAMNUMBER;
1612 :    
1613 :     if (dwFlags & ~MFT_SET_TYPE_TEST_ONLY)
1614 :     return E_INVALIDARG;
1615 :    
1616 :     HRESULT hr = S_OK;
1617 :    
1618 :     EnterCriticalSection(&m_mft_lock);
1619 :    
1620 : Isibaar 1899 /* Actually set the type or just test it? */
1621 : Isibaar 1898 BOOL bReallySet = ((dwFlags & MFT_SET_TYPE_TEST_ONLY) == 0);
1622 :    
1623 :     /* If we have samples pending the type can't be changed right now */
1624 :     if (HasPendingOutput())
1625 :     hr = MF_E_TRANSFORM_CANNOT_CHANGE_MEDIATYPE_WHILE_PROCESSING;
1626 :    
1627 :     if (SUCCEEDED(hr)) {
1628 :     if (pType) { /* Check the type */
1629 :     AM_MEDIA_TYPE *am;
1630 :     hr = MFCreateAMMediaTypeFromMFMediaType(pType, GUID_NULL, &am);
1631 :    
1632 :     if (SUCCEEDED(hr)) {
1633 :     if (FAILED(ChangeColorspace(am->subtype, am->formattype, am->pbFormat))) {
1634 :     DPRINTF("(MFT)InternalCheckOutputType (MF_E_INVALIDTYPE)");
1635 :     return MF_E_INVALIDTYPE;
1636 :     }
1637 :    
1638 :     CoTaskMemFree(am->pbFormat);
1639 :     CoTaskMemFree(am);
1640 :     }
1641 :     }
1642 :     }
1643 :    
1644 :     if (SUCCEEDED(hr)) {
1645 :     if (bReallySet) { /* Set the type if needed */
1646 :     hr = OnSetOutputType(pType);
1647 :     }
1648 :     }
1649 :     #ifdef XVID_USE_TRAYICON
1650 : Isibaar 1901 if (SUCCEEDED(hr) && Tray_Icon == 0) /* Create message passing window */
1651 : Isibaar 1898 {
1652 :     WNDCLASSEX wc;
1653 :    
1654 :     wc.cbSize = sizeof(WNDCLASSEX);
1655 :     wc.lpfnWndProc = msg_proc;
1656 :     wc.style = CS_HREDRAW | CS_VREDRAW;
1657 :     wc.cbWndExtra = 0;
1658 :     wc.cbClsExtra = 0;
1659 :     wc.hInstance = (HINSTANCE) g_xvid_hInst;
1660 :     wc.hbrBackground = (HBRUSH) GetStockObject(NULL_BRUSH);
1661 :     wc.lpszMenuName = NULL;
1662 :     wc.lpszClassName = "XVID_MSG_WINDOW";
1663 :     wc.hIcon = NULL;
1664 :     wc.hIconSm = NULL;
1665 :     wc.hCursor = NULL;
1666 :     RegisterClassEx(&wc);
1667 :    
1668 :     MSG_hwnd = CreateWindowEx(0, "XVID_MSG_WINDOW", NULL, 0, CW_USEDEFAULT,
1669 :     CW_USEDEFAULT, 0, 0, HWND_MESSAGE, NULL, (HINSTANCE) g_xvid_hInst, NULL);
1670 :    
1671 :     /* display the tray icon */
1672 :     NOTIFYICONDATA nid;
1673 : Isibaar 1901 ZeroMemory(&nid,sizeof(NOTIFYICONDATA));
1674 :    
1675 :     nid.cbSize = NOTIFYICONDATA_V1_SIZE;
1676 : Isibaar 1898 nid.hWnd = MSG_hwnd;
1677 : Isibaar 1901 nid.uID = 1456;
1678 : Isibaar 1898 nid.uCallbackMessage = WM_ICONMESSAGE;
1679 :     nid.hIcon = LoadIcon(g_xvid_hInst, MAKEINTRESOURCE(IDI_ICON));
1680 :     strcpy_s(nid.szTip, 19, "Xvid Video Decoder");
1681 :     nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
1682 : Isibaar 1916
1683 : Isibaar 1898 Shell_NotifyIcon(NIM_ADD, &nid);
1684 : Isibaar 1901
1685 :     DestroyIcon(nid.hIcon);
1686 :     Tray_Icon = 1;
1687 : Isibaar 1898 }
1688 :     #endif
1689 :    
1690 :     LeaveCriticalSection(&m_mft_lock);
1691 :     return hr;
1692 :     }
1693 :    
1694 :     HRESULT CXvidDecoder::MFTGetInputCurrentType(DWORD dwInputStreamID, IMFMediaType **ppType)
1695 :     {
1696 :     DPRINTF("(MFT)GetInputCurrentType");
1697 :    
1698 :     if (ppType == NULL)
1699 :     return E_POINTER;
1700 :    
1701 :     if (dwInputStreamID != 0)
1702 :     return MF_E_INVALIDSTREAMNUMBER;
1703 :    
1704 :     EnterCriticalSection(&m_mft_lock);
1705 :    
1706 :     HRESULT hr = S_OK;
1707 :    
1708 :     if (!m_pInputType)
1709 :     hr = MF_E_TRANSFORM_TYPE_NOT_SET;
1710 :    
1711 :     if (SUCCEEDED(hr)) {
1712 :     *ppType = m_pInputType;
1713 :     (*ppType)->AddRef();
1714 :     }
1715 :    
1716 :     LeaveCriticalSection(&m_mft_lock);
1717 :     return hr;
1718 :     }
1719 :    
1720 :     HRESULT CXvidDecoder::MFTGetOutputCurrentType(DWORD dwOutputStreamID, IMFMediaType **ppType)
1721 :     {
1722 :     DPRINTF("(MFT)GetOutputCurrentType");
1723 :    
1724 :     if (ppType == NULL)
1725 :     return E_POINTER;
1726 :    
1727 :     if (dwOutputStreamID != 0)
1728 :     return MF_E_INVALIDSTREAMNUMBER;
1729 :    
1730 :     EnterCriticalSection(&m_mft_lock);
1731 :    
1732 :     HRESULT hr = S_OK;
1733 :    
1734 :     if (!m_pOutputType)
1735 :     hr = MF_E_TRANSFORM_TYPE_NOT_SET;
1736 :    
1737 :     if (SUCCEEDED(hr)) {
1738 :     *ppType = m_pOutputType;
1739 :     (*ppType)->AddRef();
1740 :     }
1741 :    
1742 :     LeaveCriticalSection(&m_mft_lock);
1743 :     return hr;
1744 :     }
1745 :    
1746 :     HRESULT CXvidDecoder::MFTGetInputStatus(DWORD dwInputStreamID, DWORD *pdwFlags)
1747 :     {
1748 :     DPRINTF("(MFT)GetInputStatus");
1749 :    
1750 :     if (pdwFlags == NULL)
1751 :     return E_POINTER;
1752 :    
1753 :     if (dwInputStreamID != 0)
1754 :     return MF_E_INVALIDSTREAMNUMBER;
1755 :    
1756 :     EnterCriticalSection(&m_mft_lock);
1757 :    
1758 :     /* If there's pending output sampels we don't accept new
1759 :     input data until ProcessOutput() or Flush() was called */
1760 :     if (!HasPendingOutput()) {
1761 :     *pdwFlags = MFT_INPUT_STATUS_ACCEPT_DATA;
1762 :     }
1763 :     else {
1764 :     *pdwFlags = 0;
1765 :     }
1766 :    
1767 :     LeaveCriticalSection(&m_mft_lock);
1768 :    
1769 :     return S_OK;
1770 :     }
1771 :    
1772 :     HRESULT CXvidDecoder::MFTGetOutputStatus(DWORD *pdwFlags)
1773 :     {
1774 :     DPRINTF("(MFT)GetOutputStatus");
1775 :    
1776 :     if (pdwFlags == NULL)
1777 :     return E_POINTER;
1778 :    
1779 :     EnterCriticalSection(&m_mft_lock);
1780 :    
1781 :     /* We can render an output sample only after we
1782 :     have decoded one */
1783 :     if (HasPendingOutput()) {
1784 :     *pdwFlags = MFT_OUTPUT_STATUS_SAMPLE_READY;
1785 :     }
1786 :     else {
1787 :     *pdwFlags = 0;
1788 :     }
1789 :    
1790 :     LeaveCriticalSection(&m_mft_lock);
1791 :    
1792 :     return S_OK;
1793 :     }
1794 :    
1795 :     HRESULT CXvidDecoder::MFTSetOutputBounds(LONGLONG hnsLowerBound, LONGLONG hnsUpperBound)
1796 :     {
1797 :     DPRINTF("(MFT)SetOutputBounds");
1798 :     return E_NOTIMPL;
1799 :     }
1800 :    
1801 :     HRESULT CXvidDecoder::MFTProcessEvent(DWORD dwInputStreamID, IMFMediaEvent *pEvent)
1802 :     {
1803 :     DPRINTF("(MFT)ProcessEvent");
1804 :     return E_NOTIMPL; /* We don't handle any stream events */
1805 :     }
1806 :    
1807 :     HRESULT CXvidDecoder::MFTProcessMessage(MFT_MESSAGE_TYPE eMessage, ULONG_PTR ulParam)
1808 :     {
1809 :     DPRINTF("(MFT)ProcessMessage");
1810 :     HRESULT hr = S_OK;
1811 :    
1812 :     EnterCriticalSection(&m_mft_lock);
1813 :    
1814 :     switch (eMessage)
1815 :     {
1816 :     case MFT_MESSAGE_COMMAND_FLUSH:
1817 :     if (m_create.handle != NULL) {
1818 :     DPRINTF("(MFT)CommandFlush");
1819 :    
1820 :     xvid_dec_stats_t stats;
1821 :     int used_bytes;
1822 :    
1823 :     memset(&stats, 0, sizeof(stats));
1824 :     stats.version = XVID_VERSION;
1825 :    
1826 :     int csp = m_frame.output.csp;
1827 :    
1828 :     m_frame.output.csp = XVID_CSP_INTERNAL;
1829 :     m_frame.bitstream = NULL;
1830 :     m_frame.length = -1;
1831 :     m_frame.general = XVID_LOWDELAY;
1832 :    
1833 :     do {
1834 :     used_bytes = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1835 :     } while(used_bytes>=0 && stats.type <= 0);
1836 :    
1837 :     m_frame.output.csp = csp;
1838 :     m_frame.output.plane[1] = NULL; /* Don't display flushed samples */
1839 :    
1840 :     //m_timestamp = INVALID_TIME;
1841 :     //m_timelength = INVALID_TIME;
1842 :     //m_rtFrame = 0;
1843 :     }
1844 :     break;
1845 :    
1846 :     case MFT_MESSAGE_COMMAND_DRAIN:
1847 :     m_discont = 1; /* Set discontinuity flag */
1848 :     m_rtFrame = 0;
1849 :     break;
1850 :    
1851 :     case MFT_MESSAGE_SET_D3D_MANAGER:
1852 :     hr = E_NOTIMPL;
1853 :     break;
1854 :    
1855 :     case MFT_MESSAGE_NOTIFY_BEGIN_STREAMING:
1856 :     case MFT_MESSAGE_NOTIFY_END_STREAMING:
1857 :     break;
1858 :    
1859 :     case MFT_MESSAGE_NOTIFY_START_OF_STREAM:
1860 :     case MFT_MESSAGE_NOTIFY_END_OF_STREAM:
1861 :     break;
1862 :     }
1863 :    
1864 :     LeaveCriticalSection(&m_mft_lock);
1865 :    
1866 :     return hr;
1867 :     }
1868 :    
1869 :     HRESULT CXvidDecoder::MFTProcessInput(DWORD dwInputStreamID, IMFSample *pSample, DWORD dwFlags)
1870 :     {
1871 :     DPRINTF("(MFT)ProcessInput");
1872 :    
1873 :     if (pSample == NULL)
1874 :     return E_POINTER;
1875 :    
1876 :     if (dwInputStreamID != 0)
1877 :     return MF_E_INVALIDSTREAMNUMBER;
1878 :    
1879 :     if (dwFlags != 0)
1880 :     return E_INVALIDARG;
1881 :    
1882 :     if (!m_pInputType || !m_pOutputType) {
1883 :     return MF_E_NOTACCEPTING; /* Must have set input and output types */
1884 :     }
1885 :     else if (HasPendingOutput()) {
1886 :     return MF_E_NOTACCEPTING; /* We still have output samples to render */
1887 :     }
1888 :    
1889 :     xvid_dec_stats_t stats;
1890 :     int length;
1891 :    
1892 :     memset(&stats, 0, sizeof(stats));
1893 :     stats.version = XVID_VERSION;
1894 :    
1895 :     if (m_create.handle == NULL)
1896 :     {
1897 :     if (xvid_decore_func == NULL)
1898 :     return E_FAIL;
1899 :     if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)
1900 :     {
1901 :     DPRINTF("*** XVID_DEC_CREATE error");
1902 :     return E_FAIL;
1903 :     }
1904 :     }
1905 :    
1906 :     EnterCriticalSection(&m_mft_lock);
1907 :    
1908 :     HRESULT hr = S_OK;
1909 :     IMFMediaBuffer *pBuffer;
1910 :    
1911 :     if (SUCCEEDED(hr)) {
1912 :     hr = pSample->ConvertToContiguousBuffer(&pBuffer);
1913 :     }
1914 :    
1915 :     if (SUCCEEDED(hr)) {
1916 :     hr = pBuffer->Lock((BYTE**)&m_frame.bitstream, NULL, (DWORD *)&m_frame.length);
1917 :     }
1918 :    
1919 :     m_frame.general = XVID_LOWDELAY;
1920 :    
1921 :     if (m_discont == 1) {
1922 :     m_frame.general |= XVID_DISCONTINUITY;
1923 :     m_discont = 0;
1924 :     }
1925 :    
1926 :     if (g_config.nDeblock_Y)
1927 :     m_frame.general |= XVID_DEBLOCKY;
1928 :    
1929 :     if (g_config.nDeblock_UV)
1930 :     m_frame.general |= XVID_DEBLOCKUV;
1931 :    
1932 :     if (g_config.nDering_Y)
1933 :     m_frame.general |= XVID_DERINGY;
1934 :    
1935 :     if (g_config.nDering_UV)
1936 :     m_frame.general |= XVID_DERINGUV;
1937 :    
1938 :     if (g_config.nFilmEffect)
1939 :     m_frame.general |= XVID_FILMEFFECT;
1940 :    
1941 :     m_frame.brightness = g_config.nBrightness;
1942 :    
1943 :     m_frame.output.csp &= ~XVID_CSP_VFLIP;
1944 :     m_frame.output.csp |= rgb_flip^(g_config.nFlipVideo ? XVID_CSP_VFLIP : 0);
1945 :    
1946 :     int csp = m_frame.output.csp;
1947 :     m_frame.output.csp = XVID_CSP_INTERNAL;
1948 :    
1949 :     // Paranoid check.
1950 :     if (xvid_decore_func == NULL) {
1951 :     hr = E_FAIL;
1952 :     goto END_LOOP;
1953 :     }
1954 :    
1955 :     repeat :
1956 :     length = xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats);
1957 :    
1958 :     if (length == XVID_ERR_MEMORY) {
1959 :     hr = E_FAIL;
1960 :     goto END_LOOP;
1961 :     }
1962 :     else if (length < 0)
1963 :     {
1964 :     DPRINTF("*** XVID_DEC_DECODE");
1965 :     goto END_LOOP;
1966 :     }
1967 :    
1968 :     if (stats.type == XVID_TYPE_NOTHING && length > 0) {
1969 :     DPRINTF(" B-Frame decoder lag");
1970 :     m_frame.output.plane[1] = NULL;
1971 :     goto END_LOOP;
1972 :     }
1973 :    
1974 :     if (stats.type == XVID_TYPE_VOL)
1975 :     {
1976 :     if (stats.data.vol.width != m_create.width ||
1977 :     stats.data.vol.height != m_create.height)
1978 :     {
1979 :     DPRINTF("TODO: auto-resize");
1980 :     m_frame.output.plane[1] = NULL;
1981 :     hr = E_FAIL;
1982 :     }
1983 :    
1984 :     if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) { /* auto */
1985 :     int par_x, par_y;
1986 :     if (stats.data.vol.par == XVID_PAR_EXT) {
1987 :     par_x = stats.data.vol.par_width;
1988 :     par_y = stats.data.vol.par_height;
1989 :     } else {
1990 :     par_x = PARS[stats.data.vol.par-1][0];
1991 :     par_y = PARS[stats.data.vol.par-1][1];
1992 :     }
1993 :    
1994 :     ar_x = par_x * stats.data.vol.width;
1995 :     ar_y = par_y * stats.data.vol.height;
1996 :     }
1997 :    
1998 :     m_frame.bitstream = (BYTE*)m_frame.bitstream + length;
1999 :     m_frame.length -= length;
2000 :     goto repeat;
2001 :     }
2002 :    
2003 :     END_LOOP:
2004 :     m_frame.output.csp = csp;
2005 :    
2006 :     if (pBuffer) {
2007 :     pBuffer->Unlock();
2008 :     pBuffer->Release();
2009 :     }
2010 :    
2011 :     if (SUCCEEDED(hr)) {
2012 :     /* Try to get a timestamp */
2013 :     if (FAILED(pSample->GetSampleTime(&m_timestamp)))
2014 :     m_timestamp = INVALID_TIME;
2015 :    
2016 :     if (FAILED(pSample->GetSampleDuration(&m_timelength))) {
2017 :     m_timelength = INVALID_TIME;
2018 :     }
2019 : Isibaar 1916 if (m_timestamp != INVALID_TIME && stats.type == XVID_TYPE_IVOP) {
2020 :     m_rtFrame = m_timestamp;
2021 :     }
2022 : Isibaar 1898 }
2023 :    
2024 :     LeaveCriticalSection(&m_mft_lock);
2025 :    
2026 :     return hr;
2027 :     }
2028 :    
2029 :     HRESULT CXvidDecoder::MFTProcessOutput(DWORD dwFlags, DWORD cOutputBufferCount, MFT_OUTPUT_DATA_BUFFER *pOutputSamples, DWORD *pdwStatus)
2030 :     {
2031 :     DPRINTF("(MFT)ProcessOutput");
2032 :    
2033 :     /* Preroll in MFT ??
2034 :     Flags ?? -> TODO... */
2035 :     if (dwFlags != 0)
2036 :     return E_INVALIDARG;
2037 :    
2038 :     if (pOutputSamples == NULL || pdwStatus == NULL)
2039 :     return E_POINTER;
2040 :    
2041 :     if (cOutputBufferCount != 1) /* Must be exactly one output buffer */
2042 :     return E_INVALIDARG;
2043 :    
2044 :     if (pOutputSamples[0].pSample == NULL) /* Must have a sample */
2045 :     return E_INVALIDARG;
2046 :    
2047 :     if (!HasPendingOutput()) { /* If there's no sample we need to decode one first */
2048 :     return MF_E_TRANSFORM_NEED_MORE_INPUT;
2049 :     }
2050 :    
2051 :     EnterCriticalSection(&m_mft_lock);
2052 :    
2053 :     HRESULT hr = S_OK;
2054 :    
2055 :     BYTE *Dst = NULL;
2056 :     DWORD buffer_size;
2057 :    
2058 :     IMFMediaBuffer *pOutput = NULL;
2059 :    
2060 :     if (SUCCEEDED(hr)) {
2061 :     hr = pOutputSamples[0].pSample->GetBufferByIndex(0, &pOutput); /* Get output buffer */
2062 :     }
2063 :    
2064 :     if (SUCCEEDED(hr)) {
2065 :     hr = pOutput->GetMaxLength(&buffer_size);
2066 :     }
2067 :    
2068 :     if (SUCCEEDED(hr))
2069 :     hr = pOutput->Lock(&Dst, NULL, NULL);
2070 :    
2071 :     if (SUCCEEDED(hr)) {
2072 :     xvid_gbl_convert_t convert;
2073 :    
2074 :     memset(&convert, 0, sizeof(convert));
2075 :     convert.version = XVID_VERSION;
2076 :    
2077 :     convert.input.csp = XVID_CSP_INTERNAL;
2078 :     convert.input.plane[0] = m_frame.output.plane[0];
2079 :     convert.input.plane[1] = m_frame.output.plane[1];
2080 :     convert.input.plane[2] = m_frame.output.plane[2];
2081 :     convert.input.stride[0] = m_frame.output.stride[0];
2082 :     convert.input.stride[1] = m_frame.output.stride[1];
2083 :     convert.input.stride[2] = m_frame.output.stride[2];
2084 :    
2085 :     convert.output.csp = m_frame.output.csp;
2086 :     convert.output.plane[0] = Dst;
2087 :     convert.output.stride[0] = out_stride;
2088 :    
2089 :     convert.width = m_create.width;
2090 :     convert.height = m_create.height;
2091 :     convert.interlacing = 0;
2092 :    
2093 :     if (m_frame.output.plane[1] != NULL && Dst != NULL && xvid_global_func != NULL)
2094 :     if (xvid_global_func(0, XVID_GBL_CONVERT, &convert, NULL) < 0) /* CSP convert into output buffer */
2095 :     hr = E_FAIL;
2096 :    
2097 :     m_frame.output.plane[1] = NULL;
2098 :     }
2099 :    
2100 :     *pdwStatus = 0;
2101 :    
2102 :     if (SUCCEEDED(hr)) {
2103 :     if (SUCCEEDED(hr))
2104 :     hr = pOutputSamples[0].pSample->SetUINT32(MFSampleExtension_CleanPoint, TRUE); // key frame
2105 :    
2106 :     if (SUCCEEDED(hr)) { /* Set timestamp of output sample */
2107 :     if (m_timestamp != INVALID_TIME)
2108 :     hr = pOutputSamples[0].pSample->SetSampleTime(m_timestamp);
2109 :     else
2110 :     hr = pOutputSamples[0].pSample->SetSampleTime(m_rtFrame);
2111 :    
2112 :     if (m_timelength != INVALID_TIME)
2113 :     hr = pOutputSamples[0].pSample->SetSampleDuration(m_timelength);
2114 :     else
2115 :     hr = pOutputSamples[0].pSample->SetSampleDuration(m_duration);
2116 :    
2117 :     m_rtFrame += m_duration;
2118 :     }
2119 :    
2120 :     if (SUCCEEDED(hr))
2121 :     hr = pOutput->SetCurrentLength(m_create.width * abs(m_create.height) * 4); // XXX
2122 :     }
2123 :    
2124 :     if (pOutput) {
2125 :     pOutput->Unlock();
2126 :     pOutput->Release();
2127 :     }
2128 :    
2129 :     LeaveCriticalSection(&m_mft_lock);
2130 :    
2131 :     return hr;
2132 :     }
2133 :    
2134 :     HRESULT CXvidDecoder::OnCheckInputType(IMFMediaType *pmt)
2135 :     {
2136 :     DPRINTF("(MFT)CheckInputType");
2137 :    
2138 :     HRESULT hr = S_OK;
2139 :    
2140 :     /* Check if input type is already set. Reject any type that is not identical */
2141 :     if (m_pInputType) {
2142 :     DWORD dwFlags = 0;
2143 :     if (S_OK == m_pInputType->IsEqual(pmt, &dwFlags)) {
2144 :     return S_OK;
2145 :     }
2146 :     else {
2147 :     return MF_E_INVALIDTYPE;
2148 :     }
2149 :     }
2150 :    
2151 :     GUID majortype = {0}, subtype = {0};
2152 :     UINT32 width = 0, height = 0;
2153 :    
2154 :     hr = pmt->GetMajorType(&majortype);
2155 :    
2156 :     if (SUCCEEDED(hr)) {
2157 :     if (majortype != MFMediaType_Video) { /* Must be Video */
2158 :     hr = MF_E_INVALIDTYPE;
2159 :     }
2160 :     }
2161 :    
2162 :     if (m_hdll == NULL) {
2163 :     HRESULT hr = OpenLib();
2164 :    
2165 :     if (FAILED(hr) || (m_hdll == NULL)) // Paranoid checks.
2166 :     hr = MF_E_INVALIDTYPE;
2167 :     }
2168 :    
2169 :     if (SUCCEEDED(hr)) {
2170 :     hr = MFGetAttributeSize(pmt, MF_MT_FRAME_SIZE, &width, &height);
2171 :     }
2172 :    
2173 :     /* Check the frame size */
2174 :     if (SUCCEEDED(hr)) {
2175 :     if (width > 4096 || height > 4096) {
2176 :     hr = MF_E_INVALIDTYPE;
2177 :     }
2178 :     }
2179 :     m_create.width = width;
2180 :     m_create.height = height;
2181 :    
2182 :     if (SUCCEEDED(hr)) {
2183 :     if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
2184 :     hr = MFGetAttributeRatio(pmt, MF_MT_PIXEL_ASPECT_RATIO, (UINT32*)&ar_x, (UINT32*)&ar_y);
2185 :     }
2186 :     }
2187 :    
2188 :     /* TODO1: Make sure there really is a frame rate after all!
2189 :     TODO2: Use the framerate for something! */
2190 :     MFRatio fps = {0};
2191 :     if (SUCCEEDED(hr)) {
2192 :     hr = MFGetAttributeRatio(pmt, MF_MT_FRAME_RATE, (UINT32*)&fps.Numerator, (UINT32*)&fps.Denominator);
2193 :     }
2194 :    
2195 :     if (SUCCEEDED(hr)) {
2196 :     hr = pmt->GetGUID(MF_MT_SUBTYPE, &subtype);
2197 :     }
2198 :    
2199 : Isibaar 1949 if (subtype == CLSID_MP4V || subtype == CLSID_MP4V_UC ||
2200 :     subtype == CLSID_LMP4 || subtype == CLSID_LMP4_UC ||
2201 :     subtype == CLSID_RMP4 || subtype == CLSID_RMP4_UC ||
2202 :     subtype == CLSID_SMP4 || subtype == CLSID_SMP4_UC ||
2203 :     subtype == CLSID_HDX4 || subtype == CLSID_HDX4_UC) {
2204 : Isibaar 1898 if (!(g_config.supported_4cc & SUPPORT_MP4V)) {
2205 :     CloseLib();
2206 :     hr = MF_E_INVALIDTYPE;
2207 :     }
2208 :     else m_create.fourcc = FOURCC_MP4V;
2209 :     }
2210 :     else if (subtype == CLSID_DIVX || subtype == CLSID_DIVX_UC) {
2211 :     if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
2212 :     CloseLib();
2213 :     hr = MF_E_INVALIDTYPE;
2214 :     }
2215 :     else m_create.fourcc = FOURCC_DIVX;
2216 :     }
2217 :     else if (subtype == CLSID_DX50 || subtype == CLSID_DX50_UC) {
2218 : Isibaar 1949 if (!(g_config.supported_4cc & SUPPORT_DIVX)) {
2219 : Isibaar 1898 CloseLib();
2220 :     hr = MF_E_INVALIDTYPE;
2221 :     }
2222 :     else m_create.fourcc = FOURCC_DX50;
2223 :     }
2224 : Isibaar 1949 else if (subtype == CLSID_3IVX || subtype == CLSID_3IVX_UC ||
2225 :     subtype == CLSID_3IV0 || subtype == CLSID_3IV0_UC ||
2226 :     subtype == CLSID_3IV1 || subtype == CLSID_3IV1_UC ||
2227 :     subtype == CLSID_3IV2 || subtype == CLSID_3IV2_UC) {
2228 :     if (!(g_config.supported_4cc & SUPPORT_3IVX)) {
2229 :     CloseLib();
2230 :     hr = MF_E_INVALIDTYPE;
2231 :     }
2232 :     else m_create.fourcc = FOURCC_3IVX;
2233 :     }
2234 : Isibaar 1898 else if (subtype == CLSID_XVID || subtype == CLSID_XVID_UC) {
2235 :     m_create.fourcc = FOURCC_XVID;
2236 :     }
2237 :     else {
2238 :     DPRINTF("Unknown subtype!");
2239 :     CloseLib();
2240 :     hr = MF_E_INVALIDTYPE;
2241 :     }
2242 :    
2243 :     /* haali media splitter reports VOL information in the format header */
2244 :     if (SUCCEEDED(hr))
2245 :     {
2246 :     UINT32 cbSeqHeader = 0;
2247 :    
2248 :     (void)pmt->GetBlobSize(MF_MT_MPEG_SEQUENCE_HEADER, &cbSeqHeader);
2249 :    
2250 :     if (cbSeqHeader>0) {
2251 :     xvid_dec_stats_t stats;
2252 :     memset(&stats, 0, sizeof(stats));
2253 :     stats.version = XVID_VERSION;
2254 :    
2255 :     if (m_create.handle == NULL) {
2256 :     if (xvid_decore_func == NULL)
2257 :     hr = E_FAIL;
2258 :     if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) {
2259 :     DPRINTF("*** XVID_DEC_CREATE error");
2260 :     hr = E_FAIL;
2261 :     }
2262 :     }
2263 :    
2264 :     if (SUCCEEDED(hr)) {
2265 :     (void)pmt->GetAllocatedBlob(MF_MT_MPEG_SEQUENCE_HEADER, (UINT8 **)&m_frame.bitstream, (UINT32 *)&m_frame.length);
2266 :     m_frame.general = 0;
2267 :     m_frame.output.csp = XVID_CSP_NULL;
2268 :    
2269 :     int ret = 0;
2270 :     if ((ret=xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats)) >= 0) {
2271 :     /* honour video dimensions reported in VOL header */
2272 :     if (stats.type == XVID_TYPE_VOL) {
2273 :     m_create.width = stats.data.vol.width;
2274 :     m_create.height = stats.data.vol.height;
2275 :     }
2276 :     }
2277 :    
2278 :     if (ret == XVID_ERR_MEMORY) hr = E_FAIL;
2279 :     CoTaskMemFree(m_frame.bitstream);
2280 :     }
2281 :     }
2282 :     }
2283 :    
2284 :     return hr;
2285 :     }
2286 :    
2287 :     HRESULT CXvidDecoder::OnSetInputType(IMFMediaType *pmt)
2288 :     {
2289 :     HRESULT hr = S_OK;
2290 :     UINT32 w, h;
2291 :    
2292 :     if (m_pInputType) m_pInputType->Release();
2293 :    
2294 :     hr = MFGetAttributeSize(pmt, MF_MT_FRAME_SIZE, &w, &h);
2295 :     m_create.width = w; m_create.height = h;
2296 :    
2297 :     if (SUCCEEDED(hr))
2298 :     hr = MFGetAttributeRatio(pmt, MF_MT_FRAME_RATE, (UINT32*)&m_frameRate.Numerator, (UINT32*)&m_frameRate.Denominator);
2299 :    
2300 :     if (SUCCEEDED(hr)) { /* Store frame duration, derived from the frame rate */
2301 :     hr = MFFrameRateToAverageTimePerFrame(m_frameRate.Numerator, m_frameRate.Denominator, &m_duration);
2302 :     }
2303 :    
2304 :     if (SUCCEEDED(hr)) {
2305 :     m_pInputType = pmt;
2306 :     m_pInputType->AddRef();
2307 :     }
2308 :    
2309 :     return hr;
2310 :     }
2311 :    
2312 :     HRESULT CXvidDecoder::OnSetOutputType(IMFMediaType *pmt)
2313 :     {
2314 :     if (m_pOutputType) m_pOutputType->Release();
2315 :    
2316 :     m_pOutputType = pmt;
2317 :     m_pOutputType->AddRef();
2318 :    
2319 :     return S_OK;
2320 :     }
2321 :    
2322 :     #endif /* XVID_USE_MFT */

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4