dshow.cpp 80.8 KB
Newer Older
gbazin's avatar
 
gbazin committed
1
/*****************************************************************************
2
 * dshow.cpp : DirectShow access module for vlc
gbazin's avatar
 
gbazin committed
3
 *****************************************************************************
4
 * Copyright (C) 2002, 2003 the VideoLAN team
5
 * $Id$
gbazin's avatar
 
gbazin committed
6
 *
7
 * Author: Gildas Bazin <gbazin@videolan.org>
8
 *         Damien Fouilleul <damienf@videolan.org>
gbazin's avatar
 
gbazin committed
9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
dionoea's avatar
dionoea committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
gbazin's avatar
 
gbazin committed
23 24 25 26 27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
damienf's avatar
damienf committed
28

29 30 31 32
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

33 34 35
#define __STDC_FORMAT_MACROS 1
#include <inttypes.h>

36
#include <vlc_common.h>
37
#include <vlc_plugin.h>
zorglub's avatar
zorglub committed
38 39
#include <vlc_input.h>
#include <vlc_access.h>
40
#include <vlc_demux.h>
zorglub's avatar
zorglub committed
41
#include <vlc_vout.h>
zorglub's avatar
Round 2  
zorglub committed
42
#include <vlc_interface.h>
gbazin's avatar
 
gbazin committed
43

gbazin's avatar
gbazin committed
44
#include "common.h"
gbazin's avatar
 
gbazin committed
45 46
#include "filter.h"

gbazin's avatar
 
gbazin committed
47 48 49
/*****************************************************************************
 * Access: local prototypes
 *****************************************************************************/
50
static block_t *ReadCompressed( access_t * );
51
static int AccessControl ( access_t *, int, va_list );
gbazin's avatar
 
gbazin committed
52

gbazin's avatar
gbazin committed
53 54 55
static int Demux       ( demux_t * );
static int DemuxControl( demux_t *, int, va_list );

56
static int OpenDevice( vlc_object_t *, access_sys_t *, string, bool );
gbazin's avatar
 
gbazin committed
57
static IBaseFilter *FindCaptureDevice( vlc_object_t *, string *,
58
                                       list<string> *, bool );
59
static size_t EnumDeviceCaps( vlc_object_t *, IBaseFilter *,
60 61
                              int, int, int, int, int, int,
                              AM_MEDIA_TYPE *mt, size_t );
gbazin's avatar
gbazin committed
62 63
static bool ConnectFilters( vlc_object_t *, access_sys_t *,
                            IBaseFilter *, CaptureFilter * );
gbazin's avatar
 
gbazin committed
64 65
static int FindDevicesCallback( vlc_object_t *, char const *,
                                vlc_value_t, vlc_value_t, void * );
66 67
static int ConfigDevicesCallback( vlc_object_t *, char const *,
                                  vlc_value_t, vlc_value_t, void * );
gbazin's avatar
 
gbazin committed
68

69
static void ShowPropertyPage( IUnknown * );
70
static void ShowDeviceProperties( vlc_object_t *, ICaptureGraphBuilder2 *,
71
                                  IBaseFilter *, bool );
72
static void ShowTunerProperties( vlc_object_t *, ICaptureGraphBuilder2 *,
73
                                 IBaseFilter *, bool );
74 75
static void ConfigTuner( vlc_object_t *, ICaptureGraphBuilder2 *,
                         IBaseFilter * );
gbazin's avatar
 
gbazin committed
76

gbazin's avatar
 
gbazin committed
77
/*****************************************************************************
78
 * Module descriptor
gbazin's avatar
 
gbazin committed
79
 *****************************************************************************/
80 81 82 83 84 85
static const char *const ppsz_vdev[] = { "", "none" };
static const char *const ppsz_vdev_text[] = { N_("Default"), N_("None") };
static const char *const ppsz_adev[] = { "", "none" };
static const char *const ppsz_adev_text[] = { N_("Default"), N_("None") };
static const int pi_tuner_input[] = { 0, 1, 2 };
static const char *const ppsz_tuner_input_text[] =
86
    {N_("Default"), N_("Cable"), N_("Antenna")};
87
static const int pi_amtuner_mode[]  = { AMTUNER_MODE_DEFAULT,
88 89 90 91
                                 AMTUNER_MODE_TV,
                                 AMTUNER_MODE_FM_RADIO,
                                 AMTUNER_MODE_AM_RADIO,
                                 AMTUNER_MODE_DSS };
92
static const char *const ppsz_amtuner_mode_text[] = { N_("Default"),
93 94 95 96
                                          N_("TV"),
                                          N_("FM radio"),
                                          N_("AM radio"),
                                          N_("DSS") };
97

gbazin's avatar
 
gbazin committed
98 99
#define CACHING_TEXT N_("Caching value in ms")
#define CACHING_LONGTEXT N_( \
100
    "Caching value for DirectShow streams. " \
101
    "This value should be set in milliseconds." )
102 103
#define VDEV_TEXT N_("Video device name")
#define VDEV_LONGTEXT N_( \
zorglub's avatar
zorglub committed
104
    "Name of the video device that will be used by the " \
