research.m1stereo.tv/gitweb/ - melted/blob - src/modules/oldfilm/filter_dust.c

   1 /*
   2  * filter_dust.c -- dust filter
   3  * Copyright (c) 2007 Marco Gittler <g.marco@freenet.de>
   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
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <math.h>
  26
  27
  28 static int filter_get_image( mlt_frame this, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable )
  29 {
  30
  31         mlt_filter filter = mlt_frame_pop_service( this );
  32         int error = mlt_frame_get_image( this, image, format, width, height, 1 );
  33
  34         if ( error == 0 && *image && *format == mlt_image_yuv422 )
  35         {
  36
  37                 int h = *height;
  38                 int w = *width;
  39
  40                 int maxdia = mlt_properties_get_int( MLT_FILTER_PROPERTIES( filter ), "maxdiameter" );
  41                 int maxcount = mlt_properties_get_int( MLT_FILTER_PROPERTIES( filter ), "maxcount" );
  42                 int im=rand()%maxcount;
  43
  44                 while (im--){
  45                         int type=im%2;
  46                         int y1=rand()%h;
  47                         int x1=rand()%w;
  48                         int dx=rand()%maxdia;
  49                         int dy=rand()%maxdia;
  50                         int x=0,y=0;//,v=0;
  51                         for (x=-dx;x<dx;x++)
  52                                 for (y=-dy;y<dy;y++){
  53                                         if (x1+x<w && x1+x>0 && y1+y<h && y1+y>0 && x!=0 && y!=0){
  54                                                 //uint8_t *pix=*image+(y+y1)*w*2+(x+x1)*2;
  55                                                 //v=255*(abs(x)+abs(y))/dx;
  56                                                 switch(type){
  57                                                         case 0:
  58                                                                 //v=((51*sqrt(y*y+x*x))/255);
  59                                                                 *(*image+(y+y1)*w*2+(x+x1)*2)=*(*image+(y+y1)*w*2+(x+x1)*2)/((51*sqrt(y*y+x*x))/255);
  60                                                                 /*if (v!=0)
  61                                                                         *pix/=v;
  62                                                                 else
  63                                                                         *pix=0;*/
  64                                                                 break;
  65                                                         case 1:
  66                                                                 //v=((51*sqrt(y*y+x*x))/255);
  67                                                                 //v=255*(x+y)/dx;
  68                                                                 *(*image+(y+y1)*w*2+(x+x1)*2)=*(*image+(y+y1)*w*2+(x+x1)*2)*((51*sqrt(y*y+x*x))/255);
  69                                                                 /*if ((*pix*v)<255)
  70                                                                         *pix*=v;
  71                                                                 else
  72                                                                         *pix=255;*/
  73                                                                 break;
  74                                                 }
  75                                         }
  76                                 }
  77                 }
  78         }
  79
  80         return error;
  81 }
  82
  83 static mlt_frame filter_process( mlt_filter this, mlt_frame frame )
  84 {
  85
  86         mlt_frame_push_service( frame, this );
  87         mlt_frame_push_get_image( frame, filter_get_image );
  88         return frame;
  89 }
  90
  91
  92 mlt_filter filter_dust_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
  93 {
  94         mlt_filter this = mlt_filter_new( );
  95         if ( this != NULL )
  96         {
  97                 this->process = filter_process;
  98                 mlt_properties_set( MLT_FILTER_PROPERTIES( this ), "maxdiameter", "10" );
  99                 mlt_properties_set( MLT_FILTER_PROPERTIES( this ), "maxcount", "10" );
 100         }
 101         return this;
 102 }
 103
 104