dshow.cpp 53.8 KB
Newer Older
gbazin's avatar
 
gbazin committed
1
/*****************************************************************************
2
 * dshow.cpp : DirectShow access module for vlc
gbazin's avatar
 
gbazin committed
3
 *****************************************************************************
Rocky Bernstein's avatar
Rocky Bernstein committed
4
 * Copyright (C) 2002, 2003 VideoLAN
gbazin's avatar
 
gbazin committed
5
 * $Id: dshow.cpp,v 1.25 2004/01/25 17:31:22 gbazin Exp $
gbazin's avatar
 
gbazin committed
6 7 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
 *
 * Author: Gildas Bazin <gbazin@netcourrier.com>
 *
 * 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"

gbazin's avatar
 
gbazin committed
37 38 39
/*****************************************************************************
 * Access: local prototypes
 *****************************************************************************/
gbazin's avatar
 
gbazin 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 );
gbazin's avatar
 
gbazin committed
42 43 44 45 46 47 48 49 50 51

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 );
static bool ConnectFilters( IFilterGraph *, IBaseFilter *, IPin * );

static int FindDevicesCallback( vlc_object_t *, char const *,
                                vlc_value_t, vlc_value_t, void * );
gbazin's avatar
 
gbazin committed
52 53 54

static void PropertiesPage( input_thread_t *, IBaseFilter * );

gbazin's avatar
 
gbazin committed
55 56 57 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
#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 );
}

gbazin's avatar
 
gbazin committed
97
/*****************************************************************************
98
 * Module descriptor
gbazin's avatar
 
gbazin committed
99
 *****************************************************************************/
100 101 102 103 104
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") };

gbazin's avatar
 
gbazin committed
105 106 107
#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
108
    "This value should be set in milliseconds units." )
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
#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 " \
127 128
    "(eg. I420 (default), RV24, etc.)")
#define CONFIG_TEXT N_("Device properties")
gbazin's avatar
 
gbazin committed
129
#define CONFIG_LONGTEXT N_( \
gbazin's avatar
 
gbazin committed
130
    "Show the properties dialog of the selected device.")
gbazin's avatar
 
gbazin committed
131

132 133 134 135 136 137
static int  AccessOpen ( vlc_object_t * );
static void AccessClose( vlc_object_t * );

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

gbazin's avatar
 
gbazin committed
138 139
vlc_module_begin();
    set_description( _("DirectShow input") );
140
    add_integer( "dshow-caching", (mtime_t)(0.2*CLOCK_FREQ) / 1000, NULL,
gbazin's avatar
 
gbazin committed
141
                 CACHING_TEXT, CACHING_LONGTEXT, VLC_TRUE );
142

143
    add_string( "dshow-vdev", NULL, NULL, VDEV_TEXT, VDEV_LONGTEXT, VLC_FALSE);
144 145
        change_string_list( ppsz_vdev, ppsz_vdev_text, FindDevicesCallback );

146
    add_string( "dshow-adev", NULL, NULL, ADEV_TEXT, ADEV_LONGTEXT, VLC_FALSE);
147 148
        change_string_list( ppsz_adev, ppsz_adev_text, FindDevicesCallback );

149
    add_string( "dshow-size", NULL, NULL, SIZE_TEXT, SIZE_LONGTEXT, VLC_FALSE);
150

151 152
    add_string( "dshow-chroma", NULL, NULL, CHROMA_TEXT, CHROMA_LONGTEXT,
                VLC_TRUE );
gbazin's avatar
 
gbazin committed
153 154 155 156

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

gbazin's avatar
 
gbazin committed
157
    add_shortcut( "dshow" );
gbazin's avatar
 
gbazin committed
158
    set_capability( "access", 0 );
gbazin's avatar
 
gbazin committed
159 160 161 162 163 164 165 166 167 168
    set_callbacks( AccessOpen, AccessClose );

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

vlc_module_end();

gbazin's avatar
 
