diff options
Diffstat (limited to 'src/shared/vmap/ModelContainer.cpp')
| -rw-r--r-- | src/shared/vmap/ModelContainer.cpp | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/src/shared/vmap/ModelContainer.cpp b/src/shared/vmap/ModelContainer.cpp index 94d453079d2..4a722d2585b 100644 --- a/src/shared/vmap/ModelContainer.cpp +++ b/src/shared/vmap/ModelContainer.cpp @@ -17,12 +17,17 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ + #include <iostream> #include <fstream> + #include <string.h> + #include "ModelContainer.h" #include "VMapDefinitions.h" + using namespace G3D; + namespace VMAP { //========================================================== @@ -34,25 +39,32 @@ namespace VMAP return (pMc.getBasePosition() * pMc.getNTriangles()).hashCode(); } //========================================================== + ModelContainer::ModelContainer(unsigned int pNTriangles, unsigned int pNNodes, unsigned int pNSubModel) : BaseModel(pNNodes, pNTriangles) { + iNSubModel = pNSubModel; iSubModel = 0; if(pNSubModel > 0) iSubModel = new SubModel[iNSubModel]; } + //========================================================== + bool ModelContainer::operator==(const ModelContainer& pMc2) const { if (this->iNSubModel == 0 && pMc2.iNSubModel == 0 && this->iSubModel == 0 && pMc2.iSubModel == 0) return true; return this == &pMc2; } + //========================================================== + void ModelContainer::countSubModelsAndNodesAndTriangles(AABSPTree<SubModel *>::Node& pNode, int& nSubModels, int& nNodes, int& nTriangles) { // For this node we will need a TreeNode as well as for the internal nodes ++nNodes; + nSubModels += pNode.valueArray.size(); for(int i=0;i<pNode.valueArray.size(); i++) { @@ -62,6 +74,7 @@ namespace VMAP nNodes += m->getNNodes(); nTriangles += m->getNTriangles(); } + if(pNode.child[0] != 0) { countSubModelsAndNodesAndTriangles(*pNode.child[0], nSubModels, nNodes, nTriangles); @@ -72,6 +85,7 @@ namespace VMAP } } //========================================================== + void ModelContainer::fillContainer(const AABSPTree<SubModel *>::Node& pNode, int &pSubModelPos, int &pTreeNodePos, int &pTrianglePos, Vector3& pLo, Vector3& pHi, Vector3& pFinalLo, Vector3& pFinalHi) { // TreeNode for the SubModel @@ -79,20 +93,26 @@ namespace VMAP treeNode.setSplitAxis(pNode.splitAxis); treeNode.setSplitLocation(pNode.splitLocation); int currentTreeNodePos = pTreeNodePos++; + Vector3 lo = Vector3(inf(),inf(),inf()); Vector3 hi = Vector3(-inf(),-inf(),-inf()); + for(int i=0;i<pNode.valueArray.size(); i++) { G3D::_AABSPTree::Handle<SubModel*>* h= pNode.valueArray[i]; SubModel *m = h->value; + memcpy(&getTreeNodes()[pTreeNodePos], &m->getTreeNode(0), sizeof(TreeNode) * m->getNNodes()); memcpy(&getTriangles()[pTrianglePos], &m->getTriangle(0), sizeof(TriangleBox) * m->getNTriangles()); + SubModel newModel = SubModel(m->getNTriangles(), getTriangles(), pTrianglePos, m->getNNodes(), getTreeNodes(), pTreeNodePos); newModel.setReletiveBounds(m->getReletiveBounds().getLo(), m->getReletiveBounds().getHi()); newModel.setBasePosition(m->getBasePosition()); iSubModel[pSubModelPos++] = newModel; + pTreeNodePos += m->getNNodes(); pTrianglePos += m->getNTriangles(); + AABox box = m->getAABoxBounds(); lo = lo.min(box.low()); hi = hi.max(box.high()); @@ -105,6 +125,7 @@ namespace VMAP } */ // get absolute bounds + if(pNode.child[0] != 0) { treeNode.setChildPos(0, pTreeNodePos); @@ -115,45 +136,62 @@ namespace VMAP treeNode.setChildPos(1, pTreeNodePos); fillContainer(*pNode.child[1], pSubModelPos, pTreeNodePos, pTrianglePos, lo, hi,pFinalLo,pFinalHi); } + pLo = pLo.min(lo); pHi = pHi.max(hi); + treeNode.setBounds(lo,hi); + setTreeNode(treeNode, currentTreeNodePos); + } + //========================================================== /** Create the structure out of a AABSPTree */ + ModelContainer::ModelContainer(AABSPTree<SubModel *> *pTree) { + int nSubModels, nNodes, nTriangles; nSubModels = nNodes = nTriangles = 0; countSubModelsAndNodesAndTriangles(*pTree->root, nSubModels, nNodes, nTriangles); + init(nNodes, nTriangles); + iNSubModel = nSubModels; + iSubModel = new SubModel[iNSubModel]; + int subModelPos,treeNodePos, trianglePos; subModelPos = treeNodePos = trianglePos = 0; + Vector3 lo = Vector3(inf(),inf(),inf()); Vector3 hi = Vector3(-inf(),-inf(),-inf()); Vector3 finalLo, finalHi; finalLo = lo; finalHi = hi; + fillContainer(*pTree->root, subModelPos, treeNodePos, trianglePos, lo, hi, finalLo, finalHi); setBounds(finalLo, finalHi); } + //========================================================== + ModelContainer::~ModelContainer(void) { free(); if(iSubModel != 0) delete [] iSubModel; } //========================================================== + bool ModelContainer::writeFile(const char *filename) { bool result = false; unsigned int flags=0; unsigned int size; + FILE *wf =fopen(filename,"wb"); if(wf) { @@ -161,11 +199,13 @@ namespace VMAP result = true; if(result && fwrite("CTREE01",8,1,wf) != 1) result = false; if(result && fwrite(&flags,sizeof(unsigned int),1,wf) != 1) result = false; + if(result && fwrite("POS ",4,1,wf) != 1) result = false; size = sizeof(float)*3; if(result && fwrite(&size,4,1,wf) != 1) result = false; Vector3 basePos = getBasePosition(); if(result && fwrite(&basePos,sizeof(float),3,wf) != 3) result = false; + if(result && fwrite("BOX ",4,1,wf) != 1) result = false; size = sizeof(float)*6; if(result && fwrite(&size,4,1,wf) != 1) result = false; @@ -173,28 +213,35 @@ namespace VMAP if(result && fwrite(&low,sizeof(float),3,wf) != 3) result = false; Vector3 high = iBox.high(); if(result && fwrite(&high,sizeof(float),3,wf) != 3) result = false; + if(result && fwrite("NODE",4,1,wf) != 1) result = false; size = sizeof(unsigned int)+ sizeof(TreeNode)*getNNodes(); if(result && fwrite(&size,4,1,wf) != 1) result = false; unsigned int val = getNNodes(); if(result && fwrite(&val,sizeof(unsigned int),1,wf) != 1) result = false; if(result && fwrite(getTreeNodes(),sizeof(TreeNode),getNNodes(),wf) != getNNodes()) result = false; + if(result && fwrite("TRIB",4,1,wf) != 1) result = false; size = sizeof(unsigned int)+ sizeof(TriangleBox)*getNTriangles(); if(result && fwrite(&size,4,1,wf) != 1) result = false; val = getNTriangles(); if(result && fwrite(&val,sizeof(unsigned int),1,wf) != 1) result = false; if(result && fwrite(getTriangles(),sizeof(TriangleBox),getNTriangles(),wf) != getNTriangles()) result = false; + if(result && fwrite("SUBM",4,1,wf) != 1) result = false; size = sizeof(unsigned int)+ sizeof(SubModel)*iNSubModel; if(result && fwrite(&size,4,1,wf) != 1) result = false; if(result && fwrite(&iNSubModel,sizeof(unsigned int),1,wf) != 1) result = false; if(result && fwrite(iSubModel,sizeof(SubModel),iNSubModel,wf) != iNSubModel) result = false; + fclose(wf); } + return(result); } + //=============================================================== + bool ModelContainer::readFile(const char *filename) { bool result = false; @@ -207,6 +254,7 @@ namespace VMAP if(rf) { free(); + result = true; char magic[8]; // Ignore the added magic header fread(magic,1,8,rf); @@ -219,6 +267,7 @@ namespace VMAP Vector3 basePos; if(result && fread(&basePos,sizeof(float),3,rf) != 3) result = false; setBasePosition(basePos); + //---- Box if(result && fread(chunk,4,1,rf) != 1) result = false; if(result && fread(&size,4,1,rf) != 1) result = false; @@ -226,25 +275,32 @@ namespace VMAP if(result && fread(&low,sizeof(float),3,rf) != 3) result = false; if(result && fread(&high,sizeof(float),3,rf) != 3) result = false; setBounds(low, high); + //---- TreeNodes if(result && fread(chunk,4,1,rf) != 1) result = false; if(result && fread(&size,4,1,rf) != 1) result = false; + if(result && fread(&ival,sizeof(unsigned int),1,rf) != 1) result = false; if(result) setNNodes(ival); if(result) setTreeNodeArray(new TreeNode[getNNodes()]); if(result && fread(getTreeNodes(),sizeof(TreeNode),getNNodes(),rf) != getNNodes()) result = false; + //---- TriangleBoxes if(result && fread(chunk,4,1,rf) != 1) result = false; if(result && fread(&size,4,1,rf) != 1) result = false; + if(result && fread(&ival,sizeof(unsigned int),1,rf) != 1) result = false; setNTriangles(ival); if(result) setTriangleArray(new TriangleBox[getNTriangles()]); if(result && fread(getTriangles(),sizeof(TriangleBox),getNTriangles(),rf) != getNTriangles()) result = false; + //---- SubModel if(result && fread(chunk,4,1,rf) != 1) result = false; if(result && fread(&size,4,1,rf) != 1) result = false; + if(result && fread(&iNSubModel,sizeof(unsigned int),1,rf) != 1) result = false; if(result) iSubModel = new SubModel[iNSubModel]; + if(result) { for(unsigned int i=0;i<iNSubModel && result; ++i) @@ -260,12 +316,15 @@ namespace VMAP } return result; } + //================================================================= + size_t ModelContainer::getMemUsage() { // BaseModel is included in ModelContainer return(iNSubModel * sizeof(SubModel) + BaseModel::getMemUsage() + sizeof(ModelContainer) - sizeof(BaseModel)); } + //================================================================= #ifdef _DEBUG_VMAPS #ifndef gBoxArray @@ -275,6 +334,7 @@ namespace VMAP extern bool myfound; #endif #endif + void ModelContainer::intersect(const G3D::Ray& pRay, float& pMaxDist, bool pStopAtFirstHit, G3D::Vector3& /*pOutLocation*/, G3D::Vector3& /*pOutNormal*/) const { IntersectionCallBack<SubModel> intersectCallback; @@ -282,12 +342,16 @@ namespace VMAP Ray relativeRay = Ray::fromOriginAndDirection(pRay.origin - getBasePosition(), pRay.direction); iTreeNodes[0].intersectRay(pRay, intersectCallback, pMaxDist, vna, pStopAtFirstHit, false); } + //========================================================== + bool ModelContainer::intersect(const G3D::Ray& pRay, float& pMaxDist) const { return BaseModel::intersect(getAABoxBounds(), pRay, pMaxDist); } + //================================================================= + template<typename RayCallback> void ModelContainer::intersectRay(const G3D::Ray& pRay, RayCallback& intersectCallback, float& pMaxDist, bool pStopAtFirstHit, bool intersectCallbackIsFast) { @@ -302,7 +366,9 @@ namespace VMAP { pAABox = pMc.getAABoxBounds(); } + //================================================================= + void getBounds(const ModelContainer* pMc, G3D::AABox& pAABox) { pAABox = pMc->getAABoxBounds(); |