OpenSceneGraph/include/osgAnimation/VertexInfluence

/*  -*-c++-*- 
 *  Copyright (C) 2008 Cedric Pinson <mornifle@plopbyte.net>
 *
 * This library is open source and may be redistributed and/or modified under  
 * the terms of the OpenSceneGraph Public License (OSGPL) version 0.0 or 
 * (at your option) any later version.  The full license is in LICENSE file
 * included with this distribution, and on the openscenegraph.org website.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
 * OpenSceneGraph Public License for more details.
*/

#ifndef OSGANIMATION_VERTEX_INFLUENCES_H
#define OSGANIMATION_VERTEX_INFLUENCES_H

#include <osg/Object>
#include <osgAnimation/Export>
#include <map>
#include <vector>
#include <string>

namespace osgAnimation
{

    // first is vertex index, and second the weight, the
    typedef std::pair<int, float> VertexIndexWeight;
    typedef std::vector<VertexIndexWeight> VertexList;
    class OSGANIMATION_EXPORT VertexInfluence : public VertexList
    {
    public:
        const std::string& getName() const { return _name;}
        void setName(const std::string& name) { _name = name;}

    protected:
        // the name is the bone to link to
        std::string _name;
    };

    //  typedef std::map<std::string, VertexInfluence> VertexInfluenceMap;

    class VertexInfluenceMap : public std::map<std::string, VertexInfluence> , public osg::Object
    {
    public:
        META_Object(osgAnimation, VertexInfluenceMap);

        VertexInfluenceMap() {}
        VertexInfluenceMap(const osgAnimation::VertexInfluenceMap&, const osg::CopyOp&) {}
    };


    // this class manage VertexInfluence database by mesh
    // reference bones per vertex ...
    class VertexInfluenceSet
    {
    public:
        typedef std::vector<VertexInfluence> BoneToVertexList;
    
        class BoneWeight 
        {
        public:
            BoneWeight(const std::string& name, float weight) : _boneName(name), _weight(weight) {}
            const std::string& getBoneName() const { return _boneName; }
            float getWeight() const { return _weight; }
            void setWeight(float weight) { _weight = weight; }
            bool operator==(const BoneWeight& b) const { return (_boneName == b.getBoneName() && _weight == b.getWeight()); }
        protected:
            std::string _boneName;
            float _weight;
        };

        typedef std::vector<BoneWeight> BoneWeightList;
        typedef std::map<int,BoneWeightList> VertexIndexToBoneWeightMap;
    
        class UniqVertexSetToBoneSet 
        {
        public:
            void setBones(BoneWeightList& bones) { _bones = bones;}
            const BoneWeightList& getBones() const { return _bones;}
            std::vector<int>& getVertexes() { return _vertexes;}
            const std::vector<int>& getVertexes() const { return _vertexes;}
        protected:
            std::vector<int> _vertexes;
            BoneWeightList _bones; // here we could limit matrix operation by caching (weight * matrix)
        };

        typedef std::vector<UniqVertexSetToBoneSet> UniqVertexSetToBoneSetList;

        const UniqVertexSetToBoneSetList& getUniqVertexSetToBoneSetList() const { return _uniqVertexSetToBoneSet;}
        void addVertexInfluence(const VertexInfluence& v) { _bone2Vertexes.push_back(v); }
        void buildVertex2BoneList();
        void buildUniqVertexSetToBoneSetList();
        void clear() 
        {
            _bone2Vertexes.clear(); 
            _uniqVertexSetToBoneSet.clear();
        }
    protected:
        BoneToVertexList _bone2Vertexes;
        VertexIndexToBoneWeightMap _vertex2Bones;
        UniqVertexSetToBoneSetList _uniqVertexSetToBoneSet;
    };

}

#endif