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