OpenSceneGraph/include/osgAnimation/RigTransformHardware

/*  -*-c++-*-
 *  Copyright (C) 2009 Cedric Pinson <cedric.pinson@plopbyte.net>
 *  Copyright (C) 2017 Julien Valentin <mp3butcher@hotmail.com>
 *
 * 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_RIG_TRANSFORM_HARDWARE
#define OSGANIMATION_RIG_TRANSFORM_HARDWARE 1

#include <osgAnimation/Export>
#include <osgAnimation/RigTransform>
#include <osgAnimation/VertexInfluence>
#include <osgAnimation/Bone>
#include <osg/Matrix>
#include <osg/Array>

namespace osgAnimation
{
    class RigGeometry;

    /// This class manage format for hardware skinning
    class OSGANIMATION_EXPORT RigTransformHardware : public RigTransform
    {
    public:

        RigTransformHardware();

        RigTransformHardware(const RigTransformHardware& rth, const osg::CopyOp& copyop);

        META_Object(osgAnimation,RigTransformHardware);

        typedef std::vector<osg::ref_ptr<osg::Vec4Array> > BoneWeightAttribList;
        typedef std::vector<osg::ref_ptr<Bone> > BonePalette;
        typedef std::map<std::string, unsigned int> BoneNamePaletteIndex;
        typedef std::vector<osg::Matrix> MatrixPalette;

        osg::Vec4Array* getVertexAttrib(unsigned int index);
        unsigned int getNumVertexAttrib() const     {return _boneWeightAttribArrays.size();}

        void setShader(osg::Shader* shader)         {       _shader = shader;             }
        osg::Shader* getShader() const              {       return _shader;               }

        const unsigned int &getNumBonesPerVertex() const{   return _bonesPerVertex;       }
        const unsigned int &getNumVertexes() const      {   return _nbVertexes;           }

        const BoneNamePaletteIndex& getBoneNameToPalette(){ return _boneNameToPalette;    }
        const BonePalette& getBonePalette()             {   return _bonePalette;          }
        osg::Uniform* getMatrixPaletteUniform()         {   return _uniformMatrixPalette; }

        void computeMatrixPaletteUniform(const osg::Matrix& transformFromSkeletonToGeometry, const osg::Matrix& invTransformFromSkeletonToGeometry);

        virtual void operator()(RigGeometry&);

        virtual bool prepareData(RigGeometry& );

    protected:

        unsigned int _bonesPerVertex;
        unsigned int _nbVertexes;

        BonePalette _bonePalette;
        BoneNamePaletteIndex _boneNameToPalette;
        BoneWeightAttribList _boneWeightAttribArrays;
        osg::ref_ptr<osg::Uniform> _uniformMatrixPalette;
        osg::ref_ptr<osg::Shader> _shader;

        bool _needInit;

        bool buildPalette(const BoneMap& boneMap ,const RigGeometry& rig);

    };
}

#endif