TranslationConstraint.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2018 Univ. Grenoble Alpes, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK 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
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/

#ifndef SIMULATOR_SIMULATORS_SOFA_TRANSLATIONCONSTRAINT_H
#define SIMULATOR_SIMULATORS_SOFA_TRANSLATIONCONSTRAINT_H


#include "sofa/core/behavior/ProjectiveConstraintSet.h"
#include "sofa/core/behavior/MechanicalState.h"
//#include "sofa/core/VisualModel.h"

#include <vector>
#include <map>

#include <Loads.h>

template<class DataTypes>

class TranslationConstraint : public sofa::core::behavior::ProjectiveConstraintSet<DataTypes> { //, public VisualModel
public :
    typedef typename DataTypes::VecCoord VecCoord;
    typedef typename DataTypes::VecDeriv VecDeriv;
    typedef typename DataTypes::VecCoord::iterator VecCoordIterator;
    typedef typename DataTypes::VecDeriv::iterator VecDerivIterator;
    typedef typename DataTypes::Coord Coord;
    typedef typename DataTypes::Deriv Deriv;

    typedef typename DataTypes::MatrixDeriv::RowType MatrixDerivRowType;

    TranslationConstraint(Loads* loadsList, const std::map<unsigned int, unsigned int>& atomIndexToDOFIndex, sofa::core::behavior::MechanicalState<DataTypes>* mm);

    ~TranslationConstraint() { }

    std::vector<unsigned int> getTargets() {
        return targets;
    }

    TranslationConstraint<DataTypes>* addConstraint(unsigned int index, Deriv trans);
    TranslationConstraint<DataTypes>* removeConstraint(int index);

    void projectResponse(VecDeriv& dx); 
    virtual void projectVelocity(VecDeriv& v); 
    virtual void projectPosition(VecCoord& x); 

    void projectResponse(MatrixDerivRowType& dx) {}

    void draw();
    void initTextures() { }
    void update() { }

    void setInitTime(double time);

    //TODO need to define which getclass to choose because of double inherance...
    //sofa::core::objectmodel::BaseClass* getClass() const {return NULL;}

private:

    void fixDOF(int index, int axe);
    sofa::core::behavior::MechanicalState<DataTypes>* mmodel;
    std::vector<unsigned int> targets;
    VecDeriv translations;
    VecDeriv directionsNULLs;
    VecDeriv initPos;
    double initTime;
    std::vector<Load*> loads;
    std::map<unsigned int, unsigned int> atomToDOFIndexes;
};

#endif // SIMULATOR_SIMULATORS_SOFA_TRANSLATIONCONSTRAINT_H