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