SimHitTreeCreator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_ALGS_SIMHITTREECREATOR_H
#define ISF_ALGS_SIMHITTREECREATOR_H 1
// FrameWork includes
#include "StoreGate/ReadHandleKey.h"
#include "GaudiKernel/ITHistSvc.h"
#include "AthenaBaseComps/AthAlgorithm.h"
 
// Hit Collection includes
#include "InDetSimEvent/SiHitCollection.h"
#include "InDetSimEvent/TRTUncompressedHitCollection.h"
#include "MuonSimEvent/MDTSimHitCollection.h"
#include "MuonSimEvent/RPCSimHitCollection.h"
#include "MuonSimEvent/TGCSimHitCollection.h"
#include "MuonSimEvent/CSCSimHitCollection.h"
 
// THistogram service
class ITHistSvc;
// ROOT forward declarations
class TTree;
 
namespace ISF {
 
  class SimHitTreeCreator : virtual public AthAlgorithm {
 
  public:
    SimHitTreeCreator( const std::string& name, ISvcLocator* pSvcLocator );

    virtual ~SimHitTreeCreator();

    virtual StatusCode  initialize() override final;
    virtual StatusCode  execute() override final;
 
  private:

    StatusCode createSimHitsTree();
    StatusCode fillSimHitsTree();
    //** Add information from HepMcParticleLink to TTree - common for all SimHit types */
    void addHepMcParticleLinkInfoToTree(HepMcParticleLink &HMPL);
 
    ServiceHandle<ITHistSvc>                                     m_thistSvc;         
    std::string                                                  m_validationStream; 
    TTree                                                       *m_t_simHits;        
    int                                                  m_pileup;
    int                                                  m_type;
    int                                                  m_id;
    int                                                  m_mother;
    int                                                  m_barcode;
    float                                                m_time;
    float                                                m_drift;
    float                                                m_edeposit;
    float                                                m_momentum;
    float                                                m_theta;
    float                                                m_phi;
    float                                                m_eta;
 
 
    //Inner Detector
    SG::ReadHandleKey<SiHitCollection>                              m_bcmHits;
    SG::ReadHandleKey<SiHitCollection>                              m_blmHits;
    SG::ReadHandleKey<SiHitCollection>                              m_pixHits;
    SG::ReadHandleKey<SiHitCollection>                              m_sctHits;
    SG::ReadHandleKey<TRTUncompressedHitCollection>                 m_trtHits;
    SG::ReadHandleKey<SiHitCollection>                              m_pixPileupHits;
    SG::ReadHandleKey<SiHitCollection>                              m_sctPileupHits;
    SG::ReadHandleKey<TRTUncompressedHitCollection>                 m_trtPileupHits;
 
    //MuonSpectrometer
    SG::ReadHandleKey<MDTSimHitCollection>                          m_mdtHits;
    SG::ReadHandleKey<RPCSimHitCollection>                          m_rpcHits;
    SG::ReadHandleKey<TGCSimHitCollection>                          m_tgcHits;
    SG::ReadHandleKey<CSCSimHitCollection>                          m_cscHits;
  };
}
 
#endif //> !ISF_ALGS_SIMHITTREECREATOR_H