UncertaintyEnum.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "JetUncertainties/UncertaintyEnum.h"
 
#include <math.h>
 
namespace jet
{
 
//                                              //
//  Component category enumeration methods      //
//                                              //
 
namespace CompCategory
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case Statistical:   return "Statistical";
            case Detector:      return "Detector";
            case Modelling:     return "Modelling";
            case Mixed:         return "Mixed";
            case Effective:     return "Effective";
            case Other:         return "Other";
            default:            return "UNKNOWN";
        }
    }
    
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("Statistical"))
            return Statistical;
        if (!type.CompareTo("Detector"))
            return Detector;
        if (!type.CompareTo("Modelling"))
            return Modelling;
        if (!type.CompareTo("Mixed"))
            return Mixed;
        if (!type.CompareTo("Effective"))
            return Effective;
        if (!type.CompareTo("Other"))
            return Other;
        return UNKNOWN;
    }
} // end CompCategory namespace
 
 
//                                              //
//  Component correlation enumeration methods   //
//                                              //
 
namespace CompCorrelation
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case Uncorrelated:  return "Uncorrelated";
            case Correlated:    return "Correlated";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("Uncorrelated"))
            return Uncorrelated;
        if (!type.CompareTo("Correlated"))
            return Correlated;
        return UNKNOWN;
    }
} // end CompCorrelation namespace
 
 
//                                              //
//  Component parametrization type enum methods //
//                                              //
 
namespace CompParametrization
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case Pt:            return "Pt";
            case PtEta:         return "PtEta";
            case PtAbsEta:      return "PtAbsEta";
            case PtAbsMass:     return "PtAbsMass";
            case PtMass:        return "PtMass";
            case PtMassEta:     return "PtMassEta";
            case PtMassAbsEta:  return "PtMassAbsEta";
            case PtAbsMassEta:  return "PtAbsMassEta";
            case PtAbsMassAbsEta: return "PtAbsMassAbsEta";
            case eLOGmOe:       return "eLOGmOe";
            case eLOGmOeEta:    return "eLOGmOeEta";
            case eLOGmOeAbsEta: return "eLOGmOeAbsEta";
        case PtLOGPtMassForTagSF: return "PtLOGPtMassForTagSF";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("Pt",TString::kIgnoreCase))
            return Pt;
        if (!type.CompareTo("PtEta",TString::kIgnoreCase))
            return PtEta;
        if (!type.CompareTo("PtAbsEta",TString::kIgnoreCase))
            return PtAbsEta;
        if (!type.CompareTo("PtAbsMass",TString::kIgnoreCase))
            return PtAbsMass;
        if (!type.CompareTo("PtMass",TString::kIgnoreCase))
            return PtMass;
        if (!type.CompareTo("PtMassEta",TString::kIgnoreCase))
            return PtMassEta;
        if (!type.CompareTo("PtMassAbsEta",TString::kIgnoreCase))
            return PtMassAbsEta;
        if (!type.CompareTo("PtAbsMassEta",TString::kIgnoreCase))
            return PtAbsMassEta;
        if (!type.CompareTo("PtAbsMassAbsEta",TString::kIgnoreCase))
            return PtAbsMassAbsEta;
        if (!type.CompareTo("eLOGmOe",TString::kIgnoreCase))
            return eLOGmOe;
        if (!type.CompareTo("eLOGmOeEta",TString::kIgnoreCase))
            return eLOGmOeEta;
        if (!type.CompareTo("eLOGmOeAbsEta",TString::kIgnoreCase))
            return eLOGmOeAbsEta;
        if (!type.CompareTo("PtLOGPtMassForTagSF",TString::kIgnoreCase))
            return PtLOGPtMassForTagSF;
        return UNKNOWN;
    }
 
    bool isAbsEta(const TypeEnum type)
    {
        switch(type)
        {
            case PtAbsEta:
            case PtMassAbsEta:  
            case PtAbsMassAbsEta:
            case eLOGmOeAbsEta: return true;
            
            default:            return false;
        }
    }
 
    bool includesMass(const TypeEnum type)
    {
        switch(type)
        {
            case PtAbsMass:
            case PtMass:
        case PtLOGPtMassForTagSF:
            case PtMassEta:
            case PtMassAbsEta:
            case PtAbsMassEta:
            case PtAbsMassAbsEta:
            case eLOGmOe:
            case eLOGmOeEta:
            case eLOGmOeAbsEta:
                return true;
            
            default:
                return false;
        }
    }
 
} // end CompParametrization namespace
 
 
//                                              //
//  Component parametrization type enum methods //
//                                              //
 
