dshow.cpp 56.2 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 35 36
 *
 * 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>

#include "filter.h"

Gildas Bazin's avatar
 
Gildas Bazin committed
37 38 39
/*****************************************************************************
 * Access: local prototypes
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
40 41
static ssize_t Read          ( input_thread_t *, byte_t *, size_t );
static ssize_t ReadCompressed( input_thread_t *, byte_t *, size_t );
Gildas Bazin's avatar
 
Gildas Bazin committed
42 43 44 45 46 47

static int OpenDevice( input_thread_t *, string, vlc_bool_t );
static IBaseFilter *FindCaptureDevice( vlc_object_t *, string *,
                                       list<string> *, vlc_bool_t );
static AM_MEDIA_TYPE EnumDeviceCaps( vlc_object_t *, IBaseFilter *,
                                     int, int, int, int, int, int );
48 49
static bool ConnectFilters( vlc_object_t *, IFilterGraph *,
                            IBaseFilter *, IPin * );
Gildas Bazin's avatar
 
Gildas Bazin committed
50 51 52

static int FindDevicesCallback( vlc_object_t *, char const *,
                                vlc_value_t, vlc_value_t, void * );
53 54
static int ConfigDevicesCallback( vlc_object_t *, char const *,
                                  vlc_value_t, vlc_value_t, void * );
Gildas Bazin's avatar
 
Gildas Bazin committed
55

56
static void PropertiesPage( vlc_object_t *, IBaseFilter * );
Gildas Bazin's avatar
 
Gildas Bazin committed
57

Gildas Bazin's avatar
 
Gildas Bazin committed
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
#if 0
    /* Debug only, use this to find out GUIDs */
    unsigned char p_st[];
    UuidToString( (IID *)&IID_IAMBufferNegotiation, &p_st );
    msg_Err( p_input, "BufferNegotiation: %s" , p_st );
#endif

/*
 * header:
 *  fcc  ".dsh"
 *  u32    stream count
 *      fcc "auds"|"vids"       0
 *      fcc codec               4
 *      if vids
 *          u32 width           8
 *          u32 height          12
 *          u32 padding         16
 *      if auds
 *          u32 channels        12
 *          u32 samplerate      8
 *          u32 samplesize      16
 *
 * data:
 *  u32     stream number
 *  u32     data size
 *  u8      data
 */

static void SetDWBE( uint8_t *p, uint32_t dw )
{
    p[0] = (dw >> 24)&0xff;
    p[1] = (dw >> 16)&0xff;
    p[2] = (dw >>  8)&0xff;
    p[3] = (dw      )&0xff;
}

static void SetQWBE( uint8_t *p, uint64_t qw )
{
    SetDWBE( p, (qw >> 32)&0xffffffff );
    SetDWBE( &p[4], qw&0xffffffff );
}

Gildas Bazin's avatar
 
Gildas Bazin committed
100
/*****************************************************************************
101
 * Module descriptor
Gildas Bazin's avatar
 
Gildas Bazin committed
102
 *****************************************************************************/
103 104 105 106 107
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
108 109 110
#define CACHING_TEXT N_("Caching value in ms")
#define CACHING_LONGTEXT N_( \
    "Allows you to modify the default caching value for directshow streams. " \
Rocky Bernstein's avatar
Rocky Bernstein committed
111
    "This value should be set in milliseconds units." )
112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
#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 " \
130 131
    "(eg. I420 (default), RV24, etc.)")
#define CONFIG_TEXT N_("Device properties")
Gildas Bazin's avatar
 
Gildas Bazin committed
132
#define CONFIG_LONGTEXT N_( \
Gildas Bazin's avatar
 
Gildas Bazin committed
133
    "Show the properties dialog of the selected device.")
Gildas Bazin's avatar
 
Gildas Bazin committed
134

135 136 137 138 139 140
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
141 142
vlc_module_begin();
    set_description( _("DirectShow input") );
143
    add_integer( "dshow-caching", (mtime_t)(0.2*CLOCK_FREQ) / 1000, NULL,
Gildas Bazin's avatar
 
Gildas Bazin committed
144
                 CACHING_TEXT, CACHING_LONGTEXT, VLC_TRUE );
145

146
    add_string( "dshow-vdev", NULL, NULL, VDEV_TEXT, VDEV_LONGTEXT, VLC_FALSE);
147
        change_string_list( ppsz_vdev, ppsz_vdev_text, FindDevicesCallback );
148 149
        change_action_add( FindDevicesCallback, N_("Refresh list") );
        change_action_add( ConfigDevicesCallback, N_("Configure") );
150

151
    add_string( "dshow-adev", NULL, NULL, ADEV_TEXT, ADEV_LONGTEXT, VLC_FALSE);
152
        change_string_list( ppsz_adev, ppsz_adev_text, FindDevicesCallback );
153 154
        change_action_add( FindDevicesCallback, N_("Refresh list") );
        change_action_add( ConfigDevicesCallback, N_("Configure") );
155

156
    add_string( "dshow-size", NULL, NULL, SIZE_TEXT, SIZE_LONGTEXT, VLC_FALSE);
157

158 159
    add_string( "dshow-chroma", NULL, NULL, CHROMA_TEXT, CHROMA_LONGTEXT,
                VLC_TRUE );
Gildas Bazin's avatar
 
Gildas Bazin committed
160 161 162 163

    add_bool( "dshow-config", VLC_FALSE, NULL, CONFIG_TEXT, CONFIG_LONGTEXT,
              VLC_TRUE );

Gildas Bazin's avatar
 
Gildas Bazin committed
164
    add_shortcut( "dshow" );
Gildas Bazin's avatar
 
