WebSVN – gelsvn – Diff – /trunk/apps/Pathtracer/scene.cpp


#include <Geometry/BSPTree.h>
 
#include "scene.h"
#include "mesh.h"
 
using namespace Geometry;
using namespace CGLA;
 
scene::scene(void) : cam_(0)
{
    accel_ = new BSPTree;
}
 
scene::~scene(void)
{
    delete accel_;
}
 
void scene::insert(const luminaire* obj)
{
    objs_.push_back(obj);
}
 
size_t scene::luminaires(void) const
{
    return objs_.size();
}
 
const luminaire* scene::get_luminaire(size_t i) const
{
    return objs_.at(i);
}
 
bool scene::intersect(const ray& r)
{
    Ray s;
    s.direction = r.direction;
    s.origin = r.origin;
    s.dist = r.distance;
    s.trace_depth = r.depth;
 
    return accel_->intersect(s);
}
 
bool scene::intersect(const ray& r, hit_info& hi)
{
    Ray s;
    s.direction = r.direction;
    s.origin = r.origin;
    s.trace_depth = r.depth;
    s.dist = r.distance;
 
    bool hit = accel_->intersect(s);
 
    if (hit)
    {
        //find the mesh which was hit
        const mesh& msh = dynamic_cast<const mesh&>(*objs_[s.id]);
 
        //set hit position and normals
        hi.distance = s.dist;
        hi.position = s.hit_pos;
        hi.shading_normal = s.hit_normal;
 
        const IndexedFaceSet& geometry = msh.get_trimesh().geometry;
        const Vec3i& f  = geometry.face(s.hit_face_id);
        const Vec3f p0 = geometry.vertex(f[0]);
        const Vec3f a  = geometry.vertex(f[1]) - p0;
        const Vec3f b  = geometry.vertex(f[2]) - p0;
        Vec3f face_normal = cross(a,b);
        float l = sqr_length(face_normal);
        if(l > 0.0f)
            face_normal /= sqrt(l);
 
        hi.geometric_normal = msh.transform().mul_3D_vector(face_normal);
        //hi.geometric_normal = msh.triangle_normal(s.hit_face_id);
 
        hi.inside = dot(hi.geometric_normal, -s.direction) < 0.f;
        if (hi.inside)
        {
            hi.geometric_normal = -hi.geometric_normal;
            hi.shading_normal = -hi.shading_normal;
        }
 
        bool has_texcoords = msh.get_trimesh().texcoords.no_faces() > 0;
 
        if (has_texcoords)
        {
            //compute texture coords..
            const IndexedFaceSet& texcoords = msh.get_trimesh().texcoords;
            const Vec3i& t = texcoords.face(s.hit_face_id);
            Vec3f uv0 = texcoords.vertex(t[0]);
            Vec3f uv1 = texcoords.vertex(t[1]);
            Vec3f uv2 = texcoords.vertex(t[2]);
            hi.texcoords(0) = (1.f-s.u-s.v)*uv0(0) + s.u*uv1(0) + s.v*uv2(0);
            hi.texcoords(1) = (1.f-s.u-s.v)*uv0(1) + s.u*uv1(1) + s.v*uv2(1);
 
            //compute tangent vectors..
 
 
        }
        else
        {
            //hmm.. no uvs. not good!
            hi.texcoords = Vec2f(0.f);
            //orthogonal(hi.shading_normal, hi.tangent, hi.bitangent);
        }
 
        //sample material to get bsdf
        const material* mat = msh.get_material();
        if (mat)
            mat->sample(r, hi);
 
        //convert from radiant exitance to radiance
        if (!hi.inside)
            hi.emitted = msh.exitance() / float(M_PI);
    }
 
    return hit;
}
 
const camera* scene::get_camera(void) const
{
    return cam_;
}
 
void scene::set_camera(const camera* c)
{
    cam_ = c;
}
 
