Skip to content

Commit

Permalink
Init commit. Groundtruth plugin copied from PosePublisher default plu…
Browse files Browse the repository at this point in the history
…gin.
  • Loading branch information
KiselevIlia committed Mar 12, 2024
1 parent ee77e53 commit 5b1957b
Show file tree
Hide file tree
Showing 3 changed files with 691 additions and 0 deletions.
7 changes: 7 additions & 0 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -59,4 +59,11 @@ add_library(collision_plugin SHARED src/harmonic_collision.cpp ${PROTO_SRCS})
set_property(TARGET collision_plugin PROPERTY CXX_STANDARD 17)
target_link_libraries(collision_plugin
PRIVATE gz-sim${GAZEBO_VERSION}::gz-sim${GAZEBO_VERSION}
)

add_library(groundtruth_plugin SHARED src/groundtruth_plugin.cc)
target_link_libraries(groundtruth_plugin
PRIVATE gz-sim${GAZEBO_VERSION}::gz-sim${GAZEBO_VERSION}
PRIVATE gz-plugin${GZ_PLUGIN_VER}::gz-plugin${GZ_PLUGIN_VER}
PRIVATE gz-msgs10::gz-msgs10
)
91 changes: 91 additions & 0 deletions include/groundtruth_plugin.hh
Original file line number Diff line number Diff line change
@@ -0,0 +1,91 @@
/*
* Copyright (C) 2019 Open Source Robotics Foundation
*
* 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.
*
*/
#ifndef GZ_SIM_SYSTEMS_POSEPUBLISHER_HH_
#define GZ_SIM_SYSTEMS_POSEPUBLISHER_HH_

#include <memory>
#include <gz/sim/config.hh>
#include <gz/sim/System.hh>

namespace gz
{
namespace sim
{
// Inline bracket to help doxygen filtering.
inline namespace GZ_SIM_VERSION_NAMESPACE {
namespace systems
{
// Forward declaration
class PosePublisherPrivate;

/// \brief Pose publisher system. Attach to an entity to publish the
/// transform of its child entities in the form of gz::msgs::Pose
/// messages, or a single gz::msgs::Pose_V message if
/// "use_pose_vector_msg" is true.
///
/// ## System Parameters
///
/// - `<publish_link_pose>`: Set to true to publish link pose
/// - `<publish_visual_pose>`: Set to true to publish visual pose
/// - `<publish_collision_pose>`: Set to true to publish collision pose
/// - `<publish_sensor_pose>`: Set to true to publish sensor pose
/// - `<publish_model_pose>`: Set to true to publish model pose.
/// - `<publish_nested_model_pose>`: Set to true to publish nested model
/// pose. The pose of the model that contains this system is also published
/// unless publish_model_pose is set to false
/// - `<use_pose_vector_msg>`: Set to true to publish a gz::msgs::Pose_V
/// message instead of mulitple gz::msgs::Pose messages.
/// - `<update_frequency>`: Frequency of pose publications in Hz. A negative
/// frequency publishes as fast as possible (i.e, at the rate of the
/// simulation step)
/// - `<static_publisher>`: Set to true to publish static poses on a
/// "<scoped_entity_name>/pose_static" topic. This will cause only dynamic
/// poses to be published on the "<scoped_entity_name>/pose" topic.
/// - `<static_update_frequency>`: Frequency of static pose publications in
/// Hz. A negative frequency publishes as fast as possible (i.e, at the
/// rate of the simulation step).
class PosePublisher
: public System,
public ISystemConfigure,
public ISystemPostUpdate
{
/// \brief Constructor
public: PosePublisher();

/// \brief Destructor
public: ~PosePublisher() override = default;

// Documentation inherited
public: void Configure(const Entity &_entity,
const std::shared_ptr<const sdf::Element> &_sdf,
EntityComponentManager &_ecm,
EventManager &_eventMgr) override;

// Documentation inherited
public: void PostUpdate(
const UpdateInfo &_info,
const EntityComponentManager &_ecm) override;

/// \brief Private data pointer
private: std::unique_ptr<PosePublisherPrivate> dataPtr;
};
}
}
}
}

#endif
Loading

0 comments on commit 5b1957b

Please sign in to comment.