pyrit: comparison src/kdtree.cc

equal deleted inserted replaced

-:fc18ac4833f2
+:a0a3e334744f
 };
 class SortableShapeList: public vector<SortableShape>
 {
 public:
-	SortableShapeList(ShapeList &shapes, short axis)
+	SortableShapeList(ShapeList *shapes, short axis)
 	{
 		ShapeList::iterator shape;
-		for (shape = shapes.begin(); shape != shapes.end(); shape++)
+		for (shape = shapes->begin(); shape != shapes->end(); shape++)
 			push_back(SortableShape(*shape, axis));
 	};
 };
 class SplitPos
 	return nearest_shape;
 }
 void KdNode::subdivide(BBox bbox, int depth)
 {
-	if (depth >= 20 || shapes.size() <= 4)
+	if (depth >= 20 || shapes->size() <= 4)
 	{
-		leaf = true;
+		setLeaf();
 		return;
 	}
 	// choose split axis
 	axis = 0;
 	}
 	// choose best split pos
 	const float K = 1.4; // constant, K = cost of traversal / cost of ray-triangle intersection
 	float SAV = 2*(bbox.w()*bbox.h() + bbox.w()*bbox.d() + bbox.h()*bbox.d()); // surface area of node
-	float cost = SAV * (K + shapes.size()); // initial cost = non-split cost
+	float cost = SAV * (K + shapes->size()); // initial cost = non-split cost
 	SplitPos *splitpos = NULL;
-	leaf = true;
+	bool leaf = true;
 	BBox lbb = bbox;
 	BBox rbb = bbox;
 	for (spl = splitlist.begin(); spl != splitlist.end(); spl++)
 	{
 		// calculate SAH cost of this split
 			splitpos = &*spl;
 		}
 	}
 	if (leaf)
+	{
+		setLeaf();
 		return;
+	}
-#if 1
+#if 0
 // export kd-tree as .obj for visualization
 // this would be hard to reconstruct later
 	static ofstream F("kdtree.obj");
 	Vector3 v;
 	static int f=0;
 }
 void KdTree::build()
 {
 	root = new KdNode();
-	root->shapes = shapes;
+	root->shapes = &shapes;
 	root->subdivide(bbox, 0);
 	built = true;
 }
 /* algorithm by Vlastimil Havran, Heuristic Ray Shooting Algorithms, appendix C */
 	enPt->t = a; /* set the signed distance */
 	enPt->node = NULL;
 	/* distinguish between internal and external origin */
 	if (a >= 0.0) /* a ray with external origin */
-		enPt->pb = ray.a + ray.dir * a;
+		enPt->pb = ray.o + ray.dir * a;
 	else /* a ray with internal origin */
-		enPt->pb = ray.a;
+		enPt->pb = ray.o;
 	/* setup initial exit point in the stack */
 	StackElem *exPt = new StackElem();
 	exPt->t = b;
-	exPt->pb = ray.a + ray.dir * b;
+	exPt->pb = ray.o + ray.dir * b;
 	exPt->node = NULL; /* set termination ﬂag */
 	st.push(enPt);
 	/* loop, traverse through the whole kd-tree, until an object is intersected or ray leaves the scene */
 			}
 			/* case P4 or N4 . . . traverse both children */
 			/* signed distance to the splitting plane */
-			t = (splitVal - ray.a.cell[axis]) / ray.dir.cell[axis];
+			t = (splitVal - ray.o.cell[axis]) / ray.dir.cell[axis];
 			/* setup the new exit point */
 			st.push(exPt);
 			exPt = new StackElem();
 			/* push values onto the stack */
 			exPt->t = t;
 			exPt->node = farchild;
 			exPt->pb[axis] = splitVal;
-			exPt->pb[(axis+1)%3] = ray.a.cell[(axis+1)%3] + t * ray.dir.cell[(axis+1)%3];
+			exPt->pb[(axis+1)%3] = ray.o.cell[(axis+1)%3] + t * ray.dir.cell[(axis+1)%3];
-			exPt->pb[(axis+2)%3] = ray.a.cell[(axis+2)%3] + t * ray.dir.cell[(axis+2)%3];
+			exPt->pb[(axis+2)%3] = ray.o.cell[(axis+2)%3] + t * ray.dir.cell[(axis+2)%3];
 		} /* while */
 		/* current node is the leaf . . . empty or full */
 		/* "intersect ray with each object in the object list, discarding "
 		"those lying before stack[enPt].t or farther than stack[exPt].t" */
 		Shape *nearest_shape = NULL;
 		float dist = exPt->t;
 		ShapeList::iterator shape;
-		for (shape = node->shapes.begin(); shape != node->shapes.end(); shape++)
+		for (shape = node->shapes->begin(); shape != node->shapes->end(); shape++)
 			if (*shape != origin_shape && (*shape)->intersect(ray, dist)
 			&& dist >= enPt->t)
 				nearest_shape = *shape;
 		if (nearest_shape)
 	if (!built)
 		return;
 	if (node == NULL)
 		node = root;
 	if (node->isLeaf())
-		str << "(leaf: " << node->shapes.size() << " shapes)";
+		str << "(leaf: " << node->shapes->size() << " shapes)";
 	else
 	{
 		str << "(split " << (char)('x'+node->getAxis()) << " at " << node->getSplit() << "; L=";
 		save(str, node->getLeftChild());
 		str << "; R=";

branch	pyrit
changeset 15	a0a3e334744f
parent 14	fc18ac4833f2
child 16	20bceb605f48