/home/mhx/git/github/libmoost/include/moost/utils/stringify.hpp

Go to the documentation of this file.

/* vim:set ts=3 sw=3 sts=3 et: */
#ifndef MOOST_UTILS_STRINGIFY_HPP__
#define MOOST_UTILS_STRINGIFY_HPP__

#include <map>
#include <set>
#include <vector>
#include <ostream>
#include <string>

#include <boost/lexical_cast.hpp>

#include "foreach.hpp"

// TODO: overload for google:: thingies, std::list

namespace moost { namespace utils {

template <class T>
std::string stringify(const T& val, size_t truncate = 0);

template <class T>
void streamstringify(std::ostream& os, const T& val, size_t = 0);

template <class T1, class T2>
void streamstringify(std::ostream& os, const std::pair<T1, T2>& val, size_t truncate = 0);

template <class T>
void streamstringify(std::ostream& os, const std::set<T>& val, size_t truncate = 0);

template <class T>
void streamstringify(std::ostream& os, const std::vector<T>& val, size_t truncate = 0);

template <class T1, class T2>
void streamstringify(std::ostream& os, const std::map<T1, T2>& val, size_t truncate = 0);

template <class T>
std::string stringify(const T& val, size_t truncate)
{
   std::ostringstream oss;
   streamstringify(oss, val, truncate);
   return oss.str();
}

template <class T>
void streamstringify(std::ostream& os, const T& val, size_t)
{
   os << val;
}

template <class T1, class T2>
void streamstringify(std::ostream& os, const std::pair<T1, T2>& val, size_t truncate)
{
   os << "(";
   streamstringify(os, val.first, truncate);
   os << ", ";
   streamstringify(os, val.second, truncate);
   os << ")";
}

template <class T>
void streamstringify(std::ostream& os, const std::set<T>& val, size_t truncate)
{
   size_t count = 0;
   os << "(";
   foreach (const T& v, val)
   {
      if (count)
         os << ", ";
      streamstringify(os, v, truncate);
      if (++count == truncate && count < val.size())
      {
         os << ", <+" << val.size() - count << ">";
         break;
      }
   }
   os << ")";
}

template <class T>
void streamstringify(std::ostream& os, const std::vector<T>& val, size_t truncate)
{
   size_t count = 0;
   os << "[";
   foreach (const T& v, val)
   {
      if (count)
         os << ", ";
      streamstringify(os, v, truncate);
      if (++count == truncate && count < val.size())
      {
         os << ", <+" << val.size() - count << ">";
         break;
      }
   }
   os << "]";
}

template <class T1, class T2>
void streamstringify(std::ostream& os, const std::map<T1, T2>& val, size_t truncate)
{
   typedef typename std::map<T1, T2>::value_type type_pair;
   size_t count = 0;
   os << "{";
   foreach (const type_pair& p, val)
   {
      if (count)
         os << ", ";
      streamstringify(os, p, truncate);
      if (++count == truncate && count < val.size())
      {
         os << ", <+" << val.size() - count << ">";
         break;
      }
   }
   os << "}";
}

}}

#endif