git.sesse.net Git - mlt/blob - src/modules/core/producer_colour.c

   1 /*
   2  * producer_colour.c -- raster image loader based upon gdk-pixbuf
   3  * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
   4  * Author: Dan Dennedy <dan@dennedy.org>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <framework/mlt_producer.h>
  22 #include <framework/mlt_frame.h>
  23 #include <framework/mlt_pool.h>
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28
  29
  30 typedef struct
  31 {
  32         uint8_t r, g, b, a;
  33 } rgba_color;
  34
  35 static int producer_get_frame( mlt_producer parent, mlt_frame_ptr frame, int index );
  36 static void producer_close( mlt_producer parent );
  37
  38 mlt_producer producer_colour_init( mlt_profile profile, mlt_service_type type, const char *id, char *colour )
  39 {
  40         mlt_producer producer = calloc( 1, sizeof( struct mlt_producer_s ) );
  41         if ( producer != NULL && mlt_producer_init( producer, NULL ) == 0 )
  42         {
  43                 // Get the properties interface
  44                 mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
  45
  46                 // Callback registration
  47                 producer->get_frame = producer_get_frame;
  48                 producer->close = ( mlt_destructor )producer_close;
  49
  50                 // Set the default properties
  51                 mlt_properties_set( properties, "resource", colour == NULL ? "0x000000ff" : colour );
  52                 mlt_properties_set( properties, "_resource", "" );
  53                 mlt_properties_set_double( properties, "aspect_ratio", mlt_profile_sar( profile ) );
  54
  55                 return producer;
  56         }
  57         free( producer );
  58         return NULL;
  59 }
  60
  61 rgba_color parse_color( char *color, unsigned int color_int )
  62 {
  63         rgba_color result = { 0xff, 0xff, 0xff, 0xff };
  64
  65         if ( !strcmp( color, "red" ) )
  66         {
  67                 result.r = 0xff;
  68                 result.g = 0x00;
  69                 result.b = 0x00;
  70         }
  71         else if ( !strcmp( color, "green" ) )
  72         {
  73                 result.r = 0x00;
  74                 result.g = 0xff;
  75                 result.b = 0x00;
  76         }
  77         else if ( !strcmp( color, "blue" ) )
  78         {
  79                 result.r = 0x00;
  80                 result.g = 0x00;
  81                 result.b = 0xff;
  82         }
  83         else if ( strcmp( color, "white" ) )
  84         {
  85                 result.r = ( color_int >> 24 ) & 0xff;
  86                 result.g = ( color_int >> 16 ) & 0xff;
  87                 result.b = ( color_int >> 8 ) & 0xff;
  88                 result.a = ( color_int ) & 0xff;
  89         }
  90
  91         return result;
  92 }
  93
  94 static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
  95 {
  96         // Obtain properties of frame
  97         mlt_properties properties = MLT_FRAME_PROPERTIES( frame );
  98
  99         // Obtain the producer for this frame
 100         mlt_producer producer = mlt_properties_get_data( properties, "producer_colour", NULL );
 101
 102         mlt_service_lock( MLT_PRODUCER_SERVICE( producer ) );
 103
 104         // Obtain properties of producer
 105         mlt_properties producer_props = MLT_PRODUCER_PROPERTIES( producer );
 106
 107         // Get the current and previous colour strings
 108         char *now = mlt_properties_get( producer_props, "resource" );
 109         char *then = mlt_properties_get( producer_props, "_resource" );
 110
 111         // Get the current image and dimensions cached in the producer
 112         int size = 0;
 113         uint8_t *image = mlt_properties_get_data( producer_props, "image", &size );
 114         int current_width = mlt_properties_get_int( producer_props, "_width" );
 115         int current_height = mlt_properties_get_int( producer_props, "_height" );
 116         mlt_image_format current_format = mlt_properties_get_int( producer_props, "_format" );
 117
 118         // Parse the colour
 119         if ( now && strchr( now, '/' ) )
 120         {
 121                 now = strrchr( now, '/' ) + 1;
 122                 mlt_properties_set( producer_props, "resource", now );
 123         }
 124         rgba_color color = parse_color( now, mlt_properties_get_int( producer_props, "resource" ) );
 125
 126         // See if we need to regenerate
 127         if ( strcmp( now, then ) || *width != current_width || *height != current_height || *format != current_format )
 128         {
 129                 // Color the image
 130                 int i = *width * *height + 1;
 131                 int bpp;
 132
 133                 switch ( *format )
 134                 {
 135                         case mlt_image_rgb24:
 136                                 bpp = 3;
 137                                 break;
 138                         case mlt_image_rgb24a:
 139                         case mlt_image_opengl:
 140                                 bpp = 4;
 141                                 break;
 142                         default:
 143                                 bpp = 2;
 144                                 *format = mlt_image_yuv422;
 145                                 break;
 146                 }
 147
 148                 // Allocate the image
 149                 size = *width * *height * bpp;
 150                 uint8_t *p = image = mlt_pool_alloc( size );
 151
 152                 // Update the producer
 153                 mlt_properties_set_data( producer_props, "image", image, size, mlt_pool_release, NULL );
 154                 mlt_properties_set_int( producer_props, "_width", *width );
 155                 mlt_properties_set_int( producer_props, "_height", *height );
 156                 mlt_properties_set_int( producer_props, "_format", *format );
 157                 mlt_properties_set( producer_props, "_resource", now );
 158
 159                 mlt_service_unlock( MLT_PRODUCER_SERVICE( producer ) );
 160
 161                 switch ( *format )
 162                 {
 163                 case mlt_image_yuv422:
 164                 {
 165                         int uneven = *width % 2;
 166                         int count = ( *width - uneven ) / 2 + 1;
 167                         uint8_t y, u, v;
 168
 169                         RGB2YUV_601_SCALED( color.r, color.g, color.b, y, u, v );
 170                         i = *height + 1;
 171                         while ( --i )
 172                         {
 173                                 int j = count;
 174                                 while ( --j )
 175                                 {
 176                                         *p ++ = y;
 177                                         *p ++ = u;
 178                                         *p ++ = y;
 179                                         *p ++ = v;
 180                                 }
 181                                 if ( uneven )
 182                                 {
 183                                         *p ++ = y;
 184                                         *p ++ = u;
 185                                 }
 186                         }
 187                         break;
 188                 }
 189                 case mlt_image_rgb24:
 190                         while ( --i )
 191                         {
 192                                 *p ++ = color.r;
 193                                 *p ++ = color.g;
 194                                 *p ++ = color.b;
 195                         }
 196                         break;
 197                 case mlt_image_rgb24a:
 198                 case mlt_image_opengl:
 199                         while ( --i )
 200                         {
 201                                 *p ++ = color.r;
 202                                 *p ++ = color.g;
 203                                 *p ++ = color.b;
 204                                 *p ++ = color.a;
 205                         }
 206                         break;
 207                 default:
 208                         break;
 209                 }
 210         }
 211         else
 212         {
 213                 mlt_service_unlock( MLT_PRODUCER_SERVICE( producer ) );
 214         }
 215
 216         // Create the alpha channel
 217         int alpha_size = *width * *height;
 218         uint8_t *alpha = mlt_pool_alloc( alpha_size );
 219
 220         // Initialise the alpha
 221         if ( alpha )
 222                 memset( alpha, color.a, alpha_size );
 223
 224         // Clone our image
 225         *buffer = mlt_pool_alloc( size );
 226         memcpy( *buffer, image, size );
 227
 228         // Now update properties so we free the copy after
 229         mlt_frame_set_image( frame, *buffer, size, mlt_pool_release );
 230         mlt_frame_set_alpha( frame, alpha, alpha_size, mlt_pool_release );
 231         mlt_properties_set_double( properties, "aspect_ratio", mlt_properties_get_double( producer_props, "aspect_ratio" ) );
 232         mlt_properties_set_int( properties, "real_width", *width );
 233         mlt_properties_set_int( properties, "real_height", *height );
 234
 235
 236         return 0;
 237 }
 238
 239 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
 240 {
 241         // Generate a frame
 242         *frame = mlt_frame_init( MLT_PRODUCER_SERVICE( producer ) );
 243
 244         if ( *frame != NULL )
 245         {
 246                 // Obtain properties of frame and producer
 247                 mlt_properties properties = MLT_FRAME_PROPERTIES( *frame );
 248
 249                 // Obtain properties of producer
 250                 mlt_properties producer_props = MLT_PRODUCER_PROPERTIES( producer );
 251
 252                 // Set the producer on the frame properties
 253                 mlt_properties_set_data( properties, "producer_colour", producer, 0, NULL, NULL );
 254
 255                 // Update timecode on the frame we're creating
 256                 mlt_frame_set_position( *frame, mlt_producer_position( producer ) );
 257
 258                 // Set producer-specific frame properties
 259                 mlt_properties_set_int( properties, "progressive", 1 );
 260                 mlt_properties_set_double( properties, "aspect_ratio", mlt_properties_get_double( producer_props, "aspect_ratio" ) );
 261
 262                 // colour is an alias for resource
 263                 if ( mlt_properties_get( producer_props, "colour" ) != NULL )
 264                         mlt_properties_set( producer_props, "resource", mlt_properties_get( producer_props, "colour" ) );
 265
 266                 // Push the get_image method
 267                 mlt_frame_push_get_image( *frame, producer_get_image );
 268         }
 269
 270         // Calculate the next timecode
 271         mlt_producer_prepare_next( producer );
 272
 273         return 0;
 274 }
 275
 276 static void producer_close( mlt_producer producer )
 277 {
 278         producer->close = NULL;
 279         mlt_producer_close( producer );
 280         free( producer );
 281 }