#include "raytracer.h"
#include "kdtree.h"

#include "common_sdl.h"

Camera cam(Vector(0.0f,6.0f,6.0f), Vector(0.0f,2.0f,-7.0f), Vector(0.0f,0.0f,-1.0f));
Light light(Vector(-2.0f, 10.0f, -2.0f), Colour(0.9f, 0.9f, 0.9f));

Float lx, ly, lz, cf;

void update_callback(Float*)
{
	if (lx != 0.0)
		light.pos.x += lx;
	if (ly != 0.0)
		light.pos.y += ly;
	if (lz != 0.0)
		light.pos.z += lz;
	if (cf != 0.0)
		cam.setF(cam.getF() + cf);
}

void key_callback(int key, int down)
{
	switch (key)
	{
		case SDLK_r:
			lx = -0.1f * down;
			break;
		case SDLK_t:
			lx = +0.1f * down;
			break;
		case SDLK_f:
			ly = -0.1f * down;
			break;
		case SDLK_g:
			ly = +0.1f * down;
			break;
		case SDLK_v:
			lz = -0.1f * down;
			break;
		case SDLK_b:
			lz = +0.1f * down;
			break;

		case SDLK_z:
			cf = -0.02f * down;
			break;
		case SDLK_x:
			cf = +0.02f * down;
			break;
	}
}

int main(int argc, char **argv)
{
	Raytracer rt;

	KdTree top;
	rt.setCamera(&cam);
	rt.setTop(&top);

	rt.addLight(&light);
	light.castShadows(false);

	const Colour c_white(1, 1, 1);
	const Colour c_black(0, 0, 0);

	Material mat0a(Colour(0.7f, 0.7f, 0.7f));
	mat0a.setReflectivity(0.0f);
	Box box(Vector(-12.0f, -1.2f, -20.0f), Vector(12.0f, -1.0f, 0.0f), &mat0a);
	rt.addShape(&box);

	Material mat0b(Colour(0.1f, 0.7f, 0.8f));
	mat0b.setReflectivity(0.7f);
	Box box2(Vector(-12.0f, -1.2f, -10.0f), Vector(12.0f, 10.0f, -10.2f), &mat0b);
	rt.addShape(&box2);

	Float bounds[] = {0.3f, 0.6f, 1.1f};
	Colour colours[] = {c_black, c_white, c_black};
	BoundColourMap cmap(bounds, colours);

	// spheres
	Material mat1(c_white);
	mat1.texture = new CheckersTexture(new PlanarMap(Vector(-4.5f, 2.0f, -7.0f), 0.48f), &cmap);
	Sphere sphere1(Vector(-4.5f, 2.0f, -7.0f), 1, &mat1);
	rt.addShape(&sphere1);

	Material mat2(c_white);
	mat2.texture = new CheckersTexture(new CubicMap(Vector(-1.5f, 2.0f, -7.0f), 0.48f), &cmap);
	Sphere sphere2(Vector(-1.5f, 2.0f, -7.0f), 1, &mat2);
	rt.addShape(&sphere2);

	Material mat3(c_white);
	mat3.texture = new CheckersTexture(new CylinderMap(Vector(1.5f, 2.0f, -7.0f), 0.48f), &cmap);
	Sphere sphere3(Vector(1.5f, 2.0f, -7.0f), 1, &mat3);
	rt.addShape(&sphere3);

	Material mat4(c_white);
	mat4.texture = new CheckersTexture(new SphereMap(Vector(4.5f, 2.0f, -7.0f), 0.48f), &cmap);
	Sphere sphere4(Vector(4.5f, 2.0f, -7.0f), 1, &mat4);
	rt.addShape(&sphere4);

	// cubes
	Material mat5(c_white);
	const Vector cube1_base(-4.5f, 0.0f, -7.0f);
	mat5.texture = new CheckersTexture(new PlanarMap(cube1_base, 0.48f), &cmap);
	Box cube1(cube1_base - 1.0f, cube1_base + 1.0f, &mat5);
	rt.addShape(&cube1);

	Material mat6(c_white);
	const Vector cube2_base(-1.5f, 0.0f, -7.0f);
	mat6.texture = new CheckersTexture(new CubicMap(cube2_base, 0.48f), &cmap);
	Box cube2(cube2_base - 1.0f, cube2_base + 1.0f, &mat6);
	rt.addShape(&cube2);

	Material mat7(c_white);
	const Vector cube3_base(1.5f, 0.0f, -7.0f);
	mat7.texture = new CheckersTexture(new CylinderMap(cube3_base, 0.48f), &cmap);
	Box cube3(cube3_base - 1.0f, cube3_base + 1.0f, &mat7);
	rt.addShape(&cube3);

	Material mat8(c_white);
	const Vector cube4_base(4.5f, 0.0f, -7.0f);
	mat8.texture = new CheckersTexture(new SphereMap(cube4_base, 0.48f), &cmap);
	Box cube4(cube4_base - 1.0f, cube4_base + 1.0f, &mat8);
	rt.addShape(&cube4);

	mat1.setReflectivity(0);
	mat2.setReflectivity(0);
	mat3.setReflectivity(0);
	mat4.setReflectivity(0);
	mat5.setReflectivity(0);
	mat6.setReflectivity(0);
	mat7.setReflectivity(0);
	mat8.setReflectivity(0);

	top.optimize();

	w = 1024;
	h = 600;

	/* run interactive mode */
	loop_sdl(rt, cam, update_callback, key_callback);

	/* render image */
	if (argc == 2 && !strcmp(argv[1], "-r"))
	{
		pyrit_verbosity = 2;
		rt.ambientOcclusion(300, 5.0f, 0.5f);
		DefaultSampler sampler(w, h);
		sampler.setOversample(2);
		sampler.setSubsample(1);
		rt.setSampler(&sampler);
		rt.render();
		sampler.getPixmap().writePNG("textures.png");
	}
	
	return 0;
}