MissingMassProb.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
// Class handling the probability calculation of the MissingMassCalculator
// author Michael Huebner <michael.huebner@no.spam.cern.ch>
 
#ifndef DITAUMASSTOOLS_MISSINGMASSPROB_H
#define DITAUMASSTOOLS_MISSINGMASSPROB_H
 
// Local include(s):
#include "DiTauMassTools/HelperFunctions.h"
 
// ROOT include(s):
#include <TF1.h>
#include <Math/Vector4D.h> //for PtEtaPhiMVector
 
#include <string>
#include <list>
 
class TFile;
 
namespace DiTauMassTools{
  using ROOT::Math::PtEtaPhiMVector;
 
class MissingMassCalculator;
class MissingMassInput;
 
class MissingMassProb {
 
  public:
    MissingMassProb(MMCCalibrationSet::e aset, const std::string& paramFilePath);
    ~MissingMassProb();
 
    double apply(MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2, bool constant=false, bool oneTau=false, bool twoTau=false);
 
    void setParamAngle(const PtEtaPhiMVector & tauvec, int tau, int tautype);
    void setParamRatio(int tau, int tautype);
    void setParamNuMass();
    TF1* GetFormulaAngle1() {return m_formulaAngle1;}
    TF1* GetFormulaAngle2() {return m_formulaAngle2;}
    TF1* GetFormulaRatio1() {return m_formulaRatio1;}
    TF1* GetFormulaRatio2() {return m_formulaRatio2;}
    TF1* GetFormulaNuMass() {return m_formulaNuMass;}
 
    void SetAllowUseHT(bool allowUseHT) { m_allowUseHT=allowUseHT;}
    bool GetAllowUseHT() { return m_allowUseHT;}
 
    void SetUseHT(bool val) { m_UseHT=val; }
    bool GetUseHT() { return m_UseHT; } // if use HT
 
    void SetUseTauProbability(bool val) { m_fUseTauProbability=val; }
    bool GetUseTauProbability() { return m_fUseTauProbability; }
 
    void SetUseMnuProbability(bool val) { m_fUseMnuProbability=val; }
    bool GetUseMnuProbability() { return m_fUseMnuProbability; }
 
    void SetUseDphiLL(bool val) {m_fUseDphiLL = val;}
    bool GetUseDphiLL() { return m_fUseDphiLL; }
 
    double MetProbability(MissingMassInput& preparedInput, const double & met1,const  double & met2,const  double & MetSigma1, const double & MetSigma2);
    double dTheta3Dparam(const int & parInd, const int & tau_type, const double & P_tau,const double *par); 
 
    double dTheta3d_probabilityFast(MissingMassInput& preparedInput, const int & tau_type,const double & dTheta3d,const double & P_tau);
    double myDelThetaHadFunc(double *x, double *par);
    double myDelThetaLepFunc(double *x, double *par);
 
    double MHtProbability(MissingMassInput& preparedInput, const double & d_mhtX, const double & d_mhtY, const double & mht, 
                   const double & trueMetGuess, const double & mht_offset);
    double MHtProbabilityHH(MissingMassInput& preparedInput, const double & d_mhtX, const double & d_mhtY, const double & mht, 
                 const double & trueMetGuess, const double & mht_offset);
 
    void MET(MissingMassInput& preparedInput);
    double mEtAndTauProbability(MissingMassInput& preparedInput);
    double MnuProbability(MissingMassInput& preparedInput, double mnu, double binsize);
    double MnuProbability(MissingMassInput& preparedInput, double mnu);
    double TauProbability(MissingMassInput& preparedInput, const int & type1, const PtEtaPhiMVector & vis1, const PtEtaPhiMVector & nu1, 
                   const int & type2, const PtEtaPhiMVector & vis2, const PtEtaPhiMVector & nu2);
    double TauProbability(MissingMassInput& preparedInput, const int & type1, const PtEtaPhiMVector & vis1, const PtEtaPhiMVector & nu1, 
                   const int & type2, const PtEtaPhiMVector & vis2, const PtEtaPhiMVector & nu2, const double & detmet);
    double TauProbabilityLFV(MissingMassInput& preparedInput, const int & type1, const PtEtaPhiMVector & vis1, const PtEtaPhiMVector & nu1);
 
    static double MetProbabilityWrapper(MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2);
    static double mEtAndTauProbabilityWrapper(MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2);
    static double dTheta3d_probabilityFastWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
    static double TauProbabilityWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
    static double MnuProbabilityWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
    static double MnuProbabilityNewWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
 
     static double dTheta3d_probabilityNewWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
     static double TauProbabilityNewWrapper( MissingMassProb* prob, MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2 );
 
    std::list<std::function<double(MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2)>> m_probListConstant;
    std::list<std::function<double(MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2)>> m_probListOneTau;
    std::list<std::function<double(MissingMassInput& preparedInput, const int & tau_type1, const int & tau_type2, const PtEtaPhiMVector & tauvec1, const PtEtaPhiMVector & tauvec2, const PtEtaPhiMVector nuvec1, const PtEtaPhiMVector & nuvec2)>> m_probListTwoTau;
  private:
    //SpeedUp static array for efficient access
    thread_local static double s_fit_param[2][3][6][5];
    //cache quantities for efficient NuPSolution calculation
 
    TF1 *m_formulaAngle1 = new TF1("formulaAngle1", "[0]*exp(-[2]*(log((x+[3])/[1]))**2)");
    TF1 *m_formulaAngle2 = new TF1("formulaAngle2", "[0]*exp(-[2]*(log((x+[3])/[1]))**2)");
    TF1 *m_formulaRatio1{};
    TF1 *m_formulaRatio2{};
    TF1 *m_formulaRatioLep1 = new TF1("formulaRatio1", "gaus(0)+expo(3)");
    TF1 *m_formulaRatioLep2 = new TF1("formulaRatio2", "gaus(0)+expo(3)");
    TF1 *m_formulaRatioHad1 = new TF1("formulaRatio1", "gaus(0)");
    TF1 *m_formulaRatioHad2 = new TF1("formulaRatio2", "gaus(0)");
    TF1 *m_formulaNuMass = new TF1("formulaNuMass", "pol6");
    std::vector<TF1*> m_paramVectorAngle;
    std::vector<TF1*> m_paramVectorAngleLep;
    std::vector<TF1*> m_paramVectorRatio;
    std::vector<TF1*> m_paramVectorRatioLep;
    std::vector<TF1*> m_paramVectorNuMass;
 
    std::string m_paramFilePath;
    TFile *m_fParams{};
    MMCCalibrationSet::e m_mmcCalibrationSet{};
 
    bool m_allowUseHT{};
    bool m_UseHT{};
    bool m_fUseTauProbability{}; // switch to apply TauProbability
    bool m_fUseMnuProbability{}; // switch to apply MnuProbability
    bool m_fUseDphiLL{}; //for leplep
 
  // comment by Sasha to keep this for the future
  //inline double AngularProbability(PtEtaPhiMVector nu_vec, PtEtaPhiMVector vis_vec, int decayType);
 
  // to be tested if helpful
  //inline double ResonanceProbability(PtEtaPhiMVector vec1, PtEtaPhiMVector vec2);
};
} // namespace DiTauMassTools
 
#endif