Skip to content

Commit 5e3e565

Browse files
committed
add namespace for data_ros_utils
1 parent 42422c9 commit 5e3e565

File tree

4 files changed

+63
-47
lines changed

4 files changed

+63
-47
lines changed

autonomy_core/map_plan/action_planner/src/global_plan_server.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,8 +257,8 @@ void GlobalPlanServer::process_goal() {
257257
}
258258

259259
// record current odometry as start
260-
start.pos = pose_to_eigen(odom_msg_->pose.pose);
261-
goal.pos = pose_to_eigen(goal_->p_final);
260+
start.pos = kr_planning_rviz_plugins::pose_to_eigen(odom_msg_->pose.pose);
261+
goal.pos = kr_planning_rviz_plugins::pose_to_eigen(goal_->p_final);
262262

263263
// call the planner
264264

@@ -395,7 +395,7 @@ bool GlobalPlanServer::global_plan_process(
395395
}
396396

397397
// publish global_path_msg_
398-
global_path_msg_ = path_to_ros(global_path);
398+
global_path_msg_ = kr_planning_rviz_plugins::path_to_ros(global_path);
399399
global_path_msg_.header.frame_id = map_frame;
400400
path_pub_.publish(global_path_msg_);
401401
}
@@ -414,7 +414,7 @@ bool GlobalPlanServer::global_plan_process(
414414
}
415415

416416
// publish
417-
global_path_msg_ = path_to_ros(global_path);
417+
global_path_msg_ = kr_planning_rviz_plugins::path_to_ros(global_path);
418418
global_path_msg_.header.frame_id = map_frame;
419419
path_pub_.publish(global_path_msg_);
420420
}

