dshow.cpp 83.3 KB
Newer Older
Gildas Bazin's avatar
 
Gildas Bazin committed
1
/*****************************************************************************
2
 * dshow.cpp : DirectShow access module for vlc
Gildas Bazin's avatar
 
Gildas Bazin committed
3
 *****************************************************************************
4
 * Copyright (C) 2002, 2003 the VideoLAN team
5
 * $Id$
Gildas Bazin's avatar
 
Gildas Bazin committed
6
 *
7
 * Author: Gildas Bazin <gbazin@videolan.org>
8
 *         Damien Fouilleul <damienf@videolan.org>
Gildas Bazin's avatar
 
Gildas Bazin 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
Antoine Cellerier's avatar
Antoine Cellerier committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
Gildas Bazin's avatar
 
Gildas Bazin committed
23 24 25 26 27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
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>
Clément Stenac's avatar
Clément Stenac committed
38 39
#include <vlc_input.h>
#include <vlc_access.h>
40
#include <vlc_demux.h>
Clément Stenac's avatar
Clément Stenac committed
41
#include <vlc_vout.h>
42
#include <vlc_dialog.h>
Gildas Bazin's avatar
 
Gildas Bazin committed
43

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

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

Gildas Bazin's avatar
Gildas Bazin 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 );
Gildas Bazin's avatar
 
Gildas Bazin 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 );
Gildas Bazin's avatar
Gildas Bazin committed
62 63
static bool ConnectFilters( vlc_object_t *, access_sys_t *,
                            IBaseFilter *, CaptureFilter * );
Gildas Bazin's avatar
 
Gildas Bazin 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 * );
Gildas Bazin's avatar
 
Gildas Bazin 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 * );
Gildas Bazin's avatar
 
Gildas Bazin committed
76

Gildas Bazin's avatar
 
Gildas Bazin committed
77
/*****************************************************************************
78
 * Module descriptor
Gildas Bazin's avatar
 
Gildas Bazin 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

Gildas Bazin's avatar
 
Gildas Bazin 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac 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")
Gildas Bazin's avatar
 
Gildas Bazin 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac committed
142
    "Select the tuner input type (Cable/Antenna)." )
143 144
#define VIDEO_IN_TEXT N_("Video input pin")
#define VIDEO_IN_LONGTEXT N_( \
Clément Stenac's avatar
Clément Stenac 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 " \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac 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_( \
Clément Stenac's avatar
Clément Stenac committed
157
  "Select the audio output type. See the \"video input\" option." )
Gildas Bazin's avatar
 
Gildas Bazin committed
158

159
#define AMTUNER_MODE_TEXT N_("AM Tuner mode")
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 );
Gildas Bazin's avatar
Gildas Bazin 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,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
191
                 CACHING_TEXT, CACHING_LONGTEXT, true )
192

193
    add_string( "dshow-vdev", NULL, NULL, VDEV_TEXT, VDEV_LONGTEXT, false)
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
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)
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
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, true )
Gildas Bazin's avatar
 
Gildas Bazin committed
206

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

209
    add_bool( "dshow-config", false, NULL, CONFIG_TEXT, CONFIG_LONGTEXT, true )
Gildas Bazin's avatar
 
Gildas Bazin committed
210

211
    add_bool( "dshow-tuner", false, NULL, TUNER_TEXT, TUNER_LONGTEXT, true )
212

213 214
    add_integer( "dshow-tuner-channel", 0, NULL, CHANNEL_TEXT, CHANNEL_LONGTEXT,
                true )
215

216 217
    add_integer( "dshow-tuner-country", 0, NULL, COUNTRY_TEXT, COUNTRY_LONGTEXT,
                true )
218 219

    add_integer( "dshow-tuner-input", 0, NULL, TUNER_INPUT_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
220 221
                 TUNER_INPUT_LONGTEXT, true )
        change_integer_list( pi_tuner_input, ppsz_tuner_input_text, NULL )
222

223
    add_integer( "dshow-video-input",  -1, NULL, VIDEO_IN_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
224
                 VIDEO_IN_LONGTEXT, true )
225 226

    add_integer( "dshow-video-output", -1, NULL, VIDEO_OUT_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
227
                 VIDEO_OUT_LONGTEXT, true )
228

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

232
    add_integer( "dshow-audio-output", -1, NULL, AUDIO_OUT_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
233
                 AUDIO_OUT_LONGTEXT, true )
234

235
    add_integer( "dshow-amtuner-mode", AMTUNER_MODE_TV, NULL,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
236 237
                AMTUNER_MODE_TEXT, AMTUNER_MODE_LONGTEXT, false)
        change_integer_list( pi_amtuner_mode, ppsz_amtuner_mode_text, NULL )
238

239
    add_integer( "dshow-audio-channels", 0, NULL, AUDIO_CHANNELS_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
240
                 AUDIO_CHANNELS_LONGTEXT, true )
241
    add_integer( "dshow-audio-samplerate", 0, NULL, AUDIO_SAMPLERATE_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
242
                 AUDIO_SAMPLERATE_LONGTEXT, true )
243
    add_integer( "dshow-audio-bitspersample", 0, NULL, AUDIO_BITSPERSAMPLE_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
244
                 AUDIO_BITSPERSAMPLE_LONGTEXT, true )
245

246 247 248
    add_shortcut( "dshow" )
    set_capability( "access_demux", 0 )
    set_callbacks( DemuxOpen, DemuxClose )
Gildas Bazin's avatar
 
Gildas Bazin committed
249

250 251 252 253
    add_submodule ()
    set_description( N_("DirectShow input") )
    set_capability( "access", 0 )
    set_callbacks( AccessOpen, AccessClose )
Gildas Bazin's avatar
 
Gildas Bazin committed
254

255
vlc_module_end ()
Gildas Bazin's avatar
 
Gildas Bazin committed
256

Gildas Bazin's avatar
Gildas Bazin committed
257
/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
258
 * DirectShow elementary stream descriptor
Gildas Bazin's avatar
Gildas Bazin committed
259
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
260 261 262 263 264 265 266 267 268 269 270 271 272 273
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;

Gildas Bazin's avatar
Gildas Bazin committed
274 275 276
    int             i_fourcc;
    es_out_id_t     *p_es;

277
    bool      b_pts;
278 279

    deque<VLCMediaSample> samples_queue;
Gildas Bazin's avatar
 
Gildas Bazin committed
280 281
} dshow_stream_t;

Gildas Bazin's avatar
Gildas Bazin committed
282
/*****************************************************************************
283
 * DirectShow utility functions
Gildas Bazin's avatar
Gildas Bazin committed
284
 *****************************************************************************/
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
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;
    }
}

