382 lines
12 KiB
C++
382 lines
12 KiB
C++
// $Id$
|
|
|
|
#include "raytrace.h"
|
|
#include <mathlib/halton.h>
|
|
|
|
// NOTE: This has to be the last file included!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
|
|
static uint32 MapDistanceToPixel(float t)
|
|
{
|
|
if (t<0) return 0xffff0000;
|
|
if (t>100) return 0xff000000;
|
|
int a=t*1000; a&=0xff;
|
|
int b=t*10; b &=0xff;
|
|
int c=t*.01; c &=0xff;
|
|
return 0xff000000+(a<<16)+(b<<8)+c;
|
|
}
|
|
|
|
#define IGAMMA (1.0/2.2)
|
|
|
|
#define MAGIC_NUMBER (1<<23)
|
|
|
|
static fltx4 Four_MagicNumbers={ MAGIC_NUMBER, MAGIC_NUMBER, MAGIC_NUMBER, MAGIC_NUMBER };
|
|
static ALIGN16 int32 Four_255s[4]= {0xff,0xff,0xff,0xff};
|
|
#define PIXMASK ( * ( reinterpret_cast< fltx4 *>( &Four_255s ) ) )
|
|
|
|
void MapLinearIntensities(FourVectors const &intens,uint32 *p1, uint32 *p2, uint32 *p3, uint32 *p4)
|
|
{
|
|
// convert four pixels worth of sse-style rgb into argb lwords
|
|
// NOTE the _mm_empty macro is voodoo. do not mess with this routine casually - simply throwing
|
|
// anything that ends up generating a fpu stack references in here would be bad news.
|
|
static fltx4 pixscale={255.0,255.0,255.0,255.0};
|
|
fltx4 r,g,b;
|
|
r=MinSIMD(pixscale,MulSIMD(pixscale,PowSIMD(intens.x,IGAMMA)));
|
|
g=MinSIMD(pixscale,MulSIMD(pixscale,PowSIMD(intens.y,IGAMMA)));
|
|
b=MinSIMD(pixscale,MulSIMD(pixscale,PowSIMD(intens.z,IGAMMA)));
|
|
// now, convert to integer
|
|
r=AndSIMD( AddSIMD( r, Four_MagicNumbers ), PIXMASK );
|
|
g=AndSIMD( AddSIMD( g, Four_MagicNumbers ), PIXMASK );
|
|
b=AndSIMD( AddSIMD( b, Four_MagicNumbers ), PIXMASK );
|
|
|
|
*(p1)=(SubInt(r, 0))|(SubInt(g, 0)<<8)|(SubInt(b, 0)<<16);
|
|
*(p2)=(SubInt(r, 1))|(SubInt(g, 1)<<8)|(SubInt(b, 1)<<16);
|
|
*(p3)=(SubInt(r, 2))|(SubInt(g, 2)<<8)|(SubInt(b, 2)<<16);
|
|
*(p4)=(SubInt(r, 3))|(SubInt(g, 3)<<8)|(SubInt(b, 3)<<16);
|
|
}
|
|
|
|
static ALIGN16 int32 signmask[4]={0x80000000,0x80000000,0x80000000,0x80000000};
|
|
static ALIGN16 int32 all_ones[4]={-1,-1,-1,-1};
|
|
static fltx4 all_zeros={0,0,0,0};
|
|
static fltx4 TraceLimit={1.0e20,1.0e20,1.0e20,1.0e20};
|
|
|
|
void RayTracingEnvironment::RenderScene(
|
|
int width, int height, // width and height of desired rendering
|
|
int stride, // actual width in pixels of target buffer
|
|
uint32 *output_buffer, // pointer to destination
|
|
Vector CameraOrigin, // eye position
|
|
Vector ULCorner, // word space coordinates of upper left
|
|
// monitor corner
|
|
Vector URCorner, // top right corner
|
|
Vector LLCorner, // lower left
|
|
Vector LRCorner, // lower right
|
|
RayTraceLightingMode_t lmode)
|
|
{
|
|
// first, compute deltas
|
|
Vector dxvector=URCorner;
|
|
dxvector-=ULCorner;
|
|
dxvector*=(1.0/width);
|
|
Vector dxvectortimes2=dxvector;
|
|
dxvectortimes2+=dxvector;
|
|
|
|
Vector dyvector=LLCorner;
|
|
dyvector-=ULCorner;
|
|
dyvector*=(1.0/height);
|
|
|
|
|
|
// block_offsets-relative offsets for eahc of the 4 pixels in the block, in sse format
|
|
FourVectors block_offsets;
|
|
block_offsets.LoadAndSwizzle(Vector(0,0,0),dxvector,dyvector,dxvector+dyvector);
|
|
|
|
FourRays myrays;
|
|
myrays.origin.DuplicateVector(CameraOrigin);
|
|
|
|
// tmprays is used fo rthe case when we cannot trace 4 rays at once.
|
|
FourRays tmprays;
|
|
tmprays.origin.DuplicateVector(CameraOrigin);
|
|
|
|
// now, we will ray trace pixels. we will do the rays in a 2x2 pattern
|
|
for(int y=0;y<height;y+=2)
|
|
{
|
|
Vector SLoc=dyvector;
|
|
SLoc*=((float) y);
|
|
SLoc+=ULCorner;
|
|
uint32 *dest=output_buffer+y*stride;
|
|
for(int x=0;x<width;x+=2)
|
|
{
|
|
myrays.direction.DuplicateVector(SLoc);
|
|
myrays.direction+=block_offsets;
|
|
myrays.direction.VectorNormalize();
|
|
|
|
RayTracingResult rslt;
|
|
Trace4Rays(myrays,all_zeros,TraceLimit, &rslt);
|
|
if ((rslt.HitIds[0]==-1) && (rslt.HitIds[1]==-1) &&
|
|
(rslt.HitIds[2]==-1) && (rslt.HitIds[3]==-1))
|
|
MapLinearIntensities(BackgroundColor,dest,dest+1,dest+stride,dest+stride+1);
|
|
else
|
|
{
|
|
// make sure normal points back towards ray origin
|
|
fltx4 ndoti=rslt.surface_normal*myrays.direction;
|
|
fltx4 bad_dirs=AndSIMD(CmpGtSIMD(ndoti,Four_Zeros),
|
|
LoadAlignedSIMD((float *) signmask));
|
|
|
|
// flip signs of all "wrong" normals
|
|
rslt.surface_normal.x=XorSIMD(bad_dirs,rslt.surface_normal.x);
|
|
rslt.surface_normal.y=XorSIMD(bad_dirs,rslt.surface_normal.y);
|
|
rslt.surface_normal.z=XorSIMD(bad_dirs,rslt.surface_normal.z);
|
|
|
|
FourVectors intens;
|
|
intens.DuplicateVector(Vector(0,0,0));
|
|
// set up colors
|
|
FourVectors surf_colors;
|
|
surf_colors.DuplicateVector(Vector(0,0,0));
|
|
for(int i=0;i<4;i++)
|
|
{
|
|
if (rslt.HitIds[i]>=0)
|
|
{
|
|
surf_colors.X(i)=TriangleColors[rslt.HitIds[i]].x;
|
|
surf_colors.Y(i)=TriangleColors[rslt.HitIds[i]].y;
|
|
surf_colors.Z(i)=TriangleColors[rslt.HitIds[i]].z;
|
|
}
|
|
|
|
}
|
|
FourVectors surface_pos=myrays.direction;
|
|
surface_pos*=rslt.HitDistance;
|
|
surface_pos+=myrays.origin;
|
|
|
|
switch(lmode)
|
|
{
|
|
case DIRECT_LIGHTING:
|
|
{
|
|
// light all points
|
|
for(int l=0;l<LightList.Count();l++)
|
|
{
|
|
LightList[l].ComputeLightAtPoints(surface_pos,rslt.surface_normal,
|
|
intens);
|
|
}
|
|
}
|
|
break;
|
|
|
|
case DIRECT_LIGHTING_WITH_SHADOWS:
|
|
{
|
|
// light all points
|
|
for(int l=0;l<LightList.Count();l++)
|
|
{
|
|
FourVectors ldir;
|
|
ldir.DuplicateVector(LightList[l].m_Position);
|
|
ldir-=surface_pos;
|
|
fltx4 MaxT=ldir.length();
|
|
ldir.VectorNormalizeFast();
|
|
// now, compute shadow flag
|
|
FourRays myrays;
|
|
myrays.origin=surface_pos;
|
|
FourVectors epsilon=ldir;
|
|
epsilon*=0.01;
|
|
myrays.origin+=epsilon;
|
|
myrays.direction=ldir;
|
|
RayTracingResult shadowtest;
|
|
Trace4Rays(myrays,Four_Zeros,MaxT, &shadowtest);
|
|
bi32x4 unshadowed=CmpGtSIMD(shadowtest.HitDistance,MaxT);
|
|
if (! (IsAllZeros(unshadowed)))
|
|
{
|
|
FourVectors tmp;
|
|
tmp.DuplicateVector(Vector(0,0,0));
|
|
LightList[l].ComputeLightAtPoints(surface_pos,rslt.surface_normal,
|
|
tmp);
|
|
intens.x=AddSIMD(intens.x,AndSIMD(tmp.x,unshadowed));
|
|
intens.y=AddSIMD(intens.y,AndSIMD(tmp.y,unshadowed));
|
|
intens.z=AddSIMD(intens.z,AndSIMD(tmp.z,unshadowed));
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
// now, mask off non-hitting pixels
|
|
intens.VProduct(surf_colors);
|
|
bi32x4 no_hit_mask=CmpGtSIMD(rslt.HitDistance,TraceLimit);
|
|
|
|
intens.x=OrSIMD(AndSIMD(BackgroundColor.x,no_hit_mask),
|
|
AndNotSIMD(no_hit_mask,intens.x));
|
|
intens.y=OrSIMD(AndSIMD(BackgroundColor.y,no_hit_mask),
|
|
AndNotSIMD(no_hit_mask,intens.y));
|
|
intens.z=OrSIMD(AndSIMD(BackgroundColor.y,no_hit_mask),
|
|
AndNotSIMD(no_hit_mask,intens.z));
|
|
|
|
MapLinearIntensities(intens,dest,dest+1,dest+stride,dest+stride+1);
|
|
}
|
|
dest+=2;
|
|
SLoc+=dxvectortimes2;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
#define SQ(x) ((x)*(x))
|
|
|
|
void RayTracingEnvironment::ComputeVirtualLightSources(void)
|
|
{
|
|
int start_pos=0;
|
|
for(int b=0;b<3;b++)
|
|
{
|
|
int nl=LightList.Count();
|
|
int where_to_start=start_pos;
|
|
start_pos=nl;
|
|
for(int l=where_to_start;l<nl;l++)
|
|
{
|
|
DirectionalSampler_t sample_generator;
|
|
int n_desired=1*LightList[l].m_Color.Length();
|
|
if (LightList[l].m_Type==MATERIAL_LIGHT_SPOT)
|
|
n_desired*=LightList[l].m_Phi/2;
|
|
for(int try1=0;try1<n_desired;try1++)
|
|
{
|
|
LightDesc_t const &li=LightList[l];
|
|
FourRays myrays;
|
|
myrays.origin.DuplicateVector(li.m_Position);
|
|
RayTracingResult rslt;
|
|
Vector trial_dir=sample_generator.NextValue();
|
|
if (li.IsDirectionWithinLightCone(trial_dir))
|
|
{
|
|
myrays.direction.DuplicateVector(trial_dir);
|
|
Trace4Rays(myrays,all_zeros,ReplicateX4(1000.0), &rslt);
|
|
if ((rslt.HitIds[0]!=-1))
|
|
{
|
|
// make sure normal points back towards ray origin
|
|
fltx4 ndoti=rslt.surface_normal*myrays.direction;
|
|
fltx4 bad_dirs=AndSIMD(CmpGtSIMD(ndoti,Four_Zeros),
|
|
LoadAlignedSIMD((float *) signmask));
|
|
|
|
// flip signs of all "wrong" normals
|
|
rslt.surface_normal.x=XorSIMD(bad_dirs,rslt.surface_normal.x);
|
|
rslt.surface_normal.y=XorSIMD(bad_dirs,rslt.surface_normal.y);
|
|
rslt.surface_normal.z=XorSIMD(bad_dirs,rslt.surface_normal.z);
|
|
|
|
// a hit! let's make a virtual light source
|
|
|
|
// treat the virtual light as a disk with its center at the hit position
|
|
// and its radius scaled by the amount of the solid angle this probe
|
|
// represents.
|
|
float area_of_virtual_light=
|
|
4.0*M_PI*SQ( SubFloat( rslt.HitDistance, 0 ) )*(1.0/n_desired);
|
|
|
|
FourVectors intens;
|
|
intens.DuplicateVector(Vector(0,0,0));
|
|
|
|
FourVectors surface_pos=myrays.direction;
|
|
surface_pos*=rslt.HitDistance;
|
|
surface_pos+=myrays.origin;
|
|
FourVectors delta=rslt.surface_normal;
|
|
delta*=0.1;
|
|
surface_pos+=delta;
|
|
LightList[l].ComputeLightAtPoints(surface_pos,rslt.surface_normal,
|
|
intens);
|
|
FourVectors surf_colors;
|
|
surf_colors.DuplicateVector(TriangleColors[rslt.HitIds[0]]);
|
|
intens*=surf_colors;
|
|
// see if significant
|
|
LightDesc_t l1;
|
|
l1.m_Type=MATERIAL_LIGHT_SPOT;
|
|
l1.m_Position=Vector(surface_pos.X(0),surface_pos.Y(0),surface_pos.Z(0));
|
|
l1.m_Direction=Vector(rslt.surface_normal.X(0),rslt.surface_normal.Y(0),
|
|
rslt.surface_normal.Z(0));
|
|
l1.m_Color=Vector(intens.X(0),intens.Y(0),intens.Z(0));
|
|
if (l1.m_Color.Length()>0)
|
|
{
|
|
l1.m_Color*=area_of_virtual_light/M_PI;
|
|
l1.m_Range=0.0;
|
|
l1.m_Falloff=1.0;
|
|
l1.m_Attenuation0=1.0;
|
|
l1.m_Attenuation1=0.0;
|
|
l1.m_Attenuation2=1.0; // intens falls off as 1/r^2
|
|
l1.m_Theta=0;
|
|
l1.m_Phi=M_PI;
|
|
l1.RecalculateDerivedValues();
|
|
LightList.AddToTail(l1);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
static unsigned int GetSignMask(Vector const &v)
|
|
{
|
|
unsigned int ret=0;
|
|
if (v.x<0.0)
|
|
ret++;
|
|
if (v.y<0)
|
|
ret+=2;
|
|
if (v.z<0)
|
|
ret+=4;
|
|
return ret;
|
|
}
|
|
|
|
|
|
inline void RayTracingEnvironment::FlushStreamEntry(RayStream &s,int msk, RTECullMode_t cullMode )
|
|
{
|
|
assert(msk>=0);
|
|
assert(msk<8);
|
|
fltx4 tmax=s.PendingRays[msk].direction.length();
|
|
fltx4 scl=ReciprocalSaturateSIMD(tmax);
|
|
s.PendingRays[msk].direction*=scl; // normalize
|
|
RayTracingResult tmpresult;
|
|
Trace4Rays(s.PendingRays[msk],Four_Zeros,tmax,msk,&tmpresult,-1,NULL,cullMode);
|
|
// now, write out results
|
|
for(int r=0;r<4;r++)
|
|
{
|
|
RayTracingSingleResult *out=s.PendingStreamOutputs[msk][r];
|
|
out->ray_length=SubFloat( tmax, r );
|
|
out->surface_normal.x=tmpresult.surface_normal.X(r);
|
|
out->surface_normal.y=tmpresult.surface_normal.Y(r);
|
|
out->surface_normal.z=tmpresult.surface_normal.Z(r);
|
|
out->HitID=tmpresult.HitIds[r];
|
|
out->HitDistance=SubFloat( tmpresult.HitDistance, r );
|
|
}
|
|
s.n_in_stream[msk]=0;
|
|
}
|
|
|
|
void RayTracingEnvironment::AddToRayStream(RayStream &s,
|
|
Vector const &start,Vector const &end,
|
|
RayTracingSingleResult *rslt_out,
|
|
RTECullMode_t cullMode )
|
|
{
|
|
Vector delta=end;
|
|
delta-=start;
|
|
int msk=GetSignMask(delta);
|
|
assert(msk>=0);
|
|
assert(msk<8);
|
|
int pos=s.n_in_stream[msk];
|
|
assert(pos<4);
|
|
s.PendingRays[msk].origin.X(pos)=start.x;
|
|
s.PendingRays[msk].origin.Y(pos)=start.y;
|
|
s.PendingRays[msk].origin.Z(pos)=start.z;
|
|
s.PendingRays[msk].direction.X(pos)=delta.x;
|
|
s.PendingRays[msk].direction.Y(pos)=delta.y;
|
|
s.PendingRays[msk].direction.Z(pos)=delta.z;
|
|
s.PendingStreamOutputs[msk][pos]=rslt_out;
|
|
if (pos==3)
|
|
{
|
|
FlushStreamEntry(s,msk,cullMode);
|
|
}
|
|
else
|
|
s.n_in_stream[msk]++;
|
|
}
|
|
|
|
void RayTracingEnvironment::FinishRayStream(RayStream &s, RTECullMode_t cullMode )
|
|
{
|
|
for(int msk=0;msk<8;msk++)
|
|
{
|
|
int cnt=s.n_in_stream[msk];
|
|
if (cnt)
|
|
{
|
|
// fill in unfilled entries with dups of first
|
|
for(int c=cnt;c<4;c++)
|
|
{
|
|
s.PendingRays[msk].origin.X(c) = s.PendingRays[msk].origin.X(0);
|
|
s.PendingRays[msk].origin.Y(c) = s.PendingRays[msk].origin.Y(0);
|
|
s.PendingRays[msk].origin.Z(c) = s.PendingRays[msk].origin.Z(0);
|
|
s.PendingRays[msk].direction.X(c) = s.PendingRays[msk].direction.X(0);
|
|
s.PendingRays[msk].direction.Y(c) = s.PendingRays[msk].direction.Y(0);
|
|
s.PendingRays[msk].direction.Z(c) = s.PendingRays[msk].direction.Z(0);
|
|
s.PendingStreamOutputs[msk][c]=s.PendingStreamOutputs[msk][0];
|
|
}
|
|
FlushStreamEntry(s,msk,cullMode);
|
|
}
|
|
}
|
|
}
|