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