Retro68/Raytracer/raytracer2.cc

#ifdef __APPLE__
#include <Carbon/Carbon.h>
#include <ApplicationServices/ApplicationServices.h>
#define PSTR(x) ("\p" x)

#else

#include <Quickdraw.h>
#include <MacMemory.h>
#include <Sound.h>
#include <Events.h>
#include <Fonts.h>
#include <NumberFormatting.h>

#ifdef __GNUC__
#include "MacUtils.h"
QDGlobals qd;
#else
#define PSTR(x) ("\p" x)
#endif

#endif

#include "fixed.h"

#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <vector>

using std::sqrt;
using std::floor;



inline int floor_to_int(float f)
{
	return static_cast<int>(std::floor(f));
}

template<class T>
class vec3
{
public:
	T x,y,z;
	vec3(T x, T y, T z) : x(x), y(y), z(z) {}
	vec3() : x(0), y(0), z(0) {}
	
	vec3<T> operator+(const vec3<T>& other) const { return vec3(x + other.x, y + other.y, z + other.z); }
	vec3<T> operator-(const vec3<T>& other) const { return vec3(x - other.x, y - other.y, z - other.z); }
	vec3<T> operator*(T a) const { return vec3(x*a, y*a, z*a); }
	
	T length() const { return sqrt(x*x + y*y + z*z); }
	T operator*(const vec3<T>& other) const { return x*other.x + y*other.y + z*other.z; }
	
#if 1
	vec3<T> normalize() const { 
		T l = length();
		//if(l == 0)
		//	return *this;
		//else
			return (*this) * (T(1) / length());
	}
#else
	vec3<T> normalize() const { 
		T l = length();
		if(l == 0)
			return *this;
		else
			return vec3<T>(x/l, y/l, z/l);
	}
#endif
};

template<class T>
bool hitSphere(vec3<T> p0, vec3<T> dir, T& t)
{
	const vec3<T> center(0.0f, 1.0f, -6.0f);
	const T r = 2.0f;
	vec3<T> p0c(p0 - center);
	
	/*
	(x-xc)^2 + (y-yc)^2 + (z-zc)^2 = r^2;
	(x0c + dx * t)^2 + (y0c + dy * t)^2 + (z0c + dz * t)^2 = r^2;
	x0c^2 + 2*x0c*dx*t + dx^2*t^2 + y0c^2 + 2*y0c*dy*t + dy^2*t^2 + z0c^2 + 2 * z0c*dz*t + dz^2*t^2 = r^2
	
	(dx^2 + dy^2 + dz^2)*t^2 + (2*x0c*dx + 2*y0c&dy + 2*z0c*dz) * t + x0c^2+y0c^2+z0c^2-r^2
	*/
	T a = dir*dir;
	T b = 2*(p0c*dir);
	T c = p0c*p0c - r*r;
	
	T D = b*b - 4 * a * c;
	
	if(D >= 0)
	{
		t = (-b - sqrt(D)) / (2*a);
		return t >= 0;
	}
	return false;
}


template<class T>
T ray(int n, vec3<T> p0, vec3<T> dir)
{
#if 1
	static const vec3<T> light = vec3<T>(-2,4,3).normalize();
	
	if(1){
		const vec3<T> center(0.0f, 1.0f, -6.0f);
		const T r = 2.0f;
		vec3<T> p0c(p0 - center);
		
		/*
		(x-xc)^2 + (y-yc)^2 + (z-zc)^2 = r^2;
		(x0c + dx * t)^2 + (y0c + dy * t)^2 + (z0c + dz * t)^2 = r^2;
		x0c^2 + 2*x0c*dx*t + dx^2*t^2 + y0c^2 + 2*y0c*dy*t + dy^2*t^2 + z0c^2 + 2 * z0c*dz*t + dz^2*t^2 = r^2
		
		(dx^2 + dy^2 + dz^2)*t^2 + (2*x0c*dx + 2*y0c&dy + 2*z0c*dz) * t + x0c^2+y0c^2+z0c^2-r^2
		*/
		T a = dir*dir;
		T b = 2*(p0c*dir);
		T c = p0c*p0c - r*r;
		
		T D = b*b - 4 * a * c;
		
		if(D >= 0)
		{
			T t = (-b -  sqrt(D)) / (2*a);
			if(t > 0)
			{
				vec3<T> p = p0 + dir * t;
				
				vec3<T> dir2 = (p - center) * (T(1)/r);
				
				T l = dir2*dir;
				
				T reflected;
				if(n)
					reflected = ray(n-1, p, dir - dir2*(l*2));
				else
					reflected = 0.0f;
				
				
				T lambert = dir2 * light;
				
				return T(0.2f) + T(0.4f) * std::max(T(0),lambert) + T(0.4f) * reflected;
			}
		}
	}

	if(dir.y < 0)
	{
		T t = (T(-1.5f) - p0.y) / dir.y;
		vec3<T> p = p0 + dir*t;
		
		T color;
		if( (floor_to_int(p.x)
			+ floor_to_int(p.z)) % 2 )
			color = 0.8f;
		else
			color = 0.1f;
		
		T ts;
		if(hitSphere(p, light, ts))
			color *= T(0.2f);
		
		return std::min(T(1), color + T(0.5f) * ray(n-1, p,vec3<T>(dir.x, -dir.y, dir.z)));
	}
#endif
	return std::max(T(0), dir.y * T(0.3f));
}

