]> git.sesse.net Git - mlt/blobdiff - src/modules/core/consumer_multi.c
Move the aspect ratio for multi consumer from mlt_frame.
[mlt] / src / modules / core / consumer_multi.c
index 7b5cc707e1d2e7690bfde912857c6c163af85c43..ffcc608619a7bc4b50967861c72efe769bb2cd5e 100644 (file)
@@ -22,6 +22,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <pthread.h>
+#include <sys/time.h>
 
 // Forward references
 static int start( mlt_consumer consumer );
@@ -29,6 +30,7 @@ static int stop( mlt_consumer consumer );
 static int is_stopped( mlt_consumer consumer );
 static void *consumer_thread( void *arg );
 static void consumer_close( mlt_consumer consumer );
+static void purge( mlt_consumer consumer );
 
 static mlt_properties normalisers = NULL;
 
@@ -41,25 +43,36 @@ mlt_consumer consumer_multi_init( mlt_profile profile, mlt_service_type type, co
 
        if ( consumer )
        {
+               mlt_properties properties = MLT_CONSUMER_PROPERTIES(consumer);
+
+               // Set defaults
+               mlt_properties_set( properties, "resource", arg );
+               mlt_properties_set_int( properties, "real_time", -1 );
+               mlt_properties_set_int( properties, "terminate_on_pause", 1 );
+
+               // Init state
+               mlt_properties_set_int( properties, "joined", 1 );
+
                // Assign callbacks
                consumer->close = consumer_close;
                consumer->start = start;
                consumer->stop = stop;
                consumer->is_stopped = is_stopped;
-
-               mlt_properties_set( MLT_CONSUMER_PROPERTIES(consumer), "resource", arg );
+               consumer->purge = purge;
        }
 
        return consumer;
 }
 
-static mlt_consumer create_consumer( mlt_profile profile, char *id )
+static mlt_consumer create_consumer( mlt_profile profile, char *id, char *arg )
 {
        char *myid = id ? strdup( id ) : NULL;
-       char *arg = myid ? strchr( myid, ':' ) : NULL;
-       if ( arg != NULL )
-               *arg ++ = '\0';
-       mlt_consumer consumer = mlt_factory_consumer( profile, myid, arg );
+       char *myarg = ( myid && !arg ) ? strchr( myid, ':' ) : NULL;
+       if ( myarg )
+               *myarg ++ = '\0';
+       else
+               myarg = arg;
+       mlt_consumer consumer = mlt_factory_consumer( profile, myid, myarg );
        if ( myid )
                free( myid );
        return consumer;
@@ -72,17 +85,27 @@ static void create_filter( mlt_profile profile, mlt_service service, char *effec
        if ( arg != NULL )
                *arg ++ = '\0';
 
-       // The swscale and avcolor_space filters require resolution as arg to test compatibility
-       if ( strncmp( effect, "swscale", 7 ) == 0 || strncmp( effect, "avcolo", 6 ) == 0 )
-               arg = (char*) mlt_properties_get_int( MLT_SERVICE_PROPERTIES( service ), "_real_width" );
-
-       mlt_filter filter = mlt_factory_filter( profile, id, arg );
-       if ( filter != NULL )
+       // We cannot use GLSL-based filters here.
+       if ( strncmp( effect, "movit.", 6 ) && strncmp( effect, "glsl.", 5 ) )
        {
-               mlt_properties_set_int( MLT_FILTER_PROPERTIES( filter ), "_loader", 1 );
-               mlt_service_attach( service, filter );
-               mlt_filter_close( filter );
-               *created = 1;
+               mlt_filter filter;
+               // The swscale and avcolor_space filters require resolution as arg to test compatibility
+               if ( strncmp( effect, "swscale", 7 ) == 0 || strncmp( effect, "avcolo", 6 ) == 0 )
+               {
+                       int width = mlt_properties_get_int( MLT_SERVICE_PROPERTIES( service ), "meta.media.width" );
+                       filter = mlt_factory_filter( profile, id, &width );
+               }
+               else
+               {
+                       filter = mlt_factory_filter( profile, id, arg );
+               }
+               if ( filter )
+               {
+                       mlt_properties_set_int( MLT_FILTER_PROPERTIES( filter ), "_loader", 1 );
+                       mlt_service_attach( service, filter );
+                       mlt_filter_close( filter );
+                       *created = 1;
+               }
        }
        free( id );
 }
@@ -120,12 +143,27 @@ static void attach_normalisers( mlt_profile profile, mlt_service service )
 
        // Attach the audio and video format converters
        int created = 0;
+       // movit.convert skips setting the frame->convert_image pointer if GLSL cannot be used.
+       mlt_filter filter = mlt_factory_filter( profile, "movit.convert", NULL );
+       if ( filter != NULL )
+       {
+               mlt_properties_set_int( MLT_FILTER_PROPERTIES( filter ), "_loader", 1 );
+               mlt_service_attach( service, filter );
+               mlt_filter_close( filter );
+               created = 1;
+       }
+       // avcolor_space and imageconvert only set frame->convert_image if it has not been set.
        create_filter( profile, service, "avcolor_space", &created );
        if ( !created )
                create_filter( profile, service, "imageconvert", &created );
        create_filter( profile, service, "audioconvert", &created );
 }
 
