kopia lustrzana https://github.com/peterhinch/micropython-nano-gui
70 wiersze
2.7 KiB
Python
70 wiersze
2.7 KiB
Python
# grid.py nano-gui widget providing the Grid class: a 2d array of Label instances.
|
|
|
|
# Released under the MIT License (MIT). See LICENSE.
|
|
# Copyright (c) 2023 Peter Hinch
|
|
|
|
from gui.core.nanogui import DObject, Writer
|
|
from gui.core.colors import *
|
|
from gui.widgets.label import Label
|
|
from extras.parse2d import do_args
|
|
|
|
# lwidth may be integer Label width in pixels or a tuple/list of widths
|
|
class Grid(DObject):
|
|
def __init__(self, writer, row, col, lwidth, nrows, ncols, invert=False, fgcolor=None, bgcolor=BLACK, bdcolor=None, align=0):
|
|
self.nrows = nrows
|
|
self.ncols = ncols
|
|
self.ncells = nrows * ncols
|
|
self.cheight = writer.height + 4 # Cell height including borders
|
|
# Build column width list. Column width is Label width + 4.
|
|
if isinstance(lwidth, int):
|
|
self.cwidth = [lwidth + 4] * ncols
|
|
else: # Ensure len(.cwidth) == ncols
|
|
self.cwidth = [w + 4 for w in lwidth][:ncols]
|
|
self.cwidth.extend([lwidth[-1] + 4] * (ncols - len(lwidth)))
|
|
width = sum(self.cwidth) - 4 # Dimensions of widget interior
|
|
height = nrows * self.cheight - 4
|
|
super().__init__(writer, row, col, height, width, fgcolor, bgcolor, bdcolor)
|
|
self.cells = []
|
|
r = row
|
|
c = col
|
|
for _ in range(self.nrows):
|
|
for cw in self.cwidth:
|
|
self.cells.append(Label(writer, r, c, cw - 4, invert, fgcolor, bgcolor, False, align)) # No border
|
|
c += cw
|
|
r += self.cheight
|
|
c = col
|
|
|
|
def __getitem__(self, *args):
|
|
indices = do_args(args, self.nrows, self.ncols)
|
|
for i in indices:
|
|
yield self.cells[i]
|
|
|
|
# allow grid[r, c] = "foo" or kwargs for Label:
|
|
# grid[r, c] = {"text": str(n), "fgcolor" : RED}
|
|
def __setitem__(self, *args):
|
|
x = args[1] # Value
|
|
indices = do_args(args[: -1], self.nrows, self.ncols)
|
|
for i in indices:
|
|
try:
|
|
z = next(x) # May be a generator
|
|
except StopIteration:
|
|
pass # Repeat last value
|
|
except TypeError:
|
|
z = x
|
|
v = self.cells[i].value # method of Label
|
|
_ = v(**z) if isinstance(x, dict) else v(z)
|
|
|
|
def show(self):
|
|
super().show() # Draw border
|
|
if self.has_border: # Draw grid
|
|
dev = self.device
|
|
color = self.bdcolor
|
|
x = self.col - 2 # Border top left corner
|
|
y = self.row - 2
|
|
dy = self.cheight
|
|
for row in range(1, self.nrows):
|
|
dev.hline(x, y + row * dy, self.width + 4, color)
|
|
for cw in self.cwidth[:-1]:
|
|
x += cw
|
|
dev.vline(x, y, self.height + 4, color)
|