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