few refactoring

This commit is contained in:
Julien Valentin 2017-08-28 14:23:15 +02:00
parent f995c9187e
commit e216833286
6 changed files with 97 additions and 30 deletions

View File

@ -36,6 +36,7 @@ namespace osgAnimation
bool init(MorphGeometry&); bool init(MorphGeometry&);
virtual void operator()(MorphGeometry&); virtual void operator()(MorphGeometry&);
protected: protected:
bool _needInit; bool _needInit;

View File

@ -37,35 +37,43 @@ namespace osgAnimation
virtual void operator()(RigGeometry&); virtual void operator()(RigGeometry&);
class BonePtrWeight class BonePtrWeight: public BoneWeight
{ {
public: public:
BonePtrWeight(Bone* bone, float weight) : _bone(bone), _weight(weight) {} BonePtrWeight(const std::string& name, float weight, Bone *bptr=0) :BoneWeight(name,weight), _boneptr(bptr) {}
const Bone* getBone() const { return _bone.get(); } BonePtrWeight(const BonePtrWeight &bw2) : BoneWeight(bw2), _boneptr(bw2._boneptr) {}
float getWeight() const { return _weight; }
void setWeight(float w) { _weight = w; } const Bone * getBonePtr()const{return _boneptr.get();}
void setBonePtr(Bone*b){_boneptr=b;}
bool operator==(const BonePtrWeight& b) const { return (getBoneName() == b.getBoneName() && getWeight() == b.getWeight()); }
//default order : sort by weight desc and bonename if equal
/*bool operator<(const BoneWeight& bw2)const{ if (_weight > bw2._weight)return true;
if (_weight < bw2._weight)return false;
return(_boneName<bw2._boneName);
}*/
protected: protected:
osg::observer_ptr<Bone> _bone; osg::observer_ptr< Bone > _boneptr;
float _weight;
}; };
typedef std::vector<BonePtrWeight> BonePtrWeightList;
typedef std::vector<unsigned int> IndexList;
typedef std::vector<BonePtrWeight> BonePtrWeightList; /// map a set of boneinfluence to a list of vertex indices sharing this set
typedef std::vector<unsigned int> VertexList;
class VertexGroup class VertexGroup
{ {
public: public:
BonePtrWeightList& getBoneWeights() { return _boneweights; } inline BonePtrWeightList& getBoneWeights() { return _boneweights; }
VertexList& getVertexes() { return _vertexes; }
void resetMatrix() inline IndexList& getVertexes() { return _vertexes; }
inline void resetMatrix()
{ {
_result.set(0, 0, 0, 0, _result.set(0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 1); 0, 0, 0, 1);
} }
void accummulateMatrix(const osg::Matrix& invBindMatrix, const osg::Matrix& matrix, osg::Matrix::value_type weight) inline void accummulateMatrix(const osg::Matrix& invBindMatrix, const osg::Matrix& matrix, osg::Matrix::value_type weight)
{ {
osg::Matrix m = invBindMatrix * matrix; osg::Matrix m = invBindMatrix * matrix;
osg::Matrix::value_type* ptr = m.ptr(); osg::Matrix::value_type* ptr = m.ptr();
@ -86,7 +94,7 @@ namespace osgAnimation
ptrresult[13] += ptr[13] * weight; ptrresult[13] += ptr[13] * weight;
ptrresult[14] += ptr[14] * weight; ptrresult[14] += ptr[14] * weight;
} }
void computeMatrixForVertexSet() inline void computeMatrixForVertexSet()
{ {
if (_boneweights.empty()) if (_boneweights.empty())
{ {
@ -98,7 +106,7 @@ namespace osgAnimation
for(BonePtrWeightList::iterator bwit=_boneweights.begin();bwit!=_boneweights.end();++bwit ) for(BonePtrWeightList::iterator bwit=_boneweights.begin();bwit!=_boneweights.end();++bwit )
{ {
const Bone* bone = bwit->getBone(); const Bone* bone = bwit->getBonePtr();
if (!bone) if (!bone)
{ {
osg::notify(osg::WARN) << this << " RigTransformSoftware::computeMatrixForVertexSet Warning a bone is null, skip it" << std::endl; osg::notify(osg::WARN) << this << " RigTransformSoftware::computeMatrixForVertexSet Warning a bone is null, skip it" << std::endl;
@ -110,10 +118,10 @@ namespace osgAnimation
accummulateMatrix(invBindMatrix, matrix, w); accummulateMatrix(invBindMatrix, matrix, w);
} }
} }
const osg::Matrix& getMatrix() const { return _result;} inline const osg::Matrix& getMatrix() const { return _result;}
protected: protected:
BonePtrWeightList _boneweights; BonePtrWeightList _boneweights;
VertexList _vertexes; IndexList _vertexes;
osg::Matrix _result; osg::Matrix _result;
}; };
@ -128,8 +136,8 @@ namespace osgAnimation
uniq.computeMatrixForVertexSet(); uniq.computeMatrixForVertexSet();
osg::Matrix matrix = transform * uniq.getMatrix() * invTransform; osg::Matrix matrix = transform * uniq.getMatrix() * invTransform;
const VertexList& vertexes = uniq.getVertexes(); const IndexList& vertexes = uniq.getVertexes();
for(VertexList::const_iterator vertIDit=vertexes.begin(); vertIDit!=vertexes.end(); ++vertIDit) for(IndexList::const_iterator vertIDit=vertexes.begin(); vertIDit!=vertexes.end(); ++vertIDit)
{ {
dst[*vertIDit] = src[*vertIDit] * matrix; dst[*vertIDit] = src[*vertIDit] * matrix;
} }
@ -146,8 +154,8 @@ namespace osgAnimation
uniq.computeMatrixForVertexSet(); uniq.computeMatrixForVertexSet();
osg::Matrix matrix = transform * uniq.getMatrix() * invTransform; osg::Matrix matrix = transform * uniq.getMatrix() * invTransform;
const VertexList& vertexes = uniq.getVertexes(); const IndexList& vertexes = uniq.getVertexes();
for(VertexList::const_iterator vertIDit=vertexes.begin(); vertIDit!=vertexes.end(); ++vertIDit) for(IndexList::const_iterator vertIDit=vertexes.begin(); vertIDit!=vertexes.end(); ++vertIDit)
{ {
dst[*vertIDit] = osg::Matrix::transform3x3(src[*vertIDit],matrix); dst[*vertIDit] = osg::Matrix::transform3x3(src[*vertIDit],matrix);
} }
@ -163,6 +171,7 @@ namespace osgAnimation
bool _needInit; bool _needInit;
std::map<std::string,bool> _invalidInfluence; std::map<std::string,bool> _invalidInfluence;
}; };
} }

