]> git.sesse.net Git - mlt/blob - src/modules/core/producer_colour.c
970cacb9536caf946a9b99c090a618e6372f1bbb
[mlt] / 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 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 "producer_colour.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( 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 = producer_close;
49
50                 // Set the default properties
51                 mlt_properties_set( properties, "resource", colour == NULL ? "0x000000ff" : colour );
52                 
53                 return producer;
54         }
55         free( producer );
56         return NULL;
57 }
58
59 rgba_color parse_color( char *color )
60 {
61         rgba_color result = { 0xff, 0xff, 0xff, 0xff };
62
63         if ( !strncmp( color, "0x", 2 ) )
64         {
65                 unsigned int temp = 0;
66                 sscanf( color + 2, "%x", &temp );
67                 result.r = ( temp >> 24 ) & 0xff;
68                 result.g = ( temp >> 16 ) & 0xff;
69                 result.b = ( temp >> 8 ) & 0xff;
70                 result.a = ( temp ) & 0xff;
71         }
72         else if ( !strcmp( color, "red" ) )
73         {
74                 result.r = 0xff;
75                 result.g = 0x00;
76                 result.b = 0x00;
77         }
78         else if ( !strcmp( color, "green" ) )
79         {
80                 result.r = 0x00;
81                 result.g = 0xff;
82                 result.b = 0x00;
83         }
84         else if ( !strcmp( color, "blue" ) )
85         {
86                 result.r = 0x00;
87                 result.g = 0x00;
88                 result.b = 0xff;
89         }
90         else if ( strcmp( color, "white" ) )
91         {
92                 unsigned int temp = 0;
93                 sscanf( color, "%d", &temp );
94                 result.r = ( temp >> 24 ) & 0xff;
95                 result.g = ( temp >> 16 ) & 0xff;
96                 result.b = ( temp >> 8 ) & 0xff;
97                 result.a = ( temp ) & 0xff;
98         }
99
100         return result;
101 }
102
103 static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
104 {
105         // May need to know the size of the image to clone it
106         int size = 0;
107         
108         // Obtain properties of frame
109         mlt_properties properties = mlt_frame_properties( frame );
110
111         // Obtain the producer for this frame
112         mlt_producer producer = mlt_properties_get_data( properties, "producer_colour", NULL );
113
114         // Obtain properties of producer
115         mlt_properties producer_props = mlt_producer_properties( producer );
116
117         // Get the current image and dimensions cached in the producer
118         uint8_t *image = mlt_properties_get_data( producer_props, "image", &size );
119         int current_width = mlt_properties_get_int( producer_props, "width" );
120         int current_height = mlt_properties_get_int( producer_props, "height" );
121
122         // See if we need to regenerate
123         if ( *width != current_width || *height != current_height )
124         {
125                 // Allocate the image
126                 size = *width * *height * 2;
127                 image = mlt_pool_alloc( size );
128
129                 // Update the producer
130                 mlt_properties_set_data( producer_props, "image", image, size, mlt_pool_release, NULL );
131                 mlt_properties_set_int( producer_props, "width", *width );
132                 mlt_properties_set_int( producer_props, "height", *height );
133
134                 // Color the image
135                 rgba_color color = parse_color( mlt_properties_get( producer_props, "resource" ) );
136                 uint8_t y, u, v;
137                 int i;
138                 RGB2YUV( color.r, color.g, color.b, y, u, v );
139                 color.r = y;
140                 color.g = u;
141                 color.b = y;
142                 color.a = v;
143                 for ( i = 0; i < size; i += 4 )
144                         memcpy( &image[ i ], &color, 4 );
145         }
146
147         // Update the frame
148         mlt_properties_set_data( properties, "image", image, size, NULL, NULL );
149         mlt_properties_set_int( properties, "width", *width );
150         mlt_properties_set_int( properties, "height", *height );
151         
152         // Clone if necessary
153         if ( writable )
154         {
155                 // Clone our image
156                 uint8_t *copy = mlt_pool_alloc( size );
157                 memcpy( copy, image, size );
158
159                 // We're going to pass the copy on
160                 image = copy;
161
162                 // Now update properties so we free the copy after
163                 mlt_properties_set_data( properties, "image", copy, size, mlt_pool_release, NULL );
164         }
165
166         // Pass on the image
167         *buffer = image;
168         *format = mlt_image_yuv422;
169
170         return 0;
171 }
172
173 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
174 {
175         // Generate a frame
176         *frame = mlt_frame_init( );
177
178         if ( *frame != NULL )
179         {
180                 // Obtain properties of frame and producer
181                 mlt_properties properties = mlt_frame_properties( *frame );
182
183                 // Obtain properties of producer
184                 mlt_properties producer_props = mlt_producer_properties( producer );
185
186                 // Set the producer on the frame properties
187                 mlt_properties_set_data( properties, "producer_colour", producer, 0, NULL, NULL );
188
189                 // Update timecode on the frame we're creating
190                 mlt_frame_set_position( *frame, mlt_producer_position( producer ) );
191
192                 // Set producer-specific frame properties
193                 mlt_properties_set_int( properties, "progressive", 1 );
194
195                 // colour is an alias for resource
196                 if ( mlt_properties_get( producer_props, "colour" ) != NULL )
197                         mlt_properties_set( producer_props, "resource", mlt_properties_get( producer_props, "colour" ) );
198                 
199                 // Push the get_image method
200                 mlt_frame_push_get_image( *frame, producer_get_image );
201         }
202
203         // Calculate the next timecode
204         mlt_producer_prepare_next( producer );
205
206         return 0;
207 }
208
209 static void producer_close( mlt_producer producer )
210 {
211         producer->close = NULL;
212         mlt_producer_close( producer );
213         free( producer );
214 }
215