logging.hpp

Go to the documentation of this file.

//   GAMBIT: Global and Modular BSM Inference Tool
//   *********************************************

#ifndef __logging_hpp__
#define __logging_hpp__

// Standard libraries
#include <string>
#include <set>
#include <map>
#include <vector>
#include <deque>
#include <fstream>
#include <chrono> 
#include <omp.h>

// Gambit
#include "gambit/Logs/logger.hpp" // Minimal declarations needed to use logger -- most code should only need to include this.
#include "gambit/Logs/log_tags.hpp"
#include "gambit/Utils/file_lock.hpp"
//#include "gambit/Utils/util_functions.hpp"


namespace Gambit
{
  // Forward declarations from Utils. Saves including the entire header for a trivial thing
  namespace Utils
  {
    typedef std::chrono::time_point<std::chrono::system_clock> time_point;
    time_point get_clock_now();
  }

  namespace Logging
  {

    // Global reference start time. Can only be used in this compile unit.
    static const Utils::time_point start_time(Utils::get_clock_now());

    // Function to retrieve the 'msgtypes' set
    const std::set<LogTag>& msgtypes();

    // Function to retrieve the 'flags' set
    const std::set<LogTag>& flags();
 
    // Function to retrieve the 'echoes' set
    const std::set<LogTag>& echoes();

    // Function to return the next unused tag index
    // (needed by module and backend macros so they can determine what tag they are allowed to use)
    int getfreetag();

    void checktags();

    struct Message
    {
        std::string message;
        std::set<int> tags;
        Utils::time_point received_at;
        Message(const std::string& msgIN, 
                const std::set<int>& tagsIN)
          : message(msgIN), 
            tags(tagsIN), 
            received_at(Utils::get_clock_now())
        {}
    };

    struct SortedMessage
    {
        const std::string& message; //lives on in original Message object, so can be a reference safely I think
        const Utils::time_point& received_at; //ditto
        // couldn't figure out how to nicely initialise these in the constructor list, so not const
        // now happens in body of constructor
        std::set<LogTag> type_tags;      //message types
        std::set<int> component_tags;    //gambit components, modules, and backends
        std::set<LogTag> flag_tags;      //extra message flags      
        std::set<LogTag> echo_tags;      //message echo flags      
        // Constructor (does the sorting of the tags)
        SortedMessage(const Message& mail);
    };


    //%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    //% Logger class declarations                           %
    //%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
  
    class BaseLogger
    {
      public:
        virtual ~BaseLogger();

        virtual void write(const SortedMessage&) = 0;

        virtual void flush() = 0;
    };

    class StdLogger : public BaseLogger
    {
      public:
        // Attach logger object to an existing stream
        StdLogger(std::ostream&);
    
        // Opens a file and takes care of the stream itself
        StdLogger(const std::string&);
  
        virtual ~StdLogger();

        virtual void write(const SortedMessage&);
        
        virtual void flush();
 
        void writetags(const std::set<LogTag>&);   
        void writetags(const std::set<int>&);   
 
      private:
        std::ofstream my_own_fstream; //Don't use this except in constructor
        std::ostream& my_stream;
 
        int MPIrank;
        int MPIsize;
    };

  } //end namespace Logging

} // end namespace Gambit

#endif