gbazin committed
169 170 171 172 173 174 175 176 177 178
/****************************************************************************
 * 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;
gbazin's avatar
 
gbazin committed
179
    vlc_bool_t      b_invert;
gbazin's avatar
 
gbazin committed
180 181 182 183 184 185 186 187

    union
    {
      VIDEOINFOHEADER video;
      WAVEFORMATEX    audio;

    } header;

188
    vlc_bool_t      b_pts;
gbazin's avatar
 
gbazin committed
189 190 191

} dshow_stream_t;

gbazin's avatar
 
gbazin committed
192 193 194 195 196
/****************************************************************************
 * Access descriptor declaration
 ****************************************************************************/
struct access_sys_t
{
197 198 199
    vlc_mutex_t lock;
    vlc_cond_t  wait;

gbazin's avatar
 
gbazin committed
200 201 202 203 204 205
    IFilterGraph  *p_graph;
    IMediaControl *p_control;

    /* header */
    int     i_header_size;
    int     i_header_pos;
gbazin's avatar
 
gbazin committed
206
    uint8_t *p_header;
gbazin's avatar
 
gbazin committed
207

gbazin's avatar
 
gbazin committed
208
    /* list of elementary streams */
gbazin's avatar
 
gbazin committed
209 210 211
    dshow_stream_t **pp_streams;
    int            i_streams;
    int            i_current_stream;
gbazin's avatar
 
gbazin committed
212 213 214 215 216

    /* misc properties */
    int            i_width;
    int            i_height;
    int            i_chroma;
gbazin's avatar
 
gbazin committed
217
    int            b_audio;
gbazin's avatar
 
gbazin committed
218 219 220 221 222 223 224 225 226
};

/*****************************************************************************
 * 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;
227
    vlc_value_t    val;
gbazin's avatar
 
gbazin committed
228

229
    /* Get/parse options and open device(s) */
gbazin's avatar
 
gbazin committed
230
    string vdevname, adevname;
231
    int i_width = 0, i_height = 0, i_chroma = 0;
gbazin's avatar
 
gbazin committed
232

233 234 235 236
    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 );
gbazin's avatar
 
gbazin committed
237

238 239 240 241 242 243 244 245
    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 )
gbazin's avatar
 
gbazin committed
246
    {
247
        if( !strcmp( val.psz_string, "subqcif" ) )
gbazin's avatar
 
gbazin committed
248
        {
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276
            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')
gbazin's avatar
 
gbazin committed
277
            {
278
                i_height = strtol( psz_parser + 1, &psz_parser, 0 );
gbazin's avatar
 
gbazin committed
279
            }
280
            msg_Dbg( p_input, "Width x Height %dx%d", i_width, i_height );
gbazin's avatar
 
gbazin committed
281 282
        }
    }
283
    if( val.psz_string ) free( val.psz_string );
gbazin's avatar
 
gbazin committed
284

285 286 287 288 289 290 291 292
    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 );
gbazin's avatar
 
gbazin committed
293 294 295 296 297 298 299 300 301 302 303 304 305

    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
306

307 308 309
    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;
gbazin's avatar
 
gbazin committed
310 311

    /* Initialize OLE/COM */
312
    CoInitialize( 0 );
gbazin's avatar
 
gbazin committed
313 314 315 316

    /* create access private data */
    p_input->p_access_data = p_sys =
        (access_sys_t *)malloc( sizeof( access_sys_t ) );
gbazin's avatar
 
gbazin committed
317 318 319 320

    /* Initialize some data */
    p_sys->i_streams = 0;
    p_sys->pp_streams = (dshow_stream_t **)malloc( 1 );
gbazin's avatar
 
gbazin committed
321 322 323
    p_sys->i_width = i_width;
    p_sys->i_height = i_height;
    p_sys->i_chroma = i_chroma;
gbazin's avatar
 
gbazin committed
324
    p_sys->b_audio = VLC_TRUE;
gbazin's avatar
 
gbazin committed
325

gbazin's avatar
 
