SimG4SaveTrajectory.h

Go to the documentation of this file.

#ifndef SIMG4COMPONENTS_G4SAVETRAJECTORY
#define SIMG4COMPONENTS_G4SAVETRAJECTORY
 
// Gaudi
#include "GaudiKernel/AlgTool.h"
 
// FCCSW
#include "k4FWCore/DataHandle.h"
#include "SimG4Interface/ISimG4SaveOutputTool.h"
class IGeoSvc;
 
#if __has_include("edm4hep/TrackerHit3DCollection.h")
#include "edm4hep/TrackerHit3DCollection.h"
#else
#include "edm4hep/TrackerHitCollection.h"
namespace edm4hep {
  using TrackerHit3DCollection = edm4hep::TrackerHitCollection;
}
#endif
 
class SimG4SaveTrajectory : public AlgTool, virtual public ISimG4SaveOutputTool {
public:
  explicit SimG4SaveTrajectory(const std::string& aType, const std::string& aName, const IInterface* aParent);
  virtual ~SimG4SaveTrajectory();
  virtual StatusCode initialize();
  virtual StatusCode finalize();
  virtual StatusCode saveOutput(const G4Event& aEvent) final;
 
private:
  ServiceHandle<IGeoSvc> m_geoSvc;
  mutable DataHandle<edm4hep::TrackerHit3DCollection> m_trackHits{"Hits/Trajectory",
                                                                      Gaudi::DataHandle::Writer, this};
};
 
#endif /* SIMG4COMPONENTS_G4SAVETRAJECTORY */