Gildas Bazin committed
165
    set_capability( "access", 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
166 167 168 169 170 171 172 173 174 175
    set_callbacks( AccessOpen, AccessClose );

    add_submodule();
    set_description( _("DirectShow demuxer") );
    add_shortcut( "dshow" );
    set_capability( "demux", 200 );
    set_callbacks( DemuxOpen, DemuxClose );

vlc_module_end();

Gildas Bazin's avatar
 
Gildas Bazin committed
176 177 178 179 180 181 182 183 184 185
/****************************************************************************
 * DirectShow elementary stream descriptor
 ****************************************************************************/
typedef struct dshow_stream_t
{
    string          devicename;
    IBaseFilter     *p_device_filter;
    CaptureFilter   *p_capture_filter;
    AM_MEDIA_TYPE   mt;
    int             i_fourcc;
Gildas Bazin's avatar
 
Gildas Bazin committed
186
    vlc_bool_t      b_invert;
Gildas Bazin's avatar
 
Gildas Bazin committed
187 188 189 190 191 192 193 194

    union
    {
      VIDEOINFOHEADER video;
      WAVEFORMATEX    audio;

    } header;

195
    vlc_bool_t      b_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
196 197 198

} dshow_stream_t;

Gildas Bazin's avatar
 
Gildas Bazin committed
199 200 201 202 203
/****************************************************************************
 * Access descriptor declaration
 ****************************************************************************/
struct access_sys_t
{
204 205 206
    vlc_mutex_t lock;
    vlc_cond_t  wait;

Gildas Bazin's avatar
 
Gildas Bazin committed
207 208 209 210 211 212
    IFilterGraph  *p_graph;
    IMediaControl *p_control;

    /* header */
    int     i_header_size;
    int     i_header_pos;
Gildas Bazin's avatar
 
Gildas Bazin committed
213
    uint8_t *p_header;
Gildas Bazin's avatar
 
Gildas Bazin committed
214

Gildas Bazin's avatar
 
Gildas Bazin committed
215
    /* list of elementary streams */
Gildas Bazin's avatar
 
Gildas Bazin committed
216 217 218
    dshow_stream_t **pp_streams;
    int            i_streams;
    int            i_current_stream;
Gildas Bazin's avatar
 
Gildas Bazin committed
219 220 221 222 223

    /* misc properties */
    int            i_width;
    int            i_height;
    int            i_chroma;
Gildas Bazin's avatar
 
Gildas Bazin committed
224
    int            b_audio;
Gildas Bazin's avatar
 
Gildas Bazin committed
225 226 227 228 229 230 231 232 233
};

/*****************************************************************************
 * Open: open direct show device
 *****************************************************************************/
static int AccessOpen( vlc_object_t *p_this )
{
    input_thread_t *p_input = (input_thread_t *)p_this;
    access_sys_t   *p_sys;
234
    vlc_value_t    val;
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

240 241 242 243
    var_Create( p_input, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT);
    var_Get( p_input, "dshow-vdev", &val );
    if( val.psz_string ) vdevname = string( val.psz_string );
    if( val.psz_string ) free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
244

245 246 247 248 249 250 251 252
    var_Create( p_input, "dshow-adev", VLC_VAR_STRING | VLC_VAR_DOINHERIT);
    var_Get( p_input, "dshow-adev", &val );
    if( val.psz_string ) adevname = string( val.psz_string );
    if( val.psz_string ) free( val.psz_string );

    var_Create( p_input, "dshow-size", VLC_VAR_STRING | VLC_VAR_DOINHERIT);
    var_Get( p_input, "dshow-size", &val );
    if( val.psz_string && *val.psz_string )
Gildas Bazin's avatar
 
Gildas Bazin committed
253
    {
254
        if( !strcmp( val.psz_string, "subqcif" ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
255
        {
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283
            i_width  = 128; i_height = 96;
        }
        else if( !strcmp( val.psz_string, "qsif" ) )
        {
            i_width  = 160; i_height = 120;
        }
        else if( !strcmp( val.psz_string, "qcif" ) )
        {
            i_width  = 176; i_height = 144;
        }
        else if( !strcmp( val.psz_string, "sif" ) )
        {
            i_width  = 320; i_height = 240;
        }
        else if( !strcmp( val.psz_string, "cif" ) )
        {
            i_width  = 352; i_height = 288;
        }
        else if( !strcmp( val.psz_string, "vga" ) )
        {
            i_width  = 640; i_height = 480;
        }
        else
        {
            /* Width x Height */
            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
284
            {
285
                i_height = strtol( psz_parser + 1, &psz_parser, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
286
            }
287
            msg_Dbg( p_input, "Width x Height %dx%d", i_width, i_height );
Gildas Bazin's avatar
 
Gildas Bazin committed
288 289
        }
    }
290
    if( val.psz_string ) free( val.psz_string );
Gildas Bazin's avatar
 
Gildas Bazin committed
291

292 293 294 295 296 297 298 299
    var_Create( p_input, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT);
    var_Get( p_input, "dshow-chroma", &val );
    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
300 301 302 303 304 305 306 307 308 309 310 311 312

    p_input->pf_read        = Read;
    p_input->pf_seek        = NULL;
    p_input->pf_set_area    = NULL;
    p_input->pf_set_program = NULL;

    vlc_mutex_lock( &p_input->stream.stream_lock );
    p_input->stream.b_pace_control = 0;
    p_input->stream.b_seekable = 0;
    p_input->stream.p_selected_area->i_size = 0;
    p_input->stream.p_selected_area->i_tell = 0;
    p_input->stream.i_method = INPUT_METHOD_FILE;
    vlc_mutex_unlock( &p_input->stream.stream_lock );
Sam Hocevar's avatar
Sam Hocevar committed
313

314 315 316
    var_Create( p_input, "dshow-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT);
    var_Get( p_input, "dshow-caching", &val );
    p_input->i_pts_delay = val.i_int * 1000;
Gildas Bazin's avatar
 
Gildas Bazin committed
317 318

    /* Initialize OLE/COM */
319
    CoInitialize( 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
320 321 322 323

    /* create access private data */
    p_input->p_access_data = p_sys =
        (access_sys_t *)malloc( sizeof( access_sys_t ) );
Gildas Bazin's avatar
 
Gildas Bazin committed
324 325 326 327

    /* Initialize some data */
    p_sys->i_streams = 0;
    p_sys->pp_streams = (dshow_stream_t **)malloc( 1 );
Gildas Bazin's avatar
 
Gildas Bazin committed
328 329 330
    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
331
    p_sys->b_audio = VLC_TRUE;
Gildas Bazin's avatar
 
Gildas Bazin committed
332

Gildas Bazin's avatar
 
Gildas Bazin committed
333 334 335 336 337 338
    /* Create header */
    p_sys->i_header_size = 8;
    p_sys->p_header      = (uint8_t *)malloc( p_sys->i_header_size );
    memcpy(  &p_sys->p_header[0], ".dsh", 4 );
    SetDWBE( &p_sys->p_header[4], 1 );
    p_sys->i_header_pos = p_sys->i_header_size;
Gildas Bazin's avatar
 
Gildas Bazin committed
339

Gildas Bazin's avatar
 
Gildas Bazin committed
340 341 342 343 344 345 346
    /* Build directshow graph */
    CoCreateInstance( CLSID_FilterGraph, 0, CLSCTX_INPROC,
                      (REFIID)IID_IFilterGraph, (void **)&p_sys->p_graph );

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

Gildas Bazin's avatar
 
Gildas Bazin committed
347
    if( OpenDevice( p_input, vdevname, 0 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
348 349 350 351
    {
        msg_Err( p_input, "can't open video");
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
352
    if( p_sys->b_audio && OpenDevice( p_input, adevname, 1 ) != VLC_SUCCESS )
Gildas Bazin's avatar
 
Gildas Bazin committed
353 354 355 356
    {
        msg_Err( p_input, "can't open audio");
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
357
    if( !p_sys->i_streams )
Gildas Bazin's avatar
 
Gildas Bazin committed
358
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
359
        /* Release directshow objects */
360
        if( p_sys->p_control ) p_sys->p_control->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
361
        p_sys->p_graph->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
362 363

        /* Uninitialize OLE/COM */
Sam Hocevar's avatar
Sam Hocevar committed
364
        CoUninitialize();
Gildas Bazin's avatar
 
Gildas Bazin committed
365

Gildas Bazin's avatar
 
Gildas Bazin committed
366
        free( p_sys->p_header );
367
        free( p_sys->pp_streams );
Gildas Bazin's avatar
 
Gildas Bazin committed
368 369 370 371
        free( p_sys );
        return VLC_EGENERIC;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
372 373
    /* Initialize some data */
    p_sys->i_current_stream = 0;
374 375 376 377
    p_input->i_mtu += p_sys->i_header_size + 16 /* data header size */;

    vlc_mutex_init( p_input, &p_sys->lock );
    vlc_cond_init( p_input, &p_sys->wait );
Gildas Bazin's avatar
 
Gildas Bazin committed
378 379 380 381

    /* Everything is ready. Let's rock baby */
    p_sys->p_control->Run();

Gildas Bazin's avatar
 
Gildas Bazin committed
382 383 384 385 386 387 388 389 390 391 392 393
    return VLC_SUCCESS;
}

/*****************************************************************************
 * AccessClose: close device
 *****************************************************************************/
static void AccessClose( vlc_object_t *p_this )
{
    input_thread_t *p_input = (input_thread_t *)p_this;
    access_sys_t    *p_sys  = p_input->p_access_data;

    /* Stop capturing stuff */
394
    //p_sys->p_control->Stop(); /* FIXME?: we get stuck here sometimes */
Gildas Bazin's avatar
 
Gildas Bazin committed
395 396 397
    p_sys->p_control->Release();

    /* Remove filters from graph */
Gildas Bazin's avatar
 
Gildas Bazin committed
398 399 400
    for( int i = 0; i < p_sys->i_streams; i++ )
    {
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_capture_filter );
401
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_device_filter );
Gildas Bazin's avatar
 
Gildas Bazin committed
402
        p_sys->pp_streams[i]->p_capture_filter->Release();
403
        p_sys->pp_streams[i]->p_device_filter->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
404 405
    }
    p_sys->p_graph->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
406 407

    /* Uninitialize OLE/COM */
Sam Hocevar's avatar
Sam Hocevar committed
408
    CoUninitialize();
Gildas Bazin's avatar
 
Gildas Bazin committed
409

Gildas Bazin's avatar
 
Gildas Bazin committed
410 411 412
    free( p_sys->p_header );
    for( int i = 0; i < p_sys->i_streams; i++ ) delete p_sys->pp_streams[i];
    free( p_sys->pp_streams );
Gildas Bazin's avatar
 
Gildas Bazin committed
413 414 415 416 417 418
    free( p_sys );
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
419 420
static bool ConnectFilters( vlc_object_t *p_this, IFilterGraph *p_graph,
                            IBaseFilter *p_filter, IPin *p_input_pin )
Gildas Bazin's avatar
 
Gildas Bazin committed
421 422
{
    IEnumPins *p_enumpins;
423
    IPin *p_pin;
Gildas Bazin's avatar
 
Gildas Bazin committed
424 425 426

    if( S_OK != p_filter->EnumPins( &p_enumpins ) ) return false;

427
    while( S_OK == p_enumpins->Next( 1, &p_pin, NULL ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
428
    {
429 430 431 432 433
        PIN_DIRECTION pin_dir;
        p_pin->QueryDirection( &pin_dir );

        if( pin_dir == PINDIR_OUTPUT &&
            S_OK == p_graph->ConnectDirect( p_pin, p_input_pin, 0 ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
434
        {
435
            p_pin->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
436 437 438
            p_enumpins->Release();
            return true;
        }
439
        p_pin->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
440 441 442 443 444 445
    }

    p_enumpins->Release();
    return false;
}

Gildas Bazin's avatar
 
Gildas Bazin committed
446 447
static int OpenDevice( input_thread_t *p_input, string devicename,
                       vlc_bool_t b_audio )
Gildas Bazin's avatar
 
Gildas Bazin committed
448 449 450 451
{
    access_sys_t *p_sys = p_input->p_access_data;
    list<string> list_devices;

452
    /* Enumerate devices and display their names */
Gildas Bazin's avatar
 
Gildas Bazin committed
453
    FindCaptureDevice( (vlc_object_t *)p_input, NULL, &list_devices, b_audio );
Gildas Bazin's avatar
 
Gildas Bazin committed
454

455 456 457
    if( !list_devices.size() )
        return VLC_EGENERIC;

Gildas Bazin's avatar
 
Gildas Bazin committed
458 459
    list<string>::iterator iter;
    for( iter = list_devices.begin(); iter != list_devices.end(); iter++ )
Gildas Bazin's avatar
 
Gildas Bazin committed
460
        msg_Dbg( p_input, "found device: %s", iter->c_str() );
Gildas Bazin's avatar
 
Gildas Bazin committed
461 462 463 464 465 466 467 468

    /* 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
469 470 471 472 473 474 475 476 477 478 479
    // a capture/preview device, such as a desktop USB video camera.
    IBaseFilter *p_device_filter =
        FindCaptureDevice( (vlc_object_t *)p_input, &devicename,
                           NULL, b_audio );
    if( p_device_filter )
        msg_Dbg( p_input, "using device: %s", devicename.c_str() );
    else
    {
        msg_Err( p_input, "can't use device: %s", devicename.c_str() );
        return VLC_EGENERIC;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
480

Gildas Bazin's avatar
 
Gildas Bazin committed
481
    AM_MEDIA_TYPE media_type =
482 483 484
        EnumDeviceCaps( (vlc_object_t *)p_input, p_device_filter,
                        p_sys->i_chroma, p_sys->i_width, p_sys->i_height,
                        0, 0, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
485

Gildas Bazin's avatar
 
Gildas Bazin committed
486
    /* Create and add our capture filter */
Gildas Bazin's avatar
 
Gildas Bazin committed
487
    CaptureFilter *p_capture_filter = new CaptureFilter( p_input, media_type );
Gildas Bazin's avatar
 
Gildas Bazin committed
488
    p_sys->p_graph->AddFilter( p_capture_filter, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
489 490 491

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

Gildas Bazin's avatar
 
Gildas Bazin committed
494
    /* Attempt to connect one of this device's capture output pins */
Gildas Bazin's avatar
 
Gildas Bazin committed
495
    msg_Dbg( p_input, "connecting filters" );
496
    if( ConnectFilters( VLC_OBJECT(p_input), p_sys->p_graph, p_device_filter,
Gildas Bazin's avatar
 
Gildas Bazin committed
497
                        p_capture_filter->CustomGetPin() ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
498 499
    {
        /* Success */
Gildas Bazin's avatar
 
Gildas Bazin committed
500
        dshow_stream_t dshow_stream;
Gildas Bazin's avatar
 
Gildas Bazin committed
501
        dshow_stream.b_invert = VLC_FALSE;
502
        dshow_stream.b_pts = VLC_FALSE;
Gildas Bazin's avatar
 
Gildas Bazin committed
503 504
        dshow_stream.mt =
            p_capture_filter->CustomGetPin()->CustomGetMediaType();
Gildas Bazin's avatar
 
Gildas Bazin committed
505

Gildas Bazin's avatar
 
Gildas Bazin committed
506
        if( dshow_stream.mt.majortype == MEDIATYPE_Video )
Gildas Bazin's avatar
 
Gildas Bazin committed
507
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
508 509
            msg_Dbg( p_input, "MEDIATYPE_Video");

Gildas Bazin's avatar
 
Gildas Bazin committed
510 511 512 513 514
            /* Packed RGB formats */
            if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB1 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '1' );
            if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB4 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '4' );
Gildas Bazin's avatar
 
Gildas Bazin committed
515
            if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB8 )
Gildas Bazin's avatar
 
Gildas Bazin committed
516
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' );
Gildas Bazin's avatar
 
Gildas Bazin committed
517 518 519 520 521 522 523 524 525 526
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB555 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'V', '1', '5' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB565 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'V', '1', '6' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB24 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'V', '2', '4' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB32 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'V', '3', '2' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_ARGB32 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'G', 'B', 'A' );
Gildas Bazin's avatar
 
Gildas Bazin committed
527 528 529 530

            /* Packed YUV formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YVYU )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', 'Y', 'U' );
Gildas Bazin's avatar
 
Gildas Bazin committed
531 532 533 534
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YUYV )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', 'V' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_Y411 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'I', '4', '1', 'N' );
Gildas Bazin's avatar
 
Gildas Bazin committed
535 536 537 538 539 540 541 542 543
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_Y211 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', '2', '1', '1' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YUY2 ||
                     dshow_stream.mt.subtype == MEDIASUBTYPE_UYVY )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'U', 'Y', '2' );

            /* Planar YUV formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_I420 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'I', '4', '2', '0' );
Gildas Bazin's avatar
 
Gildas Bazin committed
544 545
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_Y41P )
                dshow_stream.i_fourcc = VLC_FOURCC( 'I', '4', '1', '1' );
Gildas Bazin's avatar
 
Gildas Bazin committed
546 547
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YV12 ||
                     dshow_stream.mt.subtype == MEDIASUBTYPE_IYUV )
Gildas Bazin's avatar
 
Gildas Bazin committed
548
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' );
Gildas Bazin's avatar
 
Gildas Bazin committed
549 550 551
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YVU9 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', 'U', '9' );

Gildas Bazin's avatar
 
Gildas Bazin committed
552 553 554 555 556 557 558 559
            /* DV formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_dvsl )
                dshow_stream.i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'l' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_dvsd )
                dshow_stream.i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'd' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_dvhd )
                dshow_stream.i_fourcc = VLC_FOURCC( 'd', 'v', 'h', 'd' );

Gildas Bazin's avatar
 
Gildas Bazin committed
560 561 562 563
            /* MPEG video formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_MPEG2_VIDEO )
                dshow_stream.i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'v' );

Gildas Bazin's avatar
 
Gildas Bazin committed
564 565 566 567 568
            else goto fail;

            dshow_stream.header.video =
                *(VIDEOINFOHEADER *)dshow_stream.mt.pbFormat;

Gildas Bazin's avatar
 
Gildas Bazin committed
569 570 571
            int i_height = dshow_stream.header.video.bmiHeader.biHeight;

            /* Check if the image is inverted (bottom to top) */
Gildas Bazin's avatar
 
Gildas Bazin committed
572 573 574 575
            if( dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'G', 'B', '1' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'G', 'B', '4' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'G', 'B', '8' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'V', '1', '5' ) ||
Gildas Bazin's avatar
 
Gildas Bazin committed
576 577 578 579 580 581 582 583 584
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'V', '1', '6' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'V', '2', '4' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'V', '3', '2' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'R', 'G', 'B', 'A' ) )
            {
                if( i_height > 0 ) dshow_stream.b_invert = VLC_TRUE;
                else i_height = - i_height;
            }

Gildas Bazin's avatar
 
Gildas Bazin committed
585 586 587 588 589
            /* Check if we are dealing with a DV stream */
            if( dshow_stream.i_fourcc == VLC_FOURCC( 'd', 'v', 's', 'l' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'd', 'v', 's', 'd' ) ||
                dshow_stream.i_fourcc == VLC_FOURCC( 'd', 'v', 'h', 'd' ) )
            {
Gildas Bazin's avatar
 
Gildas Bazin committed
590
                p_input->pf_read = ReadCompressed;
Gildas Bazin's avatar
 
Gildas Bazin committed
591 592 593 594
                if( !p_input->psz_demux || !*p_input->psz_demux )
                {
                    p_input->psz_demux = "rawdv";
                }
Gildas Bazin's avatar
 
Gildas Bazin committed
595
                p_sys->b_audio = VLC_FALSE;
Gildas Bazin's avatar
 
Gildas Bazin committed
596 597
            }

Gildas Bazin's avatar
 
Gildas Bazin committed
598 599 600 601 602 603 604 605 606 607
            /* Check if we are dealing with an MPEG video stream */
            if( dshow_stream.i_fourcc == VLC_FOURCC( 'm', 'p', '2', 'v' ) )
            {
                p_input->pf_read = ReadCompressed;
                if( !p_input->psz_demux || !*p_input->psz_demux )
                {
                    p_input->psz_demux = "mpgv";
                }
                p_sys->b_audio = VLC_FALSE;
            }
Gildas Bazin's avatar
 
Gildas Bazin committed
608

Gildas Bazin's avatar
 
Gildas Bazin committed
609 610 611 612 613 614 615 616 617
            /* Add video stream to header */
            p_sys->i_header_size += 20;
            p_sys->p_header = (uint8_t *)realloc( p_sys->p_header,
                                                  p_sys->i_header_size );
            memcpy(  &p_sys->p_header[p_sys->i_header_pos], "vids", 4 );
            memcpy(  &p_sys->p_header[p_sys->i_header_pos + 4],
                     &dshow_stream.i_fourcc, 4 );
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 8],
                     dshow_stream.header.video.bmiHeader.biWidth );
Gildas Bazin's avatar
 
Gildas Bazin committed
618
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 12], i_height );
Gildas Bazin's avatar
 
Gildas Bazin committed
619 620
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 16], 0 );
            p_sys->i_header_pos = p_sys->i_header_size;
621 622 623

            /* Greatly simplifies the reading routine */
            int i_mtu = dshow_stream.header.video.bmiHeader.biWidth *
Gildas Bazin's avatar
 
Gildas Bazin committed
624 625
                i_height * 4;
            p_input->i_mtu = __MAX( p_input->i_mtu, (unsigned int)i_mtu );
Gildas Bazin's avatar
 
Gildas Bazin committed
626 627
        }

Gildas Bazin's avatar
 
Gildas Bazin committed
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
        else if( dshow_stream.mt.majortype == MEDIATYPE_Audio &&
                 dshow_stream.mt.formattype == FORMAT_WaveFormatEx )
        {
            msg_Dbg( p_input, "MEDIATYPE_Audio");

            if( dshow_stream.mt.subtype == MEDIASUBTYPE_PCM )
                dshow_stream.i_fourcc = VLC_FOURCC( 'a', 'r', 'a', 'w' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_IEEE_FLOAT )
                dshow_stream.i_fourcc = VLC_FOURCC( 'f', 'l', '3', '2' );
            else goto fail;

            dshow_stream.header.audio =
                *(WAVEFORMATEX *)dshow_stream.mt.pbFormat;

            /* Add audio stream to header */
            p_sys->i_header_size += 20;
            p_sys->p_header = (uint8_t *)realloc( p_sys->p_header,
                                                  p_sys->i_header_size );
            memcpy(  &p_sys->p_header[p_sys->i_header_pos], "auds", 4 );
            memcpy(  &p_sys->p_header[p_sys->i_header_pos + 4],
                     &dshow_stream.i_fourcc, 4 );
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 8],
                     dshow_stream.header.audio.nChannels );
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 12],
                     dshow_stream.header.audio.nSamplesPerSec );
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 16],
                     dshow_stream.header.audio.wBitsPerSample );
            p_sys->i_header_pos = p_sys->i_header_size;
