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 julia N, slider steps

Browse Source
This commit is contained in:
csd4ni3l
2025-05-26 16:43:52 +02:00
parent 65b271d1a8
commit 675af0d5ca
6 changed files with 27 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
game/julia.py
View File

@@ -37,7 +37,7 @@ class JuliaViewer(arcade.gui.UIView):
def on_show_view(self):
super().on_show_view()
self.shader_program, self.julia_image = create_julia_shader(self.window.width, self.window.height, self.settings_dict.get("julia_precision", "Single").lower(), self.settings_dict.get("julia_escape_radius", 2), self.settings_dict.get("julia_type", "Classic swirling"))
self.shader_program, self.julia_image = create_julia_shader(self.window.width, self.window.height, self.settings_dict.get("julia_precision", "Single").lower(), self.settings_dict.get("julia_escape_radius", 2), self.settings_dict.get("julia_type", "Classic swirling"), int(self.settings_dict.get("julia_n", 2)))
self.julia_sprite = pyglet.sprite.Sprite(img=self.julia_image)
@@ -64,7 +64,7 @@ class JuliaViewer(arcade.gui.UIView):
def create_image(self):
with self.shader_program:
self.shader_program['u_maxIter'] = self.max_iter
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)

2
game/mandelbrot.py
View File

@@ -64,7 +64,7 @@ class MandelbrotViewer(arcade.gui.UIView):
def create_image(self):
with self.shader_program:
self.shader_program['u_maxIter'] = self.max_iter
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)

16
game/shader.py
View File

@@ -40,8 +40,6 @@ void main() {
if (iters != u_maxIter) {
float t = float(iters) / float(u_maxIter);
float pow_amount = 0.7;
t = pow(t, pow_amount);
value.r = 9.0 * (1.0 - t) * t * t * t;
value.g = 15.0 * (1.0 - t) * (1.0 - t) * t * t;
@@ -102,7 +100,8 @@ layout(location = 0, rgba32f) uniform image2D img_output;
void main() {
ivec2 texel_coord = ivec2(gl_GlobalInvocationID.xy);
int R = {escape_radius};
float R = {escape_radius};
int n = {julia_n};
{vec2type} c = {vec2type}{julia_c};
{vec2type} z = map_pixel({floattype}(texel_coord.x), {floattype}(texel_coord.y), u_resolution, u_real_range, u_imag_range);
@@ -110,9 +109,9 @@ void main() {
int iters = 0;
while ((z.x * z.x + z.y * z.y) < pow(R, 2) && iters < u_maxIter) {
{floattype} xtemp = z.x * z.x - z.y * z.y;
z.y = 2 * z.x * z.y + c.y;
z.x = xtemp + c.x;
{floattype} xtmp = pow((z.x * z.x + z.y * z.y), (n / 2)) * cos(n * atan(z.y, z.x)) + c.x;
z.y = pow((z.x * z.x + z.y * z.y), (n / 2)) * sin(n * atan(z.y, z.x)) + c.y;
z.x = xtmp;
iters = iters + 1;
}
@@ -121,8 +120,6 @@ void main() {
if (iters != u_maxIter) {
float t = float(iters) / float(u_maxIter);
float pow_amount = 0.7;
t = pow(t, pow_amount);
value.r = 9.0 * (1.0 - t) * t * t * t;
value.g = 15.0 * (1.0 - t) * (1.0 - t) * t * t;
@@ -152,7 +149,7 @@ def create_sierpinsky_carpet_shader(width, height, precision="single"):
return shader_program, sierpinsky_carpet_image
def create_julia_shader(width, height, precision="single", escape_radius=2, julia_type="Classic swirling"):
def create_julia_shader(width, height, precision="single", escape_radius=2, julia_type="Classic swirling", julia_n=2):
shader_source = julia_compute_source
if precision == "single":
@@ -166,6 +163,7 @@ def create_julia_shader(width, height, precision="single", escape_radius=2, juli
shader_source = shader_source.replace("{julia_c}", str(julia_c))
shader_source = shader_source.replace("{escape_radius}", str(escape_radius))
shader_source = shader_source.replace("{julia_n}", str(julia_n))
shader_program = pyglet.graphics.shader.ComputeShaderProgram(shader_source)

2
game/sierpinsky_carpet.py
View File

@@ -49,7 +49,7 @@ class SierpinskyCarpetViewer(arcade.gui.UIView):
def create_image(self):
with self.shader_program:
self.shader_program['u_depth'] = self.depth
self.shader_program['u_depth'] = int(self.depth)
self.shader_program['u_zoom'] = int(self.zoom)
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)