modules/access/dshow/filter.h

   1 /*****************************************************************************
   2  * filter.h : DirectShow access module for vlc:
   3  * CapturePin, CaptureFilter, CaptureEnumPins implementations
   4  *****************************************************************************
   5  * Copyright (C) 2002-2004, 2008 the VideoLAN team
   6  * $Id$
   7  *
   8  * Author: Gildas Bazin <gbazin@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28
  29 #include <deque>
  30 using namespace std;
  31
  32 #ifndef _MSC_VER
  33 #   include <wtypes.h>
  34 #   include <unknwn.h>
  35 #   include <ole2.h>
  36 #   include <limits.h>
  37 #   ifdef _WINGDI_
  38 #      undef _WINGDI_
  39 #   endif
  40 #   define _WINGDI_ 1
  41 #   define AM_NOVTABLE
  42 #   define _OBJBASE_H_
  43 #   undef _X86_
  44 #   ifndef _I64_MAX
  45 #     define _I64_MAX LONG_LONG_MAX
  46 #   endif
  47 #   define LONGLONG long long
  48 #endif
  49
  50 #include <dshow.h>
  51 #include "vlc_dshow.h"
  52
  53 typedef struct VLCMediaSample
  54 {
  55     IMediaSample *p_sample;
  56     mtime_t i_timestamp;
  57
  58 } VLCMediaSample;
  59
  60 /* */
  61 void WINAPI FreeMediaType( AM_MEDIA_TYPE& mt );
  62 HRESULT WINAPI CopyMediaType( AM_MEDIA_TYPE *pmtTarget,
  63                               const AM_MEDIA_TYPE *pmtSource );
  64
  65 int GetFourCCFromMediaType(const AM_MEDIA_TYPE &media_type);
  66
  67 /****************************************************************************
  68  * Declaration of our dummy directshow filter pin class
  69  ****************************************************************************/
  70 class CaptureFilter;
  71 class CapturePin: public IPin, public IMemInputPin
  72 {
  73     friend class CaptureEnumMediaTypes;
  74
  75     vlc_object_t *p_input;
  76     access_sys_t *p_sys;
  77     CaptureFilter  *p_filter;
  78
  79     IPin *p_connected_pin;
  80
  81     AM_MEDIA_TYPE *media_types;
  82     size_t media_type_count;
  83
  84     AM_MEDIA_TYPE cx_media_type;
  85
  86     deque<VLCMediaSample> samples_queue;
  87
  88     long i_ref;
  89
  90   public:
  91     CapturePin( vlc_object_t *_p_input, access_sys_t *p_sys,
  92                 CaptureFilter *_p_filter,
  93                 AM_MEDIA_TYPE *mt, size_t mt_count );
  94     virtual ~CapturePin();
  95
  96     /* IUnknown methods */
  97     STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
  98     STDMETHODIMP_(ULONG) AddRef();
  99     STDMETHODIMP_(ULONG) Release();
 100
 101     /* IPin methods */
 102     STDMETHODIMP Connect( IPin * pReceivePin, const AM_MEDIA_TYPE *pmt );
 103     STDMETHODIMP ReceiveConnection( IPin * pConnector,
 104                                     const AM_MEDIA_TYPE *pmt );
 105     STDMETHODIMP Disconnect();
 106     STDMETHODIMP ConnectedTo( IPin **pPin );
 107     STDMETHODIMP ConnectionMediaType( AM_MEDIA_TYPE *pmt );
 108     STDMETHODIMP QueryPinInfo( PIN_INFO * pInfo );
 109     STDMETHODIMP QueryDirection( PIN_DIRECTION * pPinDir );
 110     STDMETHODIMP QueryId( LPWSTR * Id );
 111     STDMETHODIMP QueryAccept( const AM_MEDIA_TYPE *pmt );
 112     STDMETHODIMP EnumMediaTypes( IEnumMediaTypes **ppEnum );
 113     STDMETHODIMP QueryInternalConnections( IPin* *apPin, ULONG *nPin );
 114     STDMETHODIMP EndOfStream( void );
 115
 116     STDMETHODIMP BeginFlush( void );
 117     STDMETHODIMP EndFlush( void );
 118     STDMETHODIMP NewSegment( REFERENCE_TIME tStart, REFERENCE_TIME tStop,
 119                              double dRate );
 120
 121     /* IMemInputPin methods */
 122     STDMETHODIMP GetAllocator( IMemAllocator **ppAllocator );
 123     STDMETHODIMP NotifyAllocator(  IMemAllocator *pAllocator, BOOL bReadOnly );
 124     STDMETHODIMP GetAllocatorRequirements( ALLOCATOR_PROPERTIES *pProps );
 125     STDMETHODIMP Receive( IMediaSample *pSample );
 126     STDMETHODIMP ReceiveMultiple( IMediaSample **pSamples, long nSamples,
 127                                   long *nSamplesProcessed );
 128     STDMETHODIMP ReceiveCanBlock( void );
 129
 130     /* Custom methods */
 131     HRESULT CustomGetSample( VLCMediaSample * );
 132     HRESULT CustomGetSamples( deque<VLCMediaSample> &external_queue );
 133
 134     AM_MEDIA_TYPE &CustomGetMediaType();
 135 };
 136
 137 /****************************************************************************
 138  * Declaration of our dummy directshow filter class
 139  ****************************************************************************/
 140 class CaptureFilter : public IBaseFilter
 141 {
 142     friend class CapturePin;
 143
 144     vlc_object_t   *p_input;
 145     CapturePin     *p_pin;
 146     IFilterGraph   *p_graph;
 147     //AM_MEDIA_TYPE  media_type;
 148     FILTER_STATE   state;
 149
 150     long i_ref;
 151
 152   public:
 153     CaptureFilter( vlc_object_t *_p_input, access_sys_t *p_sys,
 154                    AM_MEDIA_TYPE *mt, size_t mt_count );
 155     virtual ~CaptureFilter();
 156
 157     /* IUnknown methods */
 158     STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
 159     STDMETHODIMP_(ULONG) AddRef();
 160     STDMETHODIMP_(ULONG) Release();
 161
 162     /* IPersist method */
 163     STDMETHODIMP GetClassID(CLSID *pClsID);
 164
 165     /* IMediaFilter methods */
 166     STDMETHODIMP GetState(DWORD dwMSecs, FILTER_STATE *State);
 167     STDMETHODIMP SetSyncSource(IReferenceClock *pClock);
 168     STDMETHODIMP GetSyncSource(IReferenceClock **pClock);
 169     STDMETHODIMP Stop();
 170     STDMETHODIMP Pause();
 171     STDMETHODIMP Run(REFERENCE_TIME tStart);
 172
 173     /* IBaseFilter methods */
 174     STDMETHODIMP EnumPins( IEnumPins ** ppEnum );
 175     STDMETHODIMP FindPin( LPCWSTR Id, IPin ** ppPin );
 176     STDMETHODIMP QueryFilterInfo( FILTER_INFO * pInfo );
 177     STDMETHODIMP JoinFilterGraph( IFilterGraph * pGraph, LPCWSTR pName );
 178     STDMETHODIMP QueryVendorInfo( LPWSTR* pVendorInfo );
 179
 180     /* Custom methods */
 181     CapturePin *CustomGetPin();
 182 };
 183
 184 /****************************************************************************
 185  * Declaration of our dummy directshow enumpins class
 186  ****************************************************************************/
 187 class CaptureEnumPins : public IEnumPins
 188 {
 189     vlc_object_t *p_input;
 190     CaptureFilter  *p_filter;
 191
 192     int i_position;
 193     long i_ref;
 194
 195 public:
 196     CaptureEnumPins( vlc_object_t *_p_input, CaptureFilter *_p_filter,
 197                      CaptureEnumPins *pEnumPins );
 198     virtual ~CaptureEnumPins();
 199
 200     // IUnknown
 201     STDMETHODIMP QueryInterface( REFIID riid, void **ppv );
 202     STDMETHODIMP_(ULONG) AddRef();
 203     STDMETHODIMP_(ULONG) Release();
 204
 205     // IEnumPins
 206     STDMETHODIMP Next( ULONG cPins, IPin ** ppPins, ULONG * pcFetched );
 207     STDMETHODIMP Skip( ULONG cPins );
 208     STDMETHODIMP Reset();
 209     STDMETHODIMP Clone( IEnumPins **ppEnum );
 210 };
 211
 212 /****************************************************************************
 213  * Declaration of our dummy directshow enummediatypes class
 214  ****************************************************************************/
 215 class CaptureEnumMediaTypes : public IEnumMediaTypes
 216 {
 217     vlc_object_t *p_input;
 218     CapturePin     *p_pin;
 219     AM_MEDIA_TYPE cx_media_type;
 220
 221     size_t i_position;
 222     long i_ref;
 223
 224 public:
 225     CaptureEnumMediaTypes( vlc_object_t *_p_input, CapturePin *_p_pin,
 226                            CaptureEnumMediaTypes *pEnumMediaTypes );
 227
 228     virtual ~CaptureEnumMediaTypes();
 229
 230     // IUnknown
 231     STDMETHODIMP QueryInterface( REFIID riid, void **ppv );
 232     STDMETHODIMP_(ULONG) AddRef();
 233     STDMETHODIMP_(ULONG) Release();
 234
 235     // IEnumMediaTypes
 236     STDMETHODIMP Next( ULONG cMediaTypes, AM_MEDIA_TYPE ** ppMediaTypes,
 237                        ULONG * pcFetched );
 238     STDMETHODIMP Skip( ULONG cMediaTypes );
 239     STDMETHODIMP Reset();
 240     STDMETHODIMP Clone( IEnumMediaTypes **ppEnum );
 241 };