View File

@ -23,15 +23,32 @@
namespace osgAnimation namespace osgAnimation
{ {
// first is bonename, and second the weight
struct BoneWeight: public std::pair<std::string, float>
{
BoneWeight( std::string f,float s):
std::pair<std::string,float>(f,s){}
inline const std::string& getBoneName()const{return first;}
inline void setBoneName(const std::string&s){first=s;}
inline const float &getWeight()const{return second;}
inline void setWeight(float i){second=i;}
};
// first is vertex index, and second the weight
struct IndexWeight: public std::pair<unsigned int, float>
{
IndexWeight( unsigned int f,float s): std::pair<unsigned int,float>(f,s){}
inline const unsigned int& getIndex()const{return first;}
inline void setIndex(unsigned int i){first=i;}
inline const float &getWeight()const{return second;}
inline void setWeight(float i){second=i;}
};
// first is vertex index, and second the weight, the
typedef std::pair<unsigned int, float> IndexWeight;
typedef std::vector<IndexWeight> VertexList; typedef std::vector<IndexWeight> VertexList;
class OSGANIMATION_EXPORT BoneInfluenceList : public VertexList class OSGANIMATION_EXPORT BoneInfluenceList : public VertexList
{ {
public: public:
const std::string& getName() const { return _name;} const std::string& getBoneName() const { return _name;}
void setName(const std::string& name) { _name = name;} void setBoneName(const std::string& name) { _name = name;}
protected: protected:
// the name is the bone to link to // the name is the bone to link to

View File

@ -34,6 +34,46 @@ RigTransformSoftware::RigTransformSoftware(const RigTransformSoftware& rts,const
} }
typedef std::vector<RigTransformSoftware::BonePtrWeight> BoneWeightList;
// sort by name and weight
struct SortByNameAndWeight : public std::less<RigTransformSoftware::BonePtrWeight>
{
bool operator()(const RigTransformSoftware::BonePtrWeight& b0,
const RigTransformSoftware::BonePtrWeight& b1) const
{
if (b0.getBoneName() < b1.getBoneName())
return true;
else if (b0.getBoneName() > b1.getBoneName())
return false;
if (b0.getWeight() < b1.getWeight())
return true;
return false;
}
};
struct SortByBoneWeightList : public std::less<BoneWeightList>
{
bool operator()(const BoneWeightList& b0,
const BoneWeightList& b1) const
{
if (b0.size() < b1.size())
return true;
else if (b0.size() > b1.size())
return false;
int size = b0.size();
for (int i = 0; i < size; i++)
{
bool result = SortByNameAndWeight()(b0[i], b1[i]);
if (result)
return true;
else if (SortByNameAndWeight()(b1[i], b0[i]))
return false;
}
return false;
}
};
bool RigTransformSoftware::init(RigGeometry& geom) bool RigTransformSoftware::init(RigGeometry& geom)
{ {
if (!geom.getSkeleton()) if (!geom.getSkeleton())
@ -142,7 +182,7 @@ void RigTransformSoftware::initVertexSetFromBones(const BoneMap& map, const Vert
continue; continue;
} }
Bone* bone = it->second.get(); Bone* bone = it->second.get();
boneList.push_back(BonePtrWeight(bone, weight)); boneList.push_back(BonePtrWeight(bone->getName(), weight, bone));
sumOfWeight += weight; sumOfWeight += weight;
} }
// if a bone referenced by a vertexinfluence is missed it can make the sum less than 1.0 // if a bone referenced by a vertexinfluence is missed it can make the sum less than 1.0

View File

@ -40,10 +40,10 @@ void VertexInfluenceSet::buildVertex2BoneList(unsigned int numvertices)
IndexWeight viw = vi[i]; IndexWeight viw = vi[i];
int index = viw.first; int index = viw.first;
float weight = viw.second; float weight = viw.second;
if (vi.getName().empty()){ if (vi.getBoneName().empty()){
OSG_WARN << "VertexInfluenceSet::buildVertex2BoneList warning vertex " << index << " is not assigned to a bone" << std::endl; OSG_WARN << "VertexInfluenceSet::buildVertex2BoneList warning vertex " << index << " is not assigned to a bone" << std::endl;
} }
_vertex2Bones[index].push_back(BoneWeight(vi.getName(), weight)); _vertex2Bones[index].push_back(BoneWeight(vi.getBoneName(), weight));
} }
} }

View File

@ -21,7 +21,7 @@ static bool readInfluenceMap( osgDB::InputStream& is, osgAnimation::RigGeometry&
viSize = is.readSize(); is >> is.BEGIN_BRACKET; viSize = is.readSize(); is >> is.BEGIN_BRACKET;
osgAnimation::BoneInfluenceList vi; osgAnimation::BoneInfluenceList vi;
vi.setName( name ); vi.setBoneName( name );
vi.reserve( viSize ); vi.reserve( viSize );
for ( unsigned int j=0; j<viSize; ++j ) for ( unsigned int j=0; j<viSize; ++j )
{ {