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 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.
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.
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.
21 #include "producer_ppm.h"
22 #include <framework/mlt_frame.h>
26 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index );
27 static void producer_close( mlt_producer parent );
29 mlt_producer producer_ppm_init( void *command )
31 producer_ppm this = calloc( sizeof( struct producer_ppm_s ), 1 );
32 if ( this != NULL && mlt_producer_init( &this->parent, this ) == 0 )
34 mlt_producer producer = &this->parent;
36 producer->get_frame = producer_get_frame;
37 producer->close = producer_close;
39 if ( command != NULL )
40 this->command = strdup( command );
48 static int producer_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
50 // Get the frames properties
51 mlt_properties properties = mlt_frame_properties( this );
54 uint8_t *rgb = mlt_properties_get_data( properties, "image", NULL );
56 // Get width and height
57 *width = mlt_properties_get_int( properties, "width" );
58 *height = mlt_properties_get_int( properties, "height" );
60 // Convert to requested format
61 if ( *format == mlt_image_yuv422 )
63 uint8_t *image = malloc( *width * *height * 2 );
64 mlt_convert_rgb24_to_yuv422( rgb, *width, *height, *width * 3, image );
65 mlt_properties_set_data( properties, "image", image, *width * *height * 2, free, NULL );
68 else if ( *format == mlt_image_rgb24 )
76 static void producer_ppm_position( producer_ppm this, uint64_t requested )
78 if ( requested != this->expected )
80 if ( this->video != NULL )
81 pclose( this->video );
83 if ( this->audio != NULL )
84 pclose( this->audio );
88 // This is the next frame we expect
89 this->expected = mlt_producer_frame( &this->parent ) + 1;
92 FILE *producer_ppm_run_video( producer_ppm this )
94 if ( this->video == NULL )
96 if ( this->command == NULL )
98 this->video = popen( "image2raw -n -a -r 3 -ppm /usr/share/pixmaps/*.png", "r" );
102 char command[ 1024 ];
103 float fps = mlt_producer_get_fps( &this->parent );
104 float position = mlt_producer_position( &this->parent );
105 sprintf( command, "ffmpeg -i \"%s\" -ss %f -f imagepipe -r %f -img ppm -", this->command, position, fps );
106 this->video = popen( command, "r" );
112 FILE *producer_ppm_run_audio( producer_ppm this )
114 if ( this->audio == NULL )
116 if ( this->command != NULL )
118 char command[ 1024 ];
119 float position = mlt_producer_position( &this->parent );
120 sprintf( command, "ffmpeg -i \"%s\" -ss %f -f s16le -ar 48000 -ac 2 -", this->command, position );
121 this->audio = popen( command, "r" );
128 static int producer_get_audio( mlt_frame this, int16_t **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
130 // Get the frames properties
131 mlt_properties properties = mlt_frame_properties( this );
133 FILE *pipe = mlt_properties_get_data( properties, "audio.pipe", NULL );
140 int size = *samples * *channels * 2;
143 *buffer = malloc( size );
146 fread( *buffer, size, 1, pipe );
148 // Pass the data on the frame properties
149 mlt_properties_set_data( properties, "audio", *buffer, size, free, NULL );
154 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
156 producer_ppm this = producer->child;
160 // Construct a test frame
161 *frame = mlt_frame_init( );
163 // Are we at the position expected?
164 producer_ppm_position( this, mlt_producer_frame( producer ) );
167 FILE *video = producer_ppm_run_video( this );
169 // Open the audio pipe
170 FILE *audio = producer_ppm_run_audio( this );
172 // Get the frames properties
173 mlt_properties properties = mlt_frame_properties( *frame );
176 if ( video != NULL && fscanf( video, "P6\n%d %d\n255\n", &width, &height ) == 2 )
179 uint8_t *image = malloc( width * height * 3 );
182 fread( image, width * height * 3, 1, video );
184 // Pass the data on the frame properties
185 mlt_properties_set_data( properties, "image", image, width * height * 3, free, NULL );
186 mlt_properties_set_int( properties, "width", width );
187 mlt_properties_set_int( properties, "height", height );
189 // Push the image callback
190 mlt_frame_push_get_image( *frame, producer_get_image );
196 // Set the audio pipe
197 mlt_properties_set_data( properties, "audio.pipe", audio, 0, NULL, NULL );
199 // Hmm - register audio callback
200 ( *frame )->get_audio = producer_get_audio;
203 // Update timecode on the frame we're creating
204 mlt_frame_set_timecode( *frame, mlt_producer_position( producer ) );
206 // Calculate the next timecode
207 mlt_producer_prepare_next( producer );
212 static void producer_close( mlt_producer parent )
214 producer_ppm this = parent->child;
216 pclose( this->video );
218 pclose( this->audio );
219 free( this->command );
220 parent->close = NULL;
221 mlt_producer_close( parent );