]> git.sesse.net Git - mlt/blob - src/modules/core/transition_region.c
Merge ../mlt++
[mlt] / src / modules / core / transition_region.c
1 /*
2  * transition_region.c -- region transition
3  * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
4  * Author: Charles Yates <charles.yates@pandora.be>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library 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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "transition_region.h"
22 #include "transition_composite.h"
23
24 #include <framework/mlt.h>
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29
30 static int create_instance( mlt_transition this, char *name, char *value, int count )
31 {
32         // Return from this function
33         int error = 0;
34
35         // Duplicate the value
36         char *type = strdup( value );
37
38         // Pointer to filter argument
39         char *arg = type == NULL ? NULL : strchr( type, ':' );
40
41         // New filter being created
42         mlt_filter filter = NULL;
43
44         // Cleanup type and arg
45         if ( arg != NULL )
46                 *arg ++ = '\0';
47
48         // Create the filter
49         mlt_profile profile = mlt_service_profile( MLT_TRANSITION_SERVICE( this ) );
50         filter = mlt_factory_filter( profile, type, arg );
51
52         // If we have a filter, then initialise and store it
53         if ( filter != NULL )
54         {
55                 // Properties of this
56                 mlt_properties properties = MLT_TRANSITION_PROPERTIES( this );
57
58                 // String to hold the property name
59                 char id[ 256 ];
60
61                 // String to hold the passdown key
62                 char key[ 256 ];
63
64                 // Construct id
65                 sprintf( id, "_filter_%d", count );
66
67                 // Counstruct key
68                 sprintf( key, "%s.", name );
69
70                 // Just in case, let's assume that the filter here has a composite
71                 //mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "composite.geometry", "0%,0%:100%x100%" );
72                 //mlt_properties_set_int( MLT_FILTER_PROPERTIES( filter ), "composite.fill", 1 );
73
74                 // Pass all the key properties on the filter down
75                 mlt_properties_pass( MLT_FILTER_PROPERTIES( filter ), properties, key );
76
77                 // Ensure that filter is assigned
78                 mlt_properties_set_data( properties, id, filter, 0, ( mlt_destructor )mlt_filter_close, NULL );
79         }
80         else
81         {
82                 // Indicate that an error has occurred
83                 error = 1;
84         }
85
86         // Cleanup
87         free( type );
88
89         // Return error condition
90         return error;
91 }
92
93 static uint8_t *filter_get_alpha_mask( mlt_frame this )
94 {
95         uint8_t *alpha = NULL;
96
97         // Obtain properties of frame
98         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
99
100         // Get the shape frame
101         mlt_frame shape_frame = mlt_properties_get_data( properties, "shape_frame", NULL );
102
103         // Get the width and height of the image
104         int region_width = mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "width" );
105         int region_height = mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "height" );
106         uint8_t *image = NULL;
107         mlt_image_format format = mlt_image_yuv422;
108                                         
109         // Get the shape image to trigger alpha creation
110         mlt_properties_set_int( MLT_FRAME_PROPERTIES( shape_frame ), "distort", 1 );
111         mlt_frame_get_image( shape_frame, &image, &format, &region_width, &region_height, 0 );
112
113         alpha = mlt_frame_get_alpha_mask( shape_frame );
114
115         // Generate from the Y component of the image if no alpha available
116         if ( alpha == NULL )
117         {
118                 int size = region_width * region_height;
119                 uint8_t *p = mlt_pool_alloc( size );
120                 alpha = p;
121                 while ( size -- )
122                 {
123                         *p ++ = ( int )( ( ( *image ++ - 16 ) * 299 ) / 255 );
124                         image ++;
125                 }
126                 mlt_properties_set_data( MLT_FRAME_PROPERTIES( this ), "alpha", alpha, region_width * region_height, mlt_pool_release, NULL );
127         }
128         else
129         {
130                 mlt_properties_set_data( MLT_FRAME_PROPERTIES( this ), "alpha", alpha, region_width * region_height, NULL, NULL );
131         }
132
133         this->get_alpha_mask = NULL;
134
135         return alpha;
136 }
137
138 /** Do it :-).
139 */
140
141 static int transition_get_image( mlt_frame frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
142 {
143         // Error we will return
144         int error = 0;
145
146         // We will get the 'b frame' from the frame stack
147         mlt_frame b_frame = mlt_frame_pop_frame( frame );
148
149         // Get the watermark transition object
150         mlt_transition this = mlt_frame_pop_service( frame );
151
152         // Get the properties of the transition
153         mlt_properties properties = MLT_TRANSITION_PROPERTIES( this );
154
155         // Get the composite from the transition
156         mlt_transition composite = mlt_properties_get_data( properties, "composite", NULL );
157
158         // Look for the first filter
159         mlt_filter filter = mlt_properties_get_data( properties, "_filter_0", NULL );
160
161         // Get the unique id of the filter (used to reacquire the producer position)
162         char *name = mlt_properties_get( properties, "_unique_id" );
163
164         // Get the original producer position
165         mlt_position position = mlt_properties_get_position( MLT_FRAME_PROPERTIES( frame ), name );
166
167         // Create a composite if we don't have one
168         if ( composite == NULL )
169         {
170                 // Create composite via the factory
171                 mlt_profile profile = mlt_service_profile( MLT_TRANSITION_SERVICE( this ) );
172                 composite = mlt_factory_transition( profile, "composite", NULL );
173
174                 // If we have one
175                 if ( composite != NULL )
176                 {
177                         // Get the properties
178                         mlt_properties composite_properties = MLT_TRANSITION_PROPERTIES( composite );
179
180                         // We want to ensure that we don't get a wobble...
181                         //mlt_properties_set_int( composite_properties, "distort", 1 );
182                         mlt_properties_set_int( composite_properties, "progressive", 1 );
183
184                         // Pass all the composite. properties on the transition down
185                         mlt_properties_pass( composite_properties, properties, "composite." );
186
187                         // Register the composite for reuse/destruction
188                         mlt_properties_set_data( properties, "composite", composite, 0, ( mlt_destructor )mlt_transition_close, NULL );
189                 }
190         }
191         else
192         {
193                 // Pass all current properties down
194                 mlt_properties composite_properties = MLT_TRANSITION_PROPERTIES( composite );
195                 mlt_properties_pass( composite_properties, properties, "composite." );
196         }
197
198         // Create filters
199         if ( filter == NULL )
200         {
201                 // Loop Variable
202                 int i = 0;
203
204                 // Number of filters created
205                 int count = 0;
206
207                 // Loop for all properties
208                 for ( i = 0; i < mlt_properties_count( properties ); i ++ )
209                 {
210                         // Get the name of this property
211                         char *name = mlt_properties_get_name( properties, i );
212
213                         // If the name does not contain a . and matches filter
214                         if ( strchr( name, '.' ) == NULL && !strncmp( name, "filter", 6 ) )
215                         {
216                                 // Get the filter constructor
217                                 char *value = mlt_properties_get_value( properties, i );
218
219                                 // Create an instance
220                                 if ( create_instance( this, name, value, count ) == 0 )
221                                         count ++;
222                         }
223                 }
224         
225                 // Look for the first filter again
226                 filter = mlt_properties_get_data( properties, "_filter_0", NULL );
227         }
228         else
229         {
230                 // Pass all properties down
231                 mlt_filter temp = NULL;
232
233                 // Loop Variable
234                 int i = 0;
235
236                 // Number of filters found
237                 int count = 0;
238
239                 // Loop for all properties
240                 for ( i = 0; i < mlt_properties_count( properties ); i ++ )
241                 {
242                         // Get the name of this property
243                         char *name = mlt_properties_get_name( properties, i );
244
245                         // If the name does not contain a . and matches filter
246                         if ( strchr( name, '.' ) == NULL && !strncmp( name, "filter", 6 ) )
247                         {
248                                 // Strings to hold the id and pass down key
249                                 char id[ 256 ];
250                                 char key[ 256 ];
251
252                                 // Construct id and key
253                                 sprintf( id, "_filter_%d", count );
254                                 sprintf( key, "%s.", name );
255
256                                 // Get the filter
257                                 temp = mlt_properties_get_data( properties, id, NULL );
258
259                                 if ( temp != NULL )
260                                 {
261                                         mlt_properties_pass( MLT_FILTER_PROPERTIES( temp ), properties, key );
262                                         count ++;
263                                 }
264                         }
265                 }
266         }
267
268         // Get the image
269         error = mlt_frame_get_image( frame, image, format, width, height, 1 );
270
271         // Only continue if we have both filter and composite
272         if ( composite != NULL )
273         {
274                 // Get the resource of this filter (could be a shape [rectangle/circle] or an alpha provider of choice
275                 const char *resource =  mlt_properties_get( properties, "resource" );
276
277                 // Get the old resource in case it's changed
278                 char *old_resource =  mlt_properties_get( properties, "_old_resource" );
279
280                 // String to hold the filter to query on
281                 char id[ 256 ];
282
283                 // Index to hold the count
284                 int i = 0;
285
286                 // We will get the 'b frame' from the composite only if it's NULL
287                 if ( b_frame == NULL )
288                 {
289                         // Copy the region
290                         b_frame = composite_copy_region( composite, frame, position );
291
292                         // Ensure a destructor
293                         mlt_properties_set_data( MLT_FRAME_PROPERTIES( frame ), name, b_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
294                 }
295
296                 // Set the position of the b_frame
297                 mlt_frame_set_position( b_frame, position );
298
299                 // Make sure the filter is in the correct position
300                 while ( filter != NULL )
301                 {
302                         // Stack this filter
303                         if ( mlt_properties_get_int( MLT_FILTER_PROPERTIES( filter ), "off" ) == 0 )
304                                 mlt_filter_process( filter, b_frame );
305
306                         // Generate the key for the next
307                         sprintf( id, "_filter_%d", ++ i );
308
309                         // Get the next filter
310                         filter = mlt_properties_get_data( properties, id, NULL );
311                 }
312
313                 // Allow filters to be attached to a region filter
314                 filter = mlt_properties_get_data( properties, "_region_filter", NULL );
315                 if ( filter != NULL )
316                         mlt_service_apply_filters( MLT_FILTER_SERVICE( filter ), b_frame, 0 );
317
318                 // Hmm - this is probably going to go wrong....
319                 mlt_frame_set_position( frame, position );
320
321                 // Get the b frame and process with composite if successful
322                 mlt_transition_process( composite, frame, b_frame );
323
324                 // If we have a shape producer copy the alpha mask from the shape frame to the b_frame
325                 if ( strcmp( resource, "rectangle" ) != 0 )
326                 {
327                         // Get the producer from the transition
328                         mlt_producer producer = mlt_properties_get_data( properties, "producer", NULL );
329
330                         // If We have no producer then create one
331                         if ( producer == NULL || ( old_resource != NULL && strcmp( resource, old_resource ) ) )
332                         {
333                                 // Get the factory producer service
334                                 char *factory = mlt_properties_get( properties, "factory" );
335
336                                 // Store the old resource
337                                 mlt_properties_set( properties, "_old_resource", resource );
338
339                                 // Special case circle resource
340                                 if ( strcmp( resource, "circle" ) == 0 )
341                                         resource = "pixbuf:<svg width='100' height='100'><circle cx='50' cy='50' r='50' fill='black'/></svg>";
342
343                                 // Create the producer
344                                 mlt_profile profile = mlt_service_profile( MLT_TRANSITION_SERVICE( this ) );
345                                 producer = mlt_factory_producer( profile, factory, resource );
346
347                                 // If we have one
348                                 if ( producer != NULL )
349                                 {
350                                         // Get the producer properties
351                                         mlt_properties producer_properties = MLT_PRODUCER_PROPERTIES( producer );
352
353                                         // Ensure that we loop
354                                         mlt_properties_set( producer_properties, "eof", "loop" );
355
356                                         // Now pass all producer. properties on the transition down
357                                         mlt_properties_pass( producer_properties, properties, "producer." );
358
359                                         // Register the producer for reuse/destruction
360                                         mlt_properties_set_data( properties, "producer", producer, 0, ( mlt_destructor )mlt_producer_close, NULL );
361                                 }
362                         }
363
364                         // Now use the shape producer
365                         if ( producer != NULL )
366                         {
367                                 // We will get the alpha frame from the producer
368                                 mlt_frame shape_frame = NULL;
369
370                                 // Make sure the producer is in the correct position
371                                 mlt_producer_seek( producer, position );
372
373                                 // Get the shape frame
374                                 if ( mlt_service_get_frame( MLT_PRODUCER_SERVICE( producer ), &shape_frame, 0 ) == 0 )
375                                 {
376                                         // Ensure that the shape frame will be closed
377                                         mlt_properties_set_data( MLT_FRAME_PROPERTIES( b_frame ), "shape_frame", shape_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
378
379                                         // Specify the callback for evaluation
380                                         b_frame->get_alpha_mask = filter_get_alpha_mask;
381                                 }
382                         }
383                 }
384
385                 // Get the image
386                 error = mlt_frame_get_image( frame, image, format, width, height, 0 );
387         }
388
389         return error;
390 }
391
392 /** Filter processing.
393 */
394
395 static mlt_frame transition_process( mlt_transition this, mlt_frame a_frame, mlt_frame b_frame )
396 {
397         // Get the properties of the frame
398         mlt_properties properties = MLT_FRAME_PROPERTIES( a_frame );
399
400         // Get a unique name to store the frame position
401         char *name = mlt_properties_get( MLT_TRANSITION_PROPERTIES( this ), "_unique_id" );
402
403         // Assign the current position to the name
404         mlt_properties_set_position( properties, name, mlt_frame_get_position( a_frame ) );
405
406         // Push the transition on to the frame
407         mlt_frame_push_service( a_frame, this );
408
409         // Push the b_frame on to the stack
410         mlt_frame_push_frame( a_frame, b_frame );
411
412         // Push the transition method
413         mlt_frame_push_get_image( a_frame, transition_get_image );
414
415         // Return the frame
416         return a_frame;
417 }
418
419 /** Constructor for the transition.
420 */
421
422 mlt_transition transition_region_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
423 {
424         // Create a new transition
425         mlt_transition this = mlt_transition_new( );
426
427         // Further initialisation
428         if ( this != NULL )
429         {
430                 // Get the properties from the transition
431                 mlt_properties properties = MLT_TRANSITION_PROPERTIES( this );
432
433                 // Assign the transition process method
434                 this->process = transition_process;
435
436                 // Default factory
437                 mlt_properties_set( properties, "factory", "fezzik" );
438                 
439                 // Resource defines the shape of the region
440                 mlt_properties_set( properties, "resource", arg == NULL ? "rectangle" : arg );
441
442                 // Inform apps and framework that this is a video only transition
443                 mlt_properties_set_int( properties, "_transition_type", 1 );
444         }
445
446         // Return the transition
447         return this;
448 }
449