Align multiple DEMs with TF (#34)

* Remove terrain alignment

* Make frame_id a member of GridMapGeo

* Broadcast Tf

* Load multiple tif files

include/grid_map_geo/grid_map_geo.hpp

@@ -61,7 +61,7 @@ struct Location {
 
 class GridMapGeo {
  public:
-  GridMapGeo();
+  GridMapGeo(const std::string frame_id = "map");
   virtual ~GridMapGeo();
 
   /**
@@ -97,33 +97,24 @@ class GridMapGeo {
     origin = maporigin_.position;
   };
 
-  /**
-   * @brief Set the Altitude Origin object
-   *
-   * @param altitude
-   */
-  void setAltitudeOrigin(const double altitude) { localorigin_altitude_ = altitude; };
-
   /**
    * @brief Helper function for loading terrain from path
    *
    * @param map_path Path to dsm path (Supported formats are *.tif)
-   * @param algin_terrain Geo align terrain
    * @param color_map_path  Path to color raster files to visualize terrain texture (Supported formats are *.tif)
    * @return true Successfully loaded terrain
    * @return false Failed to load terrain
    */
-  bool Load(const std::string& map_path, bool algin_terrain, const std::string color_map_path = "");
+  bool Load(const std::string& map_path, const std::string color_map_path = "");
 
   /**
    * @brief Initialize grid map from a geotiff file
    *
    * @param path Path to dsm path (Supported formats are *.tif)
-   * @param align_terrain
    * @return true Successfully loaded terrain
    * @return false Failed to load terrain
    */
-  bool initializeFromGeotiff(const std::string& path, bool align_terrain = true);
+  bool initializeFromGeotiff(const std::string& path);
 
   /**
    * @brief Load a color layer from a geotiff file (orthomosaic)
@@ -187,9 +178,7 @@ class GridMapGeo {
 
  protected:
   grid_map::GridMap grid_map_;
-  double localorigin_e_{789823.93};  // duerrboden berghaus
-  double localorigin_n_{177416.56};
-  double localorigin_altitude_{0.0};
   Location maporigin_;
+  std::string frame_id_{""};
 };
 #endif

launch/load_multiple_tif.launch

@@ -0,0 +1,22 @@
+<launch>
+    <arg name="visualization" default="true"/>
+    <arg name="location" default="sargans"/>
+    <node pkg="tf" type="static_transform_publisher" name="world_map" args="0 0 0 0 0 0 world map 10"/>
+
+    <node pkg="grid_map_geo" type="test_tif_loader" name="first_tif_loader" output="screen">
+        <param name="frame_id" value="map"/>
+        <param name="tif_path" value="$(find grid_map_geo)/resources/sertig.tif"/>
+        <!-- <param name="color_path" value="$(find grid_map_geo)/resources/sertig_color.tif"/> -->
+    </node>
+
+    <node pkg="grid_map_geo" type="test_tif_loader" name="second_tif_loader2" output="screen">
+        <param name="frame_id" value="dischma_valley"/>
+        <param name="tif_path" value="$(find terrain_models)/models/dischma_valley.tif"/>
+        <param name="color_path" value="$(find terrain_models)/models/dischma_valley_color.tif"/>
+        <param name="topic_name" value="elevation2"/>
+    </node>
+
+    <group if="$(arg visualization)">
+        <node type="rviz" name="rviz" pkg="rviz" args="-d $(find grid_map_geo)/launch/config.rviz" />
+    </group>
+</launch>

src/grid_map_geo.cpp

@@ -43,20 +43,23 @@
 #include <grid_map_core/iterators/CircleIterator.hpp>
 #include <grid_map_core/iterators/GridMapIterator.hpp>
 
-GridMapGeo::GridMapGeo() {}
+#include <geometry_msgs/TransformStamped.h>
+#include <tf2_ros/static_transform_broadcaster.h>
+
+GridMapGeo::GridMapGeo(const std::string frame_id) { frame_id_ = frame_id; }
 
 GridMapGeo::~GridMapGeo() {}
 
-bool GridMapGeo::Load(const std::string &map_path, bool algin_terrain, const std::string color_map_path) {
-  bool loaded = initializeFromGeotiff(map_path, algin_terrain);
+bool GridMapGeo::Load(const std::string &map_path, const std::string color_map_path) {
+  bool loaded = initializeFromGeotiff(map_path);
   if (!color_map_path.empty()) {  // Load color layer if the color path is nonempty
     bool color_loaded = addColorFromGeotiff(color_map_path);
   }
   if (!loaded) return false;
   return true;
 }
 
-bool GridMapGeo::initializeFromGeotiff(const std::string &path, bool align_terrain) {
+bool GridMapGeo::initializeFromGeotiff(const std::string &path) {
   GDALAllRegister();
   GDALDataset *dataset = (GDALDataset *)GDALOpen(path.c_str(), GA_ReadOnly);
   if (!dataset) {
@@ -80,6 +83,8 @@ bool GridMapGeo::initializeFromGeotiff(const std::string &path, bool align_terra
   const char *pszProjection = dataset->GetProjectionRef();
   std::cout << std::endl << "Wkt ProjectionRef: " << pszProjection << std::endl;
 
+  const OGRSpatialReference *spatial_ref = dataset->GetSpatialRef();
+  std::string name_coordinate = spatial_ref->GetAttrValue("geogcs");
   // Get image metadata
   unsigned width = dataset->GetRasterXSize();
   unsigned height = dataset->GetRasterYSize();
@@ -99,24 +104,9 @@ bool GridMapGeo::initializeFromGeotiff(const std::string &path, bool align_terra
 
   Eigen::Vector2d position{Eigen::Vector2d::Zero()};
 
-  /// TODO: Generalize to set local origin as center of map position
-  // Eigen::Vector3d origin_lv03 =
-  //     transformCoordinates(ESPG::WGS84, std::string(pszProjection), localorigin_wgs84_.position);
-  // localorigin_e_ = origin_lv03(0);
-  // localorigin_n_ = origin_lv03(1);
-  // localorigin_altitude_ = origin_lv03(2);
-  // if (align_terrain) {
-  //   std::cout << "[GridMapGeo] Aligning terrain!" << std::endl;
-  //   double map_position_x = mapcenter_e - localorigin_e_;
-  //   double map_position_y = mapcenter_n - localorigin_n_;
-  //   position = Eigen::Vector2d(map_position_x, map_position_y);
-  // } else {
-  //   std::cout << "[GridMapGeo] Not aligning terrain!" << std::endl;
-  // }
-
   grid_map_.setGeometry(length, resolution, position);
   /// TODO: Use TF for geocoordinates
-  grid_map_.setFrameId("map");
+  grid_map_.setFrameId(frame_id_);
   grid_map_.add("elevation");
   GDALRasterBand *elevationBand = dataset->GetRasterBand(1);
 
@@ -133,17 +123,21 @@ bool GridMapGeo::initializeFromGeotiff(const std::string &path, bool align_terra
     layer_elevation(x, y) = data[gridMapIndex(0) + width * gridMapIndex(1)];
   }
 
-  /// TODO: This is a workaround with the problem of gdal 3 not translating altitude correctly.
-  /// This section just levels the current position to the ground
-  double altitude{0.0};
-  if (grid_map_.isInside(Eigen::Vector2d(0.0, 0.0))) {
-    altitude = grid_map_.atPosition("elevation", Eigen::Vector2d(0.0, 0.0));
-  }
+  static tf2_ros::StaticTransformBroadcaster static_broadcaster;
+  geometry_msgs::TransformStamped static_transformStamped;
+
+  static_transformStamped.header.stamp = ros::Time::now();
+  static_transformStamped.header.frame_id = name_coordinate;
+  static_transformStamped.child_frame_id = frame_id_;
+  static_transformStamped.transform.translation.x = mapcenter_e;
+  static_transformStamped.transform.translation.y = mapcenter_n;
+  static_transformStamped.transform.translation.z = 0.0;
+  static_transformStamped.transform.rotation.x = 0.0;
+  static_transformStamped.transform.rotation.y = 0.0;
+  static_transformStamped.transform.rotation.z = 0.0;
+  static_transformStamped.transform.rotation.w = 1.0;
+  static_broadcaster.sendTransform(static_transformStamped);
 
-  // Eigen::Translation3d meshlab_translation(0.0, 0.0, -altitude);
-  // Eigen::AngleAxisd meshlab_rotation(Eigen::AngleAxisd::Identity());
-  // Eigen::Isometry3d transform = meshlab_translation * meshlab_rotation;  // Apply affine transformation.
-  // grid_map_ = grid_map_.getTransformedMap(transform, "elevation", grid_map_.getFrameId(), true);
   return true;
 }