/* * DirectShow capture interface * Copyright (c) 2010 Ramiro Polla * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef AVDEVICE_DSHOW_CAPTURE_H #define AVDEVICE_DSHOW_CAPTURE_H #define DSHOWDEBUG 0 #include "avdevice.h" #define COBJMACROS #define WIN32_LEAN_AND_MEAN #include #define NO_DSHOW_STRSAFE #include #include #include "libavcodec/internal.h" #include "libavcodec/packet_internal.h" /* EC_DEVICE_LOST is not defined in MinGW dshow headers. */ #ifndef EC_DEVICE_LOST #define EC_DEVICE_LOST 0x1f #endif long ff_copy_dshow_media_type(AM_MEDIA_TYPE *dst, const AM_MEDIA_TYPE *src); void ff_print_VIDEO_STREAM_CONFIG_CAPS(const VIDEO_STREAM_CONFIG_CAPS *caps); void ff_print_AUDIO_STREAM_CONFIG_CAPS(const AUDIO_STREAM_CONFIG_CAPS *caps); void ff_print_AM_MEDIA_TYPE(const AM_MEDIA_TYPE *type); void ff_printGUID(const GUID *g); extern const AVClass *ff_dshow_context_class_ptr; #define dshowdebug(...) ff_dlog(&ff_dshow_context_class_ptr, __VA_ARGS__) static inline void nothing(void *foo) { } struct GUIDoffset { const GUID *iid; int offset; }; enum dshowDeviceType { VideoDevice = 0, AudioDevice = 1, }; enum dshowSourceFilterType { VideoSourceDevice = 0, AudioSourceDevice = 1, }; #define DECLARE_QUERYINTERFACE(prefix, class, ...) \ long WINAPI \ ff_dshow_##prefix##_QueryInterface(class *this, const GUID *riid, void **ppvObject) \ { \ struct GUIDoffset ifaces[] = __VA_ARGS__; \ int i; \ dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_QueryInterface(%p, %p, %p)\n", this, riid, ppvObject); \ ff_printGUID(riid); \ if (!ppvObject) \ return E_POINTER; \ for (i = 0; i < sizeof(ifaces)/sizeof(ifaces[0]); i++) { \ if (IsEqualGUID(riid, ifaces[i].iid)) { \ void *obj = (void *) ((uint8_t *) this + ifaces[i].offset); \ ff_dshow_##prefix##_AddRef(this); \ dshowdebug("\tfound %d with offset %d\n", i, ifaces[i].offset); \ *ppvObject = (void *) obj; \ return S_OK; \ } \ } \ dshowdebug("\tE_NOINTERFACE\n"); \ *ppvObject = NULL; \ return E_NOINTERFACE; \ } #define DECLARE_ADDREF(prefix, class) \ unsigned long WINAPI \ ff_dshow_##prefix##_AddRef(class *this) \ { \ dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_AddRef(%p)\t%ld\n", this, this->ref+1); \ return InterlockedIncrement(&this->ref); \ } #define DECLARE_RELEASE(prefix, class) \ unsigned long WINAPI \ ff_dshow_##prefix##_Release(class *this) \ { \ long ref = InterlockedDecrement(&this->ref); \ dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Release(%p)\t%ld\n", this, ref); \ if (!ref) \ ff_dshow_##prefix##_Destroy(this); \ return ref; \ } #define DECLARE_DESTROY(prefix, class, func) \ void ff_dshow_##prefix##_Destroy(class *this) \ { \ dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Destroy(%p)\n", this); \ func(this); \ if (this) { \ if (this->vtbl) \ CoTaskMemFree(this->vtbl); \ CoTaskMemFree(this); \ } \ } #define DECLARE_CREATE(prefix, class, setup, ...) \ class *ff_dshow_##prefix##_Create(__VA_ARGS__) \ { \ class *this = CoTaskMemAlloc(sizeof(class)); \ void *vtbl = CoTaskMemAlloc(sizeof(*this->vtbl)); \ dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Create(%p)\n", this); \ if (!this || !vtbl) \ goto fail; \ ZeroMemory(this, sizeof(class)); \ ZeroMemory(vtbl, sizeof(*this->vtbl)); \ this->ref = 1; \ this->vtbl = vtbl; \ if (!setup) \ goto fail; \ dshowdebug("created ff_dshow_"AV_STRINGIFY(prefix)" %p\n", this); \ return this; \ fail: \ ff_dshow_##prefix##_Destroy(this); \ dshowdebug("could not create ff_dshow_"AV_STRINGIFY(prefix)"\n"); \ return NULL; \ } #define SETVTBL(vtbl, prefix, fn) \ do { (vtbl)->fn = (void *) ff_dshow_##prefix##_##fn; } while(0) /***************************************************************************** * Forward Declarations ****************************************************************************/ typedef struct DShowPin DShowPin; typedef struct DShowMemInputPin DShowMemInputPin; typedef struct DShowEnumPins DShowEnumPins; typedef struct DShowEnumMediaTypes DShowEnumMediaTypes; typedef struct DShowFilter DShowFilter; /***************************************************************************** * DShowPin ****************************************************************************/ struct DShowPin { IPinVtbl *vtbl; long ref; DShowFilter *filter; IPin *connectedto; AM_MEDIA_TYPE type; IMemInputPinVtbl *imemvtbl; }; long WINAPI ff_dshow_pin_QueryInterface (DShowPin *, const GUID *, void **); unsigned long WINAPI ff_dshow_pin_AddRef (DShowPin *); unsigned long WINAPI ff_dshow_pin_Release (DShowPin *); long WINAPI ff_dshow_pin_Connect (DShowPin *, IPin *, const AM_MEDIA_TYPE *); long WINAPI ff_dshow_pin_ReceiveConnection (DShowPin *, IPin *, const AM_MEDIA_TYPE *); long WINAPI ff_dshow_pin_Disconnect (DShowPin *); long WINAPI ff_dshow_pin_ConnectedTo (DShowPin *, IPin **); long WINAPI ff_dshow_pin_ConnectionMediaType (DShowPin *, AM_MEDIA_TYPE *); long WINAPI ff_dshow_pin_QueryPinInfo (DShowPin *, PIN_INFO *); long WINAPI ff_dshow_pin_QueryDirection (DShowPin *, PIN_DIRECTION *); long WINAPI ff_dshow_pin_QueryId (DShowPin *, wchar_t **); long WINAPI ff_dshow_pin_QueryAccept (DShowPin *, const AM_MEDIA_TYPE *); long WINAPI ff_dshow_pin_EnumMediaTypes (DShowPin *, IEnumMediaTypes **); long WINAPI ff_dshow_pin_QueryInternalConnections(DShowPin *, IPin **, unsigned long *); long WINAPI ff_dshow_pin_EndOfStream (DShowPin *); long WINAPI ff_dshow_pin_BeginFlush (DShowPin *); long WINAPI ff_dshow_pin_EndFlush (DShowPin *); long WINAPI ff_dshow_pin_NewSegment (DShowPin *, REFERENCE_TIME, REFERENCE_TIME, double); long WINAPI ff_dshow_meminputpin_QueryInterface (DShowMemInputPin *, const GUID *, void **); unsigned long WINAPI ff_dshow_meminputpin_AddRef (DShowMemInputPin *); unsigned long WINAPI ff_dshow_meminputpin_Release (DShowMemInputPin *); long WINAPI ff_dshow_meminputpin_GetAllocator (DShowMemInputPin *, IMemAllocator **); long WINAPI ff_dshow_meminputpin_NotifyAllocator (DShowMemInputPin *, IMemAllocator *, BOOL); long WINAPI ff_dshow_meminputpin_GetAllocatorRequirements(DShowMemInputPin *, ALLOCATOR_PROPERTIES *); long WINAPI ff_dshow_meminputpin_Receive (DShowMemInputPin *, IMediaSample *); long WINAPI ff_dshow_meminputpin_ReceiveMultiple (DShowMemInputPin *, IMediaSample **, long, long *); long WINAPI ff_dshow_meminputpin_ReceiveCanBlock (DShowMemInputPin *); void ff_dshow_pin_Destroy(DShowPin *); DShowPin *ff_dshow_pin_Create (DShowFilter *filter); void ff_dshow_meminputpin_Destroy(DShowMemInputPin *); /***************************************************************************** * DShowEnumPins ****************************************************************************/ struct DShowEnumPins { IEnumPinsVtbl *vtbl; long ref; int pos; DShowPin *pin; DShowFilter *filter; }; long WINAPI ff_dshow_enumpins_QueryInterface(DShowEnumPins *, const GUID *, void **); unsigned long WINAPI ff_dshow_enumpins_AddRef (DShowEnumPins *); unsigned long WINAPI ff_dshow_enumpins_Release (DShowEnumPins *); long WINAPI ff_dshow_enumpins_Next (DShowEnumPins *, unsigned long, IPin **, unsigned long *); long WINAPI ff_dshow_enumpins_Skip (DShowEnumPins *, unsigned long); long WINAPI ff_dshow_enumpins_Reset (DShowEnumPins *); long WINAPI ff_dshow_enumpins_Clone (DShowEnumPins *, DShowEnumPins **); void ff_dshow_enumpins_Destroy(DShowEnumPins *); DShowEnumPins *ff_dshow_enumpins_Create (DShowPin *pin, DShowFilter *filter); /***************************************************************************** * DShowEnumMediaTypes ****************************************************************************/ struct DShowEnumMediaTypes { IEnumMediaTypesVtbl *vtbl; long ref; int pos; AM_MEDIA_TYPE type; }; long WINAPI ff_dshow_enummediatypes_QueryInterface(DShowEnumMediaTypes *, const GUID *, void **); unsigned long WINAPI ff_dshow_enummediatypes_AddRef (DShowEnumMediaTypes *); unsigned long WINAPI ff_dshow_enummediatypes_Release (DShowEnumMediaTypes *); long WINAPI ff_dshow_enummediatypes_Next (DShowEnumMediaTypes *, unsigned long, AM_MEDIA_TYPE **, unsigned long *); long WINAPI ff_dshow_enummediatypes_Skip (DShowEnumMediaTypes *, unsigned long); long WINAPI ff_dshow_enummediatypes_Reset (DShowEnumMediaTypes *); long WINAPI ff_dshow_enummediatypes_Clone (DShowEnumMediaTypes *, DShowEnumMediaTypes **); void ff_dshow_enummediatypes_Destroy(DShowEnumMediaTypes *); DShowEnumMediaTypes *ff_dshow_enummediatypes_Create(const AM_MEDIA_TYPE *type); /***************************************************************************** * DShowFilter ****************************************************************************/ struct DShowFilter { IBaseFilterVtbl *vtbl; long ref; const wchar_t *name; DShowPin *pin; FILTER_INFO info; FILTER_STATE state; IReferenceClock *clock; enum dshowDeviceType type; void *priv_data; int stream_index; int64_t start_time; void (*callback)(void *priv_data, int index, uint8_t *buf, int buf_size, int64_t time, enum dshowDeviceType type); }; long WINAPI ff_dshow_filter_QueryInterface (DShowFilter *, const GUID *, void **); unsigned long WINAPI ff_dshow_filter_AddRef (DShowFilter *); unsigned long WINAPI ff_dshow_filter_Release (DShowFilter *); long WINAPI ff_dshow_filter_GetClassID (DShowFilter *, CLSID *); long WINAPI ff_dshow_filter_Stop (DShowFilter *); long WINAPI ff_dshow_filter_Pause (DShowFilter *); long WINAPI ff_dshow_filter_Run (DShowFilter *, REFERENCE_TIME); long WINAPI ff_dshow_filter_GetState (DShowFilter *, DWORD, FILTER_STATE *); long WINAPI ff_dshow_filter_SetSyncSource (DShowFilter *, IReferenceClock *); long WINAPI ff_dshow_filter_GetSyncSource (DShowFilter *, IReferenceClock **); long WINAPI ff_dshow_filter_EnumPins (DShowFilter *, IEnumPins **); long WINAPI ff_dshow_filter_FindPin (DShowFilter *, const wchar_t *, IPin **); long WINAPI ff_dshow_filter_QueryFilterInfo(DShowFilter *, FILTER_INFO *); long WINAPI ff_dshow_filter_JoinFilterGraph(DShowFilter *, IFilterGraph *, const wchar_t *); long WINAPI ff_dshow_filter_QueryVendorInfo(DShowFilter *, wchar_t **); void ff_dshow_filter_Destroy(DShowFilter *); DShowFilter *ff_dshow_filter_Create (void *, void *, enum dshowDeviceType); /***************************************************************************** * dshow_ctx ****************************************************************************/ struct dshow_ctx { const AVClass *class; IGraphBuilder *graph; char *device_name[2]; char *device_unique_name[2]; int video_device_number; int audio_device_number; int list_options; int list_devices; int audio_buffer_size; int crossbar_video_input_pin_number; int crossbar_audio_input_pin_number; char *video_pin_name; char *audio_pin_name; int show_video_device_dialog; int show_audio_device_dialog; int show_video_crossbar_connection_dialog; int show_audio_crossbar_connection_dialog; int show_analog_tv_tuner_dialog; int show_analog_tv_tuner_audio_dialog; char *audio_filter_load_file; char *audio_filter_save_file; char *video_filter_load_file; char *video_filter_save_file; int use_video_device_timestamps; IBaseFilter *device_filter[2]; IPin *device_pin[2]; DShowFilter *capture_filter[2]; DShowPin *capture_pin[2]; HANDLE mutex; HANDLE event[2]; /* event[0] is set by DirectShow * event[1] is set by callback() */ PacketListEntry *pktl; int eof; int64_t curbufsize[2]; unsigned int video_frame_num; IMediaControl *control; IMediaEvent *media_event; enum AVPixelFormat pixel_format; enum AVCodecID video_codec_id; char *framerate; int requested_width; int requested_height; AVRational requested_framerate; int sample_rate; int sample_size; int channels; }; /***************************************************************************** * CrossBar ****************************************************************************/ HRESULT ff_dshow_try_setup_crossbar_options(ICaptureGraphBuilder2 *graph_builder2, IBaseFilter *device_filter, enum dshowDeviceType devtype, AVFormatContext *avctx); void ff_dshow_show_filter_properties(IBaseFilter *pFilter, AVFormatContext *avctx); #endif /* AVDEVICE_DSHOW_CAPTURE_H */