gbazin committed
326 327 328 329 330 331
    /* 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;
gbazin's avatar
 
gbazin committed
332

gbazin's avatar
 
gbazin committed
333 334 335 336 337 338 339
    /* 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 );

gbazin's avatar
 
gbazin committed
340
    if( OpenDevice( p_input, vdevname, 0 ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
341 342 343 344
    {
        msg_Err( p_input, "can't open video");
    }

gbazin's avatar
 
gbazin committed
345
    if( p_sys->b_audio && OpenDevice( p_input, adevname, 1 ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
346 347 348 349
    {
        msg_Err( p_input, "can't open audio");
    }

gbazin's avatar
 
gbazin committed
350
    if( !p_sys->i_streams )
gbazin's avatar
 
gbazin committed
351
    {
gbazin's avatar
 
gbazin committed
352
        /* Release directshow objects */
353
        if( p_sys->p_control ) p_sys->p_control->Release();
gbazin's avatar
 
gbazin committed
354
        p_sys->p_graph->Release();
gbazin's avatar
 
gbazin committed
355 356

        /* Uninitialize OLE/COM */
Sam Hocevar's avatar
Sam Hocevar committed
357
        CoUninitialize();
gbazin's avatar
 
gbazin committed
358

gbazin's avatar
 
gbazin committed
359
        free( p_sys->p_header );
360
        free( p_sys->pp_streams );
gbazin's avatar
 
gbazin committed
361 362 363 364
        free( p_sys );
        return VLC_EGENERIC;
    }

gbazin's avatar
 
gbazin committed
365 366
    /* Initialize some data */
    p_sys->i_current_stream = 0;
367 368 369 370
    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 );
gbazin's avatar
 
gbazin committed
371 372 373 374

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

gbazin's avatar
 
gbazin committed
375 376 377 378 379 380 381 382 383 384 385 386
    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 */
387
    //p_sys->p_control->Stop(); /* FIXME?: we get stuck here sometimes */
gbazin's avatar
 
gbazin committed
388 389 390
    p_sys->p_control->Release();

    /* Remove filters from graph */
gbazin's avatar
 
gbazin committed
391 392 393
    for( int i = 0; i < p_sys->i_streams; i++ )
    {
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_capture_filter );
394
        p_sys->p_graph->RemoveFilter( p_sys->pp_streams[i]->p_device_filter );
gbazin's avatar
 
gbazin committed
395
        p_sys->pp_streams[i]->p_capture_filter->Release();
396
        p_sys->pp_streams[i]->p_device_filter->Release();
gbazin's avatar
 
gbazin committed
397 398
    }
    p_sys->p_graph->Release();
gbazin's avatar
 
gbazin committed
399 400

    /* Uninitialize OLE/COM */
Sam Hocevar's avatar
Sam Hocevar committed
401
    CoUninitialize();
gbazin's avatar
 
gbazin committed
402

gbazin's avatar
 
gbazin committed
403 404 405
    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 );
gbazin's avatar
 
gbazin committed
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433
    free( p_sys );
}

/****************************************************************************
 * ConnectFilters
 ****************************************************************************/
static bool ConnectFilters( IFilterGraph *p_graph, IBaseFilter *p_filter,
                            IPin *p_input_pin )
{
    IEnumPins *p_enumpins;
    IPin *p_output_pin;
    ULONG i_fetched;

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

    while( S_OK == p_enumpins->Next( 1, &p_output_pin, &i_fetched ) )
    {
        if( S_OK == p_graph->ConnectDirect( p_output_pin, p_input_pin, 0 ) )
        {
            p_enumpins->Release();
            return true;
        }
    }

    p_enumpins->Release();
    return false;
}

gbazin's avatar
 
gbazin committed
434 435
static int OpenDevice( input_thread_t *p_input, string devicename,
                       vlc_bool_t b_audio )
gbazin's avatar
 
