dshow.cpp 55.5 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
 *****************************************************************************
Rocky Bernstein's avatar
Rocky Bernstein committed
4
 * Copyright (C) 2002, 2003 VideoLAN
5
 * $Id$
Gildas Bazin's avatar
 
Gildas Bazin committed
6
 *
7
 * Author: Gildas Bazin <gbazin@videolan.org>
Gildas Bazin's avatar
 
Gildas Bazin committed
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
 *
 * 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
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include <vlc/vlc.h>
#include <vlc/input.h>
#include <vlc/vout.h>

Gildas Bazin's avatar
Gildas Bazin committed
35
#include "common.h"
Gildas Bazin's avatar
 
Gildas Bazin committed
36 37
#include "filter.h"

Gildas Bazin's avatar
 
Gildas Bazin committed
38 39 40
/*****************************************************************************
 * Access: local prototypes
 *****************************************************************************/
41 42
static int ReadCompressed( access_t *, byte_t *, int );
static int AccessControl ( access_t *, int, va_list );
Gildas Bazin's avatar
 
Gildas Bazin committed
43

Gildas Bazin's avatar
Gildas Bazin committed
44 45 46 47
static int Demux       ( demux_t * );
static int DemuxControl( demux_t *, int, va_list );

static int OpenDevice( vlc_object_t *, access_sys_t *, string, vlc_bool_t );
Gildas Bazin's avatar
 
Gildas Bazin committed
48 49
static IBaseFilter *FindCaptureDevice( vlc_object_t *, string *,
                                       list<string> *, vlc_bool_t );
50
static size_t EnumDeviceCaps( vlc_object_t *, IBaseFilter *,
51 52
                              int, int, int, int, int, int,
                              AM_MEDIA_TYPE *mt, size_t );
Gildas Bazin's avatar
Gildas Bazin committed
53 54
static bool ConnectFilters( vlc_object_t *, access_sys_t *,
                            IBaseFilter *, CaptureFilter * );
Gildas Bazin's avatar
 
Gildas Bazin committed
55 56
static int FindDevicesCallback( vlc_object_t *, char const *,
                                vlc_value_t, vlc_value_t, void * );
57 58
static int ConfigDevicesCallback( vlc_object_t *, char const *,
                                  vlc_value_t, vlc_value_t, void * );
Gildas Bazin's avatar
 
Gildas Bazin committed
59

60 61 62
static void ShowPropertyPage( IUnknown * );
static void ShowDeviceProperties( vlc_object_t *, ICaptureGraphBuilder2 *, 
                                  IBaseFilter *, vlc_bool_t b_audio );
Gildas Bazin's avatar
 
Gildas Bazin committed
63

Gildas Bazin's avatar
 
Gildas Bazin committed
64
/*****************************************************************************
65
 * Module descriptor
Gildas Bazin's avatar
 
Gildas Bazin committed
66
 *****************************************************************************/
67 68 69 70 71
static char *ppsz_vdev[] = { "", "none" };
static char *ppsz_vdev_text[] = { N_("Default"), N_("None") };
static char *ppsz_adev[] = { "", "none" };
static char *ppsz_adev_text[] = { N_("Default"), N_("None") };

Gildas Bazin's avatar
 
Gildas Bazin committed
72 73
#define CACHING_TEXT N_("Caching value in ms")
#define CACHING_LONGTEXT N_( \
74
    "Allows you to modify the default caching value for DirectShow streams. " \
Rocky Bernstein's avatar
Rocky Bernstein committed
75
    "This value should be set in milliseconds units." )
76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
#define VDEV_TEXT N_("Video device name")
#define VDEV_LONGTEXT N_( \
    "You can specify the name of the video device that will be used by the " \
    "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_( \
    "You can specify the name of the audio device that will be used by the " \
    "DirectShow plugin. If you don't specify anything, the default device " \
    "will be used.")
#define SIZE_TEXT N_("Video size")
#define SIZE_LONGTEXT N_( \
    "You can specify the size of the video that will be displayed by the " \
    "DirectShow plugin. If you don't specify anything the default size for " \
    "your device will be used.")
#define CHROMA_TEXT N_("Video input chroma format")
#define CHROMA_LONGTEXT N_( \
    "Force the DirectShow video input to use a specific chroma format " \
94 95
    "(eg. I420 (default), RV24, etc.)")
#define CONFIG_TEXT N_("Device properties")
Gildas Bazin's avatar
 
Gildas Bazin committed
96
#define CONFIG_LONGTEXT N_( \
97 98
    "Show the properties dialog of the selected device before starting the " \
    "stream.")
Gildas Bazin's avatar
 
Gildas Bazin committed
99

Gildas Bazin's avatar
Gildas Bazin committed
100 101 102
static int  CommonOpen ( vlc_object_t *, access_sys_t *, vlc_bool_t );
static void CommonClose( vlc_object_t *, access_sys_t * );

103 104 105 106 107 108
static int  AccessOpen ( vlc_object_t * );
static void AccessClose( vlc_object_t * );

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

Gildas Bazin's avatar
 
Gildas Bazin committed
109
vlc_module_begin();
110
    set_shortname( _("DirectShow") );
Gildas Bazin's avatar
 
Gildas Bazin committed
111
    set_description( _("DirectShow input") );
112
    add_integer( "dshow-caching", (mtime_t)(0.2*CLOCK_FREQ) / 1000, NULL,
Gildas Bazin's avatar
 
Gildas Bazin committed
113
                 CACHING_TEXT, CACHING_LONGTEXT, VLC_TRUE );
