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