]> git.sesse.net Git - vlc/blobdiff - modules/access/qtcapture.m
posix: drop support for non-UTF-8 operating systems
[vlc] / modules / access / qtcapture.m
index 3c4da2cf36cb04836daa1a9165b8b92ae71e4c76..ce41a0d6e72d22b0da849a033e8da9008d7c7db0 100644 (file)
@@ -1,30 +1,29 @@
 /*****************************************************************************
-* qtcapture.m: qtkit (Mac OS X) based capture module
-*****************************************************************************
-* Copyright (C) 2008-2011 the VideoLAN team
-*
-* Authors: Pierre d'Herbemont <pdherbemont@videolan.org>
-*
-*****************************************************************************
-* This library is free software; you can redistribute it and/or
-* modify it under the terms of the GNU Lesser General Public
-* License as published by the Free Software Foundation;
-* version 2 of the License.
-*
-* This library 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
-* Lesser General Public License for more details.
-*
-* You should have received a copy of the GNU Lesser General Public
-* License along with this library; if not, write to the Free Software
-* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
-*
-*****************************************************************************/
+ * qtcapture.m: qtkit (Mac OS X) based capture module
+ *****************************************************************************
+ * Copyright © 2008-2011 VLC authors and VideoLAN
+ *
+ * Authors: Pierre d'Herbemont <pdherbemont@videolan.org>
+ *
+ ****************************************************************************
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
 
 /*****************************************************************************
-* Preamble
-*****************************************************************************/
+ * Preamble
+ *****************************************************************************/
 
 #ifdef HAVE_CONFIG_H
 # include "config.h"
 #include <vlc_demux.h>
 #include <vlc_interface.h>
 #include <vlc_dialog.h>
+#include <vlc_access.h>
 
 #import <QTKit/QTKit.h>
 #import <CoreAudio/CoreAudio.h>
 
+#define QTKIT_WIDTH_TEXT N_("Video Capture width")
+#define QTKIT_WIDTH_LONGTEXT N_("Video Capture width in pixel")
+#define QTKIT_HEIGHT_TEXT N_("Video Capture height")
+#define QTKIT_HEIGHT_LONGTEXT N_("Video Capture height in pixel")
+
 /*****************************************************************************
 * Local prototypes
 *****************************************************************************/
@@ -59,6 +64,10 @@ vlc_module_begin ()
    add_shortcut( "qtcapture" )
    set_capability( "access_demux", 10 )
    set_callbacks( Open, Close )
+   add_integer("qtcapture-width", 640, QTKIT_WIDTH_TEXT, QTKIT_WIDTH_LONGTEXT, true)
+      change_integer_range (80, 1280)
+   add_integer("qtcapture-height", 480, QTKIT_HEIGHT_TEXT, QTKIT_HEIGHT_LONGTEXT, true)
+      change_integer_range (60, 480)
 vlc_module_end ()
 
 
@@ -70,6 +79,7 @@ vlc_module_end ()
     CVImageBufferRef currentImageBuffer;
     mtime_t currentPts;
     mtime_t previousPts;
+    long timeScale;
 }
 - (id)init;
 - (void)outputVideoFrame:(CVImageBufferRef)videoFrame withSampleBuffer:(QTSampleBuffer *)sampleBuffer fromConnection:(QTCaptureConnection *)connection;
@@ -85,6 +95,7 @@ vlc_module_end ()
         currentImageBuffer = nil;
         currentPts = 0;
         previousPts = 0;
+        timeScale = 0;
     }
     return self;
 }
@@ -98,6 +109,11 @@ vlc_module_end ()
     [super dealloc];
 }
 
