dshow.cpp 78.3 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. " \
zorglub's avatar
zorglub committed
101
    "This value should be set in millisecondss." )
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, TV, AM_RADIO, FM_RADIO or DSS.")

163 164
#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
165
    "Select audio input format with the given number of audio channels (if non 0)" )
166 167 168 169 170 171 172 173 174

#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)" )

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

242 243 244 245
    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 );
246 247
    add_integer( "dshow-audio-bitspersample", 0, NULL, AUDIO_BITSPERSAMPLE_TEXT,
                 AUDIO_BITSPERSAMPLE_LONGTEXT, true );
248

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

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

vlc_module_end();

gbazin's avatar
gbazin committed
260
/*****************************************************************************
gbazin's avatar
 
gbazin committed
261
 * DirectShow elementary stream descriptor
gbazin's avatar
gbazin committed
262
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
263 264 265 266 267 268 269 270 271 272 273 274 275 276
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
277 278 279
    int             i_fourcc;
    es_out_id_t     *p_es;

280
    bool      b_pts;
gbazin's avatar
 
gbazin committed
281 282
} dshow_stream_t;

gbazin's avatar
gbazin committed
283
/*****************************************************************************
284
 * DirectShow utility functions
gbazin's avatar
gbazin committed
285
 *****************************************************************************/
286 287 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
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
340
/*****************************************************************************
gbazin's avatar
gbazin committed
341
 * CommonOpen: open direct show device
gbazin's avatar
 
gbazin committed
342
 *****************************************************************************/
gbazin's avatar
gbazin committed
343
static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
344
                       bool b_access_demux )
gbazin's avatar
 
gbazin committed
345
{
346
    vlc_value_t  val;
gbazin's avatar
gbazin committed
347
    int i;
gbazin's avatar
 
gbazin committed
348

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

354 355
    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
356 357
    var_Create( p_this, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-vdev", &val );
358
    if( val.psz_string ) vdevname = string( val.psz_string );
359
    free( val.psz_string );
gbazin's avatar
 
gbazin committed
360

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

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

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

398
    p_sys->b_chroma = false;
gbazin's avatar
gbazin committed
399 400
    var_Create( p_this, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-chroma", &val );
401 402 403 404
    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] );
405
        p_sys->b_chroma = true;
406
    }
407
    free( val.psz_string );
gbazin's avatar
 
gbazin committed
408

409
    var_Create( p_this, "dshow-fps", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
410 411 412 413 414 415 416
    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 );

417 418 419
    var_Create( p_this, "dshow-amtuner-mode",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );

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

422 423 424 425 426
    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
427
    /* Initialize OLE/COM */
428
    CoInitialize( 0 );
gbazin's avatar
 
gbazin committed
429

gbazin's avatar
 
gbazin committed
430 431
    /* Initialize some data */
    p_sys->i_streams = 0;
432
    p_sys->pp_streams = 0;
gbazin's avatar
 
gbazin committed
433 434 435
    p_sys->i_width = i_width;
    p_sys->i_height = i_height;
    p_sys->i_chroma = i_chroma;
gbazin's avatar
 
gbazin committed
436

437 438 439
    p_sys->p_graph = NULL;
    p_sys->p_capture_graph_builder2 = NULL;
    p_sys->p_control = NULL;
gbazin's avatar
 
gbazin committed
440

441
    vlc_mutex_init( &p_sys->lock );
442 443
    vlc_cond_init( p_this, &p_sys->wait );

444 445
    /* Build directshow graph */
    CreateDirectShowGraph( p_sys );
gbazin's avatar
 
gbazin committed
446

gbazin's avatar
gbazin committed
447
    if( OpenDevice( p_this, p_sys, vdevname, 0 ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
448
    {
gbazin's avatar
gbazin committed
449
        msg_Err( p_this, "can't open video");
gbazin's avatar
 
gbazin committed
450
    }
gbazin's avatar
gbazin committed
451 452 453 454 455 456 457 458 459 460 461 462 463 464 465
    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') )
            {
466
                b_audio = false;
467 468 469 470

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

476 477
        if( p_stream->mt.majortype == MEDIATYPE_Stream )
        {
478
            b_audio = false;
479 480 481 482 483 484

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

            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 );
            }
494
        }
gbazin's avatar
gbazin committed
495 496 497
    }

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

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

517 518 519 520 521
        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
522

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

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

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

            if( SUCCEEDED( pXbar->QueryInterface( IID_IBaseFilter,
                                                  (void **)&p_XF ) ) )
555 556 557 558
            {
                ShowPropertyPage( p_XF );
                p_XF->Release();
            }
559
        }
560 561
    }

gbazin's avatar
 
gbazin committed
562 563
    /* Initialize some data */
    p_sys->i_current_stream = 0;
564

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

gbazin's avatar
 
gbazin committed
567 568 569 570
    return VLC_SUCCESS;
}

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

gbazin's avatar
gbazin committed
579 580 581
    p_sys = (access_sys_t *)malloc( sizeof( access_sys_t ) );
    memset( p_sys, 0, sizeof( access_sys_t ) );
    p_demux->p_sys = (demux_sys_t *)p_sys;
gbazin's avatar
 
gbazin committed
582

583
    if( CommonOpen( p_this, p_sys, true ) != VLC_SUCCESS )
gbazin's avatar
gbazin committed
584 585 586 587
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
588

gbazin's avatar
gbazin committed
589 590 591
    /* Everything is ready. Let's rock baby */
    msg_Dbg( p_this, "Playing...");
    p_sys->p_control->Run();
592

gbazin's avatar
gbazin committed
593 594 595 596 597 598 599
    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++ )
600
    {
gbazin's avatar
gbazin committed
601 602
        dshow_stream_t *p_stream = p_sys->pp_streams[i];
        es_format_t fmt;
603

gbazin's avatar
gbazin committed
604 605 606
        if( p_stream->mt.majortype == MEDIATYPE_Video )
        {
            es_format_Init( &fmt, VIDEO_ES, p_stream->i_fourcc );
607

gbazin's avatar
gbazin committed
608 609 610
            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;
611

gbazin's avatar
gbazin committed
612 613 614 615 616
            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);
            }
617

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

            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
633 634 635 636
        }
        else if( p_stream->mt.majortype == MEDIATYPE_Audio )
        {
            es_format_Init( &fmt, AUDIO_ES, p_stream->i_fourcc );
637

gbazin's avatar
gbazin committed
638 639 640 641 642 643 644 645 646 647
            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 );
648
    }
gbazin's avatar
gbazin committed
649

650
    return VLC_SUCCESS;
651 652
}

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

gbazin's avatar
gbazin committed
661 662
    p_access->p_sys = p_sys = (access_sys_t *)malloc( sizeof( access_sys_t ) );
    memset( p_sys, 0, sizeof( access_sys_t ) );
663

664
    if( CommonOpen( p_this, p_sys, false ) != VLC_SUCCESS )
gbazin's avatar
gbazin committed
665 666 667 668
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
669

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

gbazin's avatar
gbazin committed
672 673
    /* Check if we need to force demuxers */
    if( !p_access->psz_demux || !*p_access->psz_demux )
gbazin's avatar
 
gbazin committed
674
    {
gbazin's avatar
gbazin committed
675 676 677
        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') )
678
        {
Rafaël Carré's avatar
Rafaël Carré committed
679
            free( p_access->psz_demux );
gbazin's avatar
gbazin committed
680 681 682 683
            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
684 685
            free( p_access->psz_demux );
            p_access->psz_demux = strdup( "mpgv" );
686
        }
gbazin's avatar
 
gbazin committed
687
    }
gbazin's avatar
 
gbazin committed
688

gbazin's avatar
gbazin committed
689
    /* Setup Access */
690 691
    p_access->pf_read = NULL;
    p_access->pf_block = ReadCompressed;
gbazin's avatar
gbazin committed
692 693 694 695 696
    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;
697
    p_access->info.b_eof = false;
gbazin's avatar
gbazin committed
698 699 700 701 702 703 704 705 706
    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;
707
}
gbazin's avatar
 
gbazin committed
708

gbazin's avatar
gbazin committed
709 710 711 712
/*****************************************************************************
 * CommonClose: close device
 *****************************************************************************/
static void CommonClose( vlc_object_t *p_this, access_sys_t *p_sys )
713
{
zorglub's avatar
zorglub committed
714
    msg_Dbg( p_this, "releasing DirectShow");
715

gbazin's avatar
gbazin committed
716
    DeleteDirectShowGraph( p_sys );
717

gbazin's avatar
gbazin committed
718 719
    /* Uninitialize OLE/COM */
    CoUninitialize();
720

721 722
    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
723 724 725

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

gbazin's avatar
gbazin committed
727
    free( p_sys );
728 729
}

gbazin's avatar
gbazin committed
730 731 732 733
/*****************************************************************************
 * AccessClose: close device
 *****************************************************************************/
static void AccessClose( vlc_object_t *p_this )
734
{
gbazin's avatar
gbazin committed
735 736
    access_t     *p_access = (access_t *)p_this;
    access_sys_t *p_sys    = p_access->p_sys;
737

gbazin's avatar
gbazin committed
738 739
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
740

gbazin's avatar
gbazin committed
741 742
    CommonClose( p_this, p_sys );
}
743

gbazin's avatar
gbazin committed
744 745 746 747 748 749 750
/*****************************************************************************
 * 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;
751

gbazin's avatar
gbazin committed
752 753
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
754

gbazin's avatar
gbazin committed
755
    CommonClose( p_this, p_sys );
gbazin's avatar
 
gbazin committed
756 757 758 759 760
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
gbazin's avatar
gbazin committed
761 762
static bool ConnectFilters( vlc_object_t *p_this, access_sys_t *p_sys,
                            IBaseFilter *p_filter,
763
                            CaptureFilter *p_capture_filter )
gbazin's avatar
 
gbazin committed
764
{
765
    CapturePin *p_input_pin = p_capture_filter->CustomGetPin();
gbazin's avatar
 
gbazin committed
766

767
    AM_MEDIA_TYPE mediaType = p_input_pin->CustomGetMediaType();
gbazin's avatar
 
gbazin committed
768

769
    if( p_sys->p_capture_graph_builder2 )
gbazin's avatar
 
gbazin committed
770
    {
771
        if( FAILED(p_sys->p_capture_graph_builder2->
gbazin's avatar
gbazin committed
772 773
                RenderStream( &PIN_CATEGORY_CAPTURE, &mediaType.majortype,
                              p_filter, 0, (IBaseFilter *)p_capture_filter )) )
774 775 776
        {
            return false;
        }
777

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

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