gbazin committed
436 437 438 439
{
    access_sys_t *p_sys = p_input->p_access_data;
    list<string> list_devices;

440
    /* Enumerate devices and display their names */
gbazin's avatar
 
gbazin committed
441
    FindCaptureDevice( (vlc_object_t *)p_input, NULL, &list_devices, b_audio );
gbazin's avatar
 
gbazin committed
442

443 444 445
    if( !list_devices.size() )
        return VLC_EGENERIC;

gbazin's avatar
 
gbazin committed
446 447
    list<string>::iterator iter;
    for( iter = list_devices.begin(); iter != list_devices.end(); iter++ )
gbazin's avatar
 
gbazin committed
448
        msg_Dbg( p_input, "found device: %s", iter->c_str() );
gbazin's avatar
 
gbazin committed
449 450 451 452 453 454 455 456

    /* 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
gbazin's avatar
 
gbazin committed
457 458 459 460 461 462 463 464 465 466 467
    // 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;
    }
gbazin's avatar
 
gbazin committed
468

gbazin's avatar
 
gbazin committed
469
    AM_MEDIA_TYPE media_type =
470 471 472
        EnumDeviceCaps( (vlc_object_t *)p_input, p_device_filter,
                        p_sys->i_chroma, p_sys->i_width, p_sys->i_height,
                        0, 0, 0 );
gbazin's avatar
 
gbazin committed
473

gbazin's avatar
 
gbazin committed
474
    /* Create and add our capture filter */
gbazin's avatar
 
gbazin committed
475
    CaptureFilter *p_capture_filter = new CaptureFilter( p_input, media_type );
gbazin's avatar
 
gbazin committed
476
    p_sys->p_graph->AddFilter( p_capture_filter, 0 );
gbazin's avatar
 
gbazin committed
477 478 479

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

gbazin's avatar
 
gbazin committed
482
    /* Attempt to connect one of this device's capture output pins */
gbazin's avatar
 
gbazin committed
483
    msg_Dbg( p_input, "connecting filters" );
gbazin's avatar
 
gbazin committed
484 485
    if( ConnectFilters( p_sys->p_graph, p_device_filter,
                        p_capture_filter->CustomGetPin() ) )
gbazin's avatar
 
gbazin committed
486 487
    {
        /* Success */
gbazin's avatar
 
gbazin committed
488
        dshow_stream_t dshow_stream;
gbazin's avatar
 
gbazin committed
489
        dshow_stream.b_invert = VLC_FALSE;
490
        dshow_stream.b_pts = VLC_FALSE;
gbazin's avatar
 
gbazin committed
491 492
        dshow_stream.mt =
            p_capture_filter->CustomGetPin()->CustomGetMediaType();
gbazin's avatar
 
gbazin committed
493

gbazin's avatar
 
gbazin committed
494
        if( dshow_stream.mt.majortype == MEDIATYPE_Video )
gbazin's avatar
 
gbazin committed
495
        {
gbazin's avatar
 
gbazin committed
496 497
            msg_Dbg( p_input, "MEDIATYPE_Video");

gbazin's avatar
 
gbazin committed
498 499 500 501 502
            /* 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' );
gbazin's avatar
 
gbazin committed
503
            if( dshow_stream.mt.subtype == MEDIASUBTYPE_RGB8 )
gbazin's avatar
 
gbazin committed
504
                dshow_stream.i_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' );
gbazin's avatar
 
gbazin committed
505 506 507 508 509 510 511 512 513 514
            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' );
gbazin's avatar
 
gbazin committed
515 516 517 518

            /* Packed YUV formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YVYU )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', 'Y', 'U' );
gbazin's avatar
 
gbazin committed
519 520 521 522
            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' );
gbazin's avatar
 
gbazin committed
523 524 525 526 527 528 529 530 531
            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' );
gbazin's avatar
 
gbazin committed
532 533
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_Y41P )
                dshow_stream.i_fourcc = VLC_FOURCC( 'I', '4', '1', '1' );
gbazin's avatar
 
gbazin committed
534 535
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YV12 ||
                     dshow_stream.mt.subtype == MEDIASUBTYPE_IYUV )
gbazin's avatar
 
gbazin committed
536
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', '1', '2' );
gbazin's avatar
 
gbazin committed
537 538 539
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_YVU9 )
                dshow_stream.i_fourcc = VLC_FOURCC( 'Y', 'V', 'U', '9' );

gbazin's avatar
 
gbazin committed
540 541 542 543 544 545 546 547
            /* 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' );

gbazin's avatar
 
gbazin committed
548 549 550 551
            /* MPEG video formats */
            else if( dshow_stream.mt.subtype == MEDIASUBTYPE_MPEG2_VIDEO )
                dshow_stream.i_fourcc = VLC_FOURCC( 'm', 'p', '2', 'v' );

gbazin's avatar
 
gbazin committed
552 553 554 555 556
            else goto fail;

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

gbazin's avatar
 
gbazin committed
557 558 559
            int i_height = dshow_stream.header.video.bmiHeader.biHeight;

            /* Check if the image is inverted (bottom to top) */
gbazin's avatar
 
gbazin committed
560 561 562 563
            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' ) ||
gbazin's avatar
 
gbazin committed
564 565 566 567 568 569 570 571 572
                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;
            }

gbazin's avatar
 
gbazin committed
573 574 575 576 577
            /* 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' ) )
            {
gbazin's avatar
 
gbazin committed
578
                p_input->pf_read = ReadCompressed;
gbazin's avatar
 
gbazin committed
579 580 581 582
                if( !p_input->psz_demux || !*p_input->psz_demux )
                {
                    p_input->psz_demux = "rawdv";
                }
gbazin's avatar
 
gbazin committed
583
                p_sys->b_audio = VLC_FALSE;
gbazin's avatar
 
gbazin committed
584 585
            }

gbazin's avatar
 
gbazin committed
586 587 588 589 590 591 592 593 594 595
            /* 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;
            }
gbazin's avatar
 
gbazin committed
596

gbazin's avatar
 
gbazin committed
597 598 599 600 601 602 603 604 605
            /* 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 );
gbazin's avatar
 
gbazin committed
606
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 12], i_height );
gbazin's avatar
 
gbazin committed
607 608
            SetDWBE( &p_sys->p_header[p_sys->i_header_pos + 16], 0 );
            p_sys->i_header_pos = p_sys->i_header_size;
609 610 611

            /* Greatly simplifies the reading routine */
            int i_mtu = dshow_stream.header.video.bmiHeader.biWidth *
gbazin's avatar
 
gbazin committed
612 613
                i_height * 4;
            p_input->i_mtu = __MAX( p_input->i_mtu, (unsigned int)i_mtu );
gbazin's avatar
 
gbazin committed
614 615
        }

gbazin's avatar
 
gbazin committed
616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643
        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;
644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667

            /* 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();
gbazin's avatar
 
gbazin committed
668
            p_input->i_mtu = __MAX( p_input->i_mtu, (unsigned int)i_mtu );
gbazin's avatar
 
gbazin committed
669
        }
gbazin's avatar
 
gbazin committed
670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704

        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 );

        if(val.i_int) PropertiesPage( p_input, p_device_filter );
gbazin's avatar
 
gbazin committed
705

gbazin's avatar
 
gbazin committed
706 707 708
        /* Add directshow elementary stream to our list */
        dshow_stream.p_device_filter = p_device_filter;
        dshow_stream.p_capture_filter = p_capture_filter;
gbazin's avatar
 
gbazin committed
709

gbazin's avatar
 
gbazin committed
710 711 712 713 714 715 716
        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 );
