src/framework/mlt_frame.c
[melted] / 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 *width = *width >> 1 << 1;
267
268 if ( get_image != NULL )
269 {
270 mlt_position position = mlt_frame_get_position( this );
271 error = get_image( this, buffer, format, width, height, writable );
272 mlt_frame_set_position( this, position );
273 }
274 else if ( mlt_properties_get_data( properties, "image", NULL ) != NULL )
275 {
276 *format = mlt_image_yuv422;
277 *buffer = mlt_properties_get_data( properties, "image", NULL );
278 *width = mlt_properties_get_int( properties, "width" );
279 *height = mlt_properties_get_int( properties, "height" );
280 }
281 else if ( producer != NULL )
282 {
283 mlt_frame test_frame = NULL;
284 mlt_service_get_frame( MLT_PRODUCER_SERVICE( producer ), &test_frame, 0 );
285 if ( test_frame != NULL )
286 {
287 mlt_properties test_properties = MLT_FRAME_PROPERTIES( test_frame );
288 mlt_properties_set_double( test_properties, "consumer_aspect_ratio", mlt_properties_get_double( properties, "consumer_aspect_ratio" ) );
289 mlt_properties_set( test_properties, "rescale.interp", mlt_properties_get( properties, "rescale.interp" ) );
290 mlt_frame_get_image( test_frame, buffer, format, width, height, writable );
291 mlt_properties_set_data( properties, "test_card_frame", test_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
292 mlt_properties_set_data( properties, "image", *buffer, *width * *height * 2, NULL, NULL );
293 mlt_properties_set_int( properties, "width", *width );
294 mlt_properties_set_int( properties, "height", *height );
295 mlt_properties_set_double( properties, "aspect_ratio", mlt_frame_get_aspect_ratio( test_frame ) );
296 }
297 else
298 {
299 mlt_properties_set_data( properties, "test_card_producer", NULL, 0, NULL, NULL );
300 mlt_frame_get_image( this, buffer, format, width, height, writable );
301 }
302 }
303 else
304 {
305 register uint8_t *p;
306 register uint8_t *q;
307 int size = 0;
308
309 *width = *width == 0 ? 720 : *width;
310 *height = *height == 0 ? 576 : *height;
311 size = *width * *height;
312
313 mlt_properties_set_int( properties, "width", *width );
314 mlt_properties_set_int( properties, "height", *height );
315 mlt_properties_set_int( properties, "aspect_ratio", 0 );
316
317 switch( *format )
318 {
319 case mlt_image_none:
320 size = 0;
321 *buffer = NULL;
322 break;
323 case mlt_image_rgb24:
324 size *= 3;
325 size += *width * 3;
326 *buffer = mlt_pool_alloc( size );
327 if ( *buffer )
328 memset( *buffer, 255, size );
329 break;
330 case mlt_image_rgb24a:
331 size *= 4;
332 size += *width * 4;
333 *buffer = mlt_pool_alloc( size );
334 if ( *buffer )
335 memset( *buffer, 255, size );
336 break;
337 case mlt_image_yuv422:
338 size *= 2;
339 size += *width * 2;
340 *buffer = mlt_pool_alloc( size );
341 p = *buffer;
342 q = p + size;
343 while ( p != NULL && p != q )
344 {
345 *p ++ = 235;
346 *p ++ = 128;
347 }
348 break;
349 case mlt_image_yuv420p:
350 size = size * 3 / 2;
351 *buffer = mlt_pool_alloc( size );
352 if ( *buffer )
353 memset( *buffer, 255, size );
354 break;
355 }
356
357 mlt_properties_set_data( properties, "image", *buffer, size, ( mlt_destructor )mlt_pool_release, NULL );
358 mlt_properties_set_int( properties, "test_image", 1 );
359 }
360
361 mlt_properties_set_int( properties, "scaled_width", *width );
362 mlt_properties_set_int( properties, "scaled_height", *height );
363
364 return error;
365 }
366
367 uint8_t *mlt_frame_get_alpha_mask( mlt_frame this )
368 {
369 uint8_t *alpha = NULL;
370 if ( this != NULL )
371 {
372 if ( this->get_alpha_mask != NULL )
373 alpha = this->get_alpha_mask( this );
374 if ( alpha == NULL )
375 alpha = mlt_properties_get_data( &this->parent, "alpha", NULL );
376 if ( alpha == NULL )
377 {
378 int size = mlt_properties_get_int( &this->parent, "scaled_width" ) * mlt_properties_get_int( &this->parent, "scaled_height" );
379 alpha = mlt_pool_alloc( size );
380 memset( alpha, 255, size );
381 mlt_properties_set_data( &this->parent, "alpha", alpha, size, mlt_pool_release, NULL );
382 }
383 }
384 return alpha;
385 }
386
387 int mlt_frame_get_audio( mlt_frame this, int16_t **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
388 {
389 mlt_get_audio get_audio = mlt_frame_pop_audio( this );
390 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
391 int hide = mlt_properties_get_int( properties, "test_audio" );
392
393 if ( hide == 0 && get_audio != NULL )
394 {
395 mlt_position position = mlt_frame_get_position( this );
396 get_audio( this, buffer, format, frequency, channels, samples );
397 mlt_frame_set_position( this, position );
398 }
399 else if ( mlt_properties_get_data( properties, "audio", NULL ) )
400 {
401 *buffer = mlt_properties_get_data( properties, "audio", NULL );
402 *frequency = mlt_properties_get_int( properties, "audio_frequency" );
403 *channels = mlt_properties_get_int( properties, "audio_channels" );
404 *samples = mlt_properties_get_int( properties, "audio_samples" );
405 }
406 else
407 {
408 int size = 0;
409 *samples = *samples <= 0 ? 1920 : *samples;
410 *channels = *channels <= 0 ? 2 : *channels;
411 *frequency = *frequency <= 0 ? 48000 : *frequency;
412 size = *samples * *channels * sizeof( int16_t );
413 *buffer = mlt_pool_alloc( size );
414 if ( *buffer != NULL )
415 memset( *buffer, 0, size );
416 mlt_properties_set_data( properties, "audio", *buffer, size, ( mlt_destructor )mlt_pool_release, NULL );
417 mlt_properties_set_int( properties, "test_audio", 1 );
418 }
419
420 mlt_properties_set_int( properties, "audio_frequency", *frequency );
421 mlt_properties_set_int( properties, "audio_channels", *channels );
422 mlt_properties_set_int( properties, "audio_samples", *samples );
423
424 if ( mlt_properties_get( properties, "meta.volume" ) )
425 {
426 double value = mlt_properties_get_double( properties, "meta.volume" );
427 if ( value == 0.0 )
428 {
429 memset( *buffer, 0, *samples * *channels * 2 );
430 mlt_properties_set_double( properties, "meta.volume", 1.0 );
431 }
432 else if ( value != 1.0 )
433 {
434 int total = *samples * *channels;
435 int16_t *p = *buffer;
436 while ( total -- )
437 {
438 *p = *p * value;
439 p ++;
440 }
441 mlt_properties_set_double( properties, "meta.volume", 1.0 );
442 }
443 }
444
445 return 0;
446 }
447
448 unsigned char *mlt_frame_get_waveform( mlt_frame this, int w, int h )
449 {
450 int16_t *pcm = NULL;
451 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
452 mlt_audio_format format = mlt_audio_pcm;
453 int frequency = 32000; // lower frequency available?
454 int channels = 2;
455 double fps = mlt_properties_get_double( properties, "fps" );
456 int samples = mlt_sample_calculator( fps, frequency, mlt_frame_get_position( this ) );
457
458 // Get the pcm data
459 mlt_frame_get_audio( this, &pcm, &format, &frequency, &channels, &samples );
460
461 // Make an 8-bit buffer large enough to hold rendering
462 int size = w * h;
463 unsigned char *bitmap = ( unsigned char* )mlt_pool_alloc( size );
464 if ( bitmap != NULL )
465 memset( bitmap, 0, size );
466 mlt_properties_set_data( properties, "waveform", bitmap, size, ( mlt_destructor )mlt_pool_release, NULL );
467
468 // Render vertical lines
469 int16_t *ubound = pcm + samples * channels;
470 int skip = samples / w - 1;
471 int i, j, k;
472
473 // Iterate sample stream and along x coordinate
474 for ( i = 0; i < w && pcm < ubound; i++ )
475 {
476 // pcm data has channels interleaved
477 for ( j = 0; j < channels; j++ )
478 {
479 // Determine sample's magnitude from 2s complement;
480 int pcm_magnitude = *pcm < 0 ? ~(*pcm) + 1 : *pcm;
481 // The height of a line is the ratio of the magnitude multiplied by
482 // half the vertical resolution
483 int height = ( int )( ( double )( pcm_magnitude ) / 32768 * h / 2 );
484 // Determine the starting y coordinate - left channel above center,
485 // right channel below - currently assumes 2 channels
486 int displacement = ( h / 2 ) - ( 1 - j ) * height;
487 // Position buffer pointer using y coordinate, stride, and x coordinate
488 unsigned char *p = &bitmap[ i + displacement * w ];
489
490 // Draw vertical line
491 for ( k = 0; k < height; k++ )
492 p[ w * k ] = 0xFF;
493
494 pcm++;
495 }
496 pcm += skip * channels;
497 }
498
499 return bitmap;
500 }
501
502 mlt_producer mlt_frame_get_original_producer( mlt_frame this )
503 {
504 if ( this != NULL )
505 return mlt_properties_get_data( MLT_FRAME_PROPERTIES( this ), "_producer", NULL );
506 return NULL;
507 }
508
509 void mlt_frame_close( mlt_frame this )
510 {
511 if ( this != NULL && mlt_properties_dec_ref( MLT_FRAME_PROPERTIES( this ) ) <= 0 )
512 {
513 mlt_deque_close( this->stack_image );
514 mlt_deque_close( this->stack_audio );
515 while( mlt_deque_peek_back( this->stack_service ) )
516 mlt_service_close( mlt_deque_pop_back( this->stack_service ) );
517 mlt_deque_close( this->stack_service );
518 mlt_properties_close( &this->parent );
519 free( this );
520 }
521 }
522
523 /***** convenience functions *****/
524
525 int mlt_convert_rgb24a_to_yuv422( uint8_t *rgba, int width, int height, int stride, uint8_t *yuv, uint8_t *alpha )
526 {
527 int ret = 0;
528 register int y0, y1, u0, u1, v0, v1;
529 register int r, g, b;
530 register uint8_t *d = yuv;
531 register int i, j;
532
533 for ( i = 0; i < height; i++ )
534 {
535 register uint8_t *s = rgba + ( stride * i );
536 for ( j = 0; j < ( width / 2 ); j++ )
537 {
538 r = *s++;
539 g = *s++;
540 b = *s++;
541 *alpha++ = *s++;
542 RGB2YUV (r, g, b, y0, u0 , v0);
543 r = *s++;
544 g = *s++;
545 b = *s++;
546 *alpha++ = *s++;
547 RGB2YUV (r, g, b, y1, u1 , v1);
548 *d++ = y0;
549 *d++ = (u0+u1) >> 1;
550 *d++ = y1;
551 *d++ = (v0+v1) >> 1;
552 }
553 if ( width % 2 )
554 {
555 r = *s++;
556 g = *s++;
557 b = *s++;
558 *alpha++ = *s++;
559 RGB2YUV (r, g, b, y0, u0 , v0);
560 *d++ = y0;
561 *d++ = u0;
562 }
563 }
564 return ret;
565 }
566
567 int mlt_convert_rgb24_to_yuv422( uint8_t *rgb, int width, int height, int stride, uint8_t *yuv )
568 {
569 int ret = 0;
570 register int y0, y1, u0, u1, v0, v1;
571 register int r, g, b;
572 register uint8_t *d = yuv;
573 register int i, j;
574
575 for ( i = 0; i < height; i++ )
576 {
577 register uint8_t *s = rgb + ( stride * i );
578 for ( j = 0; j < ( width / 2 ); j++ )
579 {
580 r = *s++;
581 g = *s++;
582 b = *s++;
583 RGB2YUV (r, g, b, y0, u0 , v0);
584 r = *s++;
585 g = *s++;
586 b = *s++;
587 RGB2YUV (r, g, b, y1, u1 , v1);
588 *d++ = y0;
589 *d++ = (u0+u1) >> 1;
590 *d++ = y1;
591 *d++ = (v0+v1) >> 1;
592 }
593 if ( width % 2 )
594 {
595 r = *s++;
596 g = *s++;
597 b = *s++;
598 RGB2YUV (r, g, b, y0, u0 , v0);
599 *d++ = y0;
600 *d++ = u0;
601 }
602 }
603 return ret;
604 }
605
606 int mlt_convert_yuv420p_to_yuv422( uint8_t *yuv420p, int width, int height, int stride, uint8_t *yuv )
607 {
608 int ret = 0;
609 register int i, j;
610
611 int half = width >> 1;
612
613 uint8_t *Y = yuv420p;
614 uint8_t *U = Y + width * height;
615 uint8_t *V = U + width * height / 4;
616
617 register uint8_t *d = yuv;
618
619 for ( i = 0; i < height; i++ )
620 {
621 register uint8_t *u = U + ( i / 2 ) * ( half );
622 register uint8_t *v = V + ( i / 2 ) * ( half );
623
624 for ( j = 0; j < half; j++ )
625 {
626 *d ++ = *Y ++;
627 *d ++ = *u ++;
628 *d ++ = *Y ++;
629 *d ++ = *v ++;
630 }
631 }
632 return ret;
633 }
634
635 uint8_t *mlt_resize_alpha( uint8_t *input, int owidth, int oheight, int iwidth, int iheight )
636 {
637 uint8_t *output = NULL;
638
639 if ( input != NULL && ( iwidth != owidth || iheight != oheight ) && ( owidth > 6 && oheight > 6 ) )
640 {
641 iwidth = iwidth - ( iwidth % 2 );
642 owidth = owidth - ( owidth % 2 );
643
644 output = mlt_pool_alloc( owidth * oheight );
645
646 // Coordinates (0,0 is middle of output)
647 int y;
648
649 // Calculate ranges
650 int out_x_range = owidth / 2;
651 int out_y_range = oheight / 2;
652 int in_x_range = iwidth / 2 < out_x_range ? iwidth / 2 : out_x_range;
653 int in_y_range = iheight / 2 < out_y_range ? iheight / 2 : out_y_range;
654
655 // Output pointers
656 uint8_t *out_line = output;
657 uint8_t *out_ptr = out_line;
658
659 // Calculate a middle and possibly invalid pointer in the input
660 uint8_t *in_middle = input + iwidth * ( iheight / 2 ) + ( iwidth / 2 );
661 int in_line = - in_y_range * iwidth - in_x_range;
662
663 int elements;
664
665 // Fill whole section with black
666 y = out_y_range - ( iheight / 2 );
667 int blank_elements = owidth * y;
668 elements = blank_elements;
669 while ( elements -- )
670 *out_line ++ = 0;
671
672 int active_width = iwidth;
673 int inactive_width = out_x_range - in_x_range;
674 uint8_t *p = NULL;
675 uint8_t *end = NULL;
676
677 // Loop for the entirety of our output height.
678 while ( iheight -- )
679 {
680 // Start at the beginning of the line
681 out_ptr = out_line;
682
683 // Fill the outer part with black
684 elements = inactive_width;
685 while ( elements -- )
686 *out_ptr ++ = 0;
687
688 // We're in the input range for this row.
689 p = in_middle + in_line;
690 end = out_ptr + active_width;
691 while ( out_ptr != end )
692 *out_ptr ++ = *p ++;
693
694 // Fill the outer part with black
695 elements = inactive_width;
696 while ( elements -- )
697 *out_ptr ++ = 0;
698
699 // Move to next input line
700 in_line += iwidth;
701
702 // Move to next output line
703 out_line += owidth;
704 }
705
706 // Fill whole section with black
707 elements = blank_elements;
708 while ( elements -- )
709 *out_line ++ = 0;
710 }
711
712 return output;
713 }
714
715 void mlt_resize_yuv422( uint8_t *output, int owidth, int oheight, uint8_t *input, int iwidth, int iheight )
716 {
717 // Calculate strides
718 int istride = iwidth * 2;
719 int ostride = owidth * 2;
720
721 iwidth = iwidth - ( iwidth % 2 );
722 owidth = owidth - ( owidth % 2 );
723 //iheight = iheight - ( iheight % 2 );
724 //oheight = oheight - ( oheight % 2 );
725
726 // Optimisation point
727 if ( output == NULL || input == NULL || ( owidth <= 6 || oheight <= 6 || iwidth <= 6 || oheight <= 6 ) )
728 {
729 return;
730 }
731 else if ( iwidth == owidth && iheight == oheight )
732 {
733 memcpy( output, input, iheight * istride );
734 return;
735 }
736
737 // Coordinates (0,0 is middle of output)
738 int y;
739
740 // Calculate ranges
741 int out_x_range = owidth / 2;
742 int out_y_range = oheight / 2;
743 int in_x_range = iwidth / 2 < out_x_range ? iwidth / 2 : out_x_range;
744 int in_y_range = iheight / 2 < out_y_range ? iheight / 2 : out_y_range;
745
746 // Output pointers
747 uint8_t *out_line = output;
748 uint8_t *out_ptr = out_line;
749
750 // Calculate a middle and possibly invalid pointer in the input
751 uint8_t *in_middle = input + istride * ( iheight / 2 ) + iwidth;
752 int in_line = - in_y_range * istride - in_x_range * 2;
753
754 int elements;
755
756 // Fill whole section with black
757 y = out_y_range - ( iheight / 2 );
758 int blank_elements = ostride * y / 2;
759 elements = blank_elements;
760 while ( elements -- )
761 {
762 *out_line ++ = 16;
763 *out_line ++ = 128;
764 }
765
766 int active_width = 2 * iwidth;
767 int left_inactive_width = out_x_range - in_x_range;
768 int right_inactive_width = left_inactive_width;
769 uint8_t *p = NULL;
770 uint8_t *end = NULL;
771
772 if ( in_line % 4 )
773 {
774 active_width -= 2;
775 in_middle += 2;
776 right_inactive_width += 2;
777 }
778
779 // Loop for the entirety of our output height.
780 while ( iheight -- )
781 {
782 // Start at the beginning of the line
783 out_ptr = out_line;
784
785 // Fill the outer part with black
786 elements = left_inactive_width;
787 while ( elements -- )
788 {
789 *out_ptr ++ = 16;
790 *out_ptr ++ = 128;
791 }
792
793 // We're in the input range for this row.
794 p = in_middle + in_line;
795 end = out_ptr + active_width;
796 while ( out_ptr != end )
797 {
798 *out_ptr ++ = *p ++;
799 *out_ptr ++ = *p ++;
800 }
801
802 // Fill the outer part with black
803 elements = right_inactive_width;
804 while ( elements -- )
805 {
806 *out_ptr ++ = 16;
807 *out_ptr ++ = 128;
808 }
809
810 // Move to next input line
811 in_line += istride;
812
813 // Move to next output line
814 out_line += ostride;
815 }
816
817 // Fill whole section with black
818 elements = blank_elements;
819 while ( elements -- )
820 {
821 *out_line ++ = 16;
822 *out_line ++ = 128;
823 }
824 }
825
826 /** A resizing function for yuv422 frames - this does not rescale, but simply
827 resizes. It assumes yuv422 images available on the frame so use with care.
828 */
829
830 uint8_t *mlt_frame_resize_yuv422( mlt_frame this, int owidth, int oheight )
831 {
832 // Get properties
833 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
834
835 // Get the input image, width and height
836 uint8_t *input = mlt_properties_get_data( properties, "image", NULL );
837 uint8_t *alpha = mlt_frame_get_alpha_mask( this );
838
839 int iwidth = mlt_properties_get_int( properties, "width" );
840 int iheight = mlt_properties_get_int( properties, "height" );
841
842 // If width and height are correct, don't do anything
843 if ( iwidth != owidth || iheight != oheight )
844 {
845 // Create the output image
846 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
847
848 // Call the generic resize
849 mlt_resize_yuv422( output, owidth, oheight, input, iwidth, iheight );
850
851 // Now update the frame
852 mlt_properties_set_data( properties, "image", output, owidth * ( oheight + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
853 mlt_properties_set_int( properties, "width", owidth );
854 mlt_properties_set_int( properties, "height", oheight );
855
856 // We should resize the alpha too
857 alpha = mlt_resize_alpha( alpha, owidth, oheight, iwidth, iheight );
858 if ( alpha != NULL )
859 {
860 mlt_properties_set_data( properties, "alpha", alpha, owidth * ( oheight + 1 ), ( mlt_destructor )mlt_pool_release, NULL );
861 this->get_alpha_mask = NULL;
862 }
863
864 // Return the output
865 return output;
866 }
867 // No change, return input
868 return input;
869 }
870
871 /** A rescaling function for yuv422 frames - low quality, and provided for testing
872 only. It assumes yuv422 images available on the frame so use with care.
873 */
874
875 uint8_t *mlt_frame_rescale_yuv422( mlt_frame this, int owidth, int oheight )
876 {
877 // Get properties
878 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
879
880 // Get the input image, width and height
881 uint8_t *input = mlt_properties_get_data( properties, "image", NULL );
882 int iwidth = mlt_properties_get_int( properties, "width" );
883 int iheight = mlt_properties_get_int( properties, "height" );
884
885 // If width and height are correct, don't do anything
886 if ( iwidth != owidth || iheight != oheight )
887 {
888 // Create the output image
889 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
890
891 // Calculate strides
892 int istride = iwidth * 2;
893 int ostride = owidth * 2;
894
895 iwidth = iwidth - ( iwidth % 4 );
896
897 // Derived coordinates
898 int dy, dx;
899
900 // Calculate ranges
901 int out_x_range = owidth / 2;
902 int out_y_range = oheight / 2;
903 int in_x_range = iwidth / 2;
904 int in_y_range = iheight / 2;
905
906 // Output pointers
907 register uint8_t *out_line = output;
908 register uint8_t *out_ptr;
909
910 // Calculate a middle pointer
911 uint8_t *in_middle = input + istride * in_y_range + in_x_range * 2;
912 uint8_t *in_line;
913
914 // Generate the affine transform scaling values
915 register int scale_width = ( iwidth << 16 ) / owidth;
916 register int scale_height = ( iheight << 16 ) / oheight;
917 register int base = 0;
918
919 int outer = out_x_range * scale_width;
920 int bottom = out_y_range * scale_height;
921
922 // Loop for the entirety of our output height.
923 for ( dy = - bottom; dy < bottom; dy += scale_height )
924 {
925 // Start at the beginning of the line
926 out_ptr = out_line;
927
928 // Pointer to the middle of the input line
929 in_line = in_middle + ( dy >> 16 ) * istride;
930
931 // Loop for the entirety of our output row.
932 for ( dx = - outer; dx < outer; dx += scale_width )
933 {
934 base = dx >> 15;
935 base &= 0xfffffffe;
936 *out_ptr ++ = *( in_line + base );
937 base &= 0xfffffffc;
938 *out_ptr ++ = *( in_line + base + 1 );
939 dx += scale_width;
940 base = dx >> 15;
941 base &= 0xfffffffe;
942 *out_ptr ++ = *( in_line + base );
943 base &= 0xfffffffc;
944 *out_ptr ++ = *( in_line + base + 3 );
945 }
946
947 // Move to next output line
948 out_line += ostride;
949 }
950
951 // Now update the frame
952 mlt_properties_set_data( properties, "image", output, owidth * ( oheight + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
953 mlt_properties_set_int( properties, "width", owidth );
954 mlt_properties_set_int( properties, "height", oheight );
955
956 // Return the output
957 return output;
958 }
959
960 // No change, return input
961 return input;
962 }
963
964 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 )
965 {
966 int ret = 0;
967 int16_t *src, *dest;
968 int frequency_src = *frequency, frequency_dest = *frequency;
969 int channels_src = *channels, channels_dest = *channels;
970 int samples_src = *samples, samples_dest = *samples;
971 int i, j;
972 double d = 0, s = 0;
973
974 mlt_frame_get_audio( that, &src, format, &frequency_src, &channels_src, &samples_src );
975 mlt_frame_get_audio( this, &dest, format, &frequency_dest, &channels_dest, &samples_dest );
976
977 int silent = mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "silent_audio" );
978 mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "silent_audio", 0 );
979 if ( silent )
980 memset( dest, 0, samples_dest * channels_dest * sizeof( int16_t ) );
981
982 silent = mlt_properties_get_int( MLT_FRAME_PROPERTIES( that ), "silent_audio" );
983 mlt_properties_set_int( MLT_FRAME_PROPERTIES( that ), "silent_audio", 0 );
984 if ( silent )
985 memset( src, 0, samples_src * channels_src * sizeof( int16_t ) );
986
987 if ( channels_src > 6 )
988 channels_src = 0;
989 if ( channels_dest > 6 )
990 channels_dest = 0;
991 if ( samples_src > 4000 )
992 samples_src = 0;
993 if ( samples_dest > 4000 )
994 samples_dest = 0;
995
996 // determine number of samples to process
997 *samples = samples_src < samples_dest ? samples_src : samples_dest;
998 *channels = channels_src < channels_dest ? channels_src : channels_dest;
999 *buffer = dest;
1000 *frequency = frequency_dest;
1001
1002 // Compute a smooth ramp over start to end
1003 float weight = weight_start;
1004 float weight_step = ( weight_end - weight_start ) / *samples;
1005
1006 // Mixdown
1007 for ( i = 0; i < *samples; i++ )
1008 {
1009 for ( j = 0; j < *channels; j++ )
1010 {
1011 if ( j < channels_dest )
1012 d = (double) dest[ i * channels_dest + j ];
1013 if ( j < channels_src )
1014 s = (double) src[ i * channels_src + j ];
1015 dest[ i * channels_dest + j ] = s * weight + d * ( 1.0 - weight );
1016 }
1017 weight += weight_step;
1018 }
1019
1020 return ret;
1021 }
1022
1023 int mlt_sample_calculator( float fps, int frequency, int64_t position )
1024 {
1025 int samples = 0;
1026
1027 if ( ( int )( fps * 100 ) == 2997 )
1028 {
1029 samples = frequency / 30;
1030
1031 switch ( frequency )
1032 {
1033 case 48000:
1034 if ( position % 5 != 0 )
1035 samples += 2;
1036 break;
1037 case 44100:
1038 if ( position % 300 == 0 )
1039 samples = 1471;
1040 else if ( position % 30 == 0 )
1041 samples = 1470;
1042 else if ( position % 2 == 0 )
1043 samples = 1472;
1044 else
1045 samples = 1471;
1046 break;
1047 case 32000:
1048 if ( position % 30 == 0 )
1049 samples = 1068;
1050 else if ( position % 29 == 0 )
1051 samples = 1067;
1052 else if ( position % 4 == 2 )
1053 samples = 1067;
1054 else
1055 samples = 1068;
1056 break;
1057 default:
1058 samples = 0;
1059 }
1060 }
1061 else if ( fps != 0 )
1062 {
1063 samples = frequency / fps;
1064 }
1065
1066 return samples;
1067 }