autonomy_core/map_plan/action_planner/src/local_plan_server.cpp

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -325,10 +325,10 @@ void LocalPlanServer::process_goal() {
325325
MPL::Waypoint3D start, goal;
326326
// instead of using current odometry as start, we use the given start position
327327
// for consistency between old and new trajectories in replan process
328-
start.pos = pose_to_eigen(goal_->p_init);
329-
start.vel = twist_to_eigen(goal_->v_init);
330-
start.acc = twist_to_eigen(goal_->a_init);
331-
start.jrk = twist_to_eigen(goal_->j_init);
328+
start.pos = kr_planning_rviz_plugins::pose_to_eigen(goal_->p_init);
329+
start.vel = kr_planning_rviz_plugins::twist_to_eigen(goal_->v_init);
330+
start.acc = kr_planning_rviz_plugins::twist_to_eigen(goal_->a_init);
331+
start.jrk = kr_planning_rviz_plugins::twist_to_eigen(goal_->j_init);
332332

333333
// Important: define use position, velocity, acceleration or jerk as control
334334
// inputs, note that the lowest order "false" term will be used as control
@@ -342,10 +342,10 @@ void LocalPlanServer::process_goal() {
342342
// in trajectory_tracker)
343343
start.use_yaw = false;
344344

345-
goal.pos = pose_to_eigen(goal_->p_final);
346-
goal.vel = twist_to_eigen(goal_->v_final);
347-
goal.acc = twist_to_eigen(goal_->a_final);
348-
goal.jrk = twist_to_eigen(goal_->j_final);
345+
goal.pos = kr_planning_rviz_plugins::pose_to_eigen(goal_->p_final);
346+
goal.vel = kr_planning_rviz_plugins::twist_to_eigen(goal_->v_final);
347+
goal.acc = kr_planning_rviz_plugins::twist_to_eigen(goal_->a_final);
348+
goal.jrk = kr_planning_rviz_plugins::twist_to_eigen(goal_->j_final);
349349
goal.use_yaw = start.use_yaw;
350350
goal.use_pos = start.use_pos;
351351
goal.use_vel = start.use_vel;
@@ -440,7 +440,7 @@ bool LocalPlanServer::local_plan_process(
440440
vec_Vec3f sg;
441441
sg.push_back(start.pos);
442442
sg.push_back(goal.pos);
443-
kr_planning_msgs::Path sg_msg = path_to_ros(sg);
443+
kr_planning_msgs::Path sg_msg = kr_planning_rviz_plugins::path_to_ros(sg);
444444
std::string map_frame; // set frame id
445445
map_frame = map.header.frame_id;
446446
sg_msg.header.frame_id = map_frame;
@@ -458,8 +458,8 @@ bool LocalPlanServer::local_plan_process(
458458
}
459459

460460
// for visualization: publish expanded nodes as a point cloud
461-
sensor_msgs::PointCloud expanded_ps =
462-
vec_to_cloud(mp_planner_util_->getExpandedNodes());
461+
sensor_msgs::PointCloud expanded_ps = kr_planning_rviz_plugins::vec_to_cloud(
462+
mp_planner_util_->getExpandedNodes());
463463
expanded_ps.header.frame_id = map_frame;
464464
expanded_cloud_pub.publish(expanded_ps);
465465

autonomy_core/map_plan/mapper/src/local_global_mapper.cpp

Lines changed: 37 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ LocalGlobalMapperNode::LocalGlobalMapperNode(const ros::NodeHandle& nh)
44
: nh_(nh) {
55
initParams();
66

7-
cloud_sub = nh_.subscribe(cloud_name_, 1,
8-
&LocalGlobalMapperNode::cloudCallback, this);
7+
cloud_sub = nh_.subscribe(
8+
cloud_name_, 1, &LocalGlobalMapperNode::cloudCallback, this);
99

1010
global_map_pub =
1111
nh_.advertise<kr_planning_msgs::VoxelMap>("global_voxel_map", 1, true);
@@ -116,9 +116,12 @@ void LocalGlobalMapperNode::globalMapInit() {
116116
const double global_res = global_map_info_.resolution;
117117
int8_t global_val_default = 0;
118118
// Initialize the mapper
119-
global_voxel_mapper_.reset(new mapper::VoxelMapper(
120-
global_origin, global_dim_d, global_res, global_val_default,
121-
global_decay_times_to_empty_));
119+
global_voxel_mapper_.reset(
120+
new mapper::VoxelMapper(global_origin,
121+
global_dim_d,
122+
global_res,
123+
global_val_default,
124+
global_decay_times_to_empty_));
122125

123126
// build the array for map inflation
124127
global_infla_array_.clear();
@@ -153,9 +156,12 @@ void LocalGlobalMapperNode::storageMapInit() {
153156
const double res = storage_map_info_.resolution;
154157
int8_t storage_val_default = 0;
155158
// Initialize the mapper
156-
storage_voxel_mapper_.reset(new mapper::VoxelMapper(
157-
storage_origin, storage_dim_d, res, storage_val_default,
158-
local_decay_times_to_empty_));
159+
storage_voxel_mapper_.reset(
160+
new mapper::VoxelMapper(storage_origin,
161+
storage_dim_d,
162+
res,
163+
storage_val_default,
164+
local_decay_times_to_empty_));
159165
}
160166

161167
void LocalGlobalMapperNode::localInflaInit() {
@@ -213,15 +219,17 @@ void LocalGlobalMapperNode::getLidarPoses(
213219
// for real robot, the point cloud frame_id may not exist in the tf tree,
214220
// manually defining it here.
215221
// TODO(xu): make this automatic
216-
auto tf_map_lidar = tf_listener.LookupTransform(map_frame_, lidar_frame_,
217-
cloud_header.stamp);
218-
auto tf_odom_lidar = tf_listener.LookupTransform(odom_frame_, lidar_frame_,
219-
cloud_header.stamp);
222+
auto tf_map_lidar = tf_listener.LookupTransform(
223+
map_frame_, lidar_frame_, cloud_header.stamp);
224+
auto tf_odom_lidar = tf_listener.LookupTransform(
225+
odom_frame_, lidar_frame_, cloud_header.stamp);
220226
if ((!tf_map_lidar) || (!tf_odom_lidar)) {
221227
ROS_WARN(
222228
"[Mapper real-robot:] Failed to get transform (either from %s to %s; "
223229
"or from %s to %s)",
224-
lidar_frame_.c_str(), map_frame_.c_str(), lidar_frame_.c_str(),
230+
lidar_frame_.c_str(),
231+
map_frame_.c_str(),
232+
lidar_frame_.c_str(),
225233
odom_frame_.c_str());
226234
return;
227235
} else {
@@ -237,13 +245,15 @@ void LocalGlobalMapperNode::getLidarPoses(
237245
ROS_WARN(
238246
"[Mapper simulation:] Failed to get transform map to lidar (from %s "
239247
"to %s)",
240-
cloud_header.frame_id.c_str(), map_frame_.c_str());
248+
cloud_header.frame_id.c_str(),
249+
map_frame_.c_str());
241250
return;
242251
} else if (!tf_odom_lidar) {
243252
ROS_WARN(
244253
"[Mapper simulation:] Failed to get transform odom to lidar (from %s "
245254
"to %s)",
246-
cloud_header.frame_id.c_str(), odom_frame_.c_str());
255+
cloud_header.frame_id.c_str(),
256+
odom_frame_.c_str());
247257
return;
248258
} else {
249259
*pose_map_lidar_ptr = *tf_map_lidar;
@@ -262,8 +272,10 @@ void LocalGlobalMapperNode::processCloud(const sensor_msgs::PointCloud& cloud) {
262272
geometry_msgs::Pose pose_odom_lidar;
263273
getLidarPoses(cloud.header, &pose_map_lidar, &pose_odom_lidar);
264274

265-
const Eigen::Affine3d T_map_lidar = toTF(pose_map_lidar);
266-
const Eigen::Affine3d T_odom_lidar = toTF(pose_odom_lidar);
275+
const Eigen::Affine3d T_map_lidar =
276+
kr_planning_rviz_plugins::toTF(pose_map_lidar);
277+
const Eigen::Affine3d T_odom_lidar =
278+
kr_planning_rviz_plugins::toTF(pose_odom_lidar);
267279

268280
// This is the lidar position in the odom frame, used for raytracing &
269281
// cropping local map
@@ -283,12 +295,13 @@ void LocalGlobalMapperNode::processCloud(const sensor_msgs::PointCloud& cloud) {
283295
double min_range = 0.75; // points within this distance will be discarded
284296
double min_range_squared;
285297
min_range_squared = min_range * min_range;
286-
const auto pts = cloud_to_vec_filter(cloud, min_range_squared);
298+
const auto pts =
299+
kr_planning_rviz_plugins::cloud_to_vec_filter(cloud, min_range_squared);
287300

288301
timer.start();
289302
// local raytracing using lidar position in the map frame (not odom frame)
290-
storage_voxel_mapper_->addCloud(pts, T_map_lidar, local_infla_array_, false,
291-
local_max_raycast_);
303+
storage_voxel_mapper_->addCloud(
304+
pts, T_map_lidar, local_infla_array_, false, local_max_raycast_);
292305
ROS_DEBUG("[storage map addCloud]: %f",
293306
static_cast<double>(timer.elapsed().wall) / 1e6);
294307

@@ -311,8 +324,8 @@ void LocalGlobalMapperNode::processCloud(const sensor_msgs::PointCloud& cloud) {
311324
++counter_;
312325
if (counter_ % update_interval_ == 0) {
313326
timer.start();
314-
global_voxel_mapper_->addCloud(pts, T_map_lidar, global_infla_array_, false,
315-
global_max_raycast_);
327+
global_voxel_mapper_->addCloud(
328+
pts, T_map_lidar, global_infla_array_, false, global_max_raycast_);
316329
ROS_DEBUG("[global map addCloud]: %f",
317330
static_cast<double>(timer.elapsed().wall) / 1e6);
318331
timer.start();
@@ -334,8 +347,8 @@ void LocalGlobalMapperNode::processCloud(const sensor_msgs::PointCloud& cloud) {
334347
global_map_pub.publish(global_map);
335348
}
336349

337-
ROS_DEBUG_THROTTLE(1, "[Mapper]: Got cloud, number of points: [%zu]",
338-
cloud.points.size());
350+
ROS_DEBUG_THROTTLE(
351+
1, "[Mapper]: Got cloud, number of points: [%zu]", cloud.points.size());
339352
}
340353

341354
void LocalGlobalMapperNode::cloudCallback(

autonomy_core/state_machine/state_machine_core/src/local_global_replan_server.cpp

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
*/
66
void RePlanner::GlobalPathCb(const kr_planning_msgs::Path& path) {
77
global_path_.clear();
8-
global_path_ = ros_to_path(path); // extract the global path information
8+
global_path_ = kr_planning_rviz_plugins::ros_to_path(
9+
path); // extract the global path information
910
}
1011

1112
void RePlanner::EpochCb(const std_msgs::Int64& msg) {
@@ -282,7 +283,7 @@ void RePlanner::setup_replanner() {
282283
}
283284
auto global_result = global_plan_client_->getResult();
284285
if (global_result->success) {
285-
global_path_ = ros_to_path(
286+
global_path_ = kr_planning_rviz_plugins::ros_to_path(
286287
global_result->path); // extract the global path information
287288
ROS_WARN("initial global plan succeeded!");
288289
} else {
@@ -459,7 +460,7 @@ bool RePlanner::PlanTrajectory(int horizon) {
459460
&local_tpgoal.a_init,
460461
&local_tpgoal.j_init);
461462
Vec3f start_pos;
462-
start_pos = pose_to_eigen(local_tpgoal.p_init);
463+
start_pos = kr_planning_rviz_plugins::pose_to_eigen(local_tpgoal.p_init);
463464

464465
// Replan step 1: Global plan
465466
// ########################################################################################################
@@ -770,7 +771,8 @@ vec_Vec3f RePlanner::PathCropIntersect(const vec_Vec3f& path) {
770771
}
771772

772773
// publish for visualization
773-
kr_planning_msgs::Path local_path_msg_ = path_to_ros(cropped_path);
774+
kr_planning_msgs::Path local_path_msg_ =
775+
kr_planning_rviz_plugins::path_to_ros(cropped_path);
774776
local_path_msg_.header.frame_id = map_frame_;
775777
cropped_path_pub_.publish(local_path_msg_);
776778

@@ -815,7 +817,7 @@ vec_Vec3f RePlanner::TransformGlobalPath(const vec_Vec3f& path_original) {
815817
map_to_odom.orientation.z = transformStamped.transform.rotation.z;
816818

817819
// TF transform from the map frame to odom frame
818-
auto map_to_odom_tf = toTF(map_to_odom);
820+
auto map_to_odom_tf = kr_planning_rviz_plugins::toTF(map_to_odom);
819821
Vec3f waypoint_wrt_map;
820822

821823
vec_Vec3f path_wrt_map;
@@ -826,7 +828,8 @@ vec_Vec3f RePlanner::TransformGlobalPath(const vec_Vec3f& path_original) {
826828
}
827829

828830
// publish transformed global path for visualization
829-
kr_planning_msgs::Path path_wrt_map_msg = path_to_ros(path_wrt_map);
831+
kr_planning_msgs::Path path_wrt_map_msg =
832+
kr_planning_rviz_plugins::path_to_ros(path_wrt_map);
830833
path_wrt_map_msg.header.frame_id = map_frame_;
831834
global_path_wrt_map_pub_.publish(path_wrt_map_msg);
832835
return path_wrt_map;
@@ -1033,8 +1036,8 @@ RePlanner::RePlanner() : nh_("~") {
10331036
cropped_path_pub_ =
10341037
priv_nh.advertise<kr_planning_msgs::Path>("cropped_local_path", 1, true);
10351038

1036-
global_path_wrt_map_pub_ = priv_nh.advertise<kr_planning_msgs::Path>(
1037-
"global_path_wrt_map", 1, true);
1039+
global_path_wrt_map_pub_ =
1040+
priv_nh.advertise<kr_planning_msgs::Path>("global_path_wrt_map", 1, true);
10381041

10391042
priv_nh.param("max_horizon", max_horizon_, 5);
10401043
priv_nh.param("crop_radius", crop_radius_, 10.0);

0 commit comments

Comments
 (0)