mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
355 lines
17 KiB
355 lines
17 KiB
/* |
|
* 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 |
|
#include <windows.h> |
|
#define NO_DSHOW_STRSAFE |
|
#include <dshow.h> |
|
#include <dvdmedia.h> |
|
|
|
#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 */
|
|
|