SystTool.cxx

Go to the documentation of this file.

#include "NAGASH/SystTool.h"
 
using namespace std;
using namespace NAGASH;
 
const TString &SystTool::CurrentVariation()
{
    if (!isSetVar)
    {
        MSGUser()->StartTitle("SystTool::CurrentVariation");
        MSGUser()->MSG(MSGLevel::DEBUG, "No variation is set, will set to nominal");
        MSGUser()->EndTitle();
 
        SetToVariation("Nominal");
        currentvar = "Nominal";
    }
    return currentvar;
}
 
void SystTool::SetVariationCorrelation(const TString &name, double correlation)
{
    auto findresult = CorrMap.find(name);
    if (findresult != CorrMap.end())
    {
        findresult->second = correlation;
    }
    else
    {
        MSGUser()->StartTitle("SystTool::SetVariationCorrelation");
        MSGUser()->MSG(MSGLevel::WARNING, "no variation named ", name, ", this call will be ignored");
        MSGUser()->EndTitle();
    }
}
 
double SystTool::GetVariationCorrelation(const TString &name)
{
    auto findresult = CorrMap.find(name);
    if (findresult != CorrMap.end())
    {
        return findresult->second;
    }
    else
    {
        MSGUser()->StartTitle("SystTool::GetVariationCorrelation");
        MSGUser()->MSG(MSGLevel::WARNING, "no variation named ", name, ", returning zero");
        MSGUser()->EndTitle();
        return 0;
    }
}
 
const std::vector<TString> &SystTool::GetListOfVariations()
{
    if (isSystListSorted)
    {
        return SystList;
    }
    else
    {
        SystList.clear();
        for (auto &t : CorrMap)
        {
            SystList.emplace_back(t.first);
            // move Nominal to the front
            if (t.first == "Nominal")
                std::swap(SystList[0], SystList.back());
        }
 
        isSystListSorted = true;
        return SystList;
    }
}
 
std::vector<TString> SystTool::GetListOfVariations(const TString &name)
{
    auto findresult = SystListMap.find(name);
    if (findresult != SystListMap.end())
    {
        return SystListMap.find(name)->second;
    }
    else
    {
        MSGUser()->StartTitle("SystTool::GetListOfVariations");
        MSGUser()->MSG(MSGLevel::WARNING, "Variable ", name, " does not exist, return NULL vector");
        MSGUser()->EndTitle();
        return std::vector<TString>();
    }
}
 
void SystTool::SetToVariation(const TString &name)
{
    if (isSetVar)
    {
        // if SystMapFull has been made
        auto findresult = SystMapFull.find(name);
        if (findresult != SystMapFull.end())
        {
            for (auto &var : findresult->second)
            {
                *(var.Current) = *(var.Variation);
            }
            currentvar = name;
        }
        else
        {
            MSGUser()->StartTitle("SystTool::SetVariation");
            MSGUser()->MSG(MSGLevel::WARNING, "Variation ", name, " does not exist, no variation will be set");
            MSGUser()->EndTitle();
            return;
        }
    }
    else
    {
        // make SystMapFull
        SystMapFull.clear();
        for (auto &systname : SystList)
        {
            std::vector<AuxVar> tempvec;
            for (auto &t : VarMapFull)
            {
                AuxVar tempvar;
                tempvar.Current = &(t.second.CurrentVar);
                auto findresult = t.second.VarMap.find(systname);
                if (findresult != t.second.VarMap.end())
                {
                    tempvar.Variation = &(findresult->second);
                }
                else if (t.second.NominalVar.has_value())
                {
                    tempvar.Variation = &(t.second.NominalVar);
                }
                else
                {
                    MSGUser()->StartTitle("SystTool::SetVariation");
                    MSGUser()->MSG(MSGLevel::WARNING, "Variable ", t.first, " has not Nominal variation, will set to the first variation ", t.second.VarMap.begin()->first);
                    MSGUser()->EndTitle();
                    tempvar.Variation = &(t.second.VarMap.begin()->second);
                }
                tempvec.push_back(tempvar);
            }
            SystMapFull.insert(pair<TString, std::vector<AuxVar>>(systname, tempvec));
        }
        isSetVar = true;
        SetToVariation(name);
    }
}
 
int SystTool::GetNVariations()
{
    return SystList.size();
}
 
int SystTool::GetNVariations(const TString &name)
{
    auto findresult = VarMapFull.find(name);
    if (findresult != VarMapFull.end())
    {
        return findresult->second.VarMap.size();
    }
    else
    {
        MSGUser()->StartTitle("SystTool::SetVariation");
        MSGUser()->MSG(MSGLevel::WARNING, "Variable ", name, " does not exist, returnint zero");
        MSGUser()->EndTitle();
        return 0;
    }
}