initial commit

3 files changed, 127 insertions, 0 deletions

diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..c956781
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,12 @@
+PROG=brotshader
+CC=gcc
+FLAGS=-lm `pkg-config -libs raylib`
+
+.DEFAULT_GOAL := run
+
+${PROG}: ${PROG}.c
+	${CC} ${PROG}.c -o ${PROG} ${FLAGS}
+
+run: ${PROG}
+	./${PROG} 
+
diff --git a/brotshader.c b/brotshader.c
new file mode 100644
index 0000000..d0fd96b
--- /dev/null
+++ b/brotshader.c
@@ -0,0 +1,70 @@
+#include <raylib.h>
+
+const int width = 1280;
+const int height = 1280;
+
+int main() {
+	InitWindow(width, height, "brotshader");
+	
+	Shader shader = LoadShader(0, "mandelbrot.glsl");
+	
+	RenderTexture2D target = LoadRenderTexture(width, height);
+
+	float c[2] = {0, 0};
+
+	float offset[2] = {0, 0};
+	float zoom = 2.0f;
+	int max = 255;
+
+	int resolutionLoc    = GetShaderLocation(shader, "resolution");
+	int locationLoc      = GetShaderLocation(shader, "location");
+	int zoomLoc          = GetShaderLocation(shader, "zoom");
+	int maxLoc           = GetShaderLocation(shader, "max");
+	
+	float screen[2] = {(float)width, (float)height};
+
+	SetShaderValue(shader, resolutionLoc, screen, UNIFORM_VEC2);
+	SetShaderValue(shader, locationLoc, &offset, UNIFORM_VEC2);
+	SetShaderValue(shader, zoomLoc, &zoom, UNIFORM_FLOAT);
+	SetShaderValue(shader, maxLoc, &max, UNIFORM_INT);
+
+	SetTargetFPS(60);
+
+	while (!WindowShouldClose()) {
+	
+		if (IsKeyDown(KEY_UP)) offset[1] -= zoom * 0.01f;
+		if (IsKeyDown(KEY_DOWN)) offset[1] += zoom * 0.01f;
+		if (IsKeyDown(KEY_RIGHT)) offset[0] -= zoom * 0.01f;
+		if (IsKeyDown(KEY_LEFT)) offset[0] += zoom * 0.01f;
+
+		if (IsKeyDown(KEY_W)) zoom -= zoom * 0.01f;
+		if (IsKeyDown(KEY_S)) zoom += zoom * 0.01f;
+		if (IsKeyDown(KEY_A)) max -= 1;
+		if (IsKeyDown(KEY_D)) max += 1;
+
+		SetShaderValue(shader, resolutionLoc, screen, UNIFORM_VEC2);
+		SetShaderValue(shader, locationLoc, &offset, UNIFORM_VEC2);
+		SetShaderValue(shader, zoomLoc, &zoom, UNIFORM_FLOAT);
+		SetShaderValue(shader, maxLoc, &max, UNIFORM_INT);
+
+		BeginDrawing();
+
+		ClearBackground(RAYWHITE);
+
+		BeginTextureMode(target);
+		DrawRectangle(0, 0, width, height, BLACK);
+		EndTextureMode();
+
+		BeginShaderMode(shader);
+		DrawTexture(target.texture, 0, 0, WHITE);
+		EndShaderMode();
+		
+		EndDrawing();
+	}
+
+	UnloadShader(shader);
+	UnloadRenderTexture(target);
+
+	CloseWindow();
+}
+
diff --git a/mandelbrot.glsl b/mandelbrot.glsl
new file mode 100644
index 0000000..d3d0ee2
--- /dev/null
+++ b/mandelbrot.glsl
@@ -0,0 +1,45 @@
+#version 100
+precision highp float;
+
+uniform vec2 resolution;
+uniform vec2 location;
+uniform double zoom;
+
+uniform int max;
+
+vec2 compsquare(vec2 z) {
+    float temp = z.x;
+    z.x = z.x * z.x - z.y * z.y;
+    z.y = 2.0 * temp * z.y;
+    return z;
+}
+
+vec3 hsv2rgb(vec3 hue){
+    vec4 K = vec4(1.0, 2.0/3.0, 1.0/3.0, 3.0);
+    vec3 p = abs(fract(hue.xxx + K.xyz) * 6.0 - K.www);
+    return hue.z * mix(K.xxx, clamp(p - K.xxx, 0.0, 1.0), hue.y);
+}
+
+vec3 mandelbort(vec2 point){
+    vec2 z = vec2(0.0);
+    float iters = 0.0;
+    for (iters = 0.0;   iters < float(max); ++iters)
+    {
+        z = compsquare(z) + point;
+        if(dot(z, z) > 4.0) break;
+    }
+    float hue = iters / float(max);
+    if(hue > 0.98) return vec3(0.0);
+    return hsv2rgb(vec3(hue, 1.0, 1.0));
+}
+
+void main()
+{
+    vec2 uv = gl_FragCoord.xy / resolution;
+    uv.x *= resolution.x / resolution.y;
+    uv -= vec2(0.7, 0.5);
+    uv *= zoom;
+    uv -= location;
+
+    gl_FragColor = vec4(mandelbort(uv), 1.0);
+}