fart/main/Scene.cc
Josh Holtrop 7d461477b3 added operator<<(ostream, Ray) to util/Ray, main/Scene now calculating rays
git-svn-id: svn://anubis/fart/trunk@38 7f9b0f55-74a9-4bce-be96-3c2cd072584d
2009-01-23 15:15:35 +00:00

125 lines
3.3 KiB
C++

#include "Scene.h"
#include <stdlib.h>
#include <string>
#include <map>
#include <math.h>
#include "BMP.h"
#include "shapes/Shape.h"
#include "shapes/Sphere.h"
using namespace std;
Scene::Scene(map<string, const char *> options,
const char * filename)
{
m_width = 800;
m_height = 600;
m_multisample_level = 1;
m_output_file_name = "fart.bmp";
m_vfov = 60.0;
m_verbose = false;
m_data = NULL;
load(filename);
/* after loading the scene file, apply any command-line render options */
for (map<string, const char *>::iterator it = options.begin();
it != options.end();
it++)
{
if (it->first == "width")
{
m_width = atoi(it->second);
}
else if (it->first == "height")
{
m_height = atoi(it->second);
}
else if (it->first == "multisample")
{
m_multisample_level = atoi(it->second);
}
else if (it->first == "field-of-view")
{
m_vfov = atof(it->second);
}
else if (it->first == "output-file")
{
m_output_file_name = it->second;
}
else if (it->first == "verbose")
{
m_verbose = true;
}
}
}
Scene::~Scene()
{
if (m_data != NULL)
delete m_data;
for (vector<Shape *>::iterator it = m_shapes.begin();
it != m_shapes.end();
it++)
{
delete (*it);
}
}
void Scene::load(const char * filename)
{
/* TODO: parse file somehow */
Shape * shape = new Sphere(2.0);
m_transform.translate(2.0, 6.0, 1.0);
shape->setTransform(m_transform);
m_shapes.push_back(shape);
}
void Scene::render()
{
if (m_verbose)
{
cout << " *** Beginning scene render ***" << endl;
cout << "Parameters:" << endl;
cout << "----------------------------------------" << endl;
cout << " Width: " << m_width << endl;
cout << " Height: " << m_height << endl;
cout << " Multisample Level: " << m_multisample_level << endl;
cout << " Vertical Field of View: " << m_vfov << endl;
cout << "----------------------------------------" << endl;
}
m_data = new unsigned char[m_width * m_height * 3];
for (int i = 0; i < m_height; i++)
{
for (int j = 0; j < m_width; j++)
{
renderPixel(j, i, &m_data[3 * (m_width * i + j)]);
}
}
if (m_verbose)
{
cout << " *** Ending scene render ***" << endl;
cout << "Writing output file '" << m_output_file_name << '\'' << endl;
}
BMP outputImage(m_output_file_name.c_str(), m_width, m_height, m_data);
}
void Scene::renderPixel(int x, int y, unsigned char * pixel)
{
/* calculate the ray going from the camera through this pixel */
double rx = (x + 0.5) - (m_width >> 1);
double rz = (m_height >> 1) - (y + 0.5);
/* ry is calculated based on the field of view */
double ry = (m_height / 2.0) / atan(m_vfov / 2.0);
Ray ray(Vector(0, 0, 0), Vector(rx, ry, rz));
// TODO: um... real work
pixel[BMP_RED] = 0;
pixel[BMP_GREEN] = 0;
pixel[BMP_BLUE] = 0;
}