114

115
    add_string( "dshow-vdev", NULL, NULL, VDEV_TEXT, VDEV_LONGTEXT, VLC_FALSE);
116
        change_string_list( ppsz_vdev, ppsz_vdev_text, FindDevicesCallback );
117 118
        change_action_add( FindDevicesCallback, N_("Refresh list") );
        change_action_add( ConfigDevicesCallback, N_("Configure") );
119

120
    add_string( "dshow-adev", NULL, NULL, ADEV_TEXT, ADEV_LONGTEXT, VLC_FALSE);
121
        change_string_list( ppsz_adev, ppsz_adev_text, FindDevicesCallback );
122 123
        change_action_add( FindDevicesCallback, N_("Refresh list") );
        change_action_add( ConfigDevicesCallback, N_("Configure") );
124

125
    add_string( "dshow-size", NULL, NULL, SIZE_TEXT, SIZE_LONGTEXT, VLC_FALSE);
126

127 128
    add_string( "dshow-chroma", NULL, NULL, CHROMA_TEXT, CHROMA_LONGTEXT,
                VLC_TRUE );
Gildas Bazin's avatar
 
Gildas Bazin committed
129 130

    add_bool( "dshow-config", VLC_FALSE, NULL, CONFIG_TEXT, CONFIG_LONGTEXT,
131
              VLC_FALSE );
Gildas Bazin's avatar
 
Gildas Bazin committed
132

Gildas Bazin's avatar
 
Gildas Bazin committed
133
    add_shortcut( "dshow" );
Gildas Bazin's avatar
Gildas Bazin committed
134 135
    set_capability( "access_demux", 0 );
    set_callbacks( DemuxOpen, DemuxClose );
Gildas Bazin's avatar
 
Gildas Bazin committed
136 137

    add_submodule();
Gildas Bazin's avatar
Gildas Bazin committed
138
    set_description( _("DirectShow input") );
Gildas Bazin's avatar
 
Gildas Bazin committed
139
    add_shortcut( "dshow" );
Gildas Bazin's avatar
Gildas Bazin committed
140 141
    set_capability( "access2", 0 );
    set_callbacks( AccessOpen, AccessClose );
Gildas Bazin's avatar
 
Gildas Bazin committed
142 143 144

vlc_module_end();

Gildas Bazin's avatar
Gildas Bazin committed
145
/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
146
 * DirectShow elementary stream descriptor
Gildas Bazin's avatar
Gildas Bazin committed
147
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
148 149 150 151 152 153 154 155 156 157 158 159 160 161
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
162 163 164
    int             i_fourcc;
    es_out_id_t     *p_es;

165
    vlc_bool_t      b_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
166 167 168

} dshow_stream_t;

Gildas Bazin's avatar
Gildas Bazin committed
169
/*****************************************************************************
170
 * DirectShow utility functions
Gildas Bazin's avatar
Gildas Bazin committed
171
 *****************************************************************************/
172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225
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;
    }
}

226

Gildas Bazin's avatar
 
Gildas Bazin committed
227
/*****************************************************************************
Gildas Bazin's avatar
Gildas Bazin committed
228
 * CommonOpen: open direct show device
Gildas Bazin's avatar
 
Gildas Bazin committed
229
 *****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
230 231
static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
                       vlc_bool_t b_access_demux )
Gildas Bazin's avatar
 
Gildas Bazin committed
232
{
233
    vlc_value_t  val;
Gildas Bazin's avatar
Gildas Bazin committed
234
    int i;
Gildas Bazin's avatar
 
Gildas Bazin committed
235

236
    /* Get/parse options and open device(s) */
Gildas Bazin's avatar
 
Gildas Bazin committed
237
    string vdevname, adevname;
238
    int i_width = 0, i_height = 0, i_chroma = 0;
Gildas Bazin's avatar
Gildas Bazin committed
239
    vlc_bool_t b_audio = VLC_TRUE;
Gildas Bazin's avatar
 
Gildas Bazin committed
240

Gildas Bazin's avatar
Gildas Bazin committed
241
    var_Create( p_this, "dshow-config", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
242

Gildas Bazin's avatar
Gildas Bazin committed
243 244
    var_Create( p_this, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-vdev", &val );
245 246
    if( val.psz_string ) vdevname = string( val.psz_string );
    if( val.psz_string ) free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
247

Gildas Bazin's avatar
Gildas Bazin committed
248 249
    var_Create( p_this, "dshow-adev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-adev", &val );
250 251 252
    if( val.psz_string ) adevname = string( val.psz_string );
    if( val.psz_string ) free( val.psz_string );

Gildas Bazin's avatar
Gildas Bazin committed
253 254 255 256 257 258 259 260
    static struct {char *psz_size; int  i_width; int  i_height;} size_table[] =
    { { "subqcif", 128, 96 }, { "qsif", 160, 120 }, { "qcif", 176, 144 },
      { "sif", 320, 240 }, { "cif", 352, 288 }, { "cif", 640, 480 },
      { 0, 0, 0 },
    };

    var_Create( p_this, "dshow-size", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-size", &val );
261
    if( val.psz_string && *val.psz_string )
Gildas Bazin's avatar
 
Gildas Bazin committed
262
    {
Gildas Bazin's avatar
Gildas Bazin committed
263
        for( i = 0; size_table[i].psz_size; i++ )
264
        {
Gildas Bazin's avatar
Gildas Bazin committed
265 266 267 268 269 270
            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;
            }
271
        }
Gildas Bazin's avatar
Gildas Bazin committed
272
        if( !size_table[i].psz_size ) /* Try to parse "WidthxHeight" */
273 274 275 276
        {
            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
277
            {
278
                i_height = strtol( psz_parser + 1, &psz_parser, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
279
            }
Gildas Bazin's avatar
Gildas Bazin committed
280
            msg_Dbg( p_this, "Width x Height %dx%d", i_width, i_height );
Gildas Bazin's avatar
 
Gildas Bazin committed
281 282
        }
    }
283
    if( val.psz_string ) free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
284

Gildas Bazin's avatar
Gildas Bazin committed
285 286
    var_Create( p_this, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_this, "dshow-chroma", &val );
287 288 289 290 291 292
    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] );
    }
    if( val.psz_string ) free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
293

Gildas Bazin's avatar
Gildas Bazin committed
294
    var_Create( p_this, "dshow-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
Gildas Bazin's avatar
 
Gildas Bazin committed
295 296

    /* Initialize OLE/COM */
297
    CoInitialize( 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
298

Gildas Bazin's avatar
 
Gildas Bazin committed
299 300 301
    /* Initialize some data */
    p_sys->i_streams = 0;
    p_sys->pp_streams = (dshow_stream_t **)malloc( 1 );
Gildas Bazin's avatar
 
Gildas Bazin committed
302 303 304
    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
305

306 307 308
    p_sys->p_graph = NULL;
    p_sys->p_capture_graph_builder2 = NULL;
    p_sys->p_control = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
309

310 311
    /* Build directshow graph */
    CreateDirectShowGraph( p_sys );
Gildas Bazin's avatar
 
Gildas Bazin committed
312

Gildas Bazin's avatar
Gildas Bazin committed
313
    if( OpenDevice( p_this, p_sys, vdevname, 0 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
314
    {
Gildas Bazin's avatar
Gildas Bazin committed
315
        msg_Err( p_this, "can't open video");
Gildas Bazin's avatar
 
Gildas Bazin committed
316
    }
Gildas Bazin's avatar
Gildas Bazin committed
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334
    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') )
            {
                b_audio = VLC_FALSE;
            }
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
335

Gildas Bazin's avatar
Gildas Bazin committed
336 337 338 339
        if( p_stream->mt.majortype == MEDIATYPE_Stream ) b_audio = VLC_FALSE;
    }

    if( b_audio && OpenDevice( p_this, p_sys, adevname, 1 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
340
    {
Gildas Bazin's avatar
Gildas Bazin committed
341
        msg_Err( p_this, "can't open audio");
Gildas Bazin's avatar
 
Gildas Bazin committed
342
    }
Gildas Bazin's avatar
Gildas Bazin committed
343

344
    for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
Gildas Bazin's avatar
 
Gildas Bazin committed
345
    {
346 347 348 349 350
        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
351

352 353
        if( SUCCEEDED(pXbar->Route(VideoOutputIndex, VideoInputIndex)) )
        {
Gildas Bazin's avatar
Gildas Bazin committed
354
            msg_Dbg( p_this, "Crossbar at depth %d, Routed video "
355
                     "ouput %ld to video input %ld", i, VideoOutputIndex,
356
                     VideoInputIndex );
Gildas Bazin's avatar
 
Gildas Bazin committed
357

358 359 360 361 362
            if( AudioOutputIndex != -1 && AudioInputIndex != -1 )
            {
                if( SUCCEEDED( pXbar->Route(AudioOutputIndex,
                                            AudioInputIndex)) )
                {
Gildas Bazin's avatar
Gildas Bazin committed
363
                    msg_Dbg(p_this, "Crossbar at depth %d, Routed audio "
364
                            "ouput %ld to audio input %ld", i,
365 366 367 368 369 370
                            AudioOutputIndex, AudioInputIndex );
                }
            }
        }
    }

371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389
    /*
    ** Show properties pages from other filters in graph
    */
    var_Get( p_this, "dshow-config", &val );
    if( val.i_int )
    {
        for( i = p_sys->i_crossbar_route_depth-1; i >= 0 ; --i )
        {
            IAMCrossbar *pXbar = p_sys->crossbar_routes[i].pXbar;
            IBaseFilter *p_XF;
            HRESULT hr = pXbar->QueryInterface( IID_IBaseFilter, (void **)&p_XF );
            if( SUCCEEDED(hr) )
            {
                ShowPropertyPage( p_XF );
                p_XF->Release();
            }
        }        
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
390 391
    /* Initialize some data */
    p_sys->i_current_stream = 0;
Gildas Bazin's avatar
Gildas Bazin committed
392 393
    vlc_mutex_init( p_this, &p_sys->lock );
    vlc_cond_init( p_this, &p_sys->wait );
394

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

Gildas Bazin's avatar
 
Gildas Bazin committed
397 398 399 400
    return VLC_SUCCESS;
}

/*****************************************************************************
Gildas Bazin's avatar
Gildas Bazin committed
401
 * DemuxOpen: open direct show device as an access_demux module
Gildas Bazin's avatar
 
Gildas Bazin committed
402
 *****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
403
static int DemuxOpen( vlc_object_t *p_this )
Gildas Bazin's avatar
 
Gildas Bazin committed
404
{
Gildas Bazin's avatar
Gildas Bazin committed
405 406 407
    demux_t      *p_demux = (demux_t *)p_this;
    access_sys_t *p_sys;
    int i;
Gildas Bazin's avatar
 
Gildas Bazin committed
408

Gildas Bazin's avatar
Gildas Bazin committed
409 410 411
    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;
Gildas Bazin's avatar
 
Gildas Bazin committed
412

Gildas Bazin's avatar
Gildas Bazin committed
413 414 415 416 417
    if( CommonOpen( p_this, p_sys, VLC_TRUE ) != VLC_SUCCESS )
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
418

Gildas Bazin's avatar
Gildas Bazin committed
419 420 421
    /* Everything is ready. Let's rock baby */
    msg_Dbg( p_this, "Playing...");
    p_sys->p_control->Run();
422

Gildas Bazin's avatar
Gildas Bazin committed
423 424 425 426 427 428 429
    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++ )
430
    {
Gildas Bazin's avatar
Gildas Bazin committed
431 432
        dshow_stream_t *p_stream = p_sys->pp_streams[i];
        es_format_t fmt;
433

Gildas Bazin's avatar
Gildas Bazin committed
434 435 436
        if( p_stream->mt.majortype == MEDIATYPE_Video )
        {
            es_format_Init( &fmt, VIDEO_ES, p_stream->i_fourcc );
437

Gildas Bazin's avatar
Gildas Bazin committed
438 439 440
            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;
441

Gildas Bazin's avatar
Gildas Bazin committed
442 443 444 445 446
            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);
            }
447

Gildas Bazin's avatar
Gildas Bazin committed
448 449 450 451
            /* Setup rgb mask for RGB formats */
            if( p_stream->i_fourcc == VLC_FOURCC('R','V','2','4') )
            {
                /* This is in BGR format */
452 453 454
                fmt.video.i_bmask = 0x00ff0000;
                fmt.video.i_gmask = 0x0000ff00;
                fmt.video.i_rmask = 0x000000ff;
Gildas Bazin's avatar
Gildas Bazin committed
455 456 457 458 459
            }
        }
        else if( p_stream->mt.majortype == MEDIATYPE_Audio )
        {
            es_format_Init( &fmt, AUDIO_ES, p_stream->i_fourcc );
460

Gildas Bazin's avatar
Gildas Bazin committed
461 462 463 464 465 466 467 468 469 470
            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 );
471
    }
Gildas Bazin's avatar
Gildas Bazin committed
472

473
    return VLC_SUCCESS;
474 475
}

Gildas Bazin's avatar
Gildas Bazin committed
476 477 478 479
/*****************************************************************************
 * AccessOpen: open direct show device as an access module
 *****************************************************************************/
static int AccessOpen( vlc_object_t *p_this )
480
{
Gildas Bazin's avatar
Gildas Bazin committed
481 482
    access_t     *p_access = (access_t*)p_this;
    access_sys_t *p_sys;
483

Gildas Bazin's avatar
Gildas Bazin committed
484 485
    p_access->p_sys = p_sys = (access_sys_t *)malloc( sizeof( access_sys_t ) );
    memset( p_sys, 0, sizeof( access_sys_t ) );
486

Gildas Bazin's avatar
Gildas Bazin committed
487 488 489 490 491
    if( CommonOpen( p_this, p_sys, VLC_TRUE ) != VLC_SUCCESS )
    {
        CommonClose( p_this, p_sys );
        return VLC_EGENERIC;
    }
492

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

Gildas Bazin's avatar
Gildas Bazin committed
495 496
    /* Check if we need to force demuxers */
    if( !p_access->psz_demux || !*p_access->psz_demux )
Gildas Bazin's avatar
 
Gildas Bazin committed
497
    {
Gildas Bazin's avatar
Gildas Bazin committed
498 499 500
        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') )
501
        {
Gildas Bazin's avatar
Gildas Bazin committed
502 503 504 505 506
            p_access->psz_demux = strdup( "rawdv" );
        }
        else if( p_stream->i_fourcc == VLC_FOURCC('m','p','2','v') )
        {
            p_access->psz_demux = "mpgv";
507
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
508
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
509

Gildas Bazin's avatar
Gildas Bazin committed
510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527
    /* Setup Access */
    p_access->pf_read = ReadCompressed;
    p_access->pf_block = NULL;
    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;
    p_access->info.b_eof = VLC_FALSE;
    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;
528
}
Gildas Bazin's avatar
 
Gildas Bazin committed
529

Gildas Bazin's avatar
Gildas Bazin committed
530 531 532 533
/*****************************************************************************
 * CommonClose: close device
 *****************************************************************************/
static void CommonClose( vlc_object_t *p_this, access_sys_t *p_sys )
534
{
Gildas Bazin's avatar
Gildas Bazin committed
535
    msg_Dbg( p_this, "Releasing DirectShow");
536

Gildas Bazin's avatar
Gildas Bazin committed
537
    DeleteDirectShowGraph( p_sys );
538

Gildas Bazin's avatar
Gildas Bazin committed
539 540
    /* Uninitialize OLE/COM */
    CoUninitialize();
541

Gildas Bazin's avatar
Gildas Bazin committed
542 543
    /* Remove filters from graph */
    for( int i = 0; i < p_sys->i_streams; i++ )
544
    {
Gildas Bazin's avatar
Gildas Bazin committed
545
        delete p_sys->pp_streams[i];
546
    }
Gildas Bazin's avatar
Gildas Bazin committed
547 548 549 550
    free( p_sys->pp_streams );

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

Gildas Bazin's avatar
Gildas Bazin committed
552
    free( p_sys );
553 554
}

Gildas Bazin's avatar
Gildas Bazin committed
555 556 557 558
/*****************************************************************************
 * AccessClose: close device
 *****************************************************************************/
static void AccessClose( vlc_object_t *p_this )
559
{
Gildas Bazin's avatar
Gildas Bazin committed
560 561
    access_t     *p_access = (access_t *)p_this;
    access_sys_t *p_sys    = p_access->p_sys;
562

Gildas Bazin's avatar
Gildas Bazin committed
563 564
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
565

Gildas Bazin's avatar
Gildas Bazin committed
566 567
    CommonClose( p_this, p_sys );
}
568

Gildas Bazin's avatar
Gildas Bazin committed
569 570 571 572 573 574 575
/*****************************************************************************
 * 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;
576

Gildas Bazin's avatar
Gildas Bazin committed
577 578
    /* Stop capturing stuff */
    p_sys->p_control->Stop();
579

Gildas Bazin's avatar
Gildas Bazin committed
580
    CommonClose( p_this, p_sys );
Gildas Bazin's avatar
 
Gildas Bazin committed
581 582 583 584 585
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
Gildas Bazin's avatar
Gildas Bazin committed
586 587
static bool ConnectFilters( vlc_object_t *p_this, access_sys_t *p_sys,
                            IBaseFilter *p_filter,
588
                            CaptureFilter *p_capture_filter )
Gildas Bazin's avatar
 
Gildas Bazin committed
589
{
590
    CapturePin *p_input_pin = p_capture_filter->CustomGetPin();
Gildas Bazin's avatar
 
Gildas Bazin committed
591

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

594
    if( p_sys->p_capture_graph_builder2 )
Gildas Bazin's avatar
 
Gildas Bazin committed
595
    {
596
        if( FAILED(p_sys->p_capture_graph_builder2->
Gildas Bazin's avatar
Gildas Bazin committed
597 598
                RenderStream( &PIN_CATEGORY_CAPTURE, &mediaType.majortype,
                              p_filter, 0, (IBaseFilter *)p_capture_filter )) )
599 600 601
        {
            return false;
        }
602

603 604 605 606 607
        // Sort out all the possible video inputs
        // The class needs to be given the capture filters ANALOGVIDEO input pin
        IEnumPins *pins = 0;
        if( mediaType.majortype == MEDIATYPE_Video &&
            SUCCEEDED(p_filter->EnumPins(&pins)) )
Gildas Bazin's avatar
 
Gildas Bazin committed
608
        {
609 610 611 612 613 614 615 616
            IPin        *pP = 0;
            ULONG        n;
            PIN_INFO     pinInfo;
            BOOL         Found = FALSE;
            IKsPropertySet *pKs=0;
            GUID guid;
            DWORD dw;

Gildas Bazin's avatar
Gildas Bazin committed
617
            while( !Found && ( S_OK == pins->Next(1, &pP, &n) ) )
618
            {
Gildas Bazin's avatar
Gildas Bazin committed
619
                if( S_OK == pP->QueryPinInfo(&pinInfo) )
620
                {
Gildas Bazin's avatar
Gildas Bazin committed
621 622 623 624
                    // is this pin an ANALOGVIDEOIN input pin?
                    if( pinInfo.dir == PINDIR_INPUT &&
                        pP->QueryInterface( IID_IKsPropertySet,
                                            (void **)&pKs ) == S_OK )
625
                    {
Gildas Bazin's avatar
Gildas Bazin committed
626 627 628
                        if( pKs->Get( AMPROPSETID_Pin,
                                      AMPROPERTY_PIN_CATEGORY, NULL, 0,
                                      &guid, sizeof(GUID), &dw ) == S_OK )
629
                        {
Gildas Bazin's avatar
Gildas Bazin committed
630
                            if( guid == PIN_CATEGORY_ANALOGVIDEOIN )
631
                            {
Gildas Bazin's avatar
Gildas Bazin committed
632
                                // recursively search crossbar routes
633
                                FindCrossbarRoutes( p_this, p_sys, pP, 0 );
Gildas Bazin's avatar
Gildas Bazin committed
634 635
                                // found it
                                Found = TRUE;
636 637
                            }
                        }
Gildas Bazin's avatar
Gildas Bazin committed
638
                        pKs->Release();
639 640 641 642 643 644
                    }
                    pinInfo.pFilter->Release();
                }
                pP->Release();
            }
            pins->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
645
        }
646
        return true;
Gildas Bazin's avatar
 
Gildas Bazin committed
647
    }
648 649 650 651
    else
    {
        IEnumPins *p_enumpins;
        IPin *p_pin;
Gildas Bazin's avatar
 
Gildas Bazin committed
652

653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
        if( S_OK != p_filter->EnumPins( &p_enumpins ) ) return false;

        while( S_OK == p_enumpins->Next( 1, &p_pin, NULL ) )
        {
            PIN_DIRECTION pin_dir;
            p_pin->QueryDirection( &pin_dir );

            if( pin_dir == PINDIR_OUTPUT &&
                p_sys->p_graph->ConnectDirect( p_pin, (IPin *)p_input_pin,
                                               0 ) == S_OK )
            {
                p_pin->Release();
                p_enumpins->Release();
                return true;
            }
            p_pin->Release();
        }

        p_enumpins->Release();
        return false;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
674 675
}

676
/*
Gildas Bazin's avatar
Gildas Bazin committed
677 678 679
 * get fourcc priority from arbritary preference, the higher the better
 */
static int GetFourCCPriority( int i_fourcc )
680 681 682
{
    switch( i_fourcc )
    {
Gildas Bazin's avatar
Gildas Bazin committed
683 684 685 686 687 688 689 690 691 692 693 694
    case VLC_FOURCC('I','4','2','0'):
    case VLC_FOURCC('f','l','3','2'):
        return 9;
    case VLC_FOURCC('Y','V','1','2'):
    case VLC_FOURCC('a','r','a','w'):
        return 8;
    case VLC_FOURCC('R','V','2','4'):
        return 7;
    case VLC_FOURCC('Y','U','Y','2'):
    case VLC_FOURCC('R','V','3','2'):
    case VLC_FOURCC('R','G','B','A'):
        return 6;
695
    }
Gildas Bazin's avatar
Gildas Bazin committed
696

697 698 699 700 701
    return 0;
}

#define MAX_MEDIA_TYPES 32

Gildas Bazin's avatar
Gildas Bazin committed
702 703
static int OpenDevice( vlc_object_t *p_this, access_sys_t *p_sys,
                       string devicename, vlc_bool_t b_audio )
Gildas Bazin's avatar
 
Gildas Bazin committed
704
{
705 706 707
    /* See if device is already opened */
    for( int i = 0; i < p_sys->i_streams; i++ )
    {
Gildas Bazin's avatar
Gildas Bazin committed
708 709
        if( devicename.size() &&
            p_sys->pp_streams[i]->devicename == devicename )
710 711 712 713 714 715
        {
            /* Already opened */
            return VLC_SUCCESS;
        }
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
716 717
    list<string> list_devices;

718
    /* Enumerate devices and display their names */
Gildas Bazin's avatar
Gildas Bazin committed
719
    FindCaptureDevice( p_this, NULL, &list_devices, b_audio );
Gildas Bazin's avatar
 
Gildas Bazin committed
720

721 722 723
    if( !list_devices.size() )
        return VLC_EGENERIC;

Gildas Bazin's avatar
 
Gildas Bazin committed
724 725
    list<string>::iterator iter;
    for( iter = list_devices.begin(); iter != list_devices.end(); iter++ )
Gildas Bazin's avatar
Gildas Bazin committed
726
        msg_Dbg( p_this, "found device: %s", iter->c_str() );
Gildas Bazin's avatar
 
Gildas Bazin committed
727 728 729 730 731 732 733 734

    /* If no device name was specified, pick the 1st one */
    if( devicename.size() == 0 )
    {
        devicename = *list_devices.begin();
    }

    // Use the system device enumerator and class enumerator to find
Gildas Bazin's avatar
 
Gildas Bazin committed
735 736
    // a capture/preview device, such as a desktop USB video camera.
    IBaseFilter *p_device_filter =
Gildas Bazin's avatar
Gildas Bazin committed
737
        FindCaptureDevice( p_this, &devicename, 0, b_audio );
Gildas Bazin's avatar
 
Gildas Bazin committed
738
    if( p_device_filter )
Gildas Bazin's avatar
Gildas Bazin committed
739
        msg_Dbg( p_this, "using device: %s", devicename.c_str() );
Gildas Bazin's avatar
 
Gildas Bazin committed
740 741
    else
    {
Gildas Bazin's avatar
Gildas Bazin committed
742
        msg_Err( p_this, "can't use device: %s, unsupported device type",
743
                 devicename.c_str() );
Gildas Bazin's avatar
 
Gildas Bazin committed
744 745
        return VLC_EGENERIC;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
746

747 748
    size_t mt_count = 0;
    AM_MEDIA_TYPE *mt = NULL;
Gildas Bazin's avatar
Gildas Bazin committed
749

750
    if( !b_audio )
Gildas Bazin's avatar
Gildas Bazin committed
751
    {
752
        // insert prefered video media type
753
        AM_MEDIA_TYPE mtr;
Gildas Bazin's avatar
Gildas Bazin committed
754 755 756 757 758 759 760 761 762 763 764
        VIDEOINFOHEADER vh;

        mtr.majortype            = MEDIATYPE_Video;
        mtr.subtype              = MEDIASUBTYPE_I420;
        mtr.bFixedSizeSamples    = TRUE;
        mtr.bTemporalCompression = FALSE;
        mtr.lSampleSize          = 0;
        mtr.pUnk                 = NULL;
        mtr.formattype           = FORMAT_VideoInfo;
        mtr.cbFormat             = sizeof(vh);
        mtr.pbFormat             = (BYTE *)&vh;
765 766 767

        memset(&vh, 0, sizeof(vh));

Gildas Bazin's avatar
Gildas Bazin committed
768 769 770 771 772 773 774
        vh.bmiHeader.biSize   = sizeof(vh.bmiHeader);
        vh.bmiHeader.biWidth  = p_sys->i_width > 0 ? p_sys->i_width: 320;
        vh.bmiHeader.biHeight = p_sys->i_height > 0 ? p_sys->i_height : 240;
        vh.bmiHeader.biPlanes      = 1;
        vh.bmiHeader.biBitCount    = 12;
        vh.bmiHeader.biCompression = VLC_FOURCC('I','4','2','0');
        vh.bmiHeader.biSizeImage   = p_sys->i_width * 12 * p_sys->i_height / 8;
775

Gildas Bazin's avatar
Gildas Bazin committed
776 777 778
        mt_count = 1;
        mt = (AM_MEDIA_TYPE *)malloc( sizeof(AM_MEDIA_TYPE)*mt_count );
        CopyMediaType(mt, &mtr);
779
    }
Gildas Bazin's avatar
Gildas Bazin committed
780 781
    else
    {
782
        // insert prefered audio media type
783
        AM_MEDIA_TYPE mtr;
Gildas Bazin's avatar
Gildas Bazin committed
784 785 786 787 788 789 790 791 792 793 794
        WAVEFORMATEX wf;

        mtr.majortype            = MEDIATYPE_Audio;
        mtr.subtype              = MEDIASUBTYPE_PCM;
        mtr.bFixedSizeSamples    = TRUE;
        mtr.bTemporalCompression = FALSE;
        mtr.lSampleSize          = 0;
        mtr.pUnk                 = NULL;
        mtr.formattype           = FORMAT_WaveFormatEx;
        mtr.cbFormat             = sizeof(wf);
        mtr.pbFormat             = (BYTE *)&wf;
795 796 797

        memset(&wf, 0, sizeof(wf));

Gildas Bazin's avatar
Gildas Bazin committed
798 799 800 801 802 803 804
        wf.wFormatTag = WAVE_FORMAT_PCM;
        wf.nChannels = 2;
        wf.nSamplesPerSec = 44100;
        wf.wBitsPerSample = 16;
        wf.nBlockAlign = wf.nSamplesPerSec * wf.wBitsPerSample / 8;
        wf.nAvgBytesPerSec = wf.nSamplesPerSec * wf.nBlockAlign;
        wf.cbSize = 0;
805

Gildas Bazin's avatar
Gildas Bazin committed
806 807 808
        mt_count = 1;
        mt = (AM_MEDIA_TYPE *)malloc( sizeof(AM_MEDIA_TYPE)*mt_count );
        CopyMediaType(mt, &mtr);
809
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
810

811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851
    // retreive acceptable media types supported by device
    
    AM_MEDIA_TYPE media_types[MAX_MEDIA_TYPES];
    size_t media_count = EnumDeviceCaps( p_this, p_device_filter, p_sys->i_chroma,
                                      p_sys->i_width, p_sys->i_height,
                                      0, 0, 0, media_types, MAX_MEDIA_TYPES );

    if( media_count > 0 )
    {
        mt = (AM_MEDIA_TYPE *)realloc( mt, sizeof(AM_MEDIA_TYPE)*(mt_count+media_count) );

        // Order and copy returned media types according to arbitrary
        // fourcc priority
        for( size_t c = 0; c < media_count; c++ )
        {
            int slot_priority =
                GetFourCCPriority(GetFourCCFromMediaType(media_types[c]));
            size_t slot_copy = c;
            for( size_t d = c+1; d < media_count; d++ )
            {
                int priority =
                    GetFourCCPriority(GetFourCCFromMediaType(media_types[d]));
                if( priority > slot_priority )
                {
                    slot_priority = priority;
                    slot_copy = d;
                }
            }
            if( slot_copy != c )
            {
                mt[c+mt_count] = media_types[slot_copy];
                media_types[slot_copy] = media_types[c];
            }
            else
            {
                mt[c+mt_count] = media_types[c];
            }
        }
        mt_count += media_count;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
852
    /* Create and add our capture filter */
853
    CaptureFilter *p_capture_filter =
Gildas Bazin's avatar
Gildas Bazin committed
854
        new CaptureFilter( p_this, p_sys, mt, mt_count );
Gildas Bazin's avatar
 
Gildas Bazin committed
855
    p_sys->p_graph->AddFilter( p_capture_filter, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
856 857 858

    /* Add the device filter to the graph (seems necessary with VfW before
     * accessing pin attributes). */
Gildas Bazin's avatar
 
Gildas Bazin committed
859
    p_sys->p_graph->AddFilter( p_device_filter, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
860

Gildas Bazin's avatar
 
Gildas Bazin committed
861
    /* Attempt to connect one of this device's capture output pins */
Gildas Bazin's avatar
Gildas Bazin committed
862 863
    msg_Dbg( p_this, "connecting filters" );
    if( ConnectFilters( p_this, p_sys, p_device_filter, p_capture_filter ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
864 865
    {
        /* Success */
Gildas Bazin's avatar
Gildas Bazin committed
866
        msg_Dbg( p_this, "filters connected successfully !" );
867

Gildas Bazin's avatar
 
Gildas Bazin committed
868
        dshow_stream_t dshow_stream;
869
        dshow_stream.b_pts = VLC_FALSE;
Gildas Bazin's avatar
Gildas Bazin committed
870
        dshow_stream.p_es = 0;
871 872 873
        dshow_stream.mt =
            p_capture_filter->CustomGetPin()->CustomGetMediaType();

874
        /* Show Device properties. Done here so the VLC stream is setup with the
875 876
         * proper parameters. */
        vlc_value_t val;
Gildas Bazin's avatar
Gildas Bazin committed
877
        var_Get( p_this, "dshow-config", &val );
878 879
        if( val.i_int )
        {
880
            ShowDeviceProperties( p_this, p_sys->p_capture_graph_builder2, p_device_filter, b_audio );
881
        }
882
        
Gildas Bazin's avatar
 
Gildas Bazin committed
883 884
        dshow_stream.mt =
            p_capture_filter->CustomGetPin()->CustomGetMediaType();
Gildas Bazin's avatar
 
Gildas Bazin committed
885

Gildas Bazin's avatar
Gildas Bazin committed
886 887
        dshow_stream.i_fourcc = GetFourCCFromMediaType( dshow_stream.mt );
        if( dshow_stream.i_fourcc )
888 889 890 891 892
        {
            if( dshow_stream.mt.majortype == MEDIATYPE_Video )
            {
                dshow_stream.header.video =
                    *(VIDEOINFOHEADER *)dshow_stream.mt.pbFormat;
Gildas Bazin's avatar
Gildas Bazin committed
893 894 895
                msg_Dbg( p_this, "MEDIATYPE_Video" );
                msg_Dbg( p_this, "selected video pin accepts format: %4.4s",
                         (char *)&dshow_stream.i_fourcc);
896 897 898 899 900
            }
            else if( dshow_stream.mt.majortype == MEDIATYPE_Audio )
            {
                dshow_stream.header.audio =
                    *(WAVEFORMATEX *)dshow_stream.mt.pbFormat;
Gildas Bazin's avatar
Gildas Bazin committed
901 902 903
                msg_Dbg( p_this, "MEDIATYPE_Audio" );
                msg_Dbg( p_this, "selected audio pin accepts format: %4.4s",
                         (char *)&dshow_stream.i_fourcc);
904 905 906
            }
            else if( dshow_stream.mt.majortype == MEDIATYPE_Stream )
            {
Gildas Bazin's avatar
Gildas Bazin committed
907 908
                msg_Dbg( p_this, "MEDIATYPE_Stream" );
                msg_Dbg( p_this, "selected stream pin accepts format: %4.4s",
909 910 911 912
                         (char *)&dshow_stream.i_fourcc);
            }
            else
            {
Gildas Bazin's avatar
Gildas Bazin committed
913
                msg_Dbg( p_this, "unknown stream majortype" );
914 915
                goto fail;
            }
Gildas Bazin's avatar
 
Gildas Bazin committed
916

917 918 919
            /* Add directshow elementary stream to our list */
            dshow_stream.p_device_filter = p_device_filter;
            dshow_stream.p_capture_filter = p_capture_filter;
Gildas Bazin's avatar
 
Gildas Bazin committed
920

921 922 923 924
            p_sys->pp_streams = (dshow_stream_t **)realloc( p_sys->pp_streams,
                sizeof(dshow_stream_t *) * (p_sys->i_streams + 1) );
            p_sys->pp_streams[p_sys->i_streams] = new dshow_stream_t;
            *p_sys->pp_streams[p_sys->i_streams++] = dshow_stream;
Gildas Bazin's avatar
 
Gildas Bazin committed
925

926 927
            return VLC_SUCCESS;
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
928 929
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
930
 fail:
Gildas Bazin's avatar
 
Gildas Bazin committed
931
    /* Remove filters from graph */
Gildas Bazin's avatar
 
Gildas Bazin committed
932 933
    p_sys->p_graph->RemoveFilter( p_device_filter );
    p_sys->p_graph->RemoveFilter( p_capture_filter );
Gildas Bazin's avatar
 
Gildas Bazin committed
934 935

    /* Release objects */
Gildas Bazin's avatar
 
Gildas Bazin committed
936 937
    p_device_filter->Release();
    p_capture_filter->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
938 939 940 941 942 943

    return VLC_EGENERIC;
}

static IBaseFilter *
FindCaptureDevice( vlc_object_t *p_this, string *p_devicename,
Gildas Bazin's avatar
 
Gildas Bazin committed
944
                   list<string> *p_listdevices, vlc_bool_t b_audio )
Gildas Bazin's avatar
 
Gildas Bazin committed
945
{
Gildas Bazin's avatar
 
Gildas Bazin committed
946 947 948
    IBaseFilter *p_base_filter = NULL;
    IMoniker *p_moniker = NULL;
    ULONG i_fetched;
Gildas Bazin's avatar
 
Gildas Bazin committed
949 950 951
    HRESULT hr;

    /* Create the system device enumerator */
Gildas Bazin's avatar
 
Gildas Bazin committed
952
    ICreateDevEnum *p_dev_enum = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
953 954

    hr = CoCreateInstance( CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC,
Gildas Bazin's avatar
 
Gildas Bazin committed
955
                           IID_ICreateDevEnum, (void **)&p_dev_enum );
Gildas Bazin's avatar
 
Gildas Bazin committed
956 957
    if( FAILED(hr) )
    {
958
        msg_Err( p_this, "failed to create the device enumerator (0x%lx)", hr);
Gildas Bazin's avatar
 
Gildas Bazin committed
959 960 961 962
        return NULL;
    }

    /* Create an enumerator for the video capture devices */
Gildas Bazin's avatar
 
Gildas Bazin committed
963 964 965 966 967 968 969 970
    IEnumMoniker *p_class_enum = NULL;
    if( !b_audio )
        hr = p_dev_enum->CreateClassEnumerator( CLSID_VideoInputDeviceCategory,
                                                &p_class_enum, 0 );
    else
        hr = p_dev_enum->CreateClassEnumerator( CLSID_AudioInputDeviceCategory,
                                                &p_class_enum, 0 );
    p_dev_enum->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
971 972
    if( FAILED(hr) )
    {
973
        msg_Err( p_this, "failed to create the class enumerator (0x%lx)", hr );
Gildas Bazin's avatar
 
Gildas Bazin committed
974 975 976
        <