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