RecordingEnvelope.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTruthBase_RECORDINGENVELOPE_H
#define MCTruthBase_RECORDINGENVELOPE_H
 
// Athena headers
#include "TrackRecord/TrackRecordCollection.h"
#include "StoreGate/WriteHandle.h"
 
// STL headers
#include <string>
 
class G4LogicalVolume;
class G4Step;
class G4VPhysicalVolume;
 
 
class RecordingEnvelope
{
public:
 
  RecordingEnvelope(const std::string& envelopeVolumeName, const std::string& trackRecordCollectionName);
 
  ~RecordingEnvelope();
 
  bool Initialize();
 
  void BeginOfEvent();
 
  inline int GetLevel() const;
 
  inline std::string GetVolumeName() const;
 
  inline const G4LogicalVolume* GetLogicalVolume() const;
 
  inline std::string GetTrackRecordCollectionName() const;
 
  void AddTrackRecord(const G4Step* aStep);
private:
  bool checkDaughters(const G4LogicalVolume *possibleParent, const G4VPhysicalVolume *thePhysicalVolume, int& level) const;
 
  int m_level;
 
 
  G4LogicalVolume *m_logicalVolume;
 
  const std::string m_envelopeVolumeName;
 
  SG::WriteHandle<TrackRecordCollection> m_trackRecordCollection;
};
 
std::string RecordingEnvelope::GetVolumeName() const
{
  return m_envelopeVolumeName;
}
 
const G4LogicalVolume* RecordingEnvelope::GetLogicalVolume() const
{
  return m_logicalVolume;
}
 
std::string RecordingEnvelope::GetTrackRecordCollectionName() const
{
  return m_trackRecordCollection.name();
}
 
int RecordingEnvelope::GetLevel() const
{
  return m_level;
}
 
#endif //MCTruthBase_RECORDINGENVELOPE_H