fix for in/out during serialisation
[melted] / src / modules / westley / consumer_westley.c
index 4e8fbb0..35afa6e 100644 (file)
@@ -24,6 +24,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <pthread.h>
+#include <unistd.h>
 #include <libxml/tree.h>
 
 #define ID_SIZE 128
@@ -40,6 +41,7 @@ struct serialise_context_s
        int transition_count;
        int pass;
        mlt_properties hide_map;
+       char *root;
 };
 typedef struct serialise_context_s* serialise_context;
 
@@ -165,7 +167,7 @@ mlt_consumer consumer_westley_init( char *arg )
        return NULL;
 }
 
-static inline void serialise_properties( mlt_properties properties, xmlNode *node )
+static inline void serialise_properties( serialise_context context, mlt_properties properties, xmlNode *node )
 {
        int i;
        xmlNode *p;
@@ -179,16 +181,24 @@ static inline void serialise_properties( mlt_properties properties, xmlNode *nod
                         mlt_properties_get_value( properties, i ) != NULL &&
                         strcmp( name, "westley" ) != 0 &&
                         strcmp( name, "in" ) != 0 &&
-                        strcmp( name, "out" ) != 0 )
+                        strcmp( name, "out" ) != 0 && 
+                        strcmp( name, "id" ) != 0 && 
+                        strcmp( name, "root" ) != 0 && 
+                        strcmp( name, "width" ) != 0 &&
+                        strcmp( name, "height" ) != 0 )
                {
+                       char *value = mlt_properties_get_value( properties, i );
                        p = xmlNewChild( node, NULL, "property", NULL );
                        xmlNewProp( p, "name", mlt_properties_get_name( properties, i ) );
-                       xmlNodeSetContent( p, mlt_properties_get_value( properties, i ) );
+                       if ( strcmp( context->root, "" ) && !strncmp( value, context->root, strlen( context->root ) ) )
+                               xmlNodeSetContent( p, value + strlen( context->root ) + 1 );
+                       else
+                               xmlNodeSetContent( p, value );
                }
        }
 }
 
-static inline void serialise_producer_filters( serialise_context context, mlt_service service, xmlNode *node )
+static inline void serialise_service_filters( serialise_context context, mlt_service service, xmlNode *node )
 {
        int i;
        xmlNode *p;
@@ -202,12 +212,23 @@ static inline void serialise_producer_filters( serialise_context context, mlt_se
                {
                        // Get a new id - if already allocated, do nothing
                        char *id = westley_get_id( context, mlt_filter_service( filter ), westley_filter );
-                       if ( id == NULL )
-                               continue;
-
-                       p = xmlNewChild( node, NULL, "filter", NULL );
-                       xmlNewProp( p, "id", id );
-                       serialise_properties( properties, p );
+                       if ( id != NULL )
+                       {
+                               int in = mlt_properties_get_position( properties, "in" );
+                               int out = mlt_properties_get_position( properties, "out" );
+                               p = xmlNewChild( node, NULL, "filter", NULL );
+                               xmlNewProp( p, "id", id );
+                               if ( in != 0 || out != 0 )
+                               {
+                                       char temp[ 20 ];
+                                       sprintf( temp, "%d", in );
+                                       xmlNewProp( p, "in", temp );
+                                       sprintf( temp, "%d", out );
+                                       xmlNewProp( p, "out", temp );
+                               }
+                               serialise_properties( context, properties, p );
+                               serialise_service_filters( context, mlt_filter_service( filter ), p );
+                       }
                }
        }
 }
