diff --git a/game/lorenz_attractor_simulator/game.py b/game/lorenz_attractor_simulator/game.py new file mode 100644 index 0000000..db95c47 --- /dev/null +++ b/game/lorenz_attractor_simulator/game.py @@ -0,0 +1,111 @@ +import pyglet, arcade, arcade.gui, os, json, time + +from game.lorenz_attractor_simulator.shader import create_shader + +class Game(arcade.gui.UIView): + def __init__(self, pypresence_client): + super().__init__() + + self.pypresence_client = pypresence_client + self.pypresence_client.update(state="Playing a simulator", details="Lorenz Attractor Simulator") + + self.anchor = self.add_widget(arcade.gui.UIAnchorLayout(size_hint=(1, 1))) + + self.settings_box = self.anchor.add(arcade.gui.UIBoxLayout(align="center", size_hint=(0.2, 1)).with_background(color=arcade.color.GRAY), anchor_x="right", anchor_y="bottom") + self.settings_label = self.settings_box.add(arcade.gui.UILabel(text="Settings", font_size=24)) + + if os.path.exists("data.json"): + with open("data.json", "r") as file: + self.settings = json.load(file) + + else: + self.settings = {} + + if not "lorenz_attractor_simulator" in self.settings: + self.settings["lorenz_attractor_simulator"] = { + "sigma": 10, + "rho": 28, + "beta": 2.66666666, + "steps": 50, + "decay_factor": 0.999, + "speed": 1 + } + + self.delta_time = 0 + self.should_clear = False + + self.last_update = time.perf_counter() + + def setup(self): + self.shader_program, self.lorenz_image = create_shader(int(self.window.width * 0.8), self.window.height) + + self.image_sprite = pyglet.sprite.Sprite(self.lorenz_image) + + def on_show_view(self): + super().on_show_view() + + self.add_setting("Sigma: {value}", 1, 50, 0.1, "sigma") + self.add_setting("Rho: {value}", 1, 100, 0.1, "rho") + self.add_setting("Beta: {value}", 0.1, 10, 0.01, "beta") + self.add_setting("Steps: {value}", 50, 1000, 10, "steps") + self.add_setting("Decay multiplier: {value}", 0.8, 1, 0.001, "decay_factor") + self.add_setting("Speed: {value}", 0.1, 100, 0.1, "speed") + + self.setup() + + def add_setting(self, text, min_value, max_value, step, settings_key): + label = self.settings_box.add(arcade.gui.UILabel(text.format(value=self.settings["lorenz_attractor_simulator"][settings_key]))) + slider = self.settings_box.add(arcade.gui.UISlider(value=self.settings["lorenz_attractor_simulator"][settings_key], min_value=min_value, max_value=max_value, step=step)) + slider._render_steps = lambda surface: None + + slider.on_change = lambda event, label=label: self.change_value(label, text, settings_key, event.new_value) + + def change_value(self, label, text, settings_key, value): + label.text = text.format(value=value) + + self.settings["lorenz_attractor_simulator"][settings_key] = value + + self.should_clear = True + + def on_update(self, delta_time): + if time.perf_counter() - self.last_update >= 1 / 15: + if self.should_clear: + self.should_clear = False + + self.delta_time = 0 + pyglet.gl.glClearTexImage( + self.lorenz_image.id, 0, + pyglet.gl.GL_RGBA, pyglet.gl.GL_FLOAT, + None + ) + + current_settings = self.settings["lorenz_attractor_simulator"] + + with self.shader_program: + self.shader_program["sigma"] = current_settings["sigma"] + self.shader_program["rho"] = current_settings["rho"] + self.shader_program["beta"] = current_settings["beta"] + self.shader_program["steps"] = int(current_settings["steps"]) + self.shader_program["dt"] = self.delta_time + self.shader_program["resolution"] = (int(self.window.width * 0.8), self.window.height) + self.shader_program["decay_factor"] = current_settings["decay_factor"] + + self.shader_program.dispatch(int(self.window.width * 0.8) // 32, self.window.height // 32) + + pyglet.gl.glMemoryBarrier(pyglet.gl.GL_SHADER_IMAGE_ACCESS_BARRIER_BIT) + + self.delta_time += (current_settings["speed"] * 0.00005) + + def on_key_press(self, symbol, modifiers): + if symbol == arcade.key.ESCAPE: + 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)) + + + def on_draw(self): + super().on_draw() + + self.image_sprite.draw() \ No newline at end of file diff --git a/game/lorenz_attractor_simulator/shader.py b/game/lorenz_attractor_simulator/shader.py new file mode 100644 index 0000000..6f37900 --- /dev/null +++ b/game/lorenz_attractor_simulator/shader.py @@ -0,0 +1,75 @@ +import pyglet, pyglet.graphics + +from pyglet.gl import GL_NEAREST + +shader_source = """#version 430 core +uniform vec2 resolution; + +uniform float sigma; +uniform float rho; +uniform float beta; +uniform float dt; +uniform int steps; +uniform float decay_factor; + +layout (local_size_x = 32, local_size_y = 32, local_size_z = 1) in; +layout (location = 0, rgba32f) uniform image2D img_output; + +vec2 project (vec3 p) { + return (vec2(p.x + 25, p.z) * 0.02); +} + +void main() { + ivec2 texel_coord = ivec2(gl_GlobalInvocationID.xy); + + if (texel_coord.x >= int(resolution.x) || texel_coord.y >= int(resolution.y)) { + return; + } + + vec4 current_color = imageLoad(img_output, texel_coord); + current_color.rgb *= decay_factor; + imageStore(img_output, texel_coord, current_color); + + for (int seed_offset = 0; seed_offset < 4; seed_offset++) { + float seedx = (float(texel_coord.x + seed_offset * 0.25) / resolution.x - 0.5) * 20.0; + float seedy = (float(texel_coord.y + seed_offset * 0.25) / resolution.y - 0.5) * 20.0; + float seedz = 15.0 + seed_offset * 2.0; + + vec3 p = vec3(seedx, seedy, seedz); + + for (int i = 0; i < steps; i++) { + float dx = sigma * (p.y - p.x); + float dy = p.x * (rho - p.z) - p.y; + float dz = p.x * p.y - beta * p.z; + + p += vec3(dx, dy, dz) * dt; + + vec2 uv = project(p); + ivec2 coord = ivec2(uv * resolution); + + if (coord.x >= 0 && coord.x < int(resolution.x) && + coord.y >= 0 && coord.y < int(resolution.y)) { + + vec4 old_color = imageLoad(img_output, coord); + float intensity = 0.008; + + vec3 new_color = old_color.rgb + vec3(intensity, intensity * 0.6, intensity * 0.9); + new_color = min(new_color, vec3(1.0)); + imageStore(img_output, coord, vec4(new_color, 1.0)); + + } + } + } +} + +""" + +def create_shader(width, height): + shader_program = pyglet.graphics.shader.ComputeShaderProgram(shader_source) + + lorenz_attractor_image = pyglet.image.Texture.create(width, height, internalformat=pyglet.gl.GL_RGBA32F, min_filter=GL_NEAREST, mag_filter=GL_NEAREST) + + uniform_location = shader_program["img_output"] + lorenz_attractor_image.bind_image_texture(unit=uniform_location) + + return shader_program, lorenz_attractor_image \ No newline at end of file diff --git a/game/voronoi_diagram_simulator/game.py b/game/voronoi_diagram_simulator/game.py index a020718..4bd7ec3 100644 --- a/game/voronoi_diagram_simulator/game.py +++ b/game/voronoi_diagram_simulator/game.py @@ -48,7 +48,7 @@ class Game(arcade.gui.UIView): 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)) + 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() diff --git a/menus/main.py b/menus/main.py index 6c06402..b0bc72e 100644 --- a/menus/main.py +++ b/menus/main.py @@ -73,6 +73,9 @@ class Main(arcade.gui.UIView): self.voronoi_diagram_simulator_button = self.box.add(arcade.gui.UITextureButton(text="Voronoi Diagram Simulator", texture=button_texture, texture_hovered=button_hovered_texture, width=self.window.width / 2, height=self.window.height / 12, style=big_button_style)) self.voronoi_diagram_simulator_button.on_click = lambda event: self.voronoi_diagram_simulator() + self.lorenz_attractor_simulator_button = self.box.add(arcade.gui.UITextureButton(text="Lorenz Attractor Simulator", texture=button_texture, texture_hovered=button_hovered_texture, width=self.window.width / 2, height=self.window.height / 12, style=big_button_style)) + self.lorenz_attractor_simulator_button.on_click = lambda event: self.lorenz_attractor_simulator() + self.settings_button = self.box.add(arcade.gui.UITextureButton(text="Settings", texture=button_texture, texture_hovered=button_hovered_texture, width=self.window.width / 2, height=self.window.height / 12, style=big_button_style)) self.settings_button.on_click = lambda event: self.settings() @@ -104,6 +107,10 @@ class Main(arcade.gui.UIView): from game.voronoi_diagram_simulator.game import Game self.window.show_view(Game(self.pypresence_client)) + def lorenz_attractor_simulator(self): + from game.lorenz_attractor_simulator.game import Game + self.window.show_view(Game(self.pypresence_client)) + def settings(self): from menus.settings import Settings self.window.show_view(Settings(self.pypresence_client))