mirror of
https://github.com/csd4ni3l/fractal-viewer.git
synced 2026-01-01 04:13:41 +01:00
Rename to Fractal Viewer, add Sierpinsky Carpet, use arcade 3.2.0
instead of development
This commit is contained in:
csd4ni3l
95
game/mandelbrot.py
Normal file
95
game/mandelbrot.py
Normal file
@@ -0,0 +1,95 @@
|
||||
import arcade, arcade.gui, pyglet, json
|
||||
|
||||
from PIL import Image
|
||||
|
||||
from game.shader import create_mandelbrot_shader
|
||||
from utils.constants import menu_background_color, button_style, initial_real_min, initial_real_max, initial_imag_min, initial_imag_max
|
||||
from utils.preload import button_texture, button_hovered_texture
|
||||
|
||||
class MandelbrotViewer(arcade.gui.UIView):
|
||||
def __init__(self, pypresence_client):
|
||||
super().__init__()
|
||||
|
||||
self.pypresence_client = pypresence_client
|
||||
self.real_min = initial_real_min
|
||||
self.real_max = initial_real_max
|
||||
self.imag_min = initial_imag_min
|
||||
self.imag_max = initial_imag_max
|
||||
|
||||
with open("settings.json", "r") as file:
|
||||
self.settings_dict = json.load(file)
|
||||
|
||||
self.max_iter = self.settings_dict.get("mandelbrot_max_iter", 200)
|
||||
self.zoom = 1.0
|
||||
|
||||
def on_show_view(self):
|
||||
super().on_show_view()
|
||||
|
||||
self.shader_program, self.mandelbrot_image = create_mandelbrot_shader(self.window.width, self.window.height, self.settings_dict.get("mandelbrot_precision", "Single").lower())
|
||||
|
||||
self.mandelbrot_sprite = pyglet.sprite.Sprite(img=self.mandelbrot_image)
|
||||
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Mandelbrot', 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="Protest Strike", font_size=16))
|
||||
self.max_iter_label = self.info_box.add(arcade.gui.UILabel(text=f"Max Iterations: {self.max_iter}", font_name="Protest Strike", 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.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)
|
||||
|
||||
def zoom_at(self, center_x, center_y, zoom_factor):
|
||||
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)
|
||||
|
||||
new_real_range = (self.real_max - self.real_min) / zoom_factor
|
||||
new_imag_range = (self.imag_max - self.imag_min) / zoom_factor
|
||||
|
||||
self.real_min = center_real - new_real_range / 2
|
||||
self.real_max = center_real + new_real_range / 2
|
||||
self.imag_min = center_imag - new_imag_range / 2
|
||||
self.imag_max = center_imag + new_imag_range / 2
|
||||
|
||||
def create_image(self):
|
||||
with self.shader_program:
|
||||
self.shader_program['u_maxIter'] = 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.mandelbrot_image.width, self.mandelbrot_image.height, 1, barrier=pyglet.gl.GL_ALL_BARRIER_BITS)
|
||||
|
||||
def on_mouse_press(self, x: int, y: int, button: int, modifiers: int) -> bool | None:
|
||||
super().on_mouse_press(x, y, button, modifiers)
|
||||
|
||||
if button == arcade.MOUSE_BUTTON_LEFT:
|
||||
zoom = self.settings_dict.get("mandelbrot_zoom_increase", 2)
|
||||
elif button == arcade.MOUSE_BUTTON_RIGHT:
|
||||
zoom = 1 / self.settings_dict.get("mandelbrot_zoom_increase", 2)
|
||||
else:
|
||||
return
|
||||
|
||||
self.zoom *= zoom
|
||||
|
||||
self.zoom_label.text = f"Zoom: {self.zoom}"
|
||||
|
||||
self.zoom_at(self.window.mouse.data["x"], self.window.mouse.data["y"], zoom)
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Mandelbrot', details=f'Zoom: {self.zoom}\nMax Iterations: {self.max_iter}', start=self.pypresence_client.start_time)
|
||||
|
||||
def on_draw(self):
|
||||
self.window.clear()
|
||||
self.mandelbrot_sprite.draw()
|
||||
self.ui.draw()
|
||||
93
game/play.py
93
game/play.py
@@ -1,95 +1,42 @@
|
||||
import arcade, arcade.gui, pyglet, json
|
||||
import arcade, arcade.gui
|
||||
|
||||
from PIL import Image
|
||||
|
||||
from game.shader import create_mandelbrot_shader
|
||||
from utils.constants import menu_background_color, button_style, initial_real_min, initial_real_max, initial_imag_min, initial_imag_max
|
||||
from utils.constants import button_style
|
||||
from utils.preload import button_texture, button_hovered_texture
|
||||
|
||||
class Game(arcade.gui.UIView):
|
||||
class FractalChooser(arcade.gui.UIView):
|
||||
def __init__(self, pypresence_client):
|
||||
super().__init__()
|
||||
|
||||
self.pypresence_client = pypresence_client
|
||||
self.real_min = initial_real_min
|
||||
self.real_max = initial_real_max
|
||||
self.imag_min = initial_imag_min
|
||||
self.imag_max = initial_imag_max
|
||||
|
||||
with open("settings.json", "r") as file:
|
||||
self.settings_dict = json.load(file)
|
||||
|
||||
self.max_iter = self.settings_dict.get("max_iter", 200)
|
||||
self.zoom = 1.0
|
||||
|
||||
def on_show_view(self):
|
||||
super().on_show_view()
|
||||
|
||||
self.shader_program, self.mandelbrot_image = create_mandelbrot_shader(self.window.width, self.window.height, self.settings_dict.get("precision", "Single").lower())
|
||||
|
||||
self.mandelbrot_sprite = pyglet.sprite.Sprite(img=self.mandelbrot_image)
|
||||
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Mandelbrot', 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="Protest Strike", font_size=16))
|
||||
self.max_iter_label = self.info_box.add(arcade.gui.UILabel(text=f"Max Iterations: {self.max_iter}", font_name="Protest Strike", font_size=16))
|
||||
self.grid = self.add_widget(arcade.gui.UIGridLayout(row_count=3, column_count=3, horizontal_spacing=10, vertical_spacing=10))
|
||||
self.anchor.add(self.grid, anchor_x="center", anchor_y="center")
|
||||
|
||||
self.title_label = self.anchor.add(arcade.gui.UILabel(text="Choose a fractal to view.", font_name="Protest Strike", font_size=32), anchor_x="center", anchor_y="top", align_y=-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)
|
||||
|
||||
def zoom_at(self, center_x, center_y, zoom_factor):
|
||||
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.mandelbrot_button = self.grid.add(arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='Mandelbrot', style=button_style, width=200, height=200), row=0, column=0)
|
||||
self.mandelbrot_button.on_click = lambda event: self.mandelbrot()
|
||||
|
||||
new_real_range = (self.real_max - self.real_min) / zoom_factor
|
||||
new_imag_range = (self.imag_max - self.imag_min) / zoom_factor
|
||||
self.sierpinsky_carpet_button = self.grid.add(arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='Sierpinsky Carpet', style=button_style, width=200, height=200), row=0, column=1)
|
||||
self.sierpinsky_carpet_button.on_click = lambda event: self.sierpinsky_carpet()
|
||||
|
||||
self.real_min = center_real - new_real_range / 2
|
||||
self.real_max = center_real + new_real_range / 2
|
||||
self.imag_min = center_imag - new_imag_range / 2
|
||||
self.imag_max = center_imag + new_imag_range / 2
|
||||
def main_exit(self):
|
||||
from menus.main import Main
|
||||
self.window.show_view(Main(self.pypresence_client))
|
||||
|
||||
def create_image(self):
|
||||
with self.shader_program:
|
||||
self.shader_program['u_maxIter'] = 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.mandelbrot_image.width, self.mandelbrot_image.height, 1, barrier=pyglet.gl.GL_ALL_BARRIER_BITS)
|
||||
def mandelbrot(self):
|
||||
from game.mandelbrot import MandelbrotViewer
|
||||
self.window.show_view(MandelbrotViewer(self.pypresence_client))
|
||||
|
||||
def on_mouse_press(self, x: int, y: int, button: int, modifiers: int) -> bool | None:
|
||||
super().on_mouse_press(x, y, button, modifiers)
|
||||
|
||||
if button == arcade.MOUSE_BUTTON_LEFT:
|
||||
zoom = self.settings_dict.get("zoom_increase", 2)
|
||||
elif button == arcade.MOUSE_BUTTON_RIGHT:
|
||||
zoom = 1 / self.settings_dict.get("zoom_increase", 2)
|
||||
else:
|
||||
return
|
||||
|
||||
self.zoom *= zoom
|
||||
|
||||
self.zoom_label.text = f"Zoom: {self.zoom}"
|
||||
|
||||
self.zoom_at(self.window.mouse.data["x"], self.window.mouse.data["y"], zoom)
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Mandelbrot', details=f'Zoom: {self.zoom}\nMax Iterations: {self.max_iter}', start=self.pypresence_client.start_time)
|
||||
|
||||
def on_draw(self):
|
||||
self.window.clear()
|
||||
self.mandelbrot_sprite.draw()
|
||||
self.ui.draw()
|
||||
def sierpinsky_carpet(self):
|
||||
from game.sierpinsky_carpet import SierpinskyCarpetViewer
|
||||
self.window.show_view(SierpinskyCarpetViewer(self.pypresence_client))
|
||||
|
||||
@@ -50,6 +50,43 @@ void main() {
|
||||
}
|
||||
"""
|
||||
|
||||
sierpinsky_carpet_compute_source = """#version 430 core
|
||||
|
||||
uniform int u_depth;
|
||||
uniform int u_zoom;
|
||||
layout (local_size_x = 1, local_size_y = 1, local_size_z = 1) in;
|
||||
layout(location = 0, rgba32f) uniform image2D img_output;
|
||||
|
||||
void main() {
|
||||
ivec2 coord = ivec2(gl_GlobalInvocationID.xy);
|
||||
bool isHole = false;
|
||||
|
||||
for (int i = 0; i < u_depth; ++i) {
|
||||
if (coord.x % 3 == 1 && coord.y % 3 == 1) {
|
||||
isHole = true;
|
||||
break;
|
||||
}
|
||||
coord /= 3;
|
||||
}
|
||||
|
||||
vec4 color = isHole ? vec4(0, 0, 0, 1) : vec4(1, 1, 1, 1);
|
||||
imageStore(img_output, ivec2(gl_GlobalInvocationID.xy), color);
|
||||
}
|
||||
|
||||
"""
|
||||
|
||||
def create_sierpinsky_carpet_shader(width, height, precision="single"):
|
||||
shader_source = sierpinsky_carpet_compute_source
|
||||
|
||||
shader_program = pyglet.graphics.shader.ComputeShaderProgram(shader_source)
|
||||
|
||||
sierpinsky_carpet_image = pyglet.image.Texture.create(width, height, internalformat=pyglet.gl.GL_RGBA32F)
|
||||
|
||||
uniform_location = shader_program['img_output']
|
||||
sierpinsky_carpet_image.bind_image_texture(unit=uniform_location)
|
||||
|
||||
return shader_program, sierpinsky_carpet_image
|
||||
|
||||
def create_mandelbrot_shader(width, height, precision="single"):
|
||||
shader_source = mandelbrot_compute_source
|
||||
|
||||
|
||||
81
game/sierpinsky_carpet.py
Normal file
81
game/sierpinsky_carpet.py
Normal file
@@ -0,0 +1,81 @@
|
||||
import arcade, arcade.gui, pyglet, json
|
||||
|
||||
from PIL import Image
|
||||
|
||||
from game.shader import create_sierpinsky_carpet_shader
|
||||
from utils.constants import menu_background_color, button_style
|
||||
from utils.preload import button_texture, button_hovered_texture
|
||||
|
||||
class SierpinskyCarpetViewer(arcade.gui.UIView):
|
||||
def __init__(self, pypresence_client):
|
||||
super().__init__()
|
||||
|
||||
self.pypresence_client = pypresence_client
|
||||
|
||||
with open("settings.json", "r") as file:
|
||||
self.settings_dict = json.load(file)
|
||||
|
||||
self.depth = self.settings_dict.get("sierpinsky_depth", 10)
|
||||
self.zoom = 1.0
|
||||
self.click_center = (self.width / 2, self.height / 2)
|
||||
|
||||
def on_show_view(self):
|
||||
super().on_show_view()
|
||||
|
||||
self.shader_program, self.sierpinsky_carpet_image = create_sierpinsky_carpet_shader(self.window.width, self.window.height, self.settings_dict.get("precision", "Single").lower())
|
||||
|
||||
self.sierpinsky_carpet_sprite = pyglet.sprite.Sprite(img=self.sierpinsky_carpet_image)
|
||||
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Sierpinsky Carpet', details=f'Zoom: {self.zoom}\nDepth: {self.depth}', 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="Protest Strike", font_size=16))
|
||||
self.depth_label = self.info_box.add(arcade.gui.UILabel(text=f"Depth: {self.depth}", font_name="Protest Strike", 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.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)
|
||||
|
||||
def create_image(self):
|
||||
with self.shader_program:
|
||||
self.shader_program['u_depth'] = self.depth
|
||||
#self.shader_program['u_zoom'] = int(self.zoom)
|
||||
#self.shader_program['u_resolution'] = self.window.size
|
||||
#self.shader_program['u_center'] = self.click_center
|
||||
self.shader_program.dispatch(self.sierpinsky_carpet_image.width, self.sierpinsky_carpet_image.height, 1, barrier=pyglet.gl.GL_ALL_BARRIER_BITS)
|
||||
|
||||
def on_mouse_press(self, x: int, y: int, button: int, modifiers: int) -> bool | None:
|
||||
super().on_mouse_press(x, y, button, modifiers)
|
||||
|
||||
if button == arcade.MOUSE_BUTTON_LEFT:
|
||||
zoom = self.settings_dict.get("sierpinsky_zoom_increase", 2)
|
||||
elif button == arcade.MOUSE_BUTTON_RIGHT:
|
||||
zoom = 1 / self.settings_dict.get("sierpinsky_zoom_increase", 2)
|
||||
else:
|
||||
return
|
||||
|
||||
self.zoom *= zoom
|
||||
|
||||
self.click_center = (x, y)
|
||||
|
||||
self.zoom_label.text = f"Zoom: {self.zoom}"
|
||||
|
||||
self.create_image()
|
||||
|
||||
self.pypresence_client.update(state='Viewing Sierpinsky Carpet', details=f'Zoom: {self.zoom}\nMax Iterations: {self.depth}', start=self.pypresence_client.start_time)
|
||||
|
||||
def on_draw(self):
|
||||
self.window.clear()
|
||||
self.sierpinsky_carpet_sprite.draw()
|
||||
self.ui.draw()
|
||||
Reference in New Issue
Block a user