NewLine.h

Go to the documentation of this file.

#ifndef LARDATA_RECOBASE_DUMPERS_NEWLINE_H
#define LARDATA_RECOBASE_DUMPERS_NEWLINE_H 1

// C/C++ standard libraries
#include <string>
#include <utility> // std::move()

namespace recob {
  namespace dumper {

    struct IndentOptions_t {
      std::string indent;       
      bool appendFirst = false; 

      IndentOptions_t(std::string ind = "", bool followLine = false)
        : indent(ind), appendFirst(followLine)
      {}

      IndentOptions_t& appendIndentation(std::string more)
      {
        indent += more;
        appendFirst = false;
        return *this;
      }
      IndentOptions_t& removeIndentation(std::string less)
      {
        indent.erase(std::max(indent.length() - less.length(), size_t(0)));
        return *this;
      }

    }; // IndentOptions_t

    template <typename Stream>
    class NewLine {
    public:
      NewLine(Stream& stream, IndentOptions_t indentOptions)
        : out(stream), options(std::move(indentOptions)), nLines(0)
      {}

      NewLine(Stream& stream, std::string indent = "", bool followLine = false)
        : NewLine(stream, IndentOptions_t{indent, followLine})
      {}


      unsigned int lines() const { return nLines; }

      std::string indent() const { return options.indent; }


      Stream& newLine()
      {
        if (!append()) forceNewLine();
        ++nLines;
        return out;
      }

      Stream& operator()() { return newLine(); }

      void forceNewLine() { out << "\n" << options.indent; }

      bool append() const { return (lines() == 0) && options.appendFirst; }

      void setIndent(std::string newIndent) { options.indent = newIndent; }

      void addIndent(std::string moreIndent) { options.indent += moreIndent; }

    protected:
      Stream& out;             
      IndentOptions_t options; 
      unsigned int nLines;     

    }; // class NewLine

    template <typename Stream>
    inline NewLine<Stream> makeNewLine(Stream& stream, std::string indent, bool followLine = false)
    {
      return NewLine<Stream>(stream, indent, followLine);
    }

    template <typename Stream>
    inline NewLine<Stream> makeNewLine(Stream& stream, IndentOptions_t const& options)
    {
      return NewLine<Stream>(stream, options);
    }

  } // namespace dumper
} // namespace recob

#endif // LARDATA_RECOBASE_DUMPERS_NEWLINE_H