656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679

            /* Greatly simplifies the reading routine */
            IAMBufferNegotiation *p_ambuf;
            IPin *p_pin;
            int i_mtu;

            p_capture_filter->CustomGetPin()->ConnectedTo( &p_pin );
            if( SUCCEEDED( p_pin->QueryInterface(
                  IID_IAMBufferNegotiation, (void **)&p_ambuf ) ) )
            {
                ALLOCATOR_PROPERTIES AllocProp;
                memset( &AllocProp, 0, sizeof( ALLOCATOR_PROPERTIES ) );
                p_ambuf->GetAllocatorProperties( &AllocProp );
                p_ambuf->Release();
                i_mtu = AllocProp.cbBuffer;
            }
            else
            {
                /* Worst case */
                i_mtu = dshow_stream.header.audio.nSamplesPerSec *
                        dshow_stream.header.audio.nChannels *
                        dshow_stream.header.audio.wBitsPerSample / 8;
            }
            p_pin->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
680
            p_input->i_mtu = __MAX( p_input->i_mtu, (unsigned int)i_mtu );
Gildas Bazin's avatar
 
Gildas Bazin committed
681
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715

        else if( dshow_stream.mt.majortype == MEDIATYPE_Stream )
        {
            msg_Dbg( p_input, "MEDIATYPE_Stream" );

            if( dshow_stream.mt.subtype == MEDIASUBTYPE_MPEG2_PROGRAM )
                dshow_stream.i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'p' );
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_MPEG2_TRANSPORT )
                dshow_stream.i_fourcc = VLC_FOURCC( 'm', 'p', '2', 't' );

            msg_Dbg( p_input, "selected stream pin accepts format: %4.4s",
                     (char *)&dshow_stream.i_fourcc);

            p_sys->b_audio = VLC_FALSE;
            p_sys->i_header_size = 0;
            p_sys->i_header_pos = 0;
            p_input->i_mtu = INPUT_DEFAULT_BUFSIZE;

            p_input->pf_read = ReadCompressed;
            p_input->pf_set_program = input_SetProgram;
        }

        else
        {
            msg_Dbg( p_input, "unknown stream majortype" );
            goto fail;
        }

        /* Show properties */
        vlc_value_t val;
        var_Create( p_input, "dshow-config",
                    VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
        var_Get( p_input, "dshow-config", &val );

716
        if(val.i_int) PropertiesPage( VLC_OBJECT(p_input), p_device_filter );
Gildas Bazin's avatar
 
Gildas Bazin committed
717

Gildas Bazin's avatar
 
Gildas Bazin committed
718 719 720
        /* 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
721

Gildas Bazin's avatar
 
Gildas Bazin committed
722 723 724 725 726 727 728
        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;
        SetDWBE( &p_sys->p_header[4], (uint32_t)p_sys->i_streams );
Gildas Bazin's avatar
 
Gildas Bazin committed
729 730 731 732

        return VLC_SUCCESS;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
733
 fail:
Gildas Bazin's avatar
 
Gildas Bazin committed
734
    /* Remove filters from graph */
Gildas Bazin's avatar
 
Gildas Bazin committed
735 736
    p_sys->p_graph->RemoveFilter( p_device_filter );
    p_sys->p_graph->RemoveFilter( p_capture_filter );
Gildas Bazin's avatar
 
Gildas Bazin committed
737 738

    /* Release objects */
Gildas Bazin's avatar
 
Gildas Bazin committed
739 740
    p_device_filter->Release();
    p_capture_filter->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
741 742 743 744 745 746

    return VLC_EGENERIC;
}