105 106 107 108
    "DirectShow plugin. If you don't specify anything, the default device " \
    "will be used.")
#define ADEV_TEXT N_("Audio device name")
#define ADEV_LONGTEXT N_( \
zorglub's avatar
zorglub committed
109
    "Name of the audio device that will be used by the " \
110
    "DirectShow plugin. If you don't specify anything, the default device " \
Christophe Mutricy's avatar
Christophe Mutricy committed
111
    "will be used. ")
112 113
#define SIZE_TEXT N_("Video size")
#define SIZE_LONGTEXT N_( \
zorglub's avatar
zorglub committed
114
    "Size of the video that will be displayed by the " \
115
    "DirectShow plugin. If you don't specify anything the default size for " \
Christophe Mutricy's avatar
Christophe Mutricy committed
116
    "your device will be used. You can specify a standard size (cif, d1, ...) or <width>x<height>.")
117 118 119
#define CHROMA_TEXT N_("Video input chroma format")
#define CHROMA_LONGTEXT N_( \
    "Force the DirectShow video input to use a specific chroma format " \
120
    "(eg. I420 (default), RV24, etc.)")
121 122 123
#define FPS_TEXT N_("Video input frame rate")
#define FPS_LONGTEXT N_( \
    "Force the DirectShow video input to use a specific frame rate" \
Christophe Mutricy's avatar
Christophe Mutricy committed
124
    "(eg. 0 means default, 25, 29.97, 50, 59.94, etc.)")
125
#define CONFIG_TEXT N_("Device properties")
gbazin's avatar
 
gbazin committed
126
#define CONFIG_LONGTEXT N_( \
127 128
    "Show the properties dialog of the selected device before starting the " \
    "stream.")
129 130 131
#define TUNER_TEXT N_("Tuner properties")
#define TUNER_LONGTEXT N_( \
    "Show the tuner properties [channel selection] page." )
132 133
#define CHANNEL_TEXT N_("Tuner TV Channel")
#define CHANNEL_LONGTEXT N_( \
zorglub's avatar
zorglub committed
134
    "Set the TV channel the tuner will set to " \
135 136 137
    "(0 means default)." )
#define COUNTRY_TEXT N_("Tuner country code")
#define COUNTRY_LONGTEXT N_( \
zorglub's avatar
zorglub committed
138
    "Set the tuner country code that establishes the current " \
139 140 141
    "channel-to-frequency mapping (0 means default)." )
#define TUNER_INPUT_TEXT N_("Tuner input type")
#define TUNER_INPUT_LONGTEXT N_( \
zorglub's avatar
zorglub committed
142
    "Select the tuner input type (Cable/Antenna)." )
143 144
#define VIDEO_IN_TEXT N_("Video input pin")
#define VIDEO_IN_LONGTEXT N_( \
zorglub's avatar
zorglub committed
145
  "Select the video input source, such as composite, s-video, " \
146
  "or tuner. Since these settings are hardware-specific, you should find good " \
zorglub's avatar
zorglub committed
147 148
  "settings in the \"Device config\" area, and use those numbers here. -1 " \
  "means that settings will not be changed.")
149 150
#define AUDIO_IN_TEXT N_("Audio input pin")
#define AUDIO_IN_LONGTEXT N_( \
zorglub's avatar
zorglub committed
151
  "Select the audio input source. See the \"video input\" option." )
152 153
#define VIDEO_OUT_TEXT N_("Video output pin")
#define VIDEO_OUT_LONGTEXT N_( \
zorglub's avatar
zorglub committed
154
  "Select the video output type. See the \"video input\" option." )
155 156
#define AUDIO_OUT_TEXT N_("Audio output pin")
#define AUDIO_OUT_LONGTEXT N_( \
zorglub's avatar
zorglub committed
157
  "Select the audio output type. See the \"video input\" option." )
gbazin's avatar
 
gbazin committed
158

159
#define AMTUNER_MODE_TEXT N_("AM Tuner mode")
dionoea's avatar
dionoea committed
160
#define AMTUNER_MODE_LONGTEXT N_( \
161 162
    "AM Tuner mode. Can be one of Default (0), TV (1)," \
     "AM Radio (2), FM Radio (3) or DSS (4).")
163

164 165
#define AUDIO_CHANNELS_TEXT N_("Number of audio channels")
#define AUDIO_CHANNELS_LONGTEXT N_( \
Felix Paul Kühne's avatar
Felix Paul Kühne committed
166
    "Select audio input format with the given number of audio channels (if non 0)" )
167 168 169 170 171 172 173 174 175

#define AUDIO_SAMPLERATE_TEXT N_("Audio sample rate")
#define AUDIO_SAMPLERATE_LONGTEXT N_( \
    "Select audio input format with the given sample rate (if non 0)" )

#define AUDIO_BITSPERSAMPLE_TEXT N_("Audio bits per sample")
#define AUDIO_BITSPERSAMPLE_LONGTEXT N_( \
    "Select audio input format with the given bits/sample (if non 0)" )

