reorg transition_luma to support producer
authorddennedy <ddennedy@d19143bc-622f-0410-bfdd-b5b2a6649095>
Thu, 4 Mar 2004 06:09:16 +0000 (06:09 +0000)
committerddennedy <ddennedy@d19143bc-622f-0410-bfdd-b5b2a6649095>
Thu, 4 Mar 2004 06:09:16 +0000 (06:09 +0000)
git-svn-id: https://mlt.svn.sourceforge.net/svnroot/mlt/trunk/mlt@191 d19143bc-622f-0410-bfdd-b5b2a6649095

src/modules/core/transition_luma.c

index bb44d29..97b2266 100644 (file)
@@ -19,7 +19,7 @@
  */
 
 #include "transition_luma.h"
-#include <framework/mlt_frame.h>
+#include <framework/mlt.h>
 
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <math.h>
 
-/** Luma class.
-*/
-
-typedef struct 
-{
-       struct mlt_transition_s parent;
-       int width;
-       int height;
-       uint16_t *bitmap;
-}
-transition_luma;
-
-
-// forward declarations
-static void transition_close( mlt_transition parent );
-
 /** Calculate the position for this frame.
 */
 
@@ -209,65 +193,6 @@ static void luma_composite( mlt_frame a_frame, mlt_frame b_frame, int luma_width
        }
 }
 