+- (long)timeScale
+{
+    return timeScale;
+}
+
 - (void)outputVideoFrame:(CVImageBufferRef)videoFrame withSampleBuffer:(QTSampleBuffer *)sampleBuffer fromConnection:(QTCaptureConnection *)connection
 {
     // Store the latest frame
@@ -110,7 +126,9 @@ vlc_module_end ()
     {
         imageBufferToRelease = currentImageBuffer;
         currentImageBuffer = videoFrame;
-        currentPts = (mtime_t)(1000000L / [sampleBuffer presentationTime].timeScale * [sampleBuffer presentationTime].timeValue);
+        QTTime timeStamp = [sampleBuffer presentationTime];
+        timeScale = timeStamp.timeScale;
+        currentPts = (mtime_t)(1000000L / timeScale * timeStamp.timeValue);
 
         /* Try to use hosttime of the sample if available, because iSight Pts seems broken */
         NSNumber *hosttime = (NSNumber *)[sampleBuffer attributeForKey:QTSampleBufferHostTimeAttribute];
@@ -124,23 +142,30 @@ vlc_module_end ()
     CVImageBufferRef imageBuffer;
     mtime_t pts;
 
+void * pixels;
+
     if(!currentImageBuffer || currentPts == previousPts )
         return 0;
 
     @synchronized (self)
     {
         imageBuffer = CVBufferRetain(currentImageBuffer);
-        pts = previousPts = currentPts;
+        if(imageBuffer){
+            pts = previousPts = currentPts;
+            CVPixelBufferLockBaseAddress(imageBuffer, 0);
+            pixels = CVPixelBufferGetBaseAddress(imageBuffer);
+            if(pixels)
+                memcpy( buffer, pixels, CVPixelBufferGetBytesPerRow(imageBuffer) * CVPixelBufferGetHeight(imageBuffer));
+            CVPixelBufferUnlockBaseAddress(imageBuffer, 0);
+        }
 
-        CVPixelBufferLockBaseAddress(imageBuffer, 0);
-        void * pixels = CVPixelBufferGetBaseAddress(imageBuffer);
-        memcpy( buffer, pixels, CVPixelBufferGetBytesPerRow(imageBuffer) * CVPixelBufferGetHeight(imageBuffer) );
-        CVPixelBufferUnlockBaseAddress(imageBuffer, 0);
     }
-
     CVBufferRelease(imageBuffer);
 
-    return currentPts;
+    if(pixels)
+        return currentPts;
+    else
+        return 0;
 }
 
 @end
@@ -155,6 +180,8 @@ struct demux_sys_t {
     VLCDecompressedVideoOutput * output;
     int height, width;
     es_out_id_t * p_es_video;
+    BOOL b_es_setup;
+    es_format_t fmt;
 };
 
 
@@ -191,11 +218,11 @@ static int Open( vlc_object_t *p_this )
 {
     demux_t     *p_demux = (demux_t*)p_this;
     demux_sys_t *p_sys = NULL;
-    es_format_t fmt;
     int i;
     int i_width;
     int i_height;
     int result = 0;
+    char *psz_uid = NULL;
 
     /* Only when selected */
     if( *p_demux->psz_access == '\0' )
@@ -203,6 +230,11 @@ static int Open( vlc_object_t *p_this )
 
     NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init];
 
+    if( p_demux->psz_location && *p_demux->psz_location )
+        psz_uid = strdup(p_demux->psz_location);
+    msg_Dbg( p_demux, "qtcapture uid = %s", psz_uid );
+    NSString *qtk_currdevice_uid = [[NSString alloc] initWithFormat:@"%s", psz_uid];
+
     /* Set up p_demux */
     p_demux->pf_demux = Demux;
     p_demux->pf_control = Control;
@@ -214,12 +246,39 @@ static int Open( vlc_object_t *p_this )
     if( !p_sys )
         return VLC_ENOMEM;
 
-    memset( &fmt, 0, sizeof( es_format_t ) );
+    NSArray *myVideoDevices = [[[QTCaptureDevice inputDevicesWithMediaType:QTMediaTypeVideo] arrayByAddingObjectsFromArray:[QTCaptureDevice inputDevicesWithMediaType:QTMediaTypeMuxed]] retain];
+    if([myVideoDevices count] == 0)
+    {
+        dialog_FatalWait( p_demux, _("No Input device found"),
+                         _("Your Mac does not seem to be equipped with a suitable input device. "
+                           "Please check your connectors and drivers.") );
+        msg_Err( p_demux, "Can't find any Video device" );
+
+        goto error;
+    }
+    NSUInteger ivideo;
+    NSUInteger deviceCount = [myVideoDevices count];
+    for(ivideo = 0; ivideo < deviceCount; ivideo++){
+        QTCaptureDevice *qtk_device;
+        qtk_device = [myVideoDevices objectAtIndex:ivideo];
+        msg_Dbg( p_demux, "qtcapture %lu/%lu %s %s", ivideo, deviceCount, [[qtk_device localizedDisplayName] UTF8String], [[qtk_device uniqueID] UTF8String]);
+        if([[[qtk_device uniqueID]stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]] isEqualToString:qtk_currdevice_uid]){
+            break;
+        }
+    }
+
+    memset( &p_sys->fmt, 0, sizeof( es_format_t ) );
 
     QTCaptureDeviceInput * input = nil;
     NSError *o_returnedError;