static IBaseFilter *
FindCaptureDevice( vlc_object_t *p_this, string *p_devicename,
Gildas Bazin's avatar
 
Gildas Bazin committed
747
                   list<string> *p_listdevices, vlc_bool_t b_audio )
Gildas Bazin's avatar
 
Gildas Bazin committed
748
{
Gildas Bazin's avatar
 
Gildas Bazin committed
749 750 751
    IBaseFilter *p_base_filter = NULL;
    IMoniker *p_moniker = NULL;
    ULONG i_fetched;
Gildas Bazin's avatar
 
Gildas Bazin committed
752 753 754
    HRESULT hr;

    /* Create the system device enumerator */
Gildas Bazin's avatar
 
Gildas Bazin committed
755
    ICreateDevEnum *p_dev_enum = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
756 757

    hr = CoCreateInstance( CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC,
Gildas Bazin's avatar
 
Gildas Bazin committed
758
                           IID_ICreateDevEnum, (void **)&p_dev_enum );
Gildas Bazin's avatar
 
Gildas Bazin committed
759 760 761 762 763 764 765
    if( FAILED(hr) )
    {
        msg_Err( p_this, "failed to create the device enumerator (0x%x)", hr);
        return NULL;
    }

    /* Create an enumerator for the video capture devices */
Gildas Bazin's avatar
 
Gildas Bazin committed
766 767 768 769 770 771 772 773
    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
774 775 776 777 778 779
    if( FAILED(hr) )
    {
        msg_Err( p_this, "failed to create the class enumerator (0x%x)", hr );
        return NULL;
    }

Sam Hocevar's avatar
Sam Hocevar committed
780
    /* If there are no enumerators for the requested type, then
Gildas Bazin's avatar
 
Gildas Bazin committed
781 782
     * CreateClassEnumerator will succeed, but p_class_enum will be NULL */
    if( p_class_enum == NULL )
Gildas Bazin's avatar
 
Gildas Bazin committed
783
    {
784
        msg_Err( p_this, "no capture device was detected" );
Gildas Bazin's avatar
 
Gildas Bazin committed
785 786 787 788 789 790 791 792 793
        return NULL;
    }

    /* Enumerate the devices */

    /* Note that if the Next() call succeeds but there are no monikers,
     * it will return S_FALSE (which is not a failure). Therefore, we check
     * that the return code is S_OK instead of using SUCCEEDED() macro. */

Gildas Bazin's avatar
 
Gildas Bazin committed
794
    while( p_class_enum->Next( 1, &p_moniker, &i_fetched ) == S_OK )
Gildas Bazin's avatar
 
Gildas Bazin committed
795 796
    {
        /* Getting the property page to get the device name */
Gildas Bazin's avatar
 
Gildas Bazin committed
797 798 799
        IPropertyBag *p_bag;
        hr = p_moniker->BindToStorage( 0, 0, IID_IPropertyBag,
                                       (void **)&p_bag );
Gildas Bazin's avatar
 
Gildas Bazin committed
800 801 802 803
        if( SUCCEEDED(hr) )
        {
            VARIANT var;
            var.vt = VT_BSTR;
Gildas Bazin's avatar
 
Gildas Bazin committed
804 805
            hr = p_bag->Read( L"FriendlyName", &var, NULL );
            p_bag->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
806 807 808 809 810 811 812 813 814 815 816 817 818
            if( SUCCEEDED(hr) )
            {
                int i_convert = ( lstrlenW( var.bstrVal ) + 1 ) * 2;
                char *p_buf = (char *)alloca( i_convert ); p_buf[0] = 0;
                WideCharToMultiByte( CP_ACP, 0, var.bstrVal, -1, p_buf,
                                     i_convert, NULL, NULL );
                SysFreeString(var.bstrVal);

                if( p_listdevices ) p_listdevices->push_back( p_buf );

                if( p_devicename && *p_devicename == string(p_buf) )
                {
                    /* Bind Moniker to a filter object */
Gildas Bazin's avatar
 
Gildas Bazin committed
819 820
                    hr = p_moniker->BindToObject( 0, 0, IID_IBaseFilter,
                                                  (void **)&p_base_filter );
Gildas Bazin's avatar
 
Gildas Bazin committed
821 822 823 824
                    if( FAILED(hr) )
                    {
                        msg_Err( p_this, "couldn't bind moniker to filter "
                                 "object (0x%x)", hr );
Gildas Bazin's avatar
 
Gildas Bazin committed
825 826
                        p_moniker->Release();
                        p_class_enum->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
827 828
                        return NULL;
                    }
Gildas Bazin's avatar
 
Gildas Bazin committed
829 830 831
                    p_moniker->Release();
                    p_class_enum->Release();
                    return p_base_filter;
Gildas Bazin's avatar
 
Gildas Bazin committed
832 833 834
                }
            }
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
835 836

        p_moniker->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
837 838
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
839
    p_class_enum->Release();
Gildas Bazin's avatar
 
Gildas Bazin committed
840 841 842
    return NULL;
}

Gildas Bazin's avatar
 
Gildas Bazin committed
843
static AM_MEDIA_TYPE EnumDeviceCaps( vlc_object_t *p_this,
844
                                     IBaseFilter *p_filter,
Gildas Bazin's avatar
 
Gildas Bazin committed
845
                                     int i_fourcc, int i_width, int i_height,
846 847
                                     int i_channels, int i_samplespersec,
                                     int i_bitspersample )
Gildas Bazin's avatar
 
Gildas Bazin committed
848 849 850 851
{
    IEnumPins *p_enumpins;
    IPin *p_output_pin;
    IEnumMediaTypes *p_enummt;
Gildas Bazin's avatar
 
Gildas Bazin committed
852 853
    int i_orig_fourcc = i_fourcc;
    vlc_bool_t b_found = VLC_FALSE;
Gildas Bazin's avatar
 
Gildas Bazin committed
854 855 856 857 858 859 860 861 862 863 864

    AM_MEDIA_TYPE media_type;
    media_type.majortype = GUID_NULL;
    media_type.subtype = GUID_NULL;
    media_type.formattype = GUID_NULL;
    media_type.pUnk = NULL;
    media_type.cbFormat = 0;
    media_type.pbFormat = NULL;

    if( S_OK != p_filter->EnumPins( &p_enumpins ) ) return media_type;

865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880
    while( S_OK == p_enumpins->Next( 1, &p_output_pin, NULL ) )
    {
        PIN_INFO info;

        if( S_OK == p_output_pin->QueryPinInfo( &info ) )
        {
            msg_Dbg( p_this, "EnumDeviceCaps: %s pin: %S",
                     info.dir == PINDIR_INPUT ? "input" : "output",
                     info.achName );
            if( info.pFilter ) info.pFilter->Release();
        }

        p_output_pin->Release();
    }
    p_enumpins->Reset();

Gildas Bazin's avatar
 
Gildas Bazin committed
881
    while( !b_found && p_enumpins->Next( 1, &p_output_pin, NULL ) == S_OK )
Gildas Bazin's avatar
 
Gildas Bazin committed
882
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
883 884 885 886 887
        PIN_INFO info;

        if( S_OK == p_output_pin->QueryPinInfo( &info ) )
        {
            if( info.pFilter ) info.pFilter->Release();
888 889 890 891 892 893
            if( info.dir == PINDIR_INPUT )
            {
                p_output_pin->Release();
                continue;
            }
            msg_Dbg( p_this, "EnumDeviceCaps: trying pin %S", info.achName );
Gildas Bazin's avatar
 
Gildas Bazin committed
894 895
        }

Gildas Bazin's avatar
 
Gildas Bazin committed
896
        /* Probe pin */
897
        if( SUCCEEDED( p_output_pin->EnumMediaTypes( &p_enummt ) ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
898 899 900 901 902
        {
            AM_MEDIA_TYPE *p_mt;
            while( p_enummt->Next( 1, &p_mt, NULL ) == S_OK )
            {

903
                if( p_mt->majortype == MEDIATYPE_Video )
Gildas Bazin's avatar
 
Gildas Bazin committed
904
                {
Gildas Bazin's avatar
 
Gildas Bazin committed
905
                    int i_current_fourcc = VLC_FOURCC(' ', ' ', ' ', ' ');
906 907 908

                    /* Packed RGB formats */
                    if( p_mt->subtype == MEDIASUBTYPE_RGB1 )
Gildas Bazin's avatar
 
Gildas Bazin committed
909
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '1' );
910
                    if( p_mt->subtype == MEDIASUBTYPE_RGB4 )
Gildas Bazin's avatar
 
Gildas Bazin committed
911
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '4' );
912
                    if( p_mt->subtype == MEDIASUBTYPE_RGB8 )
Gildas Bazin's avatar
 
Gildas Bazin committed
913
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' );
914
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB555 )
Gildas Bazin's avatar
 
Gildas Bazin committed
915
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '1', '5' );
916
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB565 )
Gildas Bazin's avatar
 
Gildas Bazin committed
917
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '1', '6' );
918
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB24 )
Gildas Bazin's avatar
 
Gildas Bazin committed
919
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '2', '4' );
920
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB32 )
Gildas Bazin's avatar
 
