Multitrack rearrangement and tractor cleanup
[melted] / src / framework / mlt_multitrack.c
index 77158c7..50a9426 100644 (file)
 #include "config.h"
 
 #include "mlt_multitrack.h"
+#include "mlt_playlist.h"
 #include "mlt_frame.h"
 
 #include <stdio.h>
 #include <stdlib.h>
 
+struct mlt_track_s
+{
+       mlt_producer producer;
+       mlt_event event;
+};
+
+typedef struct mlt_track_s *mlt_track;
+
 /** Private definition.
 */
 
@@ -33,7 +42,7 @@ struct mlt_multitrack_s
 {
        // We're extending producer here
        struct mlt_producer_s parent;
-       mlt_producer *list;
+       mlt_track *list;
        int size;
        int count;
 };
@@ -56,7 +65,11 @@ mlt_multitrack mlt_multitrack_init( )
                mlt_producer producer = &this->parent;
                if ( mlt_producer_init( producer, this ) == 0 )
                {
+                       mlt_properties properties = mlt_multitrack_properties( this );
                        producer->get_frame = producer_get_frame;
+                       mlt_properties_set_data( properties, "multitrack", this, 0, NULL, NULL );
+                       mlt_properties_set( properties, "log_id", "multitrack" );
+                       mlt_properties_set( properties, "resource", "<multitrack>" );
                }
                else
                {
@@ -73,7 +86,7 @@ mlt_multitrack mlt_multitrack_init( )
 
 mlt_producer mlt_multitrack_producer( mlt_multitrack this )
 {
-       return &this->parent;
+       return this != NULL ? &this->parent : NULL;
 }
 
 /** Get the service associated this multitrack.
@@ -92,10 +105,10 @@ mlt_properties mlt_multitrack_properties( mlt_multitrack this )
        return mlt_service_properties( mlt_multitrack_service( this ) );
 }
 
-/** Initialise timecode related information.
+/** Initialise position related information.
 */
 
-static void mlt_multitrack_refresh( mlt_multitrack this )
+void mlt_multitrack_refresh( mlt_multitrack this )
 {
        int i = 0;
 
@@ -103,7 +116,7 @@ static void mlt_multitrack_refresh( mlt_multitrack this )
        mlt_properties properties = mlt_multitrack_properties( this );
 
        // We need to ensure that the multitrack reports the longest track as its length
-       mlt_timecode length = 0;
+       mlt_position length = 0;
 
        // We need to ensure that fps are the same on all services
        double fps = 0;
@@ -112,13 +125,19 @@ static void mlt_multitrack_refresh( mlt_multitrack this )
        for ( i = 0; i < this->count; i ++ )
        {
                // Get the producer from this index
-               mlt_producer producer = this->list[ i ];
+               mlt_track track = this->list[ i ];
+               mlt_producer producer = track->producer;
 
                // If it's allocated then, update our stats
                if ( producer != NULL )
                {
+                       // If we have more than 1 track, we must be in continue mode
+                       if ( this->count > 1 )
+                               mlt_properties_set( mlt_producer_properties( producer ), "eof", "continue" );
+                       
                        // Determine the longest length
-                       length = mlt_producer_get_length( producer ) > length ? mlt_producer_get_length( producer ) : length;
+                       if ( !mlt_properties_get_int( mlt_producer_properties( producer ), "hide" ) )
+                               length = mlt_producer_get_playtime( producer ) > length ? mlt_producer_get_playtime( producer ) : length;
                        
                        // Handle fps
                        if ( fps == 0 )
@@ -138,12 +157,25 @@ static void mlt_multitrack_refresh( mlt_multitrack this )
        }
 
        // Update multitrack properties now - we'll not destroy the in point here
-       mlt_properties_set_timecode( properties, "length", length );
-       mlt_properties_set_timecode( properties, "out", length );
-       mlt_properties_set_timecode( properties, "playtime", length - mlt_properties_get_timecode( properties, "in" ) );
+       mlt_events_block( properties, properties );
+       mlt_properties_set_position( properties, "length", length );
+       mlt_events_unblock( properties, properties );
+       mlt_properties_set_position( properties, "out", length - 1 );
+       mlt_properties_set_double( properties, "fps", fps );
+}
+
+/** Listener for producers on the playlist.
+*/
+
+static void mlt_multitrack_listener( mlt_producer producer, mlt_multitrack this )
+{
+       mlt_multitrack_refresh( this );
 }
 
 /** Connect a producer to a given track.
+
+       Note that any producer can be connected here, but see special case treatment
+       of playlist in clip point determination below.
 */
 
 int mlt_multitrack_connect( mlt_multitrack this, mlt_producer producer, int track )
@@ -157,14 +189,28 @@ int mlt_multitrack_connect( mlt_multitrack this, mlt_producer producer, int trac
                if ( track >= this->size )
                {
                        int i;
-                       this->list = realloc( this->list, ( track + 10 ) * sizeof( mlt_producer ) );
+                       this->list = realloc( this->list, ( track + 10 ) * sizeof( mlt_track ) );
                        for ( i = this->size; i < track + 10; i ++ )
                                this->list[ i ] = NULL;
                        this->size = track + 10;
                }
-               
+
+               if ( this->list[ track ] != NULL )
+               {
+                       mlt_event_close( this->list[ track ]->event );
+                       mlt_producer_close( this->list[ track ]->producer );
+               }
+               else
+               {
+                       this->list[ track ] = malloc( sizeof( struct mlt_track_s ) );
+               }
+
                // Assign the track in our list here
-               this->list[ track ] = producer;
+               this->list[ track ]->producer = producer;
+               this->list[ track ]->event = mlt_events_listen( mlt_producer_properties( producer ), this, 
+                                                                        "producer-changed", ( mlt_listener )mlt_multitrack_listener );
+               mlt_properties_inc_ref( mlt_producer_properties( producer ) );
+               mlt_event_inc_ref( this->list[ track ]->event );
                
                // Increment the track count if need be
                if ( track >= this->count )
@@ -177,6 +223,141 @@ int mlt_multitrack_connect( mlt_multitrack this, mlt_producer producer, int trac
        return result;
 }
 
+/** Get the number of tracks.
+*/
+
+int mlt_multitrack_count( mlt_multitrack this )
+{
+       return this->count;     
+}
+
+/** Get an individual track as a producer.
+*/
+
+mlt_producer mlt_multitrack_track( mlt_multitrack this, int track )
+{
+       mlt_producer producer = NULL;
+       
+       if ( this->list != NULL && track < this->count )
+               producer = this->list[ track ]->producer;
+
+       return producer;
+}
+
+static int position_compare( const void *p1, const void *p2 )
+{
+       return *( mlt_position * )p1 - *( mlt_position * )p2;
+}
+
+static int add_unique( mlt_position *array, int size, mlt_position position )
+{
+       int i = 0;
+       for ( i = 0; i < size; i ++ )
+               if ( array[ i ] == position )
+                       break;
+       if ( i == size )
+               array[ size ++ ] = position;
+       return size;
+}
+
+/** Determine the clip point.
+
+       Special case here: a 'producer' has no concept of multiple clips - only the 
+       playlist and multitrack producers have clip functionality. Further to that a 
+       multitrack determines clip information from any connected tracks that happen 
+       to be playlists.
+
+       Additionally, it must locate clips in the correct order, for example, consider
+       the following track arrangement:
+
+       playlist1 |0.0     |b0.0      |0.1          |0.1         |0.2           |
+       playlist2 |b1.0  |1.0           |b1.1     |1.1             |
+
+       Note - b clips represent blanks. They are also reported as clip positions.
+
+       When extracting clip positions from these playlists, we should get a sequence of:
+
+       0.0, 1.0, b0.0, 0.1, b1.1, 1.1, 0.1, 0.2, [out of playlist2], [out of playlist1]
+*/
+
+mlt_position mlt_multitrack_clip( mlt_multitrack this, mlt_whence whence, int index )
+{
+       mlt_position position = 0;
+       int i = 0;
+       int j = 0;
+       mlt_position *map = malloc( 1000 * sizeof( mlt_position ) );
+       int count = 0;
+
+       for ( i = 0; i < this->count; i ++ )
+       {
+               // Get the producer for this track
+               mlt_producer producer = this->list[ i ]->producer;
+
+               // If it's assigned and not a hidden track
+               if ( producer != NULL )
+               {
+                       // Get the properties of this producer
+                       mlt_properties properties = mlt_producer_properties( producer );
+
+                       // Determine if it's a playlist
+                       mlt_playlist playlist = mlt_properties_get_data( properties, "playlist", NULL );
+
+                       // Special case consideration of playlists
+                       if ( playlist != NULL )
+                       {
+                               for ( j = 0; j < mlt_playlist_count( playlist ); j ++ )
+                                       count = add_unique( map, count, mlt_playlist_clip( playlist, mlt_whence_relative_start, j ) );
+                               count = add_unique( map, count, mlt_producer_get_out( producer ) + 1 );
+                       }
+                       else
+                       {
+                               count = add_unique( map, count, 0 );
+                               count = add_unique( map, count, mlt_producer_get_out( producer ) + 1 );
+                       }
+               }
+       }
+
+       // Now sort the map
+       qsort( map, count, sizeof( mlt_position ), position_compare );
+
+       // Now locate the requested index
+       switch( whence )
+       {
+               case mlt_whence_relative_start:
+                       if ( index < count )
+                               position = map[ index ];
+                       else
+                               position = map[ count - 1 ];
+                       break;
+
+               case mlt_whence_relative_current:
+                       position = mlt_producer_position( mlt_multitrack_producer( this ) );
+                       for ( i = 0; i < count - 2; i ++ ) 
+                               if ( position >= map[ i ] && position < map[ i + 1 ] )
+                                       break;
+                       index += i;
+                       if ( index >= 0 && index < count )
+                               position = map[ index ];
+                       else if ( index < 0 )
+                               position = map[ 0 ];
+                       else
+                               position = map[ count - 1 ];
+                       break;
+
+               case mlt_whence_relative_end:
+                       if ( index < count )
+                               position = map[ count - index - 1 ];
+                       else
+                               position = map[ 0 ];
+                       break;
+       }
+
+       // Free the map
+       free( map );
+
+       return position;
+}
+
 /** Get frame method.
 
        Special case here: The multitrack must be used in a conjunction with a downstream
@@ -211,35 +392,44 @@ static int producer_get_frame( mlt_producer parent, mlt_frame_ptr frame, int ind
        if ( index < this->count && this->list[ index ] != NULL )
        {
                // Get the producer for this track
-               mlt_producer producer = this->list[ index ];
+               mlt_producer producer = this->list[ index ]->producer;
 
-               // Obtain the current timecode
-               uint64_t position = mlt_producer_frame( parent );
+               // Obtain the current position
+               mlt_position position = mlt_producer_frame( parent );
 
                // Make sure we're at the same point
-               mlt_producer_seek_frame( producer, position );
+               mlt_producer_seek( producer, position );
 
                // Get the frame from the producer
                mlt_service_get_frame( mlt_producer_service( producer ), frame, 0 );
+
+               // Indicate speed of this producer
+               mlt_properties producer_properties = mlt_producer_properties( parent );
+               double speed = mlt_properties_get_double( producer_properties, "_speed" );
+               mlt_properties properties = mlt_frame_properties( *frame );
+               mlt_properties_set_double( properties, "_speed", speed );
+               mlt_properties_set_position( properties, "_position", position );
+               mlt_properties_set_int( properties, "hide", mlt_properties_get_int( mlt_producer_properties( producer ), "hide" ) );
        }
        else
        {
                // Generate a test frame
                *frame = mlt_frame_init( );
 
-               // Let tractor know if we've reached the end
-               mlt_properties_set_int( mlt_frame_properties( *frame ), "last_track", index >= this->count );
-
-               // Update timecode on the frame we're creating
-               mlt_frame_set_timecode( *frame, mlt_producer_position( parent ) );
+               // Update position on the frame we're creating
+               mlt_frame_set_position( *frame, mlt_producer_position( parent ) );
 
                // Move on to the next frame
                if ( index >= this->count )
+               {
+                       // Let tractor know if we've reached the end
+                       mlt_properties_set_int( mlt_frame_properties( *frame ), "last_track", 1 );
+
+                       // Move to the next frame
                        mlt_producer_prepare_next( parent );
+               }
        }
 
-       fprintf( stderr, "timestamp for %d = %f\n", index, ( float )mlt_frame_get_timecode( *frame ) );
-
        return 0;
 }
 
@@ -248,12 +438,26 @@ static int producer_get_frame( mlt_producer parent, mlt_frame_ptr frame, int ind
 
 void mlt_multitrack_close( mlt_multitrack this )
 {
-       // Close the producer
-       mlt_producer_close( &this->parent );
+       if ( this != NULL && mlt_properties_dec_ref( mlt_multitrack_properties( this ) ) <= 0 )
+       {
+               int i = 0;
+               for ( i = 0; i < this->count; i ++ )
+               {
+                       if ( this->list[ i ] != NULL )
+                       {
+                               mlt_event_close( this->list[ i ]->event );
+                               mlt_producer_close( this->list[ i ]->producer );
+                               free( this->list[ i ] );
+                       }
+               }
 
-       // Free the list
-       free( this->list );
+               // Close the producer
+               mlt_producer_close( &this->parent );
 
-       // Free the object
-       free( this );
+               // Free the list
+               free( this->list );
+
+               // Free the object
+               free( this );
+       }
 }