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>
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.
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.
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
21 #include <framework/mlt_producer.h>
22 #include <framework/mlt_frame.h>
23 #include <framework/mlt_pool.h>
35 static int producer_get_frame( mlt_producer parent, mlt_frame_ptr frame, int index );
36 static void producer_close( mlt_producer parent );
38 mlt_producer producer_colour_init( mlt_profile profile, mlt_service_type type, const char *id, char *colour )
40 mlt_producer producer = calloc( 1, sizeof( struct mlt_producer_s ) );
41 if ( producer != NULL && mlt_producer_init( producer, NULL ) == 0 )
43 // Get the properties interface
44 mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
46 // Callback registration
47 producer->get_frame = producer_get_frame;
48 producer->close = ( mlt_destructor )producer_close;
50 // Set the default properties
51 mlt_properties_set( properties, "resource", ( !colour || !strcmp( colour, "" ) ) ? "0x000000ff" : colour );
52 mlt_properties_set( properties, "_resource", "" );
53 mlt_properties_set_double( properties, "aspect_ratio", mlt_profile_sar( profile ) );
61 rgba_color parse_color( char *color, unsigned int color_int )
63 rgba_color result = { 0xff, 0xff, 0xff, 0xff };
65 if ( !strcmp( color, "red" ) )
71 else if ( !strcmp( color, "green" ) )
77 else if ( !strcmp( color, "blue" ) )
83 else if ( !strcmp( color, "black" ) )
89 else if ( strcmp( color, "white" ) )
91 result.r = ( color_int >> 24 ) & 0xff;
92 result.g = ( color_int >> 16 ) & 0xff;
93 result.b = ( color_int >> 8 ) & 0xff;
94 result.a = ( color_int ) & 0xff;
100 static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
102 // Obtain properties of frame
103 mlt_properties properties = MLT_FRAME_PROPERTIES( frame );
105 // Obtain the producer for this frame
106 mlt_producer producer = mlt_properties_get_data( properties, "producer_colour", NULL );
108 mlt_service_lock( MLT_PRODUCER_SERVICE( producer ) );
110 // Obtain properties of producer
111 mlt_properties producer_props = MLT_PRODUCER_PROPERTIES( producer );
113 // Get the current and previous colour strings
114 char *now = mlt_properties_get( producer_props, "resource" );
115 char *then = mlt_properties_get( producer_props, "_resource" );
117 // Get the current image and dimensions cached in the producer
119 uint8_t *image = mlt_properties_get_data( producer_props, "image", &size );
120 int current_width = mlt_properties_get_int( producer_props, "_width" );
121 int current_height = mlt_properties_get_int( producer_props, "_height" );
122 mlt_image_format current_format = mlt_properties_get_int( producer_props, "_format" );
125 if ( now && strchr( now, '/' ) )
127 now = strdup( strrchr( now, '/' ) + 1 );
128 mlt_properties_set( producer_props, "resource", now );
130 now = mlt_properties_get( producer_props, "resource" );
132 rgba_color color = parse_color( now, mlt_properties_get_int( producer_props, "resource" ) );
134 // Choose suitable out values if nothing specific requested
135 if ( *format == mlt_image_none )
136 *format = mlt_image_rgb24a;
138 *width = mlt_service_profile( MLT_PRODUCER_SERVICE(producer) )->width;
140 *height = mlt_service_profile( MLT_PRODUCER_SERVICE(producer) )->height;
142 // See if we need to regenerate
143 if ( strcmp( now, then ) || *width != current_width || *height != current_height || *format != current_format )
146 int i = *width * *height + 1;
149 // Allocate the image
150 size = mlt_image_format_size( *format, *width, *height, &bpp );
151 uint8_t *p = image = mlt_pool_alloc( size );
153 // Update the producer
154 mlt_properties_set_data( producer_props, "image", image, size, mlt_pool_release, NULL );
155 mlt_properties_set_int( producer_props, "_width", *width );
156 mlt_properties_set_int( producer_props, "_height", *height );
157 mlt_properties_set_int( producer_props, "_format", *format );
158 mlt_properties_set( producer_props, "_resource", now );
160 mlt_service_unlock( MLT_PRODUCER_SERVICE( producer ) );
164 case mlt_image_yuv422:
166 int uneven = *width % 2;
167 int count = ( *width - uneven ) / 2 + 1;
170 RGB2YUV_601_SCALED( color.r, color.g, color.b, y, u, v );
190 case mlt_image_rgb24:
211 mlt_service_unlock( MLT_PRODUCER_SERVICE( producer ) );
214 // Create the alpha channel
215 int alpha_size = *width * *height;
216 uint8_t *alpha = mlt_pool_alloc( alpha_size );
218 // Initialise the alpha
220 memset( alpha, color.a, alpha_size );
223 *buffer = mlt_pool_alloc( size );
224 memcpy( *buffer, image, size );
226 // Now update properties so we free the copy after
227 mlt_frame_set_image( frame, *buffer, size, mlt_pool_release );
228 mlt_frame_set_alpha( frame, alpha, alpha_size, mlt_pool_release );
229 mlt_properties_set_double( properties, "aspect_ratio", mlt_properties_get_double( producer_props, "aspect_ratio" ) );
230 mlt_properties_set_int( properties, "meta.media.width", *width );
231 mlt_properties_set_int( properties, "meta.media.height", *height );
237 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
240 *frame = mlt_frame_init( MLT_PRODUCER_SERVICE( producer ) );
242 if ( *frame != NULL )
244 // Obtain properties of frame and producer
245 mlt_properties properties = MLT_FRAME_PROPERTIES( *frame );
247 // Obtain properties of producer
248 mlt_properties producer_props = MLT_PRODUCER_PROPERTIES( producer );
250 // Set the producer on the frame properties
251 mlt_properties_set_data( properties, "producer_colour", producer, 0, NULL, NULL );
253 // Update timecode on the frame we're creating
254 mlt_frame_set_position( *frame, mlt_producer_position( producer ) );
256 // Set producer-specific frame properties
257 mlt_properties_set_int( properties, "progressive", 1 );
258 mlt_profile profile = mlt_service_profile( MLT_PRODUCER_SERVICE( producer ) );
259 mlt_properties_set_double( properties, "aspect_ratio", mlt_profile_sar( profile ) );
261 // colour is an alias for resource
262 if ( mlt_properties_get( producer_props, "colour" ) != NULL )
263 mlt_properties_set( producer_props, "resource", mlt_properties_get( producer_props, "colour" ) );
265 // Push the get_image method
266 mlt_frame_push_get_image( *frame, producer_get_image );
269 // Calculate the next timecode
270 mlt_producer_prepare_next( producer );
275 static void producer_close( mlt_producer producer )
277 producer->close = NULL;
278 mlt_producer_close( producer );