Gildas Bazin committed
921
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '3', '2' );
922
                    else if( p_mt->subtype == MEDIASUBTYPE_ARGB32 )
Gildas Bazin's avatar
 
Gildas Bazin committed
923 924 925 926 927 928 929 930 931 932 933
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', 'A' );

                    /* MPEG2 video elementary stream */
                    else if( p_mt->subtype == MEDIASUBTYPE_MPEG2_VIDEO )
                        i_current_fourcc = VLC_FOURCC( 'm', 'p', '2', 'v' );

                    /* hauppauge pvr video preview */
                    else if( p_mt->subtype == MEDIASUBTYPE_PREVIEW_VIDEO )
                        i_current_fourcc = VLC_FOURCC( 'P', 'V', 'R', 'V' );

                    else i_current_fourcc = *((int *)&p_mt->subtype);
934 935 936 937 938

                    int i_current_width = p_mt->pbFormat ?
                        ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biWidth : 0;
                    int i_current_height = p_mt->pbFormat ?
                        ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biHeight : 0;
Gildas Bazin's avatar
 
Gildas Bazin committed
939 940
                    if( i_current_height < 0 )
                        i_current_height = -i_current_height; 
941 942 943

                    msg_Dbg( p_this, "EnumDeviceCaps: input pin "
                             "accepts chroma: %4.4s, width:%i, height:%i",
Gildas Bazin's avatar
 
Gildas Bazin committed
944
                             (char *)&i_current_fourcc, i_current_width,
945 946
                             i_current_height );

