Edge2QuadTopologicalMapping.h

Go to the documentation of this file.

/******************************************************************************
*       SOFA, Simulation Open-Framework Architecture, version 1.0 beta 3      *
*                (c) 2006-2008 MGH, INRIA, USTL, UJF, CNRS                    *
*                                                                             *
* This library is free software; you can redistribute it and/or modify it     *
* under the terms of the GNU Lesser General Public License as published by    *
* the Free Software Foundation; either version 2.1 of the License, or (at     *
* your option) any later version.                                             *
*                                                                             *
* 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 GNU Lesser General Public License *
* for more details.                                                           *
*                                                                             *
* You should have received a copy of the GNU Lesser General Public License    *
* along with this library; if not, write to the Free Software Foundation,     *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.          *
*******************************************************************************
*                               SOFA :: Modules                               *
*                                                                             *
* Authors: The SOFA Team and external contributors (see Authors.txt)          *
*                                                                             *
* Contact information: contact@sofa-framework.org                             *
******************************************************************************/
#ifndef SOFA_COMPONENT_TOPOLOGY_EDGE2QUADTOPOLOGICALMAPPING_H
#define SOFA_COMPONENT_TOPOLOGY_EDGE2QUADTOPOLOGICALMAPPING_H

#include <sofa/core/componentmodel/topology/TopologicalMapping.h>

#include <sofa/defaulttype/Vec.h>
#include <sofa/defaulttype/RigidTypes.h>
#include <map>

#include <sofa/core/BaseMapping.h>
#include <sofa/core/componentmodel/behavior/MechanicalState.h>


namespace sofa
{

namespace component
{

namespace topology
{

using namespace sofa::defaulttype;

using namespace sofa::component::topology;
using namespace sofa::core::componentmodel::topology;
using namespace sofa::core;
using namespace sofa::core::componentmodel::behavior;

    class Edge2QuadTopologicalMapping : public TopologicalMapping 
    {
        public:
            typedef State<Rigid3Types>::VecCoord VecCoord;
            typedef State<Rigid3Types>::Coord Coord;            
            typedef Coord::value_type Real;         
            enum { M=Coord::static_size };
            typedef defaulttype::Mat<M,M,Real> Mat;
            typedef defaulttype::Vec<M,Real> Vec;


        public:
            
            Edge2QuadTopologicalMapping(In* from, Out* to)
            :  TopologicalMapping(from, to),
            m_nbPointsOnEachCircle( initData(&m_nbPointsOnEachCircle, "nbPointsOnEachCircle", "Discretization of created circles")),
            m_radius( initData(&m_radius, "radius", "Radius of created circles")),
            object1(initData(&object1, std::string("../.."), "object1", "First object to map")),
            object2(initData(&object2, std::string(".."), "object2", "Second object to map"))
            {
            }

            virtual ~Edge2QuadTopologicalMapping()
            {}              

            virtual void init();

            
            virtual void updateTopologicalMapping();

            virtual unsigned int getFromIndex(unsigned int ind);

            template<class T>
            static bool canCreate(T*& obj, core::objectmodel::BaseContext* context, core::objectmodel::BaseObjectDescription* arg)
            {
                if (arg->findObject(arg->getAttribute("object1","../..")) == NULL)
                    std::cerr << "Cannot create "<<className(obj)<<" as object1 is missing.\n";
                if (arg->findObject(arg->getAttribute("object2","..")) == NULL)
                    std::cerr << "Cannot create "<<className(obj)<<" as object2 is missing.\n";

                if (arg->findObject(arg->getAttribute("object1","../..")) == NULL || arg->findObject(arg->getAttribute("object2","..")) == NULL)
                  return false;


                BaseMeshTopology* topoIn;
                BaseMeshTopology* topoOut;
                (dynamic_cast<sofa::core::objectmodel::BaseObject*>(arg->findObject(arg->getAttribute("object1","../.."))))->getContext()->get(topoIn);
                (dynamic_cast<sofa::core::objectmodel::BaseObject*>(arg->findObject(arg->getAttribute("object2",".."))))->getContext()->get(topoOut);

                if (dynamic_cast<In*>(topoIn) == NULL)
                    return false;
                if (dynamic_cast<Out*>(topoOut) == NULL)
                    return false;
                return BaseMapping::canCreate(obj, context, arg);
            }

            template<class T>
            static void create(T*& obj, core::objectmodel::BaseContext* context, core::objectmodel::BaseObjectDescription* arg)
            {
                BaseMeshTopology* topoIn=NULL;
                BaseMeshTopology* topoOut=NULL;
                if (arg)
                  {
                    if (arg->findObject(arg->getAttribute("object1","../..")) != NULL)
                      (dynamic_cast<sofa::core::objectmodel::BaseObject*>(arg->findObject(arg->getAttribute("object1","../.."))))->getContext()->get(topoIn);
                    if (arg->findObject(arg->getAttribute("object2","../..")) != NULL)
                      (dynamic_cast<sofa::core::objectmodel::BaseObject*>(arg->findObject(arg->getAttribute("object2",".."))))->getContext()->get(topoOut);
                  }
                obj = new T(
                    (arg?dynamic_cast<In*>(topoIn):NULL),
                    (arg?dynamic_cast<Out*>(topoOut):NULL));
                if (context) context->addObject(obj);
                if ((arg) && (arg->getAttribute("object1")))
                {
                    obj->object1.setValue( arg->getAttribute("object1") );
                    arg->removeAttribute("object1");
                }
                if ((arg) && (arg->getAttribute("object2")))
                {
                    obj->object2.setValue( arg->getAttribute("object2") );
                    arg->removeAttribute("object2");
                }
                if (arg) obj->parse(arg);
            }

        protected:
            Data<unsigned int> m_nbPointsOnEachCircle; // number of points to create along the circles around each point of the input topology (10 by default)
            Data<double> m_radius;  // radius of the circles around each point of the input topology (1 by default) 

            Data< std::string > object1;
            Data< std::string > object2;
        };

} // namespace topology

} // namespace component

} // namespace sofa

#endif // SOFA_COMPONENT_TOPOLOGY_EDGE2QUADTOPOLOGICALMAPPING_H