X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Faccess%2Fdshow%2Ffilter.cpp;h=88d8f2b78d100af48c07894bdd0a46c3e9394669;hb=6ee1e193fd896ab9a4729fde14f009d9ce629815;hp=bee689f5112f80708591e95516ee7ae920a6adb5;hpb=208c33f5b91999a766ee391780606216450218f0;p=vlc diff --git a/modules/access/dshow/filter.cpp b/modules/access/dshow/filter.cpp index bee689f511..88d8f2b78d 100644 --- a/modules/access/dshow/filter.cpp +++ b/modules/access/dshow/filter.cpp @@ -1,7 +1,7 @@ /***************************************************************************** * filter.c : DirectShow access module for vlc ***************************************************************************** - * Copyright (C) 2002 VideoLAN + * Copyright (C) 2002 the VideoLAN team * $Id$ * * Author: Gildas Bazin @@ -18,61 +18,60 @@ * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA. + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************/ /***************************************************************************** * Preamble *****************************************************************************/ -#include -#include -#include #include -#include -#include +#include #ifndef _MSC_VER /* Work-around a bug in w32api-2.5 */ # define QACONTAINERFLAGS QACONTAINERFLAGS_SOMETHINGELSE #endif +#include "common.h" #include "filter.h" #define DEBUG_DSHOW 1 -struct access_sys_t -{ - vlc_mutex_t lock; - vlc_cond_t wait; -}; +#define FILTER_NAME L"VideoLAN Capture Filter" +#define PIN_NAME L"Capture" /***************************************************************************** * DirectShow GUIDs. - * Easier to define them hear as mingw doesn't provide them all. + * Easier to define them here as mingw doesn't provide them all. *****************************************************************************/ const GUID CLSID_SystemDeviceEnum = {0x62be5d10, 0x60eb, 0x11d0, {0xbd, 0x3b, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}}; const GUID CLSID_VideoInputDeviceCategory = {0x860BB310,0x5D01,0x11d0,{0xBD,0x3B,0x00,0xA0,0xC9,0x11,0xCE,0x86}}; const GUID CLSID_AudioInputDeviceCategory = {0x33d9a762, 0x90c8, 0x11d0, {0xbd, 0x43, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}}; -const GUID IID_IPropertyBag = {0x55272A00, 0x42CB, 0x11CE, {0x81, 0x35, 0x00, 0xAA, 0x00, 0x4B, 0xB8, 0x51}}; +//const GUID IID_IPropertyBag = {0x55272A00, 0x42CB, 0x11CE, {0x81, 0x35, 0x00, 0xAA, 0x00, 0x4B, 0xB8, 0x51}}; +extern const GUID IID_IPropertyBag; const GUID IID_ICreateDevEnum = {0x29840822, 0x5b84, 0x11d0, {0xbd, 0x3b, 0x00, 0xa0, 0xc9, 0x11, 0xce, 0x86}}; const GUID IID_IFilterGraph = {0x56a8689f, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}}; const GUID IID_IMediaControl = {0x56a868b1, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}}; const GUID CLSID_FilterGraph = {0xe436ebb3, 0x524f, 0x11ce, {0x9f, 0x53, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}}; -const GUID IID_IUnknown = {0x00000000, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}}; -const GUID IID_IPersist = {0x0000010c, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}}; +//const GUID IID_IUnknown = {0x00000000, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}}; +extern const GUID IID_IUnknown; +//const GUID IID_IPersist = {0x0000010c, 0x0000, 0x0000, {0xc0,0x00, 0x00,0x00,0x00,0x00,0x00,0x46}}; +extern const GUID IID_IPersist; const GUID IID_IMediaFilter = {0x56a86899, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}}; const GUID IID_IBaseFilter = {0x56a86895, 0x0ad4, 0x11ce, {0xb0, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}}; const GUID IID_IPin = {0x56a86891, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}}; const GUID IID_IMemInputPin = {0x56a8689d, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}}; +extern const GUID IID_IMemInputPin; const GUID IID_IEnumPins = {0x56a86892, 0x0ad4, 0x11ce, {0xb0,0x3a, 0x00,0x20,0xaf,0x0b,0xa7,0x70}}; const GUID IID_IEnumMediaTypes = {0x89c31040, 0x846b, 0x11ce, {0x97,0xd3, 0x00,0xaa,0x00,0x55,0x59,0x5a}}; const GUID IID_IAMBufferNegotiation = {0x56ed71a0, 0xaf5f, 0x11d0, {0xb3, 0xf0, 0x00, 0xaa, 0x00, 0x37, 0x61, 0xc5}}; -const GUID IID_ISpecifyPropertyPages = {0xb196b28b, 0xbab4, 0x101a, {0xb6, 0x9c, 0x00, 0xaa, 0x00, 0x34, 0x1d, 0x07}}; +//const GUID IID_ISpecifyPropertyPages = {0xb196b28b, 0xbab4, 0x101a, {0xb6, 0x9c, 0x00, 0xaa, 0x00, 0x34, 0x1d, 0x07}}; +extern const GUID IID_ISpecifyPropertyPages; const GUID IID_IQualityControl = {0x56a868a5, 0x0ad4, 0x11ce, {0xb, 0x3a, 0x00, 0x20, 0xaf, 0x0b, 0xa7, 0x70}}; @@ -89,7 +88,9 @@ const GUID IID_IAMTVTuner = {0x211A8766, 0x03AC, 0x11d1, {0x8D, 0x13, 0x00, 0xAA const GUID IID_IKsPropertySet = {0x31EFAC30, 0x515C, 0x11d0, {0xA9, 0xAA, 0x00, 0xAA, 0x00, 0x61, 0xBE, 0x93}}; +/* Video Format */ +const GUID FORMAT_VideoInfo = {0x05589f80, 0xc356, 0x11ce, {0xbf, 0x01, 0x00, 0xaa, 0x00, 0x55, 0x59, 0x5a}}; /* * MEDIATYPEs and MEDIASUBTYPEs */ @@ -139,6 +140,12 @@ const GUID MEDIASUBTYPE_MPEG2_PROGRAM = {0xe06d8022, 0xdb46, 0x11cf, {0xb4, 0xd1 const GUID MEDIASUBTYPE_MPEG2_TRANSPORT = {0xe06d8023, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}}; const GUID FORMAT_MPEG2Video = {0xe06d80e3, 0xdb46, 0x11cf, {0xb4, 0xd1, 0x00, 0x80, 0x5f, 0x6c, 0xbb, 0xea}}; +/* MJPG format */ +const GUID MEDIASUBTYPE_MJPG = {0x47504A4D, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}}; + +/* DivX formats */ +const GUID MEDIASUBTYPE_DIVX = {0x58564944, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71}}; + /* Analog Video */ const GUID FORMAT_AnalogVideo = {0x482dde0, 0x7817, 0x11cf, {0x8a, 0x3, 0x0, 0xaa, 0x0, 0x6e, 0xcb, 0x65}}; @@ -166,7 +173,8 @@ const GUID PIN_CATEGORY_ANALOGVIDEOIN= {0xfb6c4283, 0x0353, 0x11d1, {0x90, 0x5f, const GUID PIN_CATEGORY_CAPTURE= {0xfb6c4281, 0x0353, 0x11d1, {0x90, 0x5f, 0x0, 0x0, 0xc0, 0xcc, 0x16, 0xba}}; const GUID LOOK_UPSTREAM_ONLY= {0xac798be0, 0x98e3, 0x11d1, {0xb3, 0xf1, 0x0, 0xaa, 0x0, 0x37, 0x61, 0xc}}; -const GUID GUID_NULL = {0x0000, 0x0000, 0x0000, {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}}; +//const GUID GUID_NULL = {0x0000, 0x0000, 0x0000, {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}}; +extern const GUID GUID_NULL; void WINAPI FreeMediaType( AM_MEDIA_TYPE& mt ) { @@ -187,7 +195,10 @@ HRESULT WINAPI CopyMediaType( AM_MEDIA_TYPE *pmtTarget, const AM_MEDIA_TYPE *pmtSource ) { *pmtTarget = *pmtSource; - if( pmtSource->cbFormat != 0 ) + + if( !pmtSource || !pmtTarget ) return S_FALSE; + + if( pmtSource->cbFormat && pmtSource->pbFormat ) { pmtTarget->pbFormat = (PBYTE)CoTaskMemAlloc( pmtSource->cbFormat ); if( pmtTarget->pbFormat == NULL ) @@ -209,15 +220,119 @@ HRESULT WINAPI CopyMediaType( AM_MEDIA_TYPE *pmtTarget, return S_OK; } +int GetFourCCFromMediaType( const AM_MEDIA_TYPE &media_type ) +{ + int i_fourcc = 0; + + if( media_type.majortype == MEDIATYPE_Video ) + { + /* currently only support this type of video info format */ + if( 1 /* media_type.formattype == FORMAT_VideoInfo */ ) + { + /* Packed RGB formats */ + if( media_type.subtype == MEDIASUBTYPE_RGB1 ) + i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '1' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB4 ) + i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '4' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB8 ) + i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB555 ) + i_fourcc = VLC_FOURCC( 'R', 'V', '1', '5' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB565 ) + i_fourcc = VLC_FOURCC( 'R', 'V', '1', '6' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB24 ) + i_fourcc = VLC_FOURCC( 'R', 'V', '2', '4' ); + else if( media_type.subtype == MEDIASUBTYPE_RGB32 ) + i_fourcc = VLC_FOURCC( 'R', 'V', '3', '2' ); + else if( media_type.subtype == MEDIASUBTYPE_ARGB32 ) + i_fourcc = VLC_FOURCC( 'R', 'G', 'B', 'A' ); + + /* Planar YUV formats */ + else if( media_type.subtype == MEDIASUBTYPE_I420 ) + i_fourcc = VLC_FOURCC( 'I', '4', '2', '0' ); + else if( media_type.subtype == MEDIASUBTYPE_Y41P ) + i_fourcc = VLC_FOURCC( 'I', '4', '1', '1' ); + else if( media_type.subtype == MEDIASUBTYPE_YV12 ) + i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' ); + else if( media_type.subtype == MEDIASUBTYPE_IYUV ) + i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' ); + else if( media_type.subtype == MEDIASUBTYPE_YVU9 ) + i_fourcc = VLC_FOURCC( 'Y', 'V', 'U', '9' ); + + /* Packed YUV formats */ + else if( media_type.subtype == MEDIASUBTYPE_YVYU ) + i_fourcc = VLC_FOURCC( 'Y', 'V', 'Y', 'U' ); + else if( media_type.subtype == MEDIASUBTYPE_YUYV ) + i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', '2' ); + else if( media_type.subtype == MEDIASUBTYPE_Y411 ) + i_fourcc = VLC_FOURCC( 'I', '4', '1', 'N' ); + else if( media_type.subtype == MEDIASUBTYPE_Y211 ) + i_fourcc = VLC_FOURCC( 'Y', '2', '1', '1' ); + else if( media_type.subtype == MEDIASUBTYPE_YUY2 ) + i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', '2' ); + else if( media_type.subtype == MEDIASUBTYPE_UYVY ) + i_fourcc = VLC_FOURCC( 'U', 'Y', 'V', 'Y' ); + + /* MPEG2 video elementary stream */ + else if( media_type.subtype == MEDIASUBTYPE_MPEG2_VIDEO ) + i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'v' ); + + /* DivX video */ + else if( media_type.subtype == MEDIASUBTYPE_DIVX ) + i_fourcc = VLC_FOURCC( 'D', 'I', 'V', 'X' ); + + /* DV formats */ + else if( media_type.subtype == MEDIASUBTYPE_dvsl ) + i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'l' ); + else if( media_type.subtype == MEDIASUBTYPE_dvsd ) + i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'd' ); + else if( media_type.subtype == MEDIASUBTYPE_dvhd ) + i_fourcc = VLC_FOURCC( 'd', 'v', 'h', 'd' ); + + /* MJPEG format */ + else if( media_type.subtype == MEDIASUBTYPE_MJPG ) + i_fourcc = VLC_FOURCC( 'M', 'J', 'P', 'G' ); + + } + } + else if( media_type.majortype == MEDIATYPE_Audio ) + { + /* currently only support this type of audio info format */ + if( media_type.formattype == FORMAT_WaveFormatEx ) + { + if( media_type.subtype == MEDIASUBTYPE_PCM ) + i_fourcc = VLC_FOURCC( 'a', 'r', 'a', 'w' ); + else if( media_type.subtype == MEDIASUBTYPE_IEEE_FLOAT ) + i_fourcc = VLC_FOURCC( 'f', 'l', '3', '2' ); + } + } + else if( media_type.majortype == MEDIATYPE_Stream ) + { + if( media_type.subtype == MEDIASUBTYPE_MPEG2_PROGRAM ) + i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'p' ); + else if( media_type.subtype == MEDIASUBTYPE_MPEG2_TRANSPORT ) + i_fourcc = VLC_FOURCC( 'm', 'p', '2', 't' ); + } + + return i_fourcc; +} + /**************************************************************************** * Implementation of our dummy directshow filter pin class ****************************************************************************/ -CapturePin::CapturePin( input_thread_t * _p_input, CaptureFilter *_p_filter, - AM_MEDIA_TYPE mt ) - : p_input( _p_input ), p_filter( _p_filter ), p_connected_pin( NULL ), - media_type( mt ), i_ref( 1 ) -{ +CapturePin::CapturePin( vlc_object_t *_p_input, access_sys_t *_p_sys, + CaptureFilter *_p_filter, + AM_MEDIA_TYPE *mt, size_t mt_count ) + : p_input( _p_input ), p_sys( _p_sys ), p_filter( _p_filter ), + p_connected_pin( NULL ), media_types(mt), media_type_count(mt_count), + i_ref( 1 ) +{ + cx_media_type.majortype = mt[0].majortype; + cx_media_type.subtype = GUID_NULL; + cx_media_type.pbFormat = NULL; + cx_media_type.cbFormat = 0; + cx_media_type.pUnk = NULL; } CapturePin::~CapturePin() @@ -225,6 +340,11 @@ CapturePin::~CapturePin() #ifdef DEBUG_DSHOW msg_Dbg( p_input, "CapturePin::~CapturePin" ); #endif + for( size_t c=0; cp_access_data; vlc_mutex_lock( &p_sys->lock ); if( samples_queue.size() ) { @@ -246,15 +365,15 @@ HRESULT CapturePin::CustomGetSample( VLCMediaSample *vlc_sample ) return S_FALSE; } -AM_MEDIA_TYPE CapturePin::CustomGetMediaType() +AM_MEDIA_TYPE &CapturePin::CustomGetMediaType() { - return media_type; + return cx_media_type; } /* IUnknown methods */ STDMETHODIMP CapturePin::QueryInterface(REFIID riid, void **ppv) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::QueryInterface" ); #endif @@ -273,7 +392,7 @@ STDMETHODIMP CapturePin::QueryInterface(REFIID riid, void **ppv) } else { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::QueryInterface() failed for: " "%04X-%02X-%02X-%02X%02X%02X%02X%02X%02X%02X%02X", (int)riid.Data1, (int)riid.Data2, (int)riid.Data3, @@ -289,7 +408,7 @@ STDMETHODIMP CapturePin::QueryInterface(REFIID riid, void **ppv) STDMETHODIMP_(ULONG) CapturePin::AddRef() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::AddRef (ref: %i)", i_ref ); #endif @@ -297,7 +416,7 @@ STDMETHODIMP_(ULONG) CapturePin::AddRef() }; STDMETHODIMP_(ULONG) CapturePin::Release() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::Release (ref: %i)", i_ref ); #endif @@ -310,88 +429,127 @@ STDMETHODIMP_(ULONG) CapturePin::Release() STDMETHODIMP CapturePin::Connect( IPin * pReceivePin, const AM_MEDIA_TYPE *pmt ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::Connect" ); -#endif - return E_NOTIMPL; + if( State_Running == p_filter->state ) + { + msg_Dbg( p_input, "CapturePin::Connect [not stopped]" ); + return VFW_E_NOT_STOPPED; + } + + if( p_connected_pin ) + { + msg_Dbg( p_input, "CapturePin::Connect [already connected]" ); + return VFW_E_ALREADY_CONNECTED; + } + + if( !pmt ) return S_OK; + + if( GUID_NULL != pmt->majortype && + media_types[0].majortype != pmt->majortype ) + { + msg_Dbg( p_input, "CapturePin::Connect [media major type mismatch]" ); + return S_FALSE; + } + + if( GUID_NULL != pmt->subtype && !GetFourCCFromMediaType(*pmt) ) + { + msg_Dbg( p_input, "CapturePin::Connect [media subtype type " + "not supported]" ); + return S_FALSE; + } + + if( pmt->pbFormat && pmt->majortype == MEDIATYPE_Video ) + { + if( !((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight || + !((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth ) + { + msg_Dbg( p_input, "CapturePin::Connect " + "[video width/height == 0 ]" ); + return S_FALSE; + } + } + + msg_Dbg( p_input, "CapturePin::Connect [OK]" ); + return S_OK; } STDMETHODIMP CapturePin::ReceiveConnection( IPin * pConnector, const AM_MEDIA_TYPE *pmt ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::ReceiveConnection" ); -#endif + if( State_Stopped != p_filter->state ) + { + msg_Dbg( p_input, "CapturePin::ReceiveConnection [not stopped]" ); + return VFW_E_NOT_STOPPED; + } - if( pmt->majortype == MEDIATYPE_Video ) + if( !pConnector || !pmt ) { - if( media_type.subtype != GUID_NULL && - media_type.subtype != pmt->subtype ) - return VFW_E_TYPE_NOT_ACCEPTED; + msg_Dbg( p_input, "CapturePin::ReceiveConnection [null pointer]" ); + return E_POINTER; + } - if( media_type.pbFormat && - ((VIDEOINFOHEADER *)media_type.pbFormat)->bmiHeader.biHeight && - ((VIDEOINFOHEADER *)media_type.pbFormat)->bmiHeader.biHeight != - ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight ) - return VFW_E_TYPE_NOT_ACCEPTED; + if( p_connected_pin ) + { + msg_Dbg( p_input, "CapturePin::ReceiveConnection [already connected]"); + return VFW_E_ALREADY_CONNECTED; + } - if( media_type.pbFormat && - ((VIDEOINFOHEADER *)media_type.pbFormat)->bmiHeader.biWidth && - ((VIDEOINFOHEADER *)media_type.pbFormat)->bmiHeader.biWidth != - ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth ) - return VFW_E_TYPE_NOT_ACCEPTED; + if( S_OK != QueryAccept(pmt) ) + { + msg_Dbg( p_input, "CapturePin::ReceiveConnection " + "[media type not accepted]" ); + return VFW_E_TYPE_NOT_ACCEPTED; } + msg_Dbg( p_input, "CapturePin::ReceiveConnection [OK]" ); + p_connected_pin = pConnector; p_connected_pin->AddRef(); - FreeMediaType( media_type ); - return CopyMediaType( &media_type, pmt ); + FreeMediaType( cx_media_type ); + return CopyMediaType( &cx_media_type, pmt ); } STDMETHODIMP CapturePin::Disconnect() { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::Disconnect" ); -#endif - -#if 0 // FIXME: This does seem to create crashes sometimes - VLCMediaSample vlc_sample; - access_sys_t *p_sys = p_input->p_access_data; - - vlc_mutex_lock( &p_sys->lock ); - while( samples_queue.size() ) + if( ! p_connected_pin ) { - vlc_sample = samples_queue.back(); - samples_queue.pop_back(); - vlc_sample.p_sample->Release(); + msg_Dbg( p_input, "CapturePin::Disconnect [not connected]" ); + return S_FALSE; } - vlc_mutex_unlock( &p_sys->lock ); -#endif - if( p_connected_pin ) p_connected_pin->Release(); + msg_Dbg( p_input, "CapturePin::Disconnect [OK]" ); + + /* samples_queue was already flushed in EndFlush() */ + + p_connected_pin->Release(); p_connected_pin = NULL; - FreeMediaType( media_type ); + //FreeMediaType( cx_media_type ); + //cx_media_type.subtype = GUID_NULL; + return S_OK; } STDMETHODIMP CapturePin::ConnectedTo( IPin **pPin ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::ConnectedTo" ); -#endif - - if( !p_connected_pin ) return VFW_E_NOT_CONNECTED; + if( !p_connected_pin ) + { + msg_Dbg( p_input, "CapturePin::ConnectedTo [not connected]" ); + return VFW_E_NOT_CONNECTED; + } p_connected_pin->AddRef(); *pPin = p_connected_pin; + msg_Dbg( p_input, "CapturePin::ConnectedTo [OK]" ); + return S_OK; } STDMETHODIMP CapturePin::ConnectionMediaType( AM_MEDIA_TYPE *pmt ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::ConnectionMediaType" ); -#endif + if( !p_connected_pin ) + { + msg_Dbg( p_input, "CapturePin::ConnectionMediaType [not connected]" ); + return VFW_E_NOT_CONNECTED; + } - return CopyMediaType( pmt, &media_type ); + return CopyMediaType( pmt, &cx_media_type ); } STDMETHODIMP CapturePin::QueryPinInfo( PIN_INFO * pInfo ) { @@ -402,8 +560,7 @@ STDMETHODIMP CapturePin::QueryPinInfo( PIN_INFO * pInfo ) pInfo->pFilter = p_filter; if( p_filter ) p_filter->AddRef(); - pInfo->achName[0] = L'\0'; - + memcpy(pInfo->achName, PIN_NAME, sizeof(PIN_NAME)); pInfo->dir = PINDIR_INPUT; return NOERROR; @@ -422,21 +579,76 @@ STDMETHODIMP CapturePin::QueryId( LPWSTR * Id ) #ifdef DEBUG_DSHOW msg_Dbg( p_input, "CapturePin::QueryId" ); #endif - return E_NOTIMPL; + + *Id = L"VideoLAN Capture Pin"; + + return S_OK; } STDMETHODIMP CapturePin::QueryAccept( const AM_MEDIA_TYPE *pmt ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CapturePin::QueryAccept" ); -#endif - if( media_type.subtype == pmt->subtype && - media_type.majortype == pmt->majortype ) return S_OK; + if( State_Stopped != p_filter->state ) + { + msg_Dbg( p_input, "CapturePin::QueryAccept [not stopped]" ); + return S_FALSE; + } - return S_FALSE; + if( media_types[0].majortype != pmt->majortype ) + { + msg_Dbg( p_input, "CapturePin::QueryAccept [media type mismatch]" ); + return S_FALSE; + } + + int i_fourcc = GetFourCCFromMediaType(*pmt); + if( !i_fourcc ) + { + msg_Dbg( p_input, "CapturePin::QueryAccept " + "[media type not supported]" ); + return S_FALSE; + } + + if( pmt->majortype == MEDIATYPE_Video ) + { + if( pmt->pbFormat && + ( (((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight == 0) || + (((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth == 0) ) ) + { + msg_Dbg( p_input, "CapturePin::QueryAccept [video size wxh == 0]"); + return S_FALSE; + } + + msg_Dbg( p_input, "CapturePin::QueryAccept [OK] " + "(width=%ld, height=%ld, chroma=%4.4s, fps=%f)", + ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biWidth, + ((VIDEOINFOHEADER *)pmt->pbFormat)->bmiHeader.biHeight, + (char *)&i_fourcc, + 10000000.0f/((float)((VIDEOINFOHEADER *)pmt->pbFormat)->AvgTimePerFrame) ); + } + else if( pmt->majortype == MEDIATYPE_Audio ) + { + msg_Dbg( p_input, "CapturePin::QueryAccept [OK] (channels=%d, " + "samples/sec=%lu, bits/samples=%d, format=%4.4s)", + ((WAVEFORMATEX *)pmt->pbFormat)->nChannels, + ((WAVEFORMATEX *)pmt->pbFormat)->nSamplesPerSec, + ((WAVEFORMATEX *)pmt->pbFormat)->wBitsPerSample, + (char *)&i_fourcc ); + } + else + { + msg_Dbg( p_input, "CapturePin::QueryAccept [OK] (stream format=%4.4s)", + (char *)&i_fourcc ); + } + + if( p_connected_pin ) + { + FreeMediaType( cx_media_type ); + CopyMediaType( &cx_media_type, pmt ); + } + + return S_OK; } STDMETHODIMP CapturePin::EnumMediaTypes( IEnumMediaTypes **ppEnum ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::EnumMediaTypes" ); #endif @@ -448,7 +660,7 @@ STDMETHODIMP CapturePin::EnumMediaTypes( IEnumMediaTypes **ppEnum ) } STDMETHODIMP CapturePin::QueryInternalConnections( IPin* *apPin, ULONG *nPin ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CapturePin::QueryInternalConnections" ); #endif return E_NOTIMPL; @@ -472,6 +684,18 @@ STDMETHODIMP CapturePin::EndFlush( void ) #ifdef DEBUG_DSHOW msg_Dbg( p_input, "CapturePin::EndFlush" ); #endif + + VLCMediaSample vlc_sample; + + vlc_mutex_lock( &p_sys->lock ); + while( samples_queue.size() ) + { + vlc_sample = samples_queue.back(); + samples_queue.pop_back(); + vlc_sample.p_sample->Release(); + } + vlc_mutex_unlock( &p_sys->lock ); + return S_OK; } STDMETHODIMP CapturePin::NewSegment( REFERENCE_TIME tStart, @@ -520,7 +744,6 @@ STDMETHODIMP CapturePin::Receive( IMediaSample *pSample ) mtime_t i_timestamp = mdate() * 10; VLCMediaSample vlc_sample = {pSample, i_timestamp}; - access_sys_t *p_sys = p_input->p_access_data; vlc_mutex_lock( &p_sys->lock ); samples_queue.push_front( vlc_sample ); @@ -565,10 +788,11 @@ STDMETHODIMP CapturePin::ReceiveCanBlock( void ) /**************************************************************************** * Implementation of our dummy directshow filter class ****************************************************************************/ - -CaptureFilter::CaptureFilter( input_thread_t * _p_input, AM_MEDIA_TYPE mt ) - : p_input( _p_input ), p_pin( new CapturePin( _p_input, this, mt ) ), - media_type( mt ), state( State_Stopped ), i_ref( 1 ) +CaptureFilter::CaptureFilter( vlc_object_t *_p_input, access_sys_t *p_sys, + AM_MEDIA_TYPE *mt, size_t mt_count ) + : p_input( _p_input ), + p_pin( new CapturePin( _p_input, p_sys, this, mt, mt_count ) ), + state( State_Stopped ), i_ref( 1 ) { } @@ -583,7 +807,7 @@ CaptureFilter::~CaptureFilter() /* IUnknown methods */ STDMETHODIMP CaptureFilter::QueryInterface( REFIID riid, void **ppv ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureFilter::QueryInterface" ); #endif @@ -613,13 +837,22 @@ STDMETHODIMP CaptureFilter::QueryInterface( REFIID riid, void **ppv ) } else { +#ifdef DEBUG_DSHOW_L1 + msg_Dbg( p_input, "CaptureFilter::QueryInterface() failed for: " + "%04X-%02X-%02X-%02X%02X%02X%02X%02X%02X%02X%02X", + (int)riid.Data1, (int)riid.Data2, (int)riid.Data3, + static_cast(riid.Data4[0]), (int)riid.Data4[1], + (int)riid.Data4[2], (int)riid.Data4[3], + (int)riid.Data4[4], (int)riid.Data4[5], + (int)riid.Data4[6], (int)riid.Data4[7] ); +#endif *ppv = NULL; return E_NOINTERFACE; } }; STDMETHODIMP_(ULONG) CaptureFilter::AddRef() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureFilter::AddRef (ref: %i)", i_ref ); #endif @@ -627,7 +860,7 @@ STDMETHODIMP_(ULONG) CaptureFilter::AddRef() }; STDMETHODIMP_(ULONG) CaptureFilter::Release() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureFilter::Release (ref: %i)", i_ref ); #endif @@ -678,6 +911,8 @@ STDMETHODIMP CaptureFilter::Stop() msg_Dbg( p_input, "CaptureFilter::Stop" ); #endif + p_pin->EndFlush(); + state = State_Stopped; return S_OK; }; @@ -724,7 +959,7 @@ STDMETHODIMP CaptureFilter::QueryFilterInfo( FILTER_INFO * pInfo ) msg_Dbg( p_input, "CaptureFilter::QueryFilterInfo" ); #endif - pInfo->achName[0] = L'\0'; + memcpy(pInfo->achName, FILTER_NAME, sizeof(FILTER_NAME)); pInfo->pGraph = p_graph; if( p_graph ) p_graph->AddRef(); @@ -760,7 +995,7 @@ CapturePin *CaptureFilter::CustomGetPin() * Implementation of our dummy directshow enumpins class ****************************************************************************/ -CaptureEnumPins::CaptureEnumPins( input_thread_t * _p_input, +CaptureEnumPins::CaptureEnumPins( vlc_object_t *_p_input, CaptureFilter *_p_filter, CaptureEnumPins *pEnumPins ) : p_input( _p_input ), p_filter( _p_filter ), i_ref( 1 ) @@ -782,7 +1017,7 @@ CaptureEnumPins::CaptureEnumPins( input_thread_t * _p_input, CaptureEnumPins::~CaptureEnumPins() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::~CaptureEnumPins" ); #endif p_filter->Release(); @@ -791,7 +1026,7 @@ CaptureEnumPins::~CaptureEnumPins() /* IUnknown methods */ STDMETHODIMP CaptureEnumPins::QueryInterface( REFIID riid, void **ppv ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::QueryInterface" ); #endif @@ -810,7 +1045,7 @@ STDMETHODIMP CaptureEnumPins::QueryInterface( REFIID riid, void **ppv ) }; STDMETHODIMP_(ULONG) CaptureEnumPins::AddRef() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::AddRef (ref: %i)", i_ref ); #endif @@ -818,7 +1053,7 @@ STDMETHODIMP_(ULONG) CaptureEnumPins::AddRef() }; STDMETHODIMP_(ULONG) CaptureEnumPins::Release() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::Release (ref: %i)", i_ref ); #endif @@ -831,7 +1066,7 @@ STDMETHODIMP_(ULONG) CaptureEnumPins::Release() STDMETHODIMP CaptureEnumPins::Next( ULONG cPins, IPin ** ppPins, ULONG * pcFetched ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::Next" ); #endif @@ -852,7 +1087,7 @@ STDMETHODIMP CaptureEnumPins::Next( ULONG cPins, IPin ** ppPins, }; STDMETHODIMP CaptureEnumPins::Skip( ULONG cPins ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::Skip" ); #endif @@ -867,7 +1102,7 @@ STDMETHODIMP CaptureEnumPins::Skip( ULONG cPins ) }; STDMETHODIMP CaptureEnumPins::Reset() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::Reset" ); #endif @@ -876,7 +1111,7 @@ STDMETHODIMP CaptureEnumPins::Reset() }; STDMETHODIMP CaptureEnumPins::Clone( IEnumPins **ppEnum ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumPins::Clone" ); #endif @@ -889,10 +1124,8 @@ STDMETHODIMP CaptureEnumPins::Clone( IEnumPins **ppEnum ) /**************************************************************************** * Implementation of our dummy directshow enummediatypes class ****************************************************************************/ - -CaptureEnumMediaTypes::CaptureEnumMediaTypes( input_thread_t * _p_input, - CapturePin *_p_pin, - CaptureEnumMediaTypes *pEnumMediaTypes ) +CaptureEnumMediaTypes::CaptureEnumMediaTypes( vlc_object_t *_p_input, + CapturePin *_p_pin, CaptureEnumMediaTypes *pEnumMediaTypes ) : p_input( _p_input ), p_pin( _p_pin ), i_ref( 1 ) { /* Hold a reference count on our filter */ @@ -901,26 +1134,29 @@ CaptureEnumMediaTypes::CaptureEnumMediaTypes( input_thread_t * _p_input, /* Are we creating a new enumerator */ if( pEnumMediaTypes == NULL ) { + CopyMediaType(&cx_media_type, &p_pin->cx_media_type); i_position = 0; } else { + CopyMediaType(&cx_media_type, &pEnumMediaTypes->cx_media_type); i_position = pEnumMediaTypes->i_position; } } CaptureEnumMediaTypes::~CaptureEnumMediaTypes() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::~CaptureEnumMediaTypes" ); #endif + FreeMediaType(cx_media_type); p_pin->Release(); } /* IUnknown methods */ STDMETHODIMP CaptureEnumMediaTypes::QueryInterface( REFIID riid, void **ppv ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::QueryInterface" ); #endif @@ -939,7 +1175,7 @@ STDMETHODIMP CaptureEnumMediaTypes::QueryInterface( REFIID riid, void **ppv ) }; STDMETHODIMP_(ULONG) CaptureEnumMediaTypes::AddRef() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::AddRef (ref: %i)", i_ref ); #endif @@ -947,7 +1183,7 @@ STDMETHODIMP_(ULONG) CaptureEnumMediaTypes::AddRef() }; STDMETHODIMP_(ULONG) CaptureEnumMediaTypes::Release() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::Release (ref: %i)", i_ref ); #endif @@ -961,50 +1197,81 @@ STDMETHODIMP CaptureEnumMediaTypes::Next( ULONG cMediaTypes, AM_MEDIA_TYPE ** ppMediaTypes, ULONG * pcFetched ) { -#ifdef DEBUG_DSHOW - msg_Dbg( p_input, "CaptureEnumMediaTypes::Next" ); +#ifdef DEBUG_DSHOW_L1 + msg_Dbg( p_input, "CaptureEnumMediaTypes::Next " ); #endif + ULONG copied = 0; + ULONG offset = 0; + ULONG max = p_pin->media_type_count; + + if( ! ppMediaTypes ) + return E_POINTER; - if( pcFetched ) *pcFetched = 0; + if( (! pcFetched) && (cMediaTypes > 1) ) + return E_POINTER; - if( cMediaTypes > 0 ) + /* + ** use connection media type as first entry in iterator if it exists + */ + copied = 0; + if( cx_media_type.subtype != GUID_NULL ) { - if( 0 == i_position++ ) + ++max; + if( i_position == 0 ) { - AM_MEDIA_TYPE mediaType = p_pin->CustomGetMediaType(); - ppMediaTypes[0] = (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE)); - if( CopyMediaType(ppMediaTypes[0], &mediaType) != S_OK ) + ppMediaTypes[copied] = + (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE)); + if( CopyMediaType(ppMediaTypes[copied], &cx_media_type) != S_OK ) return E_OUTOFMEMORY; - - if( pcFetched ) *pcFetched = 1; - if( cMediaTypes == 1 ) return S_OK; + ++i_position; + ++copied; } + } - return S_FALSE; + while( (copied < cMediaTypes) && (i_position < max) ) + { + ppMediaTypes[copied] = + (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE)); + if( CopyMediaType( ppMediaTypes[copied], + &p_pin->media_types[i_position-offset]) != S_OK ) + return E_OUTOFMEMORY; + + ++copied; + ++i_position; } - return S_OK; + if( pcFetched ) *pcFetched = copied; + + return (copied == cMediaTypes) ? S_OK : S_FALSE; }; STDMETHODIMP CaptureEnumMediaTypes::Skip( ULONG cMediaTypes ) { -#ifdef DEBUG_DSHOW + ULONG max = p_pin->media_type_count; + if( cx_media_type.subtype != GUID_NULL ) + { + max = 1; + } +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::Skip" ); #endif - return S_FALSE; + i_position += cMediaTypes; + return (i_position < max) ? S_OK : S_FALSE; }; STDMETHODIMP CaptureEnumMediaTypes::Reset() { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::Reset" ); #endif + FreeMediaType(cx_media_type); + CopyMediaType(&cx_media_type, &p_pin->cx_media_type); i_position = 0; return S_OK; }; STDMETHODIMP CaptureEnumMediaTypes::Clone( IEnumMediaTypes **ppEnum ) { -#ifdef DEBUG_DSHOW +#ifdef DEBUG_DSHOW_L1 msg_Dbg( p_input, "CaptureEnumMediaTypes::Clone" ); #endif