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

separate normal and multi julia and only use double for normal

Browse Source
This commit is contained in:
csd4ni3l
2025-05-26 20:49:12 +02:00
parent 675af0d5ca
commit 564ce565a5
1 changed files with 34 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
game/shader.py
View File

@@ -80,8 +80,7 @@ void main() {
}
"""
julia_compute_source = """#version 430 core
julia_template = """#version 430 core
uniform int u_maxIter;
uniform vec2 u_resolution;
@@ -108,13 +107,7 @@ void main() {
int iters = 0;
while ((z.x * z.x + z.y * z.y) < pow(R, 2) && iters < u_maxIter) {
{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;
}
{julia_calc}
vec4 value = vec4(0.0, 0.0, 0.0, 1.0);
@@ -130,6 +123,25 @@ void main() {
}
"""
normal_julia_calc = """while (z.x * z.x + z.y * z.y < R*R && 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;
iters = iters + 1;
}
"""
multi_julia_calc = """while ((z.x * z.x + z.y * z.y) < R*R && iters < u_maxIter) {
{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;
}
"""
def create_sierpinsky_carpet_shader(width, height, precision="single"):
shader_source = sierpinsky_carpet_compute_source
@@ -150,20 +162,26 @@ 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", julia_n=2):
shader_source = julia_compute_source
shader_source = julia_template
if julia_n == 2:
shader_source = shader_source.replace("{julia_calc}", normal_julia_calc)
if precision == "single":
shader_source = shader_source.replace("{vec2type}", "vec2").replace("{floattype}", "float")
elif precision == "double":
shader_source = shader_source.replace("{vec2type}", "dvec2").replace("{floattype}", "double")
else:
raise TypeError("Invalid Precision")
shader_source = shader_source.replace("{julia_calc}", multi_julia_calc)
shader_source = shader_source.replace("{vec2type}", "vec2").replace("{floattype}", "float") # pow and atan only support floats
shader_source = shader_source.replace("{julia_n}", str(julia_n))
julia_c = c_for_julia_type[julia_type]
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)