Gildas Bazin's avatar
 
Gildas Bazin committed
339
/*****************************************************************************
Gildas Bazin's avatar
Gildas Bazin committed
340
 * CommonOpen: open direct show device
Gildas Bazin's avatar
 
Gildas Bazin committed
341
 *****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
342
static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
343
                       bool b_access_demux )
Gildas Bazin's avatar
 
Gildas Bazin committed
344
{
345
    vlc_value_t  val;
Gildas Bazin's avatar
Gildas Bazin committed
346
    int i;
Gildas Bazin's avatar
 
Gildas Bazin committed
347

348
    /* Get/parse options and open device(s) */
Gildas Bazin's avatar
 
Gildas Bazin committed
349
    string vdevname, adevname;
350
    int i_width = 0, i_height = 0, i_chroma = 0;
351 352 353 354 355
    bool b_use_audio = true;
    bool b_use_video = true;

    /* Initialize OLE/COM */
    CoInitialize( 0 );
Gildas Bazin's avatar
 
Gildas Bazin 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 );
Gildas Bazin's avatar
Gildas Bazin committed
359 360
    var_Create( p_this, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-vdev", &val );
361 362 363 364 365 366 367 368 369
    if( val.psz_string )
    {
        msg_Dbg( p_this, "dshow-vdev: %s", val.psz_string ) ;
        /* skip none device */
        if ( strncasecmp( val.psz_string, "none", 4 ) != 0 )
            vdevname = string( val.psz_string );
        else
            b_use_video = false ;
    }
370
    free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
371

Gildas Bazin's avatar
Gildas Bazin committed
372 373
    var_Create( p_this, "dshow-adev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-adev", &val );
374 375 376 377 378 379 380 381 382
    if( val.psz_string )
    {
        msg_Dbg( p_this, "dshow-adev: %s", val.psz_string ) ;
        /* skip none device */
        if ( strncasecmp( val.psz_string, "none", 4 ) != 0 )
            adevname = string( val.psz_string );
        else
            b_use_audio = false ;
    }
383
    free( val.psz_string );
384

385
    static struct {const char *psz_size; int  i_width; int  i_height;} size_table[] =
Gildas Bazin's avatar
Gildas Bazin committed
386
    { { "subqcif", 128, 96 }, { "qsif", 160, 120 }, { "qcif", 176, 144 },
387
      { "sif", 320, 240 }, { "cif", 352, 288 }, { "d1", 640, 480 },
Gildas Bazin's avatar
Gildas Bazin committed
388 389 390 391 392
      { 0, 0, 0 },
    };

    var_Create( p_this, "dshow-size", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-size", &val );
393
    if( val.psz_string && *val.psz_string )
Gildas Bazin's avatar
 
Gildas Bazin committed
394
    {
Gildas Bazin's avatar
Gildas Bazin committed
395
        for( i = 0; size_table[i].psz_size; i++ )
396
        {
Gildas Bazin's avatar
Gildas Bazin committed
397 398 399 400 401 402
            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;
            }
403
        }
Gildas Bazin's avatar
Gildas Bazin committed
404
        if( !size_table[i].psz_size ) /* Try to parse "WidthxHeight" */
405 406 407 408
        {
            char *psz_parser;
            i_width = strtol( val.psz_string, &psz_parser, 0 );
            if( *psz_parser == 'x' || *psz_parser == 'X')
Gildas Bazin's avatar
 
Gildas Bazin committed
409
            {
410
                i_height = strtol( psz_parser + 1, &psz_parser, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
411
            }
Clément Stenac's avatar
Clément Stenac committed
412
            msg_Dbg( p_this, "width x height %dx%d", i_width, i_height );
Gildas Bazin's avatar
 
Gildas Bazin committed
413 414
        }
    }
415
    free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
416

417
    p_sys->b_chroma = false;
Gildas Bazin's avatar
Gildas Bazin committed
418 419
    var_Create( p_this, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-chroma", &val );
420 421 422 423
    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] );
424
        p_sys->b_chroma = true;
425
    }
