SpatialDetectionConverter.hpp

Go to the documentation of this file.

#pragma once

#include <depthai_bridge/ImageConverter.hpp>

#include "depthai/depthai.hpp"
#ifdef IS_ROS2
    #include <depthai_ros_msgs/msg/spatial_detection_array.hpp>

    #include "rclcpp/rclcpp.hpp"
#else
    #include <depthai_ros_msgs/SpatialDetectionArray.h>
    #include <ros/ros.h>

    #include <boost/make_shared.hpp>
    #include <boost/shared_ptr.hpp>
#endif

namespace dai {

namespace ros {

#ifdef IS_ROS2
namespace SpatialMessages = depthai_ros_msgs::msg;
using SpatialDetectionArrayPtr = SpatialMessages::SpatialDetectionArray::SharedPtr;
#else
namespace SpatialMessages = depthai_ros_msgs;
using SpatialDetectionArrayPtr = SpatialMessages::SpatialDetectionArray::Ptr;
#endif
class SpatialDetectionConverter {
   public:
    // DetectionConverter() = default;
    SpatialDetectionConverter(std::string frameName, int width, int height, bool normalized = false);

    void toRosMsg(std::shared_ptr<dai::SpatialImgDetections> inNetData, std::deque<SpatialMessages::SpatialDetectionArray>& opDetectionMsg);

    SpatialDetectionArrayPtr toRosMsgPtr(std::shared_ptr<dai::SpatialImgDetections> inNetData);

   private:
    int _width, _height;
    const std::string _frameName;
    bool _normalized;
};

}  // namespace ros

namespace rosBridge = ros;

}  // namespace dai