// Process hits.
G4bool base_SD::ProcessHits( G4Step * step, G4TouchableHistory * /*ROhist*/ ) {
MCDetectorHit hit;
hit.Edep=step->GetTotalEnergyDeposit();
G4StepPoint * prestep = step->GetPreStepPoint();
G4StepPoint * poststep = step->GetPostStepPoint();
G4TouchableHandle touchable = prestep->GetTouchableHandle();
hit.id=touchable->GetCopyNumber();
if (hit.id>=2000)
{
plugin->debug(0,"Hit id is too large... bailing out \n");
exit(-1);
}
G4Track * track = step->GetTrack();
hit.time = prestep->GetGlobalTime();
hit.track = track->GetTrackID();
// Get the true world and local coordinate positions.
G4ThreeVector trueworld =
( prestep->GetPosition() + poststep->GetPosition() ) / 2.0;
// If this first time store transformation and inverse.
if( !transformValid[hit.id] ) {
if (WorldtoLocal.size()<hit.id+1) //too small
{
WorldtoLocal.resize(hit.id+1);
LocaltoWorld.resize(hit.id+1);
}
WorldtoLocal[hit.id] = touchable->GetHistory()->GetTopTransform();
LocaltoWorld[hit.id] = WorldtoLocal[hit.id].Inverse();
transformValid[hit.id] = true;
}
G4ThreeVector Tmp = WorldtoLocal[hit.id].TransformPoint(trueworld);
hit.world=TVector3(trueworld.x(),trueworld.y(),trueworld.z());
hit.local=TVector3(Tmp.x(),Tmp.y(),Tmp.z());
// find average momentum
G4ThreeVector mom=0.5*(prestep->GetMomentum()+poststep->GetMomentum());
hit.fourmomentum.SetXYZM(mom.x()/MeV,mom.y()/MeV,mom.z()/MeV,prestep->GetMass()/MeV);
mom=WorldtoLocal[hit.id].TransformAxis(mom);
hit.localmomentum=TVector3(mom.x(),mom.y(),mom.z());
data->hits.push_back(hit);
return true;
}
uint64_t GflashCalorimeterSD::cellID(const G4GFlashSpot& aSpot) {
dd4hep::sim::Geant4VolumeManager volMgr = dd4hep::sim::Geant4Mapping::instance().volumeManager();
dd4hep::VolumeID volID = volMgr.volumeID(aSpot.GetTouchableHandle()());
if (m_seg.isValid()) {
G4ThreeVector global = aSpot.GetEnergySpot()->GetPosition();
G4ThreeVector local = aSpot.GetTouchableHandle()->GetHistory()->GetTopTransform().TransformPoint(global);
dd4hep::Position loc(local.x() * MM_2_CM, local.y() * MM_2_CM, local.z() * MM_2_CM);
dd4hep::Position glob(global.x() * MM_2_CM, global.y() * MM_2_CM, global.z() * MM_2_CM);
dd4hep::VolumeID cID = m_seg.cellID(loc, glob, volID);
return cID;
}
return volID;
}
/// Store Geant4 point and step information into tracker hit structure.
Geant4Tracker::Hit& Geant4Tracker::Hit::storePoint(const G4Step* step, const G4StepPoint* pnt) {
G4Track* trk = step->GetTrack();
G4ThreeVector pos = pnt->GetPosition();
G4ThreeVector mom = pnt->GetMomentum();
truth.trackID = trk->GetTrackID();
truth.pdgID = trk->GetDefinition()->GetPDGEncoding();
truth.deposit = step->GetTotalEnergyDeposit();
truth.time = trk->GetGlobalTime();
position.SetXYZ(pos.x(), pos.y(), pos.z());
momentum.SetXYZ(mom.x(), mom.y(), mom.z());
length = 0;
return *this;
}
long long Geant4SensitiveDetector::getCellID(G4Step* s) {
StepHandler h(s);
Geant4VolumeManager volMgr = Geant4Mapping::instance().volumeManager();
VolumeID volID = volMgr.volumeID(h.preTouchable());
Segmentation seg = m_readout.segmentation();
if ( seg.isValid() ) {
G4ThreeVector global = 0.5 * ( h.prePosG4()+h.postPosG4());
G4ThreeVector local = h.preTouchable()->GetHistory()->GetTopTransform().TransformPoint(global);
Position loc(local.x()*MM_2_CM, local.y()*MM_2_CM, local.z()*MM_2_CM);
Position glob(global.x()*MM_2_CM, global.y()*MM_2_CM, global.z()*MM_2_CM);
VolumeID cID = seg.cellID(loc,glob,volID);
return cID;
}
return volID;
}
uint64_t cellID(const DD4hep::Geometry::Segmentation& aSeg, const G4Step& aStep, bool aPreStepPoint) {
DD4hep::Simulation::Geant4VolumeManager volMgr = DD4hep::Simulation::Geant4Mapping::instance().volumeManager();
DD4hep::Geometry::VolumeManager::VolumeID volID = volMgr.volumeID(aStep.GetPreStepPoint()->GetTouchable());
if (aSeg.isValid()) {
G4ThreeVector global;
if (aPreStepPoint) {
global = aStep.GetPreStepPoint()->GetPosition();
} else {
global = 0.5 * (aStep.GetPreStepPoint()->GetPosition() + aStep.GetPostStepPoint()->GetPosition());
}
G4ThreeVector local =
aStep.GetPreStepPoint()->GetTouchable()->GetHistory()->GetTopTransform().TransformPoint(global);
DD4hep::Geometry::Position loc(local.x() * MM_2_CM, local.y() * MM_2_CM, local.z() * MM_2_CM);
DD4hep::Geometry::Position glob(global.x() * MM_2_CM, global.y() * MM_2_CM, global.z() * MM_2_CM);
DD4hep::Geometry::VolumeManager::VolumeID cID = aSeg.cellID(loc, glob, volID);
return cID;
}
return volID;
}
void BIEventAction::EndOfEventAction(const G4Event *event)
{
G4HCofThisEvent *hce = event->GetHCofThisEvent();
if (!hce) {
G4cout << "HCE not found: EndOfEventAction@BIEventAction" << G4endl;
exit(0);
}
BICommonHitsCollection *hitsCollection =
static_cast<BICommonHitsCollection *>(hce->GetHC(fHitsCollectionID));
if (!hitsCollection) {
G4cout << "HC not found: EndOfEventAction@BIEventAction" << G4endl;
exit(0);
}
G4int eventID = event->GetEventID();
G4AnalysisManager *anaMan = G4AnalysisManager::Instance();
const G4int kHit = hitsCollection->entries();
for (G4int iHit = 0; iHit < kHit; iHit++) {
BICommonHit *newHit = (*hitsCollection)[iHit];
if(fForGrid){
G4double depositEnergy = newHit->GetDepositEnergy();
anaMan->FillNtupleDColumn(0, 0, depositEnergy);
G4ThreeVector position = newHit->GetPosition();
anaMan->FillNtupleDColumn(0, 1, position.x());
anaMan->FillNtupleDColumn(0, 2, position.y());
G4int trackID = newHit->GetTrackID();
anaMan->FillNtupleIColumn(0, 3, trackID);
G4int pdgCode = newHit->GetPDGCode();
anaMan->FillNtupleIColumn(0, 4, pdgCode);
}
else {
anaMan->FillNtupleIColumn(0, 0, eventID); // EventID
G4int pdgCode = newHit->GetPDGCode();
anaMan->FillNtupleIColumn(0, 1, pdgCode);
G4double depositEnergy = newHit->GetDepositEnergy();
anaMan->FillNtupleDColumn(0, 2, depositEnergy);
G4double time = newHit->GetTime();
anaMan->FillNtupleDColumn(0, 3, time);
G4String volumeName = newHit->GetVolumeName();
anaMan->FillNtupleSColumn(0, 4, volumeName);
G4ThreeVector position = newHit->GetPosition();
anaMan->FillNtupleDColumn(0, 5, position.x());
anaMan->FillNtupleDColumn(0, 6, position.y());
anaMan->FillNtupleDColumn(0, 7, position.z());
G4ThreeVector prePosition = newHit->GetPrePosition();
anaMan->FillNtupleDColumn(0, 8, prePosition.x());
anaMan->FillNtupleDColumn(0, 9, prePosition.y());
anaMan->FillNtupleDColumn(0, 10, prePosition.z());
G4ThreeVector momentum = newHit->GetMomentum();
anaMan->FillNtupleDColumn(0, 11, momentum.x());
anaMan->FillNtupleDColumn(0, 12, momentum.y());
anaMan->FillNtupleDColumn(0, 13, momentum.z());
G4int isLast = newHit->GetIsLast();
anaMan->FillNtupleIColumn(0, 14, isLast);
G4int trackID = newHit->GetTrackID();
anaMan->FillNtupleIColumn(0, 15, trackID);
}
anaMan->AddNtupleRow(0);
}
}
// This function, when registered with the Geant4 run manager should be executed
// on each execution step to provide position (or any other information) about
// the propagated tracks for visualization
void MCPropVis::UserSteppingAction(const G4Step* step)
{
// Get the track ID for this step.
int cid = step->GetTrack()->GetTrackID();
// std::cout<<"Step! "<<cid<<" "<<step->GetTrack()->GetParticleDefinition()->GetParticleName()<<"\n";
// Container index for the track information
int ind = -1; // Intiallize to -1 to indicate index not yet found
// If the track ID is the previous one save the search, else search through
// the known tracks for the right one
if (cid == lid) ind = lind;
else
{
for (unsigned int j=0; j<ids.size(); j++)
{
if (ids[j] == cid)
{
ind = j;
break;
}
}
}
// If the index hasn't been set after this, create a new track set, get the
// particle type, and set the track's properties
if (ind == -1)
{
ids.push_back(cid);
tracks.push_back(new TEveLine);
hits.push_back(new TEvePointSet);
ind = ids.size()-1; // Index is the end of the vector
// Particle type
G4String type = step->GetTrack()->GetParticleDefinition()->GetParticleName();
// std::cout<<cid<<" "<<type<<"\n";
// Set color based on particle type
Color_t color;
if (type.compare("proton") == 0) color = kBlue;
else if (type.compare("e+") == 0) color = kYellow-2;
else if (type.compare("e-") == 0) color = kRed;
else if (type.compare("gamma") == 0) color = kGreen;
else if (type.compare("neutron") == 0) color = kCyan;
else if (type.compare("mu+") == 0) color = kGray+1;
else if (type.compare("mu-") == 0) color = kBlack;
else if (type.compare("pi+") == 0) color = kCyan+3;
else if (type.compare("pi-") == 0) color = kOrange;
else color = kViolet;
tracks[ind]->SetMainColor(color);
tracks[ind]->SetLineColor(color);
hits[ind]->SetMainColor(color);
// Register the visualization set with the manager
gEve->AddElement(tracks[ind]);
gEve->AddElement(hits[ind]);
}
// Save the current IDs for the next step
lid = cid;
lind = ind;
// Add this step point to the appropriate track object
G4ThreeVector poss = step->GetPreStepPoint()->GetPosition();
tracks[ind]->SetNextPoint(poss.x()/10,poss.y()/10,poss.z()/10);
gEve->AddElement(tracks[ind]);
// Check to see if the step is in a sensitive detector. If so, draw it as a
// hit point
if (step->GetPreStepPoint()->GetSensitiveDetector())
{
hits[ind]->SetNextPoint(poss.x()/10,poss.y()/10,poss.z()/10);
gEve->AddElement(hits[ind]);
}
// std::cout<<(step->GetTrack()->GetParticleDefinition()->GetParticleName())<<"\n";
}
