add missing va_end() in bmp.c's write_packed()
[imager.git] / conv.im
1 #define IMAGER_NO_CONTEXT
2 #include "imager.h"
3 #include "imageri.h"
4
5 /*
6   General convolution for 2d decoupled filters
7   end effects are acounted for by increasing
8   scaling the result with the sum of used coefficients
9
10   coeff: (double array) coefficients for filter
11     len: length of filter.. number of coefficients
12            note that this has to be an odd number
13            (since the filter is even);
14 */
15
16 int
17 i_conv(i_img *im, const double *coeff,int len) {
18   i_img_dim xo, yo; /* output pixel co-ordinate */
19   int c, ch, center;
20   double pc;
21   double res[MAXCHANNELS];
22   i_img *timg;
23   dIMCTXim(im);
24
25   im_log((aIMCTX,1,"i_conv(im %p, coeff %p, len %d)\n",im,coeff,len));
26   im_clear_error(aIMCTX);
27
28   if (len < 1) {
29     im_push_error(aIMCTX, 0, "there must be at least one coefficient");
30     return 0;
31   }
32  
33   center=(len-1)/2;
34
35   pc = 0;
36   for (c = 0; c < len; ++c)
37     pc += coeff[c];
38
39   if (pc == 0) {
40     i_push_error(0, "sum of coefficients is zero");
41     return 0;
42   }
43
44   timg = i_sametype(im, im->xsize, im->ysize);
45
46 #code im->bits <= 8
47   IM_COLOR rcolor;
48   /* don't move the calculation of pc up here, it depends on which pixels
49      are readable */
50   for(yo = 0; yo < im->ysize; yo++) {
51     for(xo = 0; xo < im->xsize; xo++) {
52       for(ch = 0;ch < im->channels; ch++) 
53         res[ch] = 0;
54       for(c = 0;c < len; c++) {
55         i_img_dim xi = xo + c - center;
56         if (xi < 0)
57           xi = 0;
58         else if (xi >= im->xsize)
59           xi = im->xsize - 1;
60
61         if (IM_GPIX(im, xi, yo, &rcolor)!=-1) {
62           for(ch = 0; ch < im->channels; ch++) 
63             res[ch] += (rcolor.channel[ch])  *coeff[c];
64         }
65       }
66       im_assert(pc != 0);
67       for(ch = 0; ch < im->channels; ch++) {
68         double temp = res[ch] / pc;
69         rcolor.channel[ch] = 
70           temp < 0 ? 0 : temp > IM_SAMPLE_MAX ? IM_SAMPLE_MAX : (IM_SAMPLE_T)temp;
71       }
72       IM_PPIX(timg, xo, yo, &rcolor);
73     }
74   }
75
76   for(xo = 0; xo < im->xsize; xo++) {
77     for(yo = 0;yo < im->ysize; yo++) {
78       for(ch =  0; ch < im->channels; ch++)
79         res[ch] = 0;
80       for(c = 0; c < len; c++) {
81         i_img_dim yi = yo + c - center;
82         if (yi < 0)
83           yi = 0;
84         else if (yi >= im->ysize)
85           yi = im->ysize - 1;
86         if (IM_GPIX(timg, xo, yi, &rcolor) != -1) {
87           for(ch = 0;ch < im->channels; ch++) 
88             res[ch] += (rcolor.channel[ch]) * coeff[c];
89         }
90       }
91       im_assert(pc != 0);
92       for(ch = 0;ch < im->channels; ch++) {
93         double temp = res[ch] / pc;
94         rcolor.channel[ch] = 
95           temp < 0 ? 0 : temp > IM_SAMPLE_MAX ? IM_SAMPLE_MAX : (IM_SAMPLE_T)temp;
96       }
97       IM_PPIX(im, xo, yo,&rcolor);
98     }
99   }
100 #/code
101
102   i_img_destroy(timg);
103
104   return 1;
105 }