MSGTool.cxx

Go to the documentation of this file.

//***************************************************************************************
//***************************************************************************************
 
#include "NAGASH/MSGTool.h"
 
using namespace NAGASH;
using namespace std;
 
MSGTool::MSGTool(MSGLevel level, const TString &name)
{
    DefinedLevel = level;
    FocusName = name;
 
    if (DefinedLevel == MSGLevel::FOCUS)
        OutputLevel = MSGLevel::INFO;
    else
        OutputLevel = level;
}
 
MSGTool::MSGTool(const MSGTool &msg, int ThreadNumber)
{
    TString num;
    num.Form("_JobNum%d", ThreadNumber);
    OutputLevel = msg.OutputLevel;
    DefinedLevel = msg.DefinedLevel;
    FocusName = msg.FocusName;
    isCreateLog = msg.isCreateLog;
    isOpenLog = msg.isOpenLog;
    LogFileName = msg.LogFileName + num;
    FuncTitle = msg.FuncTitle;
 
    if (isOpenLog)
        OpenLog(LogFileName);
    if (isCreateLog)
        CreateLog(LogFileName);
}
 
MSGTool::~MSGTool()
{
    CloseLog();
}
 
void MSGTool::CreateLog(const TString &name)
{
    if (tolog == true)
        CloseLog();
 
    outfile = ofstream(name.TString::Data(), ios::out);
    LogFileName = name;
    tolog = true;
    isCreateLog = true;
    isOpenLog = false;
}
 
void MSGTool::OpenLog(const TString &name)
{
    if (tolog == true)
        CloseLog();
 
    outfile = ofstream(name.TString::Data(), ios::app);
    LogFileName = name;
    tolog = true;
    isOpenLog = true;
    isCreateLog = false;
}
 
void MSGTool::CloseLog()
{
    if (tolog == false)
        return;
    else
    {
        tolog = false;
        isCreateLog = false;
        isOpenLog = false;
        outfile.close();
    }
}
 
void MSGTool::StartFocusRegion(const TString &name)
{
    if ((DefinedLevel == MSGLevel::FOCUS) && (FocusName.TString::EqualTo(name) == true))
        OutputLevel = MSGLevel::DEBUG;
}
 
void MSGTool::EndFocusRegion(const TString &name)
{
    if ((DefinedLevel == MSGLevel::FOCUS) && (FocusName.TString::EqualTo(name) == true))
        OutputLevel = MSGLevel::INFO;
}
 
void MSGTool::StartTitle(const TString &name)
{
    std::unique_lock msglock(MSGMutex);
    FuncTitle.push_back(name);
}
 
std::unique_ptr<MSGTool::MSGTitleGuard> MSGTool::StartTitleWithGuard(const TString &name)
{
    std::unique_lock msglock(MSGMutex);
    FuncTitle.push_back(name);
    return std::unique_ptr<MSGTitleGuard>(new MSGTitleGuard(shared_from_this()));
}
 
void MSGTool::EndTitle()
{
    std::unique_lock msglock(MSGMutex);
    if (FuncTitle.size() > 0)
        FuncTitle.pop_back();
}
 
void MSGTool::PrintTitle()
{
    std::shared_lock msglock(MSGMutex);
    if (FuncTitle.size() == 0)
        Print("Title Track Lost");
    else
    {
        for (size_t i = 0; i < FuncTitle.size(); i++)
        {
            if (i == FuncTitle.size() - 1)
                Print(FuncTitle[i]);
            else
                Print(FuncTitle[i], "->");
        }
    }
}
 
void MSGTool::PrintLevel(MSGLevel level)
{
    if (level == MSGLevel::FATAL)
        Print("FATAL");
    if (level == MSGLevel::ERROR)
        Print("ERROR");
    if (level == MSGLevel::WARNING)
        Print("WARNING");
    if (level == MSGLevel::INFO)
        Print("INFO");
    if (level == MSGLevel::DEBUG)
        Print("DEBUG");
}
 
MSGLevel MSGTool::SetOutputLevel(MSGLevel level)
{
    auto prev = OutputLevel;
    OutputLevel = level;
    return prev;
}
 
MSGLevel MSGTool::Level() const
{
    return OutputLevel;
}