2011-04-17 02:42:00 +00:00
/* GStreamer
* Copyright ( C ) 2011 David Schleef < ds @ entropywave . com >
*
* This library is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation ; either
* version 2 of the License , or ( at your option ) any later version .
*
* 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
* Library General Public License for more details .
*
* You should have received a copy of the GNU Library General Public
* License along with this library ; if not , write to the
* Free Software Foundation , Inc . , 51 Franklin Street , Suite 500 ,
* Boston , MA 02110 - 1335 , USA .
*/
/**
* SECTION : element - gstavcsrc
*
* The avcsrc element captures video from an OS / X AVC Video Services
* devices , typically a FireWire camera .
*
* < refsect2 >
* < title > Example launch line < / title >
* | [
* gst - launch - v avcsrc ! decodebin ! osxvideosink
* ] |
*
* This pipeline captures from an AVC source , decodes the stream ( either
* DV or HDV ) , and displays the video .
* < / refsect2 >
*/
# ifdef HAVE_CONFIG_H
# include "config.h"
# endif
# include <AVCVideoServices/AVCVideoServices.h>
using namespace AVS ;
# include <gst/gst.h>
# include <gst/base/gstbasesrc.h>
# include "gstavcsrc.h"
GST_DEBUG_CATEGORY_STATIC ( gst_avc_src_debug_category ) ;
# define GST_CAT_DEFAULT gst_avc_src_debug_category
/* prototypes */
static void gst_avc_src_set_property ( GObject * object ,
guint property_id , const GValue * value , GParamSpec * pspec ) ;
static void gst_avc_src_get_property ( GObject * object ,
guint property_id , GValue * value , GParamSpec * pspec ) ;
static void gst_avc_src_dispose ( GObject * object ) ;
static void gst_avc_src_finalize ( GObject * object ) ;
static GstCaps * gst_avc_src_get_caps ( GstBaseSrc * src ) ;
static gboolean gst_avc_src_start ( GstBaseSrc * src ) ;
static gboolean gst_avc_src_stop ( GstBaseSrc * src ) ;
static gboolean gst_avc_src_is_seekable ( GstBaseSrc * src ) ;
static gboolean gst_avc_src_unlock ( GstBaseSrc * src ) ;
static gboolean gst_avc_src_event ( GstBaseSrc * src , GstEvent * event ) ;
static GstFlowReturn
gst_avc_src_create ( GstBaseSrc * src , guint64 offset , guint size ,
GstBuffer * * buf ) ;
static gboolean gst_avc_src_query ( GstBaseSrc * src , GstQuery * query ) ;
static gboolean gst_avc_src_unlock_stop ( GstBaseSrc * src ) ;
enum
{
PROP_0
} ;
/* pad templates */
static GstStaticPadTemplate gst_avc_src_src_template =
GST_STATIC_PAD_TEMPLATE ( " src " ,
GST_PAD_SRC ,
GST_PAD_ALWAYS ,
GST_STATIC_CAPS
( " video/dv,systemstream=true;video/mpegts,systemstream=true,packetsize=188 " )
) ;
/* class initialization */
# define DEBUG_INIT(bla) \
GST_DEBUG_CATEGORY_INIT ( gst_avc_src_debug_category , " avcsrc " , 0 , \
" debug category for avcsrc element " ) ;
GST_BOILERPLATE_FULL ( GstAVCSrc , gst_avc_src , GstBaseSrc ,
GST_TYPE_BASE_SRC , DEBUG_INIT ) ;
static void
gst_avc_src_base_init ( gpointer g_class )
{
GstElementClass * element_class = GST_ELEMENT_CLASS ( g_class ) ;
gst_element_class_add_pad_template ( element_class ,
gst_static_pad_template_get ( & gst_avc_src_src_template ) ) ;
2012-10-17 16:34:26 +00:00
gst_element_class_set_static_metadata ( element_class ,
2011-04-17 02:42:00 +00:00
" AVC Video Services Source " , " Video/Source " ,
" Captures DV or HDV video from Firewire port " ,
" David Schleef <ds@entropywave.com> " ) ;
}
static void
gst_avc_src_class_init ( GstAVCSrcClass * klass )
{
GObjectClass * gobject_class = G_OBJECT_CLASS ( klass ) ;
GstBaseSrcClass * base_src_class = GST_BASE_SRC_CLASS ( klass ) ;
gobject_class - > set_property = gst_avc_src_set_property ;
gobject_class - > get_property = gst_avc_src_get_property ;
gobject_class - > dispose = gst_avc_src_dispose ;
gobject_class - > finalize = gst_avc_src_finalize ;
base_src_class - > get_caps = GST_DEBUG_FUNCPTR ( gst_avc_src_get_caps ) ;
base_src_class - > start = GST_DEBUG_FUNCPTR ( gst_avc_src_start ) ;
base_src_class - > stop = GST_DEBUG_FUNCPTR ( gst_avc_src_stop ) ;
base_src_class - > is_seekable = GST_DEBUG_FUNCPTR ( gst_avc_src_is_seekable ) ;
base_src_class - > unlock = GST_DEBUG_FUNCPTR ( gst_avc_src_unlock ) ;
base_src_class - > event = GST_DEBUG_FUNCPTR ( gst_avc_src_event ) ;
base_src_class - > create = GST_DEBUG_FUNCPTR ( gst_avc_src_create ) ;
if ( 0 )
base_src_class - > query = GST_DEBUG_FUNCPTR ( gst_avc_src_query ) ;
if ( 0 )
base_src_class - > unlock_stop = GST_DEBUG_FUNCPTR ( gst_avc_src_unlock_stop ) ;
}
static void
gst_avc_src_init ( GstAVCSrc * avcsrc , GstAVCSrcClass * avcsrc_class )
{
2011-04-18 19:15:28 +00:00
gst_base_src_set_live ( GST_BASE_SRC ( avcsrc ) , TRUE ) ;
2011-04-17 02:42:00 +00:00
avcsrc - > srcpad = gst_pad_new_from_static_template ( & gst_avc_src_src_template ,
" src " ) ;
avcsrc - > queue = gst_atomic_queue_new ( 16 ) ;
avcsrc - > cond = g_cond_new ( ) ;
avcsrc - > queue_lock = g_mutex_new ( ) ;
}
void
gst_avc_src_set_property ( GObject * object , guint property_id ,
const GValue * value , GParamSpec * pspec )
{
/* GstAVCSrc *avcsrc = GST_AVC_SRC (object); */
switch ( property_id ) {
default :
G_OBJECT_WARN_INVALID_PROPERTY_ID ( object , property_id , pspec ) ;
break ;
}
}
void
gst_avc_src_get_property ( GObject * object , guint property_id ,
GValue * value , GParamSpec * pspec )
{
/* GstAVCSrc *avcsrc = GST_AVC_SRC (object); */
switch ( property_id ) {
default :
G_OBJECT_WARN_INVALID_PROPERTY_ID ( object , property_id , pspec ) ;
break ;
}
}
void
gst_avc_src_dispose ( GObject * object )
{
/* GstAVCSrc *avcsrc = GST_AVC_SRC (object); */
/* clean up as possible. may be called multiple times */
G_OBJECT_CLASS ( parent_class ) - > dispose ( object ) ;
}
void
gst_avc_src_finalize ( GObject * object )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( object ) ;
/* clean up object here */
gst_atomic_queue_unref ( avcsrc - > queue ) ;
g_cond_free ( avcsrc - > cond ) ;
g_mutex_free ( avcsrc - > queue_lock ) ;
G_OBJECT_CLASS ( parent_class ) - > finalize ( object ) ;
}
static GstCaps *
gst_avc_src_get_caps ( GstBaseSrc * src )
{
/* GstAVCSrc *avcsrc = GST_AVC_SRC (src); */
return gst_caps_from_string ( " video/mpegts,systemstream=true,packetsize=188 " ) ;
}
2011-04-19 19:01:15 +00:00
# define kNumCyclesInMPEGReceiverSegment 200
# define kNumSegmentsInMPEGReceiverProgram 10
2011-04-17 02:42:00 +00:00
void
MPEGReceiverMessageReceivedProc ( UInt32 msg , UInt32 param1 , UInt32 param2 ,
void * pRefCon )
{
}
IOReturn
MyStructuredDataPushProc ( UInt32 CycleDataCount ,
MPEGReceiveCycleData * pCycleData , void * pRefCon )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( pRefCon ) ;
if ( avcsrc ) {
2011-04-19 19:00:53 +00:00
UInt32 numPackets = 0 ;
for ( UInt32 cycle = 0 ; cycle < CycleDataCount ; cycle + + )
numPackets + = pCycleData [ cycle ] . tsPacketCount ;
GstBuffer * buffer ;
buffer = gst_buffer_new_and_alloc ( numPackets * kMPEG2TSPacketSize ) ;
guint8 * data = GST_BUFFER_DATA ( buffer ) ;
2011-04-18 19:16:00 +00:00
for ( UInt32 cycle = 0 ; cycle < CycleDataCount ; cycle + + ) {
2011-04-19 17:53:23 +00:00
GST_LOG ( " Received cycle %lu of %lu - %lu packets (fw time %lx) " ,
cycle , CycleDataCount , pCycleData [ cycle ] . tsPacketCount ,
pCycleData [ cycle ] . fireWireTimeStamp ) ;
2011-04-18 19:16:00 +00:00
for ( UInt32 sourcePacket = 0 ; sourcePacket < pCycleData [ cycle ] . tsPacketCount ;
2011-04-17 02:42:00 +00:00
sourcePacket + + ) {
2011-04-19 19:00:53 +00:00
memcpy ( data ,
2011-04-17 02:42:00 +00:00
pCycleData [ cycle ] . pBuf [ sourcePacket ] , kMPEG2TSPacketSize ) ;
2011-04-19 19:00:53 +00:00
data + = kMPEG2TSPacketSize ;
2011-04-17 02:42:00 +00:00
2011-04-19 17:53:54 +00:00
avcsrc - > packets_enqueued + + ;
2011-04-17 02:42:00 +00:00
}
}
2011-04-19 19:00:53 +00:00
gst_atomic_queue_push ( avcsrc - > queue , buffer ) ;
2011-04-17 02:42:00 +00:00
g_mutex_lock ( avcsrc - > queue_lock ) ;
g_cond_signal ( avcsrc - > cond ) ;
g_mutex_unlock ( avcsrc - > queue_lock ) ;
}
return 0 ;
}
static gboolean
gst_avc_src_start ( GstBaseSrc * src )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GST_DEBUG_OBJECT ( avcsrc , " start " ) ;
avcsrc - > unlock = FALSE ;
// Create a AVCDeviceController
2011-04-18 19:37:57 +00:00
if ( ! avcsrc - > pAVCDeviceController )
CreateAVCDeviceController ( & avcsrc - > pAVCDeviceController ) ;
if ( ! avcsrc - > pAVCDeviceController ) {
2011-04-17 02:42:00 +00:00
// TODO: This should never happen (unless we've run out of memory), but we should handle it cleanly anyway
GST_ERROR ( " Failed to create AVC device controller. " ) ;
return FALSE ;
}
GST_INFO ( " Created AVC device controller. " ) ;
2011-04-18 19:37:57 +00:00
if ( avcsrc - > deviceIndex > = CFArrayGetCount ( avcsrc - > pAVCDeviceController - > avcDeviceArray ) ) {
GST_ERROR ( " Failed to find AVC device %d " , avcsrc - > deviceIndex ) ;
2011-04-17 02:42:00 +00:00
return FALSE ;
}
2011-04-18 19:37:57 +00:00
avcsrc - > pAVCDevice = ( AVCDevice * )
CFArrayGetValueAtIndex ( avcsrc - > pAVCDeviceController - > avcDeviceArray ,
avcsrc - > deviceIndex ) ;
2011-04-17 02:42:00 +00:00
2011-04-18 19:37:57 +00:00
if ( ! avcsrc - > pAVCDevice ) {
GST_ERROR ( " Failed to find AVC device %d " , avcsrc - > deviceIndex ) ;
2011-04-17 02:42:00 +00:00
return FALSE ;
}
2011-04-18 19:37:57 +00:00
GST_INFO ( " Found device with GUID 0x%016llX \n " , avcsrc - > pAVCDevice - > guid ) ;
2011-04-17 02:42:00 +00:00
2011-04-18 19:37:57 +00:00
avcsrc - > pAVCDevice - > openDevice ( nil , nil ) ;
2011-04-17 02:42:00 +00:00
2011-04-18 19:37:57 +00:00
avcsrc - > pAVCDeviceStream = avcsrc - > pAVCDevice - > CreateMPEGReceiverForDevicePlug ( 0 , nil , // We'll install the structured callback later (MyStructuredDataPushProc),
2011-04-17 02:42:00 +00:00
nil ,
MPEGReceiverMessageReceivedProc ,
nil ,
nil , kNumCyclesInMPEGReceiverSegment , kNumSegmentsInMPEGReceiverProgram ) ;
2011-04-18 19:37:57 +00:00
avcsrc - > pAVCDeviceStream - > pMPEGReceiver - > registerStructuredDataPushCallback
2011-04-17 02:42:00 +00:00
( MyStructuredDataPushProc ,
kNumCyclesInMPEGReceiverSegment , ( void * ) avcsrc ) ;
2011-04-18 19:37:57 +00:00
avcsrc - > pAVCDevice - > StartAVCDeviceStream ( avcsrc - > pAVCDeviceStream ) ;
2011-04-17 02:42:00 +00:00
return TRUE ;
}
static gboolean
gst_avc_src_stop ( GstBaseSrc * src )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GstBuffer * buffer ;
GST_DEBUG_OBJECT ( avcsrc , " stop " ) ;
2011-04-18 19:37:57 +00:00
// Stop the stream
avcsrc - > pAVCDevice - > StopAVCDeviceStream ( avcsrc - > pAVCDeviceStream ) ;
// Destroy the stream
avcsrc - > pAVCDevice - > DestroyAVCDeviceStream ( avcsrc - > pAVCDeviceStream ) ;
avcsrc - > pAVCDeviceStream = nil ;
// Forget about the device (don't destroy it; pAVCDeviceController manages it)
avcsrc - > pAVCDevice = nil ;
2011-04-17 02:42:00 +00:00
2011-04-19 17:53:54 +00:00
GST_DEBUG ( " Packets enqueued = %llu " , avcsrc - > packets_enqueued ) ;
GST_DEBUG ( " Packets dequeued = %llu " , avcsrc - > packets_dequeued ) ;
2011-04-17 02:42:00 +00:00
while ( ( buffer = GST_BUFFER ( gst_atomic_queue_pop ( avcsrc - > queue ) ) ) ! = NULL ) {
gst_buffer_unref ( buffer ) ;
}
return TRUE ;
}
static gboolean
gst_avc_src_is_seekable ( GstBaseSrc * src )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GST_DEBUG_OBJECT ( avcsrc , " is_seekable " ) ;
return FALSE ;
}
static gboolean
gst_avc_src_unlock ( GstBaseSrc * src )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GST_DEBUG_OBJECT ( avcsrc , " unlock " ) ;
g_mutex_lock ( avcsrc - > queue_lock ) ;
avcsrc - > unlock = TRUE ;
g_cond_signal ( avcsrc - > cond ) ;
g_mutex_unlock ( avcsrc - > queue_lock ) ;
return TRUE ;
}
static gboolean
gst_avc_src_event ( GstBaseSrc * src , GstEvent * event )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
2011-04-19 17:53:54 +00:00
GST_DEBUG_OBJECT ( avcsrc , " event of type '%s' " , GST_EVENT_TYPE_NAME ( event ) ) ;
GST_DEBUG ( " Packets enqueued = %llu, dequeued = %llu " ,
avcsrc - > packets_enqueued , avcsrc - > packets_dequeued ) ;
2011-04-17 02:42:00 +00:00
return TRUE ;
}
static GstFlowReturn
gst_avc_src_create ( GstBaseSrc * src , guint64 offset , guint size ,
GstBuffer * * buf )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GstBuffer * buffer ;
GST_DEBUG_OBJECT ( avcsrc , " create " ) ;
g_mutex_lock ( avcsrc - > queue_lock ) ;
buffer = GST_BUFFER ( gst_atomic_queue_pop ( avcsrc - > queue ) ) ;
while ( buffer = = NULL & & ! avcsrc - > unlock ) {
g_cond_wait ( avcsrc - > cond , avcsrc - > queue_lock ) ;
buffer = GST_BUFFER ( gst_atomic_queue_pop ( avcsrc - > queue ) ) ;
}
g_mutex_unlock ( avcsrc - > queue_lock ) ;
if ( avcsrc - > unlock ) {
if ( buffer )
gst_buffer_unref ( buffer ) ;
2012-02-08 15:37:13 +00:00
return GST_FLOW_FLUSHING ;
2011-04-17 02:42:00 +00:00
}
gst_buffer_set_caps ( buffer , GST_PAD_CAPS ( avcsrc - > srcpad ) ) ;
* buf = buffer ;
2011-04-19 17:53:54 +00:00
avcsrc - > packets_dequeued + + ;
2011-04-17 02:42:00 +00:00
return GST_FLOW_OK ;
}
static gboolean
gst_avc_src_query ( GstBaseSrc * src , GstQuery * query )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GST_DEBUG_OBJECT ( avcsrc , " query " ) ;
return TRUE ;
}
static gboolean
gst_avc_src_unlock_stop ( GstBaseSrc * src )
{
GstAVCSrc * avcsrc = GST_AVC_SRC ( src ) ;
GST_DEBUG_OBJECT ( avcsrc , " stop " ) ;
return TRUE ;
}