157 lines
3.0 KiB
GLSL

#version 450 core
#define MAX_STEPS 64
const vec2 eps = vec2(0.001, 0.0);
layout (set = 0, binding = 1) uniform sampler2D tex;
layout (set = 0, binding = 0) uniform Matrices {
vec3 cam_pos;
float time;
vec4 viewport;
vec3 cam_dir;
uint n_objects;
};
/*
enum Shape {
SPHERE,
BOX,
PLANE,
};
*/
#define SPHERE 0
#define BOX 1
#define PLANE 2
struct Object {
vec4 center;
vec4 dimensions;
uint id;
uint shape;
};
layout (set = 0, binding = 2) readonly buffer Objects {
Object objects[];
};
layout (location = 0) in vec2 pos;
layout (location = 0) out vec4 fragColor;
/* joins two parts of a scene */
vec2 op_union(vec2 v1, float v2, float id) {
if(v1.x > v2)
return vec2(v2, id);
return v1;
}
/* subtracts sdf from scene */
float op_subtract(float v1, float v2) {
return max(v1, -v2);
}
float sphere(vec3 p, vec3 c, float r) {
return abs(length(p-c) - r);
}
float box(vec3 p, vec3 c, vec3 r) {
p -= c;
vec3 q = abs(p) - r;
return length(max(q,0.0)) + min(max(q.x, max(q.y,q.z)), 0.0);
}
float plane(vec3 p, vec3 norm, float d) {
return dot(p, normalize(norm)) + d;
}
float obj_to_sdf(vec3 p, uint n) {
switch(objects[n].shape) {
case SPHERE:
return sphere(p, objects[n].center.xyz, objects[n].dimensions.x);
break;
case BOX:
return box(p, objects[n].center.xyz, objects[n].dimensions.xyz);
break;
case PLANE:
return plane(p, objects[n].center.xyz, objects[n].dimensions.x);
break;
}
}
float map(vec2 p) {
return sin(p.x)*sin(p.y);
}
float terrain(vec3 p) {
/* render terrain with bumps */
return box(p, vec3(0.0), vec3(100.0, 0.1, 100.0)) - abs(map(p.xz));
}
/* <distance, id> */
vec2 sdf(vec3 pos) {
vec2 d = vec2(100000000.0, -1.0);
for(uint i = 0; i < n_objects; i++) {
d = op_union(d, obj_to_sdf(pos, i), 1.0);
// d = op_union(d, sphere(pos, objects[i].center.xyz, objects[i].dimensions.x), 1.0);
}
/*float dsphere = sphere(pos, vec3(0.0, 3.0*sin(time)-10.0, 0.0), 1.0);
float dbox = box(pos, vec3(0.0, -10.0, 0.0), vec3(10.0, 1.0, 10.0));
float dpellet = sphere(pos, vec3(0.0, 10.0*sin(time) - 10.0, 0.0), 0.1);
float d = op_union(op_subtract(dbox, dsphere), dpellet);*/
//d = op_union(d, plane(pos, normalize(vec3(1.0)), 50.0));
return d;
}
vec2 raycast(vec3 dir) {
float t = 0.0;
for(int i = 0; i < MAX_STEPS; i++) {
vec2 dt = sdf(cam_pos + dir * t);
if(dt.y == -1.0)
return dt;
if(dt.x < 0.0001*t)
// return vec2(t, dt.y);
return vec2(t, float(i)/MAX_STEPS);
else if(dt.x > 2000.0)
return vec2(dt.x, -1.0);
t += dt.x;
}
return vec2(0.0, -1.0);
}
vec3 norm(vec3 pos) {
return normalize(
vec3(
sdf(pos+eps.xyy).x,
sdf(pos+eps.yxy).x,
sdf(pos+eps.yyx).x
) - sdf(pos).x
);
}
vec3 raygen() {
vec3 forward = cam_dir;
vec3 right = normalize(cross(forward, vec3(0.0, 1.0, 0.0)));
vec3 up = normalize(cross(right, forward));
return normalize(pos.x * right + pos.y * up + forward * 2.0);
}
void main() {
vec3 dir = raygen();
vec2 d = raycast(dir);
vec3 p = d.x*dir + cam_pos;
if(d.y != -1.0)
fragColor = vec4(d.y);
else
fragColor = vec4(0.0);
}