]> git.sesse.net Git - vlc/blobdiff - modules/access/dshow/dshow.cpp
All: string review (refs: #438)
[vlc] / modules / access / dshow / dshow.cpp
index 7e669109a22a8d0def667187d0723570a7c366bf..e563ef27e25771e17c6b736e25bb315814b7f3f8 100644 (file)
@@ -1,7 +1,7 @@
 /*****************************************************************************
  * dshow.cpp : DirectShow access module for vlc
  *****************************************************************************
- * Copyright (C) 2002, 2003 VideoLAN
+ * Copyright (C) 2002, 2003 the VideoLAN team
  * $Id$
  *
  * Author: Gildas Bazin <gbazin@videolan.org>
@@ -18,7 +18,7 @@
  *
  * 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.
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  *****************************************************************************/
 
 /*****************************************************************************
@@ -99,6 +99,10 @@ static char *ppsz_tuner_input_text[] =
 #define CHROMA_LONGTEXT N_( \
     "Force the DirectShow video input to use a specific chroma format " \
     "(eg. I420 (default), RV24, etc.)")
+#define FPS_TEXT N_("Video input frame rate")
+#define FPS_LONGTEXT N_( \
+    "Force the DirectShow video input to use a specific frame rate" \
+    "(eg. 0 means default, 25, 29.97, 50, 59.94, etc.)")
 #define CONFIG_TEXT N_("Device properties")
 #define CONFIG_LONGTEXT N_( \
     "Show the properties dialog of the selected device before starting the " \
@@ -130,6 +134,8 @@ static void DemuxClose ( vlc_object_t * );
 vlc_module_begin();
     set_shortname( _("DirectShow") );
     set_description( _("DirectShow input") );
+    set_category( CAT_INPUT );
+    set_subcategory( SUBCAT_INPUT_ACCESS );
     add_integer( "dshow-caching", (mtime_t)(0.2*CLOCK_FREQ) / 1000, NULL,
                  CACHING_TEXT, CACHING_LONGTEXT, VLC_TRUE );
 
@@ -148,11 +154,14 @@ vlc_module_begin();
     add_string( "dshow-chroma", NULL, NULL, CHROMA_TEXT, CHROMA_LONGTEXT,
                 VLC_TRUE );
 
+    add_float( "dshow-fps", 0.0f, NULL, FPS_TEXT, FPS_LONGTEXT,
+                VLC_TRUE );
+
     add_bool( "dshow-config", VLC_FALSE, NULL, CONFIG_TEXT, CONFIG_LONGTEXT,
-              VLC_FALSE );
+              VLC_TRUE );
 
     add_bool( "dshow-tuner", VLC_FALSE, NULL, TUNER_TEXT, TUNER_LONGTEXT,
-              VLC_FALSE );
+              VLC_TRUE );
 
     add_integer( "dshow-tuner-channel", 0, NULL, CHANNEL_TEXT,
                  CHANNEL_LONGTEXT, VLC_TRUE );
@@ -197,7 +206,6 @@ typedef struct dshow_stream_t
     es_out_id_t     *p_es;
 
     vlc_bool_t      b_pts;
-
 } dshow_stream_t;
 
 /*****************************************************************************
@@ -273,7 +281,6 @@ static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
 
     var_Create( p_this, "dshow-config", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
     var_Create( p_this, "dshow-tuner", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
-
     var_Create( p_this, "dshow-vdev", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
     var_Get( p_this, "dshow-vdev", &val );
     if( val.psz_string ) vdevname = string( val.psz_string );
@@ -316,15 +323,18 @@ static int CommonOpen( vlc_object_t *p_this, access_sys_t *p_sys,
     }
     if( val.psz_string ) free( val.psz_string );
 
+    p_sys->b_chroma = VLC_FALSE;
     var_Create( p_this, "dshow-chroma", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
     var_Get( p_this, "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] );
+        p_sys->b_chroma = VLC_TRUE;
     }
     if( val.psz_string ) free( val.psz_string );
 
+    var_Create( p_this, "dshow-fps", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
     var_Create( p_this, "dshow-tuner-channel",
                 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
     var_Create( p_this, "dshow-tuner-country",
@@ -520,6 +530,13 @@ static int DemuxOpen( vlc_object_t *p_this )
                 fmt.video.i_gmask = 0x0000ff00;
                 fmt.video.i_rmask = 0x000000ff;
             }
+
+            if( p_stream->header.video.AvgTimePerFrame )
+            {
+                fmt.video.i_frame_rate = 10000000;
+                fmt.video.i_frame_rate_base =
+                    p_stream->header.video.AvgTimePerFrame;
+            }
         }
         else if( p_stream->mt.majortype == MEDIATYPE_Audio )
         {
@@ -815,88 +832,11 @@ static int OpenDevice( vlc_object_t *p_this, access_sys_t *p_sys,
                         p_sys->i_width, p_sys->i_height,
                         0, 0, 0, media_types, MAX_MEDIA_TYPES );
 
-    /* Find out if the pin handles MEDIATYPE_Stream, in which case we
-     * won't add a prefered media type as this doesn't seem to work well
-     * -- to investigate. */
-    vlc_bool_t b_stream_type = VLC_FALSE;
-    for( size_t i = 0; i < media_count; i++ )
-    {
-        if( media_types[i].majortype == MEDIATYPE_Stream )
-        {
-            b_stream_type = VLC_TRUE;
-            break;
-        }
-    }
-
-    size_t mt_count = 0;
     AM_MEDIA_TYPE *mt = NULL;
 
-    if( !b_stream_type && !b_audio )
-    {
-        // Insert prefered video media type
-        AM_MEDIA_TYPE mtr;
-        VIDEOINFOHEADER vh;
-
-        mtr.majortype            = MEDIATYPE_Video;
-        mtr.subtype              = MEDIASUBTYPE_I420;
-        mtr.bFixedSizeSamples    = TRUE;
-        mtr.bTemporalCompression = FALSE;
-        mtr.pUnk                 = NULL;
-        mtr.formattype           = FORMAT_VideoInfo;
-        mtr.cbFormat             = sizeof(vh);
-        mtr.pbFormat             = (BYTE *)&vh;
-
-        memset(&vh, 0, sizeof(vh));
-
-        vh.bmiHeader.biSize   = sizeof(vh.bmiHeader);
-        vh.bmiHeader.biWidth  = p_sys->i_width > 0 ? p_sys->i_width : 320;
-        vh.bmiHeader.biHeight = p_sys->i_height > 0 ? p_sys->i_height : 240;
-        vh.bmiHeader.biPlanes      = 3;
-        vh.bmiHeader.biBitCount    = 12;
-        vh.bmiHeader.biCompression = VLC_FOURCC('I','4','2','0');
-        vh.bmiHeader.biSizeImage   = vh.bmiHeader.biWidth * 12 *
-            vh.bmiHeader.biHeight / 8;
-        mtr.lSampleSize            = vh.bmiHeader.biSizeImage;
-
-        mt_count = 1;
-        mt = (AM_MEDIA_TYPE *)malloc( sizeof(AM_MEDIA_TYPE)*mt_count );
-        CopyMediaType(mt, &mtr);
-    }
-    else if( !b_stream_type )
-    {
-        // Insert prefered audio media type
-        AM_MEDIA_TYPE mtr;
-        WAVEFORMATEX wf;
-
-        mtr.majortype            = MEDIATYPE_Audio;
-        mtr.subtype              = MEDIASUBTYPE_PCM;
-        mtr.bFixedSizeSamples    = TRUE;
-        mtr.bTemporalCompression = FALSE;
-        mtr.lSampleSize          = 0;
-        mtr.pUnk                 = NULL;
-        mtr.formattype           = FORMAT_WaveFormatEx;
-        mtr.cbFormat             = sizeof(wf);
-        mtr.pbFormat             = (BYTE *)&wf;
-
-        memset(&wf, 0, sizeof(wf));
-
-        wf.wFormatTag = WAVE_FORMAT_PCM;
-        wf.nChannels = 2;
-        wf.nSamplesPerSec = 44100;
-        wf.wBitsPerSample = 16;
-        wf.nBlockAlign = wf.nSamplesPerSec * wf.wBitsPerSample / 8;
-        wf.nAvgBytesPerSec = wf.nSamplesPerSec * wf.nBlockAlign;
-        wf.cbSize = 0;
-
-        mt_count = 1;
-        mt = (AM_MEDIA_TYPE *)malloc( sizeof(AM_MEDIA_TYPE)*mt_count );
-        CopyMediaType(mt, &mtr);
-    }
-
     if( media_count > 0 )
     {
-        mt = (AM_MEDIA_TYPE *)realloc( mt, sizeof(AM_MEDIA_TYPE) *
-                                       (mt_count + media_count) );
+        mt = (AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE) * media_count);
 
         // Order and copy returned media types according to arbitrary
         // fourcc priority
@@ -917,20 +857,25 @@ static int OpenDevice( vlc_object_t *p_this, access_sys_t *p_sys,
             }
             if( slot_copy != c )
             {
-                mt[c+mt_count] = media_types[slot_copy];
+                mt[c] = media_types[slot_copy];
                 media_types[slot_copy] = media_types[c];
             }
             else
             {
-                mt[c+mt_count] = media_types[c];
+                mt[c] = media_types[c];
             }
         }
