73 lines
2.2 KiB
Python
Executable File
73 lines
2.2 KiB
Python
Executable File
#!/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)
|