176
static int  CommonOpen ( vlc_object_t *, access_sys_t *, bool );
gbazin's avatar
gbazin committed
177 178
static void CommonClose( vlc_object_t *, access_sys_t * );

179 180 181 182 183 184
static int  AccessOpen ( vlc_object_t * );
static void AccessClose( vlc_object_t * );

static int  DemuxOpen  ( vlc_object_t * );
static void DemuxClose ( vlc_object_t * );

185 186 187 188 189
vlc_module_begin ()
    set_shortname( N_("DirectShow") )
    set_description( N_("DirectShow input") )
    set_category( CAT_INPUT )
    set_subcategory( SUBCAT_INPUT_ACCESS )
190
    add_integer( "dshow-caching", (mtime_t)(0.2*CLOCK_FREQ) / 1000, NULL,
191
                 CACHING_TEXT, CACHING_LONGTEXT, true );
192

193
    add_string( "dshow-vdev", NULL, NULL, VDEV_TEXT, VDEV_LONGTEXT, false)
194
        change_string_list( ppsz_vdev, ppsz_vdev_text, FindDevicesCallback );
195 196
        change_action_add( FindDevicesCallback, N_("Refresh list") )
        change_action_add( ConfigDevicesCallback, N_("Configure") )
197

198
    add_string( "dshow-adev", NULL, NULL, ADEV_TEXT, ADEV_LONGTEXT, false)
199
        change_string_list( ppsz_adev, ppsz_adev_text, FindDevicesCallback );
200 201
        change_action_add( FindDevicesCallback, N_("Refresh list") )
        change_action_add( ConfigDevicesCallback, N_("Configure") )
202

203
    add_string( "dshow-size", NULL, NULL, SIZE_TEXT, SIZE_LONGTEXT, false)
204

205
    add_string( "dshow-chroma", NULL, NULL, CHROMA_TEXT, CHROMA_LONGTEXT,
206
                true );
gbazin's avatar
 
gbazin committed
207

208
    add_float( "dshow-fps", 0.0f, NULL, FPS_TEXT, FPS_LONGTEXT,
209
                true );
210

211 212
    add_bool( "dshow-config", false, NULL, CONFIG_TEXT, CONFIG_LONGTEXT,
              true );
gbazin's avatar
 
gbazin committed
213

214 215
    add_bool( "dshow-tuner", false, NULL, TUNER_TEXT, TUNER_LONGTEXT,
              true );
216

217
    add_integer( "dshow-tuner-channel", 0, NULL, CHANNEL_TEXT,
218
                 CHANNEL_LONGTEXT, true );
219 220

    add_integer( "dshow-tuner-country", 0, NULL, COUNTRY_TEXT,
221
                 COUNTRY_LONGTEXT, true );
222 223

    add_integer( "dshow-tuner-input", 0, NULL, TUNER_INPUT_TEXT,
224
                 TUNER_INPUT_LONGTEXT, true );
225
        change_integer_list( pi_tuner_input, ppsz_tuner_input_text, NULL );
226

227
    add_integer( "dshow-video-input",  -1, NULL, VIDEO_IN_TEXT,
228
                 VIDEO_IN_LONGTEXT, true );
229 230

    add_integer( "dshow-audio-input",  -1, NULL, AUDIO_IN_TEXT,
231
                 AUDIO_IN_LONGTEXT, true );
232 233

    add_integer( "dshow-video-output", -1, NULL, VIDEO_OUT_TEXT,
234
                 VIDEO_OUT_LONGTEXT, true );
235 236

    add_integer( "dshow-audio-output", -1, NULL, AUDIO_OUT_TEXT,
237
                 AUDIO_OUT_LONGTEXT, true );
238

dionoea's avatar
dionoea committed
239
    add_integer( "dshow-amtuner-mode", AMTUNER_MODE_TV, NULL,
240
                AMTUNER_MODE_TEXT, AMTUNER_MODE_LONGTEXT, false);
241
        change_integer_list( pi_amtuner_mode, ppsz_amtuner_mode_text, NULL );
242

243 244 245 246
    add_integer( "dshow-audio-channels", 0, NULL, AUDIO_CHANNELS_TEXT,
                 AUDIO_CHANNELS_LONGTEXT, true );
    add_integer( "dshow-audio-samplerate", 0, NULL, AUDIO_SAMPLERATE_TEXT,
                 AUDIO_SAMPLERATE_LONGTEXT, true );
247 248
    add_integer( "dshow-audio-bitspersample", 0, NULL, AUDIO_BITSPERSAMPLE_TEXT,
                 AUDIO_BITSPERSAMPLE_LONGTEXT, true );
249

250 251 252
    add_shortcut( "dshow" )
    set_capability( "access_demux", 0 )
    set_callbacks( DemuxOpen, DemuxClose )
gbazin's avatar
 
gbazin committed
253

254 255 256 257
    add_submodule ()
    set_description( N_("DirectShow input") )
    set_capability( "access", 0 )
    set_callbacks( AccessOpen, AccessClose )
gbazin's avatar
 
gbazin committed
258

259
vlc_module_end ()
gbazin's avatar
 
