object.hpp

Go to the documentation of this file.

/*
 * Copyright 2020 Robert Bosch GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
#ifndef CLOE_COMPONENT_OBJECT_HPP_
#define CLOE_COMPONENT_OBJECT_HPP_

#include <memory>  // for shared_ptr<>
#include <vector>  // for vector<>

#include <Eigen/Geometry>             // for Isometry3d, Vector3d
#include <fable/json/with_eigen.hpp>  // for to_json

#include <cloe/core.hpp>  // for Json

namespace cloe {

struct Object {
  enum class Type { Unknown, Static, Dynamic };

  enum class Class { Unknown, Pedestrian, Bike, Motorbike, Car, Truck, Trailer };

  int id{-1};

  double exist_prob{1.0};

  Type type{Type::Unknown};

  Class classification{Class::Unknown};

  Eigen::Isometry3d pose{Eigen::Isometry3d()};

  Eigen::Vector3d dimensions{Eigen::Vector3d::Zero()};

  Eigen::Vector3d cog_offset{Eigen::Vector3d::Zero()};

  Eigen::Vector3d velocity{Eigen::Vector3d::Zero()};

  Eigen::Vector3d acceleration{Eigen::Vector3d::Zero()};

  Eigen::Vector3d angular_velocity{Eigen::Vector3d::Zero()};

  Object() = default;

  friend void to_json(Json& j, const Object& o) {
    j = Json{
        {"id", o.id},
        {"exist_prob", o.exist_prob},
        {"type", o.type},
        {"class", o.classification},
        {"pose", o.pose},
        {"dimensions", o.dimensions},
        {"cog_offset", o.cog_offset},
        {"velocity", o.velocity},
        {"velocity_norm", static_cast<double>(o.velocity.norm())},
        {"acceleration", o.acceleration},
        {"angular_velocity", o.angular_velocity},
    };
  }
};

using Objects = std::vector<std::shared_ptr<Object>>;

inline void to_json(Json& j, const Objects& os) {
  j = Json::array();
  for (const auto& o : os) {
    assert(o != nullptr);
    j.push_back(*o);
  }
}

// clang-format off
ENUM_SERIALIZATION(Object::Type, ({
      {Object::Type::Static, "static"},
      {Object::Type::Dynamic, "dynamic"},
      {Object::Type::Unknown, "unknown"},
}))

ENUM_SERIALIZATION(Object::Class, ({
      {Object::Class::Unknown, "unknown"},
      {Object::Class::Pedestrian, "pedestrian"},
      {Object::Class::Bike, "bicycle"},
      {Object::Class::Motorbike, "motorcycle"},
      {Object::Class::Car, "car"},
      {Object::Class::Truck, "truck"},
      {Object::Class::Trailer, "trailer"},
}))
// clang-format on

}  // namespace cloe

#endif  // CLOE_COMPONENT_OBJECT_HPP_