ColorMapFilter.cpp

Go to the documentation of this file.

/*
 * ColorMapFilter.cpp
 *
 *  Created on: Sep 23, 2017
 *      Author: Peter Fankhauser
 *   Institute: ETH Zurich, ANYbotics
 */

#include <grid_map_filters/ColorMapFilter.hpp>

#include <grid_map_core/grid_map_core.hpp>
#include <pluginlib/class_list_macros.h>

#include <Eigen/Dense>

using namespace filters;

namespace grid_map {

template<typename T>
ColorMapFilter<T>::ColorMapFilter()
    : min_(0.0),
      max_(1.0)
{
}

template<typename T>
ColorMapFilter<T>::~ColorMapFilter()
{
}

template<typename T>
bool ColorMapFilter<T>::configure()
{
  if (!FilterBase<T>::getParam(std::string("input_layer"), inputLayer_)) {
    ROS_ERROR("Color map filter did not find parameter `input_layer`.");
    return false;
  }
  ROS_DEBUG("Color map filter input_layer = %s.", inputLayer_.c_str());

  if (!FilterBase<T>::getParam(std::string("output_layer"), outputLayer_)) {
    ROS_ERROR("Color map filter did not find parameter `output_layer`.");
    return false;
  }
  ROS_DEBUG("Color map filter output_layer = %s.", outputLayer_.c_str());

  if (!FilterBase<T>::getParam(std::string("min/value"), min_)) {
    ROS_ERROR("Color map filter did not find parameter `min/value`.");
    return false;
  }
  if (!FilterBase<T>::getParam(std::string("max/value"), max_)) {
    ROS_ERROR("Color map filter did not find parameter `max/value`.");
    return false;
  }

  std::vector<double> minColor;
  if (!FilterBase<T>::getParam(std::string("min/color"), minColor)) {
    ROS_ERROR("Color map filter did not find parameter `min/color`.");
    return false;
  }
  if (minColor.size() != 3) {
    ROS_ERROR("Color map filter parameter `min/color` needs to be of size 3.");
    return false;
  }
  minColor_ << minColor[0], minColor[1], minColor[2];

  std::vector<double> maxColor;
  if (!FilterBase<T>::getParam(std::string("max/color"), maxColor)) {
    ROS_ERROR("Color map filter did not find parameter `max/color`.");
    return false;
  }
  if (maxColor.size() != 3) {
    ROS_ERROR("Color map filter parameter `max/color` needs to be of size 3.");
    return false;
  }
  maxColor_ << maxColor[0], maxColor[1], maxColor[2];

  return true;
}

template<typename T>
bool ColorMapFilter<T>::update(const T& mapIn, T& mapOut)
{
  mapOut = mapIn;
  auto& input = mapIn[inputLayer_];
  mapOut.add(outputLayer_);
  auto& output = mapOut[outputLayer_];

  const double range = max_ - min_;
  const Eigen::Vector3f colorRange = maxColor_ - minColor_;

  // For each cell in map.
  for (Eigen::Index i = 0; i < output.size(); ++i) {
    if (!std::isfinite(input(i))) continue;
    const double value = std::min<float>(std::max<float>(input(i), min_), max_);
    const double factor = (value - min_) / range;
    const Eigen::Vector3f color = minColor_ + factor * colorRange;
    colorVectorToValue(color, output(i));
  }

  return true;
}

} /* namespace */

PLUGINLIB_EXPORT_CLASS(grid_map::ColorMapFilter<grid_map::GridMap>, filters::FilterBase<grid_map::GridMap>)