Changeset 413 for pwntchaTweet

Timestamp:

Jan 4, 2005, 6:06:04 PM (16 years ago)

Author:

Sam Hocevar

Message:

• fixed a bug in the SDL code: alpha mask should be set to 0.
• implemented filter_black_stuff to remove black dots.
• allow threshold to be negative in filter_equalize; in this case, invert colours.
• implemented filter_crop.

Location:

pwntcha/trunk/src

Files:

• common.h (modified) (3 diffs)
• filters.c (modified) (3 diffs)
• image.c (modified) (3 diffs)

pwntcha/trunk/src/common.h

@@ -28 +28 @@
 
 /* image operations */
-struct image *image_load(char *name);
+struct image *image_load(const char *name);
 struct image *image_new(int width, int height);
 void image_free(struct image *img);
+void image_save(struct image *img, const char *name);
 void image_display(struct image *img);
 int getgray(struct image *img, int x, int y, int *g);
@@ -40 +41 @@
 struct image *filter_fill_holes(struct image *img);
 struct image *filter_dup(struct image *img);
+struct image *filter_black_stuff(struct image *img);
 struct image *filter_detect_lines(struct image *img);
 struct image *filter_equalize(struct image *img, int threshold);
@@ -46 +48 @@
 struct image *filter_median(struct image *img);
 struct image *filter_contrast(struct image *img);
+struct image *filter_crop(struct image *img,
+                          int xmin, int ymin, int xmax, int ymax);
 int filter_count(struct image *img);
 

pwntcha/trunk/src/filters.c

@@ -121 +121 @@
 }
 
+struct image *filter_black_stuff(struct image *img)
+{
+    struct image *dst;
+    int x, y;
+    int r, ra, rb, g, b;
+
+    dst = image_new(img->width, img->height);
+
+    /* Remove vertical stuff */
+    for(y = 0; y < img->height; y++)
+        for(x = 0; x < img->width; x++)
+        {
+            getpixel(img, x, y, &r, &g, &b);
+            setpixel(dst, x, y, r, g, b);
+            if(y > 0 && y < img->height - 1)
+            {
+                getpixel(img, x, y - 1, &ra, &g, &b);
+                getpixel(img, x, y + 1, &rb, &g, &b);
+                if(r < ra && (r - ra) * (r - rb) > 5000)
+                    setpixel(dst, x, y, ra, ra, ra);
+            }
+        }
+
+    /* Remove horizontal stuff */
+    for(y = 0; y < img->height; y++)
+        for(x = 0; x < img->width; x++)
+        {
+            getpixel(img, x, y, &r, &g, &b);
+            if(x > 0 && x < img->width - 1)
+            {
+                getpixel(dst, x - 1, y, &ra, &g, &b);
+                getpixel(dst, x + 1, y, &rb, &g, &b);
+                if(r < ra && (r - ra) * (r - rb) > 5000)
+                    setpixel(dst, x, y, ra, ra, ra);
+            }
+        }
+
+    return dst;
+}
+
 struct image *filter_detect_lines(struct image *img)
 {
@@ -166 +206 @@
     int x, y;
     int r, g, b;
-
-    dst = image_new(img->width, img->height);
-
-    for(y = 0; y < img->height; y++)
-        for(x = 0; x < img->width; x++)
-        {
-            getpixel(img, x, y, &r, &g, &b);
-            if(r < threshold) r = 0; else r = 255;
-            setpixel(dst, x, y, r, r, r);
+    int min = 0, max = 255;
+
+    dst = image_new(img->width, img->height);
+
+    if(threshold < 0)
+    {
+        min = 255;
+        max = 0;
+        threshold = -threshold;
+    }
+
+    for(y = 0; y < img->height; y++)
+        for(x = 0; x < img->width; x++)
+        {
+            getpixel(img, x, y, &r, &g, &b);
+            if(r < threshold)
+                setpixel(dst, x, y, min, min, min);
+            else
+                setpixel(dst, x, y, max, max, max);
         }
 
@@ -330 +380 @@
 }
 
+struct image *filter_crop(struct image *img,
+                          int xmin, int ymin, int xmax, int ymax)
+{
+    struct image *dst;
+    int x, y;
+    int r, g, b;
+
+    if(xmin < 0)
+        xmin = 0;
+    if(ymin < 0)
+        ymin = 0;
+    if(xmax >= img->width)
+        xmax = img->width - 1;
+    if(ymax >= img->height)
+        ymax = img->height - 1;
+
+    if(xmin >= xmax || ymin >= ymax)
+        return NULL;
+
+    dst = image_new(xmax - xmin, ymax - ymin);
+
+    for(y = 0; y < dst->height; y++)
+        for(x = 0; x < dst->width; x++)
+        {
+            getpixel(img, xmin + x, ymin + y, &r, &g, &b);
+            setpixel(dst, x, y, r, g, b);
+        }
+
+    return dst;
+}
+
 int filter_count(struct image *img)
 {

pwntcha/trunk/src/image.c

@@ -27 +27 @@
 #endif
 
-struct image *image_load(char *name)
+struct image *image_load(const char *name)
 {
     struct image *img;
@@ -77 +77 @@
     gmask = 0x00ff0000;
     bmask = 0x0000ff00;
-    amask = 0x000000ff;
+    amask = 0x00000000;
 #   else
     rmask = 0x000000ff;
     gmask = 0x0000ff00;
     bmask = 0x00ff0000;
-    amask = 0xff000000;
+    amask = 0x00000000;
 #   endif
     priv = SDL_CreateRGBSurface(SDL_SWSURFACE, width, height, 32,
@@ -135 +135 @@
 
     free(img);
+}
+
+void image_save(struct image *img, const char *name)
+{
+#if defined(HAVE_SDL_IMAGE_H)
+    SDL_SaveBMP(img->priv, name);
+#elif defined(HAVE_IMLIB2_H)
+    imlib_context_set_image(img->priv);
+    imlib_save_image(name);
+#elif defined(HAVE_CV_H)
+    cvSaveImage(name, img->priv);
+#endif
 }