+static void on_frame_show( void *dummy, mlt_properties properties, mlt_frame frame )
+{
+       mlt_events_fire( properties, "consumer-frame-show", frame, NULL );
+}
+
 static mlt_consumer generate_consumer( mlt_consumer consumer, mlt_properties props, int index )
 {
        mlt_profile profile = NULL;
@@ -133,7 +171,8 @@ static mlt_consumer generate_consumer( mlt_consumer consumer, mlt_properties pro
                profile = mlt_profile_init( mlt_properties_get( props, "mlt_profile" ) );
        if ( !profile )
                profile = mlt_profile_clone( mlt_service_profile( MLT_CONSUMER_SERVICE(consumer) ) );
-       mlt_consumer nested = create_consumer( profile, mlt_properties_get( props, "consumer" ) );
+       mlt_consumer nested = create_consumer( profile, mlt_properties_get( props, "mlt_service" ),
+               mlt_properties_get( props, "target" ) );
 
        if ( nested )
        {
@@ -154,6 +193,14 @@ static mlt_consumer generate_consumer( mlt_consumer consumer, mlt_properties pro
                mlt_properties_inherit( nested_props, props );
 
                attach_normalisers( profile, MLT_CONSUMER_SERVICE(nested) );
+
+               // Relay the first available consumer-frame-show event
+               mlt_event event = mlt_properties_get_data( properties, "frame-show-event", NULL );
+               if ( !event )
+               {
+                       event = mlt_events_listen( nested_props, properties, "consumer-frame-show", (mlt_listener) on_frame_show );
+                       mlt_properties_set_data( properties, "frame-show-event", event, 0, /*mlt_event_close*/ NULL, NULL );
+               }
        }
        else
        {
@@ -215,7 +262,7 @@ static void foreach_consumer_init( mlt_consumer consumer )
                                int i, count;
 
                                if ( !p ) break;
-                               mlt_properties_set( p, "consumer", mlt_properties_get( properties, key ) );
+                               mlt_properties_set( p, "mlt_service", mlt_properties_get( properties, key ) );
                                snprintf( key, sizeof(key), "%d.", index );
 
                                count = mlt_properties_count( properties );
@@ -245,7 +292,14 @@ static void foreach_consumer_start( mlt_consumer consumer )
        do {
                snprintf( key, sizeof(key), "%d.consumer", index++ );
                nested = mlt_properties_get_data( properties, key, NULL );
-               if ( nested ) mlt_consumer_start( nested );
+               if ( nested )
+               {
+                       mlt_properties nested_props = MLT_CONSUMER_PROPERTIES(nested);
+                       mlt_properties_set_position( nested_props, "_multi_position", 0 );
+                       mlt_properties_set_data( nested_props, "_multi_audio", NULL, 0, NULL, NULL );
+                       mlt_properties_set_int( nested_props, "_multi_samples", 0 );
+                       mlt_consumer_start( nested );
+               }
        } while ( nested );
 }
 
@@ -273,7 +327,96 @@ static void foreach_consumer_put( mlt_consumer consumer, mlt_frame frame )
        do {
                snprintf( key, sizeof(key), "%d.consumer", index++ );
                nested = mlt_properties_get_data( properties, key, NULL );
-               if ( nested ) mlt_consumer_put_frame( nested, mlt_frame_clone( frame, 0 ) );
+               if ( nested )
+               {
+                       mlt_properties nested_props = MLT_CONSUMER_PROPERTIES(nested);
+                       double self_fps = mlt_properties_get_double( properties, "fps" );
+                       double nested_fps = mlt_properties_get_double( nested_props, "fps" );
+                       mlt_position nested_pos = mlt_properties_get_position( nested_props, "_multi_position" );
+                       mlt_position self_pos = mlt_frame_get_position( frame );
+                       double self_time = self_pos / self_fps;
+                       double nested_time = nested_pos / nested_fps;
+
+                       // get the audio for the current frame
+                       uint8_t *buffer = NULL;
+                       mlt_audio_format format = mlt_audio_s16;
+                       int channels = mlt_properties_get_int( properties, "channels" );
+                       int frequency = mlt_properties_get_int( properties, "frequency" );
+                       int current_samples = mlt_sample_calculator( self_fps, frequency, self_pos );
+                       mlt_frame_get_audio( frame, (void**) &buffer, &format, &frequency, &channels, &current_samples );
+                       int current_size = mlt_audio_format_size( format, current_samples, channels );
+
+                       // get any leftover audio
+                       int prev_size = 0;
+                       uint8_t *prev_buffer = mlt_properties_get_data( nested_props, "_multi_audio", &prev_size );
+                       uint8_t *new_buffer = NULL;
+                       if ( prev_size > 0 )
+                       {
+                               new_buffer = mlt_pool_alloc( prev_size + current_size );
+                               memcpy( new_buffer, prev_buffer, prev_size );
+                               memcpy( new_buffer + prev_size, buffer, current_size );
+                               buffer = new_buffer;
+                       }
+                       current_size += prev_size;
+                       current_samples += mlt_properties_get_int( nested_props, "_multi_samples" );
+
+                       while ( nested_time <= self_time )
+                       {
+                               // put ideal number of samples into cloned frame
+                               int deeply = index > 1 ? 1 : 0;
+                               mlt_frame clone_frame = mlt_frame_clone( frame, deeply );
+                               mlt_properties clone_props = MLT_FRAME_PROPERTIES( clone_frame );
+                               int nested_samples = mlt_sample_calculator( nested_fps, frequency, nested_pos );
+                               // -10 is an optimization to avoid tiny amounts of leftover samples
+                               nested_samples = nested_samples > current_samples - 10 ? current_samples : nested_samples;
+                               int nested_size = mlt_audio_format_size( format, nested_samples, channels );
+                               if ( nested_size > 0 )
+                               {
+                                       prev_buffer = mlt_pool_alloc( nested_size );
+                                       memcpy( prev_buffer, buffer, nested_size );
+                               }
+                               else
+                               {
+                                       prev_buffer = NULL;
+                                       nested_size = 0;
+                               }
+                               mlt_frame_set_audio( clone_frame, prev_buffer, format, nested_size, mlt_pool_release );
+                               mlt_properties_set_int( clone_props, "audio_samples", nested_samples );
+                               mlt_properties_set_int( clone_props, "audio_frequency", frequency );
+                               mlt_properties_set_int( clone_props, "audio_channels", channels );
+
+                               // chomp the audio
+                               current_samples -= nested_samples;
+                               current_size -= nested_size;
+                               buffer += nested_size;
+
+                               // Fix some things
+                               mlt_properties_set_int( clone_props, "meta.media.width",
+                                       mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "width" ) );
+                               mlt_properties_set_int( clone_props, "meta.media.height",
+                                       mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "height" ) );
+
+                               // send frame to nested consumer
+                               mlt_consumer_put_frame( nested, clone_frame );
+                               mlt_properties_set_position( nested_props, "_multi_position", ++nested_pos );
+                               nested_time = nested_pos / nested_fps;
+                       }
+
+                       // save any remaining audio
+                       if ( current_size > 0 )
+                       {
+                               prev_buffer = mlt_pool_alloc( current_size );
+                               memcpy( prev_buffer, buffer, current_size );
+                       }
+                       else
+                       {
+                               prev_buffer = NULL;
+                               current_size = 0;
+                       }
+                       mlt_pool_release( new_buffer );
+                       mlt_properties_set_data( nested_props, "_multi_audio", prev_buffer, current_size, mlt_pool_release, NULL );
+                       mlt_properties_set_int( nested_props, "_multi_samples", current_samples );
+               }
        } while ( nested );
 }
 
