1 /* GStreamer
2  * Copyright (C) 2007 Sebastien Moutte <sebastien@moutte.net>
3  *
4  * gstdshow.cpp:
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  */
21 
22 #include <gst/video/video-format.h>
23 
24 #include "gstdshow.h"
25 #include "gstdshowfakesink.h"
26 #include "gstdshowvideosrc.h"
27 
28 GST_DEBUG_CATEGORY_EXTERN (dshowsrcwrapper_debug);
29 #define GST_CAT_DEFAULT dshowsrcwrapper_debug
30 
31 gchar *
wchar_to_gchar(WCHAR * w)32 wchar_to_gchar (WCHAR * w)
33 {
34   return g_utf16_to_utf8 ((const gunichar2 *) w, wcslen (w), NULL, NULL, NULL);
35 }
36 
37 const GUID MEDIASUBTYPE_I420
38     = { 0x30323449, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B,
39     0x71}
40 };
41 
42 void
gst_dshow_free_mediatype(AM_MEDIA_TYPE * pmt)43 gst_dshow_free_mediatype (AM_MEDIA_TYPE * pmt)
44 {
45   if (pmt != NULL) {
46     if (pmt->cbFormat != 0) {
47       CoTaskMemFree ((PVOID) pmt->pbFormat);
48       pmt->cbFormat = 0;
49       pmt->pbFormat = NULL;
50     }
51     if (pmt->pUnk != NULL) {
52       /* Unecessary because pUnk should not be used, but safest. */
53       pmt->pUnk->Release ();
54       pmt->pUnk = NULL;
55     }
56 
57     CoTaskMemFree (pmt);
58   }
59 }
60 
61 void
gst_dshow_free_pin_mediatype(gpointer pt)62 gst_dshow_free_pin_mediatype (gpointer pt)
63 {
64   GstCapturePinMediaType *pin_mediatype = (GstCapturePinMediaType *) pt;
65   if (pin_mediatype) {
66     if (pin_mediatype->capture_pin) {
67       pin_mediatype->capture_pin->Release ();
68       pin_mediatype->capture_pin = NULL;
69     }
70     if (pin_mediatype->mediatype) {
71       gst_dshow_free_mediatype (pin_mediatype->mediatype);
72       pin_mediatype->mediatype = NULL;
73     }
74   }
75 }
76 
77 GstCapturePinMediaType *
gst_dshow_new_pin_mediatype(IPin * pin)78 gst_dshow_new_pin_mediatype (IPin * pin)
79 {
80   GstCapturePinMediaType *pin_mediatype = g_new0 (GstCapturePinMediaType, 1);
81 
82   pin->AddRef ();
83   pin_mediatype->capture_pin = pin;
84 
85   return pin_mediatype;
86 }
87 
88 GstCapturePinMediaType *
gst_dshow_new_pin_mediatype_from_enum_mediatypes(IPin * pin,IEnumMediaTypes * enum_mediatypes)89 gst_dshow_new_pin_mediatype_from_enum_mediatypes (IPin * pin, IEnumMediaTypes *enum_mediatypes)
90 {
91   GstCapturePinMediaType *pin_mediatype = gst_dshow_new_pin_mediatype (pin);
92   VIDEOINFOHEADER *video_info = NULL;
93 
94   HRESULT hres = enum_mediatypes->Next (1, &pin_mediatype->mediatype, NULL);
95   if (hres != S_OK || !pin_mediatype->mediatype) {
96     gst_dshow_free_pin_mediatype (pin_mediatype);
97     return NULL;
98   }
99 
100   video_info = (VIDEOINFOHEADER *) pin_mediatype->mediatype->pbFormat;
101 
102   pin_mediatype->defaultWidth = video_info->bmiHeader.biWidth;
103   pin_mediatype->defaultHeight = video_info->bmiHeader.biHeight;
104   pin_mediatype->defaultFPS = (gint) (10000000 / video_info->AvgTimePerFrame);
105   pin_mediatype->granularityWidth = 1;
106   pin_mediatype->granularityHeight = 1;
107 
108   return pin_mediatype;
109 }
110 
111 GstCapturePinMediaType *
gst_dshow_new_pin_mediatype_from_streamcaps(IPin * pin,gint id,IAMStreamConfig * streamcaps)112 gst_dshow_new_pin_mediatype_from_streamcaps (IPin * pin, gint id, IAMStreamConfig * streamcaps)
113 {
114   GstCapturePinMediaType *pin_mediatype = gst_dshow_new_pin_mediatype (pin);
115   VIDEOINFOHEADER *video_info = NULL;
116 
117   HRESULT hres = streamcaps->GetStreamCaps (id, &pin_mediatype->mediatype,
118       (BYTE *) & pin_mediatype->vscc);
119   if (FAILED (hres) || !pin_mediatype->mediatype) {
120     gst_dshow_free_pin_mediatype (pin_mediatype);
121     return NULL;
122   }
123 
124   video_info = (VIDEOINFOHEADER *) pin_mediatype->mediatype->pbFormat;
125 
126   pin_mediatype->defaultWidth = video_info->bmiHeader.biWidth;
127   pin_mediatype->defaultHeight = video_info->bmiHeader.biHeight;
128   pin_mediatype->defaultFPS = (gint) (10000000 / video_info->AvgTimePerFrame);
129   pin_mediatype->granularityWidth = pin_mediatype->vscc.OutputGranularityX;
130   pin_mediatype->granularityHeight = pin_mediatype->vscc.OutputGranularityY;
131 
132   return pin_mediatype;
133 }
134 
135 void
gst_dshow_free_pins_mediatypes(GList * pins_mediatypes)136 gst_dshow_free_pins_mediatypes (GList * pins_mediatypes)
137 {
138   g_list_free_full (pins_mediatypes,
139       (GDestroyNotify) gst_dshow_free_pin_mediatype);
140 }
141 
142 gboolean
gst_dshow_check_mediatype(AM_MEDIA_TYPE * media_type,const GUID sub_type,const GUID format_type)143 gst_dshow_check_mediatype (AM_MEDIA_TYPE * media_type, const GUID sub_type,
144     const GUID format_type)
145 {
146   RPC_STATUS rpcstatus;
147 
148   g_return_val_if_fail (media_type != NULL, FALSE);
149 
150   return
151       UuidCompare (&media_type->subtype, (UUID *) & sub_type,
152       &rpcstatus) == 0 && rpcstatus == RPC_S_OK &&
153       //IsEqualGUID (&media_type->subtype, &sub_type)
154       UuidCompare (&media_type->formattype, (UUID *) & format_type,
155       &rpcstatus) == 0 && rpcstatus == RPC_S_OK;
156 }
157 
158 gboolean
gst_dshow_get_pin_from_filter(IBaseFilter * filter,PIN_DIRECTION pindir,IPin ** pin)159 gst_dshow_get_pin_from_filter (IBaseFilter * filter, PIN_DIRECTION pindir,
160     IPin ** pin)
161 {
162   gboolean ret = FALSE;
163   IEnumPins *enumpins = NULL;
164   IPin *pintmp = NULL;
165   HRESULT hres;
166   *pin = NULL;
167 
168   hres = filter->EnumPins (&enumpins);
169   if (FAILED (hres)) {
170     return ret;
171   }
172 
173   while (enumpins->Next (1, &pintmp, NULL) == S_OK) {
174     PIN_DIRECTION pindirtmp;
175     hres = pintmp->QueryDirection (&pindirtmp);
176     if (hres == S_OK && pindir == pindirtmp) {
177       *pin = pintmp;
178       ret = TRUE;
179       break;
180     }
181     pintmp->Release ();
182   }
183   enumpins->Release ();
184 
185   return ret;
186 }
187 
188 gboolean
gst_dshow_find_filter(CLSID input_majortype,CLSID input_subtype,CLSID output_majortype,CLSID output_subtype,gchar * prefered_filter_name,IBaseFilter ** filter)189 gst_dshow_find_filter (CLSID input_majortype, CLSID input_subtype,
190     CLSID output_majortype, CLSID output_subtype,
191     gchar * prefered_filter_name, IBaseFilter ** filter)
192 {
193   gboolean ret = FALSE;
194   HRESULT hres;
195   GUID arrayInTypes[2];
196   GUID arrayOutTypes[2];
197   IFilterMapper2 *mapper = NULL;
198   IEnumMoniker *enum_moniker = NULL;
199   IMoniker *moniker = NULL;
200   ULONG fetched;
201   gchar *prefered_filter_upper = NULL;
202   gboolean exit = FALSE;
203 
204   /* initialize output parameter */
205   if (filter)
206     *filter = NULL;
207 
208   /* create a private copy of prefered filter substring in upper case */
209   if (prefered_filter_name) {
210     prefered_filter_upper = g_strdup (prefered_filter_name);
211     _strupr (prefered_filter_upper);
212   }
213 
214   hres = CoCreateInstance (CLSID_FilterMapper2, NULL, CLSCTX_INPROC,
215       IID_IFilterMapper2, (void **) &mapper);
216   if (FAILED (hres))
217     goto clean;
218 
219   memcpy (&arrayInTypes[0], &input_majortype, sizeof (CLSID));
220   memcpy (&arrayInTypes[1], &input_subtype, sizeof (CLSID));
221   memcpy (&arrayOutTypes[0], &output_majortype, sizeof (CLSID));
222   memcpy (&arrayOutTypes[1], &output_subtype, sizeof (CLSID));
223 
224   hres =
225       mapper->EnumMatchingFilters (&enum_moniker, 0, FALSE,
226       MERIT_DO_NOT_USE + 1, TRUE, 1, arrayInTypes, NULL, NULL, FALSE, TRUE, 1,
227       arrayOutTypes, NULL, NULL);
228   if (FAILED (hres))
229     goto clean;
230 
231   enum_moniker->Reset ();
232 
233   while (hres = enum_moniker->Next (1, &moniker, &fetched), hres == S_OK
234       && !exit) {
235     IBaseFilter *filter_temp = NULL;
236     IPropertyBag *property_bag = NULL;
237     gchar *friendly_name = NULL;
238 
239     hres =
240         moniker->BindToStorage (NULL, NULL, IID_IPropertyBag,
241         (void **) &property_bag);
242     if (SUCCEEDED (hres) && property_bag) {
243       VARIANT varFriendlyName;
244       VariantInit (&varFriendlyName);
245 
246       hres = property_bag->Read (L"FriendlyName", &varFriendlyName, NULL);
247       if (hres == S_OK && varFriendlyName.bstrVal) {
248         friendly_name = wchar_to_gchar (varFriendlyName.bstrVal);
249         if (friendly_name)
250           _strupr (friendly_name);
251         SysFreeString (varFriendlyName.bstrVal);
252       }
253       property_bag->Release ();
254     }
255 
256     hres =
257         moniker->BindToObject (NULL, NULL, IID_IBaseFilter,
258         (void **) &filter_temp);
259     if (SUCCEEDED (hres) && filter_temp) {
260       ret = TRUE;
261       if (filter) {
262         if (*filter)
263           (*filter)->Release ();
264 
265         *filter = filter_temp;
266         (*filter)->AddRef ();
267 
268         if (prefered_filter_upper && friendly_name &&
269             strstr (friendly_name, prefered_filter_upper))
270           exit = TRUE;
271       }
272 
273       /* if we just want to know if the formats are supported OR
274          if we don't care about what will be the filter used
275          => we can stop enumeration */
276       if (!filter || !prefered_filter_upper)
277         exit = TRUE;
278 
279       filter_temp->Release ();
280     }
281 
282     g_free (friendly_name);
283     moniker->Release ();
284   }
285 
286 clean:
287   g_free (prefered_filter_upper);
288   if (enum_moniker)
289     enum_moniker->Release ();
290   if (mapper)
291     mapper->Release ();
292 
293   return ret;
294 }
295 
296 void
gst_dshow_device_entry_free(DshowDeviceEntry * entry)297 gst_dshow_device_entry_free (DshowDeviceEntry * entry)
298 {
299   if (entry) {
300     g_free (entry->device);
301     entry->device = NULL;
302     g_free (entry->device_name);
303     entry->device_name = NULL;
304     if (entry->caps) {
305       gst_caps_unref (entry->caps);
306       entry->caps = NULL;
307     }
308     if (entry->moniker) {
309       entry->moniker->Release ();
310       entry->moniker = NULL;
311     }
312   }
313 }
314 
315 void
gst_dshow_device_list_free(GList * devices)316 gst_dshow_device_list_free (GList * devices)
317 {
318   GList *cur;
319 
320   for (cur = devices; cur != NULL; cur = cur->next)
321     gst_dshow_device_entry_free ((DshowDeviceEntry *) cur->data);
322 
323   g_list_free (devices);
324 }
325 
326 GList *
gst_dshow_enumerate_devices(const GUID * device_category,gboolean getcaps)327 gst_dshow_enumerate_devices (const GUID * device_category, gboolean getcaps)
328 {
329   GList *result = NULL;
330   ICreateDevEnum *devices_enum = NULL;
331   IEnumMoniker *enum_moniker = NULL;
332   IMoniker *moniker = NULL;
333   HRESULT hres = S_FALSE;
334   ULONG fetched;
335   gint devidx = -1;
336 
337   hres = CoCreateInstance (CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
338       IID_ICreateDevEnum, (void **) &devices_enum);
339   if (hres != S_OK) {
340     GST_ERROR ("Failed to create System Device Enumerator");
341     goto clean;
342   }
343 
344   hres = devices_enum->CreateClassEnumerator (*device_category,
345       &enum_moniker, 0);
346   if (hres != S_OK || !enum_moniker) {
347     GST_ERROR ("Failed to create audio/video class device enumerator");
348     goto clean;
349   }
350 
351   enum_moniker->Reset ();
352 
353   while (enum_moniker->Next (1, &moniker, &fetched) == S_OK) {
354     IPropertyBag *property_bag = NULL;
355     hres = moniker->BindToStorage (NULL, NULL, IID_IPropertyBag,
356         (void **) &property_bag);
357     if (SUCCEEDED (hres) && property_bag) {
358       VARIANT varFriendlyName;
359       VariantInit (&varFriendlyName);
360 
361       hres = property_bag->Read (L"FriendlyName", &varFriendlyName, NULL);
362       if (hres == S_OK && varFriendlyName.bstrVal) {
363         gchar *friendly_name = wchar_to_gchar (varFriendlyName.bstrVal);
364 
365         devidx++;
366         GST_DEBUG ("Found device idx=%d: device-name='%s'",
367             devidx, friendly_name);
368 
369         WCHAR *wszDisplayName = NULL;
370         hres = moniker->GetDisplayName (NULL, NULL, &wszDisplayName);
371         if (hres == S_OK && wszDisplayName) {
372           DshowDeviceEntry *entry = g_new0 (DshowDeviceEntry, 1);
373           gchar *device_path = NULL;
374           GstCaps *caps = NULL;
375 
376           device_path = wchar_to_gchar (wszDisplayName);
377           CoTaskMemFree (wszDisplayName);
378 
379           /* getting caps can be slow, so make it optional when enumerating */
380           if (getcaps) {
381             IBindCtx *lpbc = NULL;
382             hres = CreateBindCtx (0, &lpbc);
383             if (SUCCEEDED (hres)) {
384               IBaseFilter *video_cap_filter = NULL;
385               hres = moniker->BindToObject (lpbc, NULL, IID_IBaseFilter,
386                 (LPVOID *) & video_cap_filter);
387               if (video_cap_filter) {
388                 caps = gst_dshowvideosrc_getcaps_from_capture_filter (video_cap_filter, NULL);
389                 video_cap_filter->Release ();
390               }
391               lpbc->Release ();
392             }
393           }
394 
395           entry->device = device_path;
396           entry->device_name = friendly_name;
397           entry->device_index = devidx;
398           entry->caps = caps;
399           entry->moniker = moniker;
400           moniker = NULL;
401           result = g_list_append (result, entry);
402         } else {
403           g_free (friendly_name);
404         }
405         SysFreeString (varFriendlyName.bstrVal);
406       }
407       property_bag->Release ();
408     }
409     if (moniker) {
410       moniker->Release ();
411     }
412   }
413 
414 clean:
415   if (enum_moniker) {
416     enum_moniker->Release ();
417   }
418 
419   if (devices_enum) {
420     devices_enum->Release ();
421   }
422 
423   return result;
424 }
425 
426 DshowDeviceEntry *
gst_dshow_select_device(const GUID * device_category,const gchar * device,const gchar * device_name,const gint device_index)427 gst_dshow_select_device (const GUID * device_category,
428     const gchar * device, const gchar * device_name, const gint device_index)
429 {
430   GList *devices = NULL;
431   GList *item = NULL;
432   DshowDeviceEntry *selected = NULL;
433 
434   GST_DEBUG ("Trying to select device-index=%d, device-name='%s', device='%s'",
435     device_index, device_name, device);
436 
437   devices = gst_dshow_enumerate_devices (&CLSID_VideoInputDeviceCategory, FALSE);
438 
439   for (item = devices; item != NULL; item = item->next) {
440     DshowDeviceEntry *entry = (DshowDeviceEntry *) item->data;
441 
442     /* device will be used first, then device-name, then device-index */
443     if (device && g_strcmp0 (device, entry->device) == 0) {
444       selected = entry;
445       break;
446     } else if (!device && device_name && g_strcmp0 (device_name, entry->device_name) == 0) {
447       selected = entry;
448       break;
449     } else if (!device && !deviceName && device_index == entry->device_index) {
450       selected = entry;
451       break;
452     }
453   }
454 
455   if (selected) {
456     devices = g_list_remove (devices, selected);
457     GST_DEBUG ("Selected device-index=%d, device-name='%s', device='%s'",
458       selected->device_index, selected->device_name, selected->device);
459   } else {
460     GST_DEBUG ("No matching device found");
461   }
462 
463   gst_dshow_device_list_free (devices);
464 
465   return selected;
466 }
467 
468 IBaseFilter *
gst_dshow_create_capture_filter(IMoniker * moniker)469 gst_dshow_create_capture_filter (IMoniker *moniker)
470 {
471   HRESULT hres = S_OK;
472   IBindCtx *lpbc = NULL;
473   IBaseFilter *video_cap_filter = NULL;
474 
475   g_assert (moniker != NULL);
476 
477   hres = CreateBindCtx (0, &lpbc);
478   if (SUCCEEDED (hres)) {
479     hres = moniker->BindToObject (lpbc, NULL, IID_IBaseFilter,
480         (LPVOID *) & video_cap_filter);
481     lpbc->Release ();
482   }
483 
484   return video_cap_filter;
485 }
486 
487 gchar *
gst_dshow_getdevice_from_devicename(const GUID * device_category,gchar ** device_name,gint * device_index)488 gst_dshow_getdevice_from_devicename (const GUID * device_category,
489     gchar ** device_name, gint * device_index)
490 {
491   gchar *ret = NULL;
492   ICreateDevEnum *devices_enum = NULL;
493   IEnumMoniker *enum_moniker = NULL;
494   IMoniker *moniker = NULL;
495   HRESULT hres = S_FALSE;
496   ULONG fetched;
497   gboolean bfound = FALSE;
498   gint devidx = -1;
499 
500   hres = CoCreateInstance (CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
501       IID_ICreateDevEnum, (void **) &devices_enum);
502   if (hres != S_OK) {
503     GST_ERROR ("Failed to create System Device Enumerator");
504     goto clean;
505   }
506 
507   hres = devices_enum->CreateClassEnumerator (*device_category,
508       &enum_moniker, 0);
509   if (hres != S_OK || !enum_moniker) {
510     GST_ERROR ("Failed to create audio/video class device enumerator");
511     goto clean;
512   }
513 
514   enum_moniker->Reset ();
515 
516   while (hres = enum_moniker->Next (1, &moniker, &fetched), hres == S_OK
517       && !bfound) {
518     IPropertyBag *property_bag = NULL;
519     hres =
520         moniker->BindToStorage (NULL, NULL, IID_IPropertyBag,
521         (void **) &property_bag);
522     if (SUCCEEDED (hres) && property_bag) {
523       VARIANT varFriendlyName;
524       VariantInit (&varFriendlyName);
525 
526       hres = property_bag->Read (L"FriendlyName", &varFriendlyName, NULL);
527       if (hres == S_OK && varFriendlyName.bstrVal) {
528         gchar *friendly_name = wchar_to_gchar (varFriendlyName.bstrVal);
529 
530         devidx++;
531         GST_DEBUG ("Found device idx=%d: device-name='%s'",
532             devidx, friendly_name);
533 
534         if ((!*device_name || !**device_name) && devidx == *device_index) {
535           g_free (*device_name);
536           *device_name = g_strdup (friendly_name);
537         }
538 
539         if ((*device_name && **device_name)
540             && _stricmp (*device_name, friendly_name) == 0) {
541           WCHAR *wszDisplayName = NULL;
542           hres = moniker->GetDisplayName (NULL, NULL, &wszDisplayName);
543           if (hres == S_OK && wszDisplayName) {
544             *device_index = devidx;
545             ret = wchar_to_gchar (wszDisplayName);
546             CoTaskMemFree (wszDisplayName);
547           }
548           bfound = TRUE;
549         }
550         SysFreeString (varFriendlyName.bstrVal);
551       }
552       property_bag->Release ();
553     }
554     moniker->Release ();
555   }
556 
557 clean:
558   if (enum_moniker) {
559     enum_moniker->Release ();
560   }
561 
562   if (devices_enum) {
563     devices_enum->Release ();
564   }
565 
566   return ret;
567 }
568 
569 gboolean
gst_dshow_show_propertypage(IBaseFilter * base_filter)570 gst_dshow_show_propertypage (IBaseFilter * base_filter)
571 {
572   gboolean ret = FALSE;
573   ISpecifyPropertyPages *pProp = NULL;
574   HRESULT hres =
575       base_filter->QueryInterface (IID_ISpecifyPropertyPages, (void **) &pProp);
576   if (SUCCEEDED (hres)) {
577     /* Get the filter's name and IUnknown pointer. */
578     FILTER_INFO FilterInfo;
579     CAUUID caGUID;
580     IUnknown *pFilterUnk = NULL;
581     hres = base_filter->QueryFilterInfo (&FilterInfo);
582     base_filter->QueryInterface (IID_IUnknown, (void **) &pFilterUnk);
583 
584     /* Show the page. */
585     pProp->GetPages (&caGUID);
586     pProp->Release ();
587     OleCreatePropertyFrame (GetDesktopWindow (), 0, 0, FilterInfo.achName,
588         1, &pFilterUnk, caGUID.cElems, caGUID.pElems, 0, 0, NULL);
589 
590     pFilterUnk->Release ();
591     FilterInfo.pGraph->Release ();
592     CoTaskMemFree (caGUID.pElems);
593   }
594   return ret;
595 }
596 
597 GstVideoFormat
gst_dshow_guid_to_gst_video_format(AM_MEDIA_TYPE * mediatype)598 gst_dshow_guid_to_gst_video_format (AM_MEDIA_TYPE *mediatype)
599 {
600   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_I420, FORMAT_VideoInfo))
601     return GST_VIDEO_FORMAT_I420;
602 
603   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_RGB24, FORMAT_VideoInfo))
604     return GST_VIDEO_FORMAT_BGR;
605 
606   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_YUY2, FORMAT_VideoInfo))
607     return GST_VIDEO_FORMAT_YUY2;
608 
609   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_UYVY, FORMAT_VideoInfo))
610     return GST_VIDEO_FORMAT_UYVY;
611 
612   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_RGB32, FORMAT_VideoInfo))
613     return GST_VIDEO_FORMAT_BGRx;
614 
615   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_RGB565, FORMAT_VideoInfo))
616     return GST_VIDEO_FORMAT_BGR16;
617 
618   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_RGB555, FORMAT_VideoInfo))
619     return GST_VIDEO_FORMAT_BGR15;
620 
621   if (gst_dshow_check_mediatype (mediatype, MEDIASUBTYPE_RGB8, FORMAT_VideoInfo))
622     return GST_VIDEO_FORMAT_GRAY8;
623 
624   return GST_VIDEO_FORMAT_UNKNOWN;
625 }
626 
627 gboolean
gst_dshow_is_pin_connected(IPin * pin)628 gst_dshow_is_pin_connected (IPin * pin)
629 {
630   IPin *tmp_pin = NULL;
631   gboolean res;
632   HRESULT hres;
633 
634   g_assert (pin);
635   hres = pin->ConnectedTo (&tmp_pin);
636   res = (hres != VFW_E_NOT_CONNECTED);
637   if (tmp_pin)
638     tmp_pin->Release ();
639 
640   return res;
641 }
642 
643 GstCaps *
gst_dshow_new_video_caps(GstVideoFormat video_format,const gchar * name,GstCapturePinMediaType * pin_mediatype)644 gst_dshow_new_video_caps (GstVideoFormat video_format, const gchar * name,
645     GstCapturePinMediaType * pin_mediatype)
646 {
647   GstCaps *video_caps = NULL;
648   GstStructure *video_structure = NULL;
649   gint min_w, max_w;
650   gint min_h, max_h;
651   gint min_fr, max_fr;
652 
653   /* raw video format */
654   switch (video_format) {
655     case GST_VIDEO_FORMAT_BGR:
656       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("BGR"));
657       break;
658     case GST_VIDEO_FORMAT_I420:
659       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("I420"));
660 	  break;
661     case GST_VIDEO_FORMAT_YUY2:
662       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("YUY2"));
663       break;
664     case GST_VIDEO_FORMAT_UYVY:
665       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("UYVY"));
666       break;
667     case GST_VIDEO_FORMAT_BGRx:
668       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("BGRx"));
669       break;
670     case GST_VIDEO_FORMAT_BGR16:
671       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("BGR16"));
672       break;
673     case GST_VIDEO_FORMAT_BGR15:
674       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("BGR15"));
675       break;
676     case GST_VIDEO_FORMAT_GRAY8:
677       video_caps = gst_caps_from_string (GST_VIDEO_CAPS_MAKE ("GRAY8"));
678       break;
679     default:
680       break;
681   }
682 
683   /* other video format */
684   if (!video_caps) {
685     if (g_ascii_strncasecmp (name, "video/x-dv, systemstream=FALSE", 31) == 0) {
686       video_caps = gst_caps_new_simple ("video/x-dv",
687           "systemstream", G_TYPE_BOOLEAN, FALSE,
688           "format", G_TYPE_STRING, "dvsd",
689           NULL);
690     } else if (g_ascii_strncasecmp (name, "video/x-dv, systemstream=TRUE", 31) == 0) {
691       video_caps = gst_caps_new_simple ("video/x-dv",
692           "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
693       return video_caps;
694     } else if (g_ascii_strncasecmp (name, "image/jpeg", 10) == 0) {
695       video_caps = gst_caps_new_simple ("image/jpeg", NULL);
696     } else if (g_ascii_strncasecmp (name, "video/x-h264", 12) == 0) {
697       video_caps = gst_caps_new_simple ("video/x-h264", NULL);
698     }
699   }
700 
701   if (!video_caps)
702     return NULL;
703 
704   video_structure = gst_caps_get_structure (video_caps, 0);
705 
706   /* Hope GST_TYPE_INT_RANGE_STEP will exits in future gstreamer releases  */
707   /* because we could use :  */
708   /* "width", GST_TYPE_INT_RANGE_STEP, video_default->minWidth, video_default->maxWidth,  video_default->granularityWidth */
709   /* instead of : */
710   /* "width", GST_TYPE_INT_RANGE, video_default->minWidth, video_default->maxWidth */
711 
712   /* For framerate we do not need a step (granularity) because  */
713   /* "The IAMStreamConfig::SetFormat method will set the frame rate to the closest  */
714   /* value that the filter supports" as it said in the VIDEO_STREAM_CONFIG_CAPS dshwo doc */
715 
716   min_w = pin_mediatype->vscc.MinOutputSize.cx;
717   max_w = pin_mediatype->vscc.MaxOutputSize.cx;
718   min_h = pin_mediatype->vscc.MinOutputSize.cy;
719   max_h = pin_mediatype->vscc.MaxOutputSize.cy;
720   min_fr = (gint) (10000000 / pin_mediatype->vscc.MaxFrameInterval);
721   max_fr = (gint)(10000000 / pin_mediatype->vscc.MinFrameInterval);
722 
723   if (min_w == max_w)
724     gst_structure_set (video_structure, "width", G_TYPE_INT, min_w, NULL);
725   else
726      gst_structure_set (video_structure,
727        "width", GST_TYPE_INT_RANGE, min_w, max_w, NULL);
728 
729   if (min_h == max_h)
730     gst_structure_set (video_structure, "height", G_TYPE_INT, min_h, NULL);
731   else
732      gst_structure_set (video_structure,
733        "height", GST_TYPE_INT_RANGE, min_h, max_h, NULL);
734 
735   if (min_fr == max_fr)
736     gst_structure_set (video_structure, "framerate",
737         GST_TYPE_FRACTION, min_fr, 1, NULL);
738   else
739      gst_structure_set (video_structure, "framerate",
740          GST_TYPE_FRACTION_RANGE, min_fr, 1, max_fr, 1, NULL);
741 
742   return video_caps;
743 }
744 
gst_dshow_configure_latency(IPin * pCapturePin,guint bufSizeMS)745 bool gst_dshow_configure_latency (IPin *pCapturePin, guint bufSizeMS)
746 {
747   HRESULT hr;
748   ALLOCATOR_PROPERTIES alloc_prop;
749   IAMBufferNegotiation * pNeg = NULL;
750   hr = pCapturePin->QueryInterface(IID_IAMBufferNegotiation, (void **)&pNeg);
751 
752   if(!SUCCEEDED (hr))
753     return FALSE;
754 
755   alloc_prop.cbAlign = -1;  // -1 means no preference.
756   alloc_prop.cbBuffer = bufSizeMS;
757   alloc_prop.cbPrefix = -1;
758   alloc_prop.cBuffers = -1;
759   hr = pNeg->SuggestAllocatorProperties (&alloc_prop);
760   return SUCCEEDED (hr);
761 }
762