vkdoom_m/wadsrc/static/shaders/lightmap/trace_light.glsl
2024-06-15 18:08:37 +08:00

92 lines
2.7 KiB
GLSL

#include <shaders/lightmap/montecarlo.glsl>
vec3 TracePointLightRay(vec3 origin, vec3 lightpos, float tmin, vec3 rayColor);
vec3 TraceLight(vec3 origin, vec3 normal, LightInfo light, float extraDistance)
{
const float minDistance = 0.01;
vec3 incoming = vec3(0.0);
float dist = distance(light.RelativeOrigin, origin) + extraDistance;
if (dist > minDistance && dist < light.Radius)
{
vec3 dir = normalize(light.RelativeOrigin - origin);
float distAttenuation = max(1.0 - (dist / light.Radius), 0.0);
float angleAttenuation = max(dot(normal, dir), 0.0);
float spotAttenuation = 1.0;
if (light.OuterAngleCos > -1.0)
{
float cosDir = dot(dir, light.SpotDir);
spotAttenuation = smoothstep(light.OuterAngleCos, light.InnerAngleCos, cosDir);
spotAttenuation = max(spotAttenuation, 0.0);
}
float attenuation = distAttenuation * angleAttenuation * spotAttenuation;
if (attenuation > 0.0)
{
vec3 rayColor = light.Color.rgb * (attenuation * light.Intensity);
#if defined(USE_SOFTSHADOWS)
if (light.SoftShadowRadius != 0.0)
{
vec3 v = (abs(dir.x) > abs(dir.y)) ? vec3(0.0, 1.0, 0.0) : vec3(1.0, 0.0, 0.0);
vec3 xdir = normalize(cross(dir, v));
vec3 ydir = cross(dir, xdir);
float lightsize = light.SoftShadowRadius;
int step_count = 10;
for (int i = 0; i < step_count; i++)
{
vec2 gridoffset = getVogelDiskSample(i, step_count, gl_FragCoord.x + gl_FragCoord.y * 13.37) * lightsize;
vec3 pos = light.Origin + xdir * gridoffset.x + ydir * gridoffset.y;
incoming += TracePointLightRay(origin, pos, minDistance, rayColor) / float(step_count);
}
}
else
{
incoming += TracePointLightRay(origin, light.Origin, minDistance, rayColor);
}
#else
incoming += TracePointLightRay(origin, light.Origin, minDistance, rayColor);
#endif
}
}
return incoming;
}
vec3 TracePointLightRay(vec3 origin, vec3 lightpos, float tmin, vec3 rayColor)
{
vec3 dir = normalize(lightpos - origin);
float tmax = distance(origin, lightpos);
for (int i = 0; i < 3; i++)
{
TraceResult result = TraceFirstHit(origin, tmin, dir, tmax);
// Stop if we hit nothing - the point light is visible.
if (result.primitiveIndex == -1)
return rayColor;
SurfaceInfo surface = GetSurface(result.primitiveIndex);
// Pass through surface texture
rayColor = PassRayThroughSurface(surface, GetSurfaceUV(result.primitiveIndex, result.primitiveWeights), rayColor);
// Stop if there is no light left
if (rayColor.r + rayColor.g + rayColor.b <= 0.0)
return vec3(0.0);
// Move to surface hit point
origin += dir * result.t;
tmax -= result.t;
// Move through the portal, if any
TransformRay(surface.PortalIndex, origin, dir);
}
return vec3(0.0);
}