setpoint_base.hpp

/****************************************************************************
 * Copyright (c) 2023 PX4 Development Team.
 * SPDX-License-Identifier: BSD-3-Clause
 ****************************************************************************/
 
#pragma once
 
#include <cstdint>
#include <functional>
#include <memory>
#include <px4_msgs/msg/setpoint_config.hpp>
#include <px4_msgs/msg/setpoint_config_reply.hpp>
#include <rclcpp/rclcpp.hpp>
 
#include "context.hpp"
#include "exception.hpp"
 
namespace px4_ros2 {
 
class SetpointBase : public std::enable_shared_from_this<SetpointBase> {
 public:
  using ShouldActivateCB = std::function<void()>;
  using SetpointType = decltype(px4_msgs::msg::SetpointConfig::type);
 
  explicit SetpointBase(Context& context) { context.addSetpointType(this); }
 
  virtual ~SetpointBase() = default;
 
  std::shared_ptr<SetpointBase> getSharedPtr()
  {
    try {
      return shared_from_this();
    } catch (const std::bad_weak_ptr& exception) {
      throw Exception("Setpoint must be instantiated with std::make_shared<>");
    }
    return {};
  }
 
  virtual SetpointType getSetpointType() = 0;
 
  virtual void clearOptionalRequirements(px4_msgs::msg::SetpointConfigReply& setpoint_config_reply)
  {
  }
 
  virtual float desiredUpdateRateHz() { return 50.f; }
 
  void setShouldActivateCallback(const ShouldActivateCB& should_activate_cb)
  {
    _should_activate_cb = should_activate_cb;
  }
  void setActive(bool active) { _active = active; }
 
 protected:
  void onUpdate()
  {
    if (!_active && _should_activate_cb) {
      _should_activate_cb();
    }
  }
 
 private:
  ShouldActivateCB _should_activate_cb;
  bool _active{false};
};
 
} /* namespace px4_ros2 */