X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Faccess%2Fqtcapture.m;h=ce41a0d6e72d22b0da849a033e8da9008d7c7db0;hb=cff6fc15e4ddc8e31ddd2d7584c7d96ff74c19b0;hp=a39b7a2fb0e913fb251b07079e4f7643beed9e35;hpb=90d9d6dda3dc1a6260ac5e3eb355c83b377da201;p=vlc diff --git a/modules/access/qtcapture.m b/modules/access/qtcapture.m index a39b7a2fb0..ce41a0d6e7 100644 --- a/modules/access/qtcapture.m +++ b/modules/access/qtcapture.m @@ -1,42 +1,49 @@ /***************************************************************************** -* qtcapture.m: qtkit (Mac OS X) based capture module -***************************************************************************** -* Copyright (C) 2008 the VideoLAN team -* -* Authors: Pierre d'Herbemont -* -***************************************************************************** -* 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 + * + **************************************************************************** + * 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" #endif -#include +#include #include #include -#include #include +#include +#include +#include #import +#import + +#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 @@ -49,15 +56,19 @@ static int Control( demux_t *, int, va_list ); /***************************************************************************** * Module descriptor *****************************************************************************/ -vlc_module_begin(); - set_shortname( N_("Quicktime Capture") ); - set_description( N_("Quicktime Capture") ); - set_category( CAT_INPUT ); - set_subcategory( SUBCAT_INPUT_ACCESS ); - add_shortcut( "qtcapture" ); - set_capability( "access_demux", 10 ); - set_callbacks( Open, Close ); -vlc_module_end(); +vlc_module_begin () + set_shortname( N_("Quicktime Capture") ) + set_description( N_("Quicktime Capture") ) + set_category( CAT_INPUT ) + set_subcategory( SUBCAT_INPUT_ACCESS ) + 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 () /***************************************************************************** @@ -65,64 +76,96 @@ vlc_module_end(); *****************************************************************************/ @interface VLCDecompressedVideoOutput : QTCaptureDecompressedVideoOutput { - CVImageBufferRef currentImageBuffer; + CVImageBufferRef currentImageBuffer; + mtime_t currentPts; + mtime_t previousPts; + long timeScale; } - (id)init; - (void)outputVideoFrame:(CVImageBufferRef)videoFrame withSampleBuffer:(QTSampleBuffer *)sampleBuffer fromConnection:(QTCaptureConnection *)connection; -- (BOOL)copyCurrentFrameToBuffer:(void *)buffer; +- (mtime_t)copyCurrentFrameToBuffer:(void *)buffer; @end /* Apple sample code */ @implementation VLCDecompressedVideoOutput : QTCaptureDecompressedVideoOutput - (id)init { - if( self = [super init] ) - { - currentImageBuffer = nil; - } - return self; + if( self = [super init] ) + { + currentImageBuffer = nil; + currentPts = 0; + previousPts = 0; + timeScale = 0; + } + return self; } - (void)dealloc { - @synchronized (self) { - CVBufferRelease(currentImageBuffer); - currentImageBuffer = nil; - } - [super dealloc]; + @synchronized (self) + { + CVBufferRelease(currentImageBuffer); + currentImageBuffer = nil; + } + [super dealloc]; +} + +- (long)timeScale +{ + return timeScale; } - (void)outputVideoFrame:(CVImageBufferRef)videoFrame withSampleBuffer:(QTSampleBuffer *)sampleBuffer fromConnection:(QTCaptureConnection *)connection { - // Store the latest frame - // This must be done in a @synchronized block because this delegate method is not called on the main thread - CVImageBufferRef imageBufferToRelease; + // Store the latest frame + // This must be done in a @synchronized block because this delegate method is not called on the main thread + CVImageBufferRef imageBufferToRelease; - CVBufferRetain(videoFrame); + CVBufferRetain(videoFrame); - @synchronized (self) { - imageBufferToRelease = currentImageBuffer; - currentImageBuffer = videoFrame; - } - CVBufferRelease(imageBufferToRelease); + @synchronized (self) + { + imageBufferToRelease = currentImageBuffer; + currentImageBuffer = videoFrame; + 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]; + if( hosttime ) currentPts = (mtime_t)AudioConvertHostTimeToNanos([hosttime unsignedLongLongValue])/1000; + } + CVBufferRelease(imageBufferToRelease); } -- (BOOL)copyCurrentFrameToBuffer:(void *)buffer +- (mtime_t)copyCurrentFrameToBuffer:(void *)buffer { - CVImageBufferRef imageBuffer; + CVImageBufferRef imageBuffer; + mtime_t pts; - @synchronized (self) { - if(!currentImageBuffer) return NO; - imageBuffer = CVBufferRetain(currentImageBuffer); - } +void * pixels; - CVPixelBufferLockBaseAddress(imageBuffer, 0); - void * pixels = CVPixelBufferGetBaseAddress(imageBuffer); - memcpy( buffer, pixels, CVPixelBufferGetBytesPerRow(imageBuffer) * CVPixelBufferGetHeight(imageBuffer) ); - CVPixelBufferUnlockBaseAddress(imageBuffer, 0); + if(!currentImageBuffer || currentPts == previousPts ) + return 0; + + @synchronized (self) + { + imageBuffer = CVBufferRetain(currentImageBuffer); + if(imageBuffer){ + pts = previousPts = currentPts; + CVPixelBufferLockBaseAddress(imageBuffer, 0); + pixels = CVPixelBufferGetBaseAddress(imageBuffer); + if(pixels) + memcpy( buffer, pixels, CVPixelBufferGetBytesPerRow(imageBuffer) * CVPixelBufferGetHeight(imageBuffer)); + CVPixelBufferUnlockBaseAddress(imageBuffer, 0); + } - CVBufferRelease(imageBuffer); + } + CVBufferRelease(imageBuffer); - return YES; + if(pixels) + return currentPts; + else + return 0; } @end @@ -132,10 +175,13 @@ vlc_module_end(); *****************************************************************************/ struct demux_sys_t { - QTCaptureSession * session; - VLCDecompressedVideoOutput * output; - int height, width; - es_out_id_t * p_es_video; + QTCaptureSession * session; + QTCaptureDevice * device; + VLCDecompressedVideoOutput * output; + int height, width; + es_out_id_t * p_es_video; + BOOL b_es_setup; + es_format_t fmt; }; @@ -144,14 +190,16 @@ struct demux_sys_t { *****************************************************************************/ static int qtchroma_to_fourcc( int i_qt ) { - static struct + static const struct { unsigned int i_qt; int i_fourcc; } qtchroma_to_fourcc[] = { /* Raw data types */ - { k422YpCbCr8CodecType, VLC_FOURCC('U','Y','V','Y') }, + { '2vuy', VLC_CODEC_UYVY }, + { 'yuv2',VLC_CODEC_YUYV }, + { 'yuvs', VLC_CODEC_YUYV }, { 0, 0 } }; int i; @@ -170,12 +218,22 @@ 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 i_aspect; int result = 0; + char *psz_uid = NULL; + + /* Only when selected */ + if( *p_demux->psz_access == '\0' ) + return VLC_EGENERIC; + + 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; @@ -184,85 +242,145 @@ static int Open( vlc_object_t *p_this ) p_demux->info.i_title = 0; p_demux->info.i_seekpoint = 0; - NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init]; + p_demux->p_sys = p_sys = calloc( 1, sizeof( demux_sys_t ) ); + if( !p_sys ) + return VLC_ENOMEM; - msg_Dbg( p_demux, "QTCapture Probed" ); + 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" ); - p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) ); - if( !p_sys ) return VLC_ENOMEM; + 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, 0, sizeof( demux_sys_t ) ); - memset( &fmt, 0, sizeof( es_format_t ) ); + memset( &p_sys->fmt, 0, sizeof( es_format_t ) ); QTCaptureDeviceInput * input = nil; - QTCaptureSession * session = nil; - VLCDecompressedVideoOutput * output = nil; - - QTCaptureDevice * device = [QTCaptureDevice defaultInputDeviceWithMediaType: QTMediaTypeVideo]; - if( !device ) + NSError *o_returnedError; + 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 ) { - msg_Err( p_demux, "Can't open any Video device" ); + 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; } - if( ![device open: nil /* FIXME */] ) + if( ![p_sys->device open: &o_returnedError] ) { - msg_Err( p_demux, "Can't open any Video device" ); + msg_Err( p_demux, "Unable to open the capture device (%ld)", [o_returnedError code] ); goto error; } - input = [[QTCaptureDeviceInput alloc] initWithDevice: device]; - if( !device ) + if( [p_sys->device isInUseByAnotherApplication] == YES ) { - msg_Err( p_demux, "Can't create a capture session" ); + msg_Err( p_demux, "default capture device is exclusively in use by another application" ); goto error; } - output = [[VLCDecompressedVideoOutput alloc] init]; - - session = [[QTCaptureSession alloc] init]; - - bool ret = [session addInput:input error:nil /* FIXME */]; - if( !ret ) + input = [[QTCaptureDeviceInput alloc] initWithDevice: p_sys->device]; + if( !input ) { - msg_Err( p_demux, "Can't add the video device as input" ); + msg_Err( p_demux, "can't create a valid capture input facility" ); goto error; } - ret = [session addOutput:output error:nil /* FIXME */]; - if( !ret ) + p_sys->output = [[VLCDecompressedVideoOutput alloc] init]; + + /* Get the formats */ + NSArray *format_array = [p_sys->device formatDescriptions]; + QTFormatDescription* camera_format = NULL; + NSUInteger formatCount = [format_array count]; + for( NSUInteger k = 0; k < formatCount; k++ ) { - msg_Err( p_demux, "Can't get any output output" ); - goto error; + camera_format = [format_array objectAtIndex: k]; + + msg_Dbg(p_demux, "localized Format: %s", [[camera_format localizedFormatSummary] UTF8String] ); + msg_Dbg(p_demux, "format description: %s", [[[camera_format formatDescriptionAttributes] description] UTF8String] ); } + if( [format_array count] ) + camera_format = [format_array objectAtIndex: 0]; + else goto error; + + int qtchroma = [camera_format formatType]; + int chroma = VLC_CODEC_UYVY; - [session startRunning]; + /* Now we can init */ + es_format_Init( &p_sys->fmt, VIDEO_ES, chroma ); - int qtchroma = [[[device formatDescriptions] objectAtIndex: 0] formatType]; /* FIXME */ - int chroma = qtchroma_to_fourcc( qtchroma ); - if( !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]; + + 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 ) { - msg_Err( p_demux, "Unknown qt chroma %4.4s provided by camera", (char*)&qtchroma ); - goto error; + 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; } - es_format_Init( &fmt, VIDEO_ES, chroma ); + msg_Dbg(p_demux, "encoded_size %i %i", (int)encoded_size.width, (int)encoded_size.height ); + msg_Dbg(p_demux, "display_size %i %i", (int)display_size.width, (int)display_size.height ); + 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 - NSSize size = [[device attributeForKey:QTFormatDescriptionVideoEncodedPixelsSizeAttribute] sizeValue]; - p_sys->width = fmt.video.i_width = 640;/* size.width; FIXME */ - p_sys->height = fmt.video.i_height = 480;/* size.height; FIXME */ + p_sys->session = [[QTCaptureSession alloc] init]; - msg_Dbg( p_demux, "added new video es %4.4s %dx%d", - (char*)&fmt.i_codec, fmt.video.i_width, fmt.video.i_height ); + bool ret = [p_sys->session addInput:input error: &o_returnedError]; + if( !ret ) + { + msg_Err( p_demux, "default video capture device could not be added to capture session (%ld)", [o_returnedError code] ); + goto error; + } - p_sys->p_es_video = es_out_Add( p_demux->out, &fmt ); + ret = [p_sys->session addOutput:p_sys->output error: &o_returnedError]; + if( !ret ) + { + msg_Err( p_demux, "output could not be added to capture session (%ld)", [o_returnedError code] ); + goto error; + } - p_sys->output = [output retain]; - p_sys->session = [session retain]; + [p_sys->session startRunning]; [input release]; - [output release]; - [session release]; [pool release]; msg_Dbg( p_demux, "QTCapture: We have a video device ready!" ); @@ -270,9 +388,6 @@ static int Open( vlc_object_t *p_this ) return VLC_SUCCESS; error: [input release]; - [session release]; - [input release]; - [output release]; [pool release]; free( p_sys ); @@ -289,8 +404,18 @@ static void Close( vlc_object_t *p_this ) demux_t *p_demux = (demux_t*)p_this; demux_sys_t *p_sys = p_demux->p_sys; - [p_sys->output release]; - [p_sys->session release]; + + /* Hack: if libvlc was killed, main interface thread was, + * and poor QTKit needs it, so don't tell him. + * Else we dead lock. */ + if( vlc_object_alive(p_this->p_libvlc)) + { + // Perform this on main thread, as the framework itself will sometimes try to synchronously + // work on main thread. And this will create a dead lock. + [p_sys->session performSelectorOnMainThread:@selector(stopRunning) withObject:nil waitUntilDone:NO]; + [p_sys->output performSelectorOnMainThread:@selector(release) withObject:nil waitUntilDone:NO]; + [p_sys->session performSelectorOnMainThread:@selector(release) withObject:nil waitUntilDone:NO]; + } free( p_sys ); [pool release]; @@ -305,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" ); @@ -315,15 +439,27 @@ static int Demux( demux_t *p_demux ) NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init]; - if( ![p_sys->output copyCurrentFrameToBuffer: p_block->p_buffer] ) + @synchronized (p_sys->output) + { + p_block->i_pts = [p_sys->output copyCurrentFrameToBuffer: p_block->p_buffer]; + } + + if( !p_block->i_pts ) { /* Nothing to display yet, just forget */ block_Release( p_block ); [pool release]; + msleep( 10000 ); return 1; } - - p_block->i_pts = mdate(); /* FIXME */ + 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 ); @@ -353,15 +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; + pi64 = (int64_t*)va_arg( args, int64_t * ); + *pi64 = mdate(); + return VLC_SUCCESS; - /* TODO implement others */ default: return VLC_EGENERIC; }