106 lines
3 KiB
Python
106 lines
3 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import pyglet.shapes
|
|
|
|
from source.gui.sprite import Sprite
|
|
from source.gui.widget.abc import BoxWidget
|
|
from source.type import Distance
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from source.gui.scene.abc import Scene
|
|
|
|
|
|
class GameGrid(BoxWidget):
|
|
def __init__(self, scene: "Scene",
|
|
|
|
rows: int,
|
|
columns: int,
|
|
|
|
texture_background: pyglet.image.AbstractImage,
|
|
|
|
line_width: int = 2,
|
|
|
|
x: Distance = 0,
|
|
y: Distance = 0,
|
|
width: Distance = None,
|
|
height: Distance = None,
|
|
|
|
**kwargs):
|
|
super().__init__(scene, x, y, width, height)
|
|
|
|
self._rows = rows
|
|
self._columns = columns
|
|
|
|
self.background = Sprite(img=texture_background)
|
|
|
|
self.lines: list[pyglet.shapes.Line] = [
|
|
pyglet.shapes.Line(0, 0, 0, 0, width=line_width)
|
|
for _ in range((self._columns - 1) + (self._rows - 1))
|
|
]
|
|
|
|
self.cursor = pyglet.shapes.Rectangle(0, 0, 0, 0, color=(0, 0, 0, 100))
|
|
|
|
self._refresh_size()
|
|
|
|
def get_cell_from_rel(self, rel_x: int, rel_y: int) -> tuple[int, int]:
|
|
"""
|
|
Return the cell of the grid from a point relative position
|
|
"""
|
|
|
|
return int(rel_x / self.cell_width), int(rel_y / self.cell_height)
|
|
|
|
# refresh
|
|
|
|
def _refresh_size(self):
|
|
self.background.x, self.background.y = self.x, self.y
|
|
self.background.width, self.background.height = self.width, self.height
|
|
|
|
for column, line in enumerate(self.lines[:self._columns-1], start=1):
|
|
line.x = self.x + self.cell_width * column
|
|
line.x2 = line.x
|
|
line.y = self.y
|
|
line.y2 = self.y + self.height
|
|
|
|
for row, line in enumerate(self.lines[-self._rows+1:], start=1):
|
|
line.x = self.x
|
|
line.x2 = self.x + self.width
|
|
line.y = self.y + self.cell_height * row
|
|
line.y2 = line.y
|
|
|
|
# property
|
|
|
|
@property
|
|
def cell_width(self) -> float:
|
|
return self.width / self._columns
|
|
|
|
@property
|
|
def cell_height(self) -> float:
|
|
return self.height / self._rows
|
|
|
|
@property
|
|
def cell_size(self) -> tuple[float, float]:
|
|
return self.cell_width, self.cell_height
|
|
|
|
# event
|
|
|
|
def on_hover(self, rel_x: int, rel_y: int):
|
|
cell_x, cell_y = self.get_cell_from_rel(rel_x, rel_y)
|
|
|
|
self.cursor.x = self.x + cell_x * self.width / self._columns
|
|
self.cursor.y = self.y + cell_y * self.height / self._rows
|
|
self.cursor.width, self.cursor.height = self.cell_size
|
|
|
|
def on_hover_leave(self, rel_x: int, rel_y: int):
|
|
self.cursor.width, self.cursor.height = 0, 0
|
|
|
|
def on_release(self, rel_x: int, rel_y: int, button: int, modifiers: int):
|
|
print("click", (rel_x, rel_y), self.get_cell_from_rel(rel_x, rel_y))
|
|
|
|
def on_resize(self, width: int, height: int):
|
|
self._refresh_size()
|
|
|
|
def draw(self):
|
|
self.background.draw()
|
|
self.cursor.draw()
|
|
for line in self.lines: line.draw()
|