gbazin committed
260

gbazin's avatar
gbazin committed
261
/*****************************************************************************
gbazin's avatar
 
gbazin committed
262
 * DirectShow elementary stream descriptor
gbazin's avatar
gbazin committed
263
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
264 265 266 267 268 269 270 271 272 273 274 275 276 277
typedef struct dshow_stream_t
{
    string          devicename;
    IBaseFilter     *p_device_filter;
    CaptureFilter   *p_capture_filter;
    AM_MEDIA_TYPE   mt;

    union
    {
      VIDEOINFOHEADER video;
      WAVEFORMATEX    audio;

    } header;

gbazin's avatar
gbazin committed
278 279 280
    int             i_fourcc;
    es_out_id_t     *p_es;

281
    bool      b_pts;
282 283

    deque<VLCMediaSample> samples_queue;
gbazin's avatar
 
gbazin committed
284 285
} dshow_stream_t;

gbazin's avatar
gbazin committed
286
/*****************************************************************************
287
 * DirectShow utility functions
gbazin's avatar
gbazin committed
288
 *****************************************************************************/
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342
static void CreateDirectShowGraph( access_sys_t *p_sys )
{
    p_sys->i_crossbar_route_depth = 0;

    /* Create directshow filter graph */
    if( SUCCEEDED( CoCreateInstance( CLSID_FilterGraph, 0, CLSCTX_INPROC,
                       (REFIID)IID_IFilterGraph, (void **)&p_sys->p_graph) ) )
    {
        /* Create directshow capture graph builder if available */
        if( SUCCEEDED( CoCreateInstance( CLSID_CaptureGraphBuilder2, 0,
                         CLSCTX_INPROC, (REFIID)IID_ICaptureGraphBuilder2,
                         (void **)&p_sys->p_capture_graph_builder2 ) ) )
        {
            p_sys->p_capture_graph_builder2->
                SetFiltergraph((IGraphBuilder *)p_sys->p_graph);
        }

        p_sys->p_graph->QueryInterface( IID_IMediaControl,
                                        (void **)&p_sys->p_control );
    }
}

static void DeleteDirectShowGraph( access_sys_t *p_sys )
{
    DeleteCrossbarRoutes( p_sys );

    /* Remove filters from graph */
    for( int i = 0; i < p_sys->i_streams; i++ )
    {
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_capture_filter );
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_device_filter );
        p_sys->pp_streams[i]->p_capture_filter->Release();
        p_sys->pp_streams[i]->p_device_filter->Release();
    }

    /* Release directshow objects */
    if( p_sys->p_control )
    {
        p_sys->p_control->Release();
        p_sys->p_control = NULL;
    }
    if( p_sys->p_capture_graph_builder2 )
    {
        p_sys->p_capture_graph_builder2->Release();
        p_sys->p_capture_graph_builder2 = NULL;
    }

    if( p_sys->p_graph )
    {
        p_sys->p_graph->Release();
        p_sys->p_graph = NULL;
    }
}

gbazin's avatar
 
gbazin committed
343
/*****************************************************************************
gbazin's avatar
gbazin committed
344
 * CommonOpen: open direct show device
gbazin's avatar
 
gbazin committed
345
 *****************************************************************************/
gbazin's avatar
gbazin committed
346
static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
347
                       bool b_access_demux )
gbazin's avatar
 
