HGTD_RDOAnalysis.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef HGTD_RDO_ANALYSIS_H
#define HGTD_RDO_ANALYSIS_H
 
#include "AthenaBaseComps/AthHistogramAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "GeneratorObjects/McEventCollection.h"
#include "InDetSimData/InDetSimDataCollection.h"
 
#include "HGTD_ReadoutGeometry/HGTD_DetectorManager.h"
#include "HGTD_RawData/HGTD_RDO_Container.h"
 
class HGTD_ID;
 
namespace InDetDD {
  class HGTD_DetectorManager;
}
 
class HGTD_RDOAnalysis : public AthHistogramAlgorithm
{
 
struct SdoInfo {
  float time{0.f};
  int truth = -1; // signal=1, pileup=2, secondary=3
  bool operator<(const SdoInfo& rhs) const { return time < rhs.time; }
};
 
public:
  using AthHistogramAlgorithm::AthHistogramAlgorithm;
 
  virtual StatusCode initialize() override final;
  virtual StatusCode execute() override final;
 
private:
 
  bool isHSGoodParticle(HepMC::ConstGenParticlePtr particlePtr, const HepMC::GenEvent* hardScatterEvent, float min_pt_cut = 1000.);
 
  SG::ReadHandleKey<HGTD_RDO_Container> m_inputKey {this, "CollectionName", "HGTD_RDOs", "Input HGTD RDO collection name"}; 
  SG::ReadHandleKey<InDetSimDataCollection> m_inputTruthKey {this, "SDOCollectionName", "HGTD_SDO_Map", "Input HGTD SDO collection name"};
  SG::ReadHandleKey<McEventCollection> m_inputMcEventCollectionKey {this, "McEventCollectionName", "TruthEvent", "Input McEventCollection name"};
 
  const HGTD_DetectorManager *m_HGTD_Manager {};
  Gaudi::Property<std::string> m_HGTD_Name {this, "DetectorName", "HGTD", "HGTD detector name"};
  const HGTD_ID *m_HGTD_ID{};
  Gaudi::Property<std::string> m_HGTDID_Name {this, "PixelIDName", "HGTD_ID", "HGTD ID name"};
 
  Gaudi::Property<std::string> m_histPath {this, "HistPath", "/RDOAnalysis/HGTD/", ""};
  Gaudi::Property<std::string> m_sharedHistPath {this, "SharedHistPath", "/RDOAnalysis/histos/", ""};
  Gaudi::Property<std::string> m_ntuplePath {this, "NtuplePath", "/RDOAnalysis/ntuples/", ""};
  Gaudi::Property<std::string> m_ntupleName {this, "NtupleName", "HGTD", ""};
  Gaudi::Property<bool> m_doPosition {this, "DoPosition", true, ""};
 
 
  TTree* m_tree{};
  std::vector<int>   m_rdo_module_layer{}; // 0, 1, 2, 3
  std::vector<float> m_rdo_module_x{};     // global position in mm
  std::vector<float> m_rdo_module_y{};     // global position in mm
  std::vector<float> m_rdo_module_z{};     // global position in mm
  std::vector<unsigned long long> m_rdo_module_ID{}; //module ID (get compact)
  std::vector<float> m_rdo_hit_x{};        // global position in mm
  std::vector<float> m_rdo_hit_y{};        // global position in mm
  std::vector<float> m_rdo_hit_z{};        // global position in mm
  std::vector<float> m_rdo_hit_toa{};      // in ns
  std::vector<float> m_rdo_hit_sdo_toa{};  // in ns
  std::vector<int>   m_rdo_hit_sdo_truth_category{}; // signal=1, pileup=2, secondary=3
 
};
 
#endif // HGTD_RDO_ANALYSIS_H