git.sesse.net Git - vlc/blob - modules/access/dshow/dshow.cpp

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