1 /*
2 * DirectShow capture interface
3 * Copyright (c) 2010 Ramiro Polla
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #ifndef AVDEVICE_DSHOW_CAPTURE_H
23 #define AVDEVICE_DSHOW_CAPTURE_H
24
25 #define DSHOWDEBUG 0
26
27 #include "avdevice.h"
28
29 #define COBJMACROS
30 #define WIN32_LEAN_AND_MEAN
31 #include <windows.h>
32 #define NO_DSHOW_STRSAFE
33 #include <dshow.h>
34 #include <dvdmedia.h>
35
36 #include "libavcodec/internal.h"
37 #include "libavcodec/packet_internal.h"
38
39 /* EC_DEVICE_LOST is not defined in MinGW dshow headers. */
40 #ifndef EC_DEVICE_LOST
41 #define EC_DEVICE_LOST 0x1f
42 #endif
43
44 long ff_copy_dshow_media_type(AM_MEDIA_TYPE *dst, const AM_MEDIA_TYPE *src);
45 void ff_print_VIDEO_STREAM_CONFIG_CAPS(const VIDEO_STREAM_CONFIG_CAPS *caps);
46 void ff_print_AUDIO_STREAM_CONFIG_CAPS(const AUDIO_STREAM_CONFIG_CAPS *caps);
47 void ff_print_AM_MEDIA_TYPE(const AM_MEDIA_TYPE *type);
48 void ff_printGUID(const GUID *g);
49
50 extern const AVClass *ff_dshow_context_class_ptr;
51 #define dshowdebug(...) ff_dlog(&ff_dshow_context_class_ptr, __VA_ARGS__)
52
nothing(void * foo)53 static inline void nothing(void *foo)
54 {
55 }
56
57 struct GUIDoffset {
58 const GUID *iid;
59 int offset;
60 };
61
62 enum dshowDeviceType {
63 VideoDevice = 0,
64 AudioDevice = 1,
65 };
66
67 enum dshowSourceFilterType {
68 VideoSourceDevice = 0,
69 AudioSourceDevice = 1,
70 };
71
72 #define DECLARE_QUERYINTERFACE(prefix, class, ...) \
73 long \
74 ff_dshow_##prefix##_QueryInterface(class *this, const GUID *riid, void **ppvObject) \
75 { \
76 struct GUIDoffset ifaces[] = __VA_ARGS__; \
77 int i; \
78 dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_QueryInterface(%p, %p, %p)\n", this, riid, ppvObject); \
79 ff_printGUID(riid); \
80 if (!ppvObject) \
81 return E_POINTER; \
82 for (i = 0; i < sizeof(ifaces)/sizeof(ifaces[0]); i++) { \
83 if (IsEqualGUID(riid, ifaces[i].iid)) { \
84 void *obj = (void *) ((uint8_t *) this + ifaces[i].offset); \
85 ff_dshow_##prefix##_AddRef(this); \
86 dshowdebug("\tfound %d with offset %d\n", i, ifaces[i].offset); \
87 *ppvObject = (void *) obj; \
88 return S_OK; \
89 } \
90 } \
91 dshowdebug("\tE_NOINTERFACE\n"); \
92 *ppvObject = NULL; \
93 return E_NOINTERFACE; \
94 }
95 #define DECLARE_ADDREF(prefix, class) \
96 unsigned long \
97 ff_dshow_##prefix##_AddRef(class *this) \
98 { \
99 dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_AddRef(%p)\t%ld\n", this, this->ref+1); \
100 return InterlockedIncrement(&this->ref); \
101 }
102 #define DECLARE_RELEASE(prefix, class) \
103 unsigned long \
104 ff_dshow_##prefix##_Release(class *this) \
105 { \
106 long ref = InterlockedDecrement(&this->ref); \
107 dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Release(%p)\t%ld\n", this, ref); \
108 if (!ref) \
109 ff_dshow_##prefix##_Destroy(this); \
110 return ref; \
111 }
112
113 #define DECLARE_DESTROY(prefix, class, func) \
114 void ff_dshow_##prefix##_Destroy(class *this) \
115 { \
116 dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Destroy(%p)\n", this); \
117 func(this); \
118 if (this) { \
119 if (this->vtbl) \
120 CoTaskMemFree(this->vtbl); \
121 CoTaskMemFree(this); \
122 } \
123 }
124 #define DECLARE_CREATE(prefix, class, setup, ...) \
125 class *ff_dshow_##prefix##_Create(__VA_ARGS__) \
126 { \
127 class *this = CoTaskMemAlloc(sizeof(class)); \
128 void *vtbl = CoTaskMemAlloc(sizeof(*this->vtbl)); \
129 dshowdebug("ff_dshow_"AV_STRINGIFY(prefix)"_Create(%p)\n", this); \
130 if (!this || !vtbl) \
131 goto fail; \
132 ZeroMemory(this, sizeof(class)); \
133 ZeroMemory(vtbl, sizeof(*this->vtbl)); \
134 this->ref = 1; \
135 this->vtbl = vtbl; \
136 if (!setup) \
137 goto fail; \
138 dshowdebug("created ff_dshow_"AV_STRINGIFY(prefix)" %p\n", this); \
139 return this; \
140 fail: \
141 ff_dshow_##prefix##_Destroy(this); \
142 dshowdebug("could not create ff_dshow_"AV_STRINGIFY(prefix)"\n"); \
143 return NULL; \
144 }
145
146 #define SETVTBL(vtbl, prefix, fn) \
147 do { (vtbl)->fn = (void *) ff_dshow_##prefix##_##fn; } while(0)
148
149 /*****************************************************************************
150 * Forward Declarations
151 ****************************************************************************/
152 typedef struct DShowPin DShowPin;
153 typedef struct DShowMemInputPin DShowMemInputPin;
154 typedef struct DShowEnumPins DShowEnumPins;
155 typedef struct DShowEnumMediaTypes DShowEnumMediaTypes;
156 typedef struct DShowFilter DShowFilter;
157
158 /*****************************************************************************
159 * DShowPin
160 ****************************************************************************/
161 struct DShowPin {
162 IPinVtbl *vtbl;
163 long ref;
164 DShowFilter *filter;
165 IPin *connectedto;
166 AM_MEDIA_TYPE type;
167 IMemInputPinVtbl *imemvtbl;
168 };
169
170 long ff_dshow_pin_QueryInterface (DShowPin *, const GUID *, void **);
171 unsigned long ff_dshow_pin_AddRef (DShowPin *);
172 unsigned long ff_dshow_pin_Release (DShowPin *);
173 long ff_dshow_pin_Connect (DShowPin *, IPin *, const AM_MEDIA_TYPE *);
174 long ff_dshow_pin_ReceiveConnection (DShowPin *, IPin *, const AM_MEDIA_TYPE *);
175 long ff_dshow_pin_Disconnect (DShowPin *);
176 long ff_dshow_pin_ConnectedTo (DShowPin *, IPin **);
177 long ff_dshow_pin_ConnectionMediaType (DShowPin *, AM_MEDIA_TYPE *);
178 long ff_dshow_pin_QueryPinInfo (DShowPin *, PIN_INFO *);
179 long ff_dshow_pin_QueryDirection (DShowPin *, PIN_DIRECTION *);
180 long ff_dshow_pin_QueryId (DShowPin *, wchar_t **);
181 long ff_dshow_pin_QueryAccept (DShowPin *, const AM_MEDIA_TYPE *);
182 long ff_dshow_pin_EnumMediaTypes (DShowPin *, IEnumMediaTypes **);
183 long ff_dshow_pin_QueryInternalConnections(DShowPin *, IPin **, unsigned long *);
184 long ff_dshow_pin_EndOfStream (DShowPin *);
185 long ff_dshow_pin_BeginFlush (DShowPin *);
186 long ff_dshow_pin_EndFlush (DShowPin *);
187 long ff_dshow_pin_NewSegment (DShowPin *, REFERENCE_TIME, REFERENCE_TIME, double);
188
189 long ff_dshow_meminputpin_QueryInterface (DShowMemInputPin *, const GUID *, void **);
190 unsigned long ff_dshow_meminputpin_AddRef (DShowMemInputPin *);
191 unsigned long ff_dshow_meminputpin_Release (DShowMemInputPin *);
192 long ff_dshow_meminputpin_GetAllocator (DShowMemInputPin *, IMemAllocator **);
193 long ff_dshow_meminputpin_NotifyAllocator (DShowMemInputPin *, IMemAllocator *, BOOL);
194 long ff_dshow_meminputpin_GetAllocatorRequirements(DShowMemInputPin *, ALLOCATOR_PROPERTIES *);
195 long ff_dshow_meminputpin_Receive (DShowMemInputPin *, IMediaSample *);
196 long ff_dshow_meminputpin_ReceiveMultiple (DShowMemInputPin *, IMediaSample **, long, long *);
197 long ff_dshow_meminputpin_ReceiveCanBlock (DShowMemInputPin *);
198
199 void ff_dshow_pin_Destroy(DShowPin *);
200 DShowPin *ff_dshow_pin_Create (DShowFilter *filter);
201
202 void ff_dshow_meminputpin_Destroy(DShowMemInputPin *);
203
204 /*****************************************************************************
205 * DShowEnumPins
206 ****************************************************************************/
207 struct DShowEnumPins {
208 IEnumPinsVtbl *vtbl;
209 long ref;
210 int pos;
211 DShowPin *pin;
212 DShowFilter *filter;
213 };
214
215 long ff_dshow_enumpins_QueryInterface(DShowEnumPins *, const GUID *, void **);
216 unsigned long ff_dshow_enumpins_AddRef (DShowEnumPins *);
217 unsigned long ff_dshow_enumpins_Release (DShowEnumPins *);
218 long ff_dshow_enumpins_Next (DShowEnumPins *, unsigned long, IPin **, unsigned long *);
219 long ff_dshow_enumpins_Skip (DShowEnumPins *, unsigned long);
220 long ff_dshow_enumpins_Reset (DShowEnumPins *);
221 long ff_dshow_enumpins_Clone (DShowEnumPins *, DShowEnumPins **);
222
223 void ff_dshow_enumpins_Destroy(DShowEnumPins *);
224 DShowEnumPins *ff_dshow_enumpins_Create (DShowPin *pin, DShowFilter *filter);
225
226 /*****************************************************************************
227 * DShowEnumMediaTypes
228 ****************************************************************************/
229 struct DShowEnumMediaTypes {
230 IEnumMediaTypesVtbl *vtbl;
231 long ref;
232 int pos;
233 AM_MEDIA_TYPE type;
234 };
235
236 long ff_dshow_enummediatypes_QueryInterface(DShowEnumMediaTypes *, const GUID *, void **);
237 unsigned long ff_dshow_enummediatypes_AddRef (DShowEnumMediaTypes *);
238 unsigned long ff_dshow_enummediatypes_Release (DShowEnumMediaTypes *);
239 long ff_dshow_enummediatypes_Next (DShowEnumMediaTypes *, unsigned long, AM_MEDIA_TYPE **, unsigned long *);
240 long ff_dshow_enummediatypes_Skip (DShowEnumMediaTypes *, unsigned long);
241 long ff_dshow_enummediatypes_Reset (DShowEnumMediaTypes *);
242 long ff_dshow_enummediatypes_Clone (DShowEnumMediaTypes *, DShowEnumMediaTypes **);
243
244 void ff_dshow_enummediatypes_Destroy(DShowEnumMediaTypes *);
245 DShowEnumMediaTypes *ff_dshow_enummediatypes_Create(const AM_MEDIA_TYPE *type);
246
247 /*****************************************************************************
248 * DShowFilter
249 ****************************************************************************/
250 struct DShowFilter {
251 IBaseFilterVtbl *vtbl;
252 long ref;
253 const wchar_t *name;
254 DShowPin *pin;
255 FILTER_INFO info;
256 FILTER_STATE state;
257 IReferenceClock *clock;
258 enum dshowDeviceType type;
259 void *priv_data;
260 int stream_index;
261 int64_t start_time;
262 void (*callback)(void *priv_data, int index, uint8_t *buf, int buf_size, int64_t time, enum dshowDeviceType type);
263 };
264
265 long ff_dshow_filter_QueryInterface (DShowFilter *, const GUID *, void **);
266 unsigned long ff_dshow_filter_AddRef (DShowFilter *);
267 unsigned long ff_dshow_filter_Release (DShowFilter *);
268 long ff_dshow_filter_GetClassID (DShowFilter *, CLSID *);
269 long ff_dshow_filter_Stop (DShowFilter *);
270 long ff_dshow_filter_Pause (DShowFilter *);
271 long ff_dshow_filter_Run (DShowFilter *, REFERENCE_TIME);
272 long ff_dshow_filter_GetState (DShowFilter *, DWORD, FILTER_STATE *);
273 long ff_dshow_filter_SetSyncSource (DShowFilter *, IReferenceClock *);
274 long ff_dshow_filter_GetSyncSource (DShowFilter *, IReferenceClock **);
275 long ff_dshow_filter_EnumPins (DShowFilter *, IEnumPins **);
276 long ff_dshow_filter_FindPin (DShowFilter *, const wchar_t *, IPin **);
277 long ff_dshow_filter_QueryFilterInfo(DShowFilter *, FILTER_INFO *);
278 long ff_dshow_filter_JoinFilterGraph(DShowFilter *, IFilterGraph *, const wchar_t *);
279 long ff_dshow_filter_QueryVendorInfo(DShowFilter *, wchar_t **);
280
281 void ff_dshow_filter_Destroy(DShowFilter *);
282 DShowFilter *ff_dshow_filter_Create (void *, void *, enum dshowDeviceType);
283
284 /*****************************************************************************
285 * dshow_ctx
286 ****************************************************************************/
287 struct dshow_ctx {
288 const AVClass *class;
289
290 IGraphBuilder *graph;
291
292 char *device_name[2];
293 char *device_unique_name[2];
294
295 int video_device_number;
296 int audio_device_number;
297
298 int list_options;
299 int list_devices;
300 int audio_buffer_size;
301 int crossbar_video_input_pin_number;
302 int crossbar_audio_input_pin_number;
303 char *video_pin_name;
304 char *audio_pin_name;
305 int show_video_device_dialog;
306 int show_audio_device_dialog;
307 int show_video_crossbar_connection_dialog;
308 int show_audio_crossbar_connection_dialog;
309 int show_analog_tv_tuner_dialog;
310 int show_analog_tv_tuner_audio_dialog;
311 char *audio_filter_load_file;
312 char *audio_filter_save_file;
313 char *video_filter_load_file;
314 char *video_filter_save_file;
315
316 IBaseFilter *device_filter[2];
317 IPin *device_pin[2];
318 DShowFilter *capture_filter[2];
319 DShowPin *capture_pin[2];
320
321 HANDLE mutex;
322 HANDLE event[2]; /* event[0] is set by DirectShow
323 * event[1] is set by callback() */
324 PacketList *pktl;
325
326 int eof;
327
328 int64_t curbufsize[2];
329 unsigned int video_frame_num;
330
331 IMediaControl *control;
332 IMediaEvent *media_event;
333
334 enum AVPixelFormat pixel_format;
335 enum AVCodecID video_codec_id;
336 char *framerate;
337
338 int requested_width;
339 int requested_height;
340 AVRational requested_framerate;
341
342 int sample_rate;
343 int sample_size;
344 int channels;
345 };
346
347 /*****************************************************************************
348 * CrossBar
349 ****************************************************************************/
350 HRESULT ff_dshow_try_setup_crossbar_options(ICaptureGraphBuilder2 *graph_builder2,
351 IBaseFilter *device_filter, enum dshowDeviceType devtype, AVFormatContext *avctx);
352
353 void ff_dshow_show_filter_properties(IBaseFilter *pFilter, AVFormatContext *avctx);
354
355 #endif /* AVDEVICE_DSHOW_CAPTURE_H */
356