Add image generator and check for interestingness

Src/network-broker/image.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+
+import Image, sys, functools
+
+DISPLAY_X = 0x180
+DISPLAY_Y = 0x110
+
+SCREENSHOT_SCALE = 4
+SCREENSHOT_X = DISPLAY_X / SCREENSHOT_SCALE
+SCREENSHOT_Y = DISPLAY_Y / SCREENSHOT_SCALE
+
+SCREENSHOT_SIZE = (SCREENSHOT_X * SCREENSHOT_Y) / 2
+
+reds = [0x00, 0xff, 0x86, 0x4c, 0x88, 0x35, 0x20, 0xcf, 0x88, 0x40, 0xcb, 0x34, 0x68, 0x8b, 0x68, 0xa1]
+greens = [0x00, 0xff, 0x19, 0xc1, 0x17, 0xac, 0x07, 0xf2, 0x3e, 0x2a, 0x55, 0x34, 0x68, 0xff, 0x4a, 0xa1]
+blues = [0x00, 0xff, 0x01, 0xe3, 0xbd, 0x0a, 0xc0, 0x2d, 0x00, 0x00, 0x37, 0x34, 0x68, 0x59, 0xff, 0xa1]
+
+def create_palette():
+    out = []
+    for i in range(len(reds)):
+        out.append((reds[i], greens[i], blues[i]))
+    return out
+
+palette = create_palette()
+
+def image_from_data(data):
+    out = Image.new("RGB", (SCREENSHOT_X, SCREENSHOT_Y))
+    p = 0
+    blue_pixels = 0.0
+
+    for y in range(0, SCREENSHOT_Y):
+        for x in range(0, SCREENSHOT_X):
+            is_odd = ((x & 1) == 1)
+            mask = 0x0f
+            shift = 0
+            if is_odd:
+                mask = 0xf0
+                shift = 4
+
+            color = (ord(data[p]) & mask) >> shift
+            rgb = palette[color]
+            out.putpixel( (x, y), rgb )
+
+            if color == 6:
+                blue_pixels = blue_pixels + 1
+            if is_odd:
+                p = p + 1
+
+    # Interestingness is defined as the inverse proportion of
+    # blue pixels on the screen.
+    interestingness = 1.0 - (blue_pixels / (SCREENSHOT_X * SCREENSHOT_Y))
+    out.interestingness = interestingness
+
+    return out
+
+
+def save_image(img, filename):
+    img.save(filename)
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print "Usage: image.py datafile"
+        sys.exit(1)
+    name = sys.argv[1]
+
+    f = open(name)
+    data = f.read()
+    f.close()
+    img = image_from_data(data)
+    save_image(img, name + ".png")
+

Src/network-broker/network-broker

@@ -264,8 +264,8 @@ class Peer:
                 # Save as one of the last few screenshots
                 which = random.randrange(0,12)
                 img = image.image_from_data(self.screenshot)
-                print "Saving", which
+                if img.interestingness > 0.4 or random.randrange(0, 10) < 2:
-                img.save("%s%d.png" % (self.srv.image_dir, which))
+                    img.save("%s%d.png" % (self.srv.image_dir, which))
             except Exception, e:
                 log_info("Could not convert image data" + str(e))