printer.hpp

Go to the documentation of this file.

//====================================================
//     Preprocessor settings
//====================================================
#ifndef PRINTER_HPP
#define PRINTER_HPP
 
//====================================================
//     Headers
//====================================================
 
//My headers
#include "common.hpp"
#include "colsty.hpp"
 
//Extra headers
#include <arsenalgear/stream.hpp>
#include <arsenalgear/utils.hpp>
 
//STD headers
#include <iostream>
#include <vector>
#include <string>
#include <unordered_map>
 
namespace osm
 {
  //====================================================
  //     print (first overload)
  //====================================================
  template <typename... Args>
  inline void print( std::ostream& os = agr::null_stream, const Args&... args )
   {
    if( &os == &agr::null_stream ) std::cout << "\n";
    else if( &os == &std::cerr ) os << feat( col, "red" ) << feat( sty, "bold" );
    else if( &os == &std::clog ) os << feat( col, "lt blue" ) << feat( sty, "bold" );
 
    ( os << ... << args ) << "\n";
 
    if( &os == &std::cerr || &os == &std::clog || &os == &std::cout ) os << feat( rst, "all" );
   }
  
  //====================================================
  //     print (second overload)
  //====================================================
  template <typename... Args>
  inline void print( Args&... args )
   {
    ( std::cout << ... << args ) << "\n" << feat( rst, "all" );
   }
 
  //====================================================
  //     OS_Decorator class
  //====================================================
  class OS_Decorator
   {
    public:
 
     //====================================================
     //     Constructors and destructors
     //====================================================
     OS_Decorator();
     ~OS_Decorator();
 
     //====================================================
     //     Setters
     //====================================================
     void setColor( const std::string& color, std::ostream& os = std::cout );
     void setStyle( const std::string& style, std::ostream& os = std::cout );
 
     //====================================================
     //     Resetters
     //====================================================
     void resetColor( std::ostream& os = std::cout );
     void resetStyle( std::ostream& os = std::cout );
     void removeStyle( const std::string& style, std::ostream& os = std::cout );
     void resetFeatures( std::ostream& os = std::cout );
 
     //====================================================
     //     Getters
     //====================================================
     std::string getColor( std::ostream& os = std::cout );
     std::string getStyle( std::ostream& os = std::cout );
     std::unordered_map <std::ostream*, std::string> getColorList();
     std::unordered_map <std::ostream*, std::string> getStyleList();
     std::ostream& getCurrentStream();
 
     //====================================================
     //     Operators
     //====================================================
     const OS_Decorator& operator () ( std::ostream& os = std::cout );
 
    private:
 
     //====================================================
     //     Attributes
     //====================================================
     std::unordered_map <std::ostream*, std::string> colors, styles;
     std::ostream* current_stream;
   };
 
  //====================================================
  //     Operator << 
  //====================================================
  template <typename T>
  std::ostream& operator << ( OS_Decorator my_shell, const T& elem )
   {
    if ( my_shell.getColor( my_shell.getCurrentStream() ) != "" )
     {
      my_shell.getCurrentStream() << feat( col, my_shell.getColor( my_shell.getCurrentStream() ) );
     }
    if ( my_shell.getStyle( my_shell.getCurrentStream() ) != "" )
     {
      std::vector<std::string> list_of_styles = agr::split_string( my_shell.getStyle( my_shell.getCurrentStream() ), " " );
      for ( auto elem: list_of_styles )
       {
        my_shell.getCurrentStream() << feat( sty, elem );
       }
     }
    my_shell.getCurrentStream() << elem << feat( rst, "all" );
    
    return my_shell.getCurrentStream();
   }
 }
 
#endif