You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

84 lines
1.9 KiB
Python

3 years ago
from .mapfile import MapFile
import pygame
colors = {
'?' : (192, 192, 192), # Non-tried
'*' : (0 , 0 , 255), # Non-trimmed
'/' : (255, 255, 0) , # Non-scraped
'-' : (255, 0 , 0) , # Bad
'+' : (0 , 128, 0) , # Recovered
'@' : (0 , 255, 255), # Own position
3 years ago
None: (0 , 0 , 0) , # Not on disk
}
BORDERCOLOR = (64,64,64)
W = 1920 * 5 // 6
H = 1040
CELLSIZE = 8
CELLBORDER = 0
CELLMARGIN = 3
3 years ago
CELLOFFSET = CELLSIZE + 2*CELLBORDER + CELLMARGIN
COLUMNS = W // CELLOFFSET
ROWS = H // CELLOFFSET
CELLS = COLUMNS * ROWS
def get_symb(hist):
## Majority:
#symb, _count = max(hist.items(), key=lambda x: x[1])
#return symb
# The worst:
order = ['@', '-', '/', '*', '?', '+']
for ch in order:
if ch in hist:
return ch
3 years ago
class Visualisation:
def __init__(self, fn, **kwargs):
3 years ago
pygame.init()
self.disp = pygame.display.set_mode((W, H))
self.mapfile = MapFile(fn, **kwargs)
3 years ago
self.mapfile.load()
def draw(self):
self.disp.fill((0,0,0)) # Dark theme :-)
sz = self.mapfile.size
sqsz = sz / CELLS
for cid in range(CELLS):
start = cid * sqsz
end = (cid + 1) * sqsz
hist = self.mapfile.get_hist(start, end)
symb = get_symb(hist)
3 years ago
color = colors[symb]
line = cid // COLUMNS
col = cid % COLUMNS
pos = (line, col)
rect = pygame.Rect(col * CELLOFFSET + CELLBORDER, line * CELLOFFSET + CELLBORDER, CELLSIZE+2*CELLBORDER, CELLSIZE+2*CELLBORDER)
pygame.draw.rect(self.disp, color, rect)
if CELLBORDER != 0: pygame.draw.rect(self.disp, BORDERCOLOR, rect, width=CELLBORDER)
3 years ago
pygame.display.flip()
def reload(self):
self.mapfile.load()
self.draw()
def run(self, *, refresh=2):
pygame.time.set_timer(pygame.USEREVENT, refresh * 1000)
self.draw()
while True:
ev = pygame.event.wait()
if ev.type == pygame.QUIT or ev.type == pygame.KEYDOWN and ev.key == pygame.K_q:
return
if ev.type in [pygame.VIDEOEXPOSE, pygame.VIDEORESIZE, pygame.USEREVENT]:
self.draw()