-/** Get the image.
-*/
-
-static int transition_get_image( mlt_frame a_frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
-{
-       // Get the properties of the a frame
-       mlt_properties a_props = mlt_frame_properties( a_frame );
-
-       // Get the b frame from the stack
-       mlt_frame b_frame = mlt_frame_pop_frame( a_frame );
-
-       // Get the properties of the b frame
-       mlt_properties b_props = mlt_frame_properties( b_frame );
-
-       // Arbitrary composite defaults
-       float mix = mlt_properties_get_double( b_props, "image.mix" );
-       float frame_delta = mlt_properties_get_double( b_props, "luma.delta" );
-       int luma_width = mlt_properties_get_int( b_props, "luma.width" );
-       int luma_height = mlt_properties_get_int( b_props, "luma.height" );
-       uint16_t *luma_bitmap = mlt_properties_get_data( b_props, "luma.bitmap", NULL );
-       float luma_softness = mlt_properties_get_double( b_props, "luma.softness" );
-       int progressive = mlt_properties_get_int( b_props, "progressive" ) ||
-                       mlt_properties_get_int( a_props, "consumer_progressive" ) ||
-                       mlt_properties_get_int( b_props, "luma.progressive" );
-
-       int top_field_first =  mlt_properties_get_int( b_props, "top_field_first" );
-       int reverse = mlt_properties_get_int( b_props, "luma.reverse" );
-
-       // Since we are the consumer of the b_frame, we must pass along this
-       // consumer property from the a_frame
-       mlt_properties_set_double( b_props, "consumer_aspect_ratio", mlt_properties_get_double( a_props, "consumer_aspect_ratio" ) );
-       mlt_properties_set_double( b_props, "consumer_scale", mlt_properties_get_double( a_props, "consumer_scale" ) );
-
-       // Honour the reverse here
-       if ( mix >= 1.0 )
-               mix -= floor( mix );
-
-       mix = reverse ? 1 - mix : mix;
-       frame_delta *= reverse ? -1.0 : 1.0;
-
-       // Ensure we get scaling on the b_frame
-       mlt_properties_set( b_props, "rescale.interp", "nearest" );
-
-       if ( luma_width > 0 && luma_height > 0 && luma_bitmap != NULL )
-               // Composite the frames using a luma map
-               luma_composite( a_frame, b_frame, luma_width, luma_height, luma_bitmap, mix, frame_delta,
-                       luma_softness, progressive ? -1 : top_field_first, width, height );
-       else
-               // Dissolve the frames using the time offset for mix value
-               dissolve_yuv( a_frame, b_frame, mix, *width, *height );
-
-       // Extract the a_frame image info
-       *width = mlt_properties_get_int( a_props, "width" );
-       *height = mlt_properties_get_int( a_props, "height" );
-       *image = mlt_properties_get_data( a_props, "image", NULL );
-
-       return 0;
-}
-
 /** Load the luma map from PGM stream.
 */
 
@@ -277,24 +202,36 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
        while (1)
        {
                char line[128];
+               char comment[128];
                int i = 2;
                int maxval;
                int bpp;
                uint16_t *p;
-               
+
                line[127] = '\0';
 
                // get the magic code
                if ( fgets( line, 127, f ) == NULL )
                        break;
+
+               // skip comments
+               while ( sscanf( line, " #%s", comment ) > 0 )
+                       if ( fgets( line, 127, f ) == NULL )
+                               break;
+
                if ( line[0] != 'P' || line[1] != '5' )
                        break;
 
                // skip white space and see if a new line must be fetched
                for ( i = 2; i < 127 && line[i] != '\0' && isspace( line[i] ); i++ );
-               if ( line[i] == '\0' && fgets( line, 127, f ) == NULL )
+               if ( ( line[i] == '\0' || line[i] == '#' ) && fgets( line, 127, f ) == NULL )
                        break;
 
+               // skip comments
+               while ( sscanf( line, " #%s", comment ) > 0 )
+                       if ( fgets( line, 127, f ) == NULL )
+                               break;
+
                // get the dimensions
                if ( line[0] == 'P' )
                        i = sscanf( line, "P5 %d %d %d", width, height, &maxval );
@@ -306,6 +243,12 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
                {
                        if ( fgets( line, 127, f ) == NULL )
                                break;
+
+                       // skip comments
+                       while ( sscanf( line, " #%s", comment ) > 0 )
+                               if ( fgets( line, 127, f ) == NULL )
+                                       break;
+
                        i = sscanf( line, "%d", height );
                        if ( i == 0 )
                                break;
@@ -318,6 +261,12 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
                {
                        if ( fgets( line, 127, f ) == NULL )
                                break;
+
+                       // skip comments
+                       while ( sscanf( line, " #%s", comment ) > 0 )
+                               if ( fgets( line, 127, f ) == NULL )
+                                       break;
+
                        i = sscanf( line, "%d", &maxval );
                        if ( i == 0 )
                                break;
@@ -325,7 +274,7 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
 
                // determine if this is one or two bytes per pixel
                bpp = maxval > 255 ? 2 : 1;
-               
+
                // allocate temporary storage for the raw data
                data = mlt_pool_alloc( *width * *height * bpp );
                if ( data == NULL )
@@ -334,7 +283,7 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
                // read the raw data
                if ( fread( data, *width * *height * bpp, 1, f ) != 1 )
                        break;
-               
+
                // allocate the luma bitmap
                *map = p = (uint16_t*)mlt_pool_alloc( *width * *height * sizeof( uint16_t ) );
                if ( *map == NULL )
@@ -351,47 +300,202 @@ static void luma_read_pgm( FILE *f, uint16_t **map, int *width, int *height )
 
                break;
        }
-               
+
        if ( data != NULL )
                mlt_pool_release( data );
 }
 
+/** Generate a luma map from an RGB image.
+*/
+
+static void luma_read_yuv422( uint8_t *image, uint16_t **map, int width, int height )
+{
+       int i;
+       
+       // allocate the luma bitmap
+       uint16_t *p = *map = ( uint16_t* )mlt_pool_alloc( width * height * sizeof( uint16_t ) );
+       if ( *map == NULL )
+               return;
+
+       // proces the image data into the luma bitmap
+       for ( i = 0; i < width * height * 2; i += 2 )
+               *p++ = ( image[ i ] - 16 ) * 299; // 299 = 65535 / 219
+}
 
-/** Luma transition processing.
+/** Generate a luma map from a YUV image.
 */
+static void luma_read_rgb24( uint8_t *image, uint16_t **map, int width, int height )
+{
+}
 
-static mlt_frame transition_process( mlt_transition transition, mlt_frame a_frame, mlt_frame b_frame )
+/** Get the image.
+*/
+
+static int transition_get_image( mlt_frame a_frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
 {
-       transition_luma *this = (transition_luma*) transition->child;
+       // Get the transition object
+       mlt_transition transition = mlt_frame_pop_service( a_frame );
 
        // Get the properties of the transition
        mlt_properties properties = mlt_transition_properties( transition );
-       
+
+       // Get the properties of the a frame
+       mlt_properties a_props = mlt_frame_properties( a_frame );
+
+       // Get the b frame from the stack
+       mlt_frame b_frame = mlt_frame_pop_frame( a_frame );
+
        // Get the properties of the b frame
        mlt_properties b_props = mlt_frame_properties( b_frame );
+
+       // The cached luma map information
+       int luma_width = mlt_properties_get_int( properties, "width" );
+       int luma_height = mlt_properties_get_int( properties, "height" );
+       uint16_t *luma_bitmap = mlt_properties_get_data( properties, "bitmap", NULL );
        
        // If the filename property changed, reload the map
-       char *lumafile = mlt_properties_get( properties, "resource" );
-       if ( this->bitmap == NULL && lumafile != NULL )
+       char *resource = mlt_properties_get( properties, "resource" );
+
+       if ( luma_bitmap == NULL && resource != NULL )
        {
-               FILE *pipe = fopen( lumafile, "r" );
-               if ( pipe != NULL )
+               char *extension = extension = strrchr( resource, '.' );
+
+               // See if it is a PGM
+               if ( extension != NULL && strcmp( extension, ".pgm" ) == 0 )
+               {
+                       // Open PGM
+                       FILE *f = fopen( resource, "r" );
+                       if ( f != NULL )
+                       {
+                               // Load from PGM
+                               luma_read_pgm( f, &luma_bitmap, &luma_width, &luma_height );
+                               fclose( f );
+
+                               // Set the transition properties
+                               mlt_properties_set_int( properties, "width", luma_width );
+                               mlt_properties_set_int( properties, "height", luma_height );
+                               mlt_properties_set_data( properties, "bitmap", luma_bitmap, luma_width * luma_height * 2, mlt_pool_release, NULL );
+                       }
+               }
+               else
                {
-                       luma_read_pgm( pipe, &this->bitmap, &this->width, &this->height );
-                       fclose( pipe );
+                       // Get the factory producer service
+                       char *factory = mlt_properties_get( properties, "factory" );
+
+                       // Create the producer
+                       mlt_producer producer = mlt_factory_producer( factory, resource );
+
+                       // If we have one
+                       if ( producer != NULL )
+                       {
+                               // Get the producer properties
+                               mlt_properties producer_properties = mlt_producer_properties( producer );
+
+                               // Ensure that we loop
+                               mlt_properties_set( producer_properties, "eof", "loop" );
+
+                               // Now pass all producer. properties on the transition down
+                               mlt_properties_pass( producer_properties, properties, "producer." );
+
+                               // We will get the alpha frame from the producer
+                               mlt_frame luma_frame = NULL;
+
+                               // Get the luma frame
+                               if ( mlt_service_get_frame( mlt_producer_service( producer ), &luma_frame, 0 ) == 0 )
+                               {
+                                       uint8_t *luma_image;
+                                       mlt_image_format luma_format = mlt_image_yuv422;
+
+                                       // Request a luma image the size of transition image request
+                                       luma_width = *width;
+                                       luma_height = *height;
+
+                                       // Get image from the luma producer
+                                       mlt_frame_get_image( luma_frame, &luma_image, &luma_format, &luma_width, &luma_height, 0 );
+
+                                       // Generate the luma map
+                                       if ( luma_image != NULL && luma_format == mlt_image_yuv422 )
+                                               luma_read_yuv422( luma_image, &luma_bitmap, luma_width, luma_height );
+                                               
+                                       else if ( luma_image != NULL && luma_format == mlt_image_rgb24 )
+                                               luma_read_rgb24( luma_image, &luma_bitmap, luma_width, luma_height );
+                                       
+                                       // Set the transition properties
+                                       mlt_properties_set_int( properties, "width", luma_width );
+                                       mlt_properties_set_int( properties, "height", luma_height );
+                                       mlt_properties_set_data( properties, "bitmap", luma_bitmap, luma_width * luma_height * 2, mlt_pool_release, NULL );
+
+                                       // Cleanup the luma frame
+                                       mlt_frame_close( luma_frame );
+                               }
+
+                               // Cleanup the luma producer
+                               mlt_producer_close( producer );
+                       }
                }
        }
 
-       // Set the b frame properties
-       mlt_properties_set_double( b_props, "image.mix", position_calculate( transition, a_frame ) );
-       mlt_properties_set_double( b_props, "luma.delta", delta_calculate( transition, a_frame ) );
-       mlt_properties_set_int( b_props, "luma.width", this->width );
-       mlt_properties_set_int( b_props, "luma.height", this->height );
-       mlt_properties_set_data( b_props, "luma.bitmap", this->bitmap, 0, NULL, NULL );
-       mlt_properties_set_int( b_props, "luma.reverse", mlt_properties_get_int( properties, "reverse" ) );
-       mlt_properties_set_double( b_props, "luma.softness", mlt_properties_get_double( properties, "softness" ) );
+       // Arbitrary composite defaults
+       float mix = position_calculate( transition, a_frame );
+       float frame_delta = delta_calculate( transition, a_frame );
+       
+       float luma_softness = mlt_properties_get_double( properties, "softness" );
+       int progressive = mlt_properties_get_int( b_props, "progressive" ) ||
+                       mlt_properties_get_int( a_props, "consumer_progressive" ) ||
+                       mlt_properties_get_int( b_props, "luma.progressive" );
+       int top_field_first =  mlt_properties_get_int( b_props, "top_field_first" );
+       int reverse = mlt_properties_get_int( properties, "reverse" );
+
+       // Since we are the consumer of the b_frame, we must pass along this
+       // consumer property from the a_frame
+       mlt_properties_set_double( b_props, "consumer_aspect_ratio", mlt_properties_get_double( a_props, "consumer_aspect_ratio" ) );
+       mlt_properties_set_double( b_props, "consumer_scale", mlt_properties_get_double( a_props, "consumer_scale" ) );
+
+       // Honour the reverse here
+       if ( mix >= 1.0 )
+               mix -= floor( mix );
 
+       mix = reverse ? 1 - mix : mix;
+       frame_delta *= reverse ? -1.0 : 1.0;
+
+       // Ensure we get scaling on the b_frame
+       mlt_properties_set( b_props, "rescale.interp", "nearest" );
+
+       if ( luma_width > 0 && luma_height > 0 && luma_bitmap != NULL )
+               // Composite the frames using a luma map
+               luma_composite( a_frame, b_frame, luma_width, luma_height, luma_bitmap, mix, frame_delta,
+                       luma_softness, progressive ? -1 : top_field_first, width, height );
+       else
+               // Dissolve the frames using the time offset for mix value
+               dissolve_yuv( a_frame, b_frame, mix, *width, *height );
+
+       // Extract the a_frame image info
+       *width = mlt_properties_get_int( a_props, "width" );
+       *height = mlt_properties_get_int( a_props, "height" );
+       *image = mlt_properties_get_data( a_props, "image", NULL );
+
+       return 0;
+}
+
+
+/** Luma transition processing.
+*/
+
+static mlt_frame transition_process( mlt_transition transition, mlt_frame a_frame, mlt_frame b_frame )
+{
+       // Get a unique name to store the frame position
+       char *name = mlt_properties_get( mlt_transition_properties( transition ), "_unique_id" );
+
+       // Assign the current position to the name
+       mlt_properties_set_position( mlt_frame_properties( a_frame ), name, mlt_frame_get_position( a_frame ) );
+
+       // Push the transition on to the frame
+       mlt_frame_push_service( a_frame, transition );
+
+       // Push the transition method
        mlt_frame_push_get_image( a_frame, transition_get_image );
+       
+       // Push the b_frame on to the stack
        mlt_frame_push_frame( a_frame, b_frame );
 
        return a_frame;
@@ -402,26 +506,19 @@ static mlt_frame transition_process( mlt_transition transition, mlt_frame a_fram
 
 mlt_transition transition_luma_init( char *lumafile )
 {
-       transition_luma *this = calloc( sizeof( transition_luma ), 1 );
-       if ( this != NULL )
+       mlt_transition transition = mlt_transition_new( );
+       if ( transition != NULL )
        {
-               mlt_transition transition = &this->parent;
-               mlt_transition_init( transition, this );
+               // Set the methods
                transition->process = transition_process;
-               transition->close = transition_close;
+               
+               // Default factory
+               mlt_properties_set( mlt_transition_properties( transition ), "factory", "fezzik" );
+
+               // Set the main property
                mlt_properties_set( mlt_transition_properties( transition ), "resource", lumafile );
-               return &this->parent;
+               
+               return transition;
        }
        return NULL;
 }
-
-/** Close the transition.
-*/
-
-static void transition_close( mlt_transition parent )
-{
-       transition_luma *this = (transition_luma*) parent->child;
-       mlt_pool_release( this->bitmap );
-       free( this );
-}
-