gbazin's avatar
 
gbazin committed
717 718 719 720

        return VLC_SUCCESS;
    }

gbazin's avatar
 
gbazin committed
721
 fail:
gbazin's avatar
 
gbazin committed
722
    /* Remove filters from graph */
gbazin's avatar
 
gbazin committed
723 724
    p_sys->p_graph->RemoveFilter( p_device_filter );
    p_sys->p_graph->RemoveFilter( p_capture_filter );
gbazin's avatar
 
gbazin committed
725 726

    /* Release objects */
gbazin's avatar
 
gbazin committed
727 728
    p_device_filter->Release();
    p_capture_filter->Release();
gbazin's avatar
 
gbazin committed
729 730 731 732 733 734

    return VLC_EGENERIC;
}

static IBaseFilter *
FindCaptureDevice( vlc_object_t *p_this, string *p_devicename,
gbazin's avatar
 
gbazin committed
735
                   list<string> *p_listdevices, vlc_bool_t b_audio )
gbazin's avatar
 
gbazin committed
736
{
gbazin's avatar
 
gbazin committed
737 738 739
    IBaseFilter *p_base_filter = NULL;
    IMoniker *p_moniker = NULL;
    ULONG i_fetched;
gbazin's avatar
 
gbazin committed
740 741 742
    HRESULT hr;

    /* Create the system device enumerator */
gbazin's avatar
 
gbazin committed
743
    ICreateDevEnum *p_dev_enum = NULL;
gbazin's avatar
 
gbazin committed
744 745

    hr = CoCreateInstance( CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC,
gbazin's avatar
 
gbazin committed
746
                           IID_ICreateDevEnum, (void **)&p_dev_enum );
gbazin's avatar
 
gbazin committed
747 748 749 750 751 752 753
    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 */
gbazin's avatar
 
gbazin committed
754 755 756 757 758 759 760 761
    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();
gbazin's avatar
 
gbazin committed
762 763 764 765 766 767
    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
768
    /* If there are no enumerators for the requested type, then
gbazin's avatar
 
gbazin committed
769 770
     * CreateClassEnumerator will succeed, but p_class_enum will be NULL */
    if( p_class_enum == NULL )
gbazin's avatar
 
gbazin committed
771
    {
gbazin's avatar
 
gbazin committed
772
        msg_Err( p_this, "no capture device was detected." );
gbazin's avatar
 
gbazin committed
773 774 775 776 777 778 779 780 781
        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. */

gbazin's avatar
 
gbazin committed
782
    while( p_class_enum->Next( 1, &p_moniker, &i_fetched ) == S_OK )
gbazin's avatar
 
gbazin committed
783 784
    {
        /* Getting the property page to get the device name */
gbazin's avatar
 
gbazin committed
785 786 787
        IPropertyBag *p_bag;
        hr = p_moniker->BindToStorage( 0, 0, IID_IPropertyBag,
                                       (void **)&p_bag );
gbazin's avatar
 
gbazin committed
788 789 790 791
        if( SUCCEEDED(hr) )
        {
            VARIANT var;
            var.vt = VT_BSTR;
gbazin's avatar
 
gbazin committed
792 793
            hr = p_bag->Read( L"FriendlyName", &var, NULL );
            p_bag->Release();
gbazin's avatar
 
gbazin committed
794 795 796 797 798 799 800 801 802 803 804 805 806
            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 */
gbazin's avatar
 
gbazin committed
807 808
                    hr = p_moniker->BindToObject( 0, 0, IID_IBaseFilter,
                                                  (void **)&p_base_filter );
gbazin's avatar
 
gbazin committed
809 810 811 812
                    if( FAILED(hr) )
                    {
                        msg_Err( p_this, "couldn't bind moniker to filter "
                                 "object (0x%x)", hr );
gbazin's avatar
 
gbazin committed
813 814
                        p_moniker->Release();
                        p_class_enum->Release();
gbazin's avatar
 
gbazin committed
815 816
                        return NULL;
                    }
gbazin's avatar
 
gbazin committed
817 818 819
                    p_moniker->Release();
                    p_class_enum->Release();
                    return p_base_filter;
gbazin's avatar
 
gbazin committed
820 821 822
                }
            }
        }
gbazin's avatar
 
gbazin committed
823 824

        p_moniker->Release();
gbazin's avatar
 
gbazin committed
825 826
    }

