Add double pendulum simulator, make menu buttons smaller

game/double_pendulum_simulator/game.py

@@ -0,0 +1,135 @@
+import arcade, arcade.gui, json, os, math
+
+from utils.constants import button_style
+from utils.preload import button_texture, button_hovered_texture
+
+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="Double Pendulum 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 "double_pendulum_simulator" in self.settings:
+            self.settings["double_pendulum_simulator"] = {
+                "length_a": 200,
+                "length_b": 200,
+                "mass_a": 40,
+                "mass_b": 40,
+                "gravity": 9.81,
+                "speed": 1.0,
+                "trail_size": 500
+            }
+
+        self.theta_a, self.theta_b = math.pi / 2, math.pi / 2
+        self.omega_a, self.omega_b = 0, 0
+        self.origin_x, self.origin_y = self.window.width * 0.4, self.window.height / 2
+
+        self.trace = []
+
+    def on_show_view(self):
+        super().on_show_view()
+
+        self.add_setting("Length A: {value}", 5, 500, 5, "length_a")
+        self.add_setting("Length B: {value}", 5, 500, 5, "length_b")
+        self.add_setting("Mass A: {value}", 1, 100, 1, "mass_a")
+        self.add_setting("Mass B: {value}", 1, 100, 1, "mass_b")
+        self.add_setting("Gravity: {value}", 0, 25, 0.01, "gravity")
+        self.add_setting("Speed: {value}", 1, 100, 1, "speed")
+        self.add_setting("Trail Size: {value}", 0, 5000, 50, "trail_size")
+
+    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["double_pendulum_simulator"][settings_key])))
+        slider = self.settings_box.add(arcade.gui.UISlider(value=self.settings["double_pendulum_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["double_pendulum_simulator"][settings_key] = value
+
+    def on_draw(self):
+        super().on_draw()
+
+        current_settings = self.settings["double_pendulum_simulator"]
+        L1, L2 = current_settings["length_a"], current_settings["length_b"]
+        m1, m2 = current_settings["mass_a"], current_settings["mass_b"]
+
+        x1 = self.origin_x + L1 * math.sin(self.theta_a)
+        y1 = self.origin_y - L1 * math.cos(self.theta_a)
+
+        x2 = x1 + L2 * math.sin(self.theta_b)
+        y2 = y1 - L2 * math.cos(self.theta_b)
+
+        arcade.draw_line(self.origin_x, self.origin_y, x1, y1, arcade.color.BLACK, 2)
+        arcade.draw_line(x1, y1, x2, y2, arcade.color.BLACK, 2)
+
+        arcade.draw_circle_filled(x1, y1, m1 / 2, arcade.color.RED)
+        arcade.draw_circle_filled(x2, y2, m2 / 2, arcade.color.BLUE)
+
+        arcade.draw_points(self.trace, arcade.color.GREEN, 2)
+
+    def on_update(self, dt):
+        for _ in range(int(self.settings["double_pendulum_simulator"]["speed"])):
+            current_settings = self.settings["double_pendulum_simulator"]
+            
+            g = current_settings["gravity"]
+
+            m1, m2 = current_settings["mass_a"], current_settings["mass_b"]
+            L1, L2 = current_settings["length_a"], current_settings["length_b"]
+            t1, t2 = self.theta_a, self.theta_b
+            w1, w2 = self.omega_a, self.omega_b
+
+            num1 = -g * (2 * m1 + m2) * math.sin(t1)
+            num2 = -m2 * g * math.sin(t1 - 2 * t2)
+            num3 = -2 * math.sin(t1 - t2) * m2 * (w2 ** 2 * L2 + w1 ** 2 * L1 * math.cos(t1 - t2))
+            den = L1 * (2 * m1 + m2 - m2 * math.cos(2 * t1 - 2 * t2))
+            a1 = (num1 + num2 + num3) / den
+
+            num1 = 2 * math.sin(t1 - t2)
+            num2 = w1 ** 2 * L1 * (m1 + m2)
+            num3 = g * (m1 + m2) * math.cos(t1)
+            num4 = w2 ** 2 * L2 * m2 * math.cos(t1 - t2)
+            den = L2 * (2 * m1 + m2 - m2 * math.cos(2 * t1 - 2 * t2))
+            a2 = (num1 * (num2 + num3 + num4)) / den
+
+            self.omega_a += a1 * dt
+            self.omega_b += a2 * dt
+
+            self.theta_a += self.omega_a * dt
+            self.theta_b += self.omega_b * dt
+
+            x1 = self.origin_x + L1 * math.sin(self.theta_a)
+            y1 = self.origin_y - L1 * math.cos(self.theta_a)
+
+            x2 = x1 + L2 * math.sin(self.theta_b)
+            y2 = y1 - L2 * math.cos(self.theta_b)
+
+            if not current_settings["trail_size"] == 0:
+                self.trace.append((x2, y2))
+
+                if len(self.trace) > current_settings["trail_size"]:
+                    self.trace = self.trace[-int(current_settings["trail_size"]):]
+            else:
+                self.trace = []
+
+    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))