research.m1stereo.tv/gitweb/ - melted/blob - src/modules/vmfx/filter_chroma_hold.c

   1 /*
   2  * filter_chroma.c -- Maps a chroma key to the alpha channel
   3  * Copyright (C) 2005 Visual Media Fx Inc.
   4  * Author: Charles Yates <charles.yates@pandora.be>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU Lesser 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.
  10  *
  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 Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser 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.
  19  */
  20
  21 #include <framework/mlt_filter.h>
  22 #include <stdlib.h>
  23 #include <framework/mlt_factory.h>
  24 #include <framework/mlt_frame.h>
  25 #include <framework/mlt_producer.h>
  26 #include <framework/mlt_geometry.h>
  27
  28 static inline int in_range( uint8_t v, uint8_t c, int var )
  29 {
  30         return ( ( int )v >= c - var ) && ( ( int )v <= c + var );
  31 }
  32
  33 static inline uint8_t alpha_value( uint8_t a, uint8_t *p, uint8_t u, uint8_t v, int var, int odd )
  34 {
  35         if ( odd == 0 )
  36                 return ( in_range( *( p + 1 ), u, var ) && in_range( *( p + 3 ), v, var ) ) ? 0 : a;
  37         else
  38                 return ( in_range( ( *( p + 1 ) + *( p + 5 ) ) / 2, u, var ) && in_range( ( *( p + 3 ) + *( p + 7 ) ) / 2, v, var ) ) ? 0 : a;
  39 }
  40
  41 /** Get the images and map the chroma to the alpha of the frame.
  42 */
  43
  44 static int filter_get_image( mlt_frame frame, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
  45 {
  46         mlt_filter this = mlt_frame_pop_service( frame );
  47         int variance = 200 * mlt_properties_get_double( MLT_FILTER_PROPERTIES( this ), "variance" );
  48         int32_t key_val = mlt_properties_get_int( MLT_FILTER_PROPERTIES( this ), "key" );
  49         uint8_t r = ( key_val >> 24 ) & 0xff;
  50         uint8_t g = ( key_val >> 16 ) & 0xff;
  51         uint8_t b = ( key_val >>  8 ) & 0xff;
  52         uint8_t y, u, v;
  53
  54         RGB2YUV( r, g, b, y, u, v );
  55
  56         if ( mlt_frame_get_image( frame, image, format, width, height, writable ) == 0 )
  57         {
  58                 uint8_t alpha = 0;
  59                 uint8_t *p = *image;
  60                 int size = *width * *height / 2;
  61                 while ( size -- )
  62                 {
  63                         alpha = alpha_value( 255, p, u, v, variance, 0 );
  64                         if ( alpha )
  65                                 *( p + 1 )= 128;
  66                         alpha = alpha_value( 255, p, u, v, variance, 1 );
  67                         if ( alpha )
  68                                 *( p + 3 ) = 128;
  69                         p += 4;
  70                 }
  71         }
  72
  73         return 0;
  74 }
  75
  76 /** Filter processing.
  77 */
  78
  79 static mlt_frame filter_process( mlt_filter this, mlt_frame frame )
  80 {
  81         mlt_frame_push_service( frame, this );
  82         mlt_frame_push_service( frame, filter_get_image );
  83         return frame;
  84 }
  85
  86 /** Constructor for the filter.
  87 */
  88
  89 mlt_filter filter_chroma_hold_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
  90 {
  91         mlt_filter this = mlt_filter_new( );
  92         if ( this != NULL )
  93         {
  94                 mlt_properties_set( MLT_FILTER_PROPERTIES( this ), "key", arg == NULL ? "0xc0000000" : arg );
  95                 mlt_properties_set_double( MLT_FILTER_PROPERTIES( this ), "variance", 0.15 );
  96                 this->process = filter_process;
  97         }
  98         return this;
  99 }
 100