85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
"""Base class for geometry demo screens."""
|
|
import mcrfpy
|
|
import sys
|
|
import os
|
|
|
|
# Add scripts path for geometry module
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..', 'src', 'scripts'))
|
|
from geometry import *
|
|
|
|
|
|
class GeometryDemoScreen:
|
|
"""Base class for geometry demo screens."""
|
|
|
|
name = "Base Screen"
|
|
description = "Override this description"
|
|
|
|
def __init__(self, scene_name):
|
|
self.scene_name = scene_name
|
|
mcrfpy.createScene(scene_name)
|
|
self.ui = mcrfpy.sceneUI(scene_name)
|
|
self.timers = [] # Track timer names for cleanup
|
|
|
|
def setup(self):
|
|
"""Override to set up the screen content."""
|
|
pass
|
|
|
|
def cleanup(self):
|
|
"""Clean up timers when leaving screen."""
|
|
for timer_name in self.timers:
|
|
try:
|
|
mcrfpy.delTimer(timer_name)
|
|
except:
|
|
pass
|
|
|
|
def get_screenshot_name(self):
|
|
"""Return the screenshot filename for this screen."""
|
|
return f"{self.scene_name}.png"
|
|
|
|
def add_title(self, text, y=10):
|
|
"""Add a title caption."""
|
|
title = mcrfpy.Caption(text=text, pos=(400, y))
|
|
title.fill_color = mcrfpy.Color(255, 255, 255)
|
|
title.outline = 2
|
|
title.outline_color = mcrfpy.Color(0, 0, 0)
|
|
self.ui.append(title)
|
|
return title
|
|
|
|
def add_description(self, text, y=50):
|
|
"""Add a description caption."""
|
|
desc = mcrfpy.Caption(text=text, pos=(50, y))
|
|
desc.fill_color = mcrfpy.Color(200, 200, 200)
|
|
self.ui.append(desc)
|
|
return desc
|
|
|
|
def add_label(self, text, x, y, color=(200, 200, 200)):
|
|
"""Add a label caption."""
|
|
label = mcrfpy.Caption(text=text, pos=(x, y))
|
|
label.fill_color = mcrfpy.Color(*color)
|
|
self.ui.append(label)
|
|
return label
|
|
|
|
def create_grid(self, grid_size, pos, size, cell_size=16):
|
|
"""Create a grid for visualization."""
|
|
grid = mcrfpy.Grid(
|
|
grid_size=grid_size,
|
|
pos=pos,
|
|
size=size
|
|
)
|
|
grid.fill_color = mcrfpy.Color(20, 20, 30)
|
|
self.ui.append(grid)
|
|
return grid
|
|
|
|
def color_grid_cell(self, grid, x, y, color):
|
|
"""Color a specific grid cell."""
|
|
try:
|
|
point = grid.at(x, y)
|
|
point.color = mcrfpy.Color(*color) if isinstance(color, tuple) else color
|
|
except:
|
|
pass # Out of bounds
|
|
|
|
def add_timer(self, name, callback, interval):
|
|
"""Add a timer and track it for cleanup."""
|
|
mcrfpy.setTimer(name, callback, interval)
|
|
self.timers.append(name)
|