]> git.sesse.net Git - mlt/blob - src/framework/mlt_frame.c
src/framework/mlt_frame.c
[mlt] / src / framework / mlt_frame.c
1 /*
2  * mlt_frame.c -- interface for all frame classes
3  * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
4  * Author: Charles Yates <charles.yates@pandora.be>
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 "config.h"
22 #include "mlt_frame.h"
23 #include "mlt_producer.h"
24 #include "mlt_factory.h"
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28
29 /** Constructor for a frame.
30 */
31
32 mlt_frame mlt_frame_init( )
33 {
34         // Allocate a frame
35         mlt_frame this = calloc( sizeof( struct mlt_frame_s ), 1 );
36
37         if ( this != NULL )
38         {
39                 // Get the normalisation
40                 char *normalisation = mlt_environment( "MLT_NORMALISATION" );
41
42                 // Initialise the properties
43                 mlt_properties properties = &this->parent;
44                 mlt_properties_init( properties, this );
45
46                 // Set default properties on the frame
47                 mlt_properties_set_position( properties, "_position", 0.0 );
48                 mlt_properties_set_data( properties, "image", NULL, 0, NULL, NULL );
49
50                 if ( normalisation == NULL || strcmp( normalisation, "NTSC" ) )
51                 {
52                         mlt_properties_set_int( properties, "width", 720 );
53                         mlt_properties_set_int( properties, "height", 576 );
54                         mlt_properties_set_int( properties, "normalised_width", 720 );
55                         mlt_properties_set_int( properties, "normalised_height", 576 );
56                         mlt_properties_set_double( properties, "aspect_ratio", 59.0/54.0 );
57                 }
58                 else
59                 {
60                         mlt_properties_set_int( properties, "width", 720 );
61                         mlt_properties_set_int( properties, "height", 480 );
62                         mlt_properties_set_int( properties, "normalised_width", 720 );
63                         mlt_properties_set_int( properties, "normalised_height", 480 );
64                         mlt_properties_set_double( properties, "aspect_ratio", 10.0/11.0 );
65                 }
66
67                 mlt_properties_set_data( properties, "audio", NULL, 0, NULL, NULL );
68                 mlt_properties_set_data( properties, "alpha", NULL, 0, NULL, NULL );
69
70                 // Construct stacks for frames and methods
71                 this->stack_image = mlt_deque_init( );
72                 this->stack_audio = mlt_deque_init( );
73                 this->stack_service = mlt_deque_init( );
74         }
75
76         return this;
77 }
78
79 /** Fetch the frames properties.
80 */
81
82 mlt_properties mlt_frame_properties( mlt_frame this )
83 {
84         return this != NULL ? &this->parent : NULL;
85 }
86
87 /** Check if we have a way to derive something other than a test card.
88 */
89
90 int mlt_frame_is_test_card( mlt_frame this )
91 {
92         return mlt_deque_count( this->stack_image ) == 0 || mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "test_image" );
93 }
94
95 /** Check if we have a way to derive something other than test audio.
96 */
97
98 int mlt_frame_is_test_audio( mlt_frame this )
99 {
100         return mlt_deque_count( this->stack_audio ) == 0 || mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "test_audio" );
101 }
102
103 /** Get the aspect ratio of the frame.
104 */
105
106 double mlt_frame_get_aspect_ratio( mlt_frame this )
107 {
108         return mlt_properties_get_double( MLT_FRAME_PROPERTIES( this ), "aspect_ratio" );
109 }
110
111 /** Set the aspect ratio of the frame.
112 */
113
114 int mlt_frame_set_aspect_ratio( mlt_frame this, double value )
115 {
116         return mlt_properties_set_double( MLT_FRAME_PROPERTIES( this ), "aspect_ratio", value );
117 }
118
119 /** Get the position of this frame.
120 */
121
122 mlt_position mlt_frame_get_position( mlt_frame this )
123 {
124         int pos = mlt_properties_get_position( MLT_FRAME_PROPERTIES( this ), "_position" );
125         return pos < 0 ? 0 : pos;
126 }
127
128 /** Set the position of this frame.
129 */
130
131 int mlt_frame_set_position( mlt_frame this, mlt_position value )
132 {
133         return mlt_properties_set_position( MLT_FRAME_PROPERTIES( this ), "_position", value );
134 }
135
136 /** Stack a get_image callback.
137 */
138
139 int mlt_frame_push_get_image( mlt_frame this, mlt_get_image get_image )
140 {
141         return mlt_deque_push_back( this->stack_image, get_image );
142 }
143
144 /** Pop a get_image callback.
145 */
146
147 mlt_get_image mlt_frame_pop_get_image( mlt_frame this )
148 {
149         return mlt_deque_pop_back( this->stack_image );
150 }
151
152 /** Push a frame.
153 */
154
155 int mlt_frame_push_frame( mlt_frame this, mlt_frame that )
156 {
157         return mlt_deque_push_back( this->stack_image, that );
158 }
159
160 /** Pop a frame.
161 */
162
163 mlt_frame mlt_frame_pop_frame( mlt_frame this )
164 {
165         return mlt_deque_pop_back( this->stack_image );
166 }
167
168 /** Push a service.
169 */
170
171 int mlt_frame_push_service( mlt_frame this, void *that )
172 {
173         return mlt_deque_push_back( this->stack_image, that );
174 }
175
176 /** Pop a service.
177 */
178
179 void *mlt_frame_pop_service( mlt_frame this )
180 {
181         return mlt_deque_pop_back( this->stack_image );
182 }
183
184 /** Push a service.
185 */
186
187 int mlt_frame_push_service_int( mlt_frame this, int that )
188 {
189         return mlt_deque_push_back_int( this->stack_image, that );
190 }
191
192 /** Pop a service.
193 */
194
195 int mlt_frame_pop_service_int( mlt_frame this )
196 {
197         return mlt_deque_pop_back_int( this->stack_image );
198 }
199
200 /** Push an audio item on the stack.
201 */
202
203 int mlt_frame_push_audio( mlt_frame this, void *that )
204 {
205         return mlt_deque_push_back( this->stack_audio, that );
206 }
207
208 /** Pop an audio item from the stack
209 */
210
211 void *mlt_frame_pop_audio( mlt_frame this )
212 {
213         return mlt_deque_pop_back( this->stack_audio );
214 }
215
216 /** Return the service stack
217 */
218
219 mlt_deque mlt_frame_service_stack( mlt_frame this )
220 {
221         return this->stack_service;
222 }
223
224 /** Replace image stack with the information provided.
225
226         This might prove to be unreliable and restrictive - the idea is that a transition
227         which normally uses two images may decide to only use the b frame (ie: in the case
228         of a composite where the b frame completely obscures the a frame).
229
230         The image must be writable and the destructor for the image itself must be taken
231         care of on another frame and that frame cannot have a replace applied to it... 
232         Further it assumes that no alpha mask is in use.
233
234         For these reasons, it can only be used in a specific situation - when you have 
235         multiple tracks each with their own transition and these transitions are applied
236         in a strictly reversed order (ie: highest numbered [lowest track] is processed 
237         first).
238
239         More reliable approach - the cases should be detected during the process phase
240         and the upper tracks should simply not be invited to stack...
241 */
242
243 void mlt_frame_replace_image( mlt_frame this, uint8_t *image, mlt_image_format format, int width, int height )
244 {
245         // Remove all items from the stack
246         while( mlt_deque_pop_back( this->stack_image ) ) ;
247
248         // Update the information 
249         mlt_properties_set_data( MLT_FRAME_PROPERTIES( this ), "image", image, 0, NULL, NULL );
250         mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "width", width );
251         mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "height", height );
252         mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "format", format );
253         this->get_alpha_mask = NULL;
254 }
255
256 /** Get the image associated to the frame.
257 */
258
259 int mlt_frame_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
260 {
261         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
262         mlt_get_image get_image = mlt_frame_pop_get_image( this );
263         mlt_producer producer = mlt_properties_get_data( properties, "test_card_producer", NULL );
264         int error = 0;
265
266         if ( get_image != NULL )
267         {
268                 mlt_properties_set_int( properties, "image_count", mlt_properties_get_int( properties, "image_count" ) - 1 );
269                 mlt_position position = mlt_frame_get_position( this );
270                 error = get_image( this, buffer, format, width, height, writable );
271                 mlt_properties_set_int( properties, "width", *width );
272                 mlt_properties_set_int( properties, "height", *height );
273                 mlt_properties_set_int( properties, "format", *format );
274                 mlt_frame_set_position( this, position );
275         }
276         else if ( mlt_properties_get_data( properties, "image", NULL ) != NULL )
277         {
278                 *format = mlt_properties_get_int( properties, "format" );
279                 *buffer = mlt_properties_get_data( properties, "image", NULL );
280                 *width = mlt_properties_get_int( properties, "width" );
281                 *height = mlt_properties_get_int( properties, "height" );
282         }
283         else if ( producer != NULL )
284         {
285                 mlt_frame test_frame = NULL;
286                 mlt_service_get_frame( MLT_PRODUCER_SERVICE( producer ), &test_frame, 0 );
287                 if ( test_frame != NULL )
288                 {
289                         mlt_properties test_properties = MLT_FRAME_PROPERTIES( test_frame );
290                         mlt_properties_set_double( test_properties, "consumer_aspect_ratio", mlt_properties_get_double( properties, "consumer_aspect_ratio" ) );
291                         mlt_properties_set( test_properties, "rescale.interp", mlt_properties_get( properties, "rescale.interp" ) );
292                         mlt_frame_get_image( test_frame, buffer, format, width, height, writable );
293                         mlt_properties_set_data( properties, "test_card_frame", test_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
294                         mlt_properties_set_data( properties, "image", *buffer, *width * *height * 2, NULL, NULL );
295                         mlt_properties_set_int( properties, "width", *width );
296                         mlt_properties_set_int( properties, "height", *height );
297                         mlt_properties_set_int( properties, "format", *format );
298                         mlt_properties_set_double( properties, "aspect_ratio", mlt_frame_get_aspect_ratio( test_frame ) );
299                 }
300                 else
301                 {
302                         mlt_properties_set_data( properties, "test_card_producer", NULL, 0, NULL, NULL );
303                         mlt_frame_get_image( this, buffer, format, width, height, writable );
304                 }
305         }
306         else
307         {
308                 register uint8_t *p;
309                 register uint8_t *q;
310                 int size = 0;
311
312                 *width = *width == 0 ? 720 : *width;
313                 *height = *height == 0 ? 576 : *height;
314                 size = *width * *height;
315
316                 mlt_properties_set_int( properties, "format", *format );
317                 mlt_properties_set_int( properties, "width", *width );
318                 mlt_properties_set_int( properties, "height", *height );
319                 mlt_properties_set_int( properties, "aspect_ratio", 0 );
320
321                 switch( *format )
322                 {
323                         case mlt_image_none:
324                                 size = 0;
325                                 *buffer = NULL;
326                                 break;
327                         case mlt_image_rgb24:
328                                 size *= 3;
329                                 size += *width * 3;
330                                 *buffer = mlt_pool_alloc( size );
331                                 if ( *buffer )
332                                         memset( *buffer, 255, size );
333                                 break;
334                         case mlt_image_rgb24a:
335                         case mlt_image_opengl:
336                                 size *= 4;
337                                 size += *width * 4;
338                                 *buffer = mlt_pool_alloc( size );
339                                 if ( *buffer )
340                                         memset( *buffer, 255, size );
341                                 break;
342                         case mlt_image_yuv422:
343                                 size *= 2;
344                                 size += *width * 2;
345                                 *buffer = mlt_pool_alloc( size );
346                                 p = *buffer;
347                                 q = p + size;
348                                 while ( p != NULL && p != q )
349                                 {
350                                         *p ++ = 235;
351                                         *p ++ = 128;
352                                 }
353                                 break;
354                         case mlt_image_yuv420p:
355                                 size = size * 3 / 2;
356                                 *buffer = mlt_pool_alloc( size );
357                                 if ( *buffer )
358                                         memset( *buffer, 255, size );
359                                 break;
360                 }
361
362                 mlt_properties_set_data( properties, "image", *buffer, size, ( mlt_destructor )mlt_pool_release, NULL );
363                 mlt_properties_set_int( properties, "test_image", 1 );
364         }
365
366         mlt_properties_set_int( properties, "scaled_width", *width );
367         mlt_properties_set_int( properties, "scaled_height", *height );
368
369         return error;
370 }
371
372 uint8_t *mlt_frame_get_alpha_mask( mlt_frame this )
373 {
374         uint8_t *alpha = NULL;
375         if ( this != NULL )
376         {
377                 if ( this->get_alpha_mask != NULL )
378                         alpha = this->get_alpha_mask( this );
379                 if ( alpha == NULL )
380                         alpha = mlt_properties_get_data( &this->parent, "alpha", NULL );
381                 if ( alpha == NULL )
382                 {
383                         int size = mlt_properties_get_int( &this->parent, "scaled_width" ) * mlt_properties_get_int( &this->parent, "scaled_height" );
384                         alpha = mlt_pool_alloc( size );
385                         memset( alpha, 255, size );
386                         mlt_properties_set_data( &this->parent, "alpha", alpha, size, mlt_pool_release, NULL );
387                 }
388         }
389         return alpha;
390 }
391
392 int mlt_frame_get_audio( mlt_frame this, int16_t **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
393 {
394         mlt_get_audio get_audio = mlt_frame_pop_audio( this );
395         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
396         int hide = mlt_properties_get_int( properties, "test_audio" );
397
398         if ( hide == 0 && get_audio != NULL )
399         {
400                 mlt_position position = mlt_frame_get_position( this );
401                 get_audio( this, buffer, format, frequency, channels, samples );
402                 mlt_frame_set_position( this, position );
403         }
404         else if ( mlt_properties_get_data( properties, "audio", NULL ) )
405         {
406                 *buffer = mlt_properties_get_data( properties, "audio", NULL );
407                 *frequency = mlt_properties_get_int( properties, "audio_frequency" );
408                 *channels = mlt_properties_get_int( properties, "audio_channels" );
409                 *samples = mlt_properties_get_int( properties, "audio_samples" );
410         }
411         else
412         {
413                 int size = 0;
414                 *samples = *samples <= 0 ? 1920 : *samples;
415                 *channels = *channels <= 0 ? 2 : *channels;
416                 *frequency = *frequency <= 0 ? 48000 : *frequency;
417                 size = *samples * *channels * sizeof( int16_t );
418                 *buffer = mlt_pool_alloc( size );
419                 if ( *buffer != NULL )
420                         memset( *buffer, 0, size );
421                 mlt_properties_set_data( properties, "audio", *buffer, size, ( mlt_destructor )mlt_pool_release, NULL );
422                 mlt_properties_set_int( properties, "test_audio", 1 );
423         }
424
425         mlt_properties_set_int( properties, "audio_frequency", *frequency );
426         mlt_properties_set_int( properties, "audio_channels", *channels );
427         mlt_properties_set_int( properties, "audio_samples", *samples );
428
429         if ( mlt_properties_get( properties, "meta.volume" ) )
430         {
431                 double value = mlt_properties_get_double( properties, "meta.volume" );
432                 if ( value == 0.0 )
433                 {
434                         memset( *buffer, 0, *samples * *channels * 2 );
435                         mlt_properties_set_double( properties, "meta.volume", 1.0 );
436                 }
437                 else if ( value != 1.0 )
438                 {
439                         int total = *samples * *channels;
440                         int16_t *p = *buffer;
441                         while ( total -- )
442                         {
443                                 *p = *p * value;
444                                 p ++;
445                         }
446                         mlt_properties_set_double( properties, "meta.volume", 1.0 );
447                 }
448         }
449
450         return 0;
451 }
452
453 unsigned char *mlt_frame_get_waveform( mlt_frame this, int w, int h )
454 {
455         int16_t *pcm = NULL;
456         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
457         mlt_audio_format format = mlt_audio_pcm;
458         int frequency = 32000; // lower frequency available?
459         int channels = 2;
460         double fps = mlt_properties_get_double( properties, "fps" );
461         int samples = mlt_sample_calculator( fps, frequency, mlt_frame_get_position( this ) );
462         
463         // Get the pcm data
464         mlt_frame_get_audio( this, &pcm, &format, &frequency, &channels, &samples );
465         
466         // Make an 8-bit buffer large enough to hold rendering
467         int size = w * h;
468         unsigned char *bitmap = ( unsigned char* )mlt_pool_alloc( size );
469         if ( bitmap != NULL )
470                 memset( bitmap, 0, size );
471         mlt_properties_set_data( properties, "waveform", bitmap, size, ( mlt_destructor )mlt_pool_release, NULL );
472         
473         // Render vertical lines
474         int16_t *ubound = pcm + samples * channels;
475         int skip = samples / w - 1;
476         int i, j, k;
477         
478         // Iterate sample stream and along x coordinate
479         for ( i = 0; i < w && pcm < ubound; i++ )
480         {
481                 // pcm data has channels interleaved
482                 for ( j = 0; j < channels; j++ )
483                 {
484                         // Determine sample's magnitude from 2s complement;
485                         int pcm_magnitude = *pcm < 0 ? ~(*pcm) + 1 : *pcm;
486                         // The height of a line is the ratio of the magnitude multiplied by 
487                         // half the vertical resolution
488                         int height = ( int )( ( double )( pcm_magnitude ) / 32768 * h / 2 );
489                         // Determine the starting y coordinate - left channel above center,
490                         // right channel below - currently assumes 2 channels
491                         int displacement = ( h / 2 ) - ( 1 - j ) * height;
492                         // Position buffer pointer using y coordinate, stride, and x coordinate
493                         unsigned char *p = &bitmap[ i + displacement * w ];
494                         
495                         // Draw vertical line
496                         for ( k = 0; k < height; k++ )
497                                 p[ w * k ] = 0xFF;
498                         
499                         pcm++;
500                 }
501                 pcm += skip * channels;
502         }
503
504         return bitmap;
505 }
506
507 mlt_producer mlt_frame_get_original_producer( mlt_frame this )
508 {
509         if ( this != NULL )
510                 return mlt_properties_get_data( MLT_FRAME_PROPERTIES( this ), "_producer", NULL );
511         return NULL;
512 }
513
514 void mlt_frame_close( mlt_frame this )
515 {
516         if ( this != NULL && mlt_properties_dec_ref( MLT_FRAME_PROPERTIES( this ) ) <= 0 )
517         {
518                 mlt_deque_close( this->stack_image );
519                 mlt_deque_close( this->stack_audio );
520                 while( mlt_deque_peek_back( this->stack_service ) )
521                         mlt_service_close( mlt_deque_pop_back( this->stack_service ) );
522                 mlt_deque_close( this->stack_service );
523                 mlt_properties_close( &this->parent );
524                 free( this );
525         }
526 }
527
528 /***** convenience functions *****/
529
530 int mlt_convert_rgb24a_to_yuv422( uint8_t *rgba, int width, int height, int stride, uint8_t *yuv, uint8_t *alpha )
531 {
532         int ret = 0;
533         register int y0, y1, u0, u1, v0, v1;
534         register int r, g, b;
535         register uint8_t *d = yuv;
536         register int i, j;
537
538         for ( i = 0; i < height; i++ )
539         {
540                 register uint8_t *s = rgba + ( stride * i );
541                 for ( j = 0; j < ( width / 2 ); j++ )
542                 {
543                         r = *s++;
544                         g = *s++;
545                         b = *s++;
546                         *alpha++ = *s++;
547                         RGB2YUV (r, g, b, y0, u0 , v0);
548                         r = *s++;
549                         g = *s++;
550                         b = *s++;
551                         *alpha++ = *s++;
552                         RGB2YUV (r, g, b, y1, u1 , v1);
553                         *d++ = y0;
554                         *d++ = (u0+u1) >> 1;
555                         *d++ = y1;
556                         *d++ = (v0+v1) >> 1;
557                 }
558                 if ( width % 2 )
559                 {
560                         r = *s++;
561                         g = *s++;
562                         b = *s++;
563                         *alpha++ = *s++;
564                         RGB2YUV (r, g, b, y0, u0 , v0);
565                         *d++ = y0;
566                         *d++ = u0;
567                 }
568         }
569         return ret;
570 }
571
572 int mlt_convert_rgb24_to_yuv422( uint8_t *rgb, int width, int height, int stride, uint8_t *yuv )
573 {
574         int ret = 0;
575         register int y0, y1, u0, u1, v0, v1;
576         register int r, g, b;
577         register uint8_t *d = yuv;
578         register int i, j;
579
580         for ( i = 0; i < height; i++ )
581         {
582                 register uint8_t *s = rgb + ( stride * i );
583                 for ( j = 0; j < ( width / 2 ); j++ )
584                 {
585                         r = *s++;
586                         g = *s++;
587                         b = *s++;
588                         RGB2YUV (r, g, b, y0, u0 , v0);
589                         r = *s++;
590                         g = *s++;
591                         b = *s++;
592                         RGB2YUV (r, g, b, y1, u1 , v1);
593                         *d++ = y0;
594                         *d++ = (u0+u1) >> 1;
595                         *d++ = y1;
596                         *d++ = (v0+v1) >> 1;
597                 }
598                 if ( width % 2 )
599                 {
600                         r = *s++;
601                         g = *s++;
602                         b = *s++;
603                         RGB2YUV (r, g, b, y0, u0 , v0);
604                         *d++ = y0;
605                         *d++ = u0;
606                 }
607         }
608         return ret;
609 }
610
611 int mlt_convert_yuv420p_to_yuv422( uint8_t *yuv420p, int width, int height, int stride, uint8_t *yuv )
612 {
613         int ret = 0;
614         register int i, j;
615
616         int half = width >> 1;
617
618         uint8_t *Y = yuv420p;
619         uint8_t *U = Y + width * height;
620         uint8_t *V = U + width * height / 4;
621
622         register uint8_t *d = yuv;
623
624         for ( i = 0; i < height; i++ )
625         {
626                 register uint8_t *u = U + ( i / 2 ) * ( half );
627                 register uint8_t *v = V + ( i / 2 ) * ( half );
628
629                 for ( j = 0; j < half; j++ )
630                 {
631                         *d ++ = *Y ++;
632                         *d ++ = *u ++;
633                         *d ++ = *Y ++;
634                         *d ++ = *v ++;
635                 }
636         }
637         return ret;
638 }
639
640 uint8_t *mlt_resize_alpha( uint8_t *input, int owidth, int oheight, int iwidth, int iheight )
641 {
642         uint8_t *output = NULL;
643
644         if ( input != NULL && ( iwidth != owidth || iheight != oheight ) && ( owidth > 6 && oheight > 6 ) )
645         {
646                 uint8_t *in_line = input;
647                 uint8_t *out_line;
648
649                 output = mlt_pool_alloc( owidth * oheight );
650                 memset( output, 0, owidth * oheight );
651
652                 out_line = output + ( ( oheight - iheight ) / 2 ) * owidth;
653                 out_line += 2 * ( int )( ( owidth - iwidth ) / 2 );
654
655                 // Loop for the entirety of our output height.
656                 while ( iheight -- )
657                 {
658                         // We're in the input range for this row.
659                         memcpy( out_line, input, iwidth );
660
661                         // Move to next input line
662                         in_line += iwidth;
663
664                 // Move to next output line
665                 out_line += owidth;
666                 }
667         }
668
669         return output;
670 }
671
672 void mlt_resize_yuv422( uint8_t *output, int owidth, int oheight, uint8_t *input, int iwidth, int iheight )
673 {
674         // Calculate strides
675         int istride = iwidth * 2;
676         int ostride = owidth * 2;
677
678         // Optimisation point
679         if ( output == NULL || input == NULL || ( owidth <= 6 || oheight <= 6 || iwidth <= 6 || oheight <= 6 ) )
680         {
681                 return;
682         }
683         else if ( iwidth == owidth && iheight == oheight )
684         {
685                 memcpy( output, input, iheight * istride );
686                 return;
687         }
688
689         uint8_t *in_line = input;
690         uint8_t *out_line;
691
692         int size = owidth * oheight;
693         uint8_t *p = output;
694
695         while( size -- )
696         {
697                 *p ++ = 16;
698                 *p ++ = 128;
699         }
700
701         out_line = output + ( ( oheight - iheight ) / 2 ) * ostride;
702         out_line += 4 * ( int )( ( owidth - iwidth ) / 4 );
703                 
704         // Loop for the entirety of our output height.
705         while ( iheight -- )
706         {
707                 // We're in the input range for this row.
708                 memcpy( out_line, in_line, istride );
709
710                 // Move to next input line
711                 in_line += istride;
712
713                 // Move to next output line
714                 out_line += ostride;
715         }
716 }
717
718 /** A resizing function for yuv422 frames - this does not rescale, but simply
719         resizes. It assumes yuv422 images available on the frame so use with care.
720 */
721
722 uint8_t *mlt_frame_resize_yuv422( mlt_frame this, int owidth, int oheight )
723 {
724         // Get properties
725         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
726
727         // Get the input image, width and height
728         uint8_t *input = mlt_properties_get_data( properties, "image", NULL );
729         uint8_t *alpha = mlt_frame_get_alpha_mask( this );
730
731         int iwidth = mlt_properties_get_int( properties, "width" );
732         int iheight = mlt_properties_get_int( properties, "height" );
733
734         // If width and height are correct, don't do anything
735         if ( iwidth != owidth || iheight != oheight )
736         {
737                 // Create the output image
738                 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
739
740                 // Call the generic resize
741                 mlt_resize_yuv422( output, owidth, oheight, input, iwidth, iheight );
742
743                 // Now update the frame
744                 mlt_properties_set_data( properties, "image", output, owidth * ( oheight + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
745                 mlt_properties_set_int( properties, "width", owidth );
746                 mlt_properties_set_int( properties, "height", oheight );
747
748                 // We should resize the alpha too
749                 alpha = mlt_resize_alpha( alpha, owidth, oheight, iwidth, iheight );
750                 if ( alpha != NULL )
751                 {
752                         mlt_properties_set_data( properties, "alpha", alpha, owidth * ( oheight + 1 ), ( mlt_destructor )mlt_pool_release, NULL );
753                         this->get_alpha_mask = NULL;
754                 }
755
756                 // Return the output
757                 return output;
758         }
759         // No change, return input
760         return input;
761 }
762
763 /** A rescaling function for yuv422 frames - low quality, and provided for testing
764         only. It assumes yuv422 images available on the frame so use with care.
765 */
766
767 uint8_t *mlt_frame_rescale_yuv422( mlt_frame this, int owidth, int oheight )
768 {
769         // Get properties
770         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
771
772         // Get the input image, width and height
773         uint8_t *input = mlt_properties_get_data( properties, "image", NULL );
774         int iwidth = mlt_properties_get_int( properties, "width" );
775         int iheight = mlt_properties_get_int( properties, "height" );
776
777         // If width and height are correct, don't do anything
778         if ( iwidth != owidth || iheight != oheight )
779         {
780                 // Create the output image
781                 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
782
783                 // Calculate strides
784                 int istride = iwidth * 2;
785                 int ostride = owidth * 2;
786
787                 iwidth = iwidth - ( iwidth % 4 );
788
789                 // Derived coordinates
790                 int dy, dx;
791
792         // Calculate ranges
793         int out_x_range = owidth / 2;
794         int out_y_range = oheight / 2;
795         int in_x_range = iwidth / 2;
796         int in_y_range = iheight / 2;
797
798         // Output pointers
799         register uint8_t *out_line = output;
800         register uint8_t *out_ptr;
801
802         // Calculate a middle pointer
803         uint8_t *in_middle = input + istride * in_y_range + in_x_range * 2;
804         uint8_t *in_line;
805
806                 // Generate the affine transform scaling values
807                 register int scale_width = ( iwidth << 16 ) / owidth;
808                 register int scale_height = ( iheight << 16 ) / oheight;
809                 register int base = 0;
810
811                 int outer = out_x_range * scale_width;
812                 int bottom = out_y_range * scale_height;
813
814         // Loop for the entirety of our output height.
815         for ( dy = - bottom; dy < bottom; dy += scale_height )
816         {
817                 // Start at the beginning of the line
818                 out_ptr = out_line;
819         
820                 // Pointer to the middle of the input line
821                 in_line = in_middle + ( dy >> 16 ) * istride;
822
823                 // Loop for the entirety of our output row.
824                 for ( dx = - outer; dx < outer; dx += scale_width )
825                 {
826                                 base = dx >> 15;
827                                 base &= 0xfffffffe;
828                                 *out_ptr ++ = *( in_line + base );
829                                 base &= 0xfffffffc;
830                                 *out_ptr ++ = *( in_line + base + 1 );
831                                 dx += scale_width;
832                                 base = dx >> 15;
833                                 base &= 0xfffffffe;
834                                 *out_ptr ++ = *( in_line + base );
835                                 base &= 0xfffffffc;
836                                 *out_ptr ++ = *( in_line + base + 3 );
837                 }
838
839                 // Move to next output line
840                 out_line += ostride;
841         }
842
843                 // Now update the frame
844                 mlt_properties_set_data( properties, "image", output, owidth * ( oheight + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
845                 mlt_properties_set_int( properties, "width", owidth );
846                 mlt_properties_set_int( properties, "height", oheight );
847
848                 // Return the output
849                 return output;
850         }
851
852         // No change, return input
853         return input;
854 }
855
856 int mlt_frame_mix_audio( mlt_frame this, mlt_frame that, float weight_start, float weight_end, int16_t **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
857 {
858         int ret = 0;
859         int16_t *src, *dest;
860         int frequency_src = *frequency, frequency_dest = *frequency;
861         int channels_src = *channels, channels_dest = *channels;
862         int samples_src = *samples, samples_dest = *samples;
863         int i, j;
864         double d = 0, s = 0;
865
866         mlt_frame_get_audio( that, &src, format, &frequency_src, &channels_src, &samples_src );
867         mlt_frame_get_audio( this, &dest, format, &frequency_dest, &channels_dest, &samples_dest );
868
869         int silent = mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "silent_audio" );
870         mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "silent_audio", 0 );
871         if ( silent )
872                 memset( dest, 0, samples_dest * channels_dest * sizeof( int16_t ) );
873
874         silent = mlt_properties_get_int( MLT_FRAME_PROPERTIES( that ), "silent_audio" );
875         mlt_properties_set_int( MLT_FRAME_PROPERTIES( that ), "silent_audio", 0 );
876         if ( silent )
877                 memset( src, 0, samples_src * channels_src * sizeof( int16_t ) );
878
879         if ( channels_src > 6 )
880                 channels_src = 0;
881         if ( channels_dest > 6 )
882                 channels_dest = 0;
883         if ( samples_src > 4000 )
884                 samples_src = 0;
885         if ( samples_dest > 4000 )
886                 samples_dest = 0;
887
888         // determine number of samples to process
889         *samples = samples_src < samples_dest ? samples_src : samples_dest;
890         *channels = channels_src < channels_dest ? channels_src : channels_dest;
891         *buffer = dest;
892         *frequency = frequency_dest;
893
894         // Compute a smooth ramp over start to end
895         float weight = weight_start;
896         float weight_step = ( weight_end - weight_start ) / *samples;
897
898         // Mixdown
899         for ( i = 0; i < *samples; i++ )
900         {
901                 for ( j = 0; j < *channels; j++ )
902                 {
903                         if ( j < channels_dest )
904                                 d = (double) dest[ i * channels_dest + j ];
905                         if ( j < channels_src )
906                                 s = (double) src[ i * channels_src + j ];
907                         dest[ i * channels_dest + j ] = s * weight + d * ( 1.0 - weight );
908                 }
909                 weight += weight_step;
910         }
911
912         return ret;
913 }
914
915 /* Will this break when mlt_position is converted to double? -Zach */
916 int mlt_sample_calculator( float fps, int frequency, int64_t position )
917 {
918         int samples = 0;
919
920         if ( ( int )( fps * 100 ) == 2997 )
921         {
922                 samples = frequency / 30;
923
924                 switch ( frequency )
925                 {
926                         case 48000:
927                                 if ( position % 5 != 0 )
928                                         samples += 2;
929                                 break;
930                         case 44100:
931                                 if ( position % 300 == 0 )
932                                         samples = 1471;
933                                 else if ( position % 30 == 0 )
934                                         samples = 1470;
935                                 else if ( position % 2 == 0 )
936                                         samples = 1472;
937                                 else
938                                         samples = 1471;
939                                 break;
940                         case 32000:
941                                 if ( position % 30 == 0 )
942                                         samples = 1068;
943                                 else if ( position % 29 == 0 )
944                                         samples = 1067;
945                                 else if ( position % 4 == 2 )
946                                         samples = 1067;
947                                 else
948                                         samples = 1068;
949                                 break;
950                         default:
951                                 samples = 0;
952                 }
953         }
954         else if ( fps != 0 )
955         {
956                 samples = frequency / fps;
957         }
958
959         return samples;
960 }
961
962 int64_t mlt_sample_calculator_to_now( float fps, int frequency, int64_t frame )
963 {
964         int64_t samples = 0;
965
966         // TODO: Correct rules for NTSC and drop the * 100 hack
967         if ( ( int )( fps * 100 ) == 2997 )
968         {
969                 samples = ( ( double )( frame * frequency ) / 30 );
970                 switch( frequency )
971                 {
972                         case 48000:
973                                 samples += 2 * ( frame / 5 );
974                                 break;
975                         case 44100:
976                                 samples += frame + ( frame / 2 ) - ( frame / 30 ) + ( frame / 300 );
977                                 break;
978                         case 32000:
979                                 samples += ( 2 * frame ) - ( frame / 4 ) - ( frame / 29 );
980                                 break;
981                 }
982         }
983         else if ( fps != 0 )
984         {
985                 samples = ( ( frame * frequency ) / ( int )fps );
986         }
987
988         return samples;
989 }