426
    free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
427

428
    var_Create( p_this, "dshow-fps", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
429 430 431 432 433 434 435
    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 );

436 437 438
    var_Create( p_this, "dshow-amtuner-mode",
                VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );

Gildas Bazin's avatar
Gildas Bazin committed
439
    var_Create( p_this, "dshow-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
Gildas Bazin's avatar
 
Gildas Bazin committed
440

441 442 443 444 445
    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 );

Gildas Bazin's avatar
 
Gildas Bazin committed
446 447
    /* Initialize some data */
    p_sys->i_streams = 0;
Rémi Duraffort's avatar
Rémi Duraffort committed
448
    p_sys->pp_streams = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
449 450 451
    p_sys->i_width = i_width;
    p_sys->i_height = i_height;
    p_sys->i_chroma = i_chroma;
Gildas Bazin's avatar
 
Gildas Bazin committed
452

453 454 455
    p_sys->p_graph = NULL;
    p_sys->p_capture_graph_builder2 = NULL;
    p_sys->p_control = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
456

457 458 459
    /* Build directshow graph */
    CreateDirectShowGraph( p_sys );

460
    vlc_mutex_init( &p_sys->lock );
461
    vlc_cond_init( &p_sys->wait );
462

463 464
    if( !b_use_video && !b_use_audio )
    {
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
465
        dialog_Fatal( p_this, _("Capture failed"),
basos g's avatar
basos g committed
466
                        _("No video or audio device selected.") );
467 468
        return VLC_EGENERIC ;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
469

470 471 472 473 474
    if( !b_use_video )
        msg_Dbg( p_this, "skipping video device" ) ;
    bool b_err_video = false ;

    if( b_use_video && OpenDevice( p_this, p_sys, vdevname, 0 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
475
    {
476 477
        msg_Err( p_this, "can't open video device");
        b_err_video = true ;
Gildas Bazin's avatar
 
Gildas Bazin committed
478
    }
479 480

    if ( b_use_video && !b_err_video )
Gildas Bazin's avatar
Gildas Bazin committed
481 482 483 484 485 486 487 488 489 490 491 492 493 494
    {
        /* 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') )
            {
495
                b_use_audio = false;
496 497 498 499

                if( b_access_demux )
                {
                    /* Let the access (only) take care of that */
500
                    return VLC_EGENERIC;
501
                }
Gildas Bazin's avatar
Gildas Bazin committed
502 503
            }
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
504

505 506
        if( p_stream->mt.majortype == MEDIATYPE_Stream )
        {
507
            b_use_audio = false;
508 509 510 511 512 513

            if( b_access_demux )
            {
                /* Let the access (only) take care of that */
                return VLC_EGENERIC;
            }
514 515 516 517 518 519 520 521 522

            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 );
            }
523
        }
