]> git.sesse.net Git - mlt/blob - src/framework/mlt_animation.c
Add Catmull-Rom spline smooth animation interpolation.
[mlt] / src / framework / mlt_animation.c
1 /*
2  * mlt_animation.c -- provides the property animation API
3  * Copyright (C) 2004-2013 Ushodaya Enterprises Limited
4  * Author: Charles Yates <charles.yates@pandora.be>
5  * Author: Dan Dennedy <dan@dennedy.org>
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include "mlt_animation.h"
23 #include "mlt_tokeniser.h"
24 #include "mlt_profile.h"
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29
30 typedef struct animation_node_s *animation_node;
31 struct animation_node_s
32 {
33         struct mlt_animation_item_s item;
34         animation_node next, prev;
35 };
36
37 struct mlt_animation_s
38 {
39         char *data;
40         int length;
41         double fps;
42         locale_t locale;
43         animation_node nodes;
44 };
45
46 // Create a new geometry structure
47 mlt_animation mlt_animation_new( )
48 {
49         mlt_animation self = calloc( 1, sizeof( *self ) );
50         return self;
51 }
52
53 void mlt_animation_interpolate( mlt_animation self )
54 {
55         // Parse all items to ensure non-keyframes are calculated correctly.
56         if ( self->nodes )
57         {
58                 animation_node current = self->nodes;
59                 while ( current )
60                 {
61                         if ( !current->item.is_key )
62                         {
63                                 double progress;
64                                 mlt_property points[4];
65                                 animation_node prev = current->prev;
66                                 animation_node next = current->next;
67
68                                 while ( prev && !prev->item.is_key ) prev = prev->prev;
69                                 while ( next && !next->item.is_key ) next = next->next;
70
71                                 if ( !prev )
72                                         current->item.is_key = 1;
73                                 points[0] = prev->prev? prev->prev->item.property : prev->item.property;
74                                 points[1] = prev->item.property;
75                                 points[2] = next->item.property;
76                                 points[3] = next->next? next->next->item.property : next->item.property;
77                                 progress = current->item.frame - prev->item.frame;
78                                 progress /= next->item.frame - prev->item.frame;
79                                 mlt_property_interpolate( current->item.property, points, progress,
80                                         self->fps, self->locale, current->item.keyframe_type );
81                         }
82
83                         // Move to the next item
84                         current = current->next;
85                 }
86         }
87 }
88
89 static int mlt_animation_drop( mlt_animation self, animation_node node )
90 {
91         if ( node == self->nodes )
92         {
93                 self->nodes = node->next;
94                 if ( self->nodes ) {
95                         self->nodes->prev = NULL;
96                         self->nodes->item.is_key = 1;
97                 }
98         }
99         else if ( node->next && node->prev )
100         {
101                 node->prev->next = node->next;
102                 node->next->prev = node->prev;
103         }
104         else if ( node->next )
105         {
106                 node->next->prev = node->prev;
107         }
108         else if ( node->prev )
109         {
110                 node->prev->next = node->next;
111         }
112         mlt_property_close( node->item.property );
113         free( node );
114
115         return 0;
116 }
117
118 static void mlt_animation_clean( mlt_animation self )
119 {
120         if ( self->data )
121                 free( self->data );
122         self->data = NULL;
123         while ( self->nodes )
124                 mlt_animation_drop( self, self->nodes );
125 }
126
127 int mlt_animation_parse(mlt_animation self, const char *data, int length, double fps, locale_t locale )
128 {
129         int error = 0;
130         int i = 0;
131         struct mlt_animation_item_s item;
132         mlt_tokeniser tokens = mlt_tokeniser_init( );
133
134         // Clean the existing geometry
135         mlt_animation_clean( self );
136
137         // Update the info on the data
138         if ( data )
139                 self->data = strdup( data );
140         self->length = length;
141         self->fps = fps;
142         self->locale = locale;
143         item.property = mlt_property_init();
144
145         // Tokenise
146         if ( data )
147                 mlt_tokeniser_parse_new( tokens, (char*) data, ";" );
148
149         // Iterate through each token
150         for ( i = 0; i < mlt_tokeniser_count( tokens ); i++ )
151         {
152                 char *value = mlt_tokeniser_get_string( tokens, i );
153
154                 // If no data in keyframe, drop it (trailing semicolon)
155                 if ( !value || !strcmp( value, "" ) )
156                         continue;
157
158                 // Reset item
159                 item.frame = item.is_key = 0;
160
161                 // Now parse the item
162                 mlt_animation_parse_item( self, &item, value );
163
164                 // Now insert into place
165                 mlt_animation_insert( self, &item );
166         }
167         mlt_animation_interpolate( self );
168
169         // Cleanup
170         mlt_tokeniser_close( tokens );
171         mlt_property_close( item.property );
172
173         return error;
174 }
175
176 // Conditionally refresh in case of a change
177 int mlt_animation_refresh(mlt_animation self, const char *data, int length)
178 {
179         if ( ( length != self->length )|| ( data && ( !self->data || strcmp( data, self->data ) ) ) )
180                 return mlt_animation_parse( self, data, length, self->fps, self->locale );
181         return 0;
182 }
183
184 int mlt_animation_get_length( mlt_animation self )
185 {
186         if ( self )
187                 return self->length;
188         else
189                 return 0;
190 }
191
192 void mlt_animation_set_length( mlt_animation self, int length )
193 {
194         if ( self )
195                 self->length = length;
196 }
197
198 int mlt_animation_parse_item( mlt_animation self, mlt_animation_item item, const char *value )
199 {
200         int error = 0;
201
202         if ( value && strcmp( value, "" ) )
203         {
204                 // Determine if a position has been specified
205                 if ( strchr( value, '=' ) )
206                 {
207                         double temp;
208                         char *p = NULL;
209 #if defined(__GLIBC__) || defined(__DARWIN__)
210                         if ( self->locale )
211                                 temp = strtod_l( value, &p, self->locale );
212                         else
213 #endif
214                         temp = strtod( value, &p );
215                         // If p == value then it is likely a time clock or time code.
216                         if ( temp > -1.0 && temp < 1.0 && p != value )
217                         {
218                                 // Parse a relative time (-1, 1).
219                                 item->frame = temp * self->length;
220                         }
221                         else
222                         {
223                                 // Parse an absolute time value.
224                                 mlt_property_set_string( item->property, value );
225                                 item->frame = mlt_property_get_int( item->property, self->fps, self->locale );
226                         }
227
228                         // The character preceeding the equal sign indicates interpolation method.
229                         p = strchr( value, '=' ) - 1;
230                         if ( p[0] == '|' || p[0] == '!' )
231                                 item->keyframe_type = mlt_keyframe_discrete;
232                         else if ( p[0] == '~' )
233                                 item->keyframe_type = mlt_keyframe_smooth;
234                         else
235                                 item->keyframe_type = mlt_keyframe_linear;
236                         value = &p[2];
237                 }
238
239                 // Special case - frame < 0
240                 if ( item->frame < 0 )
241                         item->frame += self->length;
242
243                 // Set remainder of string as item value.
244                 mlt_property_set_string( item->property, value );
245                 item->is_key = 1;
246         }
247         else
248         {
249                 error = 1;
250         }
251
252         return error;
253 }
254
255 // Fetch a geometry item for an absolute position
256 int mlt_animation_get_item( mlt_animation self, mlt_animation_item item, int position )
257 {
258         int error = 0;
259         // Need to find the nearest keyframe to the position specifed
260         animation_node node = self->nodes;
261
262         // Iterate through the keyframes until we reach last or have
263         while ( node && node->next && position >= node->next->item.frame )
264                 node = node->next;
265
266         if ( node )
267         {
268                 item->keyframe_type = node->item.keyframe_type;
269
270                 // Position is before the first keyframe.
271                 if ( position < node->item.frame )
272                 {
273                         item->is_key = 0;
274                         mlt_property_pass( item->property, node->item.property );
275                 }
276                 // Item exists.
277                 else if ( position == node->item.frame )
278                 {
279                         item->is_key = node->item.is_key;
280                         mlt_property_pass( item->property, node->item.property );
281                 }
282                 // Position is after the last keyframe.
283                 else if ( !node->next )
284                 {
285                         item->is_key = 0;
286                         mlt_property_pass( item->property, node->item.property );
287                 }
288                 // Interpolation needed.
289                 else
290                 {
291                         double progress;
292                         mlt_property points[4];
293                         points[0] = node->prev? node->prev->item.property : node->item.property;
294                         points[1] = node->item.property;
295                         points[2] = node->next->item.property;
296                         points[3] = node->next->next? node->next->next->item.property : node->next->item.property;
297                         progress = position - node->item.frame;
298                         progress /= node->next->item.frame - node->item.frame;
299                         mlt_property_interpolate( item->property, points, progress,
300                                 self->fps, self->locale, item->keyframe_type );
301                         item->is_key = 0;
302                 }
303         }
304         else
305         {
306                 item->frame = item->is_key = 0;
307                 error = 1;
308         }
309         item->frame = position;
310
311         return error;
312 }
313
314 // Specify an animation item at an absolute position
315 int mlt_animation_insert( mlt_animation self, mlt_animation_item item )
316 {
317         int error = 0;
318         animation_node node = calloc( 1, sizeof( *node ) );
319         node->item.frame = item->frame;
320         node->item.is_key = 1;
321         node->item.keyframe_type = item->keyframe_type;
322         node->item.property = mlt_property_init();
323         mlt_property_pass( node->item.property, item->property );
324
325         // Determine if we need to insert or append to the list, or if it's a new list
326         if ( self->nodes )
327         {
328                 // Get the first item
329                 animation_node current = self->nodes;
330
331                 // Locate an existing nearby item
332                 while ( current->next && item->frame > current->item.frame )
333                         current = current->next;
334
335                 if ( item->frame < current->item.frame )
336                 {
337                         if ( current == self->nodes )
338                                 self->nodes = node;
339                         if ( current->prev )
340                                 current->prev->next = node;
341                         node->next = current;
342                         node->prev = current->prev;
343                         current->prev = node;
344                 }
345                 else if ( item->frame > current->item.frame )
346                 {
347                         if ( current->next )
348                                 current->next->prev = node;
349                         node->next = current->next;
350                         node->prev = current;
351                         current->next = node;
352                 }
353                 else
354                 {
355                         // Update matching node.
356                         current->item.frame = item->frame;
357                         current->item.is_key = 1;
358                         current->item.keyframe_type = item->keyframe_type;
359                         mlt_property_close( current->item.property );
360                         current->item.property = node->item.property;
361                         free( node );
362                 }
363         }
364         else
365         {
366                 // Set the first item
367                 self->nodes = node;
368         }
369
370         return error;
371 }
372
373 // Remove the keyframe at the specified position
374 int mlt_animation_remove( mlt_animation self, int position )
375 {
376         int error = 1;
377         animation_node node = self->nodes;
378
379         while ( node && position != node->item.frame )
380                 node = node->next;
381
382         if ( node && position == node->item.frame )
383                 error = mlt_animation_drop( self, node );
384
385         return error;
386 }
387
388 // Get the keyfame at the position or the next following
389 int mlt_animation_next_key( mlt_animation self, mlt_animation_item item, int position )
390 {
391         animation_node node = self->nodes;
392
393         while ( node && position > node->item.frame )
394                 node = node->next;
395
396         if ( node )
397         {
398                 item->frame = node->item.frame;
399                 item->is_key = node->item.is_key;
400                 item->keyframe_type = node->item.keyframe_type;
401                 mlt_property_pass( item->property, node->item.property );
402         }
403
404         return ( node == NULL );
405 }
406
407 // Get the keyframe at the position or the previous key
408 int mlt_animation_prev_key( mlt_animation self, mlt_animation_item item, int position )
409 {
410         animation_node node = self->nodes;
411
412         while ( node && node->next && position >= node->next->item.frame )
413                 node = node->next;
414
415         if ( node )
416         {
417                 item->frame = node->item.frame;
418                 item->is_key = node->item.is_key;
419                 item->keyframe_type = node->item.keyframe_type;
420                 mlt_property_pass( item->property, node->item.property );
421         }
422
423         return ( node == NULL );
424 }
425
426 char *mlt_animation_serialize_cut( mlt_animation self, int in, int out )
427 {
428         struct mlt_animation_item_s item;
429         char *ret = malloc( 1000 );
430         size_t used = 0;
431         size_t size = 1000;
432
433         item.property = mlt_property_init();
434         if ( in == -1 )
435                 in = 0;
436         if ( out == -1 )
437                 out = mlt_animation_get_length( self );
438
439         if ( ret )
440         {
441                 strcpy( ret, "" );
442
443                 item.frame = in;
444
445                 while ( 1 )
446                 {
447                         size_t item_len = 0;
448
449                         // If it's the first frame, then it's not necessarily a key
450                         if ( item.frame == in )
451                         {
452                                 if ( mlt_animation_get_item( self, &item, item.frame ) )
453                                         break;
454
455                                 // If the first keyframe is larger than the current position
456                                 // then do nothing here
457                                 if ( self->nodes->item.frame > item.frame )
458                                 {
459                                         item.frame ++;
460                                         continue;
461                                 }
462
463                                 // To ensure correct seeding
464                                 item.is_key = 1;
465                         }
466                         // Typically, we move from keyframe to keyframe
467                         else if ( item.frame < out )
468                         {
469                                 if ( mlt_animation_next_key( self, &item, item.frame ) )
470                                         break;
471
472                                 // Special case - crop at the out point
473                                 if ( item.frame > out )
474                                         mlt_animation_get_item( self, &item, out );
475                         }
476                         // We've handled the last keyframe
477                         else
478                         {
479                                 break;
480                         }
481
482                         // Determine length of string to be appended.
483                         if ( item.frame - in != 0 )
484                                 item_len += 20;
485                         if ( item.is_key )
486                                 item_len += strlen( mlt_property_get_string_l( item.property, self->locale ) );
487
488                         // Reallocate return string to be long enough.
489                         while ( used + item_len + 2 > size ) // +2 for ';' and NULL
490                         {
491                                 size += 1000;
492                                 ret = realloc( ret, size );
493                         }
494
495                         // Append item delimiter (;) if needed.
496                         if ( ret && used > 0 )
497                         {
498                                 used ++;
499                                 strcat( ret, ";" );
500                         }
501                         if ( ret )
502                         {
503                                 // Append keyframe time and keyframe/value delimiter (=).
504                                 const char *s;
505                                 switch (item.keyframe_type) {
506                                 case mlt_keyframe_discrete:
507                                         s = "|";
508                                         break;
509                                 case mlt_keyframe_smooth:
510                                         s = "~";
511                                         break;
512                                 default:
513                                         s = "";
514                                         break;
515                                 }
516                                 sprintf( ret + used, "%d%s=", item.frame - in, s );
517
518                                 // Append item value.
519                                 if ( item.is_key )
520                                         strcat( ret, mlt_property_get_string_l( item.property, self->locale ) );
521                                 used = strlen( ret );
522                         }
523                         item.frame ++;
524                 }
525         }
526         mlt_property_close( item.property );
527
528         return ret;
529 }
530
531 // Serialise the current geometry
532 char *mlt_animation_serialize( mlt_animation self )
533 {
534         char *ret = mlt_animation_serialize_cut( self, 0, self->length );
535         if ( ret )
536         {
537                 if ( self->data )
538                         free( self->data );
539                 self->data = ret;
540         }
541         return strdup( ret );
542 }
543
544 // Close the geometry
545 void mlt_animation_close( mlt_animation self )
546 {
547         if ( self )
548         {
549                 mlt_animation_clean( self );
550                 free( self );
551         }
552 }