typedef fixed numtype;


template<class T>
struct rand1
{
	static T get()
	{
		return T(std::rand()) / T(32767.0f * 65536.0f);
	}
};

template<>
struct rand1<fixed>
{
	static fixed get()
	{
		return fixed(std::rand() >> 15, fixed::raw());
	}
};

int main()
{
	WindowPtr win;
#if !TARGET_API_MAC_CARBON
    InitGraf(&qd.thePort);
    InitFonts();
    InitWindows();
    InitMenus();

	Rect r = qd.screenBits.bounds;
#else
	BitMap bm;
	GetQDGlobalsScreenBits(&bm);
	Rect r = bm.bounds;
#endif    
    SetRect(&r, r.left + 5, r.top + 45, r.right - 5, r.bottom -5);
    win = NewWindow(NULL, &r, PSTR("Raytracer"), true, 0, (WindowPtr)-1, false, 0);
    
#if !TARGET_API_MAC_CARBON	
	SetPort(win);
	r = win->portRect;
#else
	SetPortWindowPort(win);
	GetPortBounds(GetWindowPort(win), &r);
#endif
    EraseRect(&r);
	numtype accum = 0.0f;
	short cx = r.right /2;
	short cy = r.bottom / 2;
	
	long startTime = TickCount();
	std::vector<numtype> accumV(r.right);
	BitMap line;
	std::vector<unsigned char> bits(((r.right + 31) / 8) & ~0x3);
	SetRect(&line.bounds, 0,0,r.right,1);
	line.rowBytes = bits.size();
	line.baseAddr = (char*)(&bits[0]);
	
	for(int y = 0; y < r.bottom; y++)
	{
		std::fill(bits.begin(),bits.end(), 0);
		for(int x = 0; x < r.right; x++)
		{
			numtype pixel;
			
			// cam = (0,0,0)
			// ray = t * (x-r.right/2, - (y-r.bottom/2), -1)
			// plane: y = -2
			pixel = ray(1,vec3<numtype>(),vec3<numtype>(numtype(x-cx)/numtype(cx),-numtype(y-cy)/numtype(cx),-1).normalize());

			numtype thresh = rand1<numtype>::get();
			
			thresh = numtype(0.5f) + numtype(0.4f) * (thresh - numtype(0.5f));
			accum += pixel;
			accum += accumV[x];
			if(accum >= thresh)
				accum -= 1;
			else
			{
				//MoveTo(x,y);
				//Line(0,0);
				bits[x / 8] |= (0x80 >> (x%8));
			}
			accumV[x] = accum = accum / numtype(2);			
		}
		Rect r2;
		SetRect(&r2,0,y,r.right,y+1);
		CopyBits(&line, &win->portBits, &line.bounds, &r2, srcCopy, NULL);
		if(Button())
			return 0;
#if TARGET_API_MAC_CARBON
//		QDFlushPortBuffer(GetWindowPort(win),NULL);
#endif
	}
	long endTime = TickCount();
	
	char buf[256];
	unsigned char* pstr = (unsigned char*)buf;
	std::sprintf(buf+1, "pps = %d %ld %ld %ld %ld", (int)( (long)r.right * r.bottom * 60 / (endTime - startTime) ),
	             fixed::nMul, fixed::nIMul, fixed::nDiv, fixed::nSqrt);
	buf[0] = std::strlen(buf+1);
	
	SetRect(&r, 10, 10, 10 + StringWidth(pstr) + 10, 30);
	PaintRect(&r);
	PenMode(patXor);
	FrameRect(&r);
	MoveTo(15,25);
	TextMode(srcBic);
	DrawString(pstr);
#if TARGET_API_MAC_CARBON
	QDFlushPortBuffer(GetWindowPort(win),NULL);
#endif
	
	while(!Button())
		;
	FlushEvents(everyEvent, -1);
	return 0;
}