Gildas Bazin's avatar
 
Gildas Bazin committed
947 948 949 950 951
                    if( ( !i_fourcc || i_fourcc == i_current_fourcc ||
                          ( !i_orig_fourcc && i_current_fourcc ==
                            VLC_FOURCC('I','4','2','0') ) ) &&
                        ( !i_width || i_width == i_current_width ) &&
                        ( !i_height || i_height == i_current_height ) )
952 953 954
                    {
                        /* Pick the 1st match */
                        media_type = *p_mt;
Gildas Bazin's avatar
 
Gildas Bazin committed
955
                        i_fourcc = i_current_fourcc;
956 957
                        i_width = i_current_width;
                        i_height = i_current_height;
Gildas Bazin's avatar
 
Gildas Bazin committed
958
                        b_found = VLC_TRUE;
959 960 961 962 963
                    }
                    else
                    {
                        FreeMediaType( *p_mt );
                    }
Gildas Bazin's avatar
 
Gildas Bazin committed
964
                }
965
                else if( p_mt->majortype == MEDIATYPE_Audio )
Gildas Bazin's avatar
 
Gildas Bazin committed
966
                {
Gildas Bazin's avatar
 
Gildas Bazin committed
967
                    int i_current_fourcc;
968 969 970 971 972 973 974 975
                    int i_current_channels =
                        ((WAVEFORMATEX *)p_mt->pbFormat)->nChannels;
                    int i_current_samplespersec =
                        ((WAVEFORMATEX *)p_mt->pbFormat)->nSamplesPerSec;
                    int i_current_bitspersample =
                        ((WAVEFORMATEX *)p_mt->pbFormat)->wBitsPerSample;

                    if( p_mt->subtype == MEDIASUBTYPE_PCM )
Gildas Bazin's avatar
 
Gildas Bazin committed
976 977
                        i_current_fourcc = VLC_FOURCC( 'p', 'c', 'm', ' ' );
                    else i_current_fourcc = *((int *)&p_mt->subtype);
978 979 980 981

                    msg_Dbg( p_this, "EnumDeviceCaps: input pin "
                             "accepts format: %4.4s, channels:%i, "
                             "samples/sec:%i bits/sample:%i",
Gildas Bazin's avatar
 
Gildas Bazin committed
982
                             (char *)&i_current_fourcc, i_current_channels,
983 984 985 986 987 988 989 990 991 992 993 994 995
                             i_current_samplespersec, i_current_bitspersample);

                    if( (!i_channels || i_channels == i_current_channels) &&
                        (!i_samplespersec ||
                         i_samplespersec == i_current_samplespersec) &&
                        (!i_bitspersample ||
                         i_bitspersample == i_current_bitspersample) )
                    {
                        /* Pick the 1st match */
                        media_type = *p_mt;
                        i_channels = i_current_channels;
                        i_samplespersec = i_current_samplespersec;
                        i_bitspersample = i_current_bitspersample;
Gildas Bazin's avatar
 
Gildas Bazin committed
996
                        b_found = VLC_TRUE;
997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017

                        /* Setup a few properties like the audio latency */
                        IAMBufferNegotiation *p_ambuf;

                        if( SUCCEEDED( p_output_pin->QueryInterface(
                              IID_IAMBufferNegotiation, (void **)&p_ambuf ) ) )
                        {
                            ALLOCATOR_PROPERTIES AllocProp;
                            AllocProp.cbAlign = -1;
                            AllocProp.cbBuffer = i_channels * i_samplespersec *
                              i_bitspersample / 8 / 10 ; /*100 ms of latency*/
                            AllocProp.cbPrefix = -1;
                            AllocProp.cBuffers = -1;
                            p_ambuf->SuggestAllocatorProperties( &AllocProp );
                            p_ambuf->Release();
                        }
                    }
                    else
                    {
                        FreeMediaType( *p_mt );
                    }
Gildas Bazin's avatar
 
Gildas Bazin committed
1018
                }
