git.sesse.net Git - mlt/blob - src/modules/sdl/producer_sdl_image.c

   1 /*
   2  * producer_sdl_image.c -- Image loader which wraps SDL_image
   3  * Copyright (C) 2005 Visual Media FX
   4  * Author: Charles Yates <charles.yates@gmail.com>
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include <framework/mlt_producer.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 #include <sys/types.h>
  29 #include <sys/stat.h>
  30 #include <unistd.h>
  31 #include <dirent.h>
  32 #include <math.h>
  33
  34 #include <SDL_image.h>
  35
  36 static int producer_get_image( mlt_frame frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
  37 {
  38         mlt_properties properties = MLT_FRAME_PROPERTIES( frame );
  39         SDL_Surface *surface = mlt_properties_get_data( properties, "surface", NULL );
  40         SDL_Surface *converted = NULL;
  41         uint8_t *alpha;
  42
  43         *width = surface->w;
  44         *height = surface->h;
  45         *format = mlt_image_yuv422;
  46         *image = mlt_pool_alloc( *width * *height * 2 );
  47         alpha = mlt_pool_alloc( *width * *height );
  48
  49         if ( surface->format->BitsPerPixel != 32 && surface->format->BitsPerPixel != 24 )
  50         {
  51                 SDL_PixelFormat fmt;
  52                 fmt.BitsPerPixel = 24;
  53                 fmt.BytesPerPixel = 3;
  54                 fmt.Rshift = 16;
  55                 fmt.Gshift = 8;
  56                 fmt.Bshift = 0;
  57                 fmt.Rmask = 0xff << 16;
  58                 fmt.Gmask = 0xff << 8;
  59                 fmt.Bmask = 0xff;
  60                 converted = SDL_ConvertSurface( surface, &fmt, 0 );
  61         }
  62
  63         switch( surface->format->BitsPerPixel )
  64         {
  65                 case 32:
  66                         mlt_convert_rgb24a_to_yuv422( surface->pixels, *width, *height, surface->pitch, *image, alpha );
  67                         break;
  68                 case 24:
  69                         mlt_convert_rgb24_to_yuv422( surface->pixels, *width, *height, surface->pitch, *image );
  70                         memset( alpha, 255, *width * *height );
  71                         break;
  72                 default:
  73                         mlt_convert_rgb24_to_yuv422( converted->pixels, *width, *height, converted->pitch, *image );
  74                         memset( alpha, 255, *width * *height );
  75                         break;
  76         }
  77
  78         if ( converted )
  79                 SDL_FreeSurface( converted );
  80
  81         // Update the frame
  82         mlt_properties_set_data( properties, "image", *image, *width * *height * 2, mlt_pool_release, NULL );
  83         mlt_properties_set_data( properties, "alpha", alpha, *width * *height, mlt_pool_release, NULL );
  84         mlt_properties_set_int( properties, "width", *width );
  85         mlt_properties_set_int( properties, "height", *height );
  86
  87         return 0;
  88 }
  89
  90 static int filter_files( const struct dirent *de )
  91 {
  92         return de->d_name[ 0 ] != '.';
  93 }
  94
  95 static mlt_properties parse_file_names( char *resource )
  96 {
  97         mlt_properties properties = mlt_properties_new( );
  98
  99         if ( strstr( resource, "/.all." ) != NULL )
 100         {
 101                 char *dir_name = strdup( resource );
 102                 char *extension = strrchr( resource, '.' );
 103                 *( strstr( dir_name, "/.all." ) + 1 ) = '\0';
 104                 char fullname[ 1024 ];
 105                 strcpy( fullname, dir_name );
 106                 struct dirent **de = NULL;
 107                 int n = scandir( fullname, &de, filter_files, alphasort );
 108                 int i;
 109                 struct stat info;
 110
 111                 for (i = 0; i < n; i++ )
 112                 {
 113                         snprintf( fullname, 1023, "%s%s", dir_name, de[i]->d_name );
 114                         if ( strstr( fullname, extension ) && lstat( fullname, &info ) == 0 &&
 115                                 ( S_ISREG( info.st_mode ) || info.st_mode | S_IXUSR ) )
 116                         {
 117                                 char temp[ 20 ];
 118                                 sprintf( temp, "%d", i );
 119                                 mlt_properties_set( properties, temp, fullname );
 120                         }
 121                         free( de[ i ] );
 122                 }
 123
 124                 free( de );
 125                 free( dir_name );
 126         }
 127         else
 128         {
 129                 mlt_properties_set( properties, "0", resource );
 130         }
 131
 132         return properties;
 133 }
 134
 135 static SDL_Surface *load_image( mlt_producer producer )
 136 {
 137         mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
 138         char *resource = mlt_properties_get( properties, "resource" );
 139         char *last_resource = mlt_properties_get( properties, "_last_resource" );
 140         int image_idx = 0;
 141         char *this_resource = NULL;
 142         double ttl = mlt_properties_get_int( properties, "ttl" );
 143         mlt_position position = mlt_producer_position( producer );
 144         SDL_Surface *surface = mlt_properties_get_data( properties, "_surface", NULL );
 145         mlt_properties filenames = mlt_properties_get_data( properties, "_filenames", NULL );
 146
 147         if ( filenames == NULL )
 148         {
 149                 filenames = parse_file_names( resource );
 150                 mlt_properties_set_data( properties, "_surface", surface, 0, ( mlt_destructor )SDL_FreeSurface, 0 );
 151         }
 152
 153         if ( mlt_properties_count( filenames ) )
 154         {
 155                 image_idx = ( int )floor( ( double )position / ttl ) % mlt_properties_count( filenames );
 156                 this_resource = mlt_properties_get_value( filenames, image_idx );
 157
 158                 if ( surface == NULL || last_resource == NULL || strcmp( last_resource, this_resource ) )
 159                 {
 160                         surface = IMG_Load( this_resource );
 161                         if ( surface != NULL )
 162                         {
 163                                 surface->refcount ++;
 164                                 mlt_properties_set_data( properties, "_surface", surface, 0, ( mlt_destructor )SDL_FreeSurface, 0 );
 165                                 mlt_properties_set( properties, "_last_resource", this_resource );
 166                                 mlt_properties_set_int( properties, "_real_width", surface->w );
 167                                 mlt_properties_set_int( properties, "_real_height", surface->h );
 168                         }
 169                 }
 170                 else if ( surface != NULL )
 171                 {
 172                         surface->refcount ++;
 173                 }
 174         }
 175
 176         return surface;
 177 }
 178
 179 static int producer_get_frame( mlt_producer producer, mlt_frame_ptr frame, int index )
 180 {
 181         // Generate a frame
 182         *frame = mlt_frame_init( MLT_PRODUCER_SERVICE( producer ) );
 183
 184         if ( *frame != NULL )
 185         {
 186                 // Create the surface for the current image
 187                 SDL_Surface *surface = load_image( producer );
 188
 189                 if ( surface != NULL )
 190                 {
 191                         // Obtain properties of frame and producer
 192                         mlt_properties properties = MLT_FRAME_PROPERTIES( *frame );
 193
 194                         // Obtain properties of producer
 195                         mlt_properties producer_props = MLT_PRODUCER_PROPERTIES( producer );
 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", mlt_properties_get_double( producer_props, "aspect_ratio" ) );
 203                         mlt_properties_set_data( properties, "surface", surface, 0, ( mlt_destructor )SDL_FreeSurface, NULL );
 204                         mlt_properties_set_int( properties, "real_width", surface->w );
 205                         mlt_properties_set_int( properties, "real_height", surface->h );
 206
 207                         // Push the get_image method
 208                         mlt_frame_push_get_image( *frame, producer_get_image );
 209                 }
 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 }
 224
 225 mlt_producer producer_sdl_image_init( mlt_profile profile, mlt_service_type type, const char *id, char *file )
 226 {
 227         mlt_producer producer = calloc( 1, sizeof( struct mlt_producer_s ) );
 228         if ( producer != NULL && mlt_producer_init( producer, NULL ) == 0 )
 229         {
 230                 // Get the properties interface
 231                 mlt_properties properties = MLT_PRODUCER_PROPERTIES( producer );
 232
 233                 // Callback registration
 234                 producer->get_frame = producer_get_frame;
 235                 producer->close = ( mlt_destructor )producer_close;
 236
 237                 // Set the default properties
 238                 mlt_properties_set( properties, "resource", file );
 239                 mlt_properties_set( properties, "_resource", "" );
 240                 mlt_properties_set_double( properties, "aspect_ratio", 1 );
 241                 mlt_properties_set_int( properties, "ttl", 25 );
 242                 mlt_properties_set_int( properties, "progressive", 1 );
 243
 244                 // Validate the resource
 245                 SDL_Surface *surface = NULL;
 246                 if ( file && ( surface = load_image( producer ) ) )
 247                 {
 248                         SDL_FreeSurface( surface );
 249                         mlt_properties_set_data( properties, "_surface", NULL, 0, NULL, NULL );
 250                 }
 251                 else
 252                 {
 253                         producer_close( producer );
 254                         producer = NULL;
 255                 }
 256                 return producer;
 257         }
 258         free( producer );
 259         return NULL;
 260 }
 261
 262