TauThinningAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration.
*/
 
#ifndef TAUREC_TAUTHINNINGALG_H
#define TAUREC_TAUTHINNINGALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "xAODTau/TauJetContainer.h"
#include "xAODTau/TauTrackContainer.h"
#include "xAODPFlow/PFOContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "CaloEvent/CaloClusterCellLinkContainer.h"
#include "CaloEvent/CaloCellContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODParticleEvent/ParticleContainer.h"
#include "StoreGate/ThinningHandleKey.h"
#include "GaudiKernel/SystemOfUnits.h"

class TauThinningAlg : public AthReentrantAlgorithm
{
  using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
 public:

  virtual StatusCode initialize() override;

  virtual StatusCode execute(const EventContext& ctx) const override;
 
 
 private:
 
  // whether tau vertex correction is applied in tau reconstruction
  Gaudi::Property<bool> m_doVertexCorrection
    { this, "VertexCorrection", true, "Tau vertex correction" };
 
  Gaudi::Property<float> m_minNeutralPFOPt
    {this, "MinNeutralPFOPt", 0.0, "mininum threshold to keep NeutralPFO and related cells"};
 
  // save only tracks passing quality requirements
  Gaudi::Property<bool> m_saveOnlyGoodTracks
    { this, "SaveOnlyGoodTracks", false, "Option to save only tracks passing quality requirements"};        
 
  // the dR between the tau and the clusters
  Gaudi::Property<float> m_DrTauClusters
    { this, "maxDrTauClusters", 0.2, "The distance between the tau the clusters. Beyond this, clusters won't be saved"};
 
  // the dR between the tau and the neutral PFO
  Gaudi::Property<float> m_DrTauNeutrals
    { this, "maxDrTauNeutrals", 9999., "The distance between the tau the neutral PFO. Neutral PFO beyond the settled distance be saved"}; 
 
  // Name of the stream being thinned
  StringProperty m_streamName
    { this, "StreamName", "StreamAOD", "Name of the stream being thinned" };
 
  // Tau container to thin
  SG::ThinningHandleKey<xAOD::TauJetContainer> m_taus
    { this, "Taus", "TauJets", "Tau container to thin" };
 
  // Tau track container to thin
  SG::ThinningHandleKey<xAOD::TauTrackContainer> m_tauTracks
    { this, "TauTracks", "TauTracks", "Tau track container to thin" };
 
  // Tau neutral PFO container to thin
  SG::ThinningHandleKey<xAOD::PFOContainer> m_neutralPFOs
    { this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "Tau neutral PFO container to thin" };
 
  // Tau pi0 cluster container to thin
  SG::ThinningHandleKey<xAOD::CaloClusterContainer> m_pi0clusters
    { this, "TauPi0Clusters", "TauPi0Clusters", "Tau pi0 cluster container to thin" };
 
  // Tau pi0 cluster cell link container to thin
  SG::ThinningHandleKey<CaloClusterCellLinkContainer> m_pi0CellLinks
    { this, "TauPi0CellLinks", "TauPi0Clusters_links", "Tau pi0 cluster cell link container to thin" };
 
  // Tau final pi0 container to thin
  SG::ThinningHandleKey<xAOD::ParticleContainer> m_finalPi0s
    { this, "TauFinalPi0s", "TauFinalPi0s", "Tau final pi0 container to thin" };
 
  // Tau shot PFO container to thin
  SG::ThinningHandleKey<xAOD::PFOContainer> m_shotPFOs
    { this, "TauShotPFOs", "TauShotParticleFlowObjects", "Tau shot PFO container to thin" };
 
  // Tau shot cluster container to thin
  SG::ThinningHandleKey<xAOD::CaloClusterContainer> m_shotclusters
    { this, "TauShotClusters", "TauShotClusters", "Tau shot cluster container to thin" };
 
  // Tau shot cluster cell link container to thin
  SG::ThinningHandleKey<CaloClusterCellLinkContainer> m_shotCellLinks
    { this, "TauShotCellLinks", "TauShotClusters_links", "Tau shot cluster cell link container to thin" };
 
  // Tau hadronic PFO container to thin
  SG::ThinningHandleKey<xAOD::PFOContainer> m_hadronicPFOs
    { this, "TauHadronicPFOs", "TauHadronicParticleFlowObjects", "Tau hadronic PFO container to thin" };
  
  // Tau secondary vertex container to thin
  SG::ThinningHandleKey<xAOD::VertexContainer> m_secondaryVertices
    { this, "TauSecondaryVertices", "TauSecondaryVertices", "Tau secondary vertex container to thin" };
 
  // Cell container to thin
  SG::ThinningHandleKey<CaloCellContainer> m_cells
    { this, "Cells", "AllCalo", "Cell container to thin" };
 
  // Tau cluster cell link container to thin
  SG::ThinningHandleKey<CaloClusterCellLinkContainer> m_tauCellLinks
    { this, "TauCellLinks", "CaloCalTopoClusters_links", "Tau cluster cell link container to thin" };
 
};
 
#endif // TAUREC_TAUTHINNINGALG_H