filter.cpp
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:43k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * filter.c : DirectShow access module for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2002 the VideoLAN team
  5.  * $Id: 456aa87597aad24cf0c62df4e7a6c5a90c78ce21 $
  6.  *
  7.  * Author: Gildas Bazin <gbazin@videolan.org>
  8.  *
  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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_vout.h>
  31. #ifndef _MSC_VER
  32.     /* Work-around a bug in w32api-2.5 */
  33. #   define QACONTAINERFLAGS QACONTAINERFLAGS_SOMETHINGELSE
  34. #endif
  35. #include "common.h"
  36. #include "filter.h"
  37. #define DEBUG_DSHOW 1
  38. #define FILTER_NAME  L"VideoLAN Capture Filter"
  39. #define PIN_NAME     L"Capture"
  40. /*****************************************************************************
  41.  * DirectShow GUIDs.
  42.  * Easier to define them here as mingw doesn't provide them all.
  43.  *****************************************************************************/
  44. const GUID CLSID_SystemDeviceEnum = {0x62be5d10, 0x60eb, 0x11d0, {0xbd, 0x3b, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}};
  45. const GUID CLSID_VideoInputDeviceCategory = {0x860BB310,0x5D01,0x11d0,{0xBD,0x3B,0x00,0xA0,0xC9,0x11,0xCE,0x86}};
  46. const GUID CLSID_AudioInputDeviceCategory = {0x33d9a762, 0x90c8, 0x11d0, {0xbd, 0x43, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}};
  47. //const GUID IID_IPropertyBag = {0x55272A00, 0x42CB, 0x11CE, {0x81, 0x35, 0x00, 0xAA, 0x00, 0x4B, 0xB8, 0x51}};
  48. extern const GUID IID_IPropertyBag;
  49. const GUID IID_ICreateDevEnum = {0x29840822, 0x5b84, 0x11d0, {0xbd, 0x3b, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}};
  50. const GUID IID_IFilterGraph = {0x56a8689f, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  51. const GUID IID_IMediaControl = {0x56a868b1, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  52. const GUID CLSID_FilterGraph = {0xe436ebb3, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  53. //const GUID IID_IUnknown = {0x00000000, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}};
  54. extern const GUID IID_IUnknown;
  55. //const GUID IID_IPersist = {0x0000010c, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}};
  56. extern const GUID IID_IPersist;
  57. const GUID IID_IMediaFilter = {0x56a86899, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}};
  58. const GUID IID_IBaseFilter = {0x56a86895, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  59. const GUID IID_IPin = {0x56a86891, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}};
  60. const GUID IID_IMemInputPin = {0x56a8689d, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}};
  61. extern const GUID IID_IMemInputPin;
  62. const GUID IID_IEnumPins = {0x56a86892, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}};
  63. const GUID IID_IEnumMediaTypes = {0x89c31040, 0x846b, 0x11ce, {0x97,0xd3, 0x00,0xaa,0x00,0x55,0x59,0x5a}};
  64. const GUID IID_IAMBufferNegotiation = {0x56ed71a0, 0xaf5f, 0x11d0, {0xb3, 0xf0, 0x00, 0xaa, 0x00, 0x37, 0x61, 0xc5}};
  65. //const GUID IID_ISpecifyPropertyPages = {0xb196b28b, 0xbab4, 0x101a, {0xb6, 0x9c, 0x00, 0xaa, 0x00, 0x34, 0x1d, 0x07}};
  66. extern const GUID IID_ISpecifyPropertyPages;
  67. const GUID IID_IQualityControl = {0x56a868a5, 0x0ad4, 0x11ce, {0xb, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  68. const GUID CLSID_CaptureGraphBuilder2 = {0xBF87B6E1, 0x8C27, 0x11d0, {0xB3, 0xF0, 0x0, 0xAA, 0x00, 0x37, 0x61, 0xC5}};
  69. const GUID IID_IGraphBuilder = {0x56a868a9, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  70. const GUID IID_ICaptureGraphBuilder2 = {0x93E5A4E0, 0x2D50, 0x11d2, {0xAB, 0xFA, 0x00, 0xA0, 0xC9, 0xC6, 0xE3, 0x8D}};
  71. const GUID IID_IAMTVAudio = {0x83EC1C30, 0x23D1, 0x11d1, {0x99, 0xE6, 0x00, 0xA0, 0xC9, 0x56, 0x02, 0x66}};
  72. const GUID IID_IAMStreamConfig = {0xC6E13340, 0x30AC, 0x11d0, {0xA1, 0x8C, 0x00, 0xA0, 0xC9, 0x11, 0x89, 0x56}};
  73. const GUID IID_IAMCrossbar = {0xC6E13380, 0x30AC, 0x11d0, {0xA1, 0x8C, 0x00, 0xA0, 0xC9, 0x11, 0x89, 0x56}};
  74. const GUID IID_IAMTVTuner = {0x211A8766, 0x03AC, 0x11d1, {0x8D, 0x13, 0x00, 0xAA, 0x00, 0xBD, 0x83, 0x39}};
  75. const GUID IID_IKsPropertySet = {0x31EFAC30, 0x515C, 0x11d0, {0xA9, 0xAA, 0x00, 0xAA, 0x00, 0x61, 0xBE, 0x93}};
  76. /* Video Format */
  77. const GUID FORMAT_VideoInfo  = {0x05589f80, 0xc356, 0x11ce, {0xbf, 0x01, 0x00, 0xaa, 0x00, 0x55, 0x59, 0x5a}};
  78. /*
  79.  * MEDIATYPEs and MEDIASUBTYPEs
  80.  */
  81. const GUID MEDIATYPE_Video = {0x73646976, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  82. const GUID MEDIATYPE_Interleaved = {0x73766169, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  83. const GUID MEDIATYPE_Stream = {0xe436eb83, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  84. const GUID MEDIASUBTYPE_PREVIEW_VIDEO = {0x2859e1da, 0xb81f, 0x4fbd, {0x94, 0x3b, 0xe2, 0x37, 0x24, 0xa1, 0xab, 0xb3}};
  85. /* Packed RGB formats */
  86. const GUID MEDIASUBTYPE_RGB1 = {0xe436eb78, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  87. const GUID MEDIASUBTYPE_RGB4 = {0xe436eb79, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  88. const GUID MEDIASUBTYPE_RGB8 = {0xe436eb7a, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  89. const GUID MEDIASUBTYPE_RGB565 = {0xe436eb7b, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  90. const GUID MEDIASUBTYPE_RGB555 = {0xe436eb7c, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  91. const GUID MEDIASUBTYPE_RGB24 = {0xe436eb7d, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  92. const GUID MEDIASUBTYPE_RGB32 = {0xe436eb7e, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}};
  93. const GUID MEDIASUBTYPE_ARGB32 = {0x773c9ac0, 0x3274, 0x11d0, {0xb7, 0x24, 0x0, 0xaa, 0x0, 0x6c, 0x1a, 0x1}};
  94. /* Packed YUV formats */
  95. const GUID MEDIASUBTYPE_YUYV = {0x56595559, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  96. const GUID MEDIASUBTYPE_Y411 = {0x31313459, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  97. const GUID MEDIASUBTYPE_Y211 = {0x31313259, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  98. const GUID MEDIASUBTYPE_YUY2 = {0x32595559, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  99. const GUID MEDIASUBTYPE_YVYU = {0x55595659, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  100. const GUID MEDIASUBTYPE_UYVY = {0x59565955, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  101. const GUID MEDIASUBTYPE_HDYC = {0x43594448, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  102. /* Planar YUV formats */
  103. const GUID MEDIASUBTYPE_YVU9 = {0x39555659, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  104. const GUID MEDIASUBTYPE_YV12 = {0x32315659, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  105. const GUID MEDIASUBTYPE_IYUV = {0x56555949, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}}; /* identical to YV12 */
  106. const GUID MEDIASUBTYPE_Y41P = {0x50313459, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  107. const GUID MEDIASUBTYPE_I420 = {0x30323449, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  108. const GUID MEDIATYPE_Audio = {0x73647561, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  109. const GUID FORMAT_WaveFormatEx = {0x05589f81, 0xc356, 0x11ce, {0xbf, 0x01, 0x00, 0xaa, 0x00, 0x55, 0x59, 0x5a}};
  110. const GUID MEDIASUBTYPE_PCM = {0x00000001, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  111. const GUID MEDIASUBTYPE_IEEE_FLOAT = {0x00000003, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  112. /* DV formats */
  113. const GUID MEDIASUBTYPE_dvsd = {0x64737664, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  114. const GUID MEDIASUBTYPE_dvhd = {0x64687664, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  115. const GUID MEDIASUBTYPE_dvsl = {0x6c737664, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  116. /* MPEG2 formats */
  117. const GUID MEDIASUBTYPE_MPEG2_VIDEO = {0xe06d8026, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}};
  118. const GUID MEDIASUBTYPE_MPEG2_PROGRAM = {0xe06d8022, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}};
  119. const GUID MEDIASUBTYPE_MPEG2_TRANSPORT = {0xe06d8023, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}};
  120. const GUID FORMAT_MPEG2Video = {0xe06d80e3, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}};
  121. /* MJPG format */
  122. const GUID MEDIASUBTYPE_MJPG = {0x47504A4D, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  123. /* DivX formats */
  124. const GUID MEDIASUBTYPE_DIVX = {0x58564944, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}};
  125. /* Analog Video */
  126. const GUID FORMAT_AnalogVideo = {0x482dde0, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  127. const GUID MEDIATYPE_AnalogVideo = {0x482dde1, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xab, 0x0, 0x6e, 0xcb, 0x65}};
  128. const GUID MEDIASUBTYPE_AnalogVideo_NTSC_M = {0x482dde2, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  129. const GUID MEDIASUBTYPE_AnalogVideo_PAL_B = {0x482dde5, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  130. const GUID MEDIASUBTYPE_AnalogVideo_PAL_D = {0x482dde6, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  131. const GUID MEDIASUBTYPE_AnalogVideo_PAL_G = {0x482dde7, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  132. const GUID MEDIASUBTYPE_AnalogVideo_PAL_H = {0x482dde8, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  133. const GUID MEDIASUBTYPE_AnalogVideo_PAL_I = {0x482dde9, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  134. const GUID MEDIASUBTYPE_AnalogVideo_PAL_M = {0x482ddea, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  135. const GUID MEDIASUBTYPE_AnalogVideo_PAL_N = {0x482ddeb, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  136. const GUID MEDIASUBTYPE_AnalogVideo_PAL_N_COMBO = {0x482ddec, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  137. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_B = {0x482ddf0, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  138. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_D = {0x482ddf1, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  139. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_G = {0x482ddf2, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  140. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_H = {0x482ddf3, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  141. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_K = {0x482ddf4, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  142. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_K1 = {0x482ddf5, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  143. const GUID MEDIASUBTYPE_AnalogVideo_SECAM_L = {0x482ddf6, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}};
  144. const GUID AMPROPSETID_Pin= {0x9b00f101, 0x1567, 0x11d1, {0xb3, 0xf1, 0x0, 0xaa, 0x0, 0x37, 0x61, 0xc5}};
  145. const GUID PIN_CATEGORY_ANALOGVIDEOIN= {0xfb6c4283, 0x0353, 0x11d1, {0x90, 0x5f, 0x00, 0x00, 0xc0, 0xcc, 0x16, 0xba}};
  146. const GUID PIN_CATEGORY_CAPTURE= {0xfb6c4281, 0x0353, 0x11d1, {0x90, 0x5f, 0x0, 0x0, 0xc0, 0xcc, 0x16, 0xba}};
  147. const GUID LOOK_UPSTREAM_ONLY= {0xac798be0, 0x98e3, 0x11d1, {0xb3, 0xf1, 0x0, 0xaa, 0x0, 0x37, 0x61, 0xc}};
  148. //const GUID GUID_NULL = {0x0000, 0x0000, 0x0000, {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}};
  149. extern const GUID GUID_NULL;
  150. void WINAPI FreeMediaType( AM_MEDIA_TYPE& mt )
  151. {
  152.     if( mt.cbFormat != 0 )
  153.     {
  154.         CoTaskMemFree( (PVOID)mt.pbFormat );
  155.         mt.cbFormat = 0;
  156.         mt.pbFormat = NULL;
  157.     }
  158.     if( mt.pUnk != NULL )
  159.     {
  160.         mt.pUnk->Release();
  161.         mt.pUnk = NULL;
  162.     }
  163. }
  164. HRESULT WINAPI CopyMediaType( AM_MEDIA_TYPE *pmtTarget,
  165.                               const AM_MEDIA_TYPE *pmtSource )
  166. {
  167.     *pmtTarget = *pmtSource;
  168.     if( !pmtSource || !pmtTarget ) return S_FALSE;
  169.     if( pmtSource->cbFormat && pmtSource->pbFormat )
  170.     {
  171.         pmtTarget->pbFormat = (PBYTE)CoTaskMemAlloc( pmtSource->cbFormat );
  172.         if( pmtTarget->pbFormat == NULL )
  173.         {
  174.             pmtTarget->cbFormat = 0;
  175.             return E_OUTOFMEMORY;
  176.         }
  177.         else
  178.         {
  179.             CopyMemory( (PVOID)pmtTarget->pbFormat, (PVOID)pmtSource->pbFormat,
  180.                         pmtTarget->cbFormat );
  181.         }
  182.     }
  183.     if( pmtTarget->pUnk != NULL )
  184.     {
  185.         pmtTarget->pUnk->AddRef();
  186.     }
  187.     return S_OK;
  188. }
  189. int GetFourCCFromMediaType( const AM_MEDIA_TYPE &media_type )
  190. {
  191.     int i_fourcc = 0;
  192.     if( media_type.majortype == MEDIATYPE_Video )
  193.     {
  194.         /* currently only support this type of video info format */
  195.         if( 1 /* media_type.formattype == FORMAT_VideoInfo */ )
  196.         {
  197.             /* Packed RGB formats */
  198.             if( media_type.subtype == MEDIASUBTYPE_RGB1 )
  199.                i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '1' );
  200.             else if( media_type.subtype == MEDIASUBTYPE_RGB4 )
  201.                i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '4' );
  202.             else if( media_type.subtype == MEDIASUBTYPE_RGB8 )
  203.                i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' );
  204.             else if( media_type.subtype == MEDIASUBTYPE_RGB555 )
  205.                i_fourcc = VLC_FOURCC( 'R', 'V', '1', '5' );
  206.             else if( media_type.subtype == MEDIASUBTYPE_RGB565 )
  207.                i_fourcc = VLC_FOURCC( 'R', 'V', '1', '6' );
  208.             else if( media_type.subtype == MEDIASUBTYPE_RGB24 )
  209.                i_fourcc = VLC_FOURCC( 'R', 'V', '2', '4' );
  210.             else if( media_type.subtype == MEDIASUBTYPE_RGB32 )
  211.                i_fourcc = VLC_FOURCC( 'R', 'V', '3', '2' );
  212.             else if( media_type.subtype == MEDIASUBTYPE_ARGB32 )
  213.                i_fourcc = VLC_FOURCC( 'R', 'G', 'B', 'A' );
  214.             /* Planar YUV formats */
  215.             else if( media_type.subtype == MEDIASUBTYPE_I420 )
  216.                i_fourcc = VLC_FOURCC( 'I', '4', '2', '0' );
  217.             else if( media_type.subtype == MEDIASUBTYPE_Y41P )
  218.                i_fourcc = VLC_FOURCC( 'I', '4', '1', '1' );
  219.             else if( media_type.subtype == MEDIASUBTYPE_YV12 )
  220.                i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' );
  221.             else if( media_type.subtype == MEDIASUBTYPE_IYUV )
  222.                i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' );
  223.             else if( media_type.subtype == MEDIASUBTYPE_YVU9 )
  224.                i_fourcc = VLC_FOURCC( 'Y', 'V', 'U', '9' );
  225.             /* Packed YUV formats */
  226.             else if( media_type.subtype == MEDIASUBTYPE_YVYU )
  227.                i_fourcc = VLC_FOURCC( 'Y', 'V', 'Y', 'U' );
  228.             else if( media_type.subtype == MEDIASUBTYPE_YUYV )
  229.                i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', '2' );
  230.             else if( media_type.subtype == MEDIASUBTYPE_Y411 )
  231.                i_fourcc = VLC_FOURCC( 'I', '4', '1', 'N' );
  232.             else if( media_type.subtype == MEDIASUBTYPE_Y211 )
  233.                i_fourcc = VLC_FOURCC( 'Y', '2', '1', '1' );
  234.             else if( media_type.subtype == MEDIASUBTYPE_YUY2 )
  235.                i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', '2' );
  236.             else if( media_type.subtype == MEDIASUBTYPE_UYVY )
  237.                i_fourcc = VLC_FOURCC( 'U', 'Y', 'V', 'Y' );
  238.             /* HDYC uses UYVY sample positions but Rec709 colourimetry */
  239.             /* FIXME: When VLC understands colourspace, something will need
  240.              * to be added / changed here. Until then, just make it behave
  241.              * like UYVY */
  242.             else if( media_type.subtype == MEDIASUBTYPE_HDYC )
  243.                 i_fourcc = VLC_FOURCC( 'U', 'Y', 'V', 'Y');
  244.             /* MPEG2 video elementary stream */
  245.             else if( media_type.subtype == MEDIASUBTYPE_MPEG2_VIDEO )
  246.                i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'v' );
  247.         /* DivX video */
  248.             else if( media_type.subtype == MEDIASUBTYPE_DIVX )
  249.                i_fourcc = VLC_FOURCC( 'D', 'I', 'V', 'X' );
  250.             /* DV formats */
  251.             else if( media_type.subtype == MEDIASUBTYPE_dvsl )
  252.                i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'l' );
  253.             else if( media_type.subtype == MEDIASUBTYPE_dvsd )
  254.                i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'd' );
  255.             else if( media_type.subtype == MEDIASUBTYPE_dvhd )
  256.                i_fourcc = VLC_FOURCC( 'd', 'v', 'h', 'd' );
  257.             /* MJPEG format */
  258.             else if( media_type.subtype == MEDIASUBTYPE_MJPG )
  259.                 i_fourcc = VLC_FOURCC( 'M', 'J', 'P', 'G' );
  260.         }
  261.     }
  262.     else if( media_type.majortype == MEDIATYPE_Audio )
  263.     {
  264.         /* currently only support this type of audio info format */
  265.         if( media_type.formattype == FORMAT_WaveFormatEx )
  266.         {
  267.             if( media_type.subtype == MEDIASUBTYPE_PCM )
  268.                 i_fourcc = VLC_FOURCC( 'a', 'r', 'a', 'w' );
  269.             else if( media_type.subtype == MEDIASUBTYPE_IEEE_FLOAT )
  270.                 i_fourcc = VLC_FOURCC( 'f', 'l', '3', '2' );
  271.         }
  272.     }
  273.     else if( media_type.majortype == MEDIATYPE_Stream )
  274.     {
  275.         if( media_type.subtype == MEDIASUBTYPE_MPEG2_PROGRAM )
  276.             i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'p' );
  277.         else if( media_type.subtype == MEDIASUBTYPE_MPEG2_TRANSPORT )
  278.             i_fourcc = VLC_FOURCC( 'm', 'p', '2', 't' );
  279.     }
  280.     return i_fourcc;
  281. }
  282. /****************************************************************************
  283.  * Implementation of our dummy directshow filter pin class
  284.  ****************************************************************************/
  285. CapturePin::CapturePin( vlc_object_t *_p_input, access_sys_t *_p_sys,
  286.                         CaptureFilter *_p_filter,
  287.                         AM_MEDIA_TYPE *mt, size_t mt_count )
  288.   : p_input( _p_input ), p_sys( _p_sys ), p_filter( _p_filter ),
  289.     p_connected_pin( NULL ),  media_types(mt), media_type_count(mt_count),
  290.     i_ref( 1 )
  291. {
  292.     cx_media_type.majortype = mt[0].majortype;
  293.     cx_media_type.subtype   = GUID_NULL;
  294.     cx_media_type.pbFormat  = NULL;
  295.     cx_media_type.cbFormat  = 0;
  296.     cx_media_type.pUnk      = NULL;
  297. }
  298. CapturePin::~CapturePin()
  299. {
  300. #ifdef DEBUG_DSHOW
  301.     msg_Dbg( p_input, "CapturePin::~CapturePin" );
  302. #endif
  303.     for( size_t c=0; c<media_type_count; c++ )
  304.     {
  305.         FreeMediaType(media_types[c]);
  306.     }
  307.     FreeMediaType(cx_media_type);
  308. }
  309. /**
  310.  * Returns the complete queue of samples that have been received so far.
  311.  * Lock the p_sys->lock before calling this function.
  312.  * @param samples_queue [out] Empty queue that will get all elements from
  313.  * the pin queue.
  314.  * @return S_OK if a sample was available, S_FALSE if no sample was
  315.  * available
  316.  */
  317. HRESULT CapturePin::CustomGetSamples( deque<VLCMediaSample> &external_queue )
  318. {
  319. #if 0 //def DEBUG_DSHOW
  320.     msg_Dbg( p_input, "CapturePin::CustomGetSamples: %d samples in the queue", samples_queue.size());
  321. #endif
  322.     if( !samples_queue.empty() )
  323.     {
  324.         external_queue.swap(samples_queue);
  325.         return S_OK;
  326.     }
  327.     return S_FALSE;
  328. }
  329. /**
  330.  * Returns a sample from its sample queue. Proper locking must be done prior
  331.  * to this call. Current dshow code protects the access to any sample queue
  332.  * (audio and video) with the p_sys->lock
  333.  * @param vlc_sample [out] Address of a sample if sucessfull. Undefined
  334.  * otherwise.
  335.  * @return S_OK if a sample was available, S_FALSE if no sample was
  336.  * available
  337.  */
  338. HRESULT CapturePin::CustomGetSample( VLCMediaSample *vlc_sample )
  339. {
  340. #if 0 //def DEBUG_DSHOW
  341.     msg_Dbg( p_input, "CapturePin::CustomGetSample" );
  342. #endif
  343.     if( samples_queue.size() )
  344.     {
  345.         *vlc_sample = samples_queue.back();
  346.         samples_queue.pop_back();
  347.         return S_OK;
  348.     }
  349.     return S_FALSE;
  350. }
  351. AM_MEDIA_TYPE &CapturePin::CustomGetMediaType()
  352. {
  353.     return cx_media_type;
  354. }
  355. /* IUnknown methods */
  356. STDMETHODIMP CapturePin::QueryInterface(REFIID riid, void **ppv)
  357. {
  358. #ifdef DEBUG_DSHOW_L1
  359.     msg_Dbg( p_input, "CapturePin::QueryInterface" );
  360. #endif
  361.     if( riid == IID_IUnknown ||
  362.         riid == IID_IPin )
  363.     {
  364.         AddRef();
  365.         *ppv = (IPin *)this;
  366.         return NOERROR;
  367.     }
  368.     if( riid == IID_IMemInputPin )
  369.     {
  370.         AddRef();
  371.         *ppv = (IMemInputPin *)this;
  372.         return NOERROR;
  373.     }
  374.     else
  375.     {
  376. #ifdef DEBUG_DSHOW_L1
  377.         msg_Dbg( p_input, "CapturePin::QueryInterface() failed for: "
  378.                  "%04X-%02X-%02X-%02X%02X%02X%02X%02X%02X%02X%02X",
  379.                  (int)riid.Data1, (int)riid.Data2, (int)riid.Data3,
  380.                  static_cast<int>(riid.Data4[0]), (int)riid.Data4[1],
  381.                  (int)riid.Data4[2], (int)riid.Data4[3],
  382.                  (int)riid.Data4[4], (int)riid.Data4[5],
  383.                  (int)riid.Data4[6], (int)riid.Data4[7] );
  384. #endif
  385.         *ppv = NULL;
  386.         return E_NOINTERFACE;
  387.     }
  388. }
  389. STDMETHODIMP_(ULONG) CapturePin::AddRef()
  390. {
  391. #ifdef DEBUG_DSHOW_L1
  392.     msg_Dbg( p_input, "CapturePin::AddRef (ref: %i)", i_ref );
  393. #endif
  394.     return i_ref++;
  395. };
  396. STDMETHODIMP_(ULONG) CapturePin::Release()
  397. {
  398. #ifdef DEBUG_DSHOW_L1
  399.     msg_Dbg( p_input, "CapturePin::Release (ref: %i)", i_ref );
  400. #endif
  401.     if( !InterlockedDecrement(&i_ref) ) delete this;
  402.     return 0;
  403. };
  404. /* IPin methods */
  405. STDMETHODIMP CapturePin::Connect( IPin * pReceivePin,
  406.                                   const AM_MEDIA_TYPE *pmt )
  407. {
  408.     if( State_Running == p_filter->state )
  409.     {
  410.         msg_Dbg( p_input, "CapturePin::Connect [not stopped]" );
  411.         return VFW_E_NOT_STOPPED;
  412.     }
  413.     if( p_connected_pin )
  414.     {
  415.         msg_Dbg( p_input, "CapturePin::Connect [already connected]" );
  416.         return VFW_E_ALREADY_CONNECTED;
  417.     }
  418.     if( !pmt ) return S_OK;
  419.  
  420.     if( GUID_NULL != pmt->majortype &&
  421.         media_types[0].majortype != pmt->majortype )
  422.     {
  423.         msg_Dbg( p_input, "CapturePin::Connect [media major type mismatch]" );
  424.         return S_FALSE;
  425.     }
  426.     if( GUID_NULL != pmt->subtype && !GetFourCCFromMediaType(*pmt) )
  427.     {
  428.         msg_Dbg( p_input, "CapturePin::Connect [media subtype type "
  429.                  "not supported]" );
  430.         return S_FALSE;
  431.     }
  432.     if( pmt->pbFormat && pmt->majortype == MEDIATYPE_Video  )
  433.     {
  434.         if( !((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight ||
  435.             !((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth )
  436.         {
  437.             msg_Dbg( p_input, "CapturePin::Connect "
  438.                      "[video width/height == 0 ]" );
  439.             return S_FALSE;
  440.         }
  441.     }
  442.     msg_Dbg( p_input, "CapturePin::Connect [OK]" );
  443.     return S_OK;
  444. }
  445. STDMETHODIMP CapturePin::ReceiveConnection( IPin * pConnector,
  446.                                             const AM_MEDIA_TYPE *pmt )
  447. {
  448.     if( State_Stopped != p_filter->state )
  449.     {
  450.         msg_Dbg( p_input, "CapturePin::ReceiveConnection [not stopped]" );
  451.         return VFW_E_NOT_STOPPED;
  452.     }
  453.     if( !pConnector || !pmt )
  454.     {
  455.         msg_Dbg( p_input, "CapturePin::ReceiveConnection [null pointer]" );
  456.         return E_POINTER;
  457.     }
  458.     if( p_connected_pin )
  459.     {
  460.         msg_Dbg( p_input, "CapturePin::ReceiveConnection [already connected]");
  461.         return VFW_E_ALREADY_CONNECTED;
  462.     }
  463.     if( S_OK != QueryAccept(pmt) )
  464.     {
  465.         msg_Dbg( p_input, "CapturePin::ReceiveConnection "
  466.                  "[media type not accepted]" );
  467.         return VFW_E_TYPE_NOT_ACCEPTED;
  468.     }
  469.     msg_Dbg( p_input, "CapturePin::ReceiveConnection [OK]" );
  470.     p_connected_pin = pConnector;
  471.     p_connected_pin->AddRef();
  472.     FreeMediaType( cx_media_type );
  473.     return CopyMediaType( &cx_media_type, pmt );
  474. }
  475. STDMETHODIMP CapturePin::Disconnect()
  476. {
  477.     if( ! p_connected_pin )
  478.     {
  479.         msg_Dbg( p_input, "CapturePin::Disconnect [not connected]" );
  480.         return S_FALSE;
  481.     }
  482.     msg_Dbg( p_input, "CapturePin::Disconnect [OK]" );
  483.     /* samples_queue was already flushed in EndFlush() */
  484.     p_connected_pin->Release();
  485.     p_connected_pin = NULL;
  486.     //FreeMediaType( cx_media_type );
  487.     //cx_media_type.subtype = GUID_NULL;
  488.     return S_OK;
  489. }
  490. STDMETHODIMP CapturePin::ConnectedTo( IPin **pPin )
  491. {
  492.     if( !p_connected_pin )
  493.     {
  494.         msg_Dbg( p_input, "CapturePin::ConnectedTo [not connected]" );
  495.         return VFW_E_NOT_CONNECTED;
  496.     }
  497.     p_connected_pin->AddRef();
  498.     *pPin = p_connected_pin;
  499.     msg_Dbg( p_input, "CapturePin::ConnectedTo [OK]" );
  500.     return S_OK;
  501. }
  502. STDMETHODIMP CapturePin::ConnectionMediaType( AM_MEDIA_TYPE *pmt )
  503. {
  504.     if( !p_connected_pin )
  505.     {
  506.         msg_Dbg( p_input, "CapturePin::ConnectionMediaType [not connected]" );
  507.         return VFW_E_NOT_CONNECTED;
  508.     }
  509.     return CopyMediaType( pmt, &cx_media_type );
  510. }
  511. STDMETHODIMP CapturePin::QueryPinInfo( PIN_INFO * pInfo )
  512. {
  513. #ifdef DEBUG_DSHOW
  514.     msg_Dbg( p_input, "CapturePin::QueryPinInfo" );
  515. #endif
  516.     pInfo->pFilter = p_filter;
  517.     if( p_filter ) p_filter->AddRef();
  518.     memcpy(pInfo->achName, PIN_NAME, sizeof(PIN_NAME));
  519.     pInfo->dir = PINDIR_INPUT;
  520.     return NOERROR;
  521. }
  522. STDMETHODIMP CapturePin::QueryDirection( PIN_DIRECTION * pPinDir )
  523. {
  524. #ifdef DEBUG_DSHOW
  525.     msg_Dbg( p_input, "CapturePin::QueryDirection" );
  526. #endif
  527.     *pPinDir = PINDIR_INPUT;
  528.     return NOERROR;
  529. }
  530. STDMETHODIMP CapturePin::QueryId( LPWSTR * Id )
  531. {
  532. #ifdef DEBUG_DSHOW
  533.     msg_Dbg( p_input, "CapturePin::QueryId" );
  534. #endif
  535.     *Id = L"VideoLAN Capture Pin";
  536.     return S_OK;
  537. }
  538. STDMETHODIMP CapturePin::QueryAccept( const AM_MEDIA_TYPE *pmt )
  539. {
  540.     if( State_Stopped != p_filter->state )
  541.     {
  542.         msg_Dbg( p_input, "CapturePin::QueryAccept [not stopped]" );
  543.         return S_FALSE;
  544.     }
  545.     if( media_types[0].majortype != pmt->majortype )
  546.     {
  547.         msg_Dbg( p_input, "CapturePin::QueryAccept [media type mismatch]" );
  548.         return S_FALSE;
  549.     }
  550.     int i_fourcc = GetFourCCFromMediaType(*pmt);
  551.     if( !i_fourcc )
  552.     {
  553.         msg_Dbg( p_input, "CapturePin::QueryAccept "
  554.                  "[media type not supported]" );
  555.         return S_FALSE;
  556.     }
  557.     if( pmt->majortype == MEDIATYPE_Video )
  558.     {
  559.         if( pmt->pbFormat &&
  560.             ( (((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight == 0) ||
  561.               (((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth == 0) ) )
  562.         {
  563.             msg_Dbg( p_input, "CapturePin::QueryAccept [video size wxh == 0]");
  564.             return S_FALSE;
  565.         }
  566.         msg_Dbg( p_input, "CapturePin::QueryAccept [OK] "
  567.                  "(width=%ld, height=%ld, chroma=%4.4s, fps=%f)",
  568.                  ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth,
  569.                  ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight,
  570.                  (char *)&i_fourcc,
  571.          10000000.0f/((float)((VIDEOINFOHEADER *)pmt->pbFormat)->AvgTimePerFrame) );
  572.     }
  573.     else if( pmt->majortype == MEDIATYPE_Audio )
  574.     {
  575.         msg_Dbg( p_input, "CapturePin::QueryAccept [OK] (channels=%d, "
  576.                  "samples/sec=%lu, bits/samples=%d, format=%4.4s)",
  577.                  ((WAVEFORMATEX *)pmt->pbFormat)->nChannels,
  578.                  ((WAVEFORMATEX *)pmt->pbFormat)->nSamplesPerSec,
  579.                  ((WAVEFORMATEX *)pmt->pbFormat)->wBitsPerSample,
  580.                  (char *)&i_fourcc );
  581.     }
  582.     else
  583.     {
  584.         msg_Dbg( p_input, "CapturePin::QueryAccept [OK] (stream format=%4.4s)",
  585.                  (char *)&i_fourcc );
  586.     }
  587.     if( p_connected_pin )
  588.     {
  589.         FreeMediaType( cx_media_type );
  590.         CopyMediaType( &cx_media_type, pmt );
  591.     }
  592.     return S_OK;
  593. }
  594. STDMETHODIMP CapturePin::EnumMediaTypes( IEnumMediaTypes **ppEnum )
  595. {
  596. #ifdef DEBUG_DSHOW_L1
  597.     msg_Dbg( p_input, "CapturePin::EnumMediaTypes" );
  598. #endif
  599.     *ppEnum = new CaptureEnumMediaTypes( p_input, this, NULL );
  600.     if( *ppEnum == NULL ) return E_OUTOFMEMORY;
  601.     return NOERROR;
  602. }
  603. STDMETHODIMP CapturePin::QueryInternalConnections( IPin* *apPin, ULONG *nPin )
  604. {
  605. #ifdef DEBUG_DSHOW_L1
  606.     msg_Dbg( p_input, "CapturePin::QueryInternalConnections" );
  607. #endif
  608.     return E_NOTIMPL;
  609. }
  610. STDMETHODIMP CapturePin::EndOfStream( void )
  611. {
  612. #ifdef DEBUG_DSHOW
  613.     msg_Dbg( p_input, "CapturePin::EndOfStream" );
  614. #endif
  615.     return S_OK;
  616. }
  617. STDMETHODIMP CapturePin::BeginFlush( void )
  618. {
  619. #ifdef DEBUG_DSHOW
  620.     msg_Dbg( p_input, "CapturePin::BeginFlush" );
  621. #endif
  622.     return S_OK;
  623. }
  624. STDMETHODIMP CapturePin::EndFlush( void )
  625. {
  626. #ifdef DEBUG_DSHOW
  627.     msg_Dbg( p_input, "CapturePin::EndFlush" );
  628. #endif
  629.     VLCMediaSample vlc_sample;
  630.     vlc_mutex_lock( &p_sys->lock );
  631.     while( samples_queue.size() )
  632.     {
  633.         vlc_sample = samples_queue.back();
  634.         samples_queue.pop_back();
  635.         vlc_sample.p_sample->Release();
  636.     }
  637.     vlc_mutex_unlock( &p_sys->lock );
  638.     return S_OK;
  639. }
  640. STDMETHODIMP CapturePin::NewSegment( REFERENCE_TIME tStart,
  641.                                      REFERENCE_TIME tStop,
  642.                                      double dRate )
  643. {
  644. #ifdef DEBUG_DSHOW
  645.     msg_Dbg( p_input, "CapturePin::NewSegment" );
  646. #endif
  647.     return S_OK;
  648. }
  649. /* IMemInputPin methods */
  650. STDMETHODIMP CapturePin::GetAllocator( IMemAllocator **ppAllocator )
  651. {
  652. #ifdef DEBUG_DSHOW
  653.     msg_Dbg( p_input, "CapturePin::GetAllocator" );
  654. #endif
  655.     return VFW_E_NO_ALLOCATOR;
  656. }
  657. STDMETHODIMP CapturePin::NotifyAllocator( IMemAllocator *pAllocator,
  658.                                           BOOL bReadOnly )
  659. {
  660. #ifdef DEBUG_DSHOW
  661.     msg_Dbg( p_input, "CapturePin::NotifyAllocator" );
  662. #endif
  663.     return S_OK;
  664. }
  665. STDMETHODIMP CapturePin::GetAllocatorRequirements( ALLOCATOR_PROPERTIES *pProps )
  666. {
  667. #ifdef DEBUG_DSHOW
  668.     msg_Dbg( p_input, "CapturePin::GetAllocatorRequirements" );
  669. #endif
  670.     return E_NOTIMPL;
  671. }
  672. STDMETHODIMP CapturePin::Receive( IMediaSample *pSample )
  673. {
  674. #if 0 //def DEBUG_DSHOW
  675.     msg_Dbg( p_input, "CapturePin::Receive" );
  676. #endif
  677.     pSample->AddRef();
  678.     mtime_t i_timestamp = mdate() * 10;
  679.     VLCMediaSample vlc_sample = {pSample, i_timestamp};
  680.     vlc_mutex_lock( &p_sys->lock );
  681.     samples_queue.push_front( vlc_sample );
  682.     /* Make sure we don't cache too many samples */
  683.     if( samples_queue.size() > 10 )
  684.     {
  685.         vlc_sample = samples_queue.back();
  686.         samples_queue.pop_back();
  687.         msg_Dbg( p_input, "CapturePin::Receive trashing late input sample" );
  688.         vlc_sample.p_sample->Release();
  689.     }
  690.     vlc_cond_signal( &p_sys->wait );
  691.     vlc_mutex_unlock( &p_sys->lock );
  692.     return S_OK;
  693. }
  694. STDMETHODIMP CapturePin::ReceiveMultiple( IMediaSample **pSamples,
  695.                                           long nSamples,
  696.                                           long *nSamplesProcessed )
  697. {
  698.     HRESULT hr = S_OK;
  699.     *nSamplesProcessed = 0;
  700.     while( nSamples-- > 0 )
  701.     {
  702.          hr = Receive( pSamples[*nSamplesProcessed] );
  703.          if( hr != S_OK ) break;
  704.          (*nSamplesProcessed)++;
  705.     }
  706.     return hr;
  707. }
  708. STDMETHODIMP CapturePin::ReceiveCanBlock( void )
  709. {
  710. #ifdef DEBUG_DSHOW
  711.     msg_Dbg( p_input, "CapturePin::ReceiveCanBlock" );
  712. #endif
  713.     return S_FALSE; /* Thou shalt not block */
  714. }
  715. /****************************************************************************
  716.  * Implementation of our dummy directshow filter class
  717.  ****************************************************************************/
  718. CaptureFilter::CaptureFilter( vlc_object_t *_p_input, access_sys_t *p_sys,
  719.                               AM_MEDIA_TYPE *mt, size_t mt_count )
  720.   : p_input( _p_input ),
  721.     p_pin( new CapturePin( _p_input, p_sys, this, mt, mt_count ) ),
  722.     state( State_Stopped ), i_ref( 1 )
  723. {
  724. }
  725. CaptureFilter::~CaptureFilter()
  726. {
  727. #ifdef DEBUG_DSHOW
  728.     msg_Dbg( p_input, "CaptureFilter::~CaptureFilter" );
  729. #endif
  730.     p_pin->Release();
  731. }
  732. /* IUnknown methods */
  733. STDMETHODIMP CaptureFilter::QueryInterface( REFIID riid, void **ppv )
  734. {
  735. #ifdef DEBUG_DSHOW_L1
  736.     msg_Dbg( p_input, "CaptureFilter::QueryInterface" );
  737. #endif
  738.     if( riid == IID_IUnknown )
  739.     {
  740.         AddRef();
  741.         *ppv = (IUnknown *)this;
  742.         return NOERROR;
  743.     }
  744.     if( riid == IID_IPersist )
  745.     {
  746.         AddRef();
  747.         *ppv = (IPersist *)this;
  748.         return NOERROR;
  749.     }
  750.     if( riid == IID_IMediaFilter )
  751.     {
  752.         AddRef();
  753.         *ppv = (IMediaFilter *)this;
  754.         return NOERROR;
  755.     }
  756.     if( riid == IID_IBaseFilter )
  757.     {
  758.         AddRef();
  759.         *ppv = (IBaseFilter *)this;
  760.         return NOERROR;
  761.     }
  762.     else
  763.     {
  764. #ifdef DEBUG_DSHOW_L1
  765.         msg_Dbg( p_input, "CaptureFilter::QueryInterface() failed for: "
  766.                  "%04X-%02X-%02X-%02X%02X%02X%02X%02X%02X%02X%02X",
  767.                  (int)riid.Data1, (int)riid.Data2, (int)riid.Data3,
  768.                  static_cast<int>(riid.Data4[0]), (int)riid.Data4[1],
  769.                  (int)riid.Data4[2], (int)riid.Data4[3],
  770.                  (int)riid.Data4[4], (int)riid.Data4[5],
  771.                  (int)riid.Data4[6], (int)riid.Data4[7] );
  772. #endif
  773.         *ppv = NULL;
  774.         return E_NOINTERFACE;
  775.     }
  776. };
  777. STDMETHODIMP_(ULONG) CaptureFilter::AddRef()
  778. {
  779. #ifdef DEBUG_DSHOW_L1
  780.     msg_Dbg( p_input, "CaptureFilter::AddRef (ref: %i)", i_ref );
  781. #endif
  782.     return i_ref++;
  783. };
  784. STDMETHODIMP_(ULONG) CaptureFilter::Release()
  785. {
  786. #ifdef DEBUG_DSHOW_L1
  787.     msg_Dbg( p_input, "CaptureFilter::Release (ref: %i)", i_ref );
  788. #endif
  789.     if( !InterlockedDecrement(&i_ref) ) delete this;
  790.     return 0;
  791. };
  792. /* IPersist method */
  793. STDMETHODIMP CaptureFilter::GetClassID(CLSID *pClsID)
  794. {
  795. #ifdef DEBUG_DSHOW
  796.     msg_Dbg( p_input, "CaptureFilter::GetClassID" );
  797. #endif
  798.     return E_NOTIMPL;
  799. };
  800. /* IMediaFilter methods */
  801. STDMETHODIMP CaptureFilter::GetState(DWORD dwMSecs, FILTER_STATE *State)
  802. {
  803. #ifdef DEBUG_DSHOW
  804.     msg_Dbg( p_input, "CaptureFilter::GetState %i", state );
  805. #endif
  806.     *State = state;
  807.     return S_OK;
  808. };
  809. STDMETHODIMP CaptureFilter::SetSyncSource(IReferenceClock *pClock)
  810. {
  811. #ifdef DEBUG_DSHOW
  812.     msg_Dbg( p_input, "CaptureFilter::SetSyncSource" );
  813. #endif
  814.     return S_OK;
  815. };
  816. STDMETHODIMP CaptureFilter::GetSyncSource(IReferenceClock **pClock)
  817. {
  818. #ifdef DEBUG_DSHOW
  819.     msg_Dbg( p_input, "CaptureFilter::GetSyncSource" );
  820. #endif
  821.     *pClock = NULL;
  822.     return NOERROR;
  823. };
  824. STDMETHODIMP CaptureFilter::Stop()
  825. {
  826. #ifdef DEBUG_DSHOW
  827.     msg_Dbg( p_input, "CaptureFilter::Stop" );
  828. #endif
  829.     p_pin->EndFlush();
  830.     state = State_Stopped;
  831.     return S_OK;
  832. };
  833. STDMETHODIMP CaptureFilter::Pause()
  834. {
  835. #ifdef DEBUG_DSHOW
  836.     msg_Dbg( p_input, "CaptureFilter::Pause" );
  837. #endif
  838.     state = State_Paused;
  839.     return S_OK;
  840. };
  841. STDMETHODIMP CaptureFilter::Run(REFERENCE_TIME tStart)
  842. {
  843. #ifdef DEBUG_DSHOW
  844.     msg_Dbg( p_input, "CaptureFilter::Run" );
  845. #endif
  846.     state = State_Running;
  847.     return S_OK;
  848. };
  849. /* IBaseFilter methods */
  850. STDMETHODIMP CaptureFilter::EnumPins( IEnumPins ** ppEnum )
  851. {
  852. #ifdef DEBUG_DSHOW
  853.     msg_Dbg( p_input, "CaptureFilter::EnumPins" );
  854. #endif
  855.     /* Create a new ref counted enumerator */
  856.     *ppEnum = new CaptureEnumPins( p_input, this, NULL );
  857.     return *ppEnum == NULL ? E_OUTOFMEMORY : NOERROR;
  858. };
  859. STDMETHODIMP CaptureFilter::FindPin( LPCWSTR Id, IPin ** ppPin )
  860. {
  861. #ifdef DEBUG_DSHOW
  862.     msg_Dbg( p_input, "CaptureFilter::FindPin" );
  863. #endif
  864.     return E_NOTIMPL;
  865. };
  866. STDMETHODIMP CaptureFilter::QueryFilterInfo( FILTER_INFO * pInfo )
  867. {
  868. #ifdef DEBUG_DSHOW
  869.     msg_Dbg( p_input, "CaptureFilter::QueryFilterInfo" );
  870. #endif
  871.     memcpy(pInfo->achName, FILTER_NAME, sizeof(FILTER_NAME));
  872.     pInfo->pGraph = p_graph;
  873.     if( p_graph ) p_graph->AddRef();
  874.     return NOERROR;
  875. };
  876. STDMETHODIMP CaptureFilter::JoinFilterGraph( IFilterGraph * pGraph,
  877.                                              LPCWSTR pName )
  878. {
  879. #ifdef DEBUG_DSHOW
  880.     msg_Dbg( p_input, "CaptureFilter::JoinFilterGraph" );
  881. #endif
  882.     p_graph = pGraph;
  883.     return NOERROR;
  884. };
  885. STDMETHODIMP CaptureFilter::QueryVendorInfo( LPWSTR* pVendorInfo )
  886. {
  887. #ifdef DEBUG_DSHOW
  888.     msg_Dbg( p_input, "CaptureFilter::QueryVendorInfo" );
  889. #endif
  890.     return E_NOTIMPL;
  891. };
  892. /* Custom methods */
  893. CapturePin *CaptureFilter::CustomGetPin()
  894. {
  895.     return p_pin;
  896. }
  897. /****************************************************************************
  898.  * Implementation of our dummy directshow enumpins class
  899.  ****************************************************************************/
  900. CaptureEnumPins::CaptureEnumPins( vlc_object_t *_p_input,
  901.                                   CaptureFilter *_p_filter,
  902.                                   CaptureEnumPins *pEnumPins )
  903.   : p_input( _p_input ), p_filter( _p_filter ), i_ref( 1 )
  904. {
  905.     /* Hold a reference count on our filter */
  906.     p_filter->AddRef();
  907.     /* Are we creating a new enumerator */
  908.     if( pEnumPins == NULL )
  909.     {
  910.         i_position = 0;
  911.     }
  912.     else
  913.     {
  914.         i_position = pEnumPins->i_position;
  915.     }
  916. }
  917. CaptureEnumPins::~CaptureEnumPins()
  918. {
  919. #ifdef DEBUG_DSHOW_L1
  920.     msg_Dbg( p_input, "CaptureEnumPins::~CaptureEnumPins" );
  921. #endif
  922.     p_filter->Release();
  923. }
  924. /* IUnknown methods */
  925. STDMETHODIMP CaptureEnumPins::QueryInterface( REFIID riid, void **ppv )
  926. {
  927. #ifdef DEBUG_DSHOW_L1
  928.     msg_Dbg( p_input, "CaptureEnumPins::QueryInterface" );
  929. #endif
  930.     if( riid == IID_IUnknown ||
  931.         riid == IID_IEnumPins )
  932.     {
  933.         AddRef();
  934.         *ppv = (IEnumPins *)this;
  935.         return NOERROR;
  936.     }
  937.     else
  938.     {
  939.         *ppv = NULL;
  940.         return E_NOINTERFACE;
  941.     }
  942. };
  943. STDMETHODIMP_(ULONG) CaptureEnumPins::AddRef()
  944. {
  945. #ifdef DEBUG_DSHOW_L1
  946.     msg_Dbg( p_input, "CaptureEnumPins::AddRef (ref: %i)", i_ref );
  947. #endif
  948.     return i_ref++;
  949. };
  950. STDMETHODIMP_(ULONG) CaptureEnumPins::Release()
  951. {
  952. #ifdef DEBUG_DSHOW_L1
  953.     msg_Dbg( p_input, "CaptureEnumPins::Release (ref: %i)", i_ref );
  954. #endif
  955.     if( !InterlockedDecrement(&i_ref) ) delete this;
  956.     return 0;
  957. };
  958. /* IEnumPins */
  959. STDMETHODIMP CaptureEnumPins::Next( ULONG cPins, IPin ** ppPins,
  960.                                     ULONG * pcFetched )
  961. {
  962. #ifdef DEBUG_DSHOW_L1
  963.     msg_Dbg( p_input, "CaptureEnumPins::Next" );
  964. #endif
  965.     unsigned int i_fetched = 0;
  966.     if( i_position < 1 && cPins > 0 )
  967.     {
  968.         IPin *pPin = p_filter->CustomGetPin();
  969.         *ppPins = pPin;
  970.         pPin->AddRef();
  971.         i_fetched = 1;
  972.         i_position++;
  973.     }
  974.     if( pcFetched ) *pcFetched = i_fetched;
  975.     return (i_fetched == cPins) ? S_OK : S_FALSE;
  976. };
  977. STDMETHODIMP CaptureEnumPins::Skip( ULONG cPins )
  978. {
  979. #ifdef DEBUG_DSHOW_L1
  980.     msg_Dbg( p_input, "CaptureEnumPins::Skip" );
  981. #endif
  982.     i_position += cPins;
  983.     if( i_position > 1 )
  984.     {
  985.         return S_FALSE;
  986.     }
  987.     return S_OK;
  988. };
  989. STDMETHODIMP CaptureEnumPins::Reset()
  990. {
  991. #ifdef DEBUG_DSHOW_L1
  992.     msg_Dbg( p_input, "CaptureEnumPins::Reset" );
  993. #endif
  994.     i_position = 0;
  995.     return S_OK;
  996. };
  997. STDMETHODIMP CaptureEnumPins::Clone( IEnumPins **ppEnum )
  998. {
  999. #ifdef DEBUG_DSHOW_L1
  1000.     msg_Dbg( p_input, "CaptureEnumPins::Clone" );
  1001. #endif
  1002.     *ppEnum = new CaptureEnumPins( p_input, p_filter, this );
  1003.     if( *ppEnum == NULL ) return E_OUTOFMEMORY;
  1004.     return NOERROR;
  1005. };
  1006. /****************************************************************************
  1007.  * Implementation of our dummy directshow enummediatypes class
  1008.  ****************************************************************************/
  1009. CaptureEnumMediaTypes::CaptureEnumMediaTypes( vlc_object_t *_p_input,
  1010.     CapturePin *_p_pin, CaptureEnumMediaTypes *pEnumMediaTypes )
  1011.   : p_input( _p_input ), p_pin( _p_pin ), i_ref( 1 )
  1012. {
  1013.     /* Hold a reference count on our filter */
  1014.     p_pin->AddRef();
  1015.     /* Are we creating a new enumerator */
  1016.     if( pEnumMediaTypes == NULL )
  1017.     {
  1018.         CopyMediaType(&cx_media_type, &p_pin->cx_media_type);
  1019.         i_position = 0;
  1020.     }
  1021.     else
  1022.     {
  1023.         CopyMediaType(&cx_media_type, &pEnumMediaTypes->cx_media_type);
  1024.         i_position = pEnumMediaTypes->i_position;
  1025.     }
  1026. }
  1027. CaptureEnumMediaTypes::~CaptureEnumMediaTypes()
  1028. {
  1029. #ifdef DEBUG_DSHOW_L1
  1030.     msg_Dbg( p_input, "CaptureEnumMediaTypes::~CaptureEnumMediaTypes" );
  1031. #endif
  1032.     FreeMediaType(cx_media_type);
  1033.     p_pin->Release();
  1034. }
  1035. /* IUnknown methods */
  1036. STDMETHODIMP CaptureEnumMediaTypes::QueryInterface( REFIID riid, void **ppv )
  1037. {
  1038. #ifdef DEBUG_DSHOW_L1
  1039.     msg_Dbg( p_input, "CaptureEnumMediaTypes::QueryInterface" );
  1040. #endif
  1041.     if( riid == IID_IUnknown ||
  1042.         riid == IID_IEnumMediaTypes )
  1043.     {
  1044.         AddRef();
  1045.         *ppv = (IEnumMediaTypes *)this;
  1046.         return NOERROR;
  1047.     }
  1048.     else
  1049.     {
  1050.         *ppv = NULL;
  1051.         return E_NOINTERFACE;
  1052.     }
  1053. };
  1054. STDMETHODIMP_(ULONG) CaptureEnumMediaTypes::AddRef()
  1055. {
  1056. #ifdef DEBUG_DSHOW_L1
  1057.     msg_Dbg( p_input, "CaptureEnumMediaTypes::AddRef (ref: %i)", i_ref );
  1058. #endif
  1059.     return i_ref++;
  1060. };
  1061. STDMETHODIMP_(ULONG) CaptureEnumMediaTypes::Release()
  1062. {
  1063. #ifdef DEBUG_DSHOW_L1
  1064.     msg_Dbg( p_input, "CaptureEnumMediaTypes::Release (ref: %i)", i_ref );
  1065. #endif
  1066.     if( !InterlockedDecrement(&i_ref) ) delete this;
  1067.     return 0;
  1068. };
  1069. /* IEnumMediaTypes */
  1070. STDMETHODIMP CaptureEnumMediaTypes::Next( ULONG cMediaTypes,
  1071.                                           AM_MEDIA_TYPE ** ppMediaTypes,
  1072.                                           ULONG * pcFetched )
  1073. {
  1074. #ifdef DEBUG_DSHOW_L1
  1075.     msg_Dbg( p_input, "CaptureEnumMediaTypes::Next " );
  1076. #endif
  1077.     ULONG copied = 0;
  1078.     ULONG offset = 0;
  1079.     ULONG max = p_pin->media_type_count;
  1080.     if( ! ppMediaTypes )
  1081.         return E_POINTER;
  1082.     if( (! pcFetched)  && (cMediaTypes > 1) )
  1083.        return E_POINTER;
  1084.     /*
  1085.     ** use connection media type as first entry in iterator if it exists
  1086.     */
  1087.     copied = 0;
  1088.     if( cx_media_type.subtype != GUID_NULL )
  1089.     {
  1090.         ++max;
  1091.         if( i_position == 0 )
  1092.         {
  1093.             ppMediaTypes[copied] =
  1094.                 (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
  1095.             if( CopyMediaType(ppMediaTypes[copied], &cx_media_type) != S_OK )
  1096.                 return E_OUTOFMEMORY;
  1097.             ++i_position;
  1098.             ++copied;
  1099.         }
  1100.     }
  1101.     while( (copied < cMediaTypes) && (i_position < max)  )
  1102.     {
  1103.         ppMediaTypes[copied] =
  1104.             (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
  1105.         if( CopyMediaType( ppMediaTypes[copied],
  1106.                            &p_pin->media_types[i_position-offset]) != S_OK )
  1107.             return E_OUTOFMEMORY;
  1108.         ++copied;
  1109.         ++i_position;
  1110.     }
  1111.     if( pcFetched )  *pcFetched = copied;
  1112.     return (copied == cMediaTypes) ? S_OK : S_FALSE;
  1113. };
  1114. STDMETHODIMP CaptureEnumMediaTypes::Skip( ULONG cMediaTypes )
  1115. {
  1116.     ULONG max =  p_pin->media_type_count;
  1117.     if( cx_media_type.subtype != GUID_NULL )
  1118.     {
  1119.         max = 1;
  1120.     }
  1121. #ifdef DEBUG_DSHOW_L1
  1122.     msg_Dbg( p_input, "CaptureEnumMediaTypes::Skip" );
  1123. #endif
  1124.     i_position += cMediaTypes;
  1125.     return (i_position < max) ? S_OK : S_FALSE;
  1126. };
  1127. STDMETHODIMP CaptureEnumMediaTypes::Reset()
  1128. {
  1129. #ifdef DEBUG_DSHOW_L1
  1130.     msg_Dbg( p_input, "CaptureEnumMediaTypes::Reset" );
  1131. #endif
  1132.     FreeMediaType(cx_media_type);
  1133.     CopyMediaType(&cx_media_type, &p_pin->cx_media_type);
  1134.     i_position = 0;
  1135.     return S_OK;
  1136. };
  1137. STDMETHODIMP CaptureEnumMediaTypes::Clone( IEnumMediaTypes **ppEnum )
  1138. {
  1139. #ifdef DEBUG_DSHOW_L1
  1140.     msg_Dbg( p_input, "CaptureEnumMediaTypes::Clone" );
  1141. #endif
  1142.     *ppEnum = new CaptureEnumMediaTypes( p_input, p_pin, this );
  1143.     if( *ppEnum == NULL ) return E_OUTOFMEMORY;
  1144.     return NOERROR;
  1145. };