KinkTrkZeeTagTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// KinkTrkZeeTagTool.h, (c) ATLAS Detector software
 
#ifndef DERIVATIONFRAMEWORK_KINKTRKZEETAGTOOL_H
#define DERIVATIONFRAMEWORK_KINKTRKZEETAGTOOL_H 1
 
// Gaudi & Athena basics
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrigObjectMatching/TrigMatchTool.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
 
// DerivationFramework includes
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODMissingET/MissingETContainer.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include<string>
#include<vector>
 
namespace DerivationFramework {
 
  class KinkTrkZeeTagTool : public AthAlgTool, public IAugmentationTool {
   
  public: 
    KinkTrkZeeTagTool( const std::string& t, const std::string& n, const IInterface* p );
   
    ~KinkTrkZeeTagTool();
   
    // Athena algtool's Hooks
    StatusCode  initialize();
    StatusCode  finalize();
 
    virtual StatusCode addBranches() const;
 
  private:
    bool passTrigger(const std::vector<std::string>& triggers) const;
    bool checkTagElectron(const xAOD::Electron *ele) const;
    bool checkCluster(const xAOD::CaloCluster *clu) const;
    bool checkEleClusPair(const xAOD::Electron *ele, const xAOD::CaloCluster *clu) const;
    bool passElectronQuality(const xAOD::Electron *ele) const;
    bool passClusterQuality(const xAOD::CaloCluster *clu) const;
    bool passElectronTrigMatch(const xAOD::Electron *ele, const std::vector<std::string>& triggers) const;
 
    ToolHandle<Trig::TrigDecisionTool> m_trigDecisionTool;
    ToolHandle<TrigMatchTool> m_trigMatchTool;
    std::vector<std::string> m_trigNames;
    float m_trigMatchDeltaR;
    bool m_doTrigMatch;
 
    SG::ReadHandleKey<xAOD::ElectronContainer> m_electronSGKey { this, "ElectronContainerKey", "ElectronCollection", ""};
    std::vector<std::string> m_electronIDKeys;
    float m_electronPtCut;
    float m_electronEtaMax;
 
    SG::ReadHandleKey<xAOD::CaloClusterContainer> m_clusterSGKey { this, "ClusterContainerKey", "egammaClusters", ""};
    float m_clusterEtCut;
    float m_clusterEtaMax;
 
    float m_diEleMassLow;
    float m_diEleMassHigh;
    float m_dPhiMax;
 
    SG::WriteHandleKey<std::vector<float>> m_KinkTrkDiEleMassKey { this, "KinkTrkDiEleMassKey", "KinkTrkDiEleMass", ""};
    SG::WriteHandleKey<std::vector<float>> m_KinkTrkProbeEleEtKey { this, "KinkTrkProbeEleEtKey", "KinkTrkProbeEleEt", ""};
 
    SG::WriteHandleKey<std::vector<float> > m_diEleMassKey;
    SG::WriteHandleKey<std::vector<float> > m_probeEleEtKey;
 
  }; 
 
}
#endif