@@ -283,11 +426,27 @@ static void foreach_consumer_stop( mlt_consumer consumer )
        mlt_consumer nested = NULL;
        char key[30];
        int index = 0;
+       struct timespec tm = { 0, 1000 * 1000 };
 
        do {
                snprintf( key, sizeof(key), "%d.consumer", index++ );
                nested = mlt_properties_get_data( properties, key, NULL );
-               if ( nested ) mlt_consumer_stop( nested );
+               if ( nested )
+               {
+                       // Let consumer with terminate_on_pause stop on their own
+                       if ( mlt_properties_get_int( MLT_CONSUMER_PROPERTIES(nested), "terminate_on_pause" ) )
+                       {
+                               // Send additional dummy frame to unlatch nested consumer's threads
+                               mlt_consumer_put_frame( nested, mlt_frame_init( MLT_CONSUMER_SERVICE(consumer) ) );
+                               // wait for stop
+                               while ( !mlt_consumer_is_stopped( nested ) )
+                                       nanosleep( &tm, NULL );
+                       }
+                       else
+                       {
+                               mlt_consumer_stop( nested );
+                       }
+               }
        } while ( nested );
 }
 
@@ -307,6 +466,7 @@ static int start( mlt_consumer consumer )
 
                // Set the running state
                mlt_properties_set_int( properties, "running", 1 );
