X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/cb5eb5a01ee68e22c9ac02a6feaf52c46bfc6c0e..790d9cb736d8a5e30b6fafb0e4c00b7495199f03:/filters.py?ds=sidebyside

diff --git a/filters.py b/filters.py
index df3b49f..222d8fe 100644
--- a/filters.py
+++ b/filters.py
@@ -10,8 +10,8 @@ def edge_detection(image):
     print "edge detection"
     for x in range(2, image.size[0] - 2):
         for y in range(2, image.size[1] - 2):
-            pix = (sum([sum(
-                [image_l[a, b] 
+            pix = (sum([sum([
+                image_l[a, b] 
                     for b in range(y - 2, y + 3)]) 
                     for a in range(x - 2, x + 3)])
                 - (25 * image_l[x, y]))
@@ -21,6 +21,26 @@ def edge_detection(image):
                 pix = 0 
             new_image_l[x, y] = pix
     return new_image
+	
+def peaks(image):
+    image_l = image.load()
+    new_image = Image.new('L', image.size)
+    new_image_l = new_image.load()
+    clear()
+    print "peak extraction"
+    for x in range(2, image.size[0] - 2):
+        for y in range(2, image.size[1] - 2):
+            pix = (sum([sum([
+                - image_l[a, b] 
+                    for b in range(y - 2, y + 3)]) 
+                    for a in range(x - 2, x + 3)])
+                + (16 * image_l[x, y]))
+            if pix > 255:
+                pix = 255
+            if pix < 0:
+                pix = 0 
+            new_image_l[x, y] = pix
+    return new_image
 
 def high_pass(image, height):
     image_l = image.load()