Add image generator and check for interestingness

This commit is contained in:
simon.kagstrom 2009-11-04 16:50:24 +00:00
parent a16a2c3d18
commit 8964f2ec03
2 changed files with 74 additions and 2 deletions

View File

@ -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")

View File

@ -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
img.save("%s%d.png" % (self.srv.image_dir, which))
if img.interestingness > 0.4 or random.randrange(0, 10) < 2:
img.save("%s%d.png" % (self.srv.image_dir, which))
except Exception, e:
log_info("Could not convert image data" + str(e))