point_cloud_flags.cc

#include "drake/perception/point_cloud_flags.h"

#include <sstream>
#include <vector>

namespace drake {
namespace perception {

namespace {

// Utility for `ToString`.
// TODO(eric.cousineau): Move to `drake/common`, consider using `infix_iterator`
// per Soonho's suggestion: https://codereview.stackexchange.com/a/13209
std::ostream& join(std::ostream& os,
                   const std::vector<std::string>& elements,
                   const std::string& delim) {
  for (size_t i = 0; i < elements.size(); ++i) {
    os << elements[i];
    if (i + 1 < elements.size())
      os << delim;
  }
  return os;
}

}  // namespace

namespace pc_flags {

std::ostream& operator<<(std::ostream& os, const Fields& fields) {
  DRAKE_DEMAND(internal::kMaxBitInUse == kRGBs);
  std::vector<std::string> values;
  if (fields.contains(kXYZs))
    values.push_back("kXYZs");
  if (fields.contains(kNormals))
    values.push_back("kNormals");
  if (fields.contains(kRGBs))
    values.push_back("kRGBs");
  if (fields.has_descriptor()) {
    values.push_back(fields.descriptor_type().name());
  }
  os << "(";
  join(os, values, " | ");
  return os << ")";
}

}  // namespace pc_flags

}  // namespace perception
}  // namespace drake