forked from LeenkxTeam/LNXSDK
		
	
		
			
				
	
	
		
			306 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			306 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #ifndef GIM_QUANTIZED_SET_H_INCLUDED
 | |
| #define GIM_QUANTIZED_SET_H_INCLUDED
 | |
| 
 | |
| /*! \file btGImpactQuantizedBvh.h
 | |
| \author Francisco Leon Najera
 | |
| */
 | |
| /*
 | |
| This source file is part of GIMPACT Library.
 | |
| 
 | |
| For the latest info, see http://gimpact.sourceforge.net/
 | |
| 
 | |
| Copyright (c) 2007 Francisco Leon Najera. C.C. 80087371.
 | |
| email: projectileman@yahoo.com
 | |
| 
 | |
| 
 | |
| This software is provided 'as-is', without any express or implied warranty.
 | |
| In no event will the authors be held liable for any damages arising from the use of this software.
 | |
| Permission is granted to anyone to use this software for any purpose,
 | |
| including commercial applications, and to alter it and redistribute it freely,
 | |
| subject to the following restrictions:
 | |
| 
 | |
| 1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
 | |
| 2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
 | |
| 3. This notice may not be removed or altered from any source distribution.
 | |
| */
 | |
| 
 | |
| #include "btGImpactBvh.h"
 | |
| #include "btQuantization.h"
 | |
| #include "btGImpactQuantizedBvhStructs.h"
 | |
| 
 | |
| class GIM_QUANTIZED_BVH_NODE_ARRAY:public btAlignedObjectArray<BT_QUANTIZED_BVH_NODE>
 | |
| {
 | |
| };
 | |
| 
 | |
| 
 | |
| 
 | |
| 
 | |
| //! Basic Box tree structure
 | |
| class btQuantizedBvhTree
 | |