-
-    p_sys->device = [QTCaptureDevice defaultInputDeviceWithMediaType: QTMediaTypeVideo];
+    if( ivideo < [myVideoDevices count] )
+        p_sys->device = [myVideoDevices objectAtIndex:ivideo];
+    else
+    {
+        /* cannot found designated device, fall back to open default device */
+        msg_Dbg(p_demux, "Cannot find designated uid device as %s, falling back to default.", [qtk_currdevice_uid UTF8String]);
+        p_sys->device = [QTCaptureDevice defaultInputDeviceWithMediaType: QTMediaTypeVideo];
+    }
     if( !p_sys->device )
     {
         dialog_FatalWait( p_demux, _("No Input device found"),
@@ -254,7 +313,8 @@ static int Open( vlc_object_t *p_this )
     /* Get the formats */
     NSArray *format_array = [p_sys->device formatDescriptions];
     QTFormatDescription* camera_format = NULL;
-    for( int k = 0; k < [format_array count]; k++ )
+    NSUInteger formatCount = [format_array count];
+    for( NSUInteger k = 0; k < formatCount; k++ )
     {
         camera_format = [format_array objectAtIndex: k];
 
@@ -266,26 +326,27 @@ static int Open( vlc_object_t *p_this )
     else goto error;
 
     int qtchroma = [camera_format formatType];
-    int chroma = qtchroma_to_fourcc( qtchroma );
-    if( !chroma )
-    {
-        msg_Err( p_demux, "Unknown qt chroma %4.4s provided by camera", (char*)&qtchroma );
-        goto error;
-    }
+    int chroma = VLC_CODEC_UYVY;
 
     /* Now we can init */
-    es_format_Init( &fmt, VIDEO_ES, chroma );
+    es_format_Init( &p_sys->fmt, VIDEO_ES, chroma );
 
     NSSize encoded_size = [[camera_format attributeForKey:QTFormatDescriptionVideoEncodedPixelsSizeAttribute] sizeValue];
     NSSize display_size = [[camera_format attributeForKey:QTFormatDescriptionVideoCleanApertureDisplaySizeAttribute] sizeValue];
     NSSize par_size = [[camera_format attributeForKey:QTFormatDescriptionVideoProductionApertureDisplaySizeAttribute] sizeValue];
 
-    fmt.video.i_width = p_sys->width = encoded_size.width;
-    fmt.video.i_height = p_sys->height = encoded_size.height;
+    par_size.width = display_size.width = encoded_size.width
+        = var_InheritInteger (p_this, "qtcapture-width");
+    par_size.height = display_size.height = encoded_size.height
+        = var_InheritInteger (p_this, "qtcapture-height");
+
+    p_sys->fmt.video.i_width = p_sys->width = encoded_size.width;
+    p_sys->fmt.video.i_height = p_sys->height = encoded_size.height;
+    p_sys->fmt.video.i_frame_rate = 25.0; // cave: check with setMinimumVideoFrameInterval (see below)
     if( par_size.width != encoded_size.width )
     {
-        fmt.video.i_sar_num = (int64_t)encoded_size.height * par_size.width / encoded_size.width;
-        fmt.video.i_sar_den = encoded_size.width;
+        p_sys->fmt.video.i_sar_num = (int64_t)encoded_size.height * par_size.width / encoded_size.width;
+        p_sys->fmt.video.i_sar_den = encoded_size.width;
     }
 
     msg_Dbg(p_demux, "encoded_size %i %i", (int)encoded_size.width, (int)encoded_size.height );
@@ -293,10 +354,13 @@ static int Open( vlc_object_t *p_this )
     msg_Dbg(p_demux, "PAR size %i %i", (int)par_size.width, (int)par_size.height );
 
     [p_sys->output setPixelBufferAttributes: [NSDictionary dictionaryWithObjectsAndKeys:
+        [NSNumber numberWithUnsignedInt:kCVPixelFormatType_422YpCbCr8], (id)kCVPixelBufferPixelFormatTypeKey,
         [NSNumber numberWithInt: p_sys->height], kCVPixelBufferHeightKey,
         [NSNumber numberWithInt: p_sys->width], kCVPixelBufferWidthKey,
         [NSNumber numberWithBool:YES], (id)kCVPixelBufferOpenGLCompatibilityKey,
         nil]];
+    [p_sys->output setAutomaticallyDropsLateVideoFrames:YES];
+    [p_sys->output setMinimumVideoFrameInterval: (1/25)]; // 25 fps
 
     p_sys->session = [[QTCaptureSession alloc] init];
 
@@ -316,11 +380,6 @@ static int Open( vlc_object_t *p_this )
 
     [p_sys->session startRunning];
 
-    msg_Dbg( p_demux, "added new video es %4.4s %dx%d",
-            (char*)&fmt.i_codec, fmt.video.i_width, fmt.video.i_height );
-
-    p_sys->p_es_video = es_out_Add( p_demux->out, &fmt );
-
     [input release];
     [pool release];
 
@@ -371,8 +430,7 @@ static int Demux( demux_t *p_demux )
     demux_sys_t *p_sys = p_demux->p_sys;
     block_t *p_block;
 
-    p_block = block_New( p_demux, p_sys->width *
-                            p_sys->height * 2 /* FIXME */ );
+    p_block = block_New( p_demux, p_sys->width * p_sys->height * 2 /* FIXME */ );
     if( !p_block )
     {
         msg_Err( p_demux, "cannot get block" );
@@ -394,6 +452,14 @@ static int Demux( demux_t *p_demux )
         msleep( 10000 );
         return 1;
     }
+    else if( !p_sys->b_es_setup )
+    {
+        p_sys->fmt.video.i_frame_rate_base = [p_sys->output timeScale];
+        msg_Dbg( p_demux, "using frame rate base: %i", p_sys->fmt.video.i_frame_rate_base );
+        p_sys->p_es_video = es_out_Add( p_demux->out, &p_sys->fmt );
+        msg_Dbg( p_demux, "added new video es %4.4s %dx%d", (char*)&p_sys->fmt.i_codec, p_sys->fmt.video.i_width, p_sys->fmt.video.i_height );
+        p_sys->b_es_setup = YES;
+    }
 
     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block->i_pts );
     es_out_Send( p_demux->out, p_sys->p_es_video, p_block );
@@ -423,9 +489,14 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
 
         case DEMUX_GET_PTS_DELAY:
            pi64 = (int64_t*)va_arg( args, int64_t * );
-           *pi64 = (int64_t)DEFAULT_PTS_DELAY;
+           *pi64 = INT64_C(1000) * var_InheritInteger( p_demux, "live-caching" );
            return VLC_SUCCESS;
 
+        case DEMUX_GET_TIME:
+            pi64 = (int64_t*)va_arg( args, int64_t * );
+            *pi64 = mdate();
+            return VLC_SUCCESS;
+
         default:
            return VLC_EGENERIC;
     }