gbazin's avatar
 
gbazin committed
827
    p_class_enum->Release();
gbazin's avatar
 
gbazin committed
828 829 830
    return NULL;
}

gbazin's avatar
 
gbazin committed
831
static AM_MEDIA_TYPE EnumDeviceCaps( vlc_object_t *p_this,
832
                                     IBaseFilter *p_filter,
gbazin's avatar
 
gbazin committed
833
                                     int i_fourcc, int i_width, int i_height,
834 835
                                     int i_channels, int i_samplespersec,
                                     int i_bitspersample )
gbazin's avatar
 
gbazin committed
836 837 838 839
{
    IEnumPins *p_enumpins;
    IPin *p_output_pin;
    IEnumMediaTypes *p_enummt;
gbazin's avatar
 
gbazin committed
840 841
    int i_orig_fourcc = i_fourcc;
    vlc_bool_t b_found = VLC_FALSE;
gbazin's avatar
 
gbazin committed
842 843 844 845 846 847 848 849 850 851 852

    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;

gbazin's avatar
 
gbazin committed
853
    while( !b_found && p_enumpins->Next( 1, &p_output_pin, NULL ) == S_OK )
gbazin's avatar
 
gbazin committed
854
    {
gbazin's avatar
 
gbazin committed
855 856 857 858 859 860 861 862
        PIN_INFO info;

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

gbazin's avatar
 
gbazin committed
863
        /* Probe pin */
864
        if( SUCCEEDED( p_output_pin->EnumMediaTypes( &p_enummt ) ) )
gbazin's avatar
 
gbazin committed
865 866 867 868 869
        {
            AM_MEDIA_TYPE *p_mt;
            while( p_enummt->Next( 1, &p_mt, NULL ) == S_OK )
            {

870
                if( p_mt->majortype == MEDIATYPE_Video )
gbazin's avatar
 
gbazin committed
871
                {
gbazin's avatar
 
gbazin committed
872
                    int i_current_fourcc = VLC_FOURCC(' ', ' ', ' ', ' ');
873 874 875

                    /* Packed RGB formats */
                    if( p_mt->subtype == MEDIASUBTYPE_RGB1 )
gbazin's avatar
 
gbazin committed
876
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '1' );
877
                    if( p_mt->subtype == MEDIASUBTYPE_RGB4 )
gbazin's avatar
 
gbazin committed
878
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '4' );
879
                    if( p_mt->subtype == MEDIASUBTYPE_RGB8 )
gbazin's avatar
 
gbazin committed
880
                        i_current_fourcc = VLC_FOURCC( 'R', 'G', 'B', '8' );
881
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB555 )
gbazin's avatar
 
gbazin committed
882
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '1', '5' );
883
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB565 )
gbazin's avatar
 