namespace CompMassDef
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case FourVecMass:   return "FourVec";
            case CaloMass:      return "Calo";
            case TAMass:        return "TA";
            case CombMassQCD:   return "CombQCD";
            case CombMassWZ:    return "CombWZ";
            case CombMassHbb:   return "CombHbb";
            case CombMassTop:   return "CombTop";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("FourVec",TString::kIgnoreCase) || !type.CompareTo("FourVector",TString::kIgnoreCase))
            return FourVecMass;
        if (!type.CompareTo("Calo",TString::kIgnoreCase) || !type.CompareTo("Calorimeter",TString::kIgnoreCase))
            return CaloMass;
        if (!type.CompareTo("TA",TString::kIgnoreCase) || !type.CompareTo("TrackAssisted",TString::kIgnoreCase))
            return TAMass;
        if (type.BeginsWith("Comb",TString::kIgnoreCase) || type.BeginsWith("Combined",TString::kIgnoreCase))
        {
            if (type.EndsWith("QCD",TString::kIgnoreCase))
                return CombMassQCD;
            if (type.EndsWith("WZ",TString::kIgnoreCase))
                return CombMassWZ;
            if (type.EndsWith("Hbb",TString::kIgnoreCase))
                return CombMassHbb;
            if (type.EndsWith("Top",TString::kIgnoreCase))
                return CombMassTop;
        }
        return UNKNOWN;
    }
 
    TString getJetScaleString(const TypeEnum type)
    {
        switch(type)
        {
            case CaloMass:      return "JetJMSScaleMomentumCalo";
            case TAMass:        return "JetJMSScaleMomentumTA";
            case CombMassQCD:   return "JetJMSScaleMomentumCombQCD";
            case CombMassWZ:    return "JetJMSScaleMomentumCombWZ";
            case CombMassHbb:   return "JetJMSScaleMomentumCombHbb";
            case CombMassTop:   return "JetJMSScaleMomentumCombTop";
            default:            return "UNKNOWN_JET_MASS_SCALE";
 
        }
    }
 
} // end CompMassDef namespace
 
//                                              //
//  Component parametrization type enum methods //
//                                              //
 