Gildas Bazin's avatar
Gildas Bazin committed
524 525
    }

526 527 528 529 530 531
    if( !b_use_audio )
        msg_Dbg( p_this, "skipping audio device") ;

    bool b_err_audio = false ;

    if( b_use_audio && OpenDevice( p_this, p_sys, adevname, 1 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
532
    {
533 534 535 536 537 538 539 540 541
        msg_Err( p_this, "can't open audio device");
        b_err_audio = true ;
    }

    if( ( b_use_video && b_err_video && b_use_audio && b_err_audio ) ||
        ( !b_use_video && b_use_audio && b_err_audio ) ||
        ( b_use_video && !b_use_audio && b_err_video ) )
    {
        msg_Err( p_this, "FATAL: could not open ANY device" ) ;
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
542
        dialog_Fatal( p_this,  _("Capture failed"),
543 544 545
                        _("VLC cannot open ANY capture device."
                          "Check the error log for details.") );
        return VLC_EGENERIC ;
Gildas Bazin's avatar
 
Gildas Bazin committed
546
    }
Gildas Bazin's avatar
Gildas Bazin committed
547

548
    for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
Gildas Bazin's avatar
 
Gildas Bazin committed
549
    {
550
            var_Get( p_this, "dshow-video-input", &val );
551
            if( val.i_int >= 0 )
552 553
                    p_sys->crossbar_routes[i].VideoInputIndex=val.i_int;
            var_Get( p_this, "dshow-video-output", &val );
554
            if( val.i_int >= 0 )
555 556
                    p_sys->crossbar_routes[i].VideoOutputIndex=val.i_int;
            var_Get( p_this, "dshow-audio-input", &val );
557
            if( val.i_int >= 0 )
558 559
                    p_sys->crossbar_routes[i].AudioInputIndex=val.i_int;
            var_Get( p_this, "dshow-audio-output", &val );
560
            if( val.i_int >= 0 )
561 562
                    p_sys->crossbar_routes[i].AudioOutputIndex=val.i_int;

563 564 565 566 567
        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;
Gildas Bazin's avatar
 
Gildas Bazin committed
568

569 570
        if( SUCCEEDED(pXbar->Route(VideoOutputIndex, VideoInputIndex)) )
        {
Clément Stenac's avatar
Clément Stenac committed
571 572
            msg_Dbg( p_this, "crossbar at depth %d, routed video "
                     "output %ld to video input %ld", i, VideoOutputIndex,
573
                     VideoInputIndex );
Gildas Bazin's avatar
 
Gildas Bazin committed
574

575 576 577 578 579
            if( AudioOutputIndex != -1 && AudioInputIndex != -1 )
            {
                if( SUCCEEDED( pXbar->Route(AudioOutputIndex,
                                            AudioInputIndex)) )
                {
Clément Stenac's avatar
Clément Stenac committed
580 581
                    msg_Dbg(p_this, "crossbar at depth %d, routed audio "
                            "output %ld to audio input %ld", i,
582 583 584 585
                            AudioOutputIndex, AudioInputIndex );
                }
            }
        }
586 587 588
        else
            msg_Err( p_this, "crossbar at depth %d could not route video "
                     "output %ld to input %ld", i, VideoOutputIndex, VideoInputIndex );
589 590
    }

591 592 593 594
    /*
    ** Show properties pages from other filters in graph
    */
    var_Get( p_this, "dshow-config", &val );
595
    if( val.b_bool )
596 597 598 599 600
    {
        for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
        {
            IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar;
            IBaseFilter *p_XF;
601 602 603

            if( SUCCEEDED( pXbar->QueryInterface( IID_IBaseFilter,
                                                  (void **)&p_XF ) ) )
604 605 606 607
            {
                ShowPropertyPage( p_XF );
                p_XF->Release();
            }
608
        }
609 610
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
611 612
    /* Initialize some data */
    p_sys->i_current_stream = 0;
613

Gildas Bazin's avatar
Gildas Bazin committed
614
    if( !p_sys->i_streams ) return VLC_EGENERIC;
Gildas Bazin's avatar
 
Gildas Bazin committed
615

Gildas Bazin's avatar
 
Gildas Bazin committed
616 617 618 619
    return VLC_SUCCESS;
}

/*****************************************************************************
Gildas Bazin's avatar
Gildas Bazin committed
620
 * DemuxOpen: open direct show device as an access_demux module
Gildas Bazin's avatar
 
Gildas Bazin committed
621
 *****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
622
static int DemuxOpen( vlc_object_t *p_this )
Gildas Bazin's avatar
 
Gildas Bazin committed
623
{
Gildas Bazin's avatar
Gildas Bazin committed
624 625 626
    demux_t      *p_demux = (demux_t *)p_this;
    access_sys_t *p_sys;
    int i;
Gildas Bazin's avatar
 
Gildas Bazin committed
627

Rémi Duraffort's avatar
Rémi Duraffort committed
628
    p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
Rémi Duraffort's avatar
Rémi Duraffort committed
629 630
    if( !p_sys )
        return VLC_ENOMEM;
Gildas Bazin's avatar
Gildas Bazin committed
631
    p_demux->p_sys = (demux_sys_t *)p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
632

633
    if( CommonOpen( p_this, p_sys, true ) != VLC_SUCCESS )
Gildas Bazin's avatar
Gildas Bazin committed
634 635 636 637
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
638

Gildas Bazin's avatar
Gildas Bazin committed
639 640 641
    /* Everything is ready. Let's rock baby */
    msg_Dbg( p_this, "Playing...");
    p_sys->p_control->Run();
642

Gildas Bazin's avatar
Gildas Bazin committed
643 644 645 646 647 648 649
    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++ )
650
    {
Gildas Bazin's avatar
Gildas Bazin committed
651 652
        dshow_stream_t *p_stream = p_sys->pp_streams[i];
        es_format_t fmt;
653

Gildas Bazin's avatar
Gildas Bazin committed
654 655 656
        if( p_stream->mt.majortype == MEDIATYPE_Video )
        {
            es_format_Init( &fmt, VIDEO_ES, p_stream->i_fourcc );
657

Gildas Bazin's avatar
Gildas Bazin committed
658 659 660
            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;
661

Gildas Bazin's avatar
Gildas Bazin committed
662 663 664 665 666
            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);
            }
667

Gildas Bazin's avatar
Gildas Bazin committed
668 669 670 671
            /* Setup rgb mask for RGB formats */
            if( p_stream->i_fourcc == VLC_FOURCC('R','V','2','4') )
            {
                /* This is in BGR format */
672 673 674
                fmt.video.i_bmask = 0x00ff0000;
                fmt.video.i_gmask = 0x0000ff00;
                fmt.video.i_rmask = 0x000000ff;
Gildas Bazin's avatar
Gildas Bazin committed
675
            }
676 677 678 679 680 681 682

            if( p_stream->header.video.AvgTimePerFrame )
            {
                fmt.video.i_frame_rate = 10000000;
                fmt.video.i_frame_rate_base =
                    p_stream->header.video.AvgTimePerFrame;
            }
Gildas Bazin's avatar
Gildas Bazin committed
683 684 685 686
        }
        else if( p_stream->mt.majortype == MEDIATYPE_Audio )
        {
            es_format_Init( &fmt, AUDIO_ES, p_stream->i_fourcc );
687

Gildas Bazin's avatar
Gildas Bazin committed
688 689 690 691 692 693 694 695 696 697
            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 );