gbazin committed
884
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '1', '6' );
885
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB24 )
gbazin's avatar
 
gbazin committed
886
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '2', '4' );
887
                    else if( p_mt->subtype == MEDIASUBTYPE_RGB32 )
gbazin's avatar
 
gbazin committed
888
                        i_current_fourcc = VLC_FOURCC( 'R', 'V', '3', '2' );
889
                    else if( p_mt->subtype == MEDIASUBTYPE_ARGB32 )
gbazin's avatar
 
gbazin committed
890 891 892 893 894 895 896 897 898 899 900
                        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);
901 902 903 904 905

                    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;
gbazin's avatar
 
gbazin committed
906 907
                    if( i_current_height < 0 )
                        i_current_height = -i_current_height; 
908 909 910

                    msg_Dbg( p_this, "EnumDeviceCaps: input pin "
                             "accepts chroma: %4.4s, width:%i, height:%i",
gbazin's avatar
 
gbazin committed
911
                             (char *)&i_current_fourcc, i_current_width,
912 913
                             i_current_height );

gbazin's avatar
 
gbazin committed
914 915 916 917 918
                    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 ) )
919 920 921
                    {
                        /* Pick the 1st match */
                        media_type = *p_mt;
gbazin's avatar
 
gbazin committed
922
                        i_fourcc = i_current_fourcc;
923 924
                        i_width = i_current_width;
                        i_height = i_current_height;
gbazin's avatar
 
gbazin committed
925
                        b_found = VLC_TRUE;
926 927 928 929 930
                    }
                    else
                    {
                        FreeMediaType( *p_mt );
                    }
gbazin's avatar
 
