mirror of
https://github.com/rocky-linux/os-autoinst-distri-rocky.git
synced 2024-11-17 10:41:27 +00:00
add script for displaying needles
This commit is contained in:
parent
5598e09a03
commit
711537ef35
72
tools/viewneedles.py
Executable file
72
tools/viewneedles.py
Executable file
@ -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)
|
Loading…
Reference in New Issue
Block a user