Gildas Bazin's avatar
 
Gildas Bazin committed
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047
                else if( p_mt->majortype == MEDIATYPE_Stream )
                {
                    msg_Dbg( p_this, "EnumDeviceCaps: MEDIATYPE_Stream" );

                    int i_current_fourcc = VLC_FOURCC(' ', ' ', ' ', ' ');

                    if( p_mt->subtype == MEDIASUBTYPE_MPEG2_PROGRAM )
                        i_current_fourcc = VLC_FOURCC( 'm', 'p', '2', 'p' );
                    else if( p_mt->subtype == MEDIASUBTYPE_MPEG2_TRANSPORT )
                        i_current_fourcc = VLC_FOURCC( 'm', 'p', '2', 't' );

                    if( ( !i_fourcc || i_fourcc == i_current_fourcc ) )
                    {
                        /* Pick the 1st match */
                        media_type = *p_mt;
                        i_fourcc = i_current_fourcc;
                        b_found = VLC_TRUE;
                    }
                    else
                    {
                        FreeMediaType( *p_mt );
                    }
                }
                else
                {
                    msg_Dbg( p_this,
                             "EnumDeviceCaps: input pin: unknown format" );
                    FreeMediaType( *p_mt );
                }
1048

Gildas Bazin's avatar
 