+               mlt_properties_set_int( properties, "joined", 0 );
 
                // Construct and start nested consumers
                if ( !mlt_properties_get_data( properties, "0.consumer", NULL ) )
@@ -325,7 +485,7 @@ static int start( mlt_consumer consumer )
 static int stop( mlt_consumer consumer )
 {
        // Check that we're running
-       if ( !is_stopped( consumer ) )
+       if ( !mlt_properties_get_int( MLT_CONSUMER_PROPERTIES(consumer), "joined" ) )
        {
                mlt_properties properties = MLT_CONSUMER_PROPERTIES( consumer );
                pthread_t *thread = mlt_properties_get_data( properties, "thread", NULL );
@@ -339,6 +499,7 @@ static int stop( mlt_consumer consumer )
                        foreach_consumer_refresh( consumer );
                        pthread_join( *thread, NULL );
                }
+               mlt_properties_set_int( properties, "joined", 1 );
 
                // Stop nested consumers
                foreach_consumer_stop( consumer );
@@ -355,6 +516,27 @@ static int is_stopped( mlt_consumer consumer )
        return !mlt_properties_get_int( MLT_CONSUMER_PROPERTIES( consumer ), "running" );
 }
 
+/** Purge each of the child consumers.
+*/
+
+static void purge( mlt_consumer consumer )
+{
+       mlt_properties properties = MLT_CONSUMER_PROPERTIES( consumer );
+       if ( mlt_properties_get_int( properties, "running" ) )
+       {
+               mlt_consumer nested = NULL;
+               char key[30];
+               int index = 0;
+
+               do {
+                       snprintf( key, sizeof(key), "%d.consumer", index++ );
+                       nested = mlt_properties_get_data( properties, key, NULL );
+                       if ( nested )
+                               mlt_consumer_purge( nested );
+               } while ( nested );
+       }
+}
+
 /** The main thread - the argument is simply the consumer.
 */
 
@@ -381,30 +563,34 @@ static void *consumer_thread( void *arg )
                // Check that we have a frame to work with
                if ( frame && !terminated && !is_stopped( consumer ) )
                {
-                       if ( !mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "rendered" ) )
+                       if ( mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "rendered" ) )
+                       {
+                               if ( mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "_speed" ) == 0 )
+                                       foreach_consumer_refresh( consumer );
+                               foreach_consumer_put( consumer, frame );
+                       }
+                       else
                        {
                                int dropped = mlt_properties_get_int( properties, "_dropped" );
-                               mlt_frame_close( frame );
                                mlt_log_info( MLT_CONSUMER_SERVICE(consumer), "dropped frame %d\n", ++dropped );
                                mlt_properties_set_int( properties, "_dropped", dropped );
-                               continue;
                        }
-                       if ( mlt_properties_get_int( MLT_FRAME_PROPERTIES(frame), "_speed" ) == 0 )
-                               foreach_consumer_refresh( consumer );
-                       foreach_consumer_put( consumer, frame );
-                       mlt_events_fire( properties, "consumer-frame-show", frame, NULL );
                        mlt_frame_close( frame );
                }
                else
                {
-                       if ( frame ) mlt_frame_close( frame );
-                       foreach_consumer_put( consumer, NULL );
+                       if ( frame && terminated )
+                       {
+                               // Send this termination frame to nested consumers for their cancellation
+                               foreach_consumer_put( consumer, frame );
+                       }
+                       if ( frame )
+                               mlt_frame_close( frame );
                        terminated = 1;
                }
        }
 
        // Indicate that the consumer is stopped
-       mlt_properties_set_int( properties, "running", 0 );
        mlt_consumer_stopped( consumer );
 
        return NULL;
@@ -416,6 +602,7 @@ static void *consumer_thread( void *arg )
 static void consumer_close( mlt_consumer consumer )
 {
        mlt_consumer_stop( consumer );
+
        // Close the parent
        mlt_consumer_close( consumer );
        free( consumer );