enviroment base interface class

This commit is contained in:
Ilya Uraev 2023-10-15 16:24:30 +03:00
parent 48e17c2481
commit 71063b19aa
2 changed files with 84 additions and 5 deletions

View file

@ -1,5 +1,5 @@
#include "rclcpp/rclcpp.hpp"
#include "rclcpp_lifecycle/node_interfaces/lifecycle_node_interface.hpp"
#include "rclcpp_lifecycle/lifecycle_node.hpp"
// TODO: BaseInterface class for next enviroments based on LifecycleNodeInterface
// It's should be like ControllerInterfaceBase class from ros2_control
@ -9,16 +9,32 @@ namespace env_interface
using CallbackReturn = rclcpp_lifecycle::node_interfaces::LifecycleNodeInterface::CallbackReturn;
class EnvBaseIntrerface : public rclcpp_lifecycle::node_interfaces::LifecycleNodeInterface
class EnvInterfaceBase : public rclcpp_lifecycle::node_interfaces::LifecycleNodeInterface
{
public:
EnvBaseIntrerface() = default;
EnvInterfaceBase() = default;
virtual ~EnvBaseIntrerface() = default;
virtual ~EnvInterfaceBase() = default;
virtual CallbackReturn init(
const std::string& t_env_name,
const std::string & t_namespace = "",
const rclcpp::NodeOptions & t_node_options = rclcpp::NodeOptions()
);
const rclcpp_lifecycle::State& configure();
virtual CallbackReturn on_init() = 0;
std::shared_ptr<rclcpp_lifecycle::LifecycleNode> getNode();
std::shared_ptr<rclcpp_lifecycle::LifecycleNode> getNode() const;
const rclcpp_lifecycle::State & getState() const;
private:
std::shared_ptr<rclcpp_lifecycle::LifecycleNode> m_node;
};

View file

@ -0,0 +1,63 @@
#include "env_manager/env_base_interface.hpp"
namespace env_interface
{
CallbackReturn
EnvInterfaceBase::init(
const std::string& t_env_name,
const std::string & t_namespace,
const rclcpp::NodeOptions & t_node_options)
{
m_node = std::make_shared<rclcpp_lifecycle::LifecycleNode>(
t_env_name, t_namespace, t_node_options, false);
if (on_init() == CallbackReturn::FAILURE)
{
return CallbackReturn::FAILURE;
}
m_node->register_on_configure(
std::bind(&EnvInterfaceBase::on_configure, this, std::placeholders::_1)
);
m_node->register_on_activate(
std::bind(&EnvInterfaceBase::on_activate, this, std::placeholders::_1)
);
m_node->register_on_cleanup(
std::bind(&EnvInterfaceBase::on_cleanup, this, std::placeholders::_1)
);
m_node->register_on_deactivate(
std::bind(&EnvInterfaceBase::on_deactivate, this, std::placeholders::_1)
);
m_node->register_on_error(
std::bind(&EnvInterfaceBase::on_error, this, std::placeholders::_1)
);
m_node->register_on_shutdown(
std::bind(&EnvInterfaceBase::on_shutdown, this, std::placeholders::_1)
);
return CallbackReturn::SUCCESS;
}
const rclcpp_lifecycle::State&
EnvInterfaceBase::configure()
{
return getNode()->configure();
}
std::shared_ptr<rclcpp_lifecycle::LifecycleNode>
EnvInterfaceBase::getNode()
{
if (!m_node.get())
{
throw std::runtime_error("Lifecycle node hasn't been initialized yet");
}
return m_node;
}
const rclcpp_lifecycle::State & EnvInterfaceBase::getState() const
{
return m_node->get_current_state();
}
} // namespace env_interface