Gildas Bazin committed
1049 1050 1051 1052
                CoTaskMemFree( (PVOID)p_mt );
            }
            p_enummt->Release();
        }
1053

Gildas Bazin's avatar
 
Gildas Bazin committed
1054 1055 1056 1057 1058 1059 1060
        p_output_pin->Release();
    }

    p_enumpins->Release();
    return media_type;
}

Gildas Bazin's avatar
 
Gildas Bazin committed
1061 1062 1063 1064 1065 1066
/*****************************************************************************
 * Read: reads from the device into PES packets.
 *****************************************************************************
 * Returns -1 in case of error, 0 in case of EOF, otherwise the number of
 * bytes.
 *****************************************************************************/
1067 1068
static ssize_t Read( input_thread_t * p_input, byte_t * p_buffer,
                     size_t i_len )
Gildas Bazin's avatar
 
Gildas Bazin committed
1069
{
Gildas Bazin's avatar
 
Gildas Bazin committed
1070
    access_sys_t   *p_sys = p_input->p_access_data;
1071 1072 1073 1074 1075
    dshow_stream_t *p_stream = NULL;
    byte_t         *p_buf_orig = p_buffer;
    VLCMediaSample  sample;
    int             i_data_size;
    uint8_t         *p_data;
Gildas Bazin's avatar
 
Gildas Bazin committed
1076

1077 1078 1079 1080 1081 1082 1083
    if( p_sys->i_header_pos )
    {
        /* First header of the stream */
        memcpy( p_buffer, p_sys->p_header, p_sys->i_header_size );
        p_buffer += p_sys->i_header_size;
        p_sys->i_header_pos = 0;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
1084

1085
    while( 1 )
Gildas Bazin's avatar
 
Gildas Bazin committed
1086
    {
1087 1088 1089 1090 1091 1092 1093 1094
        /* Get new sample/frame from next elementary stream.
         * We first loop through all the elementary streams and if all our
         * fifos are empty we block until we are signaled some new data has
         * arrived. */
        vlc_mutex_lock( &p_sys->lock );

        int i_stream;
        for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
Gildas Bazin's avatar
 
Gildas Bazin committed
1095
        {
1096 1097 1098 1099 1100 1101 1102 1103
            p_stream = p_sys->pp_streams[i_stream];
            if( p_stream->mt.majortype == MEDIATYPE_Audio &&
                p_stream->p_capture_filter &&
                p_stream->p_capture_filter->CustomGetPin()
                  ->CustomGetSample( &sample ) == S_OK )
            {
                break;
            }
Gildas Bazin's avatar
 
Gildas Bazin committed
1104
        }
1105 1106
        if( i_stream == p_sys->i_streams )
        for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
Gildas Bazin's avatar
 
Gildas Bazin committed
1107
        {
1108 1109 1110 1111
            p_stream = p_sys->pp_streams[i_stream];
            if( p_stream->p_capture_filter &&
                p_stream->p_capture_filter->CustomGetPin()
                  ->CustomGetSample( &sample ) == S_OK )
Gildas Bazin's avatar
 
Gildas Bazin committed
1112
            {
1113
                break;
Gildas Bazin's avatar
 
Gildas Bazin committed
1114
            }
1115 1116 1117 1118 1119 1120 1121 1122
        }
        if( i_stream == p_sys->i_streams )
        {
            /* No data available. Wait until some data has arrived */
            vlc_cond_wait( &p_sys->wait, &p_sys->lock );
            vlc_mutex_unlock( &p_sys->lock );
            continue;
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
1123

1124
        vlc_mutex_unlock( &p_sys->lock );
Gildas Bazin's avatar
 
Gildas Bazin committed
1125

1126 1127 1128 1129 1130
        /*
         * We got our sample
         */
        i_data_size = sample.p_sample->GetActualDataLength();
        sample.p_sample->GetPointer( &p_data );
Gildas Bazin's avatar
 
Gildas Bazin committed
1131

1132 1133 1134
        REFERENCE_TIME i_pts, i_end_date;
        HRESULT hr = sample.p_sample->GetTime( &i_pts, &i_end_date );
        if( hr != VFW_S_NO_STOP_TIME && hr != S_OK ) i_pts = 0;
Gildas Bazin's avatar
 
Gildas Bazin committed
1135

1136 1137 1138 1139 1140 1141 1142
        if( !i_pts )
        {
            if( p_stream->mt.majortype == MEDIATYPE_Video || !p_stream->b_pts )