LikelihoodMultiDTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETTAGTOOLS_LIKELIHOODMULTIDTOOL_H
#define JETTAGTOOLS_LIKELIHOODMULTIDTOOL_H
/******************************************************
    @class LikelihoodMultiDTool
    DESCRIPTION:
    Helper class for likelihood calculations
********************************************************/
 
#include "AthenaBaseComps/AthAlgTool.h"
#include <string>
#include <vector>
#include <map>
 
#include "JetTagTools/HistoLimits.h"
 
#include "JetTagTools/LikelihoodComponents.h"
 
class ITHistSvc;
class TH1;
 
namespace Analysis
{
 
static const InterfaceID IID_LikelihoodMultiDTool("Analysis::LikelihoodMultiDTool", 1, 0);
  
class LikelihoodMultiDTool : public AthAlgTool
{
 
 public:
 
  LikelihoodMultiDTool(const std::string&,const std::string&,const IInterface*);
  virtual ~LikelihoodMultiDTool();
  
  StatusCode initialize();
  StatusCode finalize();
  
  static const InterfaceID& interfaceID() { return IID_LikelihoodMultiDTool; };
  
  void prepareHistosFromFile(const std::string& refFileName);
  TH1* retrieveHistoFromFile(const std::string& refFileName, 
    const std::string& histoName, const std::string& jetPrefix);
  void setLhVariableValue(std::vector<Slice>& value);
  void setLhVariableValue(Slice& value);
  std::vector<double> calculateLikelihood();
  const std::vector<double>& tagLikelihood();
  void printStatus();
  double getEff(const std::string& , const std::string& , const std::string&);
  void addLhVariableToUse(const std::string&);
  inline void resetLhVariableToUse() { m_useTheseLhVariables.clear(); }
  void setInterpolFlag(bool f) {m_DoInterpol = f;}
  void defineNbHypotheses(int n) { m_nbHypotheses = n; }
 
 private:
 
  ITHistSvc* m_histoSvc;
  
  // Filled once at the beginning of an run
  //std::vector<std::string> m_allLhVariables; //!< names of all likelihood variables in the histogram file
  std::vector<std::string> m_useTheseLhVariables; 
  
  std::vector< std::map<std::string, TH1*>* > m_mapOfAllHistos; 
    
  std::map<std::string, HistoLimits> m_histoLimitsMap1D;
  std::map<std::string, HistoLimits> m_histoLimitsMap2D;
  std::map<std::string, HistoLimits> m_histoLimitsMap3D;
 
  // Filled once per event
  std::vector<Slice> m_lhVariableValues;
  
  std::vector<double> m_likelihoodVector;
 
  int m_nhis1D;
  int m_nhis2D;
  int m_nhis3D;
 
  //
  bool m_DoInterpol;
  bool m_WriteSmoothedHistos;
 
  unsigned int m_nbHypotheses; // nb of hypotheses (default 2: b | u)
 
  // for soft-mu:
  int  m_nSmooth1D;
  bool m_normProb;
 
};
 
}
#endif