/* * 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" /* 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(class, ...) \ long WINAPI \ class##_QueryInterface(class *this, const GUID *riid, void **ppvObject) \ { \ struct GUIDoffset ifaces[] = __VA_ARGS__; \ int i; \ dshowdebug(AV_STRINGIFY(class)"_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); \ class##_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(class) \ unsigned long WINAPI \ class##_AddRef(class *this) \ { \ dshowdebug(AV_STRINGIFY(class)"_AddRef(%p)\t%ld\n", this, this->ref+1); \ return InterlockedIncrement(&this->ref); \ } #define DECLARE_RELEASE(class) \ unsigned long WINAPI \ class##_Release(class *this) \ { \ long ref = InterlockedDecrement(&this->ref); \ dshowdebug(AV_STRINGIFY(class)"_Release(%p)\t%ld\n", this, ref); \ if (!ref) \ class##_Destroy(this); \ return ref; \ } #define DECLARE_DESTROY(class, func) \ void class##_Destroy(class *this) \ { \ dshowdebug(AV_STRINGIFY(class)"_Destroy(%p)\n", this); \ func(this); \ if (this) { \ if (this->vtbl) \ CoTaskMemFree(this->vtbl); \ CoTaskMemFree(this); \ } \ } #define DECLARE_CREATE(class, setup, ...) \ class *class##_Create(__VA_ARGS__) \ { \ class *this = CoTaskMemAlloc(sizeof(class)); \ void *vtbl = CoTaskMemAlloc(sizeof(*this->vtbl)); \ dshowdebug(AV_STRINGIFY(class)"_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 "AV_STRINGIFY(class)" %p\n", this); \ return this; \ fail: \ class##_Destroy(this); \ dshowdebug("could not create "AV_STRINGIFY(class)"\n"); \ return NULL; \ } #define SETVTBL(vtbl, class, fn) \ do { (vtbl)->fn = (void *) class##_##fn; } while(0) /***************************************************************************** * Forward Declarations ****************************************************************************/ typedef struct libAVPin libAVPin; typedef struct libAVMemInputPin libAVMemInputPin; typedef struct libAVEnumPins libAVEnumPins; typedef struct libAVEnumMediaTypes libAVEnumMediaTypes; typedef struct libAVFilter libAVFilter; /***************************************************************************** * libAVPin ****************************************************************************/ struct libAVPin { IPinVtbl *vtbl; long ref; libAVFilter *filter; IPin *connectedto; AM_MEDIA_TYPE type; IMemInputPinVtbl *imemvtbl; }; long WINAPI libAVPin_QueryInterface (libAVPin *, const GUID *, void **); unsigned long WINAPI libAVPin_AddRef (libAVPin *); unsigned long WINAPI libAVPin_Release (libAVPin *); long WINAPI libAVPin_Connect (libAVPin *, IPin *, const AM_MEDIA_TYPE *); long WINAPI libAVPin_ReceiveConnection (libAVPin *, IPin *, const AM_MEDIA_TYPE *); long WINAPI libAVPin_Disconnect (libAVPin *); long WINAPI libAVPin_ConnectedTo (libAVPin *, IPin **); long WINAPI libAVPin_ConnectionMediaType (libAVPin *, AM_MEDIA_TYPE *); long WINAPI libAVPin_QueryPinInfo (libAVPin *, PIN_INFO *); long WINAPI libAVPin_QueryDirection (libAVPin *, PIN_DIRECTION *); long WINAPI libAVPin_QueryId (libAVPin *, wchar_t **); long WINAPI libAVPin_QueryAccept (libAVPin *, const AM_MEDIA_TYPE *); long WINAPI libAVPin_EnumMediaTypes (libAVPin *, IEnumMediaTypes **); long WINAPI libAVPin_QueryInternalConnections(libAVPin *, IPin **, unsigned long *); long WINAPI libAVPin_EndOfStream (libAVPin *); long WINAPI libAVPin_BeginFlush (libAVPin *); long WINAPI libAVPin_EndFlush (libAVPin *); long WINAPI libAVPin_NewSegment (libAVPin *, REFERENCE_TIME, REFERENCE_TIME, double); long WINAPI libAVMemInputPin_QueryInterface (libAVMemInputPin *, const GUID *, void **); unsigned long WINAPI libAVMemInputPin_AddRef (libAVMemInputPin *); unsigned long WINAPI libAVMemInputPin_Release (libAVMemInputPin *); long WINAPI libAVMemInputPin_GetAllocator (libAVMemInputPin *, IMemAllocator **); long WINAPI libAVMemInputPin_NotifyAllocator (libAVMemInputPin *, IMemAllocator *, BOOL); long WINAPI libAVMemInputPin_GetAllocatorRequirements(libAVMemInputPin *, ALLOCATOR_PROPERTIES *); long WINAPI libAVMemInputPin_Receive (libAVMemInputPin *, IMediaSample *); long WINAPI libAVMemInputPin_ReceiveMultiple (libAVMemInputPin *, IMediaSample **, long, long *); long WINAPI libAVMemInputPin_ReceiveCanBlock (libAVMemInputPin *); void libAVPin_Destroy(libAVPin *); libAVPin *libAVPin_Create (libAVFilter *filter); void libAVMemInputPin_Destroy(libAVMemInputPin *); /***************************************************************************** * libAVEnumPins ****************************************************************************/ struct libAVEnumPins { IEnumPinsVtbl *vtbl; long ref; int pos; libAVPin *pin; libAVFilter *filter; }; long WINAPI libAVEnumPins_QueryInterface(libAVEnumPins *, const GUID *, void **); unsigned long WINAPI libAVEnumPins_AddRef (libAVEnumPins *); unsigned long WINAPI libAVEnumPins_Release (libAVEnumPins *); long WINAPI libAVEnumPins_Next (libAVEnumPins *, unsigned long, IPin **, unsigned long *); long WINAPI libAVEnumPins_Skip (libAVEnumPins *, unsigned long); long WINAPI libAVEnumPins_Reset (libAVEnumPins *); long WINAPI libAVEnumPins_Clone (libAVEnumPins *, libAVEnumPins **); void libAVEnumPins_Destroy(libAVEnumPins *); libAVEnumPins *libAVEnumPins_Create (libAVPin *pin, libAVFilter *filter); /***************************************************************************** * libAVEnumMediaTypes ****************************************************************************/ struct libAVEnumMediaTypes { IEnumMediaTypesVtbl *vtbl; long ref; int pos; AM_MEDIA_TYPE type; }; long WINAPI libAVEnumMediaTypes_QueryInterface(libAVEnumMediaTypes *, const GUID *, void **); unsigned long WINAPI libAVEnumMediaTypes_AddRef (libAVEnumMediaTypes *); unsigned long WINAPI libAVEnumMediaTypes_Release (libAVEnumMediaTypes *); long WINAPI libAVEnumMediaTypes_Next (libAVEnumMediaTypes *, unsigned long, AM_MEDIA_TYPE **, unsigned long *); long WINAPI libAVEnumMediaTypes_Skip (libAVEnumMediaTypes *, unsigned long); long WINAPI libAVEnumMediaTypes_Reset (libAVEnumMediaTypes *); long WINAPI libAVEnumMediaTypes_Clone (libAVEnumMediaTypes *, libAVEnumMediaTypes **); void libAVEnumMediaTypes_Destroy(libAVEnumMediaTypes *); libAVEnumMediaTypes *libAVEnumMediaTypes_Create(const AM_MEDIA_TYPE *type); /***************************************************************************** * libAVFilter ****************************************************************************/ struct libAVFilter { IBaseFilterVtbl *vtbl; long ref; const wchar_t *name; libAVPin *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 libAVFilter_QueryInterface (libAVFilter *, const GUID *, void **); unsigned long WINAPI libAVFilter_AddRef (libAVFilter *); unsigned long WINAPI libAVFilter_Release (libAVFilter *); long WINAPI libAVFilter_GetClassID (libAVFilter *, CLSID *); long WINAPI libAVFilter_Stop (libAVFilter *); long WINAPI libAVFilter_Pause (libAVFilter *); long WINAPI libAVFilter_Run (libAVFilter *, REFERENCE_TIME); long WINAPI libAVFilter_GetState (libAVFilter *, DWORD, FILTER_STATE *); long WINAPI libAVFilter_SetSyncSource (libAVFilter *, IReferenceClock *); long WINAPI libAVFilter_GetSyncSource (libAVFilter *, IReferenceClock **); long WINAPI libAVFilter_EnumPins (libAVFilter *, IEnumPins **); long WINAPI libAVFilter_FindPin (libAVFilter *, const wchar_t *, IPin **); long WINAPI libAVFilter_QueryFilterInfo(libAVFilter *, FILTER_INFO *); long WINAPI libAVFilter_JoinFilterGraph(libAVFilter *, IFilterGraph *, const wchar_t *); long WINAPI libAVFilter_QueryVendorInfo(libAVFilter *, wchar_t **); void libAVFilter_Destroy(libAVFilter *); libAVFilter *libAVFilter_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; IBaseFilter *device_filter[2]; IPin *device_pin[2]; libAVFilter *capture_filter[2]; libAVPin *capture_pin[2]; HANDLE mutex; HANDLE event[2]; /* event[0] is set by DirectShow * event[1] is set by callback() */ AVPacketList *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 dshow_try_setup_crossbar_options(ICaptureGraphBuilder2 *graph_builder2, IBaseFilter *device_filter, enum dshowDeviceType devtype, AVFormatContext *avctx); void dshow_show_filter_properties(IBaseFilter *pFilter, AVFormatContext *avctx); #endif /* AVDEVICE_DSHOW_CAPTURE_H */