| {
 | |
| protected:
 | |
| 	int m_num_nodes;
 | |
| 	GIM_QUANTIZED_BVH_NODE_ARRAY m_node_array;
 | |
| 	btAABB m_global_bound;
 | |
| 	btVector3 m_bvhQuantization;
 | |
| protected:
 | |
| 	void calc_quantization(GIM_BVH_DATA_ARRAY & primitive_boxes, btScalar boundMargin = btScalar(1.0) );
 | |
| 
 | |
| 	int _sort_and_calc_splitting_index(
 | |
| 		GIM_BVH_DATA_ARRAY & primitive_boxes,
 | |
| 		 int startIndex,  int endIndex, int splitAxis);
 | |
| 
 | |
| 	int _calc_splitting_axis(GIM_BVH_DATA_ARRAY & primitive_boxes, int startIndex,  int endIndex);
 | |
| 
 | |
| 	void _build_sub_tree(GIM_BVH_DATA_ARRAY & primitive_boxes, int startIndex,  int endIndex);
 | |
| public:
 | |
| 	btQuantizedBvhTree()
 | |
| 	{
 | |
| 		m_num_nodes = 0;
 | |
| 	}
 | |
| 
 | |
| 	//! prototype functions for box tree management
 | |
| 	//!@{
 | |
| 	void build_tree(GIM_BVH_DATA_ARRAY & primitive_boxes);
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void quantizePoint(
 | |
| 		unsigned short * quantizedpoint, const btVector3 & point) const
 | |
| 	{
 | |
| 		bt_quantize_clamp(quantizedpoint,point,m_global_bound.m_min,m_global_bound.m_max,m_bvhQuantization);
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 	SIMD_FORCE_INLINE bool testQuantizedBoxOverlapp(
 | |
| 		int node_index,
 | |
| 		unsigned short * quantizedMin,unsigned short * quantizedMax) const
 | |
| 	{
 | |
| 		return m_node_array[node_index].testQuantizedBoxOverlapp(quantizedMin,quantizedMax);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void clearNodes()
 | |
| 	{
 | |
| 		m_node_array.clear();
 | |
| 		m_num_nodes = 0;
 | |
| 	}
 | |
| 
 | |
| 	//! node count
 | |
| 	SIMD_FORCE_INLINE int getNodeCount() const
 | |
| 	{
 | |
| 		return m_num_nodes;
 | |
| 	}
 | |
| 
 | |
| 	//! tells if the node is a leaf
 | |
| 	SIMD_FORCE_INLINE bool isLeafNode(int nodeindex) const
 | |
| 	{
 | |
| 		return m_node_array[nodeindex].isLeafNode();
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getNodeData(int nodeindex) const
 | |
| 	{
 | |
| 		return m_node_array[nodeindex].getDataIndex();
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void getNodeBound(int nodeindex, btAABB & bound) const
 | |
| 	{
 | |
| 		bound.m_min = bt_unquantize(
 | |
| 			m_node_array[nodeindex].m_quantizedAabbMin,
 | |
| 			m_global_bound.m_min,m_bvhQuantization);
 | |
| 
 | |
| 		bound.m_max = bt_unquantize(
 | |
| 			m_node_array[nodeindex].m_quantizedAabbMax,
 | |
| 			m_global_bound.m_min,m_bvhQuantization);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void setNodeBound(int nodeindex, const btAABB & bound)
 | |
| 	{
 | |
| 		bt_quantize_clamp(	m_node_array[nodeindex].m_quantizedAabbMin,
 | |
| 							bound.m_min,
 | |
| 							m_global_bound.m_min,
 | |
| 							m_global_bound.m_max,
 | |
| 							m_bvhQuantization);
 | |
| 
 | |
| 		bt_quantize_clamp(	m_node_array[nodeindex].m_quantizedAabbMax,
 | |
| 							bound.m_max,
 | |
| 							m_global_bound.m_min,
 | |
| 							m_global_bound.m_max,
 | |
| 							m_bvhQuantization);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getLeftNode(int nodeindex) const
 | |
| 	{
 | |
| 		return nodeindex+1;
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getRightNode(int nodeindex) const
 | |
| 	{
 | |
| 		if(m_node_array[nodeindex+1].isLeafNode()) return nodeindex+2;
 | |
| 		return nodeindex+1 + m_node_array[nodeindex+1].getEscapeIndex();
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getEscapeNodeIndex(int nodeindex) const
 | |
| 	{
 | |
| 		return m_node_array[nodeindex].getEscapeIndex();
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE const BT_QUANTIZED_BVH_NODE * get_node_pointer(int index = 0) const
 | |
| 	{
 | |
| 		return &m_node_array[index];
 | |
| 	}
 | |
| 
 | |
| 	//!@}
 | |
| };
 | |
| 
 | |
| 
 | |
| 
 | |
| //! Structure for containing Boxes
 | |
| /*!
 | |
| This class offers an structure for managing a box tree of primitives.
 | |
| Requires a Primitive prototype (like btPrimitiveManagerBase )
 | |
| */
 | |
| class btGImpactQuantizedBvh
 | |
| {
 | |
| protected:
 | |
| 	btQuantizedBvhTree m_box_tree;
 | |
| 	btPrimitiveManagerBase * m_primitive_manager;
 | |
| 
 | |
| protected:
 | |
| 	//stackless refit
 | |
| 	void refit();
 | |
| public:
 | |
| 
 | |
| 	//! this constructor doesn't build the tree. you must call	buildSet
 | |
| 	btGImpactQuantizedBvh()
 | |
| 	{
 | |
| 		m_primitive_manager = NULL;
 | |
| 	}
 | |
| 
 | |
| 	//! this constructor doesn't build the tree. you must call	buildSet
 | |
| 	btGImpactQuantizedBvh(btPrimitiveManagerBase * primitive_manager)
 | |
| 	{
 | |
| 		m_primitive_manager = primitive_manager;
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE btAABB getGlobalBox()  const
 | |
| 	{
 | |
| 		btAABB totalbox;
 | |
| 		getNodeBound(0, totalbox);
 | |
| 		return totalbox;
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void setPrimitiveManager(btPrimitiveManagerBase * primitive_manager)
 | |
| 	{
 | |
| 		m_primitive_manager = primitive_manager;
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE btPrimitiveManagerBase * getPrimitiveManager() const
 | |
| 	{
 | |
| 		return m_primitive_manager;
 | |
| 	}
 | |
| 
 | |
| 
 | |
| //! node manager prototype functions
 | |
| ///@{
 | |
| 
 | |
| 	//! this attemps to refit the box set.
 | |
| 	SIMD_FORCE_INLINE void update()
 | |
| 	{
 | |
| 		refit();
 | |
| 	}
 | |
| 
 | |
| 	//! this rebuild the entire set
 | |
| 	void buildSet();
 | |
| 
 | |
| 	//! returns the indices of the primitives in the m_primitive_manager
 | |
| 	bool boxQuery(const btAABB & box, btAlignedObjectArray<int> & collided_results) const;
 | |
| 
 | |
| 	//! returns the indices of the primitives in the m_primitive_manager
 | |
| 	SIMD_FORCE_INLINE bool boxQueryTrans(const btAABB & box,
 | |
| 		 const btTransform & transform, btAlignedObjectArray<int> & collided_results) const
 | |
| 	{
 | |
| 		btAABB transbox=box;
 | |
| 		transbox.appy_transform(transform);
 | |
| 		return boxQuery(transbox,collided_results);
 | |
| 	}
 | |
| 
 | |
| 	//! returns the indices of the primitives in the m_primitive_manager
 | |
| 	bool rayQuery(
 | |
| 		const btVector3 & ray_dir,const btVector3 & ray_origin ,
 | |
| 		btAlignedObjectArray<int> & collided_results) const;
 | |
| 
 | |
| 	//! tells if this set has hierarcht
 | |
| 	SIMD_FORCE_INLINE bool hasHierarchy() const
 | |
| 	{
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	//! tells if this set is a trimesh
 | |
| 	SIMD_FORCE_INLINE bool isTrimesh()  const
 | |
| 	{
 | |
| 		return m_primitive_manager->is_trimesh();
 | |
| 	}
 | |
| 
 | |
| 	//! node count
 | |
| 	SIMD_FORCE_INLINE int getNodeCount() const
 | |
| 	{
 | |
| 		return m_box_tree.getNodeCount();
 | |
| 	}
 | |
| 
 | |
| 	//! tells if the node is a leaf
 | |
| 	SIMD_FORCE_INLINE bool isLeafNode(int nodeindex) const
 | |
| 	{
 | |
| 		return m_box_tree.isLeafNode(nodeindex);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getNodeData(int nodeindex) const
 | |
| 	{
 | |
| 		return m_box_tree.getNodeData(nodeindex);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void getNodeBound(int nodeindex, btAABB & bound)  const
 | |
| 	{
 | |
| 		m_box_tree.getNodeBound(nodeindex, bound);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void setNodeBound(int nodeindex, const btAABB & bound)
 | |
| 	{
 | |
| 		m_box_tree.setNodeBound(nodeindex, bound);
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getLeftNode(int nodeindex) const
 | |
| 	{
 | |
| 		return m_box_tree.getLeftNode(nodeindex);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getRightNode(int nodeindex) const
 | |
| 	{
 | |
| 		return m_box_tree.getRightNode(nodeindex);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE int getEscapeNodeIndex(int nodeindex) const
 | |
| 	{
 | |
| 		return m_box_tree.getEscapeNodeIndex(nodeindex);
 | |
| 	}
 | |
| 
 | |
| 	SIMD_FORCE_INLINE void getNodeTriangle(int nodeindex,btPrimitiveTriangle & triangle) const
 | |
| 	{
 | |
| 		m_primitive_manager->get_primitive_triangle(getNodeData(nodeindex),triangle);
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 	SIMD_FORCE_INLINE const BT_QUANTIZED_BVH_NODE * get_node_pointer(int index = 0) const
 | |
| 	{
 | |
| 		return m_box_tree.get_node_pointer(index);
 | |
| 	}
 | |
| 
 | |
| #ifdef TRI_COLLISION_PROFILING
 | |
| 	static float getAverageTreeCollisionTime();
 | |
| #endif //TRI_COLLISION_PROFILING
 | |
| 
 | |
| 	static void find_collision(const btGImpactQuantizedBvh * boxset1, const btTransform & trans1,
 | |
| 		const btGImpactQuantizedBvh * boxset2, const btTransform & trans2,
 | |
| 		btPairSet & collision_pairs);
 | |
| };
 | |
| 
 | |
| #endif // GIM_BOXPRUNING_H_INCLUDED
 |