add script for displaying needles

tools/viewneedles.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+import pygame
+import sys
+import os.path
+import glob
+import json
+from pygame.locals import *
+
+def load_areas(img, jsons):
+    if img in jsons.keys():
+        f = open(jsons[img], "r")
+        img_json = f.read()
+        f.close()
+        parsed = json.loads(img_json)
+        return parsed["area"]
+    else:
+        return []
+
+if len(sys.argv) != 2:
+    print "%s directory" % sys.argv[0]
+    sys.exit()
+
+RES = (1024, 768)
+
+pygame.init()
+fpsClock = pygame.time.Clock()
+
+windowSurfaceObj = pygame.display.set_mode(RES)
+pygame.display.set_caption("OpenQA needles viewer")
+
+img_files = glob.glob(os.path.join(sys.argv[1], "*.png"))
+img_files.sort()
+json_files = glob.glob(os.path.join(sys.argv[1], "*.json"))
+
+jsons = {}
+for img in img_files:
+    json_name = os.path.splitext(img)[0] + ".json"
+    if json_name in json_files:
+        jsons[img] = json_name
+
+index = 0
+imageSurfaceObj = pygame.image.load(img_files[index])
+pygame.display.set_caption(os.path.basename(img_files[index]))
+areas = load_areas(img_files[index], jsons)
+
+while True:
+    for event in pygame.event.get():
+        if event.type == QUIT:
+            pygame.quit()
+            sys.exit()
+        elif event.type == KEYDOWN:
+            if event.key == K_LEFT:
+                index = index - 1 if index > 0 else len(img_files) - 1
+                img_index = img_files[index]
+                imageSurfaceObj = pygame.image.load(img_index)
+                pygame.display.set_caption(os.path.basename(img_index))
+                areas = load_areas(img_index, jsons)
+            elif event.key == K_RIGHT:
+                index = index + 1 if index + 1 < len(img_files) else 0
+                img_index = img_files[index]
+                imageSurfaceObj = pygame.image.load(img_index)
+                pygame.display.set_caption(os.path.basename(img_index))
+                areas = load_areas(img_index, jsons)
+            elif event.key == K_ESCAPE:
+                pygame.event.post(pygame.event.Event(QUIT))
+
+    windowSurfaceObj.blit(imageSurfaceObj, (0, 0))
+    for area in areas:
+        pygame.draw.rect(windowSurfaceObj, pygame.Color(255, 0, 0), (area["xpos"], area["ypos"], area["width"], area["height"]), 3)
+
+    pygame.display.update()
+    fpsClock.tick(30)