2 * producer_ppm.c -- simple ppm test case
3 * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
4 * Author: Charles Yates <charles.yates@pandora.be>
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>
27 typedef struct producer_ppm_s *producer_ppm;
31 struct mlt_producer_s parent;
38 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index );
39 static void producer_close( mlt_producer parent );
41 mlt_producer producer_ppm_init( mlt_profile profile, mlt_service_type type, const char *id, char *command )
43 producer_ppm this = calloc( sizeof( struct producer_ppm_s ), 1 );
44 if ( this != NULL && mlt_producer_init( &this->parent, this ) == 0 )
46 mlt_producer producer = &this->parent;
47 mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
49 producer->get_frame = producer_get_frame;
50 producer->close = ( mlt_destructor )producer_close;
52 if ( command != NULL )
54 mlt_properties_set( properties, "resource", command );
55 this->command = strdup( command );
59 mlt_properties_set( properties, "resource", "ppm test" );
68 static int producer_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
70 // Get the frames properties
71 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
73 if ( mlt_properties_get_int( properties, "has_image" ) )
76 uint8_t *rgb = mlt_properties_get_data( properties, "image", NULL );
78 // Get width and height
79 *width = mlt_properties_get_int( properties, "width" );
80 *height = mlt_properties_get_int( properties, "height" );
82 // Convert to requested format
83 if ( *format == mlt_image_yuv422 )
85 uint8_t *image = mlt_pool_alloc( *width * ( *height + 1 ) * 2 );
86 mlt_convert_rgb24_to_yuv422( rgb, *width, *height, *width * 3, image );
87 mlt_properties_set_data( properties, "image", image, *width * ( *height + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
90 else if ( *format == mlt_image_rgb24 )
97 mlt_frame_get_image( this, buffer, format, width, height, writable );
103 FILE *producer_ppm_run_video( producer_ppm this )
105 if ( this->video == NULL )
107 if ( this->command == NULL )
109 this->video = popen( "image2raw -k -r 25 -ppm /usr/share/pixmaps/*.png", "r" );
113 char command[ 1024 ];
114 float fps = mlt_producer_get_fps( &this->parent );
115 float position = mlt_producer_position( &this->parent );
116 sprintf( command, "ffmpeg -i \"%s\" -ss %f -f imagepipe -r %f -img ppm - 2>/dev/null", this->command, position, fps );
117 this->video = popen( command, "r" );
123 FILE *producer_ppm_run_audio( producer_ppm this )
125 if ( this->audio == NULL )
127 if ( this->command != NULL )
129 char command[ 1024 ];
130 float position = mlt_producer_position( &this->parent );
131 sprintf( command, "ffmpeg -i \"%s\" -ss %f -f s16le -ar 48000 -ac 2 - 2>/dev/null", this->command, position );
132 this->audio = popen( command, "r" );
138 static void producer_ppm_position( producer_ppm this, uint64_t requested )
140 if ( requested != this->expected )
142 if ( this->video != NULL )
143 pclose( this->video );
145 if ( this->audio != NULL )
146 pclose( this->audio );
150 // This is the next frame we expect
151 this->expected = mlt_producer_frame( &this->parent ) + 1;
154 this->video = producer_ppm_run_video( this );
156 // Open the audio pipe
157 this->audio = producer_ppm_run_audio( this );
161 static int producer_get_audio( mlt_frame this, int16_t **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
163 // Get the frames properties
164 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
166 FILE *pipe = mlt_properties_get_data( properties, "audio.pipe", NULL );
173 int size = *samples * *channels * 2;
176 *buffer = malloc( size );
180 fread( *buffer, size, 1, pipe );
182 memset( *buffer, 0, size );
184 // Pass the data on the frame properties
185 mlt_properties_set_data( properties, "audio", *buffer, size, free, NULL );
190 static int read_ppm_header( FILE *video, int *width, int *height )
195 fgets( temp, 132, video );
196 fgets( temp, 132, video );
197 count += sscanf( temp, "%d %d", width, height );
198 fgets( temp, 132, video );
203 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
205 producer_ppm this = producer->child;
209 // Construct a test frame
210 *frame = mlt_frame_init( MLT_PRODUCER_SERVICE( producer ) );
212 // Are we at the position expected?
213 producer_ppm_position( this, mlt_producer_frame( producer ) );
215 // Get the frames properties
216 mlt_properties properties = MLT_FRAME_PROPERTIES( *frame );
218 FILE *video = this->video;
219 FILE *audio = this->audio;
222 if ( video != NULL && read_ppm_header( video, &width, &height ) == 2 )
225 uint8_t *image = mlt_pool_alloc( width * ( height + 1 ) * 3 );
228 fread( image, width * height * 3, 1, video );
230 // Pass the data on the frame properties
231 mlt_properties_set_data( properties, "image", image, width * ( height + 1 ) * 3, ( mlt_destructor )mlt_pool_release, NULL );
232 mlt_properties_set_int( properties, "width", width );
233 mlt_properties_set_int( properties, "height", height );
234 mlt_properties_set_int( properties, "has_image", 1 );
235 mlt_properties_set_int( properties, "progressive", 1 );
236 mlt_properties_set_double( properties, "aspect_ratio", 1 );
238 // Push the image callback
239 mlt_frame_push_get_image( *frame, producer_get_image );
243 // Push the image callback
244 mlt_frame_push_get_image( *frame, producer_get_image );
247 // Set the audio pipe
248 mlt_properties_set_data( properties, "audio.pipe", audio, 0, NULL, NULL );
250 // Hmm - register audio callback
251 mlt_frame_push_audio( *frame, producer_get_audio );
253 // Update timecode on the frame we're creating
254 mlt_frame_set_position( *frame, mlt_producer_position( producer ) );
256 // Calculate the next timecode
257 mlt_producer_prepare_next( producer );
262 static void producer_close( mlt_producer parent )
264 producer_ppm this = parent->child;
266 pclose( this->video );
268 pclose( this->audio );
269 free( this->command );
270 parent->close = NULL;
271 mlt_producer_close( parent );