?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Revert "DShow: fix RGB24 Colouring" (fix #21005)
[vlc.git] / modules / access / dshow / dshow.cpp
blob68a379f1cf675cf0216786ab32114495082398bd
1 /*****************************************************************************
2 * dshow.cpp : DirectShow access and access_demux module for vlc
3 *****************************************************************************
4 * Copyright (C) 2002-2004, 2006, 2008, 2010 the VideoLAN team
5 * $Id$
6 *
7 * Author: Gildas Bazin <gbazin@videolan.org>
8 * Damien Fouilleul <damienf@videolan.org>
9 *
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 *****************************************************************************/
24
25 /*****************************************************************************
26 * Preamble
27 *****************************************************************************/
28
29 #ifdef HAVE_CONFIG_H
30 # include "config.h"
31 #endif
32
33 #include <inttypes.h>
34 #include <list>
35 #include <string>
36 #include <assert.h>
37 #include <stdexcept>
38
39 #include <vlc_common.h>
40 #include <vlc_plugin.h>
41 #include <vlc_access.h>
42 #include <vlc_demux.h>
43
44 #include <vlc_dialog.h> /* vlc_dialog_display_error */
45 #include <vlc_charset.h> /* FromWide */
46
47 #include <initguid.h>
48 #include "vlc_dshow.h"
49
50 #include "access.h"
51 #include "filter.h"
52
53 #include "../src/win32/mta_holder.h"
54
55 namespace dshow {
56
57 /*****************************************************************************
58 * Access: local prototypes
59 *****************************************************************************/
60 static block_t *ReadCompressed( stream_t *, bool * );
61 static int AccessControl ( stream_t *, int, va_list );
62
63 static int Demux ( demux_t * );
64 static int DemuxControl( demux_t *, int, va_list );
65
66 static int OpenDevice( vlc_object_t *, access_sys_t *, std::string, bool );
67 static ComPtr<IBaseFilter> FindCaptureDevice( vlc_object_t *, std::string *,
68 std::list<std::string> *, bool );
69 static size_t EnumDeviceCaps( vlc_object_t *, IBaseFilter *,
70 int, int, int, int, int, int,
71 AM_MEDIA_TYPE *mt, size_t, bool );
72 static bool ConnectFilters( vlc_object_t *, access_sys_t *,
73 IBaseFilter *, CaptureFilter * );
74 static int FindDevices( const char *, char ***, char *** );
75
76 static void ShowPropertyPage( IUnknown * );
77 static void ShowDeviceProperties( vlc_object_t *, ICaptureGraphBuilder2 *,
78 IBaseFilter *, bool );
79 static void ShowTunerProperties( vlc_object_t *, ICaptureGraphBuilder2 *,
80 IBaseFilter *, bool );
81 static void ConfigTuner( vlc_object_t *, ICaptureGraphBuilder2 *,
82 IBaseFilter * );
83
84 /*****************************************************************************
85 * Module descriptor
86 *****************************************************************************/
87 static const int pi_tuner_input[] = { 0, 1, 2 };
88 static const char *const ppsz_tuner_input_text[] =
89 {N_("Default"), N_("Cable"), N_("Antenna")};
90
91 static const int pi_amtuner_mode[] = { AMTUNER_MODE_DEFAULT,
92 AMTUNER_MODE_TV,
93 AMTUNER_MODE_FM_RADIO,
94 AMTUNER_MODE_AM_RADIO,
95 AMTUNER_MODE_DSS };
96 static const char *const ppsz_amtuner_mode_text[] = { N_("Default"),
97 N_("TV"),
98 N_("FM radio"),
99 N_("AM radio"),
100 N_("DSS") };
101
102 #define VDEV_TEXT N_("Video device name")
103 #define VDEV_LONGTEXT N_( \
104 "Name of the video device that will be used by the " \
105 "DirectShow plugin. If you don't specify anything, the default device " \
106 "will be used.")
107 #define ADEV_TEXT N_("Audio device name")
108 #define ADEV_LONGTEXT N_( \
109 "Name of the audio device that will be used by the " \
110 "DirectShow plugin. If you don't specify anything, the default device " \
111 "will be used.")
112 #define SIZE_TEXT N_("Video size")
113 #define SIZE_LONGTEXT N_( \
114 "Size of the video that will be displayed by the " \
115 "DirectShow plugin. If you don't specify anything the default size for " \
116 "your device will be used. You can specify a standard size (cif, d1, ...) or <width>x<height>.")
117 #define ASPECT_TEXT N_("Picture aspect-ratio n:m")
118 #define ASPECT_LONGTEXT N_("Define input picture aspect-ratio to use. Default is 4:3" )
119 #define CHROMA_TEXT N_("Video input chroma format")
120 #define CHROMA_LONGTEXT N_( \
121 "Force the DirectShow video input to use a specific chroma format " \
122 "(eg. I420 (default), RV24, etc.)")
123 #define FPS_TEXT N_("Video input frame rate")
124 #define FPS_LONGTEXT N_( \
125 "Force the DirectShow video input to use a specific frame rate " \
126 "(eg. 0 means default, 25, 29.97, 50, 59.94, etc.)")
127 #define CONFIG_TEXT N_("Device properties")
128 #define CONFIG_LONGTEXT N_( \
129 "Show the properties dialog of the selected device before starting the " \
130 "stream.")
131 #define TUNER_TEXT N_("Tuner properties")
132 #define TUNER_LONGTEXT N_( \
133 "Show the tuner properties [channel selection] page." )
134 #define CHANNEL_TEXT N_("Tuner TV Channel")
135 #define CHANNEL_LONGTEXT N_( \
136 "Set the TV channel the tuner will set to " \
137 "(0 means default)." )
138 #define COUNTRY_TEXT N_("Tuner country code")
139 #define COUNTRY_LONGTEXT N_( \
140 "Set the tuner country code that establishes the current " \
141 "channel-to-frequency mapping (0 means default)." )
142 #define TUNER_INPUT_TEXT N_("Tuner input type")
143 #define TUNER_INPUT_LONGTEXT N_( \
144 "Select the tuner input type (Cable/Antenna)." )
145 #define VIDEO_IN_TEXT N_("Video input pin")
146 #define VIDEO_IN_LONGTEXT N_( \
147 "Select the video input source, such as composite, s-video, " \
148 "or tuner. Since these settings are hardware-specific, you should find good " \
149 "settings in the \"Device config\" area, and use those numbers here. -1 " \
150 "means that settings will not be changed.")
151 #define AUDIO_IN_TEXT N_("Audio input pin")
152 #define AUDIO_IN_LONGTEXT N_( \
153 "Select the audio input source. See the \"video input\" option." )
154 #define VIDEO_OUT_TEXT N_("Video output pin")
155 #define VIDEO_OUT_LONGTEXT N_( \
156 "Select the video output type. See the \"video input\" option." )
157 #define AUDIO_OUT_TEXT N_("Audio output pin")
158 #define AUDIO_OUT_LONGTEXT N_( \
159 "Select the audio output type. See the \"video input\" option." )
160
161 #define AMTUNER_MODE_TEXT N_("AM Tuner mode")
162 #define AMTUNER_MODE_LONGTEXT N_( \
163 "AM Tuner mode. Can be one of Default (0), TV (1), " \
164 "AM Radio (2), FM Radio (3) or DSS (4).")
165
166 #define AUDIO_CHANNELS_TEXT N_("Number of audio channels")
167 #define AUDIO_CHANNELS_LONGTEXT N_( \
168 "Select audio input format with the given number of audio channels (if non 0)" )
169
170 #define AUDIO_SAMPLERATE_TEXT N_("Audio sample rate")
171 #define AUDIO_SAMPLERATE_LONGTEXT N_( \
172 "Select audio input format with the given sample rate (if non 0)" )
173
174 #define AUDIO_BITSPERSAMPLE_TEXT N_("Audio bits per sample")
175 #define AUDIO_BITSPERSAMPLE_LONGTEXT N_( \
176 "Select audio input format with the given bits/sample (if non 0)" )
177
178 static int CommonOpen ( vlc_object_t *, access_sys_t *, bool );
179 static void CommonClose( vlc_object_t *, access_sys_t * );
180
181 static int AccessOpen ( vlc_object_t * );
182 static void AccessClose( vlc_object_t * );
183
184 static int DemuxOpen ( vlc_object_t * );
185 static void DemuxClose ( vlc_object_t * );
186
187 } // namespace
188
189 using namespace dshow;
190
191 vlc_module_begin ()
192 set_shortname( N_("DirectShow") )
193 set_description( N_("DirectShow input") )
194 set_category( CAT_INPUT )
195 set_subcategory( SUBCAT_INPUT_ACCESS )
196 add_string( "dshow-vdev", NULL, VDEV_TEXT, VDEV_LONGTEXT, false)
197 change_string_cb( FindDevices )
198
199 add_string( "dshow-adev", NULL, ADEV_TEXT, ADEV_LONGTEXT, false)
200 change_string_cb( FindDevices )
201
202 add_string( "dshow-size", NULL, SIZE_TEXT, SIZE_LONGTEXT, false)
203 change_safe()
204
205 add_string( "dshow-aspect-ratio", "4:3", ASPECT_TEXT, ASPECT_LONGTEXT, false)
206 change_safe()
207
208 add_string( "dshow-chroma", NULL, CHROMA_TEXT, CHROMA_LONGTEXT, true )
209 change_safe()
210
211 add_float( "dshow-fps", 0.0f, FPS_TEXT, FPS_LONGTEXT, true )
212 change_safe()
213
214 add_bool( "dshow-config", false, CONFIG_TEXT, CONFIG_LONGTEXT, true )
215
216 add_bool( "dshow-tuner", false, TUNER_TEXT, TUNER_LONGTEXT, true )
217
218 add_integer( "dshow-tuner-channel", 0, CHANNEL_TEXT, CHANNEL_LONGTEXT,
219 true )
220 change_safe()
221
222 add_integer( "dshow-tuner-country", 0, COUNTRY_TEXT, COUNTRY_LONGTEXT,
223 true )
224
225 add_integer( "dshow-tuner-input", 0, TUNER_INPUT_TEXT,
226 TUNER_INPUT_LONGTEXT, true )
227 change_integer_list( pi_tuner_input, ppsz_tuner_input_text )
228 change_safe()
229
230 add_integer( "dshow-video-input", -1, VIDEO_IN_TEXT,
231 VIDEO_IN_LONGTEXT, true )
232 change_safe()
233
234 add_integer( "dshow-video-output", -1, VIDEO_OUT_TEXT,
235 VIDEO_OUT_LONGTEXT, true )
236
237 add_integer( "dshow-audio-input", -1, AUDIO_IN_TEXT,
238 AUDIO_IN_LONGTEXT, true )
239 change_safe()
240
241 add_integer( "dshow-audio-output", -1, AUDIO_OUT_TEXT,
242 AUDIO_OUT_LONGTEXT, true )
243
244 add_integer( "dshow-amtuner-mode", AMTUNER_MODE_TV,
245 AMTUNER_MODE_TEXT, AMTUNER_MODE_LONGTEXT, false)
246 change_integer_list( pi_amtuner_mode, ppsz_amtuner_mode_text )
247 change_safe()
248
249 add_integer( "dshow-audio-channels", 0, AUDIO_CHANNELS_TEXT,
250 AUDIO_CHANNELS_LONGTEXT, true )
251 add_integer( "dshow-audio-samplerate", 0, AUDIO_SAMPLERATE_TEXT,
252 AUDIO_SAMPLERATE_LONGTEXT, true )
253 add_integer( "dshow-audio-bitspersample", 0, AUDIO_BITSPERSAMPLE_TEXT,
254 AUDIO_BITSPERSAMPLE_LONGTEXT, true )
255
256 add_shortcut( "dshow" )
257 set_capability( "access", 1 )
258 set_callbacks( DemuxOpen, DemuxClose )
259
260 add_submodule ()
261 set_description( N_("DirectShow input") )
262 add_shortcut( "dshow" )
263 set_capability( "access", 0 )
264 set_callbacks( AccessOpen, AccessClose )
265
266 vlc_module_end ()
267
268 namespace dshow {
269
270 struct ComContext
271 {
272 ComContext( int mode )
273 {
274 if( FAILED( CoInitializeEx( NULL, mode ) ) )
275 throw std::runtime_error( "CoInitializeEx failed" );
276 }
277 ~ComContext()
278 {
279 CoUninitialize();
280 }
281 };
282
283 /*****************************************************************************
284 * DirectShow elementary stream descriptor
285 *****************************************************************************/
286 struct dshow_stream_t
287 {
288 std::string devicename;
289 ComPtr<IBaseFilter> p_device_filter;
290 ComPtr<CaptureFilter> p_capture_filter;
291 AM_MEDIA_TYPE mt;
292
293 union
294 {
295 VIDEOINFOHEADER video;
296 WAVEFORMATEX audio;
297
298 } header;
299
300 int i_fourcc;
301 es_out_id_t *p_es;
302
303 bool b_pts;
304
305 std::deque<VLCMediaSample> samples_queue;
306 };
307
308 /*****************************************************************************
309 * DirectShow utility functions
310 *****************************************************************************/
311 static void CreateDirectShowGraph( access_sys_t *p_sys )
312 {
313 p_sys->i_crossbar_route_depth = 0;
314
315 /* Create directshow filter graph */
316 if( SUCCEEDED( CoCreateInstance( CLSID_FilterGraph, 0, CLSCTX_INPROC,
317 (REFIID)IID_IFilterGraph, (void**)p_sys->p_graph.GetAddressOf() ) ) )
318 {
319 /* Create directshow capture graph builder if available */
320 if( SUCCEEDED( CoCreateInstance( CLSID_CaptureGraphBuilder2, 0,
321 CLSCTX_INPROC, (REFIID)IID_ICaptureGraphBuilder2,
322 (void**)p_sys->p_capture_graph_builder2.GetAddressOf() ) ) )
323 {
324 p_sys->p_capture_graph_builder2->
325 SetFiltergraph((IGraphBuilder *)p_sys->p_graph.Get() );
326 }
327
328 p_sys->p_graph->QueryInterface( IID_IMediaControl,
329 (void**)p_sys->p_control.GetAddressOf() );
330 }
331 }
332
333 static void DeleteDirectShowGraph( vlc_object_t *p_this, access_sys_t *p_sys )
334 {
335 DeleteCrossbarRoutes( p_sys );
336
337 /* Remove filters from graph */
338 msg_Dbg( p_this, "DeleteDirectShowGraph: Removing filters" );
339 std::vector<dshow_stream_t*>::iterator it = p_sys->pp_streams.begin();
340 std::vector<dshow_stream_t*>::iterator end = p_sys->pp_streams.end();
341 for( ; it != end; ++it )
342 {
343 /* RemoveFilter does an undocumented Release()
344 * but does not set item to NULL */
345 msg_Dbg( p_this, "DeleteDirectShowGraph: Removing capture filter" );
346 p_sys->p_graph->RemoveFilter( (*it)->p_capture_filter.Get() );
347 (*it)->p_capture_filter.Reset();
348
349 msg_Dbg( p_this, "DeleteDirectShowGraph: Removing device filter" );
350 p_sys->p_graph->RemoveFilter( (*it)->p_device_filter.Get() );
351 (*it)->p_device_filter.Reset();
352 }
353
354 /* Release directshow objects */
355 p_sys->p_control.Reset();
356 p_sys->p_capture_graph_builder2.Reset();
357 p_sys->p_graph.Reset();
358 }
359
360 /*****************************************************************************
361 * CommonOpen: open direct show device
362 *****************************************************************************/
363 static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
364 bool b_access_demux )
365 {
366 char *psz_val;
367
368 /* Get/parse options and open device(s) */
369 std::string vdevname, adevname;
370 int i_width = 0, i_height = 0;
371 vlc_fourcc_t i_chroma = 0;
372 bool b_use_audio = true;
373 bool b_use_video = true;
374
375 var_Create( p_this, "dshow-config", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
376 var_Create( p_this, "dshow-tuner", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
377 psz_val = var_CreateGetString( p_this, "dshow-vdev" );
378 if( psz_val )
379 {
380 msg_Dbg( p_this, "dshow-vdev: %s", psz_val ) ;
381 /* skip none device */
382 if ( strncasecmp( psz_val, "none", 4 ) != 0 )
383 vdevname = std::string( psz_val );
384 else
385 b_use_video = false ;
386 }
387 free( psz_val );
388
389 psz_val = var_CreateGetString( p_this, "dshow-adev" );
390 if( psz_val )
391 {
392 msg_Dbg( p_this, "dshow-adev: %s", psz_val ) ;
393 /* skip none device */
394 if ( strncasecmp( psz_val, "none", 4 ) != 0 )
395 adevname = std::string( psz_val );
396 else
397 b_use_audio = false ;
398 }
399 free( psz_val );
400
401 /* DShow Size */
402 static struct {
403 const char *psz_size;
404 int i_width;
405 int i_height;
406 } size_table[] =
407 { { "subqcif", 128, 96 },
408 { "qsif", 160, 120 },
409 { "qcif", 176, 144 },
410 { "sif", 320, 240 },
411 { "cif", 352, 288 },
412 { "d1", 640, 480 },
413 { 0, 0, 0 },
414 };
415
416 psz_val = var_CreateGetString( p_this, "dshow-size" );
417 if( !EMPTY_STR(psz_val) )
418 {
419 int i;
420 for( i = 0; size_table[i].psz_size; i++ )
421 {
422 if( !strcmp( psz_val, size_table[i].psz_size ) )
423 {
424 i_width = size_table[i].i_width;
425 i_height = size_table[i].i_height;
426 break;
427 }
428 }
429 if( !size_table[i].psz_size ) /* Try to parse "WidthxHeight" */
430 {
431 char *psz_parser;
432 i_width = strtol( psz_val, &psz_parser, 0 );
433 if( *psz_parser == 'x' || *psz_parser == 'X')
434 {
435 i_height = strtol( psz_parser + 1, &psz_parser, 0 );
436 }
437 msg_Dbg( p_this, "width x height %dx%d", i_width, i_height );
438 }
439 }
440 free( psz_val );
441
442 /* Chroma */
443 psz_val = var_CreateGetString( p_this, "dshow-chroma" );
444 i_chroma = vlc_fourcc_GetCodecFromString( VIDEO_ES, psz_val );
445 free( psz_val );
446
447 var_Create( p_this, "dshow-fps", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
448 var_Create( p_this, "dshow-tuner-channel",
449 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
450 var_Create( p_this, "dshow-tuner-country",
451 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
452 var_Create( p_this, "dshow-tuner-input",
453 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
454
455 var_Create( p_this, "dshow-amtuner-mode",
456 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
457
458 var_Create( p_this, "dshow-video-input", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
459 var_Create( p_this, "dshow-audio-input", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
460 var_Create( p_this, "dshow-video-output", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
461 var_Create( p_this, "dshow-audio-output", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
462
463
464 /* Initialize some data */
465 p_sys->i_width = i_width;
466 p_sys->i_height = i_height;
467 p_sys->i_chroma = i_chroma;
468
469 /* Build directshow graph */
470 CreateDirectShowGraph( p_sys );
471
472 vlc_mutex_init( &p_sys->lock );
473 vlc_cond_init( &p_sys->wait );
474
475 if( !b_use_video && !b_use_audio )
476 {
477 vlc_dialog_display_error( p_this, _("Capture failed"),
478 _("No video or audio device selected.") );
479 return VLC_EGENERIC ;
480 }
481
482 if( !b_use_video )
483 msg_Dbg( p_this, "skipping video device" ) ;
484 bool b_err_video = false ;
485
486 if( b_use_video && OpenDevice( p_this, p_sys, vdevname, 0 ) != VLC_SUCCESS )
487 {
488 msg_Err( p_this, "can't open video device");
489 b_err_video = true ;
490 }
491
492 if ( b_use_video && !b_err_video )
493 {
494 /* Check if we can handle the demuxing ourselves or need to spawn
495 * a demuxer module */
496 dshow_stream_t *p_stream = *(p_sys->pp_streams.rbegin());
497
498 if( p_stream->mt.majortype == MEDIATYPE_Video )
499 {
500 if( /* Raw DV stream */
501 p_stream->i_fourcc == VLC_CODEC_DV ||
502 /* Raw MPEG video stream */
503 p_stream->i_fourcc == VLC_CODEC_MPGV )
504 {
505 b_use_audio = false;
506
507 if( b_access_demux )
508 {
509 /* Let the access (only) take care of that */
510 return VLC_EGENERIC;
511 }
512 }
513 }
514
515 if( p_stream->mt.majortype == MEDIATYPE_Stream )
516 {
517 b_use_audio = false;
518
519 if( b_access_demux )
520 {
521 /* Let the access (only) take care of that */
522 return VLC_EGENERIC;
523 }
524
525 if( var_GetBool( p_this, "dshow-tuner" ) )
526 {
527 /* FIXME: we do MEDIATYPE_Stream here so we don't do
528 * it twice. */
529 ShowTunerProperties( p_this, p_sys->p_capture_graph_builder2.Get(),
530 p_stream->p_device_filter.Get(), 0 );
531 }
532 }
533 }
534
535 if( !b_use_audio )
536 msg_Dbg( p_this, "skipping audio device") ;
537
538 bool b_err_audio = false ;
539
540 if( b_use_audio && OpenDevice( p_this, p_sys, adevname, 1 ) != VLC_SUCCESS )
541 {
542 msg_Err( p_this, "can't open audio device");
543 b_err_audio = true ;
544 }
545
546 if( ( b_use_video && b_err_video && b_use_audio && b_err_audio ) ||
547 ( !b_use_video && b_use_audio && b_err_audio ) ||
548 ( b_use_video && !b_use_audio && b_err_video ) )
549 {
550 msg_Err( p_this, "FATAL: could not open ANY device" ) ;
551 vlc_dialog_display_error( p_this, _("Capture failed"),
552 _("VLC cannot open ANY capture device. "
553 "Check the error log for details.") );
554 return VLC_EGENERIC ;
555 }
556
557 for( int i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
558 {
559 int i_val = var_GetInteger( p_this, "dshow-video-input" );
560 if( i_val >= 0 )
561 p_sys->crossbar_routes[i].VideoInputIndex = i_val;
562 i_val = var_GetInteger( p_this, "dshow-video-output" );
563 if( i_val >= 0 )
564 p_sys->crossbar_routes[i].VideoOutputIndex = i_val;
565 i_val = var_GetInteger( p_this, "dshow-audio-input" );
566 if( i_val >= 0 )
567 p_sys->crossbar_routes[i].AudioInputIndex = i_val;
568 i_val = var_GetInteger( p_this, "dshow-audio-output" );
569 if( i_val >= 0 )
570 p_sys->crossbar_routes[i].AudioOutputIndex = i_val;
571
572 IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar.Get();
573 LONG VideoInputIndex = p_sys->crossbar_routes[i].VideoInputIndex;
574 LONG VideoOutputIndex = p_sys->crossbar_routes[i].VideoOutputIndex;
575 LONG AudioInputIndex = p_sys->crossbar_routes[i].AudioInputIndex;
576 LONG AudioOutputIndex = p_sys->crossbar_routes[i].AudioOutputIndex;
577
578 if( SUCCEEDED(pXbar->Route(VideoOutputIndex, VideoInputIndex)) )
579 {
580 msg_Dbg( p_this, "crossbar at depth %d, routed video "
581 "output %ld to video input %ld", i, VideoOutputIndex,
582 VideoInputIndex );
583
584 if( AudioOutputIndex != -1 && AudioInputIndex != -1 )
585 {
586 if( SUCCEEDED( pXbar->Route(AudioOutputIndex,
587 AudioInputIndex)) )
588 {
589 msg_Dbg(p_this, "crossbar at depth %d, routed audio "
590 "output %ld to audio input %ld", i,
591 AudioOutputIndex, AudioInputIndex );
592 }
593 }
594 }
595 else
596 msg_Err( p_this, "crossbar at depth %d could not route video "
597 "output %ld to input %ld", i, VideoOutputIndex, VideoInputIndex );
598 }
599
600 /*
601 ** Show properties pages from other filters in graph
602 */
603 if( var_GetBool( p_this, "dshow-config" ) )
604 {
605 for( int i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
606 {
607 IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar.Get();
608 ComPtr<IBaseFilter> p_XF;
609
610 if( SUCCEEDED( pXbar->QueryInterface( IID_IBaseFilter,
611 (void**)p_XF.ReleaseAndGetAddressOf() ) ) )
612 {
613 ShowPropertyPage( p_XF.Get() );
614 }
615 }
616 }
617
618 /* Initialize some data */
619 p_sys->i_current_stream = 0;
620
621 if( p_sys->pp_streams.empty() ) return VLC_EGENERIC;
622
623 return VLC_SUCCESS;
624 }
625
626 /*****************************************************************************
627 * DemuxOpen: open direct show device as an access_demux module
628 *****************************************************************************/
629 static int DemuxOpen( vlc_object_t *p_this )
630 {
631 demux_t *p_demux = (demux_t *)p_this;
632 access_sys_t *p_sys;
633
634 if (p_demux->out == NULL)
635 return VLC_EGENERIC;
636
637 p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
638 if( !p_sys )
639 return VLC_ENOMEM;
640 p_demux->p_sys = (demux_sys_t *)p_sys;
641
642 ComContext ctx( COINIT_MULTITHREADED );
643
644 if( vlc_mta_acquire( p_this ) == false )
645 {
646 msg_Err( p_this, "Failed to acquire MTA" );
647 return VLC_EGENERIC;
648 }
649
650 if( CommonOpen( p_this, p_sys, true ) != VLC_SUCCESS )
651 {
652 CommonClose( p_this, p_sys );
653 return VLC_EGENERIC;
654 }
655
656 /* Everything is ready. Let's rock baby */
657 msg_Dbg( p_this, "Playing...");
658 if( FAILED( p_sys->p_control->Run() ) )
659 {
660 msg_Err( p_this, "Failed to run graph. Capture device may be in use." );
661 CommonClose( p_this, p_sys );
662 return VLC_EGENERIC;
663 }
664
665 p_demux->pf_demux = Demux;
666 p_demux->pf_control = DemuxControl;
667
668 std::vector<dshow_stream_t*>::iterator it = p_sys->pp_streams.begin();
669 std::vector<dshow_stream_t*>::iterator end = p_sys->pp_streams.end();
670 for ( ; it != end; ++it )
671 {
672 dshow_stream_t *p_stream = *it;
673 es_format_t fmt;
674
675 if( p_stream->mt.majortype == MEDIATYPE_Video )
676 {
677 char *psz_aspect = var_CreateGetString( p_this, "dshow-aspect-ratio" );
678 char *psz_delim = !EMPTY_STR( psz_aspect ) ? strchr( psz_aspect, ':' ) : NULL;
679
680 es_format_Init( &fmt, VIDEO_ES, p_stream->i_fourcc );
681
682 fmt.video.i_width = p_stream->header.video.bmiHeader.biWidth;
683 fmt.video.i_height = p_stream->header.video.bmiHeader.biHeight;
684
685 if( psz_delim )
686 {
687 fmt.video.i_sar_num = atoi( psz_aspect ) * fmt.video.i_height;
688 fmt.video.i_sar_den = atoi( psz_delim + 1 ) * fmt.video.i_width;
689 }
690 else
691 {
692 fmt.video.i_sar_num = 4 * fmt.video.i_height;
693 fmt.video.i_sar_den = 3 * fmt.video.i_width;
694 }
695 free( psz_aspect );
696
697 if( !p_stream->header.video.bmiHeader.biCompression )
698 {
699 /* RGB DIB are coded from bottom to top */
700 fmt.video.orientation = ORIENT_BOTTOM_LEFT;
701 }
702
703 /* Setup rgb mask for RGB formats */
704 if( p_stream->i_fourcc == VLC_CODEC_RGB24 )
705 {
706 /* This is in BGR format */
707 fmt.video.i_bmask = 0x00ff0000;
708 fmt.video.i_gmask = 0x0000ff00;
709 fmt.video.i_rmask = 0x000000ff;
710 }
711
712 if( p_stream->header.video.AvgTimePerFrame )
713 {
714 fmt.video.i_frame_rate = 10000000;
715 fmt.video.i_frame_rate_base =
716 p_stream->header.video.AvgTimePerFrame;
717 }
718 }
719 else if( p_stream->mt.majortype == MEDIATYPE_Audio )
720 {
721 es_format_Init( &fmt, AUDIO_ES, p_stream->i_fourcc );
722
723 fmt.audio.i_channels = p_stream->header.audio.nChannels;
724 fmt.audio.i_rate = p_stream->header.audio.nSamplesPerSec;
725 fmt.audio.i_bitspersample = p_stream->header.audio.wBitsPerSample;
726 fmt.audio.i_blockalign = fmt.audio.i_channels *
727 fmt.audio.i_bitspersample / 8;
728 fmt.i_bitrate = fmt.audio.i_channels * fmt.audio.i_rate *
729 fmt.audio.i_bitspersample;
730 }
731
732 p_stream->p_es = es_out_Add( p_demux->out, &fmt );
733 }
734
735 p_sys->i_start = vlc_tick_now();
736 return VLC_SUCCESS;
737 }
738
739 /*****************************************************************************
740 * AccessOpen: open direct show device as an access module
741 *****************************************************************************/
742 static int AccessOpen( vlc_object_t *p_this )
743 {
744 stream_t *p_access = (stream_t*)p_this;
745 access_sys_t *p_sys;
746
747 p_access->p_sys = p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
748 if( !p_sys )
749 return VLC_ENOMEM;
750
751 ComContext ctx( COINIT_MULTITHREADED );
752
753 if( vlc_mta_acquire( p_this ) == false )
754 {
755 msg_Err( p_this, "Failed to acquire MTA" );
756 return VLC_EGENERIC;
757 }
758
759 if( CommonOpen( p_this, p_sys, false ) != VLC_SUCCESS )
760 {
761 CommonClose( p_this, p_sys );
762 return VLC_EGENERIC;
763 }
764
765 /* Setup Access */
766 p_access->pf_read = NULL;
767 p_access->pf_block = ReadCompressed;
768 p_access->pf_control = AccessControl;
769 p_access->pf_seek = NULL;
770 p_access->p_sys = p_sys;
771
772 /* Everything is ready. Let's rock baby */
773 msg_Dbg( p_this, "Playing...");
774 if( FAILED( p_sys->p_control->Run() ) )
775 {
776 msg_Err( p_this, "Failed to run graph. Capture device may be in use." );
777 CommonClose( p_this, p_sys );
778 return VLC_EGENERIC;
779 }
780
781 return VLC_SUCCESS;
782 }
783
784 /*****************************************************************************
785 * CommonClose: close device
786 *****************************************************************************/
787 static void CommonClose( vlc_object_t *p_this, access_sys_t *p_sys )
788 {
789 msg_Dbg( p_this, "releasing DirectShow");
790
791 DeleteDirectShowGraph( p_this, p_sys );
792
793 vlc_delete_all( p_sys->pp_streams );
794
795 vlc_mutex_destroy( &p_sys->lock );
796 vlc_cond_destroy( &p_sys->wait );
797
798 vlc_mta_release( p_this );
799
800 free( p_sys );
801 }
802
803 /*****************************************************************************
804 * AccessClose: close device
805 *****************************************************************************/
806 static void AccessClose( vlc_object_t *p_this )
807 {
808 stream_t *p_access = (stream_t *)p_this;
809 access_sys_t *p_sys = (access_sys_t *)p_access->p_sys;
810
811 ComContext ctx( COINIT_MULTITHREADED );
812
813 /* Stop capturing stuff */
814 p_sys->p_control->Stop();
815
816 CommonClose( p_this, p_sys );
817 }
818
819 /*****************************************************************************
820 * DemuxClose: close device
821 *****************************************************************************/
822 static void DemuxClose( vlc_object_t *p_this )
823 {
824 demux_t *p_demux = (demux_t *)p_this;
825 access_sys_t *p_sys = (access_sys_t *)p_demux->p_sys;
826
827 ComContext ctx( COINIT_MULTITHREADED );
828
829 /* Stop capturing stuff */
830 p_sys->p_control->Stop();
831
832 CommonClose( p_this, p_sys );
833 }
834
835 /****************************************************************************
836 * ConnectFilters
837 ****************************************************************************/
838 static bool ConnectFilters( vlc_object_t *p_this, access_sys_t *p_sys,
839 IBaseFilter *p_filter,
840 CaptureFilter *p_capture_filter )
841 {
842 ComPtr<CapturePin> p_input_pin = p_capture_filter->CustomGetPin();
843
844 AM_MEDIA_TYPE mediaType = p_input_pin->CustomGetMediaType();
845
846 if( p_sys->p_capture_graph_builder2 )
847 {
848 if( FAILED(p_sys->p_capture_graph_builder2->
849 RenderStream( &PIN_CATEGORY_CAPTURE, &mediaType.majortype,
850 p_filter, 0, (IBaseFilter *)p_capture_filter )) )
851 {
852 return false;
853 }
854
855 // Sort out all the possible video inputs
856 // The class needs to be given the capture filters ANALOGVIDEO input pin
857 ComPtr<IEnumPins> pins;
858 if( ( mediaType.majortype == MEDIATYPE_Video ||
859 mediaType.majortype == MEDIATYPE_Stream ) &&
860 SUCCEEDED(p_filter->EnumPins(pins.GetAddressOf())) )
861 {
862 ComPtr<IPin> pP;
863 ULONG n;
864 PIN_INFO pinInfo;
865 BOOL Found = FALSE;
866 GUID guid;
867 DWORD dw;
868
869 while( !Found && ( S_OK == pins->Next(1, pP.ReleaseAndGetAddressOf(), &n) ) )
870 {
871 if( S_OK == pP->QueryPinInfo(&pinInfo) )
872 {
873 ComPtr<IKsPropertySet> pKs;
874 // is this pin an ANALOGVIDEOIN input pin?
875 if( pinInfo.dir == PINDIR_INPUT &&
876 pP->QueryInterface( IID_IKsPropertySet,
877 (void**)pKs.GetAddressOf() ) == S_OK )
878 {
879 if( pKs->Get( AMPROPSETID_Pin,
880 AMPROPERTY_PIN_CATEGORY, NULL, 0,
881 &guid, sizeof(GUID), &dw ) == S_OK )
882 {
883 if( guid == PIN_CATEGORY_ANALOGVIDEOIN )
884 {
885 // recursively search crossbar routes
886 FindCrossbarRoutes( p_this, p_sys, pP.Get(), 0 );
887 // found it
888 Found = TRUE;
889 }
890 }
891 }
892 pinInfo.pFilter->Release();
893 }
894 }
895 msg_Dbg( p_this, "ConnectFilters: graph_builder2 available.") ;
896 if ( !Found )
897 msg_Warn( p_this, "ConnectFilters: No crossBar routes found (incompatible pin types)" ) ;
898 }
899 return true;
900 }
901 else
902 {
903 ComPtr<IEnumPins> p_enumpins;
904 ComPtr<IPin> p_pin;
905
906 if( S_OK != p_filter->EnumPins( p_enumpins.GetAddressOf() ) ) return false;
907
908 while( S_OK == p_enumpins->Next( 1, p_pin.ReleaseAndGetAddressOf(), NULL ) )
909 {
910 PIN_DIRECTION pin_dir;
911 p_pin->QueryDirection( &pin_dir );
912
913 if( pin_dir == PINDIR_OUTPUT &&
914 p_sys->p_graph->ConnectDirect( p_pin.Get(), p_input_pin.Get(),
915 0 ) == S_OK )
916 {
917 return true;
918 }
919 }
920 return false;
921 }
922 }
923
924 /*
925 * get fourcc priority from arbritary preference, the higher the better
926 */
927 static int GetFourCCPriority( int i_fourcc )
928 {
929 switch( i_fourcc )
930 {
931 case VLC_CODEC_I420:
932 case VLC_CODEC_FL32:
933 return 9;
934 case VLC_CODEC_YV12:
935 case VLC_FOURCC('a','r','a','w'):
936 return 8;
937 case VLC_CODEC_RGB24:
938 return 7;
939 case VLC_CODEC_YUYV:
940 case VLC_CODEC_RGB32:
941 case VLC_CODEC_RGBA:
942 return 6;
943 }
944
945 return 0;
946 }
947
948 #define MAX_MEDIA_TYPES 32
949
950 static int OpenDevice( vlc_object_t *p_this, access_sys_t *p_sys,
951 std::string devicename, bool b_audio )
952 {
953 ComPtr<IBaseFilter> p_device_filter;
954
955 /* See if device is already opened */
956 std::vector<dshow_stream_t*>::iterator it = p_sys->pp_streams.begin();
957 std::vector<dshow_stream_t*>::iterator end = p_sys->pp_streams.end();
958 for ( ; it != end; ++it )
959 {
960 if( devicename.size() &&
961 (*it)->devicename == devicename )
962 {
963 /* Already opened */
964 wchar_t *pwsz_devicename = ToWide( devicename.c_str() );
965
966 if( likely( pwsz_devicename ) )
967 p_sys->p_graph->FindFilterByName( pwsz_devicename, p_device_filter.GetAddressOf() );
968
969 free( pwsz_devicename );
970
971 if( !p_device_filter )
972 {
973 msg_Err( p_this, "Device '%s' already opened, but couldn't be retrieved", devicename.c_str() );
974 return VLC_EGENERIC;
975 }
976
977 break;
978 }
979 }
980
981 if( !p_device_filter )
982 {
983 std::list<std::string> list_devices;
984
985 /* Enumerate devices and display their names */
986 FindCaptureDevice( p_this, NULL, &list_devices, b_audio );
987 if( list_devices.empty() )
988 return VLC_EGENERIC;
989
990 std::list<std::string>::iterator iter;
991 for( iter = list_devices.begin(); iter != list_devices.end(); ++iter )
992 msg_Dbg( p_this, "found device: %s", iter->c_str() );
993
994 /* If no device name was specified, pick the 1st one */
995 if( devicename.size() == 0 )
996 {
997 /* When none selected */
998 devicename = *list_devices.begin();
999 msg_Dbg( p_this, "asking for default device: %s", devicename.c_str() ) ;
1000 }
1001 else
1002 msg_Dbg( p_this, "asking for device: %s", devicename.c_str() ) ;
1003
1004 // Use the system device enumerator and class enumerator to find
1005 // a capture/preview device, such as a desktop USB video camera.
1006 p_device_filter = FindCaptureDevice( p_this, &devicename, NULL, b_audio );
1007
1008 if( p_device_filter )
1009 msg_Dbg( p_this, "using device: %s", devicename.c_str() );
1010 else
1011 {
1012 msg_Err( p_this, "can't use device: %s, unsupported device type",
1013 devicename.c_str() );
1014 vlc_dialog_display_error( p_this, _("Capture failed"),
1015 _("The device you selected cannot be used, because its "
1016 "type is not supported.") );
1017 return VLC_EGENERIC;
1018 }
1019
1020 /* Add the device filter to the graph (seems necessary with VfW before
1021 * accessing pin attributes). */
1022
1023 HRESULT hr = E_FAIL;
1024 wchar_t *pwsz_devicename = ToWide( devicename.c_str() );
1025
1026 if( likely( pwsz_devicename ) )
1027 hr = p_sys->p_graph->AddFilter( p_device_filter.Get(), pwsz_devicename );
1028
1029 free( pwsz_devicename );
1030
1031 if( FAILED( hr ) )
1032 {
1033 msg_Err( p_this, "Error adding device '%s' to the graph", devicename.c_str() );
1034 return VLC_EGENERIC;
1035 }
1036 }
1037
1038 // Retreive acceptable media types supported by device
1039 AM_MEDIA_TYPE media_types[MAX_MEDIA_TYPES];
1040 size_t media_count =
1041 EnumDeviceCaps( p_this, p_device_filter.Get(), b_audio ? 0 : p_sys->i_chroma,
1042 p_sys->i_width, p_sys->i_height,
1043 b_audio ? var_CreateGetInteger( p_this, "dshow-audio-channels" ) : 0,
1044 b_audio ? var_CreateGetInteger( p_this, "dshow-audio-samplerate" ) : 0,
1045 b_audio ? var_CreateGetInteger( p_this, "dshow-audio-bitspersample" ) : 0,
1046 media_types, MAX_MEDIA_TYPES, b_audio );
1047
1048 AM_MEDIA_TYPE *mt = NULL;
1049
1050 if( media_count > 0 )
1051 {
1052 mt = (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE) * media_count);
1053
1054 // Order and copy returned media types according to arbitrary
1055 // fourcc priority
1056 for( size_t c = 0; c < media_count; c++ )
1057 {
1058 int slot_priority =
1059 GetFourCCPriority(GetFourCCFromMediaType(media_types[c]));
1060 size_t slot_copy = c;
1061 for( size_t d = c+1; d < media_count; d++ )
1062 {
1063 int priority =
1064 GetFourCCPriority(GetFourCCFromMediaType(media_types[d]));
1065 if( priority > slot_priority )
1066 {
1067 slot_priority = priority;
1068 slot_copy = d;
1069 }
1070 }
1071 if( slot_copy != c )
1072 {
1073 mt[c] = media_types[slot_copy];
1074 media_types[slot_copy] = media_types[c];
1075 }
1076 else
1077 {
1078 mt[c] = media_types[c];
1079 }
1080 }
1081 }
1082 else {
1083 /* capture device */
1084 msg_Err( p_this, "capture device '%s' does not support required parameters !", devicename.c_str() );
1085 vlc_dialog_display_error( p_this, _("Capture failed"),
1086 _("The capture device \"%s\" does not support the "
1087 "required parameters."), devicename.c_str() );
1088 return VLC_EGENERIC;
1089 }
1090
1091 /* Create and add our capture filter */
1092 ComPtr<CaptureFilter> p_capture_filter(
1093 new CaptureFilter( p_this, p_sys, mt, media_count ) );
1094 p_sys->p_graph->AddFilter( p_capture_filter.Get(), 0 );
1095
1096 /* Attempt to connect one of this device's capture output pins */
1097 msg_Dbg( p_this, "connecting filters" );
1098 if( ConnectFilters( p_this, p_sys, p_device_filter.Get(), p_capture_filter.Get() ) )
1099 {
1100 /* Success */
1101 msg_Dbg( p_this, "filters connected successfully !" );
1102
1103 dshow_stream_t dshow_stream;
1104 dshow_stream.devicename = devicename;
1105 dshow_stream.b_pts = false;
1106 dshow_stream.p_es = 0;
1107 dshow_stream.mt =
1108 p_capture_filter->CustomGetPin()->CustomGetMediaType();
1109
1110 /* Show Device properties. Done here so the VLC stream is setup with
1111 * the proper parameters. */
1112 if( var_GetBool( p_this, "dshow-config" ) )
1113 {
1114 ShowDeviceProperties( p_this, p_sys->p_capture_graph_builder2.Get(),
1115 p_device_filter.Get(), b_audio );
1116 }
1117
1118 ConfigTuner( p_this, p_sys->p_capture_graph_builder2.Get(),
1119 p_device_filter.Get() );
1120
1121 if( var_GetBool( p_this, "dshow-tuner" ) &&
1122 dshow_stream.mt.majortype != MEDIATYPE_Stream )
1123 {
1124 /* FIXME: we do MEDIATYPE_Stream later so we don't do it twice. */
1125 ShowTunerProperties( p_this, p_sys->p_capture_graph_builder2.Get(),
1126 p_device_filter.Get(), b_audio );
1127 }
1128
1129 dshow_stream.mt =
1130 p_capture_filter->CustomGetPin()->CustomGetMediaType();
1131
1132 dshow_stream.i_fourcc = GetFourCCFromMediaType( dshow_stream.mt );
1133 if( dshow_stream.i_fourcc )
1134 {
1135 if( dshow_stream.mt.majortype == MEDIATYPE_Video )
1136 {
1137 dshow_stream.header.video =
1138 *(VIDEOINFOHEADER *)dshow_stream.mt.pbFormat;
1139 msg_Dbg( p_this, "MEDIATYPE_Video" );
1140 msg_Dbg( p_this, "selected video pin accepts format: %4.4s",
1141 (char *)&dshow_stream.i_fourcc);
1142 }
1143 else if( dshow_stream.mt.majortype == MEDIATYPE_Audio )
1144 {
1145 dshow_stream.header.audio =
1146 *(WAVEFORMATEX *)dshow_stream.mt.pbFormat;
1147 msg_Dbg( p_this, "MEDIATYPE_Audio" );
1148 msg_Dbg( p_this, "selected audio pin accepts format: %4.4s",
1149 (char *)&dshow_stream.i_fourcc);
1150 }
1151 else if( dshow_stream.mt.majortype == MEDIATYPE_Stream )
1152 {
1153 msg_Dbg( p_this, "MEDIATYPE_Stream" );
1154 msg_Dbg( p_this, "selected stream pin accepts format: %4.4s",
1155 (char *)&dshow_stream.i_fourcc);
1156 }
1157 else
1158 {
1159 msg_Dbg( p_this, "unknown stream majortype" );
1160 goto fail;
1161 }
1162
1163 /* Add directshow elementary stream to our list */
1164 dshow_stream.p_device_filter = p_device_filter;
1165 dshow_stream.p_capture_filter = p_capture_filter;
1166
1167 p_sys->pp_streams.push_back( new dshow_stream_t(dshow_stream) );
1168
1169 return VLC_SUCCESS;
1170 }
1171 }
1172
1173 fail:
1174 /* Remove filters from graph */
1175 msg_Dbg( p_this, "OpenDevice: Removing filters" ) ;
1176 p_sys->p_graph->RemoveFilter( p_device_filter.Get() );
1177 p_sys->p_graph->RemoveFilter( p_capture_filter.Get() );
1178
1179 return VLC_EGENERIC;
1180 }
1181
1182 /* FindCaptureDevices:: This Function had two purposes :
1183 Returns the list of capture devices when p_listdevices != NULL
1184 Creates an IBaseFilter when p_devicename corresponds to an existing devname
1185 These actions *may* be requested whith a single call.
1186 */
1187 static ComPtr<IBaseFilter>
1188 FindCaptureDevice( vlc_object_t *p_this, std::string *p_devicename,
1189 std::list<std::string> *p_listdevices, bool b_audio )
1190 {
1191 ComPtr<IBaseFilter> p_base_filter;
1192 ComPtr<IMoniker> p_moniker;
1193 ULONG i_fetched;
1194 HRESULT hr;
1195 std::list<std::string> devicelist;
1196
1197 /* Create the system device enumerator */
1198 ComPtr<ICreateDevEnum> p_dev_enum;
1199
1200 hr = CoCreateInstance( CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC,
1201 IID_ICreateDevEnum, (void**)p_dev_enum.GetAddressOf() );
1202 if( FAILED(hr) )
1203 {
1204 msg_Err( p_this, "failed to create the device enumerator (0x%lx)", hr);
1205 return p_base_filter;
1206 }
1207
1208 /* Create an enumerator for the video capture devices */
1209 ComPtr<IEnumMoniker> p_class_enum;
1210 if( !b_audio )
1211 hr = p_dev_enum->CreateClassEnumerator( CLSID_VideoInputDeviceCategory,
1212 p_class_enum.GetAddressOf(), 0 );
1213 else
1214 hr = p_dev_enum->CreateClassEnumerator( CLSID_AudioInputDeviceCategory,
1215 p_class_enum.GetAddressOf(), 0 );
1216 if( FAILED(hr) )
1217 {
1218 msg_Err( p_this, "failed to create the class enumerator (0x%lx)", hr );
1219 return p_base_filter;
1220 }
1221
1222 /* If there are no enumerators for the requested type, then
1223 * CreateClassEnumerator will succeed, but p_class_enum will be NULL */
1224 if( p_class_enum == NULL )
1225 {
1226 msg_Err( p_this, "no %s capture device was detected", ( b_audio ? "audio" : "video" ) );
1227 return p_base_filter;
1228 }
1229
1230 /* Enumerate the devices */
1231
1232 /* Note that if the Next() call succeeds but there are no monikers,
1233 * it will return S_FALSE (which is not a failure). Therefore, we check
1234 * that the return code is S_OK instead of using SUCCEEDED() macro. */
1235
1236 while( p_class_enum->Next( 1, p_moniker.ReleaseAndGetAddressOf(), &i_fetched ) == S_OK )
1237 {
1238 /* Getting the property page to get the device name */
1239 ComPtr<IPropertyBag> p_bag;
1240 hr = p_moniker->BindToStorage( 0, 0, IID_IPropertyBag,
1241 (void**)p_bag.GetAddressOf() );
1242 if( SUCCEEDED(hr) )
1243 {
1244 VARIANT var;
1245 var.vt = VT_BSTR;
1246 hr = p_bag->Read( L"FriendlyName", &var, NULL );
1247 if( SUCCEEDED(hr) )
1248 {
1249 char *p_buf = FromWide( var.bstrVal );
1250 std::string devname(p_buf);
1251 free( p_buf) ;
1252
1253 int dup = 0;
1254 /* find out if this name is already used by a previously found device */
1255 std::list<std::string>::const_iterator iter = devicelist.begin();
1256 std::list<std::string>::const_iterator end = devicelist.end();
1257 std::string ordevname = devname ;
1258 while ( iter != end )
1259 {
1260 if( 0 == (*iter).compare( devname ) )
1261 { /* devname is on the list. Try another name with sequence
1262 number apended and then rescan until a unique entry is found*/
1263 char seq[16];
1264 snprintf(seq, 16, " #%d", ++dup);
1265 devname = ordevname + seq;
1266 iter = devicelist.begin();
1267 }
1268 else
1269 ++iter;
1270 }
1271 devicelist.push_back( devname );
1272
1273 if( p_devicename && *p_devicename == devname )
1274 {
1275 msg_Dbg( p_this, "asked for %s, binding to %s", p_devicename->c_str() , devname.c_str() ) ;
1276 /* NULL possibly means we don't need BindMoniker BindCtx ?? */
1277 hr = p_moniker->BindToObject( NULL, 0, IID_IBaseFilter,
1278 (void**)p_base_filter.GetAddressOf() );
1279 if( FAILED(hr) )
1280 {
1281 msg_Err( p_this, "couldn't bind moniker to filter "
1282 "object (0x%lx)", hr );
1283 return NULL;
1284 }
1285 return p_base_filter;
1286 }
1287 }
1288 }
1289 }
1290
1291 if( p_listdevices ) {
1292 devicelist.sort();
1293 *p_listdevices = devicelist;
1294 }
1295 return p_base_filter;
1296 }
1297
1298 static size_t EnumDeviceCaps( vlc_object_t *p_this, IBaseFilter *p_filter,
1299 int i_fourcc, int i_width, int i_height,
1300 int i_channels, int i_samplespersec,
1301 int i_bitspersample, AM_MEDIA_TYPE *mt,
1302 size_t mt_max, bool b_audio )
1303 {
1304 ComPtr<IEnumPins> p_enumpins;
1305 ComPtr<IPin> p_output_pin;
1306 ComPtr<IEnumMediaTypes> p_enummt;
1307 size_t mt_count = 0;
1308
1309 LONGLONG i_AvgTimePerFrame = 0;
1310 float r_fps = var_GetFloat( p_this, "dshow-fps" );
1311 if( r_fps )
1312 i_AvgTimePerFrame = 10000000000LL/(LONGLONG)(r_fps*1000.0f);
1313
1314 if( FAILED(p_filter->EnumPins( p_enumpins.GetAddressOf() )) )
1315 {
1316 msg_Dbg( p_this, "EnumDeviceCaps failed: no pin enumeration !");
1317 return 0;
1318 }
1319
1320 while( S_OK == p_enumpins->Next( 1, p_output_pin.ReleaseAndGetAddressOf(), NULL ) )
1321 {
1322 PIN_INFO info;
1323
1324 if( S_OK == p_output_pin->QueryPinInfo( &info ) )
1325 {
1326 msg_Dbg( p_this, "EnumDeviceCaps: %s pin: %S",
1327 info.dir == PINDIR_INPUT ? "input" : "output",
1328 info.achName );
1329 if( info.pFilter ) info.pFilter->Release();
1330 }
1331 }
1332
1333 p_enumpins->Reset();
1334
1335 while( !mt_count && p_enumpins->Next( 1, p_output_pin.ReleaseAndGetAddressOf(), NULL ) == S_OK )
1336 {
1337 PIN_INFO info;
1338
1339 if( S_OK == p_output_pin->QueryPinInfo( &info ) )
1340 {
1341 if( info.pFilter ) info.pFilter->Release();
1342 if( info.dir == PINDIR_INPUT )
1343 continue;
1344 msg_Dbg( p_this, "EnumDeviceCaps: trying pin %S", info.achName );
1345 }
1346
1347 AM_MEDIA_TYPE *p_mt;
1348
1349 /*
1350 ** Configure pin with a default compatible media if possible
1351 */
1352
1353 ComPtr<IAMStreamConfig> pSC;
1354 if( SUCCEEDED(p_output_pin->QueryInterface( IID_IAMStreamConfig,
1355 (void**)pSC.GetAddressOf() )) )
1356 {
1357 int piCount, piSize;
1358 if( SUCCEEDED(pSC->GetNumberOfCapabilities(&piCount, &piSize)) )
1359 {
1360 BYTE *pSCC= (BYTE *)CoTaskMemAlloc(piSize);
1361 if( NULL != pSCC )
1362 {
1363 int i_priority = ES_PRIORITY_NOT_DEFAULTABLE;
1364 for( int i=0; i<piCount; ++i )
1365 {
1366 if( SUCCEEDED(pSC->GetStreamCaps(i, &p_mt, pSCC)) )
1367 {
1368 int i_current_fourcc = GetFourCCFromMediaType( *p_mt );
1369 int i_current_priority = GetFourCCPriority(i_current_fourcc);
1370
1371 if( (i_fourcc && (i_current_fourcc != i_fourcc))
1372 || (i_priority > i_current_priority) )
1373 {
1374 // unwanted chroma, try next media type
1375 FreeMediaType( *p_mt );
1376 CoTaskMemFree( (PVOID)p_mt );
1377 continue;
1378 }
1379
1380 if( !b_audio && MEDIATYPE_Video == p_mt->majortype
1381 && FORMAT_VideoInfo == p_mt->formattype )
1382 {
1383 VIDEO_STREAM_CONFIG_CAPS *pVSCC = reinterpret_cast<VIDEO_STREAM_CONFIG_CAPS*>(pSCC);
1384 VIDEOINFOHEADER *pVih = reinterpret_cast<VIDEOINFOHEADER*>(p_mt->pbFormat);
1385
1386 if( i_AvgTimePerFrame )
1387 {
1388 if( pVSCC->MinFrameInterval > i_AvgTimePerFrame
1389 || i_AvgTimePerFrame > pVSCC->MaxFrameInterval )
1390 {
1391 // required frame rate not compatible, try next media type
1392 FreeMediaType( *p_mt );
1393 CoTaskMemFree( (PVOID)p_mt );
1394 continue;
1395 }
1396 pVih->AvgTimePerFrame = i_AvgTimePerFrame;
1397 }
1398
1399 if( i_width )
1400 {
1401 if(( !pVSCC->OutputGranularityX
1402 && i_width != pVSCC->MinOutputSize.cx
1403 && i_width != pVSCC->MaxOutputSize.cx)
1404 ||
1405 ( pVSCC->OutputGranularityX
1406 && ((i_width % pVSCC->OutputGranularityX)
1407 || pVSCC->MinOutputSize.cx > i_width
1408 || i_width > pVSCC->MaxOutputSize.cx )))
1409 {
1410 // required width not compatible, try next media type
1411 FreeMediaType( *p_mt );
1412 CoTaskMemFree( (PVOID)p_mt );
1413 continue;
1414 }
1415 pVih->bmiHeader.biWidth = i_width;
1416 }
1417
1418 if( i_height )
1419 {
1420 if(( !pVSCC->OutputGranularityY
1421 && i_height != pVSCC->MinOutputSize.cy
1422 && i_height != pVSCC->MaxOutputSize.cy)
1423 ||
1424 ( pVSCC->OutputGranularityY
1425 && ((i_height % pVSCC->OutputGranularityY)
1426 || pVSCC->MinOutputSize.cy > i_height
1427 || i_height > pVSCC->MaxOutputSize.cy )))
1428 {
1429 // required height not compatible, try next media type
1430 FreeMediaType( *p_mt );
1431 CoTaskMemFree( (PVOID)p_mt );
1432 continue;
1433 }
1434 pVih->bmiHeader.biHeight = i_height;
1435 }
1436
1437 // Set the sample size and image size.
1438 // (Round the image width up to a DWORD boundary.)
1439 p_mt->lSampleSize = pVih->bmiHeader.biSizeImage =
1440 ((pVih->bmiHeader.biWidth + 3) & ~3) *
1441 pVih->bmiHeader.biHeight * (pVih->bmiHeader.biBitCount>>3);
1442
1443 // no cropping, use full video input buffer
1444 memset(&(pVih->rcSource), 0, sizeof(RECT));
1445 memset(&(pVih->rcTarget), 0, sizeof(RECT));
1446
1447 // select this format as default
1448 if( SUCCEEDED( pSC->SetFormat(p_mt) ) )
1449 {
1450 i_priority = i_current_priority;
1451 if( i_fourcc )
1452 // no need to check any more media types
1453 i = piCount;
1454 }
1455 }
1456 else if( b_audio && p_mt->majortype == MEDIATYPE_Audio
1457 && p_mt->formattype == FORMAT_WaveFormatEx )
1458 {
1459 AUDIO_STREAM_CONFIG_CAPS *pASCC = reinterpret_cast<AUDIO_STREAM_CONFIG_CAPS*>(pSCC);
1460 WAVEFORMATEX *pWfx = reinterpret_cast<WAVEFORMATEX*>(p_mt->pbFormat);
1461
1462 if( i_current_fourcc && (WAVE_FORMAT_PCM == pWfx->wFormatTag) )
1463 {
1464 int val = i_channels;
1465 if( ! val )
1466 val = 2;
1467
1468 if( ( !pASCC->ChannelsGranularity
1469 && (unsigned int)val != pASCC->MinimumChannels
1470 && (unsigned int)val != pASCC->MaximumChannels)
1471 ||
1472 ( pASCC->ChannelsGranularity
1473 && ((val % pASCC->ChannelsGranularity)
1474 || (unsigned int)val < pASCC->MinimumChannels
1475 || (unsigned int)val > pASCC->MaximumChannels)))
1476 {
1477 // required number channels not available, try next media type
1478 FreeMediaType( *p_mt );
1479 CoTaskMemFree( (PVOID)p_mt );
1480 continue;
1481 }
1482 pWfx->nChannels = val;
1483
1484 val = i_samplespersec;
1485 if( ! val )
1486 val = 44100;
1487
1488 if( ( !pASCC->SampleFrequencyGranularity
1489 && (unsigned int)val != pASCC->MinimumSampleFrequency
1490 && (unsigned int)val != pASCC->MaximumSampleFrequency)
1491 ||
1492 ( pASCC->SampleFrequencyGranularity
1493 && ((val % pASCC->SampleFrequencyGranularity)
1494 || (unsigned int)val < pASCC->MinimumSampleFrequency
1495 || (unsigned int)val > pASCC->MaximumSampleFrequency )))
1496 {
1497 // required sampling rate not available, try next media type
1498 FreeMediaType( *p_mt );
1499 CoTaskMemFree( (PVOID)p_mt );
1500 continue;
1501 }
1502 pWfx->nSamplesPerSec = val;
1503
1504 val = i_bitspersample;
1505 if( ! val )
1506 {
1507 if( VLC_CODEC_FL32 == i_current_fourcc )
1508 val = 32;
1509 else
1510 val = 16;
1511 }
1512
1513 if( ( !pASCC->BitsPerSampleGranularity
1514 && (unsigned int)val != pASCC->MinimumBitsPerSample
1515 && (unsigned int)val != pASCC->MaximumBitsPerSample )
1516 ||
1517 ( pASCC->BitsPerSampleGranularity
1518 && ((val % pASCC->BitsPerSampleGranularity)
1519 || (unsigned int)val < pASCC->MinimumBitsPerSample
1520 || (unsigned int)val > pASCC->MaximumBitsPerSample )))
1521 {
1522 // required sample size not available, try next media type
1523 FreeMediaType( *p_mt );
1524 CoTaskMemFree( (PVOID)p_mt );
1525 continue;
1526 }
1527
1528 pWfx->wBitsPerSample = val;
1529 pWfx->nBlockAlign = (pWfx->wBitsPerSample * pWfx->nChannels)/8;
1530 pWfx->nAvgBytesPerSec = pWfx->nSamplesPerSec * pWfx->nBlockAlign;
1531
1532 // select this format as default
1533 if( SUCCEEDED( pSC->SetFormat(p_mt) ) )
1534 {
1535 i_priority = i_current_priority;
1536 }
1537 }
1538 }
1539 FreeMediaType( *p_mt );
1540 CoTaskMemFree( (PVOID)p_mt );
1541 }
1542 }
1543 CoTaskMemFree( (LPVOID)pSCC );
1544 if( i_priority >= ES_PRIORITY_SELECTABLE_MIN )
1545 msg_Dbg( p_this, "EnumDeviceCaps: input pin default format configured");
1546 }
1547 }
1548 }
1549
1550 /*
1551 ** Probe pin for available medias (may be a previously configured one)
1552 */
1553
1554 if( FAILED( p_output_pin->EnumMediaTypes( p_enummt.ReleaseAndGetAddressOf() ) ) )
1555 {
1556 continue;
1557 }
1558
1559 while( p_enummt->Next( 1, &p_mt, NULL ) == S_OK )
1560 {
1561 int i_current_fourcc = GetFourCCFromMediaType( *p_mt );
1562 if( !b_audio && i_current_fourcc && p_mt->majortype == MEDIATYPE_Video
1563 && p_mt->formattype == FORMAT_VideoInfo )
1564 {
1565 int i_current_width = ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biWidth;
1566 int i_current_height = ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biHeight;
1567 LONGLONG i_current_atpf = ((VIDEOINFOHEADER *)p_mt->pbFormat)->AvgTimePerFrame;
1568
1569 if( i_current_height < 0 )
1570 i_current_height = -i_current_height;
1571
1572 msg_Dbg( p_this, "EnumDeviceCaps: input pin "
1573 "accepts chroma: %4.4s, width:%i, height:%i, fps:%f",
1574 (char *)&i_current_fourcc, i_current_width,
1575 i_current_height, (10000000.0f/((float)i_current_atpf)) );
1576
1577 if( ( !i_fourcc || i_fourcc == i_current_fourcc ) &&
1578 ( !i_width || i_width == i_current_width ) &&
1579 ( !i_height || i_height == i_current_height ) &&
1580 ( !i_AvgTimePerFrame || i_AvgTimePerFrame == i_current_atpf ) &&
1581 mt_count < mt_max )
1582 {
1583 /* Pick match */
1584 mt[mt_count++] = *p_mt;
1585 }
1586 else FreeMediaType( *p_mt );
1587 }
1588 else if( b_audio && i_current_fourcc && p_mt->majortype == MEDIATYPE_Audio
1589 && p_mt->formattype == FORMAT_WaveFormatEx)
1590 {
1591 int i_current_channels =
1592 ((WAVEFORMATEX *)p_mt->pbFormat)->nChannels;
1593 int i_current_samplespersec =
1594 ((WAVEFORMATEX *)p_mt->pbFormat)->nSamplesPerSec;
1595 int i_current_bitspersample =
1596 ((WAVEFORMATEX *)p_mt->pbFormat)->wBitsPerSample;
1597
1598 msg_Dbg( p_this, "EnumDeviceCaps: input pin "
1599 "accepts format: %4.4s, channels:%i, "
1600 "samples/sec:%i bits/sample:%i",
1601 (char *)&i_current_fourcc, i_current_channels,
1602 i_current_samplespersec, i_current_bitspersample);
1603
1604 if( (!i_channels || i_channels == i_current_channels) &&
1605 (!i_samplespersec ||
1606 i_samplespersec == i_current_samplespersec) &&
1607 (!i_bitspersample ||
1608 i_bitspersample == i_current_bitspersample) &&
1609 mt_count < mt_max )
1610 {
1611 /* Pick match */
1612 mt[mt_count++] = *p_mt;
1613
1614 /* Setup a few properties like the audio latency */
1615 ComPtr<IAMBufferNegotiation> p_ambuf;
1616 if( SUCCEEDED( p_output_pin->QueryInterface(
1617 IID_IAMBufferNegotiation, (void **)p_ambuf.GetAddressOf() ) ) )
1618 {
1619 ALLOCATOR_PROPERTIES AllocProp;
1620 AllocProp.cbAlign = -1;
1621
1622 /* 100 ms of latency */
1623 AllocProp.cbBuffer = i_current_channels *
1624 i_current_samplespersec *
1625 i_current_bitspersample / 8 / 10;
1626
1627 AllocProp.cbPrefix = -1;
1628 AllocProp.cBuffers = -1;
1629 p_ambuf->SuggestAllocatorProperties( &AllocProp );
1630 }
1631 }
1632 else FreeMediaType( *p_mt );
1633 }
1634 else if( i_current_fourcc && p_mt->majortype == MEDIATYPE_Stream )
1635 {
1636 msg_Dbg( p_this, "EnumDeviceCaps: input pin "
1637 "accepts stream format: %4.4s",
1638 (char *)&i_current_fourcc );
1639
1640 if( ( !i_fourcc || i_fourcc == i_current_fourcc ) &&
1641 mt_count < mt_max )
1642 {
1643 /* Pick match */
1644 mt[mt_count++] = *p_mt;
1645 i_fourcc = i_current_fourcc;
1646 }
1647 else FreeMediaType( *p_mt );
1648 }
1649 else
1650 {
1651 const char * psz_type = "unknown";
1652 if( p_mt->majortype == MEDIATYPE_Video ) psz_type = "video";
1653 if( p_mt->majortype == MEDIATYPE_Audio ) psz_type = "audio";
1654 if( p_mt->majortype == MEDIATYPE_Stream ) psz_type = "stream";
1655 msg_Dbg( p_this, "EnumDeviceCaps: input pin media: unsupported format "
1656 "(%s %4.4s)", psz_type, (char *)&p_mt->subtype );
1657
1658 FreeMediaType( *p_mt );
1659 }
1660 CoTaskMemFree( (PVOID)p_mt );
1661 }
1662
1663 if( !mt_count && p_enummt->Reset() == S_OK )
1664 {
1665 // VLC did not find any supported MEDIATYPE for this output pin.
1666 // However the graph builder might insert converter filters in
1667 // the graph if we use a different codec in VLC filter input pin.
1668 // however, in order to avoid nasty surprises, make use of this
1669 // facility only for known unsupported codecs.
1670
1671 while( !mt_count && p_enummt->Next( 1, &p_mt, NULL ) == S_OK )
1672 {
1673 // the first four bytes of subtype GUID contains the codec FOURCC
1674 const char *pfcc = (char *)&p_mt->subtype;
1675 int i_current_fourcc = VLC_FOURCC(pfcc[0], pfcc[1], pfcc[2], pfcc[3]);
1676 if( VLC_FOURCC('H','C','W','2') == i_current_fourcc
1677 && p_mt->majortype == MEDIATYPE_Video && !b_audio )
1678 {
1679 // output format for 'Hauppauge WinTV PVR PCI II Capture'
1680 // try I420 as an input format
1681 i_current_fourcc = VLC_CODEC_I420;
1682 if( !i_fourcc || i_fourcc == i_current_fourcc )
1683 {
1684 // return alternative media type
1685 AM_MEDIA_TYPE mtr;
1686 VIDEOINFOHEADER vh;
1687
1688 mtr.majortype = MEDIATYPE_Video;
1689 mtr.subtype = MEDIASUBTYPE_I420;
1690 mtr.bFixedSizeSamples = TRUE;
1691 mtr.bTemporalCompression = FALSE;
1692 mtr.pUnk = NULL;
1693 mtr.formattype = FORMAT_VideoInfo;
1694 mtr.cbFormat = sizeof(vh);
1695 mtr.pbFormat = (BYTE *)&vh;
1696
1697 memset(&vh, 0, sizeof(vh));
1698
1699 vh.bmiHeader.biSize = sizeof(vh.bmiHeader);
1700 vh.bmiHeader.biWidth = i_width > 0 ? i_width :
1701 ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biWidth;
1702 vh.bmiHeader.biHeight = i_height > 0 ? i_height :
1703 ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biHeight;
1704 vh.bmiHeader.biPlanes = 3;
1705 vh.bmiHeader.biBitCount = 12;
1706 vh.bmiHeader.biCompression = VLC_CODEC_I420;
1707 vh.bmiHeader.biSizeImage = vh.bmiHeader.biWidth * 12 *
1708 vh.bmiHeader.biHeight / 8;
1709 mtr.lSampleSize = vh.bmiHeader.biSizeImage;
1710
1711 msg_Dbg( p_this, "EnumDeviceCaps: input pin media: using 'I420' in place of unsupported format 'HCW2'");
1712
1713 if( SUCCEEDED(CopyMediaType(mt+mt_count, &mtr)) )
1714 ++mt_count;
1715 }
1716 }
1717 FreeMediaType( *p_mt );
1718 }
1719 }
1720 }
1721
1722 return mt_count;
1723 }
1724
1725 /*****************************************************************************
1726 * ReadCompressed: reads compressed (MPEG/DV) data from the device.
1727 *****************************************************************************/
1728 static block_t *ReadCompressed( stream_t *p_access, bool *eof )
1729 {
1730 ComContext ctx( COINIT_MULTITHREADED );
1731
1732 access_sys_t *p_sys = (access_sys_t *)p_access->p_sys;
1733 /* There must be only 1 elementary stream to produce a valid stream
1734 * of MPEG or DV data */
1735 dshow_stream_t *p_stream = p_sys->pp_streams[0];
1736 VLCMediaSample sample;
1737
1738 /* Read 1 DV/MPEG frame (they contain the video and audio data) */
1739
1740 /* Get new sample/frame from the elementary stream (blocking). */
1741 vlc_mutex_lock( &p_sys->lock );
1742
1743 if( p_stream->p_capture_filter->CustomGetPin()->CustomGetSample(&sample) != S_OK )
1744 { /* No data available. Wait until some data has arrived */
1745 vlc_cond_wait( &p_sys->wait, &p_sys->lock );
1746 vlc_mutex_unlock( &p_sys->lock );
1747 return NULL;
1748 }
1749 vlc_mutex_unlock( &p_sys->lock );
1750
1751 /*
1752 * We got our sample
1753 */
1754 block_t *p_block = NULL;
1755 uint8_t *p_data;
1756 int i_data_size = sample.p_sample->GetActualDataLength();
1757 if( i_data_size == 0 )
1758 goto out;
1759
1760 p_block = block_Alloc( i_data_size );
1761 if( unlikely(p_block == NULL) )
1762 goto out;
1763
1764 sample.p_sample->GetPointer( &p_data );
1765 memcpy( p_block->p_buffer, p_data, i_data_size );
1766 /* The caller got what he wanted */
1767 out:
1768 (void) eof;
1769 return p_block;
1770 }
1771
1772 /****************************************************************************
1773 * Demux:
1774 ****************************************************************************/
1775 static int Demux( demux_t *p_demux )
1776 {
1777 ComContext ctx( COINIT_MULTITHREADED );
1778
1779 access_sys_t *p_sys = (access_sys_t *)p_demux->p_sys;
1780 int i_found_samples;
1781
1782 i_found_samples = 0;
1783 vlc_mutex_lock( &p_sys->lock );
1784
1785 while ( !i_found_samples )
1786 {
1787 /* Try to grab samples from all streams */
1788 std::vector<dshow_stream_t*>::iterator it = p_sys->pp_streams.begin();
1789 std::vector<dshow_stream_t*>::iterator end = p_sys->pp_streams.end();
1790 for ( ; it != end; ++it )
1791 {
1792 dshow_stream_t *p_stream = *it;
1793 if( p_stream->p_capture_filter &&
1794 p_stream->p_capture_filter->CustomGetPin()
1795 ->CustomGetSamples( p_stream->samples_queue ) == S_OK )
1796 {
1797 i_found_samples = 1;
1798 }
1799 }
1800
1801 if ( !i_found_samples)
1802 {
1803 /* Didn't find any audio nor video sample, just wait till the
1804 * dshow thread pushes some samples */
1805 vlc_cond_wait( &p_sys->wait, &p_sys->lock );
1806 /* Some DShow thread pushed data, or the OS broke the wait all
1807 * by itself. In all cases, it's *strongly* advised to test the
1808 * condition again, so let the loop do the test again */
1809 }
1810 }
1811
1812 vlc_mutex_unlock( &p_sys->lock );
1813
1814 std::vector<dshow_stream_t*>::iterator it = p_sys->pp_streams.begin();
1815 std::vector<dshow_stream_t*>::iterator end = p_sys->pp_streams.end();
1816 for ( ; it != end; ++it )
1817 {
1818 int i_samples;
1819 dshow_stream_t *p_stream = *it;
1820
1821 i_samples = p_stream->samples_queue.size();
1822 while ( i_samples > 0 )
1823 {
1824 int i_data_size;
1825 uint8_t *p_data;
1826 block_t *p_block;
1827 VLCMediaSample sample;
1828
1829 sample = p_stream->samples_queue.front();
1830 p_stream->samples_queue.pop_front();
1831
1832 i_data_size = sample.p_sample->GetActualDataLength();
1833 sample.p_sample->GetPointer( &p_data );
1834
1835 REFERENCE_TIME i_pts, i_end_date;
1836 HRESULT hr = sample.p_sample->GetTime( &i_pts, &i_end_date );
1837 if( hr != S_OK && hr != VFW_S_NO_STOP_TIME )
1838 {
1839 if( p_stream->mt.majortype == MEDIATYPE_Video || !p_stream->b_pts )
1840 {
1841 /* Use our data timestamp */
1842 i_pts = MSFTIME_FROM_VLC_TICK(sample.i_timestamp);
1843 p_stream->b_pts = true;
1844 }
1845 else
1846 i_pts = LONG_MIN;
1847 }
1848
1849 if( i_pts != LONG_MIN ) {
1850 i_pts += 5;
1851 }
1852 #if 0
1853 msg_Dbg( p_demux, "Read() stream: %i, size: %i, PTS: %" PRId64,
1854 i_stream, i_data_size, VLC_TICK_FROM_MSFTIME(i_pts) );
1855 #endif
1856
1857 p_block = block_Alloc( i_data_size );
1858 memcpy( p_block->p_buffer, p_data, i_data_size );
1859 p_block->i_pts = p_block->i_dts = i_pts == LONG_MIN ?
1860 VLC_TICK_INVALID : (VLC_TICK_FROM_MSFTIME(i_pts) + VLC_TICK_0);
1861
1862 if( p_block->i_pts != VLC_TICK_INVALID )
1863 es_out_SetPCR( p_demux->out, p_block->i_pts );
1864 es_out_Send( p_demux->out, p_stream->p_es, p_block );
1865
1866 i_samples--;
1867 }
1868 }
1869
1870 return 1;
1871 }
1872
1873 /*****************************************************************************
1874 * AccessControl:
1875 *****************************************************************************/
1876 static int AccessControl( stream_t *p_access, int i_query, va_list args )
1877 {
1878 access_sys_t *sys = (access_sys_t *)p_access->p_sys;
1879 bool *pb_bool;
1880
1881 switch( i_query )
1882 {
1883 case STREAM_CAN_SEEK:
1884 case STREAM_CAN_FASTSEEK:
1885 case STREAM_CAN_PAUSE:
1886 case STREAM_CAN_CONTROL_PACE:
1887 pb_bool = va_arg( args, bool * );
1888 *pb_bool = false;
1889 break;
1890
1891 case STREAM_GET_PTS_DELAY:
1892 *va_arg( args, vlc_tick_t * ) =
1893 VLC_TICK_FROM_MS( var_InheritInteger( p_access, "live-caching" ) );
1894 break;
1895
1896 case STREAM_GET_CONTENT_TYPE:
1897 {
1898 dshow_stream_t *p_stream = sys->pp_streams[0];
1899 char **type = va_arg( args, char ** );
1900
1901 /* Check if we need to force demuxers */
1902 if( p_stream->i_fourcc == VLC_CODEC_DV )
1903 *type = strdup( "video/dv" );
1904 else if( p_stream->i_fourcc == VLC_CODEC_MPGV )
1905 *type = strdup( "video/MP2P" );
1906 else
1907 return VLC_EGENERIC;
1908 break;
1909 }
1910
1911 default:
1912 return VLC_EGENERIC;
1913 }
1914
1915 return VLC_SUCCESS;
1916 }
1917
1918 /****************************************************************************
1919 * DemuxControl:
1920 ****************************************************************************/
1921 static int DemuxControl( demux_t *p_demux, int i_query, va_list args )
1922 {
1923 bool *pb;
1924 int64_t *pi64;
1925
1926 access_sys_t *p_sys = ( access_sys_t * ) p_demux->p_sys;
1927
1928 switch( i_query )
1929 {
1930 /* Special for access_demux */
1931 case DEMUX_CAN_PAUSE:
1932 case DEMUX_CAN_SEEK:
1933 case DEMUX_SET_PAUSE_STATE:
1934 case DEMUX_CAN_CONTROL_PACE:
1935 pb = va_arg( args, bool * );
1936 *pb = false;
1937 return VLC_SUCCESS;
1938
1939 case DEMUX_GET_PTS_DELAY:
1940 *va_arg( args, vlc_tick_t * ) =
1941 VLC_TICK_FROM_MS( var_InheritInteger( p_demux, "live-caching" ) );
1942 return VLC_SUCCESS;
1943
1944 case DEMUX_GET_TIME:
1945 pi64 = va_arg( args, int64_t * );
1946 *pi64 = vlc_tick_now() - p_sys->i_start;
1947 return VLC_SUCCESS;
1948
1949 /* TODO implement others */
1950 default:
1951 return VLC_EGENERIC;
1952 }
1953
1954 return VLC_EGENERIC;
1955 }
1956
1957 static int AppendAudioEnabledVDevs( vlc_object_t *p_this, std::list<std::string> &audio_list,
1958 std::list<std::string> &video_list )
1959 {
1960 ComPtr<IFilterGraph> p_graph;
1961 ComPtr<IGraphBuilder> p_gbuilder;
1962 ComPtr<ICaptureGraphBuilder2> p_cgbuilder;
1963
1964 if( FAILED( CoCreateInstance( CLSID_FilterGraph, NULL, CLSCTX_INPROC_SERVER, IID_IFilterGraph,
1965 ( void ** ) p_graph.GetAddressOf() ) ) )
1966 return VLC_EGENERIC;
1967
1968 if( FAILED( p_graph->QueryInterface( IID_IGraphBuilder, ( void ** ) p_gbuilder.GetAddressOf() ) ) )
1969 return VLC_EGENERIC;
1970
1971 if( FAILED( CoCreateInstance( CLSID_CaptureGraphBuilder2, NULL, CLSCTX_INPROC_SERVER,
1972 IID_ICaptureGraphBuilder2, ( void ** ) p_cgbuilder.GetAddressOf() ) ) )
1973 return VLC_EGENERIC;
1974
1975 if( FAILED( p_cgbuilder->SetFiltergraph( p_gbuilder.Get() ) ) )
1976 return VLC_EGENERIC;
1977
1978 for( std::list<std::string>::iterator iter = video_list.begin();
1979 iter != video_list.end();
1980 ++iter )
1981 {
1982 ComPtr<IBaseFilter> p_device;
1983 ComPtr<IPin> p_pin;
1984
1985 p_device = FindCaptureDevice( p_this, &( *iter ), NULL, false );
1986 if( !p_device ) continue;
1987
1988 if( FAILED( p_gbuilder->AddFilter( p_device.Get(), NULL ) ) )
1989 continue;
1990
1991 if( SUCCEEDED( p_cgbuilder->FindPin( p_device.Get(), PINDIR_OUTPUT, NULL, &MEDIATYPE_Audio,
1992 true, 0, p_pin.GetAddressOf() ) ) )
1993 audio_list.push_back( *iter );
1994
1995 p_gbuilder->RemoveFilter( p_device.Get() );
1996 }
1997
1998 return VLC_SUCCESS;
1999 }
2000
2001 /*****************************************************************************
2002 * config variable callback
2003 *****************************************************************************/
2004 static int FindDevices( const char *psz_name, char ***vp, char ***tp )
2005 {
2006 /* Find list of devices */
2007 std::list<std::string> list_devices;
2008 try
2009 {
2010 bool b_audio = !strcmp( psz_name, "dshow-adev" );
2011
2012 // Use STA as this most likely comes from a Qt thread, which is
2013 // initialized as STA.
2014 ComContext ctx( COINIT_APARTMENTTHREADED );
2015
2016 FindCaptureDevice( NULL, NULL, &list_devices, b_audio );
2017
2018 if( b_audio )
2019 {
2020 std::list<std::string> list_vdevs;
2021 FindCaptureDevice( NULL, NULL, &list_vdevs, false );
2022 if( !list_vdevs.empty() )
2023 AppendAudioEnabledVDevs( NULL, list_devices, list_vdevs );
2024 }
2025 }
2026 catch (const std::runtime_error& ex)
2027 {
2028 msg_Err( (vlc_object_t *)NULL, "Failed fetch devices: %s", ex.what() );
2029 }
2030
2031 unsigned count = 2 + list_devices.size(), i = 2;
2032 char **values = (char **)xmalloc( count * sizeof(*values) );
2033 char **texts = (char **)xmalloc( count * sizeof(*texts) );
2034
2035 values[0] = strdup( "" );
2036 texts[0] = strdup( _("Default") );
2037 values[1] = strdup( "none" );
2038 texts[1] = strdup( _("None") );
2039
2040 for( std::list<std::string>::iterator iter = list_devices.begin();
2041 iter != list_devices.end();
2042 ++iter )
2043 {
2044 assert( i < count );
2045 values[i] = strdup( iter->c_str() );
2046 texts[i] = strdup( iter->c_str() );
2047 i++;
2048 }
2049
2050 *vp = values;
2051 *tp = texts;
2052 return count;
2053 }
2054
2055 /*****************************************************************************
2056 * Properties
2057 *****************************************************************************/
2058
2059 static void ShowPropertyPage( IUnknown *obj )
2060 {
2061 ComPtr<ISpecifyPropertyPages> p_spec;
2062 CAUUID cauuid;
2063
2064 HRESULT hr = obj->QueryInterface( IID_ISpecifyPropertyPages,
2065 (void **)p_spec.GetAddressOf() );
2066 if( FAILED(hr) ) return;
2067
2068 if( SUCCEEDED(p_spec->GetPages( &cauuid )) )
2069 {
2070 if( cauuid.cElems > 0 )
2071 {
2072 HWND hwnd_desktop = ::GetDesktopWindow();
2073
2074 OleCreatePropertyFrame( hwnd_desktop, 30, 30, NULL, 1, &obj,
2075 cauuid.cElems, cauuid.pElems, 0, 0, NULL );
2076
2077 CoTaskMemFree( cauuid.pElems );
2078 }
2079 }
2080 }
2081
2082 static void ShowDeviceProperties( vlc_object_t *p_this,
2083 ICaptureGraphBuilder2 *p_graph,
2084 IBaseFilter *p_device_filter,
2085 bool b_audio )
2086 {
2087 HRESULT hr;
2088 msg_Dbg( p_this, "configuring Device Properties" );
2089
2090 /*
2091 * Video or audio capture filter page
2092 */
2093 ShowPropertyPage( p_device_filter );
2094
2095 /*
2096 * Audio capture pin
2097 */
2098 if( p_graph && b_audio )
2099 {
2100 ComPtr<IAMStreamConfig> p_SC;
2101
2102 msg_Dbg( p_this, "showing WDM Audio Configuration Pages" );
2103
2104 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2105 &MEDIATYPE_Audio, p_device_filter,
2106 IID_IAMStreamConfig, (void **)p_SC.GetAddressOf() );
2107 if( SUCCEEDED(hr) )
2108 {
2109 ShowPropertyPage(p_SC.Get());
2110 }
2111
2112 /*
2113 * TV Audio filter
2114 */
2115 ComPtr<IAMTVAudio> p_TVA;
2116 HRESULT hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2117 &MEDIATYPE_Audio, p_device_filter,
2118 IID_IAMTVAudio, (void **)p_TVA.GetAddressOf() );
2119 if( SUCCEEDED(hr) )
2120 {
2121 ShowPropertyPage(p_TVA.Get());
2122 }
2123 }
2124
2125 /*
2126 * Video capture pin
2127 */
2128 if( p_graph && !b_audio )
2129 {
2130 ComPtr<IAMStreamConfig> p_SC;
2131
2132 msg_Dbg( p_this, "showing WDM Video Configuration Pages" );
2133
2134 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2135 &MEDIATYPE_Interleaved, p_device_filter,
2136 IID_IAMStreamConfig, (void **)p_SC.GetAddressOf() );
2137 if( FAILED(hr) )
2138 {
2139 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2140 &MEDIATYPE_Video, p_device_filter,
2141 IID_IAMStreamConfig, (void **)p_SC.GetAddressOf() );
2142 }
2143
2144 if( FAILED(hr) )
2145 {
2146 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2147 &MEDIATYPE_Stream, p_device_filter,
2148 IID_IAMStreamConfig, (void **)p_SC.GetAddressOf() );
2149 }
2150
2151 if( SUCCEEDED(hr) )
2152 {
2153 ShowPropertyPage(p_SC.Get());
2154 }
2155 }
2156 }
2157
2158 static void ShowTunerProperties( vlc_object_t *p_this,
2159 ICaptureGraphBuilder2 *p_graph,
2160 IBaseFilter *p_device_filter,
2161 bool b_audio )
2162 {
2163 HRESULT hr;
2164 msg_Dbg( p_this, "configuring Tuner Properties" );
2165
2166 if( !p_graph || b_audio ) return;
2167
2168 ComPtr<IAMTVTuner> p_TV;
2169 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2170 &MEDIATYPE_Interleaved, p_device_filter,
2171 IID_IAMTVTuner, (void **)p_TV.GetAddressOf() );
2172 if( FAILED(hr) )
2173 {
2174 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2175 &MEDIATYPE_Video, p_device_filter,
2176 IID_IAMTVTuner, (void **)p_TV.GetAddressOf() );
2177 }
2178
2179 if( FAILED(hr) )
2180 {
2181 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE,
2182 &MEDIATYPE_Stream, p_device_filter,
2183 IID_IAMTVTuner, (void **)p_TV.GetAddressOf() );
2184 }
2185
2186 if( SUCCEEDED(hr) )
2187 {
2188 ShowPropertyPage(p_TV.Get());
2189 }
2190 }
2191
2192 static void ConfigTuner( vlc_object_t *p_this, ICaptureGraphBuilder2 *p_graph,
2193 IBaseFilter *p_device_filter )
2194 {
2195 int i_channel, i_country, i_input, i_amtuner_mode;
2196 long l_modes = 0;
2197 unsigned i_frequency;
2198 ComPtr<IAMTVTuner> p_TV;
2199 HRESULT hr;
2200
2201 if( !p_graph ) return;
2202
2203 i_channel = var_GetInteger( p_this, "dshow-tuner-channel" );
2204 i_country = var_GetInteger( p_this, "dshow-tuner-country" );
2205 i_input = var_GetInteger( p_this, "dshow-tuner-input" );
2206 i_amtuner_mode = var_GetInteger( p_this, "dshow-amtuner-mode" );
2207
2208 if( !i_channel && !i_country && !i_input ) return; /* Nothing to do */
2209
2210 msg_Dbg( p_this, "tuner config: channel %i, country %i, input type %i",
2211 i_channel, i_country, i_input );
2212
2213 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE, &MEDIATYPE_Interleaved,
2214 p_device_filter, IID_IAMTVTuner,
2215 (void **)p_TV.GetAddressOf() );
2216 if( FAILED(hr) )
2217 {
2218 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video,
2219 p_device_filter, IID_IAMTVTuner,
2220 (void **)p_TV.GetAddressOf() );
2221 }
2222
2223 if( FAILED(hr) )
2224 {
2225 hr = p_graph->FindInterface( &PIN_CATEGORY_CAPTURE, &MEDIATYPE_Stream,
2226 p_device_filter, IID_IAMTVTuner,
2227 (void **)p_TV.GetAddressOf() );
2228 }
2229
2230 if( FAILED(hr) )
2231 {
2232 msg_Dbg( p_this, "couldn't find tuner interface" );
2233 return;
2234 }
2235
2236 hr = p_TV->GetAvailableModes( &l_modes );
2237 if( SUCCEEDED(hr) && (l_modes & i_amtuner_mode) )
2238 {
2239 p_TV->put_Mode( (AMTunerModeType)i_amtuner_mode );
2240 }
2241
2242 if( i_input == 1 ) p_TV->put_InputType( 0, TunerInputCable );
2243 else if( i_input == 2 ) p_TV->put_InputType( 0, TunerInputAntenna );
2244
2245 p_TV->put_CountryCode( i_country );
2246 p_TV->put_Channel( i_channel, AMTUNER_SUBCHAN_NO_TUNE,
2247 AMTUNER_SUBCHAN_NO_TUNE );
2248 p_TV->Release();
2249 }
2250
2251 } // namespace
VLC Development tree mirror