namespace CompScaleVar
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case FourVec:       return "FourVec";
            case Pt:            return "Pt";
            case Mass:          return "Mass";
            case D12:           return "Split12";
            case D23:           return "Split23";
            case Tau21:         return "Tau21";
            case Tau32:         return "Tau32";
            case Tau21WTA:      return "Tau21WTA";
            case Tau32WTA:      return "Tau32WTA";
            case D2Beta1:       return "D2Beta1";
            case C2Beta1:       return "C2Beta1";
            case Qw:            return "Qw";
            case MassRes:       return "MassRes";
            case MassResAbs:    return "MassResAbs";
            case PtRes:         return "PtRes";
            case PtResAbs:      return "PtResAbs";
            case FourVecRes:    return "FourVecRes";
            case FourVecResAbs: return "FourVecResAbs";
            case TagScaleFactor:return "TagScaleFactor";
        case TagEfficiency: return "TagEfficiency";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("FourVec",TString::kIgnoreCase) || !type.CompareTo("4Vec",TString::kIgnoreCase) || !type.CompareTo("FourVector",TString::kIgnoreCase))
            return FourVec;
        if (!type.CompareTo("Pt",TString::kIgnoreCase))
            return Pt;
        if (!type.CompareTo("Mass",TString::kIgnoreCase))
            return Mass;
        if (!type.CompareTo("D12",TString::kIgnoreCase) || !type.CompareTo("Split12",TString::kIgnoreCase))
            return D12;
        if (!type.CompareTo("D23",TString::kIgnoreCase) || !type.CompareTo("Split23",TString::kIgnoreCase))
            return D23;
        if (!type.CompareTo("Tau21",TString::kIgnoreCase))
            return Tau21;
        if (!type.CompareTo("Tau32",TString::kIgnoreCase))
            return Tau32;
        if (!type.CompareTo("Tau21WTA",TString::kIgnoreCase))
            return Tau21WTA;
        if (!type.CompareTo("Tau32WTA",TString::kIgnoreCase))
            return Tau32WTA;
        if (!type.CompareTo("D2Beta1",TString::kIgnoreCase))
            return D2Beta1;
        if (!type.CompareTo("C2Beta1",TString::kIgnoreCase))
            return C2Beta1;
        if (!type.CompareTo("Qw",TString::kIgnoreCase))
            return Qw;
        if (!type.CompareTo("MassRes",TString::kIgnoreCase))
            return MassRes;
        if (!type.CompareTo("MassResAbs",TString::kIgnoreCase))
            return MassResAbs;
        if (!type.CompareTo("PtRes",TString::kIgnoreCase))
            return PtRes;
        if (!type.CompareTo("PtResAbs",TString::kIgnoreCase))
            return PtResAbs;
        if (!type.CompareTo("FourVecRes",TString::kIgnoreCase))
            return FourVecRes;
        if (!type.CompareTo("FourVecResAbs",TString::kIgnoreCase))
            return FourVecResAbs;
        if (!type.CompareTo("TagScaleFactor",TString::kIgnoreCase))
            return TagScaleFactor;
        if (!type.CompareTo("TagEfficiency",TString::kIgnoreCase))
        return TagEfficiency;
        return UNKNOWN;
    }
 
    bool isScaleType(const TypeEnum type)
    {
        switch (type)
        {
            case FourVec:
            case Pt:
            case Mass:
            case D12:
            case D23:
            case Tau21:
            case Tau32:
            case Tau21WTA:
            case Tau32WTA:
            case D2Beta1:
            case C2Beta1:
            case Qw:
            case TagScaleFactor:
        case TagEfficiency:
                return true;
 
            default:
                return false;
        }
    }
 
    bool isResolutionType(const TypeEnum type)
    {
        switch (type)
        {
            case MassRes:
            case MassResAbs:
            case PtRes:
            case PtResAbs:
            case FourVecRes:
            case FourVecResAbs:
                return true;
 
            default:
                return false;
        }
    }
 
    bool isAbsResolutionType(const TypeEnum type)
    {
        switch (type)
        {
            case MassResAbs:
            case PtResAbs:
            case FourVecResAbs:
                return true;
 
            default:
                return false;
        }
    }
 
    bool isRelResolutionType(const TypeEnum type)
    {
        switch (type)
        {
            case MassRes:
            case PtRes:
            case FourVecRes:
                return true;
 
            default:
                return false;
        }
    }
}
 
 
//                                              //
//  Component parametrization type enum methods //
//                                              //
 
namespace CompFlavorLabelVar
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case t_qqb:         return "t_qqb";
            case t:             return "t";
        case V_qq:          return "V_qq";
            case W_qq:          return "W_qq";
            case Z_qq:          return "Z_qq";
        case W_qq_From_t:   return "W_qq_From_t";
        case t_other:       return "t_other";
        case q:             return "q";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (type=="t_qqb")
      return t_qqb;
    if (type=="t")
      return t;
    if (type=="V_qq")
      return V_qq;
    if (type=="W_qq")
      return W_qq;
    if (type=="Z_qq")
      return Z_qq;
    if (type=="W_qq_From_t")
      return W_qq_From_t;
    if (type=="t_other")
      return t_other;
    if (type=="q")
      return q;
        return UNKNOWN;
    }
}
 
 
//                                              //
//  Component parametrization type enum methods //
//                                              //
 
namespace CompTaggerRegionVar
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case passMpassD2_2Var:  return "2Var_passMpassD2";
        case passMfailD2_2Var:  return "2Var_passMfailD2";
            case failMpassD2_2Var:  return "2Var_failMpassD2";
            case failMfailD2_2Var:  return "2Var_failMfailD2";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
      if (type.Contains("2Var",TString::kIgnoreCase))
      {
    if (type.Contains("passMpassD2",TString::kIgnoreCase))
      return passMpassD2_2Var;
        if (type.Contains("passMfailD2",TString::kIgnoreCase))
      return passMfailD2_2Var;
        if (type.Contains("failMpassD2",TString::kIgnoreCase))
      return failMpassD2_2Var;
        if (type.Contains("failMfailD2",TString::kIgnoreCase))
      return failMfailD2_2Var;
      }
      return UNKNOWN;
    }
}
 
 
//                                              //
//  Pileup component enumeration methods        //
//                                              //
 
