OStreamMessageSink.hh

 
#ifndef GRAMODS_CORE_OSTREAMMESSAGESINK
#define GRAMODS_CORE_OSTREAMMESSAGESINK
 
#include <gmCore/MessageSink.hh>
 
#include <mutex>
 
BEGIN_NAMESPACE_GMCORE;
 
class OStreamMessageSink
  : public MessageSink {
 
public:
 
  void setStream(std::ostream *s) {
    std::lock_guard<std::mutex> guard(lock);
    raw_out = s;
    shared_out = nullptr;
  }
 
  void setStream(std::shared_ptr<std::ostream> s) {
    std::lock_guard<std::mutex> guard(lock);
    raw_out = nullptr;
    shared_out = s;
  }
 
  void setStream(std::string name);
 
  void setUseAnsiColor(bool on);
 
  void output(Message msg);
 
  void setLevel(int l) { level = l; }
 
  GM_OFI_DECLARE;
 
private:
 
  std::ostream *raw_out = nullptr;
  std::shared_ptr<std::ostream> shared_out;
  std::mutex lock;
 
  bool use_ansi_color = false;
  int level = 2;
};
 
END_NAMESPACE_GMCORE;
 
#endif