mirror of
https://github.com/csd4ni3l/simulator-games.git
synced 2026-01-01 12:23:44 +01:00
98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
import pyglet, arcade, arcade.gui, os, json, numpy as np
|
|
|
|
from game.voronoi_diagram_simulator.shader import create_shader
|
|
from game.base import BaseGame
|
|
|
|
from utils.constants import button_style
|
|
from utils.preload import button_texture, button_hovered_texture
|
|
|
|
class Game(BaseGame):
|
|
def __init__(self, pypresence_client):
|
|
super().__init__(pypresence_client, "Voronoi Diagram Simulator", "voronoi_diagram_simulator", {
|
|
"edge_thickness": 0.01,
|
|
"edge_smoothness": 0.005
|
|
})
|
|
|
|
self.points = np.empty((0, 2), dtype=np.float32)
|
|
|
|
self.dragged_point = None
|
|
self.needs_redraw = True
|
|
|
|
def setup(self):
|
|
self.shader_program, self.voronoi_image, self.points_sbbo = create_shader(int(self.window.width * 0.8), self.window.height)
|
|
|
|
self.image_sprite = pyglet.sprite.Sprite(img=self.voronoi_image)
|
|
|
|
def on_show_view(self):
|
|
super().on_show_view()
|
|
|
|
self.add_setting("Edge Thickness: {value}", 0, 0.1, 0.01, "edge_thickness")
|
|
self.add_setting("Edge Smoothness: {value}", 0, 0.1, 0.01, "edge_smoothness")
|
|
|
|
self.add_point_button = self.settings_box.add(arcade.gui.UITextureButton(text="Add point", texture=button_texture, texture_hovered=button_hovered_texture, width=self.window.width * 0.2, style=button_style))
|
|
self.add_point_button.on_click = lambda event: self.add_point()
|
|
|
|
self.setup()
|
|
|
|
def add_point(self):
|
|
self.points = np.vstack([self.points, [self.window.width * 0.4, self.window.height / 2]]).astype(np.float32)
|
|
|
|
self.needs_redraw = True
|
|
|
|
def change_value(self, label, text, settings_key, value):
|
|
super().change_value(label, text, settings_key, value)
|
|
|
|
self.needs_redraw = True
|
|
|
|
def on_update(self, delta_time):
|
|
if self.needs_redraw:
|
|
self.needs_redraw = False
|
|
|
|
self.points_sbbo.set_data(self.points.tobytes())
|
|
|
|
with self.shader_program:
|
|
self.shader_program["point_count"] = len(self.points)
|
|
self.shader_program["resolution"] = (int(self.window.width * 0.8), self.window.height)
|
|
self.shader_program["edge_smoothness"] = self.settings["voronoi_diagram_simulator"]["edge_smoothness"]
|
|
self.shader_program["edge_thickness"] = self.settings["voronoi_diagram_simulator"]["edge_thickness"]
|
|
|
|
self.shader_program.dispatch(int(int(self.window.width * 0.8) / 32), int(self.window.height / 32))
|
|
|
|
def on_mouse_press(self, x, y, button, modifiers):
|
|
if not self.dragged_point:
|
|
for i, point in enumerate(self.points.reshape(-1, 2)):
|
|
if arcade.math.Vec2(x, y).distance(arcade.math.Vec2(*point)) <= 10:
|
|
self.dragged_point = i
|
|
break
|
|
|
|
def on_mouse_drag(self, x, y, dx, dy, _buttons, _modifiers):
|
|
if self.dragged_point is not None:
|
|
self.points[self.dragged_point] = [x, y]
|
|
|
|
def on_mouse_release(self, x, y, button, modifiers):
|
|
self.dragged_point = None
|
|
self.needs_redraw = True
|
|
|
|
def on_key_press(self, symbol, modifiers):
|
|
if symbol == arcade.key.ESCAPE: # overwrite is needed since shader program needs to be deleted
|
|
self.shader_program.delete()
|
|
|
|
with open("data.json", "w") as file:
|
|
file.write(json.dumps(self.settings, indent=4))
|
|
|
|
from menus.main import Main
|
|
self.window.show_view(Main(self.pypresence_client))
|
|
|
|
elif symbol == arcade.key.C:
|
|
del self.points
|
|
self.points = np.empty((0, 2), dtype=np.float32)
|
|
|
|
self.needs_redraw = True
|
|
|
|
def on_draw(self):
|
|
super().on_draw()
|
|
|
|
self.image_sprite.draw()
|
|
|
|
for point in self.points.reshape(-1, 2):
|
|
arcade.draw_circle_filled(point[0], point[1], 10, arcade.color.GRAY) |