2d7d83c4a55cadbcb496380996e971d9a57c2bf9
[melted] / src / modules / westley / consumer_westley.c
1 /*
2 * consumer_westley.c -- a libxml2 serialiser of mlt service networks
3 * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
4 * Author: Dan Dennedy <dan@dennedy.org>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software Foundation,
18 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19 */
20
21 #include "consumer_westley.h"
22 #include <framework/mlt.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <pthread.h>
27 #include <libxml/tree.h>
28
29 #define ID_SIZE 128
30
31 /** Forward references to static functions.
32 */
33
34 static int consumer_start( mlt_consumer parent );
35 static int consumer_is_stopped( mlt_consumer this );
36
37 /** This is what will be called by the factory - anything can be passed in
38 via the argument, but keep it simple.
39 */
40
41 mlt_consumer consumer_westley_init( char *arg )
42 {
43 // Create the consumer object
44 mlt_consumer this = calloc( sizeof( struct mlt_consumer_s ), 1 );
45
46 // If no malloc'd and consumer init ok
47 if ( this != NULL && mlt_consumer_init( this, NULL ) == 0 )
48 {
49 // Allow thread to be started/stopped
50 this->start = consumer_start;
51 this->is_stopped = consumer_is_stopped;
52
53 mlt_properties_set( mlt_consumer_properties( this ), "resource", arg );
54
55 // Return the consumer produced
56 return this;
57 }
58
59 // malloc or consumer init failed
60 free( this );
61
62 // Indicate failure
63 return NULL;
64 }
65
66
67 // This maintains counters for adding ids to elements
68 struct serialise_context_s
69 {
70 int producer_count;
71 int multitrack_count;
72 int playlist_count;
73 int tractor_count;
74 int filter_count;
75 int transition_count;
76 int pass;
77 mlt_properties producer_map;
78 };
79 typedef struct serialise_context_s* serialise_context;
80
81
82 static inline void serialise_properties( mlt_properties properties, xmlNode *node )
83 {
84 int i;
85 xmlNode *p;
86
87 // Enumerate the properties
88 for ( i = 0; i < mlt_properties_count( properties ); i++ )
89 {
90 char *name = mlt_properties_get_name( properties, i );
91 if ( name != NULL &&
92 name[ 0 ] != '_' &&
93 mlt_properties_get_value( properties, i ) != NULL &&
94 strcmp( name, "westley" ) != 0 )
95 {
96 #if 1
97 p = xmlNewChild( node, NULL, "property", NULL );
98 xmlNewProp( p, "name", mlt_properties_get_name( properties, i ) );
99 xmlNodeSetContent( p, mlt_properties_get_value( properties, i ) );
100 #else
101 p = node;
102 xmlNewProp( p, mlt_properties_get_name( properties, i ), mlt_properties_get_value( properties, i ) );
103 #endif
104 }
105 }
106 }
107
108 static void serialise_service( serialise_context context, mlt_service service, xmlNode *node )
109 {
110 int i;
111 xmlNode *child = node;
112 char id[ ID_SIZE + 1 ];
113 char key[ 11 ];
114 id[ ID_SIZE ] = '\0';
115 key[ 10 ] = '\0';
116
117 // Iterate over consumer/producer connections
118 while ( service != NULL )
119 {
120 mlt_properties properties = mlt_service_properties( service );
121 char *mlt_type = mlt_properties_get( properties, "mlt_type" );
122
123 // Tell about the producer
124 if ( strcmp( mlt_type, "producer" ) == 0 )
125 {
126 if ( context->pass == 0 )
127 {
128 child = xmlNewChild( node, NULL, "producer", NULL );
129
130 // Set the id
131 if ( mlt_properties_get( properties, "id" ) == NULL )
132 {
133 snprintf( id, ID_SIZE, "producer%d", context->producer_count++ );
134 xmlNewProp( child, "id", id );
135 }
136 else
137 strncpy( id, mlt_properties_get( properties, "id" ), ID_SIZE );
138 serialise_properties( properties, child );
139
140 // Add producer to the map
141 snprintf( key, 10, "%p", service );
142 mlt_properties_set( context->producer_map, key, id );
143 }
144 else
145 {
146 snprintf( key, 10, "%p", service );
147 xmlNewProp( node, "producer", mlt_properties_get( context->producer_map, key ) );
148 }
149 if ( mlt_properties_get( properties, "westley" ) != NULL )
150 break;
151 }
152
153 // Tell about the framework container producers
154 else if ( strcmp( mlt_type, "mlt_producer" ) == 0 )
155 {
156 // Recurse on multitrack's tracks
157 if ( strcmp( mlt_properties_get( properties, "resource" ), "<multitrack>" ) == 0 )
158 {
159 if ( context->pass == 0 )
160 {
161 // Iterate over the tracks
162 for ( i = 0; i < mlt_multitrack_count( MLT_MULTITRACK( service ) ); i++ )
163 {
164 serialise_service( context, MLT_SERVICE( mlt_multitrack_track( MLT_MULTITRACK( service ), i ) ), node );
165 }
166 }
167 else
168 {
169 // Iterate over the tracks to collect the producers
170 for ( i = 0; i < mlt_multitrack_count( MLT_MULTITRACK( service ) ); i++ )
171 {
172 serialise_service( context, MLT_SERVICE( mlt_multitrack_track( MLT_MULTITRACK( service ), i ) ), node );
173 }
174
175 child = xmlNewChild( node, NULL, "multitrack", NULL );
176
177 // Set the id
178 if ( mlt_properties_get( properties, "id" ) == NULL )
179 {
180 snprintf( id, ID_SIZE, "multitrack%d", context->multitrack_count++ );
181 xmlNewProp( child, "id", id );
182 }
183
184 // Iterate over the tracks
185 for ( i = 0; i < mlt_multitrack_count( MLT_MULTITRACK( service ) ); i++ )
186 {
187 xmlNode *track = xmlNewChild( child, NULL, "track", NULL );
188 snprintf( key, 10, "%p", MLT_SERVICE( mlt_multitrack_track( MLT_MULTITRACK( service ), i ) ) );
189 xmlNewProp( track, "producer", mlt_properties_get( context->producer_map, key ) );
190 }
191 }
192 break;
193 }
194
195 // Recurse on playlist's clips
196 else if ( strcmp( mlt_properties_get( properties, "resource" ), "<playlist>" ) == 0 )
197 {
198 mlt_playlist_clip_info info;
199
200 if ( context->pass == 0 )
201 {
202 // Iterate over the playlist entries to collect the producers
203 for ( i = 0; i < mlt_playlist_count( MLT_PLAYLIST( service ) ); i++ )
204 {
205 if ( ! mlt_playlist_get_clip_info( MLT_PLAYLIST( service ), &info, i ) )
206 {
207 if ( strcmp( mlt_properties_get( mlt_producer_properties( info.producer ), "mlt_service" ), "blank" ) != 0 )
208 {
209 serialise_service( context, MLT_SERVICE( info.producer ), node );
210 }
211 }
212 }
213
214 child = xmlNewChild( node, NULL, "playlist", NULL );
215
216 // Set the id
217 if ( mlt_properties_get( properties, "id" ) == NULL )
218 {
219 snprintf( id, ID_SIZE, "playlist%d", context->playlist_count++ );
220 xmlNewProp( child, "id", id );
221 }
222 else
223 strncpy( id, mlt_properties_get( properties, "id" ), ID_SIZE );
224
225 xmlNewProp( child, "in", mlt_properties_get( properties, "in" ) );
226 xmlNewProp( child, "out", mlt_properties_get( properties, "out" ) );
227
228 // Add producer to the map
229 snprintf( key, 10, "%p", service );
230 mlt_properties_set( context->producer_map, key, id );
231
232 // Iterate over the playlist entries
233 for ( i = 0; i < mlt_playlist_count( MLT_PLAYLIST( service ) ); i++ )
234 {
235 if ( ! mlt_playlist_get_clip_info( MLT_PLAYLIST( service ), &info, i ) )
236 {
237 if ( strcmp( mlt_properties_get( mlt_producer_properties( info.producer ), "mlt_service" ), "blank" ) == 0 )
238 {
239 char length[ 20 ];
240 length[ 19 ] = '\0';
241 xmlNode *entry = xmlNewChild( child, NULL, "blank", NULL );
242 snprintf( length, 19, "%lld", info.frame_count );
243 xmlNewProp( entry, "length", length );
244 }
245 else
246 {
247 xmlNode *entry = xmlNewChild( child, NULL, "entry", NULL );
248 snprintf( key, 10, "%p", MLT_SERVICE( info.producer ) );
249 xmlNewProp( entry, "producer", mlt_properties_get( context->producer_map, key ) );
250 }
251 }
252 }
253 }
254 else if ( strcmp( (const char*) node->name, "tractor" ) != 0 )
255 {
256 snprintf( key, 10, "%p", service );
257 xmlNewProp( node, "producer", mlt_properties_get( context->producer_map, key ) );
258 }
259 }
260
261 // Recurse on tractor's producer
262 else if ( strcmp( mlt_properties_get( properties, "resource" ), "<tractor>" ) == 0 )
263 {
264 if ( context->pass == 0 )
265 {
266 // Recurse on connected producer
267 serialise_service( context, mlt_service_get_producer( service ), node );
268 }
269 else
270 {
271 child = xmlNewChild( node, NULL, "tractor", NULL );
272
273 // Set the id
274 if ( mlt_properties_get( properties, "id" ) == NULL )
275 {
276 snprintf( id, ID_SIZE, "tractor%d", context->tractor_count++ );
277 xmlNewProp( child, "id", id );
278 }
279
280 xmlNewProp( child, "in", mlt_properties_get( properties, "in" ) );
281 xmlNewProp( child, "out", mlt_properties_get( properties, "out" ) );
282
283 // Recurse on connected producer
284 serialise_service( context, mlt_service_get_producer( service ), child );
285 }
286 break;
287 }
288 }
289
290 // Tell about a filter
291 else if ( strcmp( mlt_type, "filter" ) == 0 )
292 {
293 // Recurse on connected producer
294 serialise_service( context, MLT_SERVICE( MLT_FILTER( service )->producer ), node );
295
296 if ( context->pass == 1 )
297 {
298 child = xmlNewChild( node, NULL, "filter", NULL );
299
300 // Set the id
301 if ( mlt_properties_get( properties, "id" ) == NULL )
302 {
303 snprintf( id, ID_SIZE, "filter%d", context->filter_count++ );
304 xmlNewProp( child, "id", id );
305 }
306
307 serialise_properties( properties, child );
308 }
309 break;
310 }
311
312 // Tell about a transition
313 else if ( strcmp( mlt_type, "transition" ) == 0 )
314 {
315 // Recurse on connected producer
316 serialise_service( context, MLT_SERVICE( MLT_TRANSITION( service )->producer ), node );
317
318 if ( context->pass == 1 )
319 {
320 child = xmlNewChild( node, NULL, "transition", NULL );
321
322 // Set the id
323 if ( mlt_properties_get( properties, "id" ) == NULL )
324 {
325 snprintf( id, ID_SIZE, "transition%d", context->transition_count++ );
326 xmlNewProp( child, "id", id );
327 }
328
329 serialise_properties( properties, child );
330 }
331 break;
332 }
333
334 // Get the next connected service
335 service = mlt_service_get_producer( service );
336 }
337 }
338
339 static int consumer_start( mlt_consumer this )
340 {
341 mlt_service inigo = NULL;
342 xmlDoc *doc = xmlNewDoc( "1.0" );
343 xmlNode *root = xmlNewNode( NULL, "westley" );
344 xmlDocSetRootElement( doc, root );
345
346 // Get the producer service
347 mlt_service service = mlt_service_get_producer( mlt_consumer_service( this ) );
348 if ( service != NULL )
349 {
350 struct serialise_context_s *context = calloc( 1, sizeof( struct serialise_context_s ) );
351 context->producer_map = mlt_properties_new();
352
353 // Remember inigo
354 if ( mlt_properties_get( mlt_service_properties( service ), "mlt_service" ) != NULL &&
355 strcmp( mlt_properties_get( mlt_service_properties( service ), "mlt_service" ), "inigo" ) == 0 )
356 inigo = service;
357
358 // Ensure producer is a framework producer
359 mlt_properties_set( mlt_service_properties( service ), "mlt_type", "mlt_producer" );
360
361 // In pass one, we serialise the end producers and playlists,
362 // adding them to a map keyed by address.
363 serialise_service( context, service, root );
364
365 // In pass two, we serialise the tractor and reference the
366 // producers and playlists
367 context->pass++;
368 serialise_service( context, service, root );
369
370 mlt_properties_close( context->producer_map );
371 free( context );
372
373 if ( mlt_properties_get( mlt_consumer_properties( this ), "resource" ) == NULL )
374 xmlDocFormatDump( stdout, doc, 1 );
375 else
376 xmlSaveFormatFile( mlt_properties_get( mlt_consumer_properties( this ), "resource" ), doc, 1 );
377 }
378
379 xmlFreeDoc( doc );
380 mlt_consumer_stop( this );
381
382 return 0;
383 }
384
385 static int consumer_is_stopped( mlt_consumer this )
386 {
387 return 1;
388 }
389