clray

view src/scene.h @ 27:8b2f2ad14ae7

find changesets by author, revision, files, or words in the commit message
semi-fixed the kdtree construction
author John Tsiombikas <nuclear@member.fsf.org>
date Tue, 17 Aug 2010 20:35:00 +0100
parents c740ae431d51
children 97cfd9675310
line source
1 #ifndef MESH_H_
2 #define MESH_H_
3
4 #include <stdio.h>
5 #include <vector>
6 #include <list>
7
8 struct Vertex {
9 float pos[4];
10 float normal[4];
11 float tex[4];
12 float padding[4];
13 };
14
15 struct Face {
16 Vertex v[3];
17 float normal[4];
18 int matid;
19 int padding[3];
20
21 bool operator ==(const Face &f) const;
22 };
23
24 struct Material {
25 float kd[4], ks[4];
26 float kr, kt;
27 float spow;
28 float padding;
29 };
30
31 struct Mesh {
32 std::vector<Face> faces;
33 int matid;
34 };
35
36 class AABBox {
37 public:
38 float min[4], max[4];
39
40 float calc_surface_area() const;
41 };
42
43 enum {
44 KDAXIS_X,
45 KDAXIS_Y,
46 KDAXIS_Z
47 };
48
49 #define KDCLEAR(node) ((node)->axis = -1)
50 #define KDUSED(node) ((node)->axis >= 0)
51 #define KDPARENT(x) ((x) >> 1)
52 #define KDLEFT(x) ((x) << 1)
53 #define KDRIGHT(x) (((x) << 1) + 1)
54
55 struct KDNode {
56 int axis;
57 float pt;
58 AABBox aabb;
59 float cost;
60
61 KDNode *left, *right;
62 int num_faces; // cause on some implementations list::size() is O(n)
63 std::list<const Face*> faces;
64
65 KDNode();
66 };
67
68 struct KDNodeGPU {
69 int axis;
70 float pt;
71 };
72
73
74 class Scene {
75 private:
76 mutable Face *facebuf;
77 mutable int num_faces;
78
79 public:
80 std::vector<Mesh*> meshes;
81 std::vector<Material> matlib;
82
83 KDNode *kdtree;
84 std::vector<KDNode> kdtree_gpu;
85
86 Scene();
87 ~Scene();
88
89 bool add_mesh(Mesh *m);
90 int get_num_meshes() const;
91 int get_num_materials() const;
92 int get_num_faces() const;
93
94 Material *get_materials();
95 const Material *get_materials() const;
96
97 bool load(const char *fname);
98 bool load(FILE *fp);
99
100 const Face *get_face_buffer() const;
101
102 void draw_kdtree() const;
103 bool build_kdtree();
104 };
105
106 enum {
107 ACCEL_PARAM_MAX_TREE_DEPTH,
108 ACCEL_PARAM_MAX_NODE_ITEMS,
109 ACCEL_PARAM_COST_TRAVERSE,
110 ACCEL_PARAM_COST_INTERSECT,
111
112 NUM_ACCEL_PARAMS
113 };
114
115 void set_accel_param(int p, int v);
116
117 #endif /* MESH_H_ */