TrackFastSimSD Class Reference

#include <TrackFastSimSD.h>

Inheritance diagram for TrackFastSimSD:

Collaboration diagram for TrackFastSimSD:

Public Member Functions
	TrackFastSimSD (const std::string &name, const std::string &outputCollectionName, const int SD_type=0)
	~TrackFastSimSD ()
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)
	Templated method to stuff a single hit into the sensitive detector class.

Private Member Functions
TrackRecordCollection *	getTrackRecordCollection () const

Private Attributes
std::string	m_outputCollectionName
TrackRecordCollection *	m_trackRecordCollection {}
int	m_SD_type

Detailed Description

Definition at line 23 of file TrackFastSimSD.h.

Constructor & Destructor Documentation

TrackFastSimSD()

TrackFastSimSD::TrackFastSimSD	(	const std::string &	name,
		const std::string &	outputCollectionName,
		const int	SD_type = 0 )

Definition at line 26 of file TrackFastSimSD.cxx.

  : G4VSensitiveDetector( name )
  , m_outputCollectionName( outputCollectionName )
  , m_SD_type( SD_type )
{
}

~TrackFastSimSD()

TrackFastSimSD::~TrackFastSimSD ( )

inline

Definition at line 27 of file TrackFastSimSD.h.

{}

Member Function Documentation

AddHit()

template<class... Args>

bool TrackFastSimSD::AddHit ( Args &&... args )

inline

Templated method to stuff a single hit into the sensitive detector class.

This could get rather tricky, but the idea is to allow fast simulations to use the very same SD classes as the standard simulation.

Definition at line 38 of file TrackFastSimSD.h.

  {
    if (m_trackRecordCollection) {
      m_trackRecordCollection->Emplace(std::forward<Args>(args)...);
      return true;
    }
    return false;
  }

getTrackRecordCollection()

TrackRecordCollection * TrackFastSimSD::getTrackRecordCollection ( ) const

private

Definition at line 132 of file TrackFastSimSD.cxx.

{
  auto* eventManager = G4EventManager::GetEventManager();
  if (!eventManager) {
    return nullptr;
  }
 
  auto* eventInfo =
    dynamic_cast<AtlasG4EventUserInfo*>(eventManager->GetUserInformation());
  if (!eventInfo) {
    return nullptr;
  }
 
  std::shared_ptr<HitCollectionMap> hitCollections = eventInfo->GetHitCollectionMap();
  return hitCollections ? hitCollections->Find<TrackRecordCollection>(m_outputCollectionName)
                        : nullptr;
}

Initialize()

void TrackFastSimSD::Initialize ( G4HCofThisEvent * )

finaloverride

Definition at line 34 of file TrackFastSimSD.cxx.

{
  m_trackRecordCollection = getTrackRecordCollection();
}

ProcessHits()

G4bool TrackFastSimSD::ProcessHits ( G4Step * aStep, G4TouchableHistory *  )

finaloverride

Definition at line 39 of file TrackFastSimSD.cxx.

{
  if (!m_SD_type) { return false; }
  const G4StepPoint *preStep=aStep->GetPreStepPoint();
  const G4VPhysicalVolume *preVol=preStep->GetPhysicalVolume();
  const G4StepPoint *postStep=aStep->GetPostStepPoint();
  const G4VPhysicalVolume *postVol=postStep->GetPhysicalVolume();
 
  if (preVol==postVol) { return false; }
 
  const G4Track *track(aStep->GetTrack());
 
  // PDG Code
  int pdgcode = (track->GetDefinition())?track->GetDefinition()->GetPDGEncoding():0;
  if (track->GetDefinition() == G4Geantino::Definition() ) pdgcode=999;
  if (track->GetDefinition() == G4ChargedGeantino::Definition() ) pdgcode=998;
 
  // Position and Momentum
  G4ThreeVector pos=postStep->GetPosition();
  G4ThreeVector mom=postStep->GetMomentum();
 
  if(m_SD_type==2)
    {
      // need to get the local position
      const G4TouchableHistory* touchHist = static_cast<const G4TouchableHistory*>(preStep->GetTouchable());
      const G4AffineTransform trans = track->GetTouchable()->GetHistory()->GetTopTransform(); // trans from global to local
      G4ThreeVector localPos=trans.TransformPoint(pos);
      G4ThreeVector localMom=mom;
      trans.ApplyAxisTransform(localMom); // rotate momentum as needed
      const G4VSolid *shape= touchHist->GetSolid();
      const G4ThreeVector normal=shape->SurfaceNormal(localPos);
      // check if particle is entering. if it is extiting don't record it.
      if(normal.dot(localPos)>=0.) return false;
    }
 
  // Energy
  const double ener=postStep->GetTotalEnergy();
 
  // Time
  const double time=postStep->GetGlobalTime();
 
  // Barcode
  TrackHelper trHelp(track);
  const int barcode = trHelp.GetBarcode(); // FIXME barcode based
  const int id = trHelp.GetUniqueID();
  const int status = trHelp.GetStatus();
 
  //create the TimedTrackRecord
  return AddHit(pdgcode,
                status,
                ener,
                mom,
                pos,
                time,
                barcode, // FIXME barcode based
                id,
                preVol->GetName());
}

WriteTrack()

void TrackFastSimSD::WriteTrack	(	const G4Track *	track,
		const bool	originPos,
		const bool	originMom )

Definition at line 98 of file TrackFastSimSD.cxx.

{
  if (!track) { G4cout << "ERROR: the track pointer was zero" << G4endl; return; }
 
  G4VPhysicalVolume *preVol=track->GetVolume();
 
  const int pdgcode = (track->GetDefinition())?track->GetDefinition()->GetPDGEncoding():0;
 
  const G4ThreeVector pos = originPos?track->GetVertexPosition():track->GetPosition();
  const double ener=originMom?(track->GetVertexKineticEnergy()+track->GetDynamicParticle()->GetMass()):track->GetTotalEnergy();
  G4ThreeVector mom = track->GetMomentum();
  if (originMom){
    double mommag = std::sqrt(std::pow(ener,2)-std::pow(track->GetDynamicParticle()->GetMass(),2));
    mom = track->GetVertexMomentumDirection()*mommag;
  }
 
  const double time=track->GetGlobalTime();
  TrackHelper trHelp(track);
  const int barcode = trHelp.GetBarcode(); // FIXME barcode based
  const int id = trHelp.GetUniqueID();
  const int status = trHelp.GetStatus();
 
  //create the TimedTrackRecord
  AddHit(pdgcode,
         status,
         ener,
         mom,
         pos,
         time,
         barcode, // FIXME barcode based
         id,
         preVol ? preVol->GetName() : "Unknown");
}

Member Data Documentation

m_outputCollectionName

std::string TrackFastSimSD::m_outputCollectionName

private

Definition at line 50 of file TrackFastSimSD.h.

m_SD_type

int TrackFastSimSD::m_SD_type

private

Definition at line 53 of file TrackFastSimSD.h.

m_trackRecordCollection

TrackRecordCollection* TrackFastSimSD::m_trackRecordCollection {}

private

Definition at line 52 of file TrackFastSimSD.h.

{};

---

The documentation for this class was generated from the following files:

• TrackFastSimSD.h
• TrackFastSimSD.cxx