void scene::initialize(int mo, int md)
{
    assert(cam_);
    assert(!objs_.empty());
 
    std::vector<const TriMesh*> msh;
    std::vector<Mat4x4f> trs;
 
    std::vector<const luminaire*>::iterator it;
    for (it=objs_.begin(); it!=objs_.end(); ++it)
    {
        const luminaire* obj = *it;
 
        const mesh* mxx = dynamic_cast<const mesh*>(obj);
        if (!mxx)
            continue;
 
        msh.push_back(&mxx->get_trimesh());
        trs.push_back(mxx->transform());
    }
 
    accel_->init(msh, trs, mo, md);
    accel_->build();
}
 
//02566 framework, Anders Wang Kristensen, awk@imm.dtu.dk, 2007
 
 
Generated by GNU Enscript 1.6.6.
 
 
 

Rev 350	Rev 353
1	`#include <Geometry/BSPTree.h>`	1	`#include <Geometry/BSPTree.h>`
2		2
3	`#include "scene.hpp"`	3	`#include "scene.h"`
4	`#include "mesh.hpp"`	4	`#include "mesh.h"`
5		5
6	`using namespace Geometry;`	6	`using namespace Geometry;`
7	`using namespace CGLA;`	7	`using namespace CGLA;`
8		8
9	`scene::scene(void) : cam_(0)`	9	`scene::scene(void) : cam_(0)`
10	`{`	10	`{`
11	`accel_ = new BSPTree;`	11	`accel_ = new BSPTree;`
12	`}`	12	`}`
13		13
14	`scene::~scene(void)`	14	`scene::~scene(void)`
15	`{`	15	`{`
16	`delete accel_;`	16	`delete accel_;`
17	`}`	17	`}`
18		18
19	`void scene::insert(const luminaire* obj)`	19	`void scene::insert(const luminaire* obj)`
20	`{`	20	`{`
21	`objs_.push_back(obj);`	21	`objs_.push_back(obj);`
22	`}`	22	`}`
23		23
24	`size_t scene::luminaires(void) const`	24	`size_t scene::luminaires(void) const`
25	`{`	25	`{`
26	`return objs_.size();`	26	`return objs_.size();`
27	`}`	27	`}`
28		28
29	`const luminaire* scene::get_luminaire(size_t i) const`	29	`const luminaire* scene::get_luminaire(size_t i) const`
30	`{`	30	`{`
31	`return objs_.at(i);`	31	`return objs_.at(i);`
32	`}`	32	`}`
33		33
34	`bool scene::intersect(const ray& r)`	34	`bool scene::intersect(const ray& r)`
35	`{`	35	`{`
36	`Ray s;`	36	`Ray s;`
37	`s.direction = r.direction;`	37	`s.direction = r.direction;`
38	`s.origin = r.origin;`	38	`s.origin = r.origin;`
39	`s.dist = r.distance;`	39	`s.dist = r.distance;`
40	`s.trace_depth = r.depth;`	40	`s.trace_depth = r.depth;`
41		41
42	`return accel_->intersect(s);`	42	`return accel_->intersect(s);`
43	`}`	43	`}`
44		44
45	`bool scene::intersect(const ray& r, hit_info& hi)`	45	`bool scene::intersect(const ray& r, hit_info& hi)`
46	`{`	46	`{`
47	`Ray s;`	47	`Ray s;`
48	`s.direction = r.direction;`	48	`s.direction = r.direction;`
49	`s.origin = r.origin;`	49	`s.origin = r.origin;`
50	`s.trace_depth = r.depth;`	50	`s.trace_depth = r.depth;`
51	`s.dist = r.distance;`	51	`s.dist = r.distance;`
52		52
53	`bool hit = accel_->intersect(s);`	53	`bool hit = accel_->intersect(s);`
54		54
55	`if (hit)`	55	`if (hit)`
56	`{`	56	`{`
57	`//find the mesh which was hit`	57	`//find the mesh which was hit`
58	`const mesh& msh = dynamic_cast<const mesh&>(*objs_[s.id]);`	58	`const mesh& msh = dynamic_cast<const mesh&>(*objs_[s.id]);`
59		59
60	`//set hit position and normals`	60	`//set hit position and normals`
61	`hi.distance = s.dist;`	61	`hi.distance = s.dist;`
62	`hi.position = s.hit_pos;`	62	`hi.position = s.hit_pos;`
63	`hi.shading_normal = s.hit_normal;`	63	`hi.shading_normal = s.hit_normal;`
64		64
65	`const IndexedFaceSet& geometry = msh.get_trimesh().geometry;`	65	`const IndexedFaceSet& geometry = msh.get_trimesh().geometry;`
66	`const Vec3i& f = geometry.face(s.hit_face_id);`	66	`const Vec3i& f = geometry.face(s.hit_face_id);`
67	`const Vec3f p0 = geometry.vertex(f[0]);`	67	`const Vec3f p0 = geometry.vertex(f[0]);`
68	`const Vec3f a = geometry.vertex(f[1]) - p0;`	68	`const Vec3f a = geometry.vertex(f[1]) - p0;`
69	`const Vec3f b = geometry.vertex(f[2]) - p0;`	69	`const Vec3f b = geometry.vertex(f[2]) - p0;`
70	`Vec3f face_normal = cross(a,b);`	70	`Vec3f face_normal = cross(a,b);`
71	`float l = sqr_length(face_normal);`	71	`float l = sqr_length(face_normal);`
72	`if(l > 0.0f)`	72	`if(l > 0.0f)`
73	`face_normal /= sqrt(l);`	73	`face_normal /= sqrt(l);`
74		74
75	`hi.geometric_normal = msh.transform().mul_3D_vector(face_normal);`	75	`hi.geometric_normal = msh.transform().mul_3D_vector(face_normal);`
76	`//hi.geometric_normal = msh.triangle_normal(s.hit_face_id);`	76	`//hi.geometric_normal = msh.triangle_normal(s.hit_face_id);`
77		77
78	`hi.inside = dot(hi.geometric_normal, -s.direction) < 0.f;`	78	`hi.inside = dot(hi.geometric_normal, -s.direction) < 0.f;`
79	`if (hi.inside)`	79	`if (hi.inside)`
80	`{`	80	`{`
81	`hi.geometric_normal = -hi.geometric_normal;`	81	`hi.geometric_normal = -hi.geometric_normal;`
82	`hi.shading_normal = -hi.shading_normal;`	82	`hi.shading_normal = -hi.shading_normal;`
83	`}`	83	`}`
84		84
85	`bool has_texcoords = msh.get_trimesh().texcoords.no_faces() > 0;`	85	`bool has_texcoords = msh.get_trimesh().texcoords.no_faces() > 0;`
86		86
87	`if (has_texcoords)`	87	`if (has_texcoords)`
88	`{`	88	`{`
89	`//compute texture coords..`	89	`//compute texture coords..`
90	`const IndexedFaceSet& texcoords = msh.get_trimesh().texcoords;`	90	`const IndexedFaceSet& texcoords = msh.get_trimesh().texcoords;`
91	`const Vec3i& t = texcoords.face(s.hit_face_id);`	91	`const Vec3i& t = texcoords.face(s.hit_face_id);`
92	`Vec3f uv0 = texcoords.vertex(t[0]);`	92	`Vec3f uv0 = texcoords.vertex(t[0]);`
93	`Vec3f uv1 = texcoords.vertex(t[1]);`	93	`Vec3f uv1 = texcoords.vertex(t[1]);`
94	`Vec3f uv2 = texcoords.vertex(t[2]);`	94	`Vec3f uv2 = texcoords.vertex(t[2]);`
95	`hi.texcoords(0) = (1.f-s.u-s.v)uv0(0) + s.uuv1(0) + s.v*uv2(0);`	95	`hi.texcoords(0) = (1.f-s.u-s.v)uv0(0) + s.uuv1(0) + s.v*uv2(0);`
96	`hi.texcoords(1) = (1.f-s.u-s.v)uv0(1) + s.uuv1(1) + s.v*uv2(1);`	96	`hi.texcoords(1) = (1.f-s.u-s.v)uv0(1) + s.uuv1(1) + s.v*uv2(1);`
97		97
98	`//compute tangent vectors..`	98	`//compute tangent vectors..`
99		99
100		100
101	`}`	101	`}`
102	`else`	102	`else`
103	`{`	103	`{`
104	`//hmm.. no uvs. not good!`	104	`//hmm.. no uvs. not good!`
105	`hi.texcoords = Vec2f(0.f);`	105	`hi.texcoords = Vec2f(0.f);`
106	`//orthogonal(hi.shading_normal, hi.tangent, hi.bitangent);`	106	`//orthogonal(hi.shading_normal, hi.tangent, hi.bitangent);`
107	`}`	107	`}`
108		108
109	`//sample material to get bsdf`	109	`//sample material to get bsdf`
110	`const material* mat = msh.get_material();`	110	`const material* mat = msh.get_material();`
111	`if (mat)`	111	`if (mat)`
112	`mat->sample(r, hi);`	112	`mat->sample(r, hi);`
113		113
114	`//convert from radiant exitance to radiance`	114	`//convert from radiant exitance to radiance`
115	`if (!hi.inside)`	115	`if (!hi.inside)`
116	`hi.emitted = msh.exitance() / float(M_PI);`	116	`hi.emitted = msh.exitance() / float(M_PI);`
117	`}`	117	`}`
118		118
119	`return hit;`	119	`return hit;`
120	`}`	120	`}`
121		121
122	`const camera* scene::get_camera(void) const`	122	`const camera* scene::get_camera(void) const`
123	`{`	123	`{`
124	`return cam_;`	124	`return cam_;`
125	`}`	125	`}`
126		126
127	`void scene::set_camera(const camera* c)`	127	`void scene::set_camera(const camera* c)`
128	`{`	128	`{`
129	`cam_ = c;`	129	`cam_ = c;`
130	`}`	130	`}`
131		131
132	`void scene::initialize(int mo, int md)`	132	`void scene::initialize(int mo, int md)`
133	`{`	133	`{`
134	`assert(cam_);`	134	`assert(cam_);`
135	`assert(!objs_.empty());`	135	`assert(!objs_.empty());`
136		136
137	`std::vector<const TriMesh*> msh;`	137	`std::vector<const TriMesh*> msh;`
138	`std::vector<Mat4x4f> trs;`	138	`std::vector<Mat4x4f> trs;`
139		139
140	`std::vector<const luminaire*>::iterator it;`	140	`std::vector<const luminaire*>::iterator it;`
141	`for (it=objs_.begin(); it!=objs_.end(); ++it)`	141	`for (it=objs_.begin(); it!=objs_.end(); ++it)`
142	`{`	142	`{`
143	`const luminaire* obj = *it;`	143	`const luminaire* obj = *it;`
144		144
145	`const mesh* mxx = dynamic_cast<const mesh*>(obj);`	145	`const mesh* mxx = dynamic_cast<const mesh*>(obj);`
146	`if (!mxx)`	146	`if (!mxx)`
147	`continue;`	147	`continue;`
148		148
149	`msh.push_back(&mxx->get_trimesh());`	149	`msh.push_back(&mxx->get_trimesh());`
150	`trs.push_back(mxx->transform());`	150	`trs.push_back(mxx->transform());`
151	`}`	151	`}`
152		152
153	`accel_->init(msh, trs, mo, md);`	153	`accel_->init(msh, trs, mo, md);`
154	`accel_->build();`	154	`accel_->build();`
155	`}`	155	`}`
156		156
157	`//02566 framework, Anders Wang Kristensen, awk@imm.dtu.dk, 2007`	157	`//02566 framework, Anders Wang Kristensen, awk@imm.dtu.dk, 2007`
158		158
159		159
160	`Generated by GNU Enscript 1.6.6.`	160	`Generated by GNU Enscript 1.6.6.`
161		161
162		162
163		163

Subversion Repositories gelsvn

(root)/trunk/apps/Pathtracer/scene.cpp – Rev 350 → 353