You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
343 lines
11 KiB
C++
343 lines
11 KiB
C++
#include "bulletnifloader.hpp"
|
|
|
|
#include <cstdio>
|
|
#include <vector>
|
|
#include <list>
|
|
#include <stdexcept>
|
|
|
|
#include <BulletCollision/CollisionShapes/btBoxShape.h>
|
|
#include <BulletCollision/CollisionShapes/btTriangleMesh.h>
|
|
#include <BulletCollision/CollisionShapes/btScaledBvhTriangleMeshShape.h>
|
|
#include <BulletCollision/CollisionShapes/btCompoundShape.h>
|
|
|
|
#include <components/misc/stringops.hpp>
|
|
|
|
#include <components/nif/node.hpp>
|
|
#include <components/nif/data.hpp>
|
|
#include <components/nif/property.hpp>
|
|
#include <components/nif/controller.hpp>
|
|
#include <components/nif/extra.hpp>
|
|
|
|
|
|
namespace
|
|
{
|
|
|
|
osg::Matrixf getWorldTransform(const Nif::Node *node)
|
|
{
|
|
if(node->parent != NULL)
|
|
return node->trafo.toMatrix() * getWorldTransform(node->parent);
|
|
return node->trafo.toMatrix();
|
|
}
|
|
|
|
btVector3 getbtVector(const osg::Vec3f &v)
|
|
{
|
|
return btVector3(v.x(), v.y(), v.z());
|
|
}
|
|
|
|
bool pathFileNameStartsWithX(const std::string& path)
|
|
{
|
|
const std::size_t slashpos = path.find_last_of("/\\");
|
|
const std::size_t letterPos = slashpos == std::string::npos ? 0 : slashpos + 1;
|
|
return letterPos < path.size() && (path[letterPos] == 'x' || path[letterPos] == 'X');
|
|
}
|
|
|
|
}
|
|
|
|
namespace NifBullet
|
|
{
|
|
|
|
BulletNifLoader::BulletNifLoader()
|
|
: mCompoundShape(NULL)
|
|
, mStaticMesh(NULL)
|
|
{
|
|
}
|
|
|
|
BulletNifLoader::~BulletNifLoader()
|
|
{
|
|
}
|
|
|
|
osg::ref_ptr<Resource::BulletShape> BulletNifLoader::load(const Nif::NIFFilePtr& nif)
|
|
{
|
|
mShape = new Resource::BulletShape;
|
|
|
|
mCompoundShape = NULL;
|
|
mStaticMesh = NULL;
|
|
|
|
if (nif->numRoots() < 1)
|
|
{
|
|
warn("Found no root nodes in NIF.");
|
|
return mShape;
|
|
}
|
|
|
|
Nif::Record *r = nif->getRoot(0);
|
|
assert(r != NULL);
|
|
|
|
Nif::Node *node = dynamic_cast<Nif::Node*>(r);
|
|
if (node == NULL)
|
|
{
|
|
warn("First root in file was not a node, but a " +
|
|
r->recName + ". Skipping file.");
|
|
return mShape;
|
|
}
|
|
|
|
if (findBoundingBox(node))
|
|
{
|
|
std::unique_ptr<btCompoundShape> compound (new btCompoundShape);
|
|
|
|
btBoxShape* boxShape = new btBoxShape(getbtVector(mShape->mCollisionBoxHalfExtents));
|
|
btTransform transform = btTransform::getIdentity();
|
|
transform.setOrigin(getbtVector(mShape->mCollisionBoxTranslate));
|
|
compound->addChildShape(transform, boxShape);
|
|
|
|
mShape->mCollisionShape = compound.release();
|
|
return mShape;
|
|
}
|
|
else
|
|
{
|
|
// files with the name convention xmodel.nif usually have keyframes stored in a separate file xmodel.kf (see Animation::addAnimSource).
|
|
// assume all nodes in the file will be animated
|
|
const bool isAnimated = pathFileNameStartsWithX(nif->getFilename());
|
|
|
|
// If the mesh has RootCollisionNode, attached to actual root node, use it as collision mesh
|
|
const Nif::Node* rootCollisionNode = getCollisionNode(node);
|
|
if (rootCollisionNode)
|
|
handleNode(nif->getFilename(), rootCollisionNode, 0, false, isAnimated, false);
|
|
else
|
|
handleNode(nif->getFilename(), node, 0, true, isAnimated, true);
|
|
|
|
if (mCompoundShape)
|
|
{
|
|
mShape->mCollisionShape = mCompoundShape;
|
|
if (mStaticMesh)
|
|
{
|
|
btTransform trans;
|
|
trans.setIdentity();
|
|
mCompoundShape->addChildShape(trans, new Resource::TriangleMeshShape(mStaticMesh,true));
|
|
}
|
|
}
|
|
else if (mStaticMesh)
|
|
mShape->mCollisionShape = new Resource::TriangleMeshShape(mStaticMesh,true);
|
|
|
|
return mShape;
|
|
}
|
|
}
|
|
|
|
// Find a boundingBox in the node hierarchy.
|
|
// Return: use bounding box for collision?
|
|
bool BulletNifLoader::findBoundingBox(const Nif::Node* node, int flags)
|
|
{
|
|
flags |= node->flags;
|
|
|
|
if (node->hasBounds)
|
|
{
|
|
mShape->mCollisionBoxHalfExtents = node->boundXYZ;
|
|
mShape->mCollisionBoxTranslate = node->boundPos;
|
|
|
|
if (flags & Nif::NiNode::Flag_BBoxCollision)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
const Nif::NiNode *ninode = dynamic_cast<const Nif::NiNode*>(node);
|
|
if(ninode)
|
|
{
|
|
const Nif::NodeList &list = ninode->children;
|
|
for(size_t i = 0;i < list.length();i++)
|
|
{
|
|
if(!list[i].empty())
|
|
{
|
|
bool found = findBoundingBox (list[i].getPtr());
|
|
if (found)
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
const Nif::Node* BulletNifLoader::getCollisionNode(const Nif::Node* rootNode)
|
|
{
|
|
const Nif::NiNode *ninode = dynamic_cast<const Nif::NiNode*>(rootNode);
|
|
if(ninode)
|
|
{
|
|
// If root NiNode has only other NiNode as child, consider it as a wrapper, not as actual root node
|
|
const Nif::NodeList &list = ninode->children;
|
|
if (list.length() == 1 &&
|
|
rootNode->recType == Nif::RC_NiNode &&
|
|
list[0].getPtr()->recType == Nif::RC_NiNode)
|
|
{
|
|
return getCollisionNode(list[0].getPtr());
|
|
}
|
|
|
|
for(size_t i = 0; i < list.length(); i++)
|
|
{
|
|
if(list[i].empty())
|
|
continue;
|
|
|
|
const Nif::Node* childNode = list[i].getPtr();
|
|
if(childNode->recType == Nif::RC_RootCollisionNode)
|
|
return childNode;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void BulletNifLoader::handleNode(const std::string& fileName, const Nif::Node *node, int flags,
|
|
bool isCollisionNode, bool isAnimated, bool autogenerated)
|
|
{
|
|
// Accumulate the flags from all the child nodes. This works for all
|
|
// the flags we currently use, at least.
|
|
flags |= node->flags;
|
|
|
|
if (!node->controller.empty() && node->controller->recType == Nif::RC_NiKeyframeController
|
|
&& (node->controller->flags & Nif::NiNode::ControllerFlag_Active))
|
|
isAnimated = true;
|
|
|
|
isCollisionNode = isCollisionNode || (node->recType == Nif::RC_RootCollisionNode);
|
|
|
|
if (node->recType == Nif::RC_RootCollisionNode && autogenerated)
|
|
std::cerr << "Found RootCollisionNode attached to non-root node in " << fileName << ". Treat it as a common NiTriShape." << std::endl;
|
|
|
|
// Don't collide with AvoidNode shapes
|
|
if(node->recType == Nif::RC_AvoidNode)
|
|
flags |= 0x800;
|
|
|
|
// Check for extra data
|
|
Nif::Extra const *e = node;
|
|
while (!e->extra.empty())
|
|
{
|
|
// Get the next extra data in the list
|
|
e = e->extra.getPtr();
|
|
assert(e != NULL);
|
|
|
|
if (e->recType == Nif::RC_NiStringExtraData)
|
|
{
|
|
// String markers may contain important information
|
|
// affecting the entire subtree of this node
|
|
Nif::NiStringExtraData *sd = (Nif::NiStringExtraData*)e;
|
|
|
|
if (Misc::StringUtils::ciCompareLen(sd->string, "NC", 2) == 0)
|
|
{
|
|
// No collision. Use an internal flag setting to mark this.
|
|
flags |= 0x800;
|
|
}
|
|
else if (sd->string == "MRK" && autogenerated)
|
|
{
|
|
// Marker can still have collision if the model explicitely specifies it via a RootCollisionNode.
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (isCollisionNode)
|
|
{
|
|
// NOTE: a trishape with hasBounds=true, but no BBoxCollision flag should NOT go through handleNiTriShape!
|
|
// It must be ignored completely.
|
|
// (occurs in tr_ex_imp_wall_arch_04.nif)
|
|
if(!node->hasBounds && node->recType == Nif::RC_NiTriShape)
|
|
{
|
|
handleNiTriShape(static_cast<const Nif::NiTriShape*>(node), flags, getWorldTransform(node), isAnimated);
|
|
}
|
|
}
|
|
|
|
// For NiNodes, loop through children
|
|
const Nif::NiNode *ninode = dynamic_cast<const Nif::NiNode*>(node);
|
|
if(ninode)
|
|
{
|
|
const Nif::NodeList &list = ninode->children;
|
|
for(size_t i = 0;i < list.length();i++)
|
|
{
|
|
if(!list[i].empty())
|
|
handleNode(fileName, list[i].getPtr(), flags, isCollisionNode, isAnimated, autogenerated);
|
|
}
|
|
}
|
|
}
|
|
|
|
void BulletNifLoader::handleNiTriShape(const Nif::NiTriShape *shape, int flags, const osg::Matrixf &transform, bool isAnimated)
|
|
{
|
|
assert(shape != NULL);
|
|
|
|
// If the object was marked "NCO" earlier, it shouldn't collide with
|
|
// anything. So don't do anything.
|
|
if ((flags & 0x800))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!shape->skin.empty())
|
|
isAnimated = false;
|
|
|
|
if (shape->data.empty())
|
|
return;
|
|
if (shape->data->triangles.empty())
|
|
return;
|
|
|
|
if (isAnimated)
|
|
{
|
|
if (!mCompoundShape)
|
|
mCompoundShape = new btCompoundShape();
|
|
|
|
btTriangleMesh* childMesh = new btTriangleMesh();
|
|
|
|
const Nif::NiTriShapeData *data = shape->data.getPtr();
|
|
|
|
childMesh->preallocateVertices(data->vertices.size());
|
|
childMesh->preallocateIndices(data->triangles.size());
|
|
|
|
const std::vector<osg::Vec3f> &vertices = data->vertices;
|
|
const std::vector<unsigned short> &triangles = data->triangles;
|
|
|
|
for(size_t i = 0;i < data->triangles.size();i+=3)
|
|
{
|
|
osg::Vec3f b1 = vertices[triangles[i+0]];
|
|
osg::Vec3f b2 = vertices[triangles[i+1]];
|
|
osg::Vec3f b3 = vertices[triangles[i+2]];
|
|
childMesh->addTriangle(getbtVector(b1), getbtVector(b2), getbtVector(b3));
|
|
}
|
|
|
|
Resource::TriangleMeshShape* childShape = new Resource::TriangleMeshShape(childMesh,true);
|
|
|
|
float scale = shape->trafo.scale;
|
|
const Nif::Node* parent = shape;
|
|
while (parent->parent)
|
|
{
|
|
parent = parent->parent;
|
|
scale *= parent->trafo.scale;
|
|
}
|
|
osg::Quat q = transform.getRotate();
|
|
osg::Vec3f v = transform.getTrans();
|
|
childShape->setLocalScaling(btVector3(scale, scale, scale));
|
|
|
|
btTransform trans(btQuaternion(q.x(), q.y(), q.z(), q.w()), btVector3(v.x(), v.y(), v.z()));
|
|
|
|
mShape->mAnimatedShapes.insert(std::make_pair(shape->recIndex, mCompoundShape->getNumChildShapes()));
|
|
|
|
mCompoundShape->addChildShape(trans, childShape);
|
|
}
|
|
else
|
|
{
|
|
if (!mStaticMesh)
|
|
mStaticMesh = new btTriangleMesh(false);
|
|
|
|
// Static shape, just transform all vertices into position
|
|
const Nif::NiTriShapeData *data = shape->data.getPtr();
|
|
const std::vector<osg::Vec3f> &vertices = data->vertices;
|
|
const std::vector<unsigned short> &triangles = data->triangles;
|
|
|
|
mStaticMesh->preallocateVertices(data->vertices.size());
|
|
mStaticMesh->preallocateIndices(data->triangles.size());
|
|
|
|
size_t numtris = data->triangles.size();
|
|
for(size_t i = 0;i < numtris;i+=3)
|
|
{
|
|
osg::Vec3f b1 = vertices[triangles[i+0]]*transform;
|
|
osg::Vec3f b2 = vertices[triangles[i+1]]*transform;
|
|
osg::Vec3f b3 = vertices[triangles[i+2]]*transform;
|
|
mStaticMesh->addTriangle(getbtVector(b1), getbtVector(b2), getbtVector(b3));
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace NifBullet
|