CaloMuonTag.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CALOTRKMUIDTOOLS_CALOMUONTAG_H
#define CALOTRKMUIDTOOLS_CALOMUONTAG_H
 
#include <TFile.h>
#include <TH2.h>
 
#include <atomic>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ToolHandle.h"
#include "ICaloTrkMuIdTools/ICaloMuonTag.h"
 
class CaloMuonTag : public AthAlgTool, virtual public ICaloMuonTag {
public:
    CaloMuonTag(const std::string& type, const std::string& name, const IInterface* pInterface);
    virtual ~CaloMuonTag();
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
    int caloMuonTag(const std::vector<DepositInCalo>& deposits, double eta, double pt) const override;  
 
private:
    double interpolate(const std::string& cutName, double pt) const;
    int getPtBinLow(double pt) const;
    int getCutBin(const std::string& cutName) const;
 
    std::unique_ptr<TH2> m_hist{nullptr};
    std::string m_tagMode;
    StatusCode getHist(TFile* file, const std::string& histName, std::unique_ptr<TH2>& hist);
 
    std::vector<std::string> m_vecCutNames;
    std::vector<int> m_vecPtBins;
    int m_numCuts{0};
    int m_numPtBins{0};
 
    mutable std::atomic_int m_numTagged{0};
    mutable std::atomic_int m_numRejected{0};
};
 
#endif