ros2_parsers/imu_msg.h

Go to the documentation of this file.

#pragma once

#include <sensor_msgs/msg/imu.hpp>
#include "fmt/format.h"
#include "ros2_parser.h"
#include "covariance_util.h"
#include "quaternion_msg.h"
#include "header_msg.h"

class ImuMsgParser : public BuiltinMessageParser<sensor_msgs::msg::Imu>
{
 public:
  ImuMsgParser(const std::string& topic_name, PJ::PlotDataMapRef& plot_data)
      : BuiltinMessageParser<sensor_msgs::msg::Imu>(topic_name, plot_data)
        , _header_parser(topic_name + "/header", plot_data)
        , _quat_parser(topic_name + "/orientation", plot_data)
        , _orientation_covariance(topic_name + "/orientation_covariance", plot_data)
        , _lin_acc_covariance(topic_name + "/linear_acceleration_covariance", plot_data)
        , _ang_vel_covariance(topic_name + "/angular_velocity_covariance", plot_data)
  {
  }

  void parseMessageImpl(const sensor_msgs::msg::Imu& msg, double& timestamp) override
  {
    if( !_initialized )
    {
      _initialized = true;
      _data.push_back(&getSeries(_topic_name + "/angular_velocity/x"));
      _data.push_back(&getSeries(_topic_name + "/angular_velocity/y"));
      _data.push_back(&getSeries(_topic_name + "/angular_velocity/z"));

      _data.push_back(&getSeries(_topic_name + "/linear_acceleration/x"));
      _data.push_back(&getSeries(_topic_name + "/linear_acceleration/y"));
      _data.push_back(&getSeries(_topic_name + "/linear_acceleration/z"));
    }

    _header_parser.parse(msg.header, timestamp, _config.use_header_stamp);

    _data[0]->pushBack({ timestamp, msg.angular_velocity.x });
    _data[1]->pushBack({ timestamp, msg.angular_velocity.y });
    _data[2]->pushBack({ timestamp, msg.angular_velocity.z });

    _data[3]->pushBack({ timestamp, msg.linear_acceleration.x });
    _data[4]->pushBack({ timestamp, msg.linear_acceleration.y });
    _data[5]->pushBack({ timestamp, msg.linear_acceleration.z });

    _quat_parser.parseMessageImpl(msg.orientation, timestamp);

    _orientation_covariance.parse(msg.orientation_covariance, timestamp);
    _lin_acc_covariance.parse(msg.linear_acceleration_covariance, timestamp);
    _ang_vel_covariance.parse(msg.angular_velocity_covariance, timestamp);
  }

 private:
  HeaderMsgParser _header_parser;
  QuaternionMsgParser _quat_parser;
  CovarianceParser<3> _orientation_covariance;
  CovarianceParser<3> _lin_acc_covariance;
  CovarianceParser<3> _ang_vel_covariance;
  std::vector<PJ::PlotData*> _data;

  bool _initialized = false;
};