PFLeptonSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
#ifndef PFLEPTONSELECTOR_H
#define PFLEPTONSELECTOR_H

 
#include "AthContainers/ConstDataVector.h"
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/DataHandle.h"
 
#include "xAODEgamma/ElectronContainer.h"
#include "xAODMuon/MuonContainer.h"
 
class CaloCellContainer;
 
class PFLeptonSelector : public AthReentrantAlgorithm
{
 
public:
  PFLeptonSelector(const std::string& name, ISvcLocator* pSvcLocator);
  ~PFLeptonSelector(){};

  StatusCode initialize();
  StatusCode execute(const EventContext& ctx) const;
  StatusCode finalize();
 
private:
  StatusCode selectElectrons(SG::WriteHandle<ConstDataVector<xAOD::ElectronContainer>>& selectedElectronsWriteHandle,
                             const SG::WriteHandle<ConstDataVector<CaloCellContainer>>& leptonCaloCellsWriteHandle) const;

  void storeElectronCells(const xAOD::Egamma& electron,
                          SG::WriteHandle<ConstDataVector<CaloCellContainer>> leptonCaloCellsWriteHandle) const;

  StatusCode selectMuons(SG::WriteHandle<ConstDataVector<xAOD::MuonContainer>>& selectedMuonsWriteHandle,
                         const SG::WriteHandle<ConstDataVector<CaloCellContainer>>& leptonCaloCellsWriteHandle) const;

  void storeMuonCells(const xAOD::Muon& muon,
                      SG::WriteHandle<ConstDataVector<CaloCellContainer>> leptonCaloCellsWriteHandle) const;

  void storeLeptonCells(const xAOD::CaloCluster& theCluster,
                        SG::WriteHandle<ConstDataVector<CaloCellContainer>> leptonCaloCellsWriteHandle) const;

  StatusCode recordLeptonContainers(
    SG::WriteHandle<ConstDataVector<xAOD::ElectronContainer>>& selectedElectronsWriteHandle,
    SG::WriteHandle<ConstDataVector<xAOD::MuonContainer>>& selectedMuonsWriteHandle,
    SG::WriteHandle<ConstDataVector<CaloCellContainer>> leptonCaloCellsWriteHandle) const;

  SG::ReadHandleKey<xAOD::ElectronContainer> m_electronsReadHandleKey{
    this,
    "inputElectronsName",
    "Electrons",
    "ReadHandle for the ElectronContainer to be used as input"
  };

  SG::ReadHandleKey<xAOD::MuonContainer> m_muonsReadHandleKey{ this,
                                                               "inputMuonsName",
                                                               "Muons",
                                                               "ReadHandle for the MuonContainer to be used as input" };

  SG::WriteHandleKey<ConstDataVector<xAOD::ElectronContainer>> m_selectedElectronsWriteHandleKey{
    this,
    "outputElectronsName",
    "eflowRec_selectedElectrons",
    "WriteHandle for the ElectronContainer, that will be filled with electrons passing the electron ID in "
    "PFLeptonSelector::selectElectrons"
  };

  SG::WriteHandleKey<ConstDataVector<xAOD::MuonContainer>> m_selectedMuonsWriteHandleKey{
    this,
    "outputMuonsName",
    "eflowRec_selectedMuons",
    "WriteHandle for the MuonContainer, that will be filled with muons passing the muon ID in "
    "PFLeptonSelector::selectMuons"
  };

  SG::WriteHandleKey<ConstDataVector<CaloCellContainer>> m_leptonCaloCellsWriteHandleKey{
    this,
    "outputLeptonCellsName",
    "eflowRec_leptonCellContainer",
    "WriteHandle for the CaloCellContainer, that will store calorimeter cells associated to leptons"
  };

  Gaudi::Property<bool> m_storeLeptonCells{ this, "storeLeptonCells", false, "Toggle storage of lepton CaloCells" };
 
  Gaudi::Property<std::string> m_electronID{ this, "electronID", "LHMedium", "Select electron ID" };

  Gaudi::Property<bool> m_selectElectrons{ this, "selectElectrons", true, "Toggle usage of electron ID" };

  Gaudi::Property<bool> m_selectMuons{ this, "selectMuons", true, "Toggle usage of muon ID" };
};
#endif