@@ -228,8 +249,10 @@ static void serialise_producer( serialise_context context, mlt_service service,
 
                // Set the id
                xmlNewProp( child, "id", id );
-               serialise_properties( properties, child );
-               serialise_producer_filters( context, service, child );
+               xmlNewProp( child, "in", mlt_properties_get( properties, "in" ) );
+               xmlNewProp( child, "out", mlt_properties_get( properties, "out" ) );
+               serialise_properties( context, properties, child );
+               serialise_service_filters( context, service, child );
 
                // Add producer to the map
                mlt_properties_set_int( context->hide_map, id, mlt_properties_get_int( properties, "hide" ) );
@@ -245,7 +268,6 @@ static void serialise_producer( serialise_context context, mlt_service service,
 static void serialise_multitrack( serialise_context context, mlt_service service, xmlNode *node )
 {
        int i;
-       xmlNode *child = node;
        
        if ( context->pass == 0 )
        {
@@ -260,16 +282,10 @@ static void serialise_multitrack( serialise_context context, mlt_service service
                if ( id == NULL )
                        return;
 
-               // Create the multitrack node
-               child = xmlNewChild( node, NULL, "multitrack", NULL );
-               
-               // Set the id
-               xmlNewProp( child, "id", id );
-
                // Serialise the tracks
                for ( i = 0; i < mlt_multitrack_count( MLT_MULTITRACK( service ) ); i++ )
                {
-                       xmlNode *track = xmlNewChild( child, NULL, "track", NULL );
+                       xmlNode *track = xmlNewChild( node, NULL, "track", NULL );
                        int hide = 0;
                        mlt_producer producer = mlt_multitrack_track( MLT_MULTITRACK( service ), i );
 
@@ -280,10 +296,12 @@ static void serialise_multitrack( serialise_context context, mlt_service service
                        if ( hide )
                                xmlNewProp( track, "hide", hide == 1 ? "video" : ( hide == 2 ? "audio" : "both" ) );
                }
-               serialise_producer_filters( context, service, child );
+               serialise_service_filters( context, service, node );
        }
 }
 
+static void serialise_tractor( serialise_context context, mlt_service service, xmlNode *node );
+
 static void serialise_playlist( serialise_context context, mlt_service service, xmlNode *node )
 {
        int i;
@@ -307,7 +325,14 @@ static void serialise_playlist( serialise_context context, mlt_service service,
                                {
                                        char *service_s = mlt_properties_get( mlt_producer_properties( info.producer ), "mlt_service" );
                                        char *resource_s = mlt_properties_get( mlt_producer_properties( info.producer ), "resource" );
-                                       if ( service_s != NULL && strcmp( service_s, "blank" ) != 0 )
+                                       if ( resource_s != NULL && !strcmp( resource_s, "<tractor>" ) )
+                                       {
+                                               serialise_tractor( context, MLT_SERVICE( info.producer ), node );
+                                               context->pass ++;
+                                               serialise_tractor( context, MLT_SERVICE( info.producer ), node );
+                                               context->pass --;
+                                       }
+                                       else if ( service_s != NULL && strcmp( service_s, "blank" ) != 0 )
                                                serialise_service( context, MLT_SERVICE( info.producer ), node );
                                        else if ( resource_s != NULL && !strcmp( resource_s, "<playlist>" ) )
                                                serialise_playlist( context, MLT_SERVICE( info.producer ), node );
@@ -351,7 +376,7 @@ static void serialise_playlist( serialise_context context, mlt_service service,
                        }
                }
 
-               serialise_producer_filters( context, service, child );
+               serialise_service_filters( context, service, child );
        }
        else if ( strcmp( (const char*) node->name, "tractor" ) != 0 )
        {
@@ -386,7 +411,7 @@ static void serialise_tractor( serialise_context context, mlt_service service, x
 
                // Recurse on connected producer
                serialise_service( context, mlt_service_producer( service ), child );
-               serialise_producer_filters( context, service, child );
+               serialise_service_filters( context, service, child );
        }
 }
 
@@ -412,7 +437,8 @@ static void serialise_filter( serialise_context context, mlt_service service, xm
                xmlNewProp( child, "in", mlt_properties_get( properties, "in" ) );
                xmlNewProp( child, "out", mlt_properties_get( properties, "out" ) );
 
-               serialise_properties( properties, child );
+               serialise_properties( context, properties, child );
+               serialise_service_filters( context, service, child );
        }
 }
 
@@ -438,7 +464,8 @@ static void serialise_transition( serialise_context context, mlt_service service
                xmlNewProp( child, "in", mlt_properties_get( properties, "in" ) );
                xmlNewProp( child, "out", mlt_properties_get( properties, "out" ) );
 
-               serialise_properties( properties, child );
+               serialise_properties( context, properties, child );
+               serialise_service_filters( context, service, child );
        }
 }
 
@@ -511,12 +538,28 @@ static void serialise_service( serialise_context context, mlt_service service, x
 
 xmlDocPtr westley_make_doc( mlt_service service )
 {
+       mlt_properties properties = mlt_service_properties( service );
        xmlDocPtr doc = xmlNewDoc( "1.0" );
        xmlNodePtr root = xmlNewNode( NULL, "westley" );
        struct serialise_context_s *context = calloc( 1, sizeof( struct serialise_context_s ) );
        
        xmlDocSetRootElement( doc, root );
-               
+
+       // If we have root, then deal with it now
+       if ( mlt_properties_get( properties, "root" ) != NULL )
+       {
+               xmlNewProp( root, "root", mlt_properties_get( properties, "root" ) );
+               context->root = strdup( mlt_properties_get( properties, "root" ) );
+       }
+       else
+       {
+               context->root = strdup( "" );
+       }
+
+       // Assign a title property
+       if ( mlt_properties_get( properties, "title" ) != NULL )
+               xmlNewProp( root, "title", mlt_properties_get( properties, "title" ) );
+
        // Construct the context maps
        context->id_map = mlt_properties_new();
        context->hide_map = mlt_properties_new();
@@ -536,12 +579,12 @@ xmlDocPtr westley_make_doc( mlt_service service )
        // Cleanup resource
        mlt_properties_close( context->id_map );
        mlt_properties_close( context->hide_map );
+       free( context->root );
        free( context );
        
        return doc;
 }
 
-
 static int consumer_start( mlt_consumer this )
 {
        xmlDocPtr doc = NULL;
@@ -550,15 +593,49 @@ static int consumer_start( mlt_consumer this )
        mlt_service service = mlt_service_producer( mlt_consumer_service( this ) );
        if ( service != NULL )
        {
-               char *resource =  mlt_properties_get( mlt_consumer_properties( this ), "resource" );
+               mlt_properties properties = mlt_consumer_properties( this );
+               char *resource =  mlt_properties_get( properties, "resource" );
+
+               // Set the title if provided
+               if ( mlt_properties_get( properties, "title" ) )
+                       mlt_properties_set( mlt_service_properties( service ), "title", mlt_properties_get( properties, "title" ) );
+               else if ( mlt_properties_get( mlt_service_properties( service ), "title" ) == NULL )
+                       mlt_properties_set( mlt_service_properties( service ), "title", "Anonymous Submission" );
+
+               // Check for a root on the consumer properties and pass to service
+               if ( mlt_properties_get( properties, "root" ) )
+                       mlt_properties_set( mlt_service_properties( service ), "root", mlt_properties_get( properties, "root" ) );
 
+               // Specify roots in other cases...
+               if ( resource != NULL && mlt_properties_get( properties, "root" ) == NULL )
+               {
+                       // Get the current working directory
+                       char *cwd = getcwd( NULL, 0 );
+                       mlt_properties_set( mlt_service_properties( service ), "root", cwd );
+                       free( cwd );
+               }
+
+               // Make the document
                doc = westley_make_doc( service );
-               
+
+               // Handle the output
                if ( resource == NULL || !strcmp( resource, "" ) )
+               {
                        xmlDocFormatDump( stdout, doc, 1 );
+               }
+               else if ( !strcmp( resource, "buffer" ) )
+               {
+                       xmlChar *buffer = NULL;
+                       int length = 0;
+                       xmlDocDumpMemory( doc, &buffer, &length );
+                       mlt_properties_set_data( properties, "buffer", buffer, length, xmlFree, NULL );
+               }
                else
+               {
                        xmlSaveFormatFile( resource, doc, 1 );
+               }
                
+               // Close the document
                xmlFreeDoc( doc );
        }