csd4ni3l/fractal-viewer
1
0
Fork 0
mirror of https://github.com/csd4ni3l/fractal-viewer.git synced 2026-01-01 04:13:41 +01:00
Code Issues Packages Projects Releases 1 Wiki Activity

Add preturbation code which doesnt work yet, update to newest build script and run script and preload script

Browse Source
This commit is contained in:
csd4ni3l
2025-12-13 15:14:02 +01:00
parent 9424bf589a
commit 5ef274f60d
9 changed files with 775 additions and 389 deletions
Download Patch File Download Diff File Expand all files Collapse all files

169
game/iter_fractal_viewer.py
View File

@@ -1,5 +1,7 @@
import arcade, arcade.gui, pyglet, json
from mpmath import mpc
from game.shader import create_iter_calc_shader
from utils.constants import button_style, initial_real_imag
from utils.preload import button_texture, button_hovered_texture, cursor_texture
@@ -14,50 +16,96 @@ class IterFractalViewer(arcade.gui.UIView):
with open("settings.json", "r") as file:
self.settings_dict = json.load(file)
self.escape_radius = int(self.settings_dict.get(f"{self.fractal_name}_escape_radius", 2))
self.real_min, self.real_max, self.imag_min, self.imag_max = initial_real_imag[fractal_name] if fractal_name != "julia" else (-self.escape_radius, self.escape_radius, -self.escape_radius, self.escape_radius)
self.escape_radius = int(
self.settings_dict.get(f"{self.fractal_name}_escape_radius", 2)
)
self.real_min, self.real_max, self.imag_min, self.imag_max = (
initial_real_imag[fractal_name]
if fractal_name != "julia"
else (
-self.escape_radius,
self.escape_radius,
-self.escape_radius,
self.escape_radius,
)
)
self.max_iter = self.settings_dict.get(f"{self.fractal_name}_max_iter", 200)
self.zoom = 1.0
self.zoom_start_position = ()
self.zoom_rect = None
self.has_controller = False
self.dragging_with_controller = False
self.center_real = self.real_min + 0.5 * (self.real_max - self.real_min)
self.center_imag = self.imag_min + 0.5 * (self.imag_max - self.imag_min)
def on_show_view(self):
super().on_show_view()
self.shader_program, self.fractal_image = create_iter_calc_shader(
self.fractal_name,
self.fractal_name,
self.window.width,
self.window.height,
self.settings_dict.get(f"{self.fractal_name}_precision", "Single").lower(),
int(self.settings_dict.get(f"{self.fractal_name}_n", 2)), # This will work for non-exponentiable fractals as well because they dont have an _n property
int(
self.settings_dict.get(f"{self.fractal_name}_n", 2)
), # This will work for non-exponentiable fractals as well because they dont have an _n property
int(self.settings_dict.get(f"{self.fractal_name}_escape_radius", 2)),
self.settings_dict.get("julia_type", "Classic swirling")
self.settings_dict.get("julia_type", "Classic swirling"),
False,
mpc(self.real_min, self.imag_min),
int(self.max_iter),
)
self.fractal_sprite = pyglet.sprite.Sprite(img=self.fractal_image)
self.create_image()
self.pypresence_client.update(state=f'Viewing {self.fractal_name.replace("_", " ").capitalize()}', details=f'Zoom: {self.zoom}\nMax Iterations: {self.max_iter}', start=self.pypresence_client.start_time)
self.pypresence_client.update(
state=f"Viewing {self.fractal_name.replace('_', ' ').capitalize()}",
details=f"Zoom: {self.zoom}\nMax Iterations: {self.max_iter}",
start=self.pypresence_client.start_time,
)
self.setup_ui()
def main_exit(self):
from menus.main import Main
self.window.show_view(Main(self.pypresence_client))
def setup_ui(self):
self.anchor = self.add_widget(arcade.gui.UIAnchorLayout(size_hint=(1, 1)))
self.info_box = self.anchor.add(arcade.gui.UIBoxLayout(space_between=10, vertical=False), anchor_x="center", anchor_y="top")
self.zoom_label = self.info_box.add(arcade.gui.UILabel(text=f"Zoom: {self.zoom}", font_name="Roboto", font_size=16))
self.max_iter_label = self.info_box.add(arcade.gui.UILabel(text=f"Max Iterations: {self.max_iter}", font_name="Roboto", font_size=16))
self.info_box = self.anchor.add(
arcade.gui.UIBoxLayout(space_between=10, vertical=False),
anchor_x="center",
anchor_y="top",
)
self.zoom_label = self.info_box.add(
arcade.gui.UILabel(
text=f"Zoom: {self.zoom}", font_name="Roboto", font_size=16
)
)
self.max_iter_label = self.info_box.add(
arcade.gui.UILabel(
text=f"Max Iterations: {self.max_iter}",
font_name="Roboto",
font_size=16,
)
)
self.back_button = arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='<--', style=button_style, width=100, height=50)
self.back_button = arcade.gui.UITextureButton(
texture=button_texture,
texture_hovered=button_hovered_texture,
text="<--",
style=button_style,
width=100,
height=50,
)
self.back_button.on_click = lambda event: self.main_exit()
self.anchor.add(self.back_button, anchor_x="left", anchor_y="top", align_x=5, align_y=-5)
self.anchor.add(
self.back_button, anchor_x="left", anchor_y="top", align_x=5, align_y=-5
)
if self.window.get_controllers():
self.sprite_list = arcade.SpriteList()
@@ -67,7 +115,10 @@ class IterFractalViewer(arcade.gui.UIView):
def on_button_press(self, controller, button_name):
if button_name == "a":
self.zoom_start_position = self.cursor_sprite.center_x, self.cursor_sprite.center_y
self.zoom_start_position = (
self.cursor_sprite.center_x,
self.cursor_sprite.center_y,
)
self.dragging_with_controller = True
elif button_name == "start":
self.main_exit()
@@ -75,26 +126,55 @@ class IterFractalViewer(arcade.gui.UIView):
def on_button_release(self, controller, button_name):
if button_name == "a" and self.dragging_with_controller:
self.dragging_with_controller = False
self.on_mouse_release(self.cursor_sprite.center_x, self.cursor_sprite.center_y, None, None)
self.on_mouse_release(
self.cursor_sprite.center_x, self.cursor_sprite.center_y, None, None
)
def on_stick_motion(self, controller, name, vector):
if name == "leftstick":
self.cursor_sprite.center_x += vector.x * 5
self.cursor_sprite.center_y += vector.y * 5
if self.dragging_with_controller:
self.on_mouse_drag(self.cursor_sprite.center_x, self.cursor_sprite.center_y, 0, 0, None, None)
self.on_mouse_drag(
self.cursor_sprite.center_x,
self.cursor_sprite.center_y,
0,
0,
None,
None,
)
def create_image(self):
with self.shader_program:
self.shader_program['u_maxIter'] = int(self.max_iter)
self.shader_program['u_resolution'] = (self.window.width, self.window.height)
self.shader_program['u_real_range'] = (self.real_min, self.real_max)
self.shader_program['u_imag_range'] = (self.imag_min, self.imag_max)
self.shader_program.dispatch(self.fractal_image.width, self.fractal_image.height, 1, barrier=pyglet.gl.GL_ALL_BARRIER_BITS)
self.shader_program["u_maxIter"] = int(self.max_iter)
self.shader_program["u_resolution"] = (
self.window.width,
self.window.height,
)
self.shader_program["u_real_range"] = (self.real_min, self.real_max)
self.shader_program["u_imag_range"] = (self.imag_min, self.imag_max)
if False:
self.shader_program["u_center"] = (self.center_real, self.center_imag)
self.shader_program["usepreturbation"] = False
self.shader_program.dispatch(
self.fractal_image.width,
self.fractal_image.height,
1,
barrier=pyglet.gl.GL_ALL_BARRIER_BITS,
)
def on_key_press(self, symbol, modifiers):
if symbol == arcade.key.ESCAPE:
self.real_min, self.real_max, self.imag_min, self.imag_max = initial_real_imag[self.fractal_name] if self.fractal_name != "julia" else (-self.escape_radius, self.escape_radius, -self.escape_radius, self.escape_radius)
self.real_min, self.real_max, self.imag_min, self.imag_max = (
initial_real_imag[self.fractal_name]
if self.fractal_name != "julia"
else (
-self.escape_radius,
self.escape_radius,
-self.escape_radius,
self.escape_radius,
)
)
self.zoom = 1
@@ -106,9 +186,9 @@ class IterFractalViewer(arcade.gui.UIView):
self.create_image()
self.pypresence_client.update(
state=f'Viewing {self.fractal_name.replace("_", " ").capitalize()}',
details=f'Zoom: {self.zoom:.2f}\nMax Iterations: {self.max_iter}',
start=self.pypresence_client.start_time
state=f"Viewing {self.fractal_name.replace('_', ' ').capitalize()}",
details=f"Zoom: {self.zoom:.2f}\nMax Iterations: {self.max_iter}",
start=self.pypresence_client.start_time,
)
def on_mouse_drag(self, x, y, dx, dy, _buttons, _modifiers):
@@ -137,7 +217,9 @@ class IterFractalViewer(arcade.gui.UIView):
bottom = min(y0, y1)
top = max(y0, y1)
self.zoom_rect = arcade.rect.LRBT(left=left, right=right, top=top, bottom=bottom)
self.zoom_rect = arcade.rect.LRBT(
left=left, right=right, top=top, bottom=bottom
)
def on_mouse_release(self, x, y, button, modifiers):
if self.zoom_start_position and self.zoom_rect:
@@ -146,18 +228,29 @@ class IterFractalViewer(arcade.gui.UIView):
center_x = (rect.left + rect.right) / 2
center_y = (rect.bottom + rect.top) / 2
center_real = self.real_min + (center_x / self.width) * (self.real_max - self.real_min)
center_imag = self.imag_min + (center_y / self.height) * (self.imag_max - self.imag_min)
self.center_real = self.real_min + (center_x / self.width) * (
self.real_max - self.real_min
)
self.center_imag = self.imag_min + (center_y / self.height) * (
self.imag_max - self.imag_min
)
real_span = (rect.right - rect.left) / self.width * (self.real_max - self.real_min)
imag_span = (rect.top - rect.bottom) / self.height * (self.imag_max - self.imag_min)
real_span = (
(rect.right - rect.left) / self.width * (self.real_max - self.real_min)
)
imag_span = (
(rect.top - rect.bottom) / self.height * (self.imag_max - self.imag_min)
)
self.real_min = center_real - real_span / 2
self.real_max = center_real + real_span / 2
self.imag_min = center_imag - imag_span / 2
self.imag_max = center_imag + imag_span / 2
self.real_min = self.center_real - real_span / 2
self.real_max = self.center_real + real_span / 2
self.imag_min = self.center_imag - imag_span / 2
self.imag_max = self.center_imag + imag_span / 2
initial_real_range = initial_real_imag[self.fractal_name][1] - initial_real_imag[self.fractal_name][0]
initial_real_range = (
initial_real_imag[self.fractal_name][1]
- initial_real_imag[self.fractal_name][0]
)
new_real_range = self.real_max - self.real_min
self.zoom = initial_real_range / new_real_range
@@ -169,9 +262,9 @@ class IterFractalViewer(arcade.gui.UIView):
self.create_image()
self.pypresence_client.update(
state=f'Viewing {self.fractal_name.replace("_", " ").capitalize()}',
details=f'Zoom: {self.zoom:.4f}\nMax Iterations: {self.max_iter}',
start=self.pypresence_client.start_time
state=f"Viewing {self.fractal_name.replace('_', ' ').capitalize()}",
details=f"Zoom: {self.zoom:.4f}\nMax Iterations: {self.max_iter}",
start=self.pypresence_client.start_time,
)
def on_draw(self):
@@ -181,6 +274,6 @@ class IterFractalViewer(arcade.gui.UIView):
if self.has_controller:
self.sprite_list.draw()
if self.zoom_rect:
arcade.draw_rect_outline(self.zoom_rect, color=arcade.color.GRAY)