TrackFastSimSD.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRACKWRITEFASTSIM_TRACKFASTSIMSD_H
#define TRACKWRITEFASTSIM_TRACKFASTSIMSD_H
 
// Base class header
#include "G4VSensitiveDetector.hh"
 
// Athena headers
#include "TrackRecord/TrackRecordCollection.h"
 
// STL headers
#include <string>
#include <utility>
 
// G4 needed classes
class G4Step;
class G4TouchableHistory;
class G4Track;
 
class TrackFastSimSD : public G4VSensitiveDetector
{
public:
  TrackFastSimSD(const std::string& name, const std::string& outputCollectionName, const int SD_type=0);
  ~TrackFastSimSD() {}
 
  // Initialize from G4
  void Initialize(G4HCofThisEvent *) override final;
 
  G4bool ProcessHits(G4Step*, G4TouchableHistory*) override final;
  void   WriteTrack(const G4Track*, const bool, const bool);

  template <class... Args> bool AddHit(Args&&... args)
  {
    if (m_trackRecordCollection) {
      m_trackRecordCollection->Emplace(std::forward<Args>(args)...);
      return true;
    }
    return false;
  }
 
private:
  TrackRecordCollection* getTrackRecordCollection() const;
 
  std::string m_outputCollectionName;
  // Non-owning cache set by Initialize; HitCollectionMap owns the collection.
  TrackRecordCollection* m_trackRecordCollection{};
  int m_SD_type; //TrackRecorderSD=1, CosmicTRSD=2
};
 
#endif // TRACKWRITEFASTSIM_TRACKFASTSIMSD_H