namespace PileupComp
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case OffsetNPV:     return "Pileup_OffsetNPV";
            case OffsetMu:      return "Pileup_OffsetMu";
            case PtTerm:        return "Pileup_PtTerm";
            case RhoTopology:   return "Pileup_RhoTopology";
            default:            return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (type.Contains("Pileup",TString::kIgnoreCase))
        {
            if (type.Contains("OffsetNPV",TString::kIgnoreCase))
                return OffsetNPV;
            if (type.Contains("OffsetMu",TString::kIgnoreCase))
                return OffsetMu;
            if (type.Contains("PtTerm",TString::kIgnoreCase))
                return PtTerm;
            if (type.Contains("RhoTopology",TString::kIgnoreCase))
                return RhoTopology;
        }
        return UNKNOWN;
    }
}
 
 
//                                              //
//  Flavour component enumeration methods       //
//                                              //
 
namespace FlavourComp
{
    TString enumToString(const TypeEnum type)
    {
        switch (type)
        {
            case Response:          return "FlavourResponse";
            case Composition:       return "FlavourComposition";
            case bJES:              return "bJES";
            case PerJetResponse:    return "PerJetFlavourResponse";
            case PerJetResponse_Gluon: return "PerJetFlavourResponse_Gluon";
            case PerJetResponse_LQ: return "PerJetFlavourResponse_LQ";
            case PerJetResponse_B: return "PerJetFlavourResponse_B";
            case PerJetResponse_C: return "PerJetFlavourResponse_C";
            default:                return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (type.Contains("Flavour",TString::kIgnoreCase) || type.Contains("Flavor",TString::kIgnoreCase))
        {
            if (type.Contains("Response",TString::kIgnoreCase))
            {
                if (type.Contains("PerJet",TString::kIgnoreCase)){
                    if(type.Contains("Response_Gluon",TString::kIgnoreCase)) return PerJetResponse_Gluon;
                    else if(type.Contains("Response_LQ",TString::kIgnoreCase)) return PerJetResponse_LQ;
                    else if(type.Contains("Response_B",TString::kIgnoreCase)) return PerJetResponse_B;
                    else if(type.Contains("Response_C",TString::kIgnoreCase)) return PerJetResponse_C;
                    else return PerJetResponse;
                }
                else{
                        return Response;
                    }
            }
            if (type.Contains("Composition",TString::kIgnoreCase))
                return Composition;
        }
        if (type.Contains("bJES",TString::kIgnoreCase))
            return bJES;
        return UNKNOWN;
    }
}
 
 
//                                              //
//  Combined mass component enumeration methods //
//                                              //
 
namespace CombMassComp
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case Calo:      return "Calo";
            case TA:        return "TA";
            case Both:      return "Both";
            default:        return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (type.Contains("Calo",TString::kIgnoreCase))
            return Calo;
        if (type.Contains("TA",TString::kIgnoreCase))
            return TA;
        if (type.Contains("Both",TString::kIgnoreCase))
            return Both;
        return UNKNOWN;
    }
}
 
 
//                                              //
//  Jet topology enumeration methods            //
//                                              //
 
namespace JetTopology
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case QCD:   return "QCD";
            case WZ:    return "WZ";
            case Hbb:   return "Hbb";
            case Top:   return "Top";
            case MIXED: return "MIXED";
            default:    return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("QCD",TString::kIgnoreCase))
            return QCD;
        if (!type.CompareTo("WZ",TString::kIgnoreCase))
            return WZ;
        if (!type.CompareTo("Hbb",TString::kIgnoreCase))
            return Hbb;
        if (!type.CompareTo("Top",TString::kIgnoreCase))
            return Top;
        return UNKNOWN;
    }
}
 
 
//                                              //
//  Histogram interpolation enumeration methods //
//                                              //
 
namespace Interpolate
{
    TString enumToString(const TypeEnum type)
    {
        switch(type)
        {
            case None:  return "None";
            case Full:  return "Full";
            case OnlyX: return "X";
            case OnlyY: return "Y";
            default:    return "UNKNOWN";
        }
    }
 
    TypeEnum stringToEnum(const TString& type)
    {
        if (!type.CompareTo("None",TString::kIgnoreCase) || !type.CompareTo("false",TString::kIgnoreCase))
            return None;
        if (!type.CompareTo("Full",TString::kIgnoreCase) || !type.CompareTo("true",TString::kIgnoreCase))
            return Full;
        if (!type.CompareTo("OnlyX",TString::kIgnoreCase))
            return OnlyX;
        if (!type.CompareTo("OnlyY",TString::kIgnoreCase))
            return OnlyY;
        return UNKNOWN;
    }
}
 
} // end jet namespace