git.sesse.net Git - mlt/blob - src/modules/kdenlive/filter_freeze.c

   1 /*
   2  * filter_freeze.c -- simple frame freezing filter
   3  * Copyright (C) 2007 Jean-Baptiste Mardelle <jb@kdenlive.org>
   4  *
   5  * This 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  * This 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 this 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_filter.h>
  21 #include <framework/mlt_frame.h>
  22 #include <framework/mlt_producer.h>
  23 #include <framework/mlt_service.h>
  24 #include <framework/mlt_factory.h>
  25 #include <framework/mlt_property.h>
  26
  27 #include <stdio.h>
  28 #include <string.h>
  29
  30 static int filter_get_image( mlt_frame frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
  31 {
  32         // Get the image
  33         mlt_filter filter = mlt_frame_pop_service( frame );
  34         mlt_properties properties = MLT_FILTER_PROPERTIES( filter );
  35         mlt_properties props = MLT_FRAME_PROPERTIES( frame );
  36
  37         mlt_frame freeze_frame = NULL;;
  38         int freeze_before = mlt_properties_get_int( properties, "freeze_before" );
  39         int freeze_after = mlt_properties_get_int( properties, "freeze_after" );
  40         mlt_position pos = mlt_properties_get_position( properties, "frame" );
  41         mlt_position currentpos = mlt_filter_get_position( filter, frame );
  42
  43         int do_freeze = 0;
  44         if (freeze_before == 0 && freeze_after == 0) {
  45                 do_freeze = 1;
  46         } else if (freeze_before != 0 && pos > currentpos) {
  47                 do_freeze = 1;
  48         } else if (freeze_after != 0 && pos < currentpos) {
  49                 do_freeze = 1;
  50         }
  51
  52         if (do_freeze == 1) {
  53                 mlt_service_lock( MLT_FILTER_SERVICE( filter ) );
  54                 freeze_frame = mlt_properties_get_data( properties, "freeze_frame", NULL );
  55                 if ( !freeze_frame || mlt_properties_get_position( properties, "_frame" ) != pos )
  56                 {
  57                         // freeze_frame has not been fetched yet or is not useful, so fetch it and cache it.
  58                         mlt_producer producer = mlt_frame_get_original_producer(frame);
  59
  60                         // If requested freeze frame is out of the cutted producer, get parent
  61                         if ( pos < mlt_producer_get_in( producer ) || pos > mlt_producer_get_out( producer ) )
  62                         {
  63                                 producer = mlt_producer_cut_parent( producer );
  64                         }
  65                         mlt_producer_seek( producer, pos );
  66
  67                         // Get the frame
  68                         mlt_service_get_frame( mlt_producer_service(producer), &freeze_frame, 0 );
  69
  70                         mlt_properties freeze_properties = MLT_FRAME_PROPERTIES( freeze_frame );
  71                         mlt_properties_set_double( freeze_properties, "consumer_aspect_ratio", mlt_properties_get_double( props, "consumer_aspect_ratio" ) );
  72                         mlt_properties_set( freeze_properties, "rescale.interp", mlt_properties_get( props, "rescale.interp" ) );
  73                         mlt_properties_set_double( freeze_properties, "aspect_ratio", mlt_frame_get_aspect_ratio( frame ) );
  74                         mlt_properties_set_int( freeze_properties, "progressive", mlt_properties_get_int( props, "progressive" ) );
  75                         mlt_properties_set_int( freeze_properties, "consumer_deinterlace", mlt_properties_get_int( props, "consumer_deinterlace" ) || mlt_properties_get_int( properties, "deinterlace" ) );
  76                         mlt_properties_set_data( properties, "freeze_frame", freeze_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
  77                         mlt_properties_set_position( properties, "_frame", pos );
  78                 }
  79                 mlt_service_unlock( MLT_FILTER_SERVICE( filter ) );
  80
  81                 // Get frozen image
  82                 uint8_t *buffer = NULL;
  83                 int error = mlt_frame_get_image( freeze_frame, &buffer, format, width, height, 1 );
  84
  85                 // Copy it to current frame
  86                 int size = mlt_image_format_size( *format, *width, *height, NULL );
  87                 uint8_t *image_copy = mlt_pool_alloc( size );
  88                 memcpy( image_copy, buffer, size );
  89                 *image = image_copy;
  90                 mlt_frame_set_image( frame, *image, size, mlt_pool_release );
  91
  92                 uint8_t *alpha_buffer = mlt_frame_get_alpha_mask( freeze_frame );
  93                 int alphasize = *width * *height;
  94                 uint8_t *alpha_copy = mlt_pool_alloc( alphasize );
  95                 memcpy( alpha_copy, alpha_buffer, alphasize );
  96                 mlt_frame_set_alpha( frame, alpha_copy, alphasize, mlt_pool_release );
  97
  98                 return error;
  99         }
 100
 101         int error = mlt_frame_get_image( frame, image, format, width, height, 1 );
 102         return error;
 103 }
 104
 105 /** Filter processing.
 106 */
 107
 108 static mlt_frame filter_process( mlt_filter filter, mlt_frame frame )
 109 {
 110
 111         // Push the filter on to the stack
 112         mlt_frame_push_service( frame, filter );
 113
 114         // Push the frame filter
 115         mlt_frame_push_get_image( frame, filter_get_image );
 116
 117         return frame;
 118 }
 119
 120 /** Constructor for the filter.
 121 */
 122
 123 mlt_filter filter_freeze_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
 124 {
 125         mlt_filter filter = mlt_filter_new( );
 126         if ( filter != NULL )
 127         {
 128                 filter->process = filter_process;
 129                 // Set the frame which will be chosen for freeze
 130                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "frame", "0" );
 131
 132                 // If freeze_after = 1, only frames after the "frame" value will be frozen
 133                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "freeze_after", "0" );
 134
 135                 // If freeze_before = 1, only frames before the "frame" value will be frozen
 136                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "freeze_before", "0" );
 137         }
 138         return filter;
 139 }
 140
 141
 142