FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
dshow_capture.h
Go to the documentation of this file.
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_H
23 #define AVDEVICE_DSHOW_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 /* EC_DEVICE_LOST is not defined in MinGW dshow headers. */
37 #ifndef EC_DEVICE_LOST
38 #define EC_DEVICE_LOST 0x1f
39 #endif
40 
41 long ff_copy_dshow_media_type(AM_MEDIA_TYPE *dst, const AM_MEDIA_TYPE *src);
42 void ff_print_VIDEO_STREAM_CONFIG_CAPS(const VIDEO_STREAM_CONFIG_CAPS *caps);
43 void ff_print_AUDIO_STREAM_CONFIG_CAPS(const AUDIO_STREAM_CONFIG_CAPS *caps);
44 void ff_print_AM_MEDIA_TYPE(const AM_MEDIA_TYPE *type);
45 void ff_printGUID(const GUID *g);
46 
47 #if DSHOWDEBUG
49 #define dshowdebug(...) av_log(&ff_dshow_context_class_ptr, AV_LOG_DEBUG, __VA_ARGS__)
50 #else
51 #define dshowdebug(...)
52 #endif
53 
54 static inline void nothing(void *foo)
55 {
56 }
57 
58 struct GUIDoffset {
59  const GUID *iid;
60  int offset;
61 };
62 
66 };
67 
71 };
72 
73 #define DECLARE_QUERYINTERFACE(class, ...) \
74 long WINAPI \
75 class##_QueryInterface(class *this, const GUID *riid, void **ppvObject) \
76 { \
77  struct GUIDoffset ifaces[] = __VA_ARGS__; \
78  int i; \
79  dshowdebug(AV_STRINGIFY(class)"_QueryInterface(%p, %p, %p)\n", this, riid, ppvObject); \
80  ff_printGUID(riid); \
81  if (!ppvObject) \
82  return E_POINTER; \
83  for (i = 0; i < sizeof(ifaces)/sizeof(ifaces[0]); i++) { \
84  if (IsEqualGUID(riid, ifaces[i].iid)) { \
85  void *obj = (void *) ((uint8_t *) this + ifaces[i].offset); \
86  class##_AddRef(this); \
87  dshowdebug("\tfound %d with offset %d\n", i, ifaces[i].offset); \
88  *ppvObject = (void *) obj; \
89  return S_OK; \
90  } \
91  } \
92  dshowdebug("\tE_NOINTERFACE\n"); \
93  *ppvObject = NULL; \
94  return E_NOINTERFACE; \
95 }
96 #define DECLARE_ADDREF(class) \
97 unsigned long WINAPI \
98 class##_AddRef(class *this) \
99 { \
100  dshowdebug(AV_STRINGIFY(class)"_AddRef(%p)\t%ld\n", this, this->ref+1); \
101  return InterlockedIncrement(&this->ref); \
102 }
103 #define DECLARE_RELEASE(class) \
104 unsigned long WINAPI \
105 class##_Release(class *this) \
106 { \
107  long ref = InterlockedDecrement(&this->ref); \
108  dshowdebug(AV_STRINGIFY(class)"_Release(%p)\t%ld\n", this, ref); \
109  if (!ref) \
110  class##_Destroy(this); \
111  return ref; \
112 }
113 
114 #define DECLARE_DESTROY(class, func) \
115 void class##_Destroy(class *this) \
116 { \
117  dshowdebug(AV_STRINGIFY(class)"_Destroy(%p)\n", this); \
118  func(this); \
119  if (this) { \
120  if (this->vtbl) \
121  CoTaskMemFree(this->vtbl); \
122  CoTaskMemFree(this); \
123  } \
124 }
125 #define DECLARE_CREATE(class, setup, ...) \
126 class *class##_Create(__VA_ARGS__) \
127 { \
128  class *this = CoTaskMemAlloc(sizeof(class)); \
129  void *vtbl = CoTaskMemAlloc(sizeof(*this->vtbl)); \
130  dshowdebug(AV_STRINGIFY(class)"_Create(%p)\n", this); \
131  if (!this || !vtbl) \
132  goto fail; \
133  ZeroMemory(this, sizeof(class)); \
134  ZeroMemory(vtbl, sizeof(*this->vtbl)); \
135  this->ref = 1; \
136  this->vtbl = vtbl; \
137  if (!setup) \
138  goto fail; \
139  dshowdebug("created "AV_STRINGIFY(class)" %p\n", this); \
140  return this; \
141 fail: \
142  class##_Destroy(this); \
143  dshowdebug("could not create "AV_STRINGIFY(class)"\n"); \
144  return NULL; \
145 }
146 
147 #define SETVTBL(vtbl, class, fn) \
148  do { (vtbl)->fn = (void *) class##_##fn; } while(0)
149 
150 /*****************************************************************************
151  * Forward Declarations
152  ****************************************************************************/
153 typedef struct libAVPin libAVPin;
155 typedef struct libAVEnumPins libAVEnumPins;
157 typedef struct libAVFilter libAVFilter;
158 
159 /*****************************************************************************
160  * libAVPin
161  ****************************************************************************/
162 struct libAVPin {
163  IPinVtbl *vtbl;
164  long ref;
166  IPin *connectedto;
167  AM_MEDIA_TYPE type;
168  IMemInputPinVtbl *imemvtbl;
169 };
170 
171 long WINAPI libAVPin_QueryInterface (libAVPin *, const GUID *, void **);
172 unsigned long WINAPI libAVPin_AddRef (libAVPin *);
173 unsigned long WINAPI libAVPin_Release (libAVPin *);
174 long WINAPI libAVPin_Connect (libAVPin *, IPin *, const AM_MEDIA_TYPE *);
175 long WINAPI libAVPin_ReceiveConnection (libAVPin *, IPin *, const AM_MEDIA_TYPE *);
176 long WINAPI libAVPin_Disconnect (libAVPin *);
177 long WINAPI libAVPin_ConnectedTo (libAVPin *, IPin **);
178 long WINAPI libAVPin_ConnectionMediaType (libAVPin *, AM_MEDIA_TYPE *);
179 long WINAPI libAVPin_QueryPinInfo (libAVPin *, PIN_INFO *);
180 long WINAPI libAVPin_QueryDirection (libAVPin *, PIN_DIRECTION *);
181 long WINAPI libAVPin_QueryId (libAVPin *, wchar_t **);
182 long WINAPI libAVPin_QueryAccept (libAVPin *, const AM_MEDIA_TYPE *);
183 long WINAPI libAVPin_EnumMediaTypes (libAVPin *, IEnumMediaTypes **);
184 long WINAPI libAVPin_QueryInternalConnections(libAVPin *, IPin **, unsigned long *);
185 long WINAPI libAVPin_EndOfStream (libAVPin *);
186 long WINAPI libAVPin_BeginFlush (libAVPin *);
187 long WINAPI libAVPin_EndFlush (libAVPin *);
188 long WINAPI libAVPin_NewSegment (libAVPin *, REFERENCE_TIME, REFERENCE_TIME, double);
189 
190 long WINAPI libAVMemInputPin_QueryInterface (libAVMemInputPin *, const GUID *, void **);
191 unsigned long WINAPI libAVMemInputPin_AddRef (libAVMemInputPin *);
192 unsigned long WINAPI libAVMemInputPin_Release (libAVMemInputPin *);
193 long WINAPI libAVMemInputPin_GetAllocator (libAVMemInputPin *, IMemAllocator **);
194 long WINAPI libAVMemInputPin_NotifyAllocator (libAVMemInputPin *, IMemAllocator *, BOOL);
195 long WINAPI libAVMemInputPin_GetAllocatorRequirements(libAVMemInputPin *, ALLOCATOR_PROPERTIES *);
196 long WINAPI libAVMemInputPin_Receive (libAVMemInputPin *, IMediaSample *);
197 long WINAPI libAVMemInputPin_ReceiveMultiple (libAVMemInputPin *, IMediaSample **, long, long *);
199 
200 void libAVPin_Destroy(libAVPin *);
202 
204 
205 /*****************************************************************************
206  * libAVEnumPins
207  ****************************************************************************/
209  IEnumPinsVtbl *vtbl;
210  long ref;
211  int pos;
214 };
215 
216 long WINAPI libAVEnumPins_QueryInterface(libAVEnumPins *, const GUID *, void **);
217 unsigned long WINAPI libAVEnumPins_AddRef (libAVEnumPins *);
218 unsigned long WINAPI libAVEnumPins_Release (libAVEnumPins *);
219 long WINAPI libAVEnumPins_Next (libAVEnumPins *, unsigned long, IPin **, unsigned long *);
220 long WINAPI libAVEnumPins_Skip (libAVEnumPins *, unsigned long);
221 long WINAPI libAVEnumPins_Reset (libAVEnumPins *);
223 
226 
227 /*****************************************************************************
228  * libAVEnumMediaTypes
229  ****************************************************************************/
231  IEnumMediaTypesVtbl *vtbl;
232  long ref;
233  int pos;
234  AM_MEDIA_TYPE type;
235 };
236 
237 long WINAPI libAVEnumMediaTypes_QueryInterface(libAVEnumMediaTypes *, const GUID *, void **);
238 unsigned long WINAPI libAVEnumMediaTypes_AddRef (libAVEnumMediaTypes *);
239 unsigned long WINAPI libAVEnumMediaTypes_Release (libAVEnumMediaTypes *);
240 long WINAPI libAVEnumMediaTypes_Next (libAVEnumMediaTypes *, unsigned long, AM_MEDIA_TYPE **, unsigned long *);
241 long WINAPI libAVEnumMediaTypes_Skip (libAVEnumMediaTypes *, unsigned long);
244 
247 
248 /*****************************************************************************
249  * libAVFilter
250  ****************************************************************************/
251 struct libAVFilter {
252  IBaseFilterVtbl *vtbl;
253  long ref;
254  const wchar_t *name;
256  FILTER_INFO info;
257  FILTER_STATE state;
258  IReferenceClock *clock;
260  void *priv_data;
262  int64_t start_time;
263  void (*callback)(void *priv_data, int index, uint8_t *buf, int buf_size, int64_t time, enum dshowDeviceType type);
264 };
265 
266 long WINAPI libAVFilter_QueryInterface (libAVFilter *, const GUID *, void **);
267 unsigned long WINAPI libAVFilter_AddRef (libAVFilter *);
268 unsigned long WINAPI libAVFilter_Release (libAVFilter *);
269 long WINAPI libAVFilter_GetClassID (libAVFilter *, CLSID *);
270 long WINAPI libAVFilter_Stop (libAVFilter *);
271 long WINAPI libAVFilter_Pause (libAVFilter *);
272 long WINAPI libAVFilter_Run (libAVFilter *, REFERENCE_TIME);
273 long WINAPI libAVFilter_GetState (libAVFilter *, DWORD, FILTER_STATE *);
274 long WINAPI libAVFilter_SetSyncSource (libAVFilter *, IReferenceClock *);
275 long WINAPI libAVFilter_GetSyncSource (libAVFilter *, IReferenceClock **);
276 long WINAPI libAVFilter_EnumPins (libAVFilter *, IEnumPins **);
277 long WINAPI libAVFilter_FindPin (libAVFilter *, const wchar_t *, IPin **);
278 long WINAPI libAVFilter_QueryFilterInfo(libAVFilter *, FILTER_INFO *);
279 long WINAPI libAVFilter_JoinFilterGraph(libAVFilter *, IFilterGraph *, const wchar_t *);
280 long WINAPI libAVFilter_QueryVendorInfo(libAVFilter *, wchar_t **);
281 
283 libAVFilter *libAVFilter_Create (void *, void *, enum dshowDeviceType);
284 
285 /*****************************************************************************
286  * dshow_ctx
287  ****************************************************************************/
288 struct dshow_ctx {
289  const AVClass *class;
290 
291  IGraphBuilder *graph;
292 
293  char *device_name[2];
296 
310 
311  IBaseFilter *device_filter[2];
312  IPin *device_pin[2];
315 
317  HANDLE event[2]; /* event[0] is set by DirectShow
318  * event[1] is set by callback() */
320 
321  int eof;
322 
323  int64_t curbufsize[2];
324  unsigned int video_frame_num;
325 
326  IMediaControl *control;
327  IMediaEvent *media_event;
328 
331  char *framerate;
332 
336 
339  int channels;
340 };
341 
342 /*****************************************************************************
343  * CrossBar
344  ****************************************************************************/
345 HRESULT dshow_try_setup_crossbar_options(ICaptureGraphBuilder2 *graph_builder2,
346  IBaseFilter *device_filter, enum dshowDeviceType devtype, AVFormatContext *avctx);
347 
348 void dshow_show_filter_properties(IBaseFilter *pFilter, AVFormatContext *avctx);
349 
350 #endif /* AVDEVICE_DSHOW_H */