[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 1890, Tue Aug 10 14:17:40 2010 UTC
# Line 19  Line 19 
19   *  along with this program ; if not, write to the Free Software   *  along with this program ; if not, write to the Free Software
20   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
21   *   *
22   * $Id: CXvidDecoder.cpp,v 1.7 2004-04-18 07:55:11 syskin Exp $   * $Id: CXvidDecoder.cpp,v 1.18 2010-08-10 14:17:40 Isibaar Exp $
23   *   *
24   ****************************************************************************/   ****************************************************************************/
25    
# Line 36  Line 36 
36          place these paths at the top of the Tools|Options|Directories list          place these paths at the top of the Tools|Options|Directories list
37    
38          headers:          headers:
39          C:\DXVCSDK\include          C:\DX90SDK\Include
40          C:\DXVCSDK\samples\Multimedia\DirectShow\BaseClasses          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses
41    
42          libraries (optional):          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Release
43          C:\DXVCSDK\samples\Multimedia\DirectShow\BaseClasses\Release          C:\DX90SDK\Samples\C++\DirectShow\BaseClasses\Debug
44  */  */
45    
46    
# Line 82  Line 82 
82          { &MEDIATYPE_Video, &CLSID_DX50 },          { &MEDIATYPE_Video, &CLSID_DX50 },
83          { &MEDIATYPE_Video, &CLSID_DX50_UC },          { &MEDIATYPE_Video, &CLSID_DX50_UC },
84          { &MEDIATYPE_Video, &CLSID_MP4V },          { &MEDIATYPE_Video, &CLSID_MP4V },
85      { &MEDIATYPE_Video, &CLSID_MP4V_UC },
86  };  };
87    
88  const AMOVIESETUP_MEDIATYPE sudOutputPinTypes[] =  const AMOVIESETUP_MEDIATYPE sudOutputPinTypes[] =
# Line 204  Line 205 
205  #define XVID_DLL_NAME "xvidcore.dll"  #define XVID_DLL_NAME "xvidcore.dll"
206    
207  CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) :  CXvidDecoder::CXvidDecoder(LPUNKNOWN punk, HRESULT *phr) :
208      CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID)      CVideoTransformFilter(NAME("CXvidDecoder"), punk, CLSID_XVID), m_hdll (NULL)
209  {  {
210          DPRINTF("Constructor");          DPRINTF("Constructor");
211    
212        xvid_decore_func = NULL; // Hmm, some strange errors appearing if I try to initialize...
213        xvid_global_func = NULL; // ...this in constructor's init-list. So, they assigned here.
214    
215        LoadRegistryInfo();
216    
217        *phr = OpenLib();
218    }
219    
220    HRESULT CXvidDecoder::OpenLib()
221    {
222        DPRINTF("OpenLib");
223    
224        if (m_hdll != NULL)
225                    return E_UNEXPECTED; // Seems, that library already opened.
226    
227          xvid_gbl_init_t init;          xvid_gbl_init_t init;
228          memset(&init, 0, sizeof(init));          memset(&init, 0, sizeof(init));
229          init.version = XVID_VERSION;          init.version = XVID_VERSION;
230    
         ar_x = ar_y = 0;  
   
231          m_hdll = LoadLibrary(XVID_DLL_NAME);          m_hdll = LoadLibrary(XVID_DLL_NAME);
232          if (m_hdll == NULL) {          if (m_hdll == NULL) {
233                  DPRINTF("dll load failed");                  DPRINTF("dll load failed");
234                  MessageBox(0, XVID_DLL_NAME " not found","Error", 0);                  MessageBox(0, XVID_DLL_NAME " not found","Error", MB_TOPMOST);
235                  return;                  return E_FAIL;
236          }          }
237    
238          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");
239          if (xvid_global_func == NULL) {          if (xvid_global_func == NULL) {
240                  MessageBox(0, "xvid_global() not found", "Error", 0);          FreeLibrary(m_hdll);
241                  return;          m_hdll = NULL;
242                    MessageBox(0, "xvid_global() not found", "Error", MB_TOPMOST);
243                    return E_FAIL;
244          }          }
245    
246          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");
247          if (xvid_decore_func == NULL) {          if (xvid_decore_func == NULL) {
248                  MessageBox(0, "xvid_decore() not found", "Error", 0);          xvid_global_func = NULL;
249                  return;          FreeLibrary(m_hdll);
250            m_hdll = NULL;
251                    MessageBox(0, "xvid_decore() not found", "Error", MB_TOPMOST);
252                    return E_FAIL;
253          }          }
254    
255          if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0)          if (xvid_global_func(0, XVID_GBL_INIT, &init, NULL) < 0)
256          {          {
257                  MessageBox(0, "xvid_global() failed", "Error", 0);          xvid_global_func = NULL;
258                  return;          xvid_decore_func = NULL;
259            FreeLibrary(m_hdll);
260            m_hdll = NULL;
261                    MessageBox(0, "xvid_global() failed", "Error", MB_TOPMOST);
262                    return E_FAIL;
263          }          }
264    
265          memset(&m_create, 0, sizeof(m_create));          memset(&m_create, 0, sizeof(m_create));
# Line 246  Line 269 
269          memset(&m_frame, 0, sizeof(m_frame));          memset(&m_frame, 0, sizeof(m_frame));
270          m_frame.version = XVID_VERSION;          m_frame.version = XVID_VERSION;
271    
         LoadRegistryInfo();  
   
272          USE_IYUV = false;          USE_IYUV = false;
273          USE_YV12 = false;          USE_YV12 = false;
274          USE_YUY2 = false;          USE_YUY2 = false;
# Line 285  Line 306 
306                  USE_RGB32 = true;                  USE_RGB32 = true;
307                  break;                  break;
308          }          }
309    
310            switch (g_config.aspect_ratio)
311            {
312            case 0:
313            case 1:
314                    break;
315            case 2:
316                    ar_x = 4;
317                    ar_y = 3;
318                    break;
319            case 3:
320                    ar_x = 16;
321                    ar_y = 9;
322                    break;
323            case 4:
324                    ar_x = 47;
325                    ar_y = 20;
326                    break;
327            }
328    
329            return S_OK;
330  }  }
331    
332  void CXvidDecoder::CloseLib()  void CXvidDecoder::CloseLib()
333  {  {
334          DPRINTF("Destructor");          DPRINTF("CloseLib");
335    
336          if (m_create.handle != NULL) {          if ((m_create.handle != NULL) && (xvid_decore_func != NULL))
337            {
338                  xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0);                  xvid_decore_func(m_create.handle, XVID_DEC_DESTROY, 0, 0);
339                  m_create.handle = NULL;                  m_create.handle = NULL;
340          }          }
# Line 300  Line 343 
343                  FreeLibrary(m_hdll);                  FreeLibrary(m_hdll);
344                  m_hdll = NULL;                  m_hdll = NULL;
345          }          }
346        xvid_decore_func = NULL;
347        xvid_global_func = NULL;
348  }  }
349    
350  /* destructor */  /* destructor */
351    
352  CXvidDecoder::~CXvidDecoder()  CXvidDecoder::~CXvidDecoder()
353  {  {
354        DPRINTF("Destructor");
355          CloseLib();          CloseLib();
356  }  }
357    
# Line 318  Line 364 
364          DPRINTF("CheckInputType");          DPRINTF("CheckInputType");
365          BITMAPINFOHEADER * hdr;          BITMAPINFOHEADER * hdr;
366    
367            ar_x = ar_y = 0;
368    
369          if (*mtIn->Type() != MEDIATYPE_Video)          if (*mtIn->Type() != MEDIATYPE_Video)
370          {          {
371                  DPRINTF("Error: Unknown Type");                  DPRINTF("Error: Unknown Type");
# Line 325  Line 373 
373                  return VFW_E_TYPE_NOT_ACCEPTED;                  return VFW_E_TYPE_NOT_ACCEPTED;
374          }          }
375    
376        if (m_hdll == NULL)
377        {
378                    HRESULT hr = OpenLib();
379    
380            if (FAILED(hr) || (m_hdll == NULL)) // Paranoid checks.
381                            return VFW_E_TYPE_NOT_ACCEPTED;
382        }
383    
384          if (*mtIn->FormatType() == FORMAT_VideoInfo)          if (*mtIn->FormatType() == FORMAT_VideoInfo)
385          {          {
386                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format();                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtIn->Format();
387                  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);  
388          }          }
389          else if (*mtIn->FormatType() == FORMAT_VideoInfo2)          else if (*mtIn->FormatType() == FORMAT_VideoInfo2)
390          {          {
391                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format();                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 *) mtIn->Format();
392                  hdr = &vih2->bmiHeader;                  hdr = &vih2->bmiHeader;
393                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
394                  ar_x = vih2->dwPictAspectRatioX;                  ar_x = vih2->dwPictAspectRatioX;
395                  ar_y = vih2->dwPictAspectRatioY;                  ar_y = vih2->dwPictAspectRatioY;
396                    }
397                  DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);                  DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
398          }          }
399      else if (*mtIn->FormatType() == FORMAT_MPEG2Video) {
400        MPEG2VIDEOINFO * mpgvi = (MPEG2VIDEOINFO*)mtIn->Format();
401        VIDEOINFOHEADER2 * vih2 = &mpgvi->hdr;
402                    hdr = &vih2->bmiHeader;
403                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) {
404                            ar_x = vih2->dwPictAspectRatioX;
405                            ar_y = vih2->dwPictAspectRatioY;
406                    }
407                    DPRINTF("VIDEOINFOHEADER2 AR: %d:%d", ar_x, ar_y);
408    
409        /* haali media splitter reports VOL information in the format header */
410    
411        if (mpgvi->cbSequenceHeader>0) {
412    
413          xvid_dec_stats_t stats;
414                memset(&stats, 0, sizeof(stats));
415                stats.version = XVID_VERSION;
416    
417                if (m_create.handle == NULL) {
418                        if (xvid_decore_func == NULL)
419                                return E_FAIL;
420                        if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0) {
421              DPRINTF("*** XVID_DEC_CREATE error");
422                                return E_FAIL;
423                        }
424                }
425    
426          m_frame.general = 0;
427          m_frame.bitstream = (void*)mpgvi->dwSequenceHeader;
428          m_frame.length = mpgvi->cbSequenceHeader;
429          m_frame.output.csp = XVID_CSP_NULL;
430    
431          int ret = 0;
432          if ((ret=xvid_decore_func(m_create.handle, XVID_DEC_DECODE, &m_frame, &stats)) >= 0) {
433            /* honour video dimensions reported in VOL header */
434                  if (stats.type == XVID_TYPE_VOL) {
435              hdr->biWidth = stats.data.vol.width;
436              hdr->biHeight = stats.data.vol.height;
437            }
438          }
439          if (ret == XVID_ERR_MEMORY) return E_FAIL;
440        }
441      }
442          else          else
443          {          {
444                  DPRINTF("Error: Unknown FormatType");                  DPRINTF("Error: Unknown FormatType");
# Line 361  Line 456 
456    
457          switch(hdr->biCompression)          switch(hdr->biCompression)
458          {          {
459      case FOURCC_mp4v:
460          case FOURCC_MP4V:          case FOURCC_MP4V:
461                  if (!(g_config.supported_4cc & SUPPORT_MP4V)) {                  if (!(g_config.supported_4cc & SUPPORT_MP4V)) {
462                          CloseLib();                          CloseLib();
# Line 393  Line 488 
488                  CloseLib();                  CloseLib();
489                  return VFW_E_TYPE_NOT_ACCEPTED;                  return VFW_E_TYPE_NOT_ACCEPTED;
490          }          }
491    
492            m_create.fourcc = hdr->biCompression;
493    
494          return S_OK;          return S_OK;
495  }  }
496    
# Line 421  Line 519 
519                  if (ar_x != 0 && ar_y != 0) {                  if (ar_x != 0 && ar_y != 0) {
520                          vih->dwPictAspectRatioX = ar_x;                          vih->dwPictAspectRatioX = ar_x;
521                          vih->dwPictAspectRatioY = ar_y;                          vih->dwPictAspectRatioY = ar_y;
522                            forced_ar = true;
523                  } else { // just to be safe                  } else { // just to be safe
524                          vih->dwPictAspectRatioX = m_create.width;                          vih->dwPictAspectRatioX = m_create.width;
525                          vih->dwPictAspectRatioY = abs(m_create.height);                          vih->dwPictAspectRatioY = abs(m_create.height);
526                            forced_ar = false;
527                  }                  }
   
528          } else {          } else {
529    
530                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER));                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER *) mtOut->ReallocFormatBuffer(sizeof(VIDEOINFOHEADER));
# Line 533  Line 632 
632    
633    
634  /* (internal function) change colorspace */  /* (internal function) change colorspace */
635    #define CALC_BI_STRIDE(width,bitcount)  ((((width * bitcount) + 31) & ~31) >> 3)
636    
637  HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format)  HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format)
638  {  {
639            DWORD biWidth;
640    
641          if (formattype == FORMAT_VideoInfo)          if (formattype == FORMAT_VideoInfo)
642          {          {
643                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;                  VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;
644                  m_frame.output.stride[0] = (((vih->bmiHeader.biWidth * vih->bmiHeader.biBitCount) + 31) & ~31) >> 3;                  biWidth = vih->bmiHeader.biWidth;
645                    m_frame.output.stride[0] = CALC_BI_STRIDE(vih->bmiHeader.biWidth, vih->bmiHeader.biBitCount);
646                  rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);                  rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
647          }          }
648          else if (formattype == FORMAT_VideoInfo2)          else if (formattype == FORMAT_VideoInfo2)
649          {          {
650                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;                  VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;
651                  m_frame.output.stride[0] = (((vih2->bmiHeader.biWidth * vih2->bmiHeader.biBitCount) + 31) & ~31) >> 3;                  biWidth = vih2->bmiHeader.biWidth;
652                    m_frame.output.stride[0] = CALC_BI_STRIDE(vih2->bmiHeader.biWidth, vih2->bmiHeader.biBitCount);
653                  rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);                  rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
654          }          }
655          else          else
# Line 558  Line 662 
662                  DPRINTF("IYUV");                  DPRINTF("IYUV");
663                  rgb_flip = 0;                  rgb_flip = 0;
664                  m_frame.output.csp = XVID_CSP_I420;                  m_frame.output.csp = XVID_CSP_I420;
665                  m_frame.output.stride[0] = (m_frame.output.stride[0] * 2) / 3;  /* planar format fix */                  m_frame.output.stride[0] = CALC_BI_STRIDE(biWidth, 8);  /* planar format fix */
666          }          }
667          else if (subtype == MEDIASUBTYPE_YV12)          else if (subtype == MEDIASUBTYPE_YV12)
668          {          {
669                  DPRINTF("YV12");                  DPRINTF("YV12");
670                  rgb_flip = 0;                  rgb_flip = 0;
671                  m_frame.output.csp = XVID_CSP_YV12;                  m_frame.output.csp = XVID_CSP_YV12;
672                  m_frame.output.stride[0] = (m_frame.output.stride[0] * 2) / 3;  /* planar format fix */                  m_frame.output.stride[0] = CALC_BI_STRIDE(biWidth, 8);  /* planar format fix */
673          }          }
674          else if (subtype == MEDIASUBTYPE_YUY2)          else if (subtype == MEDIASUBTYPE_YUY2)
675          {          {
# Line 689  Line 793 
793    
794          if (m_create.handle == NULL)          if (m_create.handle == NULL)
795          {          {
796                    if (xvid_decore_func == NULL)
797                            return E_FAIL;
798    
799                  if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)                  if (xvid_decore_func(0, XVID_DEC_CREATE, &m_create, 0) < 0)
800                  {                  {
801              DPRINTF("*** XVID_DEC_CREATE error");              DPRINTF("*** XVID_DEC_CREATE error");
802                          return S_FALSE;                          return E_FAIL;
803                  }                  }
804          }          }
805    
# Line 748  Line 855 
855          m_frame.output.csp &= ~XVID_CSP_VFLIP;          m_frame.output.csp &= ~XVID_CSP_VFLIP;
856          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);
857    
858        // Paranoid check.
859        if (xvid_decore_func == NULL)
860                    return E_FAIL;
861    
862    
863    
864  repeat :  repeat :
865    
866          if (pIn->IsPreroll() != S_OK)          if (pIn->IsPreroll() != S_OK)
867          {          {
868                  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);
869    
870                  if (length < 0)                  if (length == XVID_ERR_MEMORY)
871                            return E_FAIL;
872                    else if (length < 0)
873                  {                  {
874              DPRINTF("*** XVID_DEC_DECODE");              DPRINTF("*** XVID_DEC_DECODE");
875                          return S_FALSE;                          return S_FALSE;
876                    } else
877                            if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1 && forced_ar == false) {
878    
879          if (stats.type != XVID_TYPE_NOTHING) {  /* dont attempt to set vmr aspect ratio if no frame was returned by decoder */
880                            // inspired by minolta! works for VMR 7 + 9
881                            IMediaSample2 *pOut2 = NULL;
882                            AM_SAMPLE2_PROPERTIES outProp2;
883                            if (SUCCEEDED(pOut->QueryInterface(IID_IMediaSample2, (void **)&pOut2)) &&
884                                    SUCCEEDED(pOut2->GetProperties(FIELD_OFFSET(AM_SAMPLE2_PROPERTIES, tStart), (PBYTE)&outProp2)))
885                            {
886                                    CMediaType mtOut2 = m_pOutput->CurrentMediaType();
887                                    VIDEOINFOHEADER2* vihOut2 = (VIDEOINFOHEADER2*)mtOut2.Format();
888    
889                                    if (*mtOut2.FormatType() == FORMAT_VideoInfo2 &&
890                                            vihOut2->dwPictAspectRatioX != ar_x && vihOut2->dwPictAspectRatioY != ar_y)
891                                    {
892                                            vihOut2->dwPictAspectRatioX = ar_x;
893                                            vihOut2->dwPictAspectRatioY = ar_y;
894                                            pOut2->SetMediaType(&mtOut2);
895                                            m_pOutput->SetMediaType(&mtOut2);
896                                    }
897                                    pOut2->Release();
898                            }
899          }
900                  }                  }
901          }          }
902          else          else
# Line 774  Line 913 
913                  m_frame.general &= ~XVID_FILMEFFECT;                  m_frame.general &= ~XVID_FILMEFFECT;
914    
915                  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);
916                  if (length < 0)                  if (length == XVID_ERR_MEMORY)
917                            return E_FAIL;
918                    else if (length < 0)
919                  {                  {
920              DPRINTF("*** XVID_DEC_DECODE");              DPRINTF("*** XVID_DEC_DECODE");
921                          return S_FALSE;                          return S_FALSE;
# Line 799  Line 940 
940                          return S_FALSE;                          return S_FALSE;
941                  }                  }
942    
 //              pOut->SetDiscontinuity(TRUE);  
943                  pOut->SetSyncPoint(TRUE);                  pOut->SetSyncPoint(TRUE);
944    
945                    if (g_config.aspect_ratio == 0 || g_config.aspect_ratio == 1) { /* auto */
946                            int par_x, par_y;
947                            if (stats.data.vol.par == XVID_PAR_EXT) {
948                                    par_x = stats.data.vol.par_width;
949                                    par_y = stats.data.vol.par_height;
950                            } else {
951                                    par_x = PARS[stats.data.vol.par-1][0];
952                                    par_y = PARS[stats.data.vol.par-1][1];
953                            }
954    
955                            ar_x = par_x * stats.data.vol.width;
956                            ar_y = par_y * stats.data.vol.height;
957                    }
958    
959                  m_frame.bitstream = (BYTE*)m_frame.bitstream + length;                  m_frame.bitstream = (BYTE*)m_frame.bitstream + length;
960                  m_frame.length -= length;                  m_frame.length -= length;
961                  goto repeat;                  goto repeat;
Legend:
Removed from v.1437  
changed lines
  Added in v.1890
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4