MRRestoringStreamsSink.h

Go to the documentation of this file.

#pragma once
 
#include "MRMeshFwd.h"
#include "MRPch/MRSpdlog.h"
#include <mutex>
#include <streambuf>
#include <string>
 
namespace MR
{
 
 
class LoggingStreambuf : public std::streambuf
{
public:
    MRMESH_API explicit LoggingStreambuf( spdlog::level::level_enum level );
 
protected:
    MRMESH_API std::streamsize xsputn( const char_type* s, std::streamsize count ) override;
    MRMESH_API int_type overflow( int_type ch = traits_type::eof() ) override;
 
private:
    spdlog::level::level_enum level_;
    std::mutex bufMutex_;
    std::string buf_;
};
 
class RestoringStreamsSink : public spdlog::sinks::base_sink<spdlog::details::null_mutex>
{
public:
    MRMESH_API RestoringStreamsSink();
    MRMESH_API ~RestoringStreamsSink();
 
private:
    virtual void sink_it_( const spdlog::details::log_msg& ) override { }
    virtual void flush_() override { }
 
    std::streambuf* coutBuf_;
    std::streambuf* cerrBuf_;
    std::streambuf* clogBuf_;
 
    LoggingStreambuf spdCoutBuf_;
    LoggingStreambuf spdCerrBuf_;
    LoggingStreambuf spdClogBuf_;
};
 
MRMESH_API void redirectSTDStreamsToLogger();
 
 
} // namespace MR