git.sesse.net Git - mlt/blob - src/modules/swfdec/producer_swfdec.c

   1 /*
   2  * producer_swfdec.c -- swfdec producer for Flash files
   3  * Copyright (C) 2010 Dan Dennedy <dan@dennedy.org>
   4  *
   5  * swfdec library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * swfdec library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with swfdec library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19
  20 #include <framework/mlt.h>
  21 #include <swfdec/swfdec.h>
  22
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <stdio.h>
  26 #include <math.h>
  27
  28 typedef struct
  29 {
  30         struct mlt_producer_s parent;
  31     SwfdecPlayer *player;
  32     SwfdecURL *url;
  33     cairo_surface_t *surface;
  34     cairo_t *cairo;
  35     mlt_position last_position;
  36         guint width;
  37         guint height;
  38 } *producer_swfdec;
  39
  40 void swfdec_open( producer_swfdec swfdec, mlt_profile profile )
  41 {
  42         mlt_properties properties = MLT_PRODUCER_PROPERTIES( &swfdec->parent );
  43
  44         // Setup the swfdec player
  45         swfdec->player = swfdec_player_new( NULL );
  46         if ( mlt_properties_get( properties, "variables") )
  47                 swfdec_player_set_variables( swfdec->player, mlt_properties_get( properties, "variables" ) );
  48         swfdec_player_set_url( swfdec->player, swfdec->url );
  49         swfdec_player_set_maximum_runtime( swfdec->player, 10000 );
  50
  51         // Setup size
  52         swfdec_player_get_default_size( swfdec->player, &swfdec->width, &swfdec->height );
  53         if ( swfdec->width == 0 || swfdec->height == 0 )
  54         {
  55                 swfdec_player_set_size( swfdec->player, profile->width, profile->height );
  56                 swfdec->width = profile->width;
  57                 swfdec->height = profile->height;
  58         }
  59
  60         // Setup scaling
  61         double scale = 1.0;
  62         if ( swfdec->width > 2 * swfdec->height )
  63                 scale = 0.5 * profile->width / swfdec->height;
  64         else if ( swfdec->height > 2 * swfdec->width )
  65                 scale = 0.5 * profile->height / swfdec->width;
  66         else
  67                 scale = (double) profile->width / MAX( swfdec->width, swfdec->height );
  68         swfdec->width = ceil( scale * swfdec->width );
  69         swfdec->height = ceil( scale * swfdec->height );
  70
  71         // Compute the centering translation
  72         double x = swfdec->width > profile->width ? (swfdec->width - profile->width) / 2 : 0;
  73         double y = swfdec->height > profile->height ? (swfdec->height - profile->height) / 2 : 0;
  74
  75         // Setup cairo
  76         swfdec->surface = cairo_image_surface_create( CAIRO_FORMAT_ARGB32, MIN(profile->width, swfdec->width), MIN(profile->height, swfdec->height) );
  77         swfdec->cairo = cairo_create( swfdec->surface );
  78         cairo_translate( swfdec->cairo, -x, -y );
  79         cairo_scale( swfdec->cairo, scale, scale );
  80 }
  81
  82 void swfdec_close( producer_swfdec swfdec )
  83 {
  84         if ( swfdec->cairo )
  85                 cairo_destroy( swfdec->cairo );
  86         swfdec->cairo = NULL;
  87         if ( swfdec->player )
  88                 g_object_unref( swfdec->player );
  89         swfdec->player = NULL;
  90         if ( swfdec->surface )
  91                 cairo_surface_destroy( swfdec->surface );
  92         swfdec->surface = NULL;
  93 }
  94
  95 // Cairo uses 32 bit native endian ARGB
  96 static void bgra_to_rgba( uint8_t *src, uint8_t* dst, int width, int height )
  97 {
  98         int n = width * height + 1;
  99
 100         while ( --n )
 101         {
 102                 *dst++ = src[2];
 103                 *dst++ = src[1];
 104                 *dst++ = src[0];
 105                 *dst++ = src[3];
 106                 src += 4;
 107         }
 108 }
 109
 110 static int get_image( mlt_frame frame, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
 111 {
 112         producer_swfdec swfdec = mlt_frame_pop_service( frame );
 113         mlt_service service = MLT_PRODUCER_SERVICE( &swfdec->parent );
 114         mlt_profile profile = mlt_service_profile( service );
 115         mlt_properties properties = MLT_FRAME_PROPERTIES( frame );
 116
 117         mlt_service_lock( service );
 118
 119         if ( !swfdec->player )
 120                 swfdec_open( swfdec, profile );
 121
 122         // Set width and height
 123         *width = swfdec->width;
 124         *height = swfdec->height;
 125         *format = mlt_image_rgb24a;
 126
 127         *buffer = mlt_pool_alloc( *width * ( *height + 1 ) * 4 );
 128         mlt_frame_set_image( frame, *buffer, *width * ( *height + 1 ) * 4, mlt_pool_release );
 129
 130         // Seek
 131         mlt_position pos = mlt_properties_get_position( properties, "swfdec.position" );
 132         if ( pos > swfdec->last_position )
 133         {
 134                 gulong msec = 1000UL * ( pos - swfdec->last_position ) * profile->frame_rate_den / profile->frame_rate_num;
 135                 while ( msec > 0 )
 136                         msec -= swfdec_player_advance( swfdec->player, msec );
 137         }
 138         else if ( pos < swfdec->last_position )
 139         {
 140                 swfdec_close( swfdec );
 141                 swfdec_open( swfdec, mlt_service_profile( service ) );
 142                 gulong msec = 1000UL * pos * profile->frame_rate_den / profile->frame_rate_num;
 143                 while ( msec > 0 )
 144                         msec -= swfdec_player_advance( swfdec->player, msec );
 145         }
 146         swfdec->last_position = pos;
 147
 148         // Render
 149         cairo_save( swfdec->cairo );
 150         //cairo_set_source_rgba( swfdec->cairo, r, g, b, a );
 151         cairo_set_operator( swfdec->cairo, CAIRO_OPERATOR_CLEAR );
 152         cairo_paint( swfdec->cairo );
 153         cairo_restore( swfdec->cairo );
 154         swfdec_player_render( swfdec->player, swfdec->cairo );
 155
 156         // Get image from surface
 157         uint8_t *image = cairo_image_surface_get_data( swfdec->surface );
 158
 159         mlt_service_unlock( service );
 160
 161         // Convert to RGBA
 162         bgra_to_rgba( image, *buffer, swfdec->width, swfdec->height );
 163
 164         return 0;
 165 }
 166
 167 static int get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
 168 {
 169         // Access the private data
 170         producer_swfdec swfdec = producer->child;
 171
 172         // Create an empty frame
 173         *frame = mlt_frame_init( MLT_PRODUCER_SERVICE( producer ) );
 174
 175         // Get the frames properties
 176         mlt_properties properties = MLT_FRAME_PROPERTIES( *frame );
 177
 178         // Update other info on the frame
 179         mlt_properties_set_int( properties, "test_image", 0 );
 180         mlt_properties_set_int( properties, "width", swfdec->width );
 181         mlt_properties_set_int( properties, "height", swfdec->height );
 182         mlt_properties_set_int( properties, "real_width", swfdec->width );
 183         mlt_properties_set_int( properties, "real_height", swfdec->height );
 184         mlt_properties_set_int( properties, "progressive", 1 );
 185         mlt_properties_set_double( properties, "aspect_ratio", 1.0 );
 186         mlt_properties_set_position( properties, "swfdec.position", mlt_producer_frame( producer ) );
 187
 188         // Push the get_image method on to the stack
 189         mlt_frame_push_service( *frame, swfdec );
 190         mlt_frame_push_get_image( *frame, get_image );
 191
 192         // Update timecode on the frame we're creating
 193         mlt_frame_set_position( *frame, mlt_producer_position( producer ) );
 194
 195         // Calculate the next timecode
 196         mlt_producer_prepare_next( producer );
 197
 198         return 0;
 199 }
 200
 201 static void producer_close( mlt_producer parent )
 202 {
 203         // Obtain swfdec
 204         producer_swfdec swfdec = parent->child;
 205
 206         // Close the file
 207         swfdec_close( swfdec );
 208         if ( swfdec->url )
 209                 swfdec_url_free( swfdec->url );
 210
 211         // Close the parent
 212         parent->close = NULL;
 213         mlt_producer_close( parent );
 214
 215         // Free the memory
 216         free( swfdec );
 217 }
 218
 219 mlt_producer producer_swfdec_init( mlt_profile profile, mlt_service_type type, const char *id, char *filename )
 220 {
 221         if ( !filename ) return NULL;
 222         producer_swfdec swfdec = calloc( 1, sizeof( *swfdec ) );
 223         mlt_producer producer = NULL;
 224
 225         if ( swfdec && mlt_producer_init( &swfdec->parent, swfdec ) == 0 )
 226         {
 227                 // Initialize swfdec and try to open the file
 228                 swfdec->url = swfdec_url_new_from_input( filename );
 229                 if ( swfdec->url )
 230                 {
 231                         // Set the return value
 232                         producer = &swfdec->parent;
 233
 234                         // Set the resource property (required for all producers)
 235                         mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
 236                         mlt_properties_set( properties, "resource", filename );
 237
 238                         // Set the callbacks
 239                         producer->close = (mlt_destructor) producer_close;
 240                         producer->get_frame = get_frame;
 241
 242                         // Set the meta media attributes
 243                         swfdec->width = profile->width;
 244                         swfdec->height = profile->height;
 245                         mlt_properties_set_int( properties, "meta.media.nb_streams", 1 );
 246                         mlt_properties_set( properties, "meta.media.0.stream.type", "video" );
 247                         mlt_properties_set( properties, "meta.media.0.codec.name", "swf" );
 248                         mlt_properties_set( properties, "meta.media.0.codec.long_name", "Adobe Flash" );
 249                         mlt_properties_set( properties, "meta.media.0.codec.pix_fmt", "bgra" );
 250                         mlt_properties_set_int( properties, "meta.media.width", profile->width );
 251                         mlt_properties_set_int( properties, "meta.media.height", profile->height );
 252                         mlt_properties_set_double( properties, "meta.media.sample_aspect_num", 1.0 );
 253                         mlt_properties_set_double( properties, "meta.media.sample_aspect_den", 1.0 );
 254                         mlt_properties_set_int( properties, "meta.media.frame_rate_num", profile->frame_rate_num );
 255                         mlt_properties_set_int( properties, "meta.media.frame_rate_den", profile->frame_rate_den );
 256                         mlt_properties_set_int( properties, "meta.media.progressive", 1 );
 257                 }
 258                 else
 259                 {
 260                         g_object_unref( swfdec->player );
 261                         mlt_producer_close( &swfdec->parent );
 262                         free( swfdec );
 263                 }
 264         }
 265         else
 266         {
 267                 free( swfdec );
 268         }
 269
 270         return producer;
 271 }
 272
 273 MLT_REPOSITORY
 274 {
 275         swfdec_init();
 276         MLT_REGISTER( producer_type, "swfdec", producer_swfdec_init );
 277 }