Merge branch 'object_detection' of github.com:SemvdH/5g_drone_ROS2 into object_detection

src/object_detection/src/lidar_reader.cpp

@@ -2,6 +2,7 @@
 #include <functional>
 #include <memory>
 #include <string>
+#include <iostream>
 
 #include "rclcpp/rclcpp.hpp"
 #include "object_detection/msg/lidar_reading.hpp"
@@ -11,34 +12,84 @@
 #include <terabee/TowerDistanceData.hpp>
 #include <terabee/ImuData.hpp>
 
+using terabee::ImuData;
 using terabee::ITerarangerTowerEvo;
 using terabee::TowerDistanceData;
-using terabee::ImuData;
 
 using namespace std::chrono_literals;
 
 /* This example creates a subclass of Node and uses std::bind() to register a
-* member function as a callback from the timer. */
+ * member function as a callback from the timer. */
+
+std::ostream &operator<<(std::ostream &os, const TowerDistanceData &d)
+{
+  os << "[";
+  for (size_t i = 0; i < d.distance.size(); i++)
+  {
+    os << d.distance[i] << (d.mask[i] ? " <new>, " : " <old>, ");
+  }
+  os << "\b\b"
+     << " ]";
+  return os;
+}
+
+std::ostream &operator<<(std::ostream &os, const ImuData &d)
+{
+  os << "[";
+  for (size_t i = 0; i < d.data.size(); i++)
+  {
+    os << d.data[i] << ", ";
+  }
+  os << "\b\b"
+     << " ]";
+  return os;
+}
 
 class LidarReader : public rclcpp::Node
 {
-  public:
+public:
-    LidarReader()
+  LidarReader()
-    : Node("lidar_reader")
+      : Node("lidar_reader")
-    { 
+  {
-      publisher_ = this->create_publisher<object_detection::msg::LidarReading>("drone/object_detection", 10);
+    publisher_ = this->create_publisher<object_detection::msg::LidarReading>("drone/object_detection", 10);
-    }
+    timer_ = this->create_wall_timer(500ms, std::bind(&LidarReader::timer_callback, this));
 
-  private:
+    ITerarangerTowerEvo::ImuMode mode(ITerarangerTowerEvo::QuaternionLinearAcc);
-    void timer_callback()
+
+    factory = terabee::ITerarangerFactory::getFactory();
+    tower = factory.createTerarangerTowerEvo("/dev/ttyACM0");
+
+    if (!tower)
     {
-      RCLCPP_INFO(this->get_logger(), "yeet");
+      RCLCPP_ERROR(this->get_logger(), "Failed to create TerarangerTowerEvo");
+      return;
     }
 
-    rclcpp::Publisher<object_detection::msg::LidarReading>::SharedPtr publisher_;
+    tower->setImuMode(mode);
+
+    if (!tower->initialize())
+    {
+      RCLCPP_ERROR(this->get_logger(), "Failed to initialize TerarangerTowerEvo");
+      return;
+    }
+  }
+
+private:
+  void timer_callback()
+  {
+    std::cout << "Distance = " << tower->getDistance() << std::endl;
+    std::cout << "IMU = " << tower->getImuData() << std::endl;
+  }
+
+  rclcpp::Publisher<object_detection::msg::LidarReading>::SharedPtr publisher_;
+  rclcpp::TimerBase::SharedPtr timer_;
+
+  // terabee tower evo variables
+  terabee::ITerarangerTowerEvo tower;
+  terabee::ITerarangerFactory factory;
 };
 
-int main(int argc, char * argv[])
+int main(int argc, char *argv[])
 {
   rclcpp::init(argc, argv);
   rclcpp::spin(std::make_shared<LidarReader>());