698
    }
Gildas Bazin's avatar
Gildas Bazin committed
699

700
    return VLC_SUCCESS;
701 702
}

Gildas Bazin's avatar
Gildas Bazin committed
703 704 705 706
/*****************************************************************************
 * AccessOpen: open direct show device as an access module
 *****************************************************************************/
static int AccessOpen( vlc_object_t *p_this )
707
{
Gildas Bazin's avatar
Gildas Bazin committed
708 709
    access_t     *p_access = (access_t*)p_this;
    access_sys_t *p_sys;
710

Rémi Duraffort's avatar
Rémi Duraffort committed
711
    p_access->p_sys = p_sys = (access_sys_t*)calloc( 1, sizeof( access_sys_t ) );
Rémi Duraffort's avatar
Rémi Duraffort committed
712 713
    if( !p_sys )
        return VLC_ENOMEM;
714

715
    if( CommonOpen( p_this, p_sys, false ) != VLC_SUCCESS )
Gildas Bazin's avatar
Gildas Bazin committed
716 717 718 719
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
720

Gildas Bazin's avatar
Gildas Bazin committed
721
    dshow_stream_t *p_stream = p_sys->pp_streams[0];
722

Gildas Bazin's avatar
Gildas Bazin committed
723 724
    /* Check if we need to force demuxers */
    if( !p_access->psz_demux || !*p_access->psz_demux )
Gildas Bazin's avatar
 
Gildas Bazin committed
725
    {
Gildas Bazin's avatar
Gildas Bazin committed
726 727 728
        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') )
729
        {
Rafaël Carré's avatar
Rafaël Carré committed
730
            free( p_access->psz_demux );
Gildas Bazin's avatar
Gildas Bazin committed
731 732 733 734
            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
735 736
            free( p_access->psz_demux );
            p_access->psz_demux = strdup( "mpgv" );
737
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
738
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
739

Gildas Bazin's avatar
Gildas Bazin committed
740
    /* Setup Access */
741 742
    p_access->pf_read = NULL;
    p_access->pf_block = ReadCompressed;
Gildas Bazin's avatar
Gildas Bazin committed
743 744 745 746 747
    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;
748
    p_access->info.b_eof = false;
Gildas Bazin's avatar
Gildas Bazin committed
749 750 751 752 753 754 755 756 757
    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;
758
}
Gildas Bazin's avatar
 
Gildas Bazin committed
759

Gildas Bazin's avatar
Gildas Bazin committed
760 761 762 763
/*****************************************************************************
 * CommonClose: close device
 *****************************************************************************/
static void CommonClose( vlc_object_t *p_this, access_sys_t *p_sys )
764
{
Clément Stenac's avatar
Clément Stenac committed
765
    msg_Dbg( p_this, "releasing DirectShow");
766

Gildas Bazin's avatar
Gildas Bazin committed
767
    DeleteDirectShowGraph( p_sys );
768

Gildas Bazin's avatar
Gildas Bazin committed
769 770
    /* Uninitialize OLE/COM */
    CoUninitialize();
771

772 773
    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 );
Gildas Bazin's avatar
Gildas Bazin committed
774 775 776

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

Gildas Bazin's avatar
Gildas Bazin committed
778
    free( p_sys );
779 780
}

Gildas Bazin's avatar
Gildas Bazin committed
781 782 783 784
/*****************************************************************************
 * AccessClose: close device
 *****************************************************************************/
static void AccessClose( vlc_object_t *p_this )
785
{
Gildas Bazin's avatar
Gildas Bazin committed
786 787
    access_t     *p_access = (access_t *)p_this;
    access_sys_t *p_sys    = p_access->p_sys;
788

Gildas Bazin's avatar
Gildas Bazin committed
789 790
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
791

Gildas Bazin's avatar
Gildas Bazin committed
792 793
    CommonClose( p_this, p_sys );
}
794

Gildas Bazin's avatar
Gildas Bazin committed
795 796 797 798 799 800 801
/*****************************************************************************
 * 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;
802

Gildas Bazin's avatar
Gildas Bazin committed
803 804
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
805

Gildas Bazin's avatar
Gildas Bazin committed
806
    CommonClose( p_this, p_sys );
Gildas Bazin's avatar
 
Gildas Bazin committed
807 808 809 810 811
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
812 813
static bool ConnectFilters( vlc_object_t *p_this, access_sys_t *p_sys,
                            IBaseFilter *p_filter,
814
                            CaptureFilter *p_capture_filter )
Gildas Bazin's avatar
 
Gildas Bazin committed
815
{
816
    CapturePin *p_input_pin = p_capture_filter->CustomGetPin();
Gildas Bazin's avatar
 
Gildas Bazin committed
817

818
    AM_MEDIA_TYPE mediaType = p_input_pin->CustomGetMediaType();
Gildas Bazin's avatar
 
Gildas Bazin committed
819

820
    if( p_sys->p_capture_graph_builder2 )
Gildas Bazin's avatar
 
Gildas Bazin committed
821
    {
822
        if( FAILED(p_sys->p_capture_graph_builder2->
Gildas Bazin's avatar
Gildas Bazin committed
823 824
                RenderStream( &PIN_CATEGORY_CAPTURE, &mediaType.majortype,
                              p_filter, 0, (IBaseFilter *)p_capture_filter )) )
825 826 827
        {
            return false;
        }
828

829 830
        // Sort out all the possible video inputs
        // The class needs to be given the capture filters ANALOGVIDEO input pin
Rémi Duraffort's avatar
Rémi Duraffort committed
831
        IEnumPins *pins = NULL;
832 833
        if( ( mediaType.majortype == MEDIATYPE_Video ||
              mediaType.majortype == MEDIATYPE_Stream ) &&
834
            SUCCEEDED(p_filter->EnumPins(&pins)) )
Gildas Bazin's avatar
 
Gildas Bazin committed
835
        {
Rémi Duraffort's avatar
Rémi Duraffort committed
836
            IPin        *pP = NULL;
837 838 839
            ULONG        n;
            PIN_INFO     pinInfo;
            BOOL         Found = FALSE;
Rémi Duraffort's avatar
Rémi Duraffort committed
840
            IKsPropertySet *pKs = NULL;
841 842 843
            GUID guid;
            DWORD dw;