ThinInDetClustersAlg.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef THINNINGUTILS_ThinInDetClustersAlg_H
#define THINNINGUTILS_ThinInDetClustersAlg_H 1
 
// FrameWork includes
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "StoreGate/ThinningHandleKey.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/TrackStateValidationContainer.h"
#include "xAODTracking/TrackMeasurementValidationContainer.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "ExpressionEvaluation/ExpressionParserUser.h"
 
// STL includes
#include <atomic>
#include <string>
#include <vector>
 
class ThinInDetClustersAlg final : public ExpressionParserUser<::AthAlgorithm>
{
 public:
  ThinInDetClustersAlg(const std::string& name, ISvcLocator* pSvcLocator);
  
  virtual ~ThinInDetClustersAlg() = default;
 
  virtual StatusCode initialize() override;
 
  virtual StatusCode execute() override; 
 
  virtual StatusCode finalize() override;
 
 private:
  unsigned int m_ntot = 0;
  unsigned int m_npass = 0;
  BooleanProperty m_thinPixelHitsOnTrack
  { this, "ThinPixelHitsOnTrack", false, ""};
  BooleanProperty m_thinSCTHitsOnTrack
  { this, "ThinSCTHitsOnTrack", false, ""};
  BooleanProperty m_thinTRTHitsOnTrack
  { this, "ThinTRTHitsOnTrack", false, ""};
  
  StringProperty m_streamName
  { this, "StreamName", "", "Name of the stream being thinned" };
 
  
  StringProperty m_selectionString
  { this, "SelectionString", "InDetTrackParticles.pt>10*GeV", "Selection string for each TrackParticle container" };
 
  SG::ThinningHandleKey<xAOD::TrackParticleContainer> m_inDetSGKey
  { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" };
 
  unsigned int m_ntot_pix_states = 0;
  unsigned int m_npass_pix_states = 0;
  SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesPixSGKey
  { this, "InDetTrackStatesPixKey", "PixelMSOSs", "" }; // original
  
  unsigned int m_ntot_pix_measurements = 0;
  unsigned int m_npass_pix_measurements = 0;
  SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsPixSGKey
  { this, "InDetTrackMeasurementsPixKey", "PixelClusters", "" };
  
  unsigned int m_ntot_sct_states = 0;
  unsigned int m_npass_sct_states = 0;
  SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesSctSGKey
  { this, "InDetTrackStatesSctKey", "SCT_MSOSs", "" };
  
  unsigned int m_ntot_sct_measurements = 0;
  unsigned int m_npass_sct_measurements = 0;
  SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsSctSGKey
  { this, "InDetTrackMeasurementsSctKey", "SCT_Clusters", "" };
  
  unsigned int m_ntot_trt_states = 0;
  unsigned int m_npass_trt_states = 0;
  SG::ThinningHandleKey<xAOD::TrackStateValidationContainer> m_statesTrtSGKey
  { this, "InDetTrackStatesTrtKey", "TRT_MSOSs", "" };
  
  unsigned int m_ntot_trt_measurements = 0;
  unsigned int m_npass_trt_measurements = 0;
  SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer> m_measurementsTrtSGKey
  { this, "InDetTrackMeasurementsTrtKey", "TRT_DriftCircles", "" };
  
  SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
 
  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;
  
  StatusCode filterTrackHits
    (MeasurementType detTypeToSelect,
     const xAOD::TrackParticleContainer& inputTrackParticles,
     const std::vector<bool>& inputMask,
     const SG::ThinningHandleKey<xAOD::TrackStateValidationContainer>& statesKey,
     const SG::ThinningHandleKey<xAOD::TrackMeasurementValidationContainer>& measurementsKey,
     unsigned int& ntot_states,
     unsigned int& npass_states,
     unsigned int& npass_measurements) const;
}; 
 
#endif //> !THINNINGUTILS_ThinInDetClustersAlg_H