/*************************************************************************************

    Grid physics library, www.github.com/paboyle/Grid 

    Source file: ./lib/Log.h

    Copyright (C) 2015

Author: Antonin Portelli <antonin.portelli@me.com>
Author: Azusa Yamaguchi <ayamaguc@staffmail.ed.ac.uk>
Author: Peter Boyle <paboyle@ph.ed.ac.uk>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License along
    with this program; if not, write to the Free Software Foundation, Inc.,
    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

    See the full license in the file "LICENSE" in the top level distribution directory
    *************************************************************************************/
    /*  END LEGAL */

#include <map>

#ifndef GRID_LOG_H
#define GRID_LOG_H
namespace Grid {
  
  // Dress the output; use std::chrono for time stamping via the StopWatch class


  class Colours{
  protected:
    bool is_active;
  public:
    std::map<std::string, std::string> colour;
    
   
    Colours(bool activate=false){
      Active(activate);
    };

    
    void Active(bool activate){
      is_active=activate;
      
      if (is_active){
	colour["BLACK"]  ="\033[30m";
	colour["RED"]    ="\033[31m";
	colour["GREEN"]  ="\033[32m";
	colour["YELLOW"] ="\033[33m";
	colour["BLUE"]   ="\033[34m";
	colour["PURPLE"] ="\033[35m";
	colour["CYAN"]   ="\033[36m";
	colour["WHITE"]  ="\033[37m";
	colour["NORMAL"] ="\033[0;39m";
      } else {
      colour["BLACK"] ="";
      colour["RED"]   ="";
      colour["GREEN"] ="";
      colour["YELLOW"]="";
      colour["BLUE"]  ="";
      colour["PURPLE"]="";
      colour["CYAN"]  ="";
      colour["WHITE"] ="";
      colour["NORMAL"]="";
      }

      
    };
    
  };
    
  
  class Logger {
  protected:
    Colours &Painter;
    int active;
    std::string name, topName;
    std::string COLOUR;

  public:
    static GridStopWatch StopWatch;
    static std::ostream devnull;
    
    std::string background() {return Painter.colour["NORMAL"];}
    std::string evidence() {return Painter.colour["YELLOW"];}
    std::string colour() {return Painter.colour[COLOUR];}
    
    Logger(std::string topNm, int on, std::string nm, Colours& col_class, std::string col)
      : active(on),
	name(nm),
	topName(topNm),
	Painter(col_class),
	COLOUR(col){} ;
  
  void Active(int on) {active = on;};
  int  isActive(void) {return active;};
  
  friend std::ostream& operator<< (std::ostream& stream, Logger& log){
      
    if ( log.active ) {
            StopWatch.Stop();
            GridTime now = StopWatch.Elapsed();
            StopWatch.Start();
            stream << log.background()<< log.topName << log.background()<< " : ";
            stream << log.colour() <<std::setw(10) << std::left << log.name << log.background() << " : ";
            stream << log.evidence()<< now << log.background() << " : " << log.colour();
            return stream;
        } else { 
            return devnull;
        }
    }
    
};
    
class GridLogger: public Logger {
public:
  GridLogger(int on, std::string nm, Colours&col_class, std::string col_key = "NORMAL"):
    Logger("Grid", on, nm, col_class, col_key){};
};

void GridLogConfigure(std::vector<std::string> &logstreams);

  extern GridLogger GridLogError;
  extern GridLogger GridLogWarning;
  extern GridLogger GridLogMessage;
  extern GridLogger GridLogDebug  ;
  extern GridLogger GridLogPerformance;
  extern GridLogger GridLogIterative  ;
  extern GridLogger GridLogIntegrator  ;
  extern Colours    GridLogColours;

  
}
#endif