StreamUtil.h

Go to the documentation of this file.

// StreamUtil.h: useful stream manipulation methods.
//
// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef LOFAR_COMMON_STREAMUTILX_H
#define LOFAR_COMMON_STREAMUTILX_H
 
#include <map>
#include <ostream>
#include <set>
#include <string>
#include <utility>
#include <vector>
 
namespace dp3 {
namespace common {
 
 
class Indent {
 public:
  Indent() { lvl++; }
  ~Indent() { lvl--; }
  static unsigned int level() { return lvl; }
  static const std::string& token() { return tok; }
 
 private:
  static unsigned int lvl;
  static const std::string tok;
};
 
inline std::ostream& indent(std::ostream& os) {
  for (unsigned int i = 0; i < Indent::level(); ++i) {
    os << Indent::token();
  }
  return os;
}
 
template <typename T, typename U>
std::ostream& operator<<(std::ostream& os, const std::pair<T, U>& p);
 
template <typename ITER>
void print(std::ostream& os, ITER begin, ITER end, const char* separator = ",",
           const char* prefix = "[", const char* postfix = "]");
 
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::vector<T>& v);
 
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::set<T>& v);
 
template <typename T, typename U>
std::ostream& operator<<(std::ostream& os, const std::map<T, U>& m);
 
template <typename T, typename U>
inline std::ostream& operator<<(std::ostream& os, const std::pair<T, U>& p) {
  os << '<' << p.first << ',' << p.second << '>';
  return os;
}
 
template <typename ITER>
inline void print(std::ostream& os, ITER begin, ITER end, const char* separator,
                  const char* prefix, const char* postfix) {
  os << prefix;
  if (begin != end) {
    os << *begin;
    ++begin;
  }
  for (; begin != end; ++begin) {
    os << separator << *begin;
  }
  os << postfix;
}
 
template <typename T>
inline std::ostream& operator<<(std::ostream& os, const std::vector<T>& v) {
  print(os, v.begin(), v.end(), ",", "[", "]");
  return os;
}
 
template <typename T>
inline std::ostream& operator<<(std::ostream& os, const std::set<T>& s) {
  print(os, s.begin(), s.end(), ",", "[", "]");
  return os;
}
 
template <typename T, typename U>
inline std::ostream& operator<<(std::ostream& os, const std::map<T, U>& m) {
  print(os, m.begin(), m.end(), ", ", "{", "}");
  return os;
}
 
}  // namespace common
}  // namespace dp3
 
#endif