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" ) + mlt_producer_get_in( mlt_frame_get_original_producer( 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                         // get parent producer
  59                         mlt_producer producer = mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) );
  60                         mlt_producer_seek( producer, pos );
  61
  62                         // Get the frame
  63                         mlt_service_get_frame( mlt_producer_service(producer), &freeze_frame, 0 );
  64
  65                         mlt_properties freeze_properties = MLT_FRAME_PROPERTIES( freeze_frame );
  66                         mlt_properties_set( freeze_properties, "rescale.interp", mlt_properties_get( props, "rescale.interp" ) );
  67                         mlt_properties_set_double( freeze_properties, "aspect_ratio", mlt_frame_get_aspect_ratio( frame ) );
  68                         mlt_properties_set_int( freeze_properties, "progressive", mlt_properties_get_int( props, "progressive" ) );
  69                         mlt_properties_set_int( freeze_properties, "consumer_deinterlace", mlt_properties_get_int( props, "consumer_deinterlace" ) || mlt_properties_get_int( properties, "deinterlace" ) );
  70                         mlt_properties_set_data( properties, "freeze_frame", freeze_frame, 0, ( mlt_destructor )mlt_frame_close, NULL );
  71                         mlt_properties_set_position( properties, "_frame", pos );
  72                 }
  73                 mlt_service_unlock( MLT_FILTER_SERVICE( filter ) );
  74
  75                 // Get frozen image
  76                 uint8_t *buffer = NULL;
  77                 int error = mlt_frame_get_image( freeze_frame, &buffer, format, width, height, 1 );
  78
  79                 // Copy it to current frame
  80                 int size = mlt_image_format_size( *format, *width, *height, NULL );
  81                 uint8_t *image_copy = mlt_pool_alloc( size );
  82                 memcpy( image_copy, buffer, size );
  83                 *image = image_copy;
  84                 mlt_frame_set_image( frame, *image, size, mlt_pool_release );
  85
  86                 uint8_t *alpha_buffer = mlt_frame_get_alpha_mask( freeze_frame );
  87                 int alphasize = *width * *height;
  88                 uint8_t *alpha_copy = mlt_pool_alloc( alphasize );
  89                 memcpy( alpha_copy, alpha_buffer, alphasize );
  90                 mlt_frame_set_alpha( frame, alpha_copy, alphasize, mlt_pool_release );
  91
  92                 return error;
  93         }
  94
  95         int error = mlt_frame_get_image( frame, image, format, width, height, 1 );
  96         return error;
  97 }
  98
  99 /** Filter processing.
 100 */
 101
 102 static mlt_frame filter_process( mlt_filter filter, mlt_frame frame )
 103 {
 104
 105         // Push the filter on to the stack
 106         mlt_frame_push_service( frame, filter );
 107
 108         // Push the frame filter
 109         mlt_frame_push_get_image( frame, filter_get_image );
 110
 111         return frame;
 112 }
 113
 114 /** Constructor for the filter.
 115 */
 116
 117 mlt_filter filter_freeze_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
 118 {
 119         mlt_filter filter = mlt_filter_new( );
 120         if ( filter != NULL )
 121         {
 122                 filter->process = filter_process;
 123                 // Set the frame which will be chosen for freeze
 124                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "frame", "0" );
 125
 126                 // If freeze_after = 1, only frames after the "frame" value will be frozen
 127                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "freeze_after", "0" );
 128
 129                 // If freeze_before = 1, only frames before the "frame" value will be frozen
 130                 mlt_properties_set( MLT_FILTER_PROPERTIES( filter ), "freeze_before", "0" );
 131         }
 132         return filter;
 133 }
 134
 135
 136