-        mt_count += media_count;
+    }
+    else {
+        /* capture device */
+        msg_Err( p_this, "capture device '%s' does not support required parameters !", devicename.c_str() );
+        p_device_filter->Release();
+        return VLC_EGENERIC;
     }
 
     /* Create and add our capture filter */
     CaptureFilter *p_capture_filter =
-        new CaptureFilter( p_this, p_sys, mt, mt_count );
+        new CaptureFilter( p_this, p_sys, mt, media_count );
     p_sys->p_graph->AddFilter( p_capture_filter, 0 );
 
     /* Add the device filter to the graph (seems necessary with VfW before
@@ -1093,11 +1038,13 @@ FindCaptureDevice( vlc_object_t *p_this, string *p_devicename,
             p_bag->Release();
             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 );
+                int i_convert = WideCharToMultiByte(CP_ACP, 0, var.bstrVal,
+                        SysStringLen(var.bstrVal), NULL, 0, NULL, NULL);
+                char *p_buf = (char *)alloca( i_convert+1 ); p_buf[0] = 0;
+                WideCharToMultiByte( CP_ACP, 0, var.bstrVal,
+                        SysStringLen(var.bstrVal), p_buf, i_convert, NULL, NULL );
                 SysFreeString(var.bstrVal);
+                p_buf[i_convert] = '\0';
 
                 if( p_listdevices ) p_listdevices->push_back( p_buf );
 
@@ -1139,6 +1086,11 @@ static size_t EnumDeviceCaps( vlc_object_t *p_this, IBaseFilter *p_filter,
     IEnumMediaTypes *p_enummt;
     size_t mt_count = 0;
 
+    LONGLONG i_AvgTimePerFrame = 0;
+    float r_fps = var_GetFloat( p_this, "dshow-fps" );
+    if( r_fps )
+        i_AvgTimePerFrame = 10000000000LL/(LONGLONG)(r_fps*1000.0f);
+
     if( FAILED(p_filter->EnumPins( &p_enumpins )) )
     {
         msg_Dbg( p_this, "EnumDeviceCaps failed: no pin enumeration !");
@@ -1177,34 +1129,207 @@ static size_t EnumDeviceCaps( vlc_object_t *p_this, IBaseFilter *p_filter,
             msg_Dbg( p_this, "EnumDeviceCaps: trying pin %S", info.achName );
         }
 
-        /* Probe pin */
-        if( !SUCCEEDED( p_output_pin->EnumMediaTypes( &p_enummt ) ) )
+        AM_MEDIA_TYPE *p_mt;
+
+        /*
+        ** Configure pin with a default compatible media if possible
+        */
+
+        IAMStreamConfig *pSC;
+        if( SUCCEEDED(p_output_pin->QueryInterface( IID_IAMStreamConfig,
+                                            (void **)&pSC )) )
+        {
+            int piCount, piSize;
+            if( SUCCEEDED(pSC->GetNumberOfCapabilities(&piCount, &piSize)) )
+            {
+                BYTE *pSCC= (BYTE *)CoTaskMemAlloc(piSize);
+                if( NULL != pSCC )
+                {
+                    int i_priority = -1;
+                    for( int i=0; i<piCount; ++i )
+                    {
+                        if( SUCCEEDED(pSC->GetStreamCaps(i, &p_mt, pSCC)) )
+                        {
+                            int i_current_fourcc = GetFourCCFromMediaType( *p_mt );
+                            int i_current_priority = GetFourCCPriority(i_current_fourcc);
+
+                            if( !i_current_fourcc
+                             || (i_fourcc && (i_current_fourcc != i_fourcc))
+                             || (i_priority > i_current_priority) )
+                            {
+                                // incompatible or unwanted chroma, try next media type
+                                FreeMediaType( *p_mt );
+                                CoTaskMemFree( (PVOID)p_mt );
+                                continue;
+                            }
+
+                            if( MEDIATYPE_Video == p_mt->majortype
+                                    && FORMAT_VideoInfo == p_mt->formattype )
+                            {
+                                VIDEO_STREAM_CONFIG_CAPS *pVSCC = reinterpret_cast<VIDEO_STREAM_CONFIG_CAPS*>(pSCC);
+                                VIDEOINFOHEADER *pVih = reinterpret_cast<VIDEOINFOHEADER*>(p_mt->pbFormat);
+
+                                if( i_AvgTimePerFrame )
+                                {
+                                    if( pVSCC->MinFrameInterval > i_AvgTimePerFrame
+                                      || i_AvgTimePerFrame > pVSCC->MaxFrameInterval )
+                                    {
+                                        // required frame rate not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pVih->AvgTimePerFrame = i_AvgTimePerFrame;
+                                }
+
+                                if( i_width )
+                                {
+                                    if( i_width % pVSCC->OutputGranularityX
+                                     || pVSCC->MinOutputSize.cx > i_width
+                                     || i_width > pVSCC->MaxOutputSize.cx )
+                                    {
+                                        // required width not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pVih->bmiHeader.biWidth = i_width;
+                                }
+
+                                if( i_height )
+                                {
+                                    if( i_height % pVSCC->OutputGranularityY
+                                     || pVSCC->MinOutputSize.cy > i_height 
+                                     || i_height > pVSCC->MaxOutputSize.cy )
+                                    {
+                                        // required height not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pVih->bmiHeader.biHeight = i_height;
+                                }
+
+                                // Set the sample size and image size.
+                                // (Round the image width up to a DWORD boundary.)
+                                p_mt->lSampleSize = pVih->bmiHeader.biSizeImage = 
+                                    ((pVih->bmiHeader.biWidth + 3) & ~3) *
+                                    pVih->bmiHeader.biHeight * (pVih->bmiHeader.biBitCount>>3);
+
+                                // no cropping, use full video input buffer
+                                memset(&(pVih->rcSource), 0, sizeof(RECT));
+                                memset(&(pVih->rcTarget), 0, sizeof(RECT));
+
+                                // select this format as default
+                                if( SUCCEEDED( pSC->SetFormat(p_mt) ) )
+                                {
+                                    i_priority = i_current_priority;
+                                    if( i_fourcc )
+                                        // no need to check any more media types 
+                                        i = piCount;
+                                }
+                            }
+                            else if( p_mt->majortype == MEDIATYPE_Audio
+                                    && p_mt->formattype == FORMAT_WaveFormatEx )
+                            {
+                                AUDIO_STREAM_CONFIG_CAPS *pASCC = reinterpret_cast<AUDIO_STREAM_CONFIG_CAPS*>(pSCC);
+                                WAVEFORMATEX *pWfx = reinterpret_cast<WAVEFORMATEX*>(p_mt->pbFormat);
+
+                                if( i_channels )
+                                {
+                                    if( i_channels % pASCC->ChannelsGranularity
+                                     || (unsigned int)i_channels < pASCC->MinimumChannels
+                                     || (unsigned int)i_channels > pASCC->MaximumChannels )
+                                    {
+                                        // required channels not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pWfx->nChannels = i_channels;
+                                }
+
+                                if( i_samplespersec )
+                                {
+                                    if( i_samplespersec % pASCC->BitsPerSampleGranularity
+                                     || (unsigned int)i_samplespersec < pASCC->MinimumSampleFrequency
+                                     || (unsigned int)i_samplespersec > pASCC->MaximumSampleFrequency )
+                                    {
+                                        // required sampling rate not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pWfx->nSamplesPerSec = i_samplespersec;
+                                }
+
+                                if( i_bitspersample )
+                                {
+                                    if( i_bitspersample % pASCC->BitsPerSampleGranularity
+                                     || (unsigned int)i_bitspersample < pASCC->MinimumBitsPerSample
+                                     || (unsigned int)i_bitspersample > pASCC->MaximumBitsPerSample )
+                                    {
+                                        // required sample size not compatible, try next media type
+                                        FreeMediaType( *p_mt );
+                                        CoTaskMemFree( (PVOID)p_mt );
+                                        continue;
+                                    }
+                                    pWfx->wBitsPerSample = i_bitspersample;
+                                }
+
+                                // select this format as default
+                                if( SUCCEEDED( pSC->SetFormat(p_mt) ) )
+                                {
+                                    i_priority = i_current_priority;
+                                    if( i_fourcc )
+                                        // no need to check any more media types 
+                                        i = piCount;
+                                }
+                            }
+                            FreeMediaType( *p_mt );
+                            CoTaskMemFree( (PVOID)p_mt );
+                        }
+                    }
+                    CoTaskMemFree( (LPVOID)pSCC );
+                    if( i_priority >= 0 )
+                        msg_Dbg( p_this, "EnumDeviceCaps: input pin default format configured");
+                }
+            }
+            pSC->Release();
+        }
+
+        /*
+        ** Probe pin for available medias (may be a previously configured one)
+        */
+
+        if( FAILED( p_output_pin->EnumMediaTypes( &p_enummt ) ) )
         {
             p_output_pin->Release();
             continue;
         }
 
-        AM_MEDIA_TYPE *p_mt;
         while( p_enummt->Next( 1, &p_mt, NULL ) == S_OK )
         {
             int i_current_fourcc = GetFourCCFromMediaType( *p_mt );
-            if( i_current_fourcc && p_mt->majortype == MEDIATYPE_Video )
+            if( i_current_fourcc && p_mt->majortype == MEDIATYPE_Video
+                && p_mt->formattype == FORMAT_VideoInfo )
             {
-                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;
+                int i_current_width = ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biWidth;
+                int i_current_height = ((VIDEOINFOHEADER *)p_mt->pbFormat)->bmiHeader.biHeight;
+                LONGLONG i_current_atpf = ((VIDEOINFOHEADER *)p_mt->pbFormat)->AvgTimePerFrame;
+
                 if( i_current_height < 0 )
                     i_current_height = -i_current_height; 
 
                 msg_Dbg( p_this, "EnumDeviceCaps: input pin "
-                         "accepts chroma: %4.4s, width:%i, height:%i",
+                         "accepts chroma: %4.4s, width:%i, height:%i, fps:%f",
                          (char *)&i_current_fourcc, i_current_width,
-                         i_current_height );
+                         i_current_height, (10000000.0f/((float)i_current_atpf)) );
 
                 if( ( !i_fourcc || i_fourcc == i_current_fourcc ) &&
                     ( !i_width || i_width == i_current_width ) &&
                     ( !i_height || i_height == i_current_height ) &&
+                    ( !i_AvgTimePerFrame || i_AvgTimePerFrame == i_current_atpf ) &&
                     mt_count < mt_max )
                 {
                     /* Pick match */
@@ -1212,7 +1337,8 @@ static size_t EnumDeviceCaps( vlc_object_t *p_this, IBaseFilter *p_filter,
                 }
                 else FreeMediaType( *p_mt );
             }
-            else if( i_current_fourcc && p_mt->majortype == MEDIATYPE_Audio )
+            else if( i_current_fourcc && p_mt->majortype == MEDIATYPE_Audio 
+                    && p_mt->formattype == FORMAT_WaveFormatEx)
             {
                 int i_current_channels =
                     ((WAVEFORMATEX *)p_mt->pbFormat)->nChannels;
@@ -1279,7 +1405,7 @@ static size_t EnumDeviceCaps( vlc_object_t *p_this, IBaseFilter *p_filter,
                 if( p_mt->majortype == MEDIATYPE_Video ) psz_type = "video";
                 if( p_mt->majortype == MEDIATYPE_Audio ) psz_type = "audio";
                 if( p_mt->majortype == MEDIATYPE_Stream ) psz_type = "stream";
-                msg_Dbg( p_this, "EnumDeviceCaps: input pin: unknown format "
+                msg_Dbg( p_this, "EnumDeviceCaps: input pin media: unknown format "
                          "(%s %4.4s)", psz_type, (char *)&p_mt->subtype );
                 FreeMediaType( *p_mt );
             }
@@ -1382,14 +1508,16 @@ static int Demux( demux_t *p_demux )
     }
     /* Try to grab a video sample */
     if( i_stream == p_sys->i_streams )
-    for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
     {
-        p_stream = p_sys->pp_streams[i_stream];
-        if( p_stream->p_capture_filter &&
-            p_stream->p_capture_filter->CustomGetPin()
-                ->CustomGetSample( &sample ) == S_OK )
+        for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
         {
-            break;
+            p_stream = p_sys->pp_streams[i_stream];
+            if( p_stream->p_capture_filter &&
+                p_stream->p_capture_filter->CustomGetPin()
+                    ->CustomGetSample( &sample ) == S_OK )
+            {
+                break;
+            }
         }
     }