Skip to content

P
pinocchio
Commit 6e35342c authored by Justin Carpentier's avatar Justin Carpentier
Browse files
Options

core: split visitors into several sub files

parent 2326b8a7
Hide whitespace changes
Inline Side-by-side
Showing with 213 additions and 191 deletions
CMakeLists.txt
View file @ 6e35342c
......@@ -227,6 +227,7 @@ MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/spatial")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/multibody")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/multibody/joint")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/multibody/liegroup")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/multibody/visitor")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/parsers/lua")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/parsers")
MAKE_DIRECTORY("${${PROJECT_NAME}_BINARY_DIR}/include/pinocchio/parsers/urdf")
......
src/multibody/visitor.hpp
View file @ 6e35342c
......@@ -6,196 +6,6 @@
#ifndef __pinocchio_multibody_visitor_hpp__
#define __pinocchio_multibody_visitor_hpp__
#define BOOST_FUSION_INVOKE_MAX_ARITY 10
#include <boost/variant/static_visitor.hpp>
#include <boost/fusion/include/invoke.hpp>
#include <boost/fusion/container/generation/make_vector.hpp>
#include "pinocchio/multibody/joint/joint-base.hpp"
namespace boost
{
namespace fusion
{
// Append the element T at the front of boost fusion vector V.
template<typename T,typename V>
typename result_of::push_front<V const, T>::type
append(T const& t,V const& v)
{ return push_front(v,t); }
// Append the elements T1 and T2 at the front of boost fusion vector V.
template<typename T1,typename T2,typename V>
typename result_of::push_front<typename result_of::push_front<V const, T2>::type const, T1>::type
append2(T1 const& t1,T2 const& t2,V const& v)
{ return push_front(push_front(v,t2),t1); }
}
}
namespace pinocchio
{
namespace fusion
{
namespace bf = boost::fusion;
typedef boost::blank NoArg;
template<typename JointVisitorDerived, typename ReturnType = void>
struct JointVisitorBase
{
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl, typename ArgsTmp>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
JointDataTpl<Scalar,Options,JointCollectionTpl> & jdata,
ArgsTmp args)
{
InternalVisitorModelAndData<JointModelTpl<Scalar,Options,JointCollectionTpl>,ArgsTmp> visitor(jdata,args);
return boost::apply_visitor(visitor,jmodel);
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
JointDataTpl<Scalar,Options,JointCollectionTpl> & jdata)
{
InternalVisitorModelAndData<JointModelTpl<Scalar,Options,JointCollectionTpl>,NoArg> visitor(jdata);
return boost::apply_visitor(visitor,jmodel);
}
template<typename JointModelDerived, typename ArgsTmp>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
typename JointModelBase<JointModelDerived>::JointDataDerived & jdata,
ArgsTmp args)
{
InternalVisitorModelAndData<JointModelDerived,ArgsTmp> visitor(jdata,args);
return visitor(jmodel.derived());
}
template<typename JointModelDerived>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
typename JointModelBase<JointModelDerived>::JointDataDerived & jdata)
{
InternalVisitorModelAndData<JointModelDerived,NoArg> visitor(jdata);
return visitor(jmodel.derived());
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl, typename ArgsTmp>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
ArgsTmp args)
{
InternalVisitorModel<ArgsTmp> visitor(args);
return boost::apply_visitor(visitor,jmodel);
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel)
{
InternalVisitorModel<NoArg> visitor;
return boost::apply_visitor(visitor,jmodel);
}
template<typename JointModelDerived, typename ArgsTmp>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
ArgsTmp args)
{
InternalVisitorModel<ArgsTmp> visitor(args);
return visitor(jmodel.derived());
}
template<typename JointModelDerived>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel)
{
InternalVisitorModel<NoArg> visitor;
return visitor(jmodel.derived());
}
private:
template<typename JointModel, typename ArgType>
struct InternalVisitorModelAndData
: public boost::static_visitor<ReturnType>
{
typedef typename JointModel::JointDataDerived JointData;
InternalVisitorModelAndData(JointData & jdata, ArgType args)
: jdata(jdata), args(args)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::append2(boost::ref(jmodel.derived()),
boost::ref(boost::get<typename JointModelBase<JointModelDerived>::JointDataDerived >(jdata)),
args));
}
ReturnType operator()(const JointModelVoid) {return;}
JointData & jdata;
ArgType args;
};
template<typename JointModel>
struct InternalVisitorModelAndData<JointModel,NoArg>
: public boost::static_visitor<ReturnType>
{
typedef typename JointModel::JointDataDerived JointData;
InternalVisitorModelAndData(JointData & jdata)
: jdata(jdata)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::make_vector(boost::ref(jmodel.derived()),
boost::ref(boost::get<typename JointModelBase<JointModelDerived>::JointDataDerived >(jdata)))
);
}
JointData & jdata;
};
template<typename ArgType, typename Dummy = void>
struct InternalVisitorModel
: public boost::static_visitor<ReturnType>
{
InternalVisitorModel(ArgType args)
: args(args)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::append(boost::ref(jmodel.derived()),
args));
}
ReturnType operator()(const JointModelVoid) {return;}
ArgType args;
};
template<typename Dummy>
struct InternalVisitorModel<NoArg,Dummy>
: public boost::static_visitor<ReturnType>
{
InternalVisitorModel() {}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return JointVisitorDerived::template algo<JointModelDerived>(jmodel.derived());
}
};
}; // struct JointVisitorBase
} // namespace fusion
} // namespace pinocchio
#include "pinocchio/multibody/visitor/joint-unary-visitor.hpp"
#endif // ifndef __pinocchio_multibody_visitor_hpp__
src/multibody/visitor/fusion.hpp 0 → 100644
View file @ 6e35342c
//
// Copyright (c) 2015-2019 CNRS INRIA
//
#ifndef __pinocchio_multibody_visitior_fusion_hpp__
#define __pinocchio_multibody_visitior_fusion_hpp__
#define BOOST_FUSION_INVOKE_MAX_ARITY 10
#include <boost/variant/static_visitor.hpp>
#include <boost/fusion/include/invoke.hpp>
#include <boost/fusion/container/generation/make_vector.hpp>
namespace boost
{
namespace fusion
{
// Append the element T at the front of boost fusion vector V.
template<typename T,typename V>
typename result_of::push_front<V const, T>::type
append(T const& t,V const& v)
{ return push_front(v,t); }
// Append the elements T1 and T2 at the front of boost fusion vector V.
template<typename T1,typename T2,typename V>
typename result_of::push_front<typename result_of::push_front<V const, T2>::type const, T1>::type
append2(T1 const& t1,T2 const& t2,V const& v)
{ return push_front(push_front(v,t2),t1); }
}
}
#endif // ifndef __pinocchio_multibody_visitior_fusion_hpp__
src/multibody/visitor/joint-unary-visitor.hpp 0 → 100644
View file @ 6e35342c
//
// Copyright (c) 2015-2019 CNRS INRIA
//
#ifndef __pinocchio_multibody_visitior_joint_unary_visitor_hpp__
#define __pinocchio_multibody_visitior_joint_unary_visitor_hpp__
#include <boost/variant.hpp>
#include "pinocchio/multibody/joint/joint-base.hpp"
#include "pinocchio/multibody/visitor/fusion.hpp"
namespace pinocchio
{
namespace fusion
{
namespace bf = boost::fusion;
typedef boost::blank NoArg;
template<typename JointVisitorDerived, typename ReturnType = void>
struct JointVisitorBase
{
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl, typename ArgsTmp>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
JointDataTpl<Scalar,Options,JointCollectionTpl> & jdata,
ArgsTmp args)
{
InternalVisitorModelAndData<JointModelTpl<Scalar,Options,JointCollectionTpl>,ArgsTmp> visitor(jdata,args);
return boost::apply_visitor(visitor,jmodel);
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
JointDataTpl<Scalar,Options,JointCollectionTpl> & jdata)
{
InternalVisitorModelAndData<JointModelTpl<Scalar,Options,JointCollectionTpl>,NoArg> visitor(jdata);
return boost::apply_visitor(visitor,jmodel);
}
template<typename JointModelDerived, typename ArgsTmp>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
typename JointModelBase<JointModelDerived>::JointDataDerived & jdata,
ArgsTmp args)
{
InternalVisitorModelAndData<JointModelDerived,ArgsTmp> visitor(jdata,args);
return visitor(jmodel.derived());
}
template<typename JointModelDerived>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
typename JointModelBase<JointModelDerived>::JointDataDerived & jdata)
{
InternalVisitorModelAndData<JointModelDerived,NoArg> visitor(jdata);
return visitor(jmodel.derived());
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl, typename ArgsTmp>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel,
ArgsTmp args)
{
InternalVisitorModel<ArgsTmp> visitor(args);
return boost::apply_visitor(visitor,jmodel);
}
template<typename Scalar, int Options, template<typename,int> class JointCollectionTpl>
static ReturnType run(const JointModelTpl<Scalar,Options,JointCollectionTpl> & jmodel)
{
InternalVisitorModel<NoArg> visitor;
return boost::apply_visitor(visitor,jmodel);
}
template<typename JointModelDerived, typename ArgsTmp>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel,
ArgsTmp args)
{
InternalVisitorModel<ArgsTmp> visitor(args);
return visitor(jmodel.derived());
}
template<typename JointModelDerived>
static ReturnType run(const JointModelBase<JointModelDerived> & jmodel)
{
InternalVisitorModel<NoArg> visitor;
return visitor(jmodel.derived());
}
private:
template<typename JointModel, typename ArgType>
struct InternalVisitorModelAndData
: public boost::static_visitor<ReturnType>
{
typedef typename JointModel::JointDataDerived JointData;
InternalVisitorModelAndData(JointData & jdata, ArgType args)
: jdata(jdata), args(args)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::append2(boost::ref(jmodel.derived()),
boost::ref(boost::get<typename JointModelBase<JointModelDerived>::JointDataDerived >(jdata)),
args));
}
ReturnType operator()(const JointModelVoid) {return;}
JointData & jdata;
ArgType args;
};
template<typename JointModel>
struct InternalVisitorModelAndData<JointModel,NoArg>
: public boost::static_visitor<ReturnType>
{
typedef typename JointModel::JointDataDerived JointData;
InternalVisitorModelAndData(JointData & jdata)
: jdata(jdata)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::make_vector(boost::ref(jmodel.derived()),
boost::ref(boost::get<typename JointModelBase<JointModelDerived>::JointDataDerived >(jdata)))
);
}
JointData & jdata;
};
template<typename ArgType, typename Dummy = void>
struct InternalVisitorModel
: public boost::static_visitor<ReturnType>
{
InternalVisitorModel(ArgType args)
: args(args)
{}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return bf::invoke(&JointVisitorDerived::template algo<JointModelDerived>,
bf::append(boost::ref(jmodel.derived()),
args));
}
ReturnType operator()(const JointModelVoid) {return;}
ArgType args;
};
template<typename Dummy>
struct InternalVisitorModel<NoArg,Dummy>
: public boost::static_visitor<ReturnType>
{
InternalVisitorModel() {}
template<typename JointModelDerived>
ReturnType operator()(const JointModelBase<JointModelDerived> & jmodel) const
{
return JointVisitorDerived::template algo<JointModelDerived>(jmodel.derived());
}
};
}; // struct JointVisitorBase
} // namespace fusion
} // namespace pinocchio
#endif // ifndef __pinocchio_multibody_visitior_joint_unary_visitor_hpp__
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment