1 files changed, 87 insertions, 0 deletions
diff --git a/unittests/rt_test/rt_test.cpp b/unittests/rt_test/rt_test.cpp
new file mode 100644
index 0000000..c9399e8
--- /dev/null
+++ b/unittests/rt_test/rt_test.cpp
@@ -0,0 +1,87 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+#include "tier0/platform.h"
+#include "tier0/progressbar.h"
+#include "bitmap/float_bm.h"
+#include "mathlib/mathlib.h"
+#include "tier2/tier2.h"
+#include "tier0/memdbgon.h"
+#include "raytrace.h"
+#include "bitmap/tgawriter.h"
+
+void main(int argc,char **argv)
+{
+	InitCommandLineProgram( argc, argv );
+
+	if (argc != 5)
+	{
+		printf("format is 'rt_test src_image dest_image xsize ysize'\n");
+	}
+	else
+	{
+		ReportProgress("reading src texture",0,0);
+		FloatBitMap_t src_texture(argv[1]);
+		int xsize = atoi( argv[3] );
+		int ysize = atoi( argv[4] );
+		
+		// render a simple scene of a terrain, using a bitmap for color data and its alpha channel for the height
+		RayTracingEnvironment rt_Env;
+		int id = 1;
+		float flXScale = (1.0/(src_texture.Width-1) );
+		float flZScale = (1.0/(src_texture.Height-1) );
+		for( int y=0; y < src_texture.Height-1; y++ )
+			for(int x=0 ; x < src_texture.Width-1; x++ )
+			{
+				Vector vecVerts[2][2];
+				for(int iy=0 ; iy < 2; iy++)
+					for(int ix=0 ; ix < 2; ix++)
+					{
+						vecVerts[ix][iy].x =  2.0* ( ( x+ix )*flXScale-0.5 );
+						if ( ( x+ix == src_texture.Width-1 ) || ( y+iy==src_texture.Height-1 ) )
+							vecVerts[ix][iy].y = 0;
+						else
+							vecVerts[ix][iy].y = 0.3*src_texture.Pixel( x+ix, y+iy, 1 );
+						vecVerts[ix][iy].z =  -2.0* ( ( y+iy )*flZScale-0.5 );
+					}
+				Vector vecColor( GammaToLinear(src_texture.Pixel(x,y,0)),
+								 GammaToLinear( src_texture.Pixel( x, y, 1 )),
+								 GammaToLinear( src_texture.Pixel( x, y, 2 )) );
+				rt_Env.AddTriangle( id++, vecVerts[0][0], vecVerts[1][0], vecVerts[1][1], vecColor );
+				rt_Env.AddTriangle( id++, vecVerts[0][0], vecVerts[0][1], vecVerts[1][1], vecColor );
+			}
+		rt_Env.AddTriangle( id++, Vector(0,0,-.2), Vector(.2,0,.2), Vector( -.2,0,.2), Vector( 0,0,1 ) );
+		printf("n triangles %d\n",id);
+		ReportProgress("Creating kd-tree",0,0);
+		float stime = Plat_FloatTime();
+		rt_Env.SetupAccelerationStructure();
+		printf("kd built time := %d\n", (int) ( Plat_FloatTime() - stime ) );
+		rt_Env.AddInfinitePointLight( Vector( 0,5, 0), Vector( .1,.1,.1 ));
+		// lets render a frame
+		uint32 *buf=reinterpret_cast<uint32 *> ( MemAlloc_AllocAligned( xsize * ysize * 4 , 16 ) );
+
+		Vector EyePos(0,2,0);
+		ReportProgress("Rendering",0,0);
+
+// 		rt_Env.RenderScene( xsize, ysize, xsize, buf, Vector( 0, 0.5, -1.0 ),
+// 							Vector( -1, 1, 0), 
+// 							Vector( 1, 1, 0 ),
+// 							Vector( -1, -1, 0 ),
+// 							Vector( 1, -1, 0 ) );
+		float curtime = Plat_FloatTime();
+		for(int i=0;i<10;i++)
+		{
+			rt_Env.RenderScene( xsize, ysize, xsize, buf,
+								EyePos,
+								Vector( -1, 0,1)-EyePos, 
+								Vector( 1, 0, 1 )-EyePos,
+								Vector( -1, 0, -1 )-EyePos,
+								Vector( 1, 0, -1 )-EyePos );
+		}
+		float etime=Plat_FloatTime()-curtime;
+		printf("pixels traced and lit per second := %f\n",(10*xsize*ysize)*(1.0/etime));
+		TGAWriter::WriteTGAFile( "test.tga", xsize, ysize, IMAGE_FORMAT_RGBA8888,
+								 reinterpret_cast<uint8 *> (buf), 4*xsize );
+		
+		MemAlloc_FreeAligned( buf );
+	}
+
+}