gbazin committed
931
                }
932
                else if( p_mt->majortype == MEDIATYPE_Audio )
gbazin's avatar
 
gbazin committed
933
                {
gbazin's avatar
 
gbazin committed
934
                    int i_current_fourcc;
935 936 937 938 939 940 941 942
                    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 )
gbazin's avatar
 
gbazin committed
943 944
                        i_current_fourcc = VLC_FOURCC( 'p', 'c', 'm', ' ' );
                    else i_current_fourcc = *((int *)&p_mt->subtype);
945 946 947 948

                    msg_Dbg( p_this, "EnumDeviceCaps: input pin "
                             "accepts format: %4.4s, channels:%i, "
                             "samples/sec:%i bits/sample:%i",
gbazin's avatar
 
gbazin committed
949
                             (char *)&i_current_fourcc, i_current_channels,
950 951 952 953 954 955 956 957 958 959 960 961 962
                             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;
gbazin's avatar
 
gbazin committed
963
                        b_found = VLC_TRUE;
964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984

                        /* 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 );
                    }
gbazin's avatar
 
gbazin committed
985
                }
gbazin's avatar
 
gbazin committed
986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
                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 );
                }
1015

gbazin's avatar
 
gbazin committed
1016 1017 1018 1019
                CoTaskMemFree( (PVOID)p_mt );
            }
            p_enummt->Release();
        }
1020

gbazin's avatar
 
gbazin committed
1021 1022 1023 1024 1025 1026 1027
        p_output_pin->Release();
    }

    p_enumpins->Release();
    return media_type;
}

gbazin's avatar
 
gbazin committed
1028 1029 1030 1031 1032 1033
/*****************************************************************************
 * Read: reads from the device into PES packets.
 *****************************************************************************
 * Returns -1 in case of error, 0 in case of EOF, otherwise the number of
 * bytes.
 *****************************************************************************/
1034 1035
static ssize_t Read( input_thread_t * p_input, byte_t * p_buffer,
                     size_t i_len )
gbazin's avatar
 
gbazin committed
1036
{
gbazin's avatar
 
gbazin committed
1037
    access_sys_t   *p_sys = p_input->p_access_data;
1038 1039 1040 1041 1042
    dshow_stream_t *p_stream = NULL;
    byte_t         *p_buf_orig = p_buffer;
    VLCMediaSample  sample;
    int             i_data_size;
    uint8_t         *p_data;
gbazin's avatar
 
gbazin committed
1043

1044 1045 1046 1047 1048 1049 1050
    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;
    }
gbazin's avatar
 
gbazin committed
1051

1052
    while( 1 )
gbazin's avatar
 
gbazin committed
1053
    {
1054 1055 1056 1057 1058 1059 1060 1061
        /* 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++ )
gbazin's avatar
 
gbazin committed
1062
        {
1063 1064 1065 1066 1067 1068 1069 1070
            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;
            }
gbazin's avatar
 
gbazin committed
1071
        }
1072 1073
        if( i_stream == p_sys->i_streams )
        for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
gbazin's avatar
 
gbazin committed
1074
        {
1075 1076 1077 1078
            p_stream = p_sys->pp_streams[i_stream];
            if( p_stream->p_capture_filter &&
                p_stream->p_capture_filter->CustomGetPin()
                  ->CustomGetSample( &sample ) == S_OK )
gbazin's avatar
 
gbazin committed
1079
            {
1080
                break;
gbazin's avatar
 
gbazin committed
1081
            }
1082 1083 1084 1085 1086 1087 1088 1089
        }
        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;
        }
gbazin's avatar
 
gbazin committed
1090

1091
        vlc_mutex_unlock( &p_sys->lock );
gbazin's avatar
 
gbazin committed
1092

1093 1094 1095 1096 1097
        /*
         * We got our sample
         */
        i_data_size = sample.p_sample->GetActualDataLength();
        sample.p_sample->GetPointer( &p_data );
gbazin's avatar
 
gbazin committed
1098

1099 1100 1101
        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;
gbazin's avatar