DiTauSelectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAUANALYSISTOOLS_DITAUSELECTIONTOOL_H
#define TAUANALYSISTOOLS_DITAUSELECTIONTOOL_H
 
/*
  author: Dirk Duschinger
  mail: dirk.duschinger@cern.ch
  contact email: antonio.de.maria@cern.ch
*/
 
// Framework include(s):
#include "AsgTools/AsgMetadataTool.h"
#include "AsgTools/AnaToolHandle.h"
#include "PATCore/IAsgSelectionTool.h"
#include "AsgDataHandles/ReadHandleKey.h"
 
// Local include(s):
#include "TauAnalysisTools/IDiTauSelectionTool.h"
#include "TauAnalysisTools/Enums.h"
#include "TauAnalysisTools/HelperFunctions.h"
 
// ROOT include(s):
#include "TH1F.h"
#include "TFile.h"
 
namespace TauAnalysisTools
{
 
class DiTauSelectionCut;
class DiTauSelectionCutPt;
class DiTauSelectionCutAbsEta;
class DiTauSelectionCutNSubjets;
 
class DiTauSelectionTool : public virtual IAsgSelectionTool,
  public virtual IDiTauSelectionTool,
  public asg::AsgMetadataTool
{
  friend class DiTauSelectionCut;
  friend class DiTauSelectionCutPt;
  friend class DiTauSelectionCutAbsEta;
  friend class DiTauSelectionCutNSubjets;
 
  ASG_TOOL_CLASS2( DiTauSelectionTool,
                   IAsgSelectionTool,
                   TauAnalysisTools::IDiTauSelectionTool )
 
public:
  DiTauSelectionTool( const std::string& name );
 
  virtual ~DiTauSelectionTool();
 
  virtual StatusCode initialize() override;
 
  virtual const asg::AcceptInfo& getAcceptInfo() const override;
 
  virtual asg::AcceptData accept( const xAOD::IParticle* p ) const override;
 
  virtual asg::AcceptData accept( const xAOD::DiTauJet& tau ) const override;
 
  virtual void setOutFile( TFile* fOutFile ) override;
 
  virtual void writeControlHistograms() override;
 
private:
 
  // Execute at each event
  virtual StatusCode beginEvent() override;
 
  template<typename T, typename U>
  void FillRegionVector(std::vector<T>& vRegion, U tMin, U tMax) const;
  template<typename T, typename U>
  void FillValueVector(std::vector<T>& vRegion, U tVal) const;
  template<typename T>
  void PrintConfigRegion(const std::string& sCutName, std::vector<T>& vRegion) const;
  template<typename T>
  void PrintConfigValue(const std::string& sCutName, std::vector<T>& vRegion) const;
  template<typename T>
  void PrintConfigValue(const std::string& sCutName, T& sVal) const;
 
  // bitmask of tau selection cuts
  int m_iSelectionCuts;
  // vector of transverse momentum cut regions
  std::vector<float> m_vPtRegion;
  // vector of absolute eta cut regions
  std::vector<float> m_vAbsEtaRegion;
  // vector of number of subjets cut regions
  std::vector<float> m_vNSubjetsRegion;
 
  float m_dPtMin;
  float m_dPtMax;
  float m_dAbsEtaMin;
  float m_dAbsEtaMax;
  float m_dNSubjetsMax;
  float m_dNSubjetsMin;
 
protected:
  TFile* m_fOutFile;
  std::shared_ptr<TH1F> m_hCutFlow;
 
private:
  std::string m_sConfigPath;
 
  std::map<DiTauSelectionCuts, std::unique_ptr<TauAnalysisTools::DiTauSelectionCut>> m_cMap;
 
  void setupCutFlowHistogram();
 
protected:
  bool m_bCreateControlPlots;
 
  asg::AcceptInfo m_aAccept;
 
 
}; // class DiTauSelectionTool
 
} // namespace TauAnalysisTools
 
#endif // TAUANALYSISTOOLS_DITAUSELECTIONTOOL_H