TrackParticleThinningBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_TRACKPARTICLETHINNINGBASE_H
#define DERIVATIONFRAMEWORK_TRACKPARTICLETHINNINGBASE_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IThinningTool.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/ThinningHandleKey.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/TrackStateValidationContainer.h"
#include "xAODTracking/TrackMeasurementValidationContainer.h"
 
#include "GaudiKernel/ToolHandle.h"
 
#include <string>
#include <atomic>
#include <vector>
 
namespace DerivationFramework {
 
  class TrackParticleThinningBase : public extends<AthAlgTool, IThinningTool> {
  public:
    using base_class::base_class;
 
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override final;
    virtual StatusCode doThinning(const EventContext& ctx) const override final;
 
  protected:
    virtual std::vector<int> updateMask(const EventContext& ctx, const xAOD::TrackParticleContainer*) const;
    SG::ThinningHandleKey<xAOD::TrackParticleContainer> m_inDetSGKey
      { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" };
 
  private:
    //Counters and keys for xAOD::TrackParticle container
    mutable std::atomic<unsigned int> m_ntot{}, m_npass{};
    StringProperty m_streamName
      { this, "StreamName", "", "Name of the stream being thinned" };
 
    //Counters and keys for xAOD::TrackStateValidation and xAOD::TrackMeasurementValidation containers
    mutable std::atomic<unsigned int> m_ntot_pix_states{}, m_npass_pix_states{};
    SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesPixSGKey
      { this, "InDetTrackStatesPixKey", "PixelMSOSs", "" };
    mutable std::atomic<unsigned int> m_ntot_pix_measurements, m_npass_pix_measurements;
    SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsPixSGKey
      { this, "InDetTrackMeasurementsPixKey", "PixelClusters", "" };
    mutable std::atomic<unsigned int> m_ntot_sct_states{}, m_npass_sct_states{};
    SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesSctSGKey
      { this, "InDetTrackStatesSctKey", "SCT_MSOSs", "" };
    mutable std::atomic<unsigned int> m_ntot_sct_measurements{}, m_npass_sct_measurements{};
    SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsSctSGKey
      { this, "InDetTrackMeasurementsSctKey", "SCT_Clusters", "" };
    mutable std::atomic<unsigned int> m_ntot_trt_states{}, m_npass_trt_states{};
    SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesTrtSGKey
      { this, "InDetTrackStatesTrtKey", "TRT_MSOSs", "" };
    mutable std::atomic<unsigned int> m_ntot_trt_measurements{}, m_npass_trt_measurements{};
    SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsTrtSGKey
      { this, "InDetTrackMeasurementsTrtKey", "TRT_DriftCircles", "" };
 
    // For P->T converter of SCT_Clusters
    SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
 
    //logic
    Gaudi::Property<bool> m_thinHitsOnTrack{this, "ThinHitsOnTrack", false};
 
    //Track state types
    // Subset of enum MeasurementType from Athena: Tracking/TrkEvent/TrkEventPrimitives/TrkEventPrimitives/TrackStateDefs.h
    enum MeasurementType {
      TrkState_unidentified = 0,
      TrkState_Pixel      = 1,
      TrkState_SCT        = 2,
      TrkState_TRT        = 3,
      TrkState_Pseudo     = 8,
      TrkState_Vertex     = 9,
      TrkState_SpacePoint = 11,
      TrkState_NumberOfMeasurementTypes=16
    };

    void selectTrackHits(const xAOD::TrackParticleContainer& inputTrackParticles,
                         const std::vector<bool>& inputMask,
                         MeasurementType detTypeToSelect,
                         std::vector<bool>& outputStatesMask, std::vector<bool>& outputMeasurementsMask) const;
 
    void filterTrackHits
    (const EventContext& ctx,
     MeasurementType detTypeToSelect,
     const xAOD::TrackParticleContainer& inputTrackParticles,
     const std::vector<bool>& inputMask,
     const SG::ThinningHandleKey<xAOD::TrackStateValidationContainer>& statesKey,
     const SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer>& measurementsKey,
     std::atomic<unsigned int>& ntot_states,
     std::atomic<unsigned int>& ntot_measurements,
     std::atomic<unsigned int>& npass_states,
     std::atomic<unsigned int>& npass_measurements) const;
  };
}
 
#endif // DERIVATIONFRAMEWORK_TRACKPARTICLETHINNINGBASE_H