Leolo's guassian2 patch

[imager.git] / maskimg.c

diff --git a/maskimg.c b/maskimg.c
index 828c3917f59b03f9e5ba1a27b8f8c0728d021b46..725b8b6ba8b5fc653b87d2da207cd8602cc6533c 100644 (file)
--- a/maskimg.c
+++ b/maskimg.c
@@ -11,6 +11,8 @@ maskimg.c - implements masked images/image subsets
 =cut
 */
 
+#define IMAGER_NO_CONTEXT
+
 #include "imager.h"
 #include "imageri.h"
 
@@ -48,6 +50,12 @@ static i_img_dim i_gsampf_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim
                            int const *chans, int chan_count);
 static i_img_dim i_gpal_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y, i_palidx *vals);
 static i_img_dim i_ppal_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y, const i_palidx *vals);
+static i_img_dim
+psamp_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y,
+              const i_sample_t *samples, const int *chans, int chan_count);
+static i_img_dim
+psampf_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y,
+              const i_fsample_t *samples, const int *chans, int chan_count);
 
 /*
 =item IIM_base_masked
@@ -89,6 +97,12 @@ static i_img IIM_base_masked =
   i_setcolors_forward, /* i_f_setcolors */
 
   i_destroy_masked, /* i_f_destroy */
+
+  NULL, /* i_f_gsamp_bits */
+  NULL, /* i_f_psamp_bits */
+
+  psamp_masked, /* i_f_psamp */
+  psampf_masked /* i_f_psampf */
 };
 
 /*
@@ -112,13 +126,15 @@ sample is treated as boolean.
 =cut
 */
 
-i_img *i_img_masked_new(i_img *targ, i_img *mask, i_img_dim x, i_img_dim y, i_img_dim w, i_img_dim h) {
+i_img *
+i_img_masked_new(i_img *targ, i_img *mask, i_img_dim x, i_img_dim y, i_img_dim w, i_img_dim h) {
   i_img *im;
   i_img_mask_ext *ext;
+  dIMCTXim(targ);
 
-  i_clear_error();
+  im_clear_error(aIMCTX);
   if (x >= targ->xsize || y >= targ->ysize) {
-    i_push_error(0, "subset outside of target image");
+    im_push_error(aIMCTX, 0, "subset outside of target image");
     return NULL;
   }
   if (mask) {
@@ -132,8 +148,10 @@ i_img *i_img_masked_new(i_img *targ, i_img *mask, i_img_dim x, i_img_dim y, i_im
   if (y+h > targ->ysize)
     h = targ->ysize - y;
 
-  im = mymalloc(sizeof(i_img));
+  im = im_img_alloc(aIMCTX);
+
   memcpy(im, &IIM_base_masked, sizeof(i_img));
+  i_tags_new(&im->tags);
   im->xsize = w;
   im->ysize = h;
   im->channels = targ->channels;
@@ -147,6 +165,8 @@ i_img *i_img_masked_new(i_img *targ, i_img *mask, i_img_dim x, i_img_dim y, i_im
   ext->samps = mymalloc(sizeof(i_sample_t) * im->xsize);
   im->ext_data = ext;
 
+  im_img_init(aIMCTX, im);
+
   return im;
 }
 
@@ -512,6 +532,141 @@ static i_img_dim i_ppal_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y,
   }
 }
 
+/*
+=item psamp_masked()
+
+i_psamp() implementation for masked images.
+
+=cut
+*/
+
+static i_img_dim
+psamp_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y,
+            const i_sample_t *samples, const int *chans, int chan_count) {
+  i_img_mask_ext *ext = MASKEXT(im);
+
+  if (y >= 0 && y < im->ysize && l < im->xsize && l >= 0) {
+    unsigned old_ch_mask = ext->targ->ch_mask;
+    i_img_dim result = 0;
+    ext->targ->ch_mask = im->ch_mask;
+    if (r > im->xsize)
+      r = im->xsize;
+    if (ext->mask) {
+      i_img_dim w = r - l;
+      i_img_dim i = 0;
+      i_img_dim x = ext->xbase + l;
+      i_img_dim work_y = y + ext->ybase;
+      i_sample_t *mask_samps = ext->samps;
+       
+      i_gsamp(ext->mask, l, r, y, mask_samps, NULL, 1);
+      /* not optimizing this yet */
+      while (i < w) {
+       if (mask_samps[i]) {
+         /* found a set mask value, try to do a run */
+         i_img_dim run_left = x;
+         const i_sample_t *run_samps = samples;
+         ++i;
+         ++x;
+         samples += chan_count;
+         
+         while (i < w && mask_samps[i]) {
+           ++i;
+           ++x;
+           samples += chan_count;
+         }
+         result += i_psamp(ext->targ, run_left, x, work_y, run_samps, chans, chan_count);
+       }
+       else {
+         ++i;
+         ++x;
+         samples += chan_count;
+         result += chan_count; /* pretend we wrote masked off pixels */
+       }
+      }
+    }
+    else {
+      result = i_psamp(ext->targ, l + ext->xbase, r + ext->xbase, 
+                      y + ext->ybase, samples, chans, chan_count);
+      im->type = ext->targ->type;
+    }
+    ext->targ->ch_mask = old_ch_mask;
+    return result;
+  }
+  else {
+    dIMCTXim(im);
+    i_push_error(0, "Image position outside of image");
+    return -1;
+  }
+}
+
+/*
+=item psampf_masked()
+
+i_psampf() implementation for masked images.
+
+=cut
+*/
+
+static i_img_dim
+psampf_masked(i_img *im, i_img_dim l, i_img_dim r, i_img_dim y,
+            const i_fsample_t *samples, const int *chans, int chan_count) {
+  i_img_mask_ext *ext = MASKEXT(im);
+
+  if (y >= 0 && y < im->ysize && l < im->xsize && l >= 0) {
+    i_img_dim result = 0;
+    unsigned old_ch_mask = ext->targ->ch_mask;
+    ext->targ->ch_mask = im->ch_mask;
+    if (r > im->xsize)
+      r = im->xsize;
+    if (ext->mask) {
+      i_img_dim w = r - l;
+      i_img_dim i = 0;
+      i_img_dim x = ext->xbase + l;
+      i_img_dim work_y = y + ext->ybase;
+      i_sample_t *mask_samps = ext->samps;
+       
+      i_gsamp(ext->mask, l, r, y, mask_samps, NULL, 1);
+      /* not optimizing this yet */
+      while (i < w) {
+       if (mask_samps[i]) {
+         /* found a set mask value, try to do a run */
+         i_img_dim run_left = x;
+         const i_fsample_t *run_samps = samples;
+         ++i;
+         ++x;
+         samples += chan_count;
+         
+         while (i < w && mask_samps[i]) {
+           ++i;
+           ++x;
+           samples += chan_count;
+         }
+         result += i_psampf(ext->targ, run_left, x, work_y, run_samps, chans, chan_count);
+       }
+       else {
+         ++i;
+         ++x;
+         samples += chan_count;
+         result += chan_count; /* pretend we wrote masked off pixels */
+       }
+      }
+    }
+    else {
+      result = i_psampf(ext->targ, l + ext->xbase, r + ext->xbase, 
+                       y + ext->ybase, samples,
+                                chans, chan_count);
+      im->type = ext->targ->type;
+    }
+    ext->targ->ch_mask = old_ch_mask;
+    return result;
+  }
+  else {
+    dIMCTXim(im);
+    i_push_error(0, "Image position outside of image");
+    return -1;
+  }
+}
+
 
 /*
 =back