560 lines
14 KiB
C++
560 lines
14 KiB
C++
|
|
#include <typeinfo>
|
|
#include <iostream>
|
|
#include "Scene.h"
|
|
#include "Light.h"
|
|
#include "parser/parser.h"
|
|
#include "parser/nodes.h"
|
|
|
|
using namespace std;
|
|
|
|
typedef vector< refptr<Node> >::const_iterator Node_Iterator;
|
|
|
|
void Scene::load(const char * filename)
|
|
{
|
|
refptr<Node> node = parse(filename);
|
|
processNode(node);
|
|
}
|
|
|
|
void Scene::processNode(refptr<Node> node)
|
|
{
|
|
if (node.isNull())
|
|
return;
|
|
|
|
if ( typeid(*node) == typeid(SceneNode) )
|
|
{
|
|
processScene(node);
|
|
}
|
|
else if ( node->isShape() )
|
|
{
|
|
refptr<Shape> shape = processShape(node);
|
|
if ( ! shape.isNull() )
|
|
m_shapes.push_back(shape);
|
|
}
|
|
else if ( typeid(*node) == typeid(LightNode) )
|
|
{
|
|
refptr<Light> light = processLight(node);
|
|
if ( ! light.isNull() )
|
|
m_lights.push_back(light);
|
|
}
|
|
else if ( typeid(*node) == typeid(OptionsNode) )
|
|
{
|
|
processOptions(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(TranslateBlockNode) )
|
|
{
|
|
processTransformBlock(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(RotateBlockNode) )
|
|
{
|
|
processTransformBlock(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(ScaleBlockNode) )
|
|
{
|
|
processTransformBlock(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(MaterialDefinitionNode) )
|
|
{
|
|
processMaterialDefinition(node);
|
|
}
|
|
else
|
|
{
|
|
cerr << __FILE__ << ": " << __LINE__
|
|
<< ": error: unrecognized node!" << endl;
|
|
}
|
|
}
|
|
|
|
void Scene::processChildren(refptr<Node> node)
|
|
{
|
|
std::vector< refptr<Node> > & children = node->getChildren();
|
|
for (int i = 0, sz = children.size(); i < sz; i++)
|
|
{
|
|
processNode(children[i]);
|
|
}
|
|
}
|
|
|
|
void Scene::processScene(refptr<Node> node)
|
|
{
|
|
/* first process any cameras present */
|
|
int cameras_found = 0;
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(CameraNode) )
|
|
{
|
|
cameras_found++;
|
|
if (cameras_found == 1)
|
|
processCamera(*it);
|
|
else
|
|
{
|
|
cerr << "Error: multiple camera definitions found!" << endl;
|
|
exit(5);
|
|
}
|
|
}
|
|
}
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) != typeid(CameraNode) )
|
|
{
|
|
processNode(*it);
|
|
}
|
|
}
|
|
}
|
|
|
|
refptr<Shape> Scene::processShape(refptr<Node> node)
|
|
{
|
|
if ( typeid(*node) == typeid(BoxNode) )
|
|
{
|
|
return processBox(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(PlaneNode) )
|
|
{
|
|
return processPlane(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(SphereNode) )
|
|
{
|
|
return processSphere(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(CylNode) )
|
|
{
|
|
return processCyl(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(IntersectNode) )
|
|
{
|
|
return processBool(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(UnionNode) )
|
|
{
|
|
return processBool(node);
|
|
}
|
|
else if ( typeid(*node) == typeid(SubtractNode) )
|
|
{
|
|
return processBool(node);
|
|
}
|
|
|
|
return refptr<Shape>(NULL);
|
|
}
|
|
|
|
void Scene::processCamera(refptr<Node> node)
|
|
{
|
|
Vector position(0, 0, 0);
|
|
Vector look_at(0, 1, 0);
|
|
Vector up(0, 0, 1);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(PositionNode) )
|
|
{
|
|
position = * (*it)->getVector();
|
|
}
|
|
else if ( typeid(**it) == typeid(LookAtNode) )
|
|
{
|
|
look_at = * (*it)->getVector();
|
|
}
|
|
else if ( typeid(**it) == typeid(UpNode) )
|
|
{
|
|
up = * (*it)->getVector();
|
|
}
|
|
else if ( typeid(**it) == typeid(VFOVNode) )
|
|
{
|
|
m_vfov = (*it)->getNumber();
|
|
}
|
|
}
|
|
|
|
m_transforms.top().lookAt(position, look_at, up);
|
|
}
|
|
|
|
void Scene::processOptions(refptr<Node> node)
|
|
{
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(WidthNode) )
|
|
{
|
|
m_width = (*it)->getInteger();
|
|
}
|
|
else if ( typeid(**it) == typeid(HeightNode) )
|
|
{
|
|
m_height = (*it)->getInteger();
|
|
}
|
|
else if ( typeid(**it) == typeid(MultisampleNode) )
|
|
{
|
|
m_multisample_level = (*it)->getInteger();
|
|
}
|
|
else if ( typeid(**it) == typeid(MaxDepthNode) )
|
|
{
|
|
m_max_depth = (*it)->getInteger();
|
|
}
|
|
}
|
|
}
|
|
|
|
void Scene::processTransformBlock(refptr<Node> node)
|
|
{
|
|
if ( typeid(*node) == typeid(TranslateBlockNode) )
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
m_transforms.top().translate(node->getVector());
|
|
}
|
|
else if ( typeid(*node) == typeid(RotateBlockNode) )
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
m_transforms.top().rotate(node->getNumber(),
|
|
node->getVector());
|
|
}
|
|
else if ( typeid(*node) == typeid(ScaleBlockNode) )
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
m_transforms.top().scale(node->getVector());
|
|
}
|
|
|
|
processChildren(node);
|
|
|
|
m_transforms.pop();
|
|
}
|
|
|
|
void Scene::processMaterialDefinition(refptr<Node> node)
|
|
{
|
|
map< string, refptr<Material> >::iterator it =
|
|
m_materials.find(node->getString());
|
|
if ( it == m_materials.end() )
|
|
{
|
|
m_materials[node->getString()] = processMaterial(node);
|
|
}
|
|
else
|
|
{
|
|
cerr << "Error: duplicate material definition for material '"
|
|
<< node->getString() << "'" << endl;
|
|
exit(4);
|
|
}
|
|
}
|
|
|
|
refptr<Material> Scene::processMaterial(refptr<Node> node)
|
|
{
|
|
if ( typeid(*node) == typeid(MaterialRefNode) )
|
|
{
|
|
map< string, refptr<Material> >::iterator it =
|
|
m_materials.find(node->getString());
|
|
if ( it == m_materials.end() )
|
|
{
|
|
cerr << "Undefined material '" << node->getString()
|
|
<< "' requested!" << endl;
|
|
exit(4);
|
|
}
|
|
|
|
return m_materials[node->getString()];
|
|
}
|
|
|
|
refptr<Material> material = new Material();
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(ColorNode) )
|
|
{
|
|
material->setDiffuseColor(Color((*it)->getVector()));
|
|
material->setAmbientColor(Color((*it)->getVector()));
|
|
}
|
|
else if ( typeid(**it) == typeid(AmbientNode) )
|
|
{
|
|
material->setAmbientColor(Color((*it)->getVector()));
|
|
}
|
|
else if ( typeid(**it) == typeid(DiffuseNode) )
|
|
{
|
|
material->setDiffuseColor(Color((*it)->getVector()));
|
|
}
|
|
else if ( typeid(**it) == typeid(SpecularNode) )
|
|
{
|
|
material->setSpecularColor(Color((*it)->getVector()));
|
|
}
|
|
else if ( typeid(**it) == typeid(ReflectanceNode) )
|
|
{
|
|
material->setReflectance((*it)->getNumber());
|
|
}
|
|
else if ( typeid(**it) == typeid(ShininessNode) )
|
|
{
|
|
material->setShininess((*it)->getNumber());
|
|
}
|
|
else if ( typeid(**it) == typeid(TransparencyNode) )
|
|
{
|
|
material->setTransparency((*it)->getNumber());
|
|
}
|
|
}
|
|
|
|
return material;
|
|
}
|
|
|
|
refptr<Shape> Scene::processBox(refptr<Node> node)
|
|
{
|
|
refptr<Vector> size = new Vector(1, 1, 1);
|
|
refptr<Material> material;
|
|
|
|
bool restore_transform = processTransforms(node);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(SizeNode) )
|
|
{
|
|
size = (*it)->getVector();
|
|
}
|
|
else if ( (*it)->isMaterial() )
|
|
{
|
|
material = processMaterial(*it);
|
|
}
|
|
}
|
|
|
|
refptr<Shape> box = new Box(size);
|
|
if ( ! material.isNull() )
|
|
box->setMaterial(material);
|
|
box->setTransform(m_transforms.top());
|
|
|
|
if (restore_transform)
|
|
m_transforms.pop();
|
|
|
|
return box;
|
|
}
|
|
|
|
refptr<Shape> Scene::processCyl(refptr<Node> node)
|
|
{
|
|
double radius1 = 1.0;
|
|
double radius2 = 1.0;
|
|
double height = 1.0;
|
|
refptr<Material> material;
|
|
|
|
bool restore_transform = processTransforms(node);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(SizeNode) )
|
|
{
|
|
const Vector & v = *(*it)->getVector();
|
|
radius1 = v[0];
|
|
radius2 = v[1];
|
|
height = v[2];
|
|
}
|
|
else if ( (*it)->isMaterial() )
|
|
{
|
|
material = processMaterial(*it);
|
|
}
|
|
}
|
|
|
|
refptr<Shape> cyl = new Cyl(radius1, radius2, height);
|
|
if ( ! material.isNull() )
|
|
cyl->setMaterial(material);
|
|
cyl->setTransform(m_transforms.top());
|
|
|
|
if (restore_transform)
|
|
m_transforms.pop();
|
|
|
|
return cyl;
|
|
}
|
|
|
|
refptr<Light> Scene::processLight(refptr<Node> node)
|
|
{
|
|
refptr<Light> light = new PointLight();
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(PositionNode) )
|
|
{
|
|
refptr<Vector> v = (*it)->getVector();
|
|
light->setPosition(m_transforms.top().transform_point(*v));
|
|
}
|
|
else if ( typeid(**it) == typeid(DiffuseNode) )
|
|
{
|
|
Color c((*it)->getVector());
|
|
light->setDiffuseColor(c);
|
|
}
|
|
else if ( typeid(**it) == typeid(SpecularNode) )
|
|
{
|
|
Color c((*it)->getVector());
|
|
light->setSpecularColor(c);
|
|
}
|
|
else if ( typeid(**it) == typeid(ColorNode) )
|
|
{
|
|
Color c((*it)->getVector());
|
|
light->setDiffuseColor(c);
|
|
light->setSpecularColor(c);
|
|
}
|
|
}
|
|
|
|
return light;
|
|
}
|
|
|
|
refptr<Shape> Scene::processPlane(refptr<Node> node)
|
|
{
|
|
Vector normal(0, 0, 1);
|
|
double dist = 0;
|
|
refptr<Material> material;
|
|
|
|
bool restore_transform = processTransforms(node);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(PlanePositionNode) )
|
|
{
|
|
normal = *(*it)->getVector();
|
|
dist = (*it)->getNumber();
|
|
}
|
|
else if ( (*it)->isMaterial() )
|
|
{
|
|
material = processMaterial(*it);
|
|
}
|
|
}
|
|
|
|
refptr<Shape> plane = new Plane(normal[0],
|
|
normal[1],
|
|
normal[2],
|
|
dist);
|
|
if ( ! material.isNull() )
|
|
plane->setMaterial(material);
|
|
plane->setTransform(m_transforms.top());
|
|
|
|
if (restore_transform)
|
|
m_transforms.pop();
|
|
|
|
return plane;
|
|
}
|
|
|
|
refptr<Shape> Scene::processSphere(refptr<Node> node)
|
|
{
|
|
double radius = 1.0;
|
|
refptr<Material> material;
|
|
|
|
bool restore_transform = processTransforms(node);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(RadiusNode) )
|
|
{
|
|
radius = (*it)->getNumber();
|
|
}
|
|
else if ( (*it)->isMaterial() )
|
|
{
|
|
material = processMaterial(*it);
|
|
}
|
|
}
|
|
|
|
refptr<Shape> sphere = new Sphere(radius);
|
|
if ( ! material.isNull() )
|
|
sphere->setMaterial(material);
|
|
sphere->setTransform(m_transforms.top());
|
|
|
|
if (restore_transform)
|
|
m_transforms.pop();
|
|
|
|
return sphere;
|
|
}
|
|
|
|
refptr<Shape> Scene::processBool(refptr<Node> node)
|
|
{
|
|
vector< refptr<Shape> > shapes;
|
|
refptr<Material> material;
|
|
|
|
bool restore_transform = processTransforms(node);
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( (*it)->isShape() )
|
|
{
|
|
refptr<Shape> shape = processShape(*it);
|
|
if ( ! shape.isNull() )
|
|
shapes.push_back(shape);
|
|
}
|
|
else if ( (*it)->isMaterial() )
|
|
{
|
|
material = processMaterial(*it);
|
|
}
|
|
}
|
|
|
|
if (shapes.size() < 2)
|
|
{
|
|
cerr << "Error: boolean objects must have 2 or more sub-objects!"
|
|
<< endl;
|
|
exit(3);
|
|
}
|
|
|
|
refptr<Shape> shape;
|
|
if ( typeid(*node) == typeid(IntersectNode) )
|
|
shape = new Intersect(shapes);
|
|
else if ( typeid(*node) == typeid(UnionNode) )
|
|
shape = new Union(shapes);
|
|
else if ( typeid(*node) == typeid(SubtractNode) )
|
|
shape = new Subtract(shapes);
|
|
else
|
|
{
|
|
cerr << __FILE__ << ": " << __LINE__
|
|
<< ": error: bool object unrecognized" << endl;
|
|
exit(3);
|
|
}
|
|
|
|
if ( ! material.isNull() )
|
|
shape->setMaterial(material);
|
|
shape->setTransform(m_transforms.top());
|
|
|
|
if (restore_transform)
|
|
m_transforms.pop();
|
|
|
|
return shape;
|
|
}
|
|
|
|
bool Scene::processTransforms(refptr<Node> node)
|
|
{
|
|
bool did_any = false;
|
|
|
|
for (Node_Iterator it = node->getChildren().begin();
|
|
it != node->getChildren().end();
|
|
it++)
|
|
{
|
|
if ( typeid(**it) == typeid(TranslateNode) )
|
|
{
|
|
if (did_any == false)
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
did_any = true;
|
|
}
|
|
m_transforms.top().translate((*it)->getVector());
|
|
}
|
|
else if ( typeid(**it) == typeid(RotateNode) )
|
|
{
|
|
if (did_any == false)
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
did_any = true;
|
|
}
|
|
m_transforms.top().rotate((*it)->getNumber(),
|
|
(*it)->getVector());
|
|
}
|
|
else if ( typeid(**it) == typeid(ScaleNode) )
|
|
{
|
|
if (did_any == false)
|
|
{
|
|
m_transforms.push(m_transforms.top());
|
|
did_any = true;
|
|
}
|
|
m_transforms.top().scale((*it)->getVector());
|
|
}
|
|
}
|
|
|
|
return did_any;
|
|
}
|