/*
- * producer_libdv.c -- a DV encoder based on libdv
+ * consumer_libdv.c -- a DV encoder based on libdv
* Copyright (C) 2003-2004 Ushodaya Enterprises Limited
* Author: Charles Yates <charles.yates@pandora.be>
*
// Local header files
#include "consumer_libdv.h"
+#include "producer_libdv.h"
// mlt Header files
#include <framework/mlt_frame.h>
#include <libdv/dv.h>
// Forward references.
+static int consumer_start( mlt_consumer this );
+static int consumer_stop( mlt_consumer this );
+static int consumer_is_stopped( mlt_consumer this );
static int consumer_encode_video( mlt_consumer this, uint8_t *dv_frame, mlt_frame frame );
static void consumer_encode_audio( mlt_consumer this, uint8_t *dv_frame, mlt_frame frame );
static void consumer_output( mlt_consumer this, uint8_t *dv_frame, int size, mlt_frame frame );
// Get properties from the consumer
mlt_properties properties = mlt_consumer_properties( this );
- // Allocate a thread
- pthread_t *thread = calloc( 1, sizeof( pthread_t ) );
-
// Assign close callback
this->close = consumer_close;
- // Assign all properties
- if ( arg == NULL || !strcmp( arg, "PAL" ) )
- mlt_properties_set_double( properties, "fps", 25 );
- else
- mlt_properties_set_double( properties, "fps", 29.97 );
+ // Interpret the argument
+ if ( arg != NULL )
+ mlt_properties_set( properties, "target", arg );
- mlt_properties_set_data( properties, "thread", thread, sizeof( pthread_t ), free, NULL );
+ // Set the encode and output handling method
mlt_properties_set_data( properties, "video", consumer_encode_video, 0, NULL, NULL );
mlt_properties_set_data( properties, "audio", consumer_encode_audio, 0, NULL, NULL );
mlt_properties_set_data( properties, "output", consumer_output, 0, NULL, NULL );
- // Create the thread (this should not happen immediately)
- mlt_properties_set_int( properties, "running", 1 );
- pthread_create( thread, NULL, consumer_thread, this );
+ // Set up start/stop/terminated callbacks
+ this->start = consumer_start;
+ this->stop = consumer_stop;
+ this->is_stopped = consumer_is_stopped;
}
else
{
return this;
}
+/** Start the consumer.
+*/
+
+static int consumer_start( mlt_consumer this )
+{
+ // Get the properties
+ mlt_properties properties = mlt_consumer_properties( this );
+
+ // Check that we're not already running
+ if ( !mlt_properties_get_int( properties, "running" ) )
+ {
+ // Allocate a thread
+ pthread_t *thread = calloc( 1, sizeof( pthread_t ) );
+ pthread_attr_t thread_attributes;
+
+ // Assign the thread to properties
+ mlt_properties_set_data( properties, "thread", thread, sizeof( pthread_t ), free, NULL );
+
+ // Set the running state
+ mlt_properties_set_int( properties, "running", 1 );
+
+ // Inherit the scheduling priority
+ pthread_attr_init( &thread_attributes );
+ pthread_attr_setinheritsched( &thread_attributes, PTHREAD_INHERIT_SCHED );
+
+ // Create the thread
+ pthread_create( thread, &thread_attributes, consumer_thread, this );
+ }
+ return 0;
+}
+
+/** Stop the consumer.
+*/
+
+static int consumer_stop( mlt_consumer this )
+{
+ // Get the properties
+ mlt_properties properties = mlt_consumer_properties( this );
+
+ // Check that we're running
+ if ( mlt_properties_get_int( properties, "running" ) )
+ {
+ // Get the thread
+ pthread_t *thread = mlt_properties_get_data( properties, "thread", NULL );
+
+ // Stop the thread
+ mlt_properties_set_int( properties, "running", 0 );
+
+ // Wait for termination
+ pthread_join( *thread, NULL );
+
+ // Close the output file :-) - this is obtuse - doesn't matter if output file
+ // exists or not - the destructor will kick in if it does
+ mlt_properties_set_data( properties, "output_file", NULL, 0, NULL, NULL );
+ }
+
+ return 0;
+}
+
+/** Determine if the consumer is stopped.
+*/
+
+static int consumer_is_stopped( mlt_consumer this )
+{
+ // Get the properties
+ mlt_properties properties = mlt_consumer_properties( this );
+ return !mlt_properties_get_int( properties, "running" );
+}
+
/** Get or create a new libdv encoder.
*/
double fps = mlt_properties_get_double( this_properties, "fps" );
// Create the encoder
- encoder = dv_encoder_new( fps != 25, 0, 0 );
+ encoder = dv_encoder_new( 0, 0, 0 );
// Encoder settings
- encoder->isPAL = fps = 25;
+ encoder->isPAL = fps == 25;
encoder->is16x9 = 0;
encoder->vlc_encode_passes = 1;
encoder->static_qno = 0;
// Store the encoder on the properties
mlt_properties_set_data( this_properties, "dv_encoder", encoder, 0, ( mlt_destructor )dv_encoder_free, NULL );
-
- // Convenience for image dimensions
- mlt_properties_set_int( this_properties, "width", 720 );
- mlt_properties_set_int( this_properties, "height", fps == 25 ? 576 : 480 );
}
// Return the encoder
// This will hold the size of the dv frame
int size = 0;
+ // Is the image rendered
+ int rendered = mlt_properties_get_int( mlt_frame_properties( frame ), "rendered" );
+
+ // Get width and height
+ int width = mlt_properties_get_int( this_properties, "width" );
+ int height = mlt_properties_get_int( this_properties, "height" );
+
// If we get an encoder, then encode the image
- if ( encoder != NULL )
+ if ( rendered && encoder != NULL )
{
// Specify desired image properties
mlt_image_format fmt = mlt_image_yuv422;
- int width = mlt_properties_get_int( this_properties, "width" );
- int height = mlt_properties_get_int( this_properties, "height" );
uint8_t *image = NULL;
// Get the image
+ mlt_events_fire( this_properties, "consumer-frame-show", frame, NULL );
mlt_frame_get_image( frame, &image, &fmt, &width, &height, 0 );
// Check that we get what we expected
else
{
// Calculate the size of the dv frame
- size = height == 576 ? frame_size_625_50 : frame_size_525_60;
+ size = height == 576 ? FRAME_SIZE_625_50 : FRAME_SIZE_525_60;
}
// Process the frame
dv_encode_full_frame( encoder, &image, e_dv_color_yuv, dv_frame );
}
}
+ else if ( encoder != NULL )
+ {
+ // Calculate the size of the dv frame (duplicate of previous)
+ size = height == 576 ? FRAME_SIZE_625_50 : FRAME_SIZE_525_60;
+ }
return size;
}
// Get the properties of the consumer
mlt_properties this_properties = mlt_consumer_properties( this );
+ // Get the properties of the frame
+ mlt_properties frame_properties = mlt_frame_properties( frame );
+
// Obtain the dv_encoder
dv_encoder_t *encoder = libdv_get_encoder( this, frame );
// Default audio args
mlt_audio_format fmt = mlt_audio_pcm;
int channels = 2;
- int frequency = 48000;
+ int frequency = mlt_properties_get_int( this_properties, "frequency" );
int samples = mlt_sample_calculator( mlt_properties_get_double( this_properties, "fps" ), frequency, count );
int16_t *pcm = NULL;
time_t start = time( NULL );
int height = mlt_properties_get_int( this_properties, "height" );
int is_pal = height == 576;
- int is_wide = 0;
+ int is_wide = mlt_properties_get_double( frame_properties, "fps" ) == ( ( double ) 16.0 / 9.0 );
// Temporary - audio buffer allocation
int16_t *audio_buffers[ 4 ];
int i = 0;
int j = 0;
for ( i = 0 ; i < 4; i ++ )
- audio_buffers[ i ] = malloc( 2 * DV_AUDIO_MAX_SAMPLES );
+ audio_buffers[ i ] = mlt_pool_alloc( 2 * DV_AUDIO_MAX_SAMPLES );
// Get the audio
mlt_frame_get_audio( frame, &pcm, &fmt, &frequency, &channels, &samples );
encoder->samples_this_frame = samples;
// Fill the audio buffers correctly
- for ( i = 0; i < samples; i ++ )
+ if ( mlt_properties_get_double( frame_properties, "_speed" ) == 1.0 )
+ {
+ for ( i = 0; i < samples; i ++ )
+ for ( j = 0; j < channels; j++ )
+ audio_buffers[ j ][ i ] = *pcm ++;
+ }
+ else
+ {
for ( j = 0; j < channels; j++ )
- audio_buffers[ j ][ i ] = *pcm ++;
+ memset( audio_buffers[ j ], 0, 2 * DV_AUDIO_MAX_SAMPLES );
+ }
// Encode audio on frame
dv_encode_full_audio( encoder, audio_buffers, channels, frequency, dv_frame );
// Temporary - free audio buffers
for ( i = 0 ; i < 4; i ++ )
- free( audio_buffers[ i ] );
+ mlt_pool_release( audio_buffers[ i ] );
}
}
static void consumer_output( mlt_consumer this, uint8_t *dv_frame, int size, mlt_frame frame )
{
- fwrite( dv_frame, size, 1, stdout );
- fflush( stdout );
+ // Get the properties
+ mlt_properties properties = mlt_consumer_properties( this );
+
+ FILE *output = stdout;
+ char *target = mlt_properties_get( properties, "target" );
+
+ if ( target != NULL )
+ {
+ output = mlt_properties_get_data( properties, "output_file", NULL );
+ if ( output == NULL )
+ {
+ output = fopen( target, "w" );
+ if ( output != NULL )
+ mlt_properties_set_data( properties, "output_file", output, 0, ( mlt_destructor )fclose, 0 );
+ }
+ }
+
+ if ( output != NULL )
+ {
+ fwrite( dv_frame, size, 1, output );
+ fflush( output );
+ }
+ else
+ {
+ fprintf( stderr, "Unable to open %s\n", target );
+ }
}
/** The main thread - the argument is simply the consumer.
int ( *output )( mlt_consumer, uint8_t *, int, mlt_frame ) = mlt_properties_get_data( properties, "output", NULL );
// Allocate a single PAL frame for encoding
- uint8_t *dv_frame = malloc( frame_size_625_50 );
-
- // Get the service associated to the consumer
- mlt_service service = mlt_consumer_service( this );
+ uint8_t *dv_frame = mlt_pool_alloc( FRAME_SIZE_625_50 );
- // Define a frame pointer
- mlt_frame frame;
+ // Frame and size
+ mlt_frame frame = NULL;
+ int size = 0;
// Loop while running
while( mlt_properties_get_int( properties, "running" ) )
{
// Get the frame
- if ( mlt_service_get_frame( service, &frame, 0 ) == 0 )
- {
- // Encode the image
- int size = video( this, dv_frame, frame );
+ frame = mlt_consumer_rt_frame( this );
- // Encode the audio
- if ( size > 0 )
- audio( this, dv_frame, frame );
-
- // Output the frame
- output( this, dv_frame, size, frame );
-
- // Close the frame
- mlt_frame_close( frame );
+ // Check that we have a frame to work with
+ if ( frame != NULL )
+ {
+ // Obtain the dv_encoder
+ if ( libdv_get_encoder( this, frame ) != NULL )
+ {
+ // Encode the image
+ size = video( this, dv_frame, frame );
+
+ // Encode the audio
+ if ( size > 0 )
+ audio( this, dv_frame, frame );
+
+ // Output the frame
+ output( this, dv_frame, size, frame );
+
+ // Close the frame
+ mlt_frame_close( frame );
+ }
+ else
+ {
+ fprintf( stderr, "Unable to obtain dv encoder.\n" );
+ }
}
}
// Tidy up
- free( dv_frame );
+ mlt_pool_release( dv_frame );
+
+ mlt_consumer_stopped( this );
return NULL;
}
static void consumer_close( mlt_consumer this )
{
- // Get the properties
- mlt_properties properties = mlt_consumer_properties( this );
-
- // Get the thread
- pthread_t *thread = mlt_properties_get_data( properties, "thread", NULL );
-
- // Stop the thread
- mlt_properties_set_int( properties, "running", 0 );
-
- // Wait for termination
- pthread_join( *thread, NULL );
+ // Stop the consumer
+ mlt_consumer_stop( this );
// Close the parent
mlt_consumer_close( this );
// Free the memory
free( this );
}
-