gbazin committed
348
{
349
    vlc_value_t  val;
gbazin's avatar
gbazin committed
350
    int i;
gbazin's avatar
 
gbazin committed
351

352
    /* Get/parse options and open device(s) */
gbazin's avatar
 
gbazin committed
353
    string vdevname, adevname;
354
    int i_width = 0, i_height = 0, i_chroma = 0;
355
    bool b_audio = true;
gbazin's avatar
 
gbazin committed
356

357 358
    var_Create( p_this, "dshow-config", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-tuner", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
gbazin's avatar
gbazin committed
359 360
    var_Create( p_this, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-vdev", &val );
361
    if( val.psz_string ) vdevname = string( val.psz_string );
362
    free( val.psz_string );
gbazin's avatar
 
gbazin committed
363

gbazin's avatar
gbazin committed
364 365
    var_Create( p_this, "dshow-adev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-adev", &val );
366
    if( val.psz_string ) adevname = string( val.psz_string );
367
    free( val.psz_string );
368

gbazin's avatar
gbazin committed
369 370
    static struct {char *psz_size; int  i_width; int  i_height;} size_table[] =
    { { "subqcif", 128, 96 }, { "qsif", 160, 120 }, { "qcif", 176, 144 },
371
      { "sif", 320, 240 }, { "cif", 352, 288 }, { "d1", 640, 480 },
gbazin's avatar
gbazin committed
372 373 374 375 376
      { 0, 0, 0 },
    };

    var_Create( p_this, "dshow-size", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-size", &val );
377
    if( val.psz_string && *val.psz_string )
gbazin's avatar
 
gbazin committed
378
    {
gbazin's avatar
gbazin committed
379
        for( i = 0; size_table[i].psz_size; i++ )
380
        {
gbazin's avatar
gbazin committed
381 382 383 384 385 386
            if( !strcmp( val.psz_string, size_table[i].psz_size ) )
            {
                i_width = size_table[i].i_width;
                i_height = size_table[i].i_height;
                break;
            }
387
        }
gbazin's avatar
gbazin committed
388
        if( !size_table[i].psz_size ) /* Try to parse "WidthxHeight" */
389 390 391 392
        {
            char *psz_parser;
            i_width = strtol( val.psz_string, &psz_parser, 0 );
            if( *psz_parser == 'x' || *psz_parser == 'X')
gbazin's avatar
 
gbazin committed
393
            {
394
                i_height = strtol( psz_parser + 1, &psz_parser, 0 );
gbazin's avatar
 
gbazin committed
395
            }
zorglub's avatar
zorglub committed
396
            msg_Dbg( p_this, "width x height %dx%d", i_width, i_height );
gbazin's avatar
 
gbazin committed
397 398
        }
    }
399
    free( val.psz_string );
gbazin's avatar
 
gbazin committed
400

401
    p_sys->b_chroma = false;
gbazin's avatar
gbazin committed
402 403
    var_Create( p_this, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-chroma", &val );
404 405 406 407
    if( val.psz_string && strlen( val.psz_string ) >= 4 )
    {
        i_chroma = VLC_FOURCC( val.psz_string[0], val.psz_string[1],
                               val.psz_string[2], val.psz_string[3] );
408
        p_sys->b_chroma = true;
409
    }
410
    free( val.psz_string );
gbazin's avatar
 
gbazin committed
411

412
    var_Create( p_this, "dshow-fps", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
413 414 415 416 417 418 419
    var_Create( p_this, "dshow-tuner-channel",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-tuner-country",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-tuner-input",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );

420 421 422
    var_Create( p_this, "dshow-amtuner-mode",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );

gbazin's avatar
gbazin committed
423
    var_Create( p_this, "dshow-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
gbazin's avatar
 
gbazin committed
424

425 426 427 428 429
    var_Create( p_this, "dshow-video-input", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-audio-input", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-video-output", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
    var_Create( p_this, "dshow-audio-output", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );

gbazin's avatar
 
gbazin committed
430
    /* Initialize OLE/COM */
431
    CoInitialize( 0 );
gbazin's avatar
 
gbazin committed
432

gbazin's avatar
 
gbazin committed
433 434
    /* Initialize some data */
    p_sys->i_streams = 0;
ivoire's avatar
ivoire committed
435
    p_sys->pp_streams = NULL;
gbazin's avatar
 
gbazin committed
436 437 438
    p_sys->i_width = i_width;
    p_sys->i_height = i_height;
    p_sys->i_chroma = i_chroma;
gbazin's avatar
 
gbazin committed
439

440 441 442
    p_sys->p_graph = NULL;
    p_sys->p_capture_graph_builder2 = NULL;
    p_sys->p_control = NULL;
gbazin's avatar
 
gbazin committed
443

444
    vlc_mutex_init( &p_sys->lock );
445
    vlc_cond_init( &p_sys->wait );
446

447 448
    /* Build directshow graph */
    CreateDirectShowGraph( p_sys );
gbazin's avatar
 
gbazin committed
449

gbazin's avatar
gbazin committed
450
    if( OpenDevice( p_this, p_sys, vdevname, 0 ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
451
    {
gbazin's avatar
gbazin committed
452
        msg_Err( p_this, "can't open video");
gbazin's avatar
 
gbazin committed
453
    }
gbazin's avatar
gbazin committed
454 455 456 457 458 459 460 461 462 463 464 465 466 467 468
    else
    {
        /* Check if we can handle the demuxing ourselves or need to spawn
         * a demuxer module */
        dshow_stream_t *p_stream = p_sys->pp_streams[p_sys->i_streams-1];

        if( p_stream->mt.majortype == MEDIATYPE_Video )
        {
            if( /* Raw DV stream */
                p_stream->i_fourcc == VLC_FOURCC('d','v','s','l') ||
                p_stream->i_fourcc == VLC_FOURCC('d','v','s','d') ||
                p_stream->i_fourcc == VLC_FOURCC('d','v','h','d') ||
                /* Raw MPEG video stream */
                p_stream->i_fourcc == VLC_FOURCC('m','p','2','v') )
            {
469
                b_audio = false;
470 471 472 473

                if( b_access_demux )
                {
                    /* Let the access (only) take care of that */
474
                    return VLC_EGENERIC;
475
                }
gbazin's avatar
gbazin committed
476 477
            }
        }
gbazin's avatar
 
gbazin committed
478

479 480
        if( p_stream->mt.majortype == MEDIATYPE_Stream )
        {
481
            b_audio = false;
482 483 484 485 486 487

            if( b_access_demux )
            {
                /* Let the access (only) take care of that */
                return VLC_EGENERIC;
            }
488 489 490 491 492 493 494 495 496

            var_Get( p_this, "dshow-tuner", &val );
            if( val.b_bool )
            {
                /* FIXME: we do MEDIATYPE_Stream here so we don't do
                 * it twice. */
                ShowTunerProperties( p_this, p_sys->p_capture_graph_builder2,
                                     p_stream->p_device_filter, 0 );
            }
497
        }
gbazin's avatar
gbazin committed
498 499 500
    }

    if( b_audio && OpenDevice( p_this, p_sys, adevname, 1 ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
501
    {
gbazin's avatar
gbazin committed
502
        msg_Err( p_this, "can't open audio");
gbazin's avatar
 
gbazin committed
503
    }
gbazin's avatar
gbazin committed
504

505
    for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
gbazin's avatar
 
gbazin committed
506
    {
507
            var_Get( p_this, "dshow-video-input", &val );
508
            if( val.i_int >= 0 )
509 510
                    p_sys->crossbar_routes[i].VideoInputIndex=val.i_int;
            var_Get( p_this, "dshow-video-output", &val );
511
            if( val.i_int >= 0 )
512 513
                    p_sys->crossbar_routes[i].VideoOutputIndex=val.i_int;
            var_Get( p_this, "dshow-audio-input", &val );
514
            if( val.i_int >= 0 )
515 516
                    p_sys->crossbar_routes[i].AudioInputIndex=val.i_int;
            var_Get( p_this, "dshow-audio-output", &val );
517
            if( val.i_int >= 0 )
518 519
                    p_sys->crossbar_routes[i].AudioOutputIndex=val.i_int;

520 521 522 523 524
        IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar;
        LONG VideoInputIndex = p_sys->crossbar_routes[i].VideoInputIndex;
        LONG VideoOutputIndex = p_sys->crossbar_routes[i].VideoOutputIndex;
        LONG AudioInputIndex = p_sys->crossbar_routes[i].AudioInputIndex;
        LONG AudioOutputIndex = p_sys->crossbar_routes[i].AudioOutputIndex;
gbazin's avatar
 
gbazin committed
525

526 527
        if( SUCCEEDED(pXbar->Route(VideoOutputIndex, VideoInputIndex)) )
        {
zorglub's avatar
zorglub committed
528 529
            msg_Dbg( p_this, "crossbar at depth %d, routed video "
                     "output %ld to video input %ld", i, VideoOutputIndex,
530
                     VideoInputIndex );
gbazin's avatar
 
gbazin committed
531

532 533 534 535 536
            if( AudioOutputIndex != -1 && AudioInputIndex != -1 )
            {
                if( SUCCEEDED( pXbar->Route(AudioOutputIndex,
                                            AudioInputIndex)) )
                {
zorglub's avatar
zorglub committed
537 538
                    msg_Dbg(p_this, "crossbar at depth %d, routed audio "
                            "output %ld to audio input %ld", i,
539 540 541 542 543 544
                            AudioOutputIndex, AudioInputIndex );
                }
            }
        }
    }

545 546 547 548
    /*
    ** Show properties pages from other filters in graph
    */
    var_Get( p_this, "dshow-config", &val );
549
    if( val.b_bool )
550 551 552 553 554
    {
        for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
        {
            IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar;
            IBaseFilter *p_XF;
555 556 557

            if( SUCCEEDED( pXbar->QueryInterface( IID_IBaseFilter,
                                                  (void **)&p_XF ) ) )
558 559 560 561
            {
                ShowPropertyPage( p_XF );
                p_XF->Release();
            }
562
        }
563 564
    }

gbazin's avatar
 
gbazin committed
565 566
    /* Initialize some data */
    p_sys->i_current_stream = 0;
567

gbazin's avatar
gbazin committed
568
    if( !p_sys->i_streams ) return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
569

gbazin's avatar
 
gbazin committed
570 571 572 573
    return VLC_SUCCESS;
}

/*****************************************************************************
gbazin's avatar
gbazin committed
574
 * DemuxOpen: open direct show device as an access_demux module
gbazin's avatar
 
gbazin committed
575
 *****************************************************************************/
gbazin's avatar
gbazin committed
576
static int DemuxOpen( vlc_object_t *p_this )
gbazin's avatar
 
gbazin committed
577
{
gbazin's avatar
gbazin committed
578 579 580
    demux_t      *p_demux = (demux_t *)p_this;
    access_sys_t *p_sys;
    int i;
gbazin's avatar
 
gbazin committed
581

ivoire's avatar
ivoire committed
582
    p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
ivoire's avatar
ivoire committed
583 584
    if( !p_sys )
        return VLC_ENOMEM;
gbazin's avatar
gbazin committed
585
    p_demux->p_sys = (demux_sys_t *)p_sys;
gbazin's avatar
 
gbazin committed
586

587
    if( CommonOpen( p_this, p_sys, true ) != VLC_SUCCESS )
gbazin's avatar
gbazin committed
588 589 590 591
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
592

gbazin's avatar
gbazin committed
593 594 595
    /* Everything is ready. Let's rock baby */
    msg_Dbg( p_this, "Playing...");
    p_sys->p_control->Run();
596

gbazin's avatar
gbazin committed
597 598 599 600 601 602 603
    p_demux->pf_demux   = Demux;
    p_demux->pf_control = DemuxControl;
    p_demux->info.i_update = 0;
    p_demux->info.i_title = 0;
    p_demux->info.i_seekpoint = 0;

    for( i = 0; i < p_sys->i_streams; i++ )
604
    {
gbazin's avatar
gbazin committed
605 606
        dshow_stream_t *p_stream = p_sys->pp_streams[i];
        es_format_t fmt;
607

gbazin's avatar
gbazin committed
608 609 610
        if( p_stream->mt.majortype == MEDIATYPE_Video )
        {
            es_format_Init( &fmt, VIDEO_ES, p_stream->i_fourcc );
611

gbazin's avatar
gbazin committed
612 613 614
            fmt.video.i_width  = p_stream->header.video.bmiHeader.biWidth;
            fmt.video.i_height = p_stream->header.video.bmiHeader.biHeight;
            fmt.video.i_aspect = 4 * VOUT_ASPECT_FACTOR / 3;
615

gbazin's avatar
gbazin committed
616 617 618 619 620
            if( !p_stream->header.video.bmiHeader.biCompression )
            {
                /* RGB DIB are coded from bottom to top */
                fmt.video.i_height = (unsigned int)(-(int)fmt.video.i_height);
            }
621

gbazin's avatar
gbazin committed
622 623 624 625
            /* Setup rgb mask for RGB formats */
            if( p_stream->i_fourcc == VLC_FOURCC('R','V','2','4') )
            {
                /* This is in BGR format */
626 627 628
                fmt.video.i_bmask = 0x00ff0000;
                fmt.video.i_gmask = 0x0000ff00;
                fmt.video.i_rmask = 0x000000ff;
gbazin's avatar
gbazin committed
629
            }
630 631 632 633 634 635 636

            if( p_stream->header.video.AvgTimePerFrame )
            {
                fmt.video.i_frame_rate = 10000000;
                fmt.video.i_frame_rate_base =
                    p_stream->header.video.AvgTimePerFrame;
            }
gbazin's avatar
gbazin committed
637 638 639 640
        }
        else if( p_stream->mt.majortype == MEDIATYPE_Audio )
        {
            es_format_Init( &fmt, AUDIO_ES, p_stream->i_fourcc );
641

gbazin's avatar
gbazin committed
642 643 644 645 646 647 648 649 650 651
            fmt.audio.i_channels = p_stream->header.audio.nChannels;
            fmt.audio.i_rate = p_stream->header.audio.nSamplesPerSec;
            fmt.audio.i_bitspersample = p_stream->header.audio.wBitsPerSample;
            fmt.audio.i_blockalign = fmt.audio.i_channels *
                fmt.audio.i_bitspersample / 8;
            fmt.i_bitrate = fmt.audio.i_channels * fmt.audio.i_rate *
                fmt.audio.i_bitspersample;
        }

        p_stream->p_es = es_out_Add( p_demux->out, &fmt );
652
    }
gbazin's avatar
gbazin committed
653

654
    return VLC_SUCCESS;
655 656
}

gbazin's avatar
gbazin committed
657 658 659 660
/*****************************************************************************
 * AccessOpen: open direct show device as an access module
 *****************************************************************************/
static int AccessOpen( vlc_object_t *p_this )
661
{
gbazin's avatar
gbazin committed
662 663
    access_t     *p_access = (access_t*)p_this;
    access_sys_t *p_sys;
664

ivoire's avatar
ivoire committed
665
    p_access->p_sys = p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
ivoire's avatar
ivoire committed
666 667
    if( !p_sys )
        return VLC_ENOMEM;
668

669
    if( CommonOpen( p_this, p_sys, false ) != VLC_SUCCESS )
gbazin's avatar
gbazin committed
670 671 672 673
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
674

gbazin's avatar
gbazin committed
675
    dshow_stream_t *p_stream = p_sys->pp_streams[0];
676

gbazin's avatar
gbazin committed
677 678
    /* Check if we need to force demuxers */
    if( !p_access->psz_demux || !*p_access->psz_demux )
gbazin's avatar
 
gbazin committed
679
    {
gbazin's avatar
gbazin committed
680 681 682
        if( p_stream->i_fourcc == VLC_FOURCC('d','v','s','l') ||
            p_stream->i_fourcc == VLC_FOURCC('d','v','s','d') ||
            p_stream->i_fourcc == VLC_FOURCC('d','v','h','d') )
683
        {
Rafaël Carré's avatar
Rafaël Carré committed
684
            free( p_access->psz_demux );
gbazin's avatar
gbazin committed
685 686 687 688
            p_access->psz_demux = strdup( "rawdv" );
        }
        else if( p_stream->i_fourcc == VLC_FOURCC('m','p','2','v') )
        {
Rafaël Carré's avatar
Rafaël Carré committed
689 690
            free( p_access->psz_demux );
            p_access->psz_demux = strdup( "mpgv" );
691
        }
gbazin's avatar
 
gbazin committed
692
    }
gbazin's avatar
 
gbazin committed
693

gbazin's avatar
gbazin committed
694
    /* Setup Access */
695 696
    p_access->pf_read = NULL;
    p_access->pf_block = ReadCompressed;
gbazin's avatar
gbazin committed
697 698 699 700 701
    p_access->pf_control = AccessControl;
    p_access->pf_seek = NULL;
    p_access->info.i_update = 0;
    p_access->info.i_size = 0;
    p_access->info.i_pos = 0;
702
    p_access->info.b_eof = false;
gbazin's avatar
gbazin committed
703 704 705 706 707 708 709 710 711
    p_access->info.i_title = 0;
    p_access->info.i_seekpoint = 0;
    p_access->p_sys = p_sys;

    /* Everything is ready. Let's rock baby */
    msg_Dbg( p_this, "Playing...");
    p_sys->p_control->Run();

    return VLC_SUCCESS;
712
}
gbazin's avatar
 
gbazin committed
713

gbazin's avatar
gbazin committed
714 715 716 717
/*****************************************************************************
 * CommonClose: close device
 *****************************************************************************/
static void CommonClose( vlc_object_t *p_this, access_sys_t *p_sys )
718
{
zorglub's avatar
zorglub committed
719
    msg_Dbg( p_this, "releasing DirectShow");
720

gbazin's avatar
gbazin committed
721
    DeleteDirectShowGraph( p_sys );
722

gbazin's avatar
gbazin committed
723 724
    /* Uninitialize OLE/COM */
    CoUninitialize();
725

726 727
    for( int i = 0; i < p_sys->i_streams; i++ ) delete p_sys->pp_streams[i];
    if( p_sys->i_streams ) free( p_sys->pp_streams );
gbazin's avatar
gbazin committed
728 729 730

    vlc_mutex_destroy( &p_sys->lock );
    vlc_cond_destroy( &p_sys->wait );
731

gbazin's avatar
gbazin committed
732
    free( p_sys );
733 734
}

gbazin's avatar
gbazin committed
735 736 737 738
/*****************************************************************************
 * AccessClose: close device
 *****************************************************************************/
static void AccessClose( vlc_object_t *p_this )
739
{
gbazin's avatar
gbazin committed
740 741
    access_t     *p_access = (access_t *)p_this;
    access_sys_t *p_sys    = p_access->p_sys;
742

gbazin's avatar
gbazin committed
743 744
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
745

gbazin's avatar
gbazin committed
746 747
    CommonClose( p_this, p_sys );
}
748

gbazin's avatar
gbazin committed
749 750 751 752 753 754 755
/*****************************************************************************
 * DemuxClose: close device
 *****************************************************************************/
static void DemuxClose( vlc_object_t *p_this )
{
    demux_t      *p_demux = (demux_t *)p_this;
    access_sys_t *p_sys   = (access_sys_t *)p_demux->p_sys;
756

gbazin's avatar
gbazin committed
757 758
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
759

gbazin's avatar
gbazin committed
760
    CommonClose( p_this, p_sys );
gbazin's avatar
 
gbazin committed
761 762 763 764 765
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
gbazin's avatar
gbazin committed
766 767
static bool ConnectFilters( vlc_object_t *p_this, access_sys_t *p_sys,
                            IBaseFilter *p_filter,
768
                            CaptureFilter *p_capture_filter )
gbazin's avatar
 
gbazin committed
769
{
770
    CapturePin *p_input_pin = p_capture_filter->CustomGetPin();
gbazin's avatar
 
gbazin committed
771

772
    AM_MEDIA_TYPE mediaType = p_input_pin->CustomGetMediaType();
gbazin's avatar
 
gbazin committed
773

774
    if( p_sys->p_capture_graph_builder2 )
gbazin's avatar
 
gbazin committed
775
    {
776
        if( FAILED(p_sys->p_capture_graph_builder2->
gbazin's avatar
gbazin committed
777 778
                RenderStream( &PIN_CATEGORY_CAPTURE, &mediaType.majortype,
                              p_filter, 0, (IBaseFilter *)p_capture_filter )) )
779 780 781
        {
            return false;
        }
782

783 784
        // Sort out all the possible video inputs
        // The class needs to be given the capture filters ANALOGVIDEO input pin
ivoire's avatar
ivoire committed
785
        IEnumPins *pins = NULL;
786 787
        if( ( mediaType.majortype == MEDIATYPE_Video ||
              mediaType.majortype == MEDIATYPE_Stream ) &&
788
            SUCCEEDED(p_filter->EnumPins(&pins)) )
gbazin's avatar
 
gbazin committed
789
        {
ivoire's avatar
ivoire committed
790
            IPin        *pP = NULL;
791 792 793
            ULONG        n;
            PIN_INFO     pinInfo;
            BOOL         Found = FALSE;
ivoire's avatar
ivoire committed
794
            IKsPropertySet *pKs = NULL;
795 796 797
            GUID guid;
            DWORD dw;

gbazin's avatar
gbazin committed
798
            while( !Found && ( S_OK == pins->Next(1, &pP, &n) ) )
799
            {
gbazin's avatar
gbazin committed
800
                if( S_OK == pP->QueryPinInfo(&pinInfo) )