123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238 |
- #ifndef __SCREEN_SPACE_RAYTRACE__
- #define __SCREEN_SPACE_RAYTRACE__
- sampler2D_float _CameraDepthTexture;
- float distanceSquared(float2 A, float2 B)
- {
- A -= B;
- return dot(A, A);
- }
- float distanceSquared(float3 A, float3 B)
- {
- A -= B;
- return dot(A, A);
- }
- void swap(inout float v0, inout float v1)
- {
- float temp = v0;
- v0 = v1;
- v1 = temp;
- }
- bool isIntersecting(float rayZMin, float rayZMax, float sceneZ, float layerThickness)
- {
- return (rayZMax >= sceneZ - layerThickness) && (rayZMin <= sceneZ);
- }
- void rayIterations(in bool traceBehindObjects, inout float2 P, inout float stepDirection, inout float end, inout int stepCount, inout int maxSteps, inout bool intersecting,
- inout float sceneZ, inout float2 dP, inout float3 Q, inout float3 dQ, inout float k, inout float dk,
- inout float rayZMin, inout float rayZMax, inout float prevZMaxEstimate, inout bool permute, inout float2 hitPixel,
- inout float2 invSize, inout float layerThickness)
- {
- bool stop = intersecting;
- UNITY_LOOP
- for (; (P.x * stepDirection) <= end && stepCount < maxSteps && !stop; P += dP, Q.z += dQ.z, k += dk, stepCount += 1)
- {
-
-
- rayZMin = prevZMaxEstimate;
-
-
- rayZMax = (dQ.z * 0.5 + Q.z) / (dk * 0.5 + k);
- prevZMaxEstimate = rayZMax;
- if (rayZMin > rayZMax)
- {
- swap(rayZMin, rayZMax);
- }
-
-
- hitPixel = permute ? P.yx : P;
- sceneZ = tex2Dlod(_CameraDepthTexture, float4(hitPixel * invSize,0,0)).r;
- sceneZ = -LinearEyeDepth(sceneZ);
- bool isBehind = (rayZMin <= sceneZ);
- intersecting = isBehind && (rayZMax >= sceneZ - layerThickness);
- stop = traceBehindObjects ? intersecting : isBehind;
- }
- P -= dP, Q.z -= dQ.z, k -= dk;
- }
- bool castDenseScreenSpaceRay
- (float3 csOrigin,
- float3 csDirection,
- float4x4 projectToPixelMatrix,
- float2 csZBufferSize,
- float3 clipInfo,
- float jitterFraction,
- int maxSteps,
- float layerThickness,
- float maxRayTraceDistance,
- out float2 hitPixel,
- int stepRate,
- bool traceBehindObjects,
- out float3 csHitPoint,
- out float stepCount) {
- float2 invSize = float2(1.0 / csZBufferSize.x, 1.0 / csZBufferSize.y);
-
- hitPixel = float2(-1, -1);
- float nearPlaneZ = -0.01;
-
- float rayLength = ((csOrigin.z + csDirection.z * maxRayTraceDistance) > nearPlaneZ) ?
- ((nearPlaneZ - csOrigin.z) / csDirection.z) :
- maxRayTraceDistance;
- float3 csEndPoint = csDirection * rayLength + csOrigin;
-
-
-
-
- float4 H0 = mul(projectToPixelMatrix, float4(csOrigin, 1.0));
- float4 H1 = mul(projectToPixelMatrix, float4(csEndPoint, 1.0));
-
-
-
-
-
-
-
- float k0 = 1.0 / H0.w;
- float k1 = 1.0 / H1.w;
-
- float2 P0 = H0.xy * k0;
- float2 P1 = H1.xy * k1;
-
- float3 Q0 = csOrigin * k0;
- float3 Q1 = csEndPoint * k1;
- #if 1 // Clipping to the screen coordinates. We could simply modify maxSteps instead
- float yMax = csZBufferSize.y - 0.5;
- float yMin = 0.5;
- float xMax = csZBufferSize.x - 0.5;
- float xMin = 0.5;
-
- float alpha = 0.0;
-
- if (P1.y > yMax || P1.y < yMin) {
- float yClip = (P1.y > yMax) ? yMax : yMin;
- float yAlpha = (P1.y - yClip) / (P1.y - P0.y);
- alpha = yAlpha;
- }
-
- if (P1.x > xMax || P1.x < xMin) {
- float xClip = (P1.x > xMax) ? xMax : xMin;
- float xAlpha = (P1.x - xClip) / (P1.x - P0.x);
- alpha = max(alpha, xAlpha);
- }
-
- P1 = lerp(P1, P0, alpha);
- k1 = lerp(k1, k0, alpha);
- Q1 = lerp(Q1, Q0, alpha);
- #endif
-
- P1 = (distanceSquared(P0, P1) < 0.0001) ? P0 + float2(0.01, 0.01) : P1;
- float2 delta = P1 - P0;
-
- bool permute = false;
- if (abs(delta.x) < abs(delta.y)) {
-
- permute = true;
-
- delta = delta.yx;
- P1 = P1.yx;
- P0 = P0.yx;
- }
-
- float stepDirection = sign(delta.x);
- float invdx = stepDirection / delta.x;
- float2 dP = float2(stepDirection, invdx * delta.y);
-
- float3 dQ = (Q1 - Q0) * invdx;
- float dk = (k1 - k0) * invdx;
- dP *= stepRate;
- dQ *= stepRate;
- dk *= stepRate;
- P0 += dP * jitterFraction;
- Q0 += dQ * jitterFraction;
- k0 += dk * jitterFraction;
-
- float3 Q = Q0;
- float k = k0;
-
-
-
-
- float prevZMaxEstimate = csOrigin.z;
- stepCount = 0.0;
- float rayZMax = prevZMaxEstimate, rayZMin = prevZMaxEstimate;
- float sceneZ = 100000;
-
-
- float end = P1.x * stepDirection;
- bool intersecting = isIntersecting(rayZMin, rayZMax, sceneZ, layerThickness);
-
-
-
- float2 P = P0;
- int originalStepCount = 0;
- rayIterations(traceBehindObjects, P, stepDirection, end, originalStepCount, maxSteps, intersecting,
- sceneZ, dP, Q, dQ, k, dk,
- rayZMin, rayZMax, prevZMaxEstimate, permute, hitPixel,
- invSize, layerThickness);
- stepCount = originalStepCount;
-
-
- Q.xy += dQ.xy * stepCount;
-
- csHitPoint = Q * (1.0 / k);
- return intersecting;
- }
- #endif // __SCREEN_SPACE_RAYTRACE__
|