/*
 * raytracer.h: Raytracer class
 *
 * This file is part of Pyrit Ray Tracer.
 *
 * Copyright 2006, 2007, 2008  Radek Brich
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#ifndef RAYTRACER_H
#define RAYTRACER_H

#include <vector>
#include <queue>

#include "common.h"
#include "container.h"
#include "scene.h"

class Raytracer;

/**
 * main ray tracer class
 */
class Raytracer
{
	Container *top;
	Sampler *sampler;
	Camera *camera;
	vector<Light*> lights;
	Colour bg_colour;
	Float ao_distance, ao_angle;
	int ao_samples;
	int num_threads;
	int max_depth;
	bool use_packets;

	Sample *sample_queue;
	int sample_queue_pos, sample_queue_size, sample_queue_count;
	bool end_of_samples;
	pthread_mutex_t sample_queue_mutex, sampler_mutex;
	pthread_cond_t sample_queue_cond, worker_ready_cond;

	Vector SphereDistribute(int i, int n, Float extent, const Vector &normal);
	Colour PhongShader(const Shape *shape,
		const Vector &P, const Vector &N, const Vector &V);
	void lightScatter(const Ray &ray, const Shape *shape, int depth,
		const Vector &P, const Vector &normal, bool from_inside, Colour &col);
#ifndef NO_SIMD
	VectorPacket PhongShader_packet(const Shape* const* shapes,
		const VectorPacket &P, const VectorPacket &N, const VectorPacket &V);
	void raytracePacket(RayPacket &rays, Colour *results);
#endif
	static void *raytrace_worker(void *d);
public:
	Raytracer(): top(NULL), camera(NULL), lights(), bg_colour(0., 0., 0.),
		ao_samples(0), num_threads(2), max_depth(3), use_packets(true)
	{
		pthread_mutex_init(&sample_queue_mutex, NULL);
		pthread_mutex_init(&sampler_mutex, NULL);
		pthread_cond_init (&sample_queue_cond, NULL);
		pthread_cond_init (&worker_ready_cond, NULL);
	};
	~Raytracer()
	{
		pthread_mutex_destroy(&sample_queue_mutex);
		pthread_mutex_destroy(&sampler_mutex);
		pthread_cond_destroy (&sample_queue_cond);
		pthread_cond_destroy (&worker_ready_cond);
	}

	void render();
	Colour raytrace(Ray &ray, int depth, const Shape *origin_shape);
	void addShape(Shape *shape) { top->addShape(shape); };
	void addLight(Light *light) { lights.push_back(light); };
	void setSampler(Sampler *sampl) { sampler = sampl; };
	Sampler *&getSampler() { return sampler; };
	void setCamera(Camera *cam) { camera = cam; };
	Camera *&getCamera() { return camera; };
	void setTop(Container *atop) { top = atop; };
	Container *&getTop() { return top; };

	void setBgColour(const Colour &bg) { bg_colour = bg; };
	void setMaxDepth(int newdepth) { max_depth = newdepth; };

	void ambientOcclusion(int samples, Float distance, Float angle);
	void setThreads(int num) { num_threads = num; };
};

#endif