Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/runAndEvent/RE04/src/RE04Trajectory.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  1 //
  2 // ********************************************************************
  3 // * License and Disclaimer                                           *
  4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.                             *
 10 // *                                                                  *
 11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                                                  *
 18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // ********************************************************************
 25 //
 26 /// \file runAndEvent/RE04/src/RE04Trajectory.cc
 27 /// \brief Implementation of the RE04Trajectory class
 28 //
 29 //
 30 #include "RE04Trajectory.hh"
 31 
 32 #include "RE04TrajectoryPoint.hh"
 33 
 34 #include "G4AttDef.hh"
 35 #include "G4AttDefStore.hh"
 36 #include "G4AttValue.hh"
 37 #include "G4ParticleTable.hh"
 38 #include "G4UIcommand.hh"
 39 #include "G4UnitsTable.hh"
 40 
 41 // #define G4ATTDEBUG
 42 #ifdef G4ATTDEBUG
 43 #  include "G4AttCheck.hh"
 44 #endif
 45 
 46 G4ThreadLocal G4Allocator<RE04Trajectory>* faTrajAllocator = 0;
 47 
 48 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 49 RE04Trajectory::RE04Trajectory()
 50   : G4VTrajectory(),
 51     fPositionRecord(0),
 52     fTrackID(0),
 53     fParentID(0),
 54     fPDGEncoding(0),
 55     fPDGCharge(0.0),
 56     fParticleName(""),
 57     fInitialKineticEnergy(0.),
 58     fInitialMomentum(G4ThreeVector())
 59 {
 60   ;
 61 }
 62 
 63 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 64 RE04Trajectory::RE04Trajectory(const G4Track* aTrack)
 65 {
 66   G4ParticleDefinition* fpParticleDefinition = aTrack->GetDefinition();
 67   fParticleName = fpParticleDefinition->GetParticleName();
 68   fPDGCharge = fpParticleDefinition->GetPDGCharge();
 69   fPDGEncoding = fpParticleDefinition->GetPDGEncoding();
 70   fTrackID = aTrack->GetTrackID();
 71   fParentID = aTrack->GetParentID();
 72   fInitialKineticEnergy = aTrack->GetKineticEnergy();
 73   fInitialMomentum = aTrack->GetMomentum();
 74   fPositionRecord = new TrajectoryPointContainer();
 75   // Following is for the first trajectory point
 76   fPositionRecord->push_back(new RE04TrajectoryPoint(aTrack->GetPosition(), aTrack->GetMaterial()));
 77 }
 78 
 79 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 80 RE04Trajectory::RE04Trajectory(RE04Trajectory& right) : G4VTrajectory()
 81 {
 82   fParticleName = right.fParticleName;
 83   fPDGCharge = right.fPDGCharge;
 84   fPDGEncoding = right.fPDGEncoding;
 85   fTrackID = right.fTrackID;
 86   fParentID = right.fParentID;
 87   fInitialKineticEnergy = right.fInitialKineticEnergy;
 88   fInitialMomentum = right.fInitialMomentum;
 89   fPositionRecord = new TrajectoryPointContainer();
 90 
 91   for (size_t i = 0; i < right.fPositionRecord->size(); i++) {
 92     RE04TrajectoryPoint* rightPoint = (RE04TrajectoryPoint*)((*(right.fPositionRecord))[i]);
 93     fPositionRecord->push_back(new RE04TrajectoryPoint(*rightPoint));
 94   }
 95 }
 96 
 97 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 98 RE04Trajectory::~RE04Trajectory()
 99 {
100   if (fPositionRecord) {
101     //  fPositionRecord->clearAndDestroy();
102     size_t i;
103     for (i = 0; i < fPositionRecord->size(); i++) {
104       delete (*fPositionRecord)[i];
105     }
106     fPositionRecord->clear();
107     delete fPositionRecord;
108   }
109 }
110 
111 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
112 void RE04Trajectory::ShowTrajectory(std::ostream& os) const
113 {
114   // Invoke the default implementation in G4VTrajectory...
115   G4VTrajectory::ShowTrajectory(os);
116   // ... or override with your own code here.
117 }
118 
119 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
120 void RE04Trajectory::DrawTrajectory() const
121 {
122   // Invoke the default implementation in G4VTrajectory...
123   G4VTrajectory::DrawTrajectory();
124   // ... or override with your own code here.
125 }
126 
127 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
128 const std::map<G4String, G4AttDef>* RE04Trajectory::GetAttDefs() const
129 {
130   G4bool isNew;
131   std::map<G4String, G4AttDef>* store = G4AttDefStore::GetInstance("RE04Trajectory", isNew);
132   if (isNew) {
133     G4String id("ID");
134     (*store)[id] = G4AttDef(id, "Track ID", "Physics", "", "G4int");
135 
136     G4String pid("PID");
137     (*store)[pid] = G4AttDef(pid, "Parent ID", "Physics", "", "G4int");
138 
139     G4String pn("PN");
140     (*store)[pn] = G4AttDef(pn, "Particle Name", "Physics", "", "G4String");
141 
142     G4String ch("Ch");
143     (*store)[ch] = G4AttDef(ch, "Charge", "Physics", "e+", "G4double");
144 
145     G4String pdg("PDG");
146     (*store)[pdg] = G4AttDef(pdg, "PDG Encoding", "Physics", "", "G4int");
147 
148     G4String ike("IKE");
149     (*store)[ike] = G4AttDef(ike, "Initial kinetic energy", "Physics", "G4BestUnit", "G4double");
150 
151     G4String iMom("IMom");
152     (*store)[iMom] = G4AttDef(iMom, "Initial momentum", "Physics", "G4BestUnit", "G4ThreeVector");
153 
154     G4String iMag("IMag");
155     (*store)[iMag] =
156       G4AttDef(iMag, "Initial momentum magnitude", "Physics", "G4BestUnit", "G4double");
157 
158     G4String ntp("NTP");
159     (*store)[ntp] = G4AttDef(ntp, "No. of points", "Physics", "", "G4int");
160   }
161   return store;
162 }
163 
164 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
165 std::vector<G4AttValue>* RE04Trajectory::CreateAttValues() const
166 {
167   std::vector<G4AttValue>* values = new std::vector<G4AttValue>;
168 
169   values->push_back(G4AttValue("ID", G4UIcommand::ConvertToString(fTrackID), ""));
170 
171   values->push_back(G4AttValue("PID", G4UIcommand::ConvertToString(fParentID), ""));
172 
173   values->push_back(G4AttValue("PN", fParticleName, ""));
174 
175   values->push_back(G4AttValue("Ch", G4UIcommand::ConvertToString(fPDGCharge), ""));
176 
177   values->push_back(G4AttValue("PDG", G4UIcommand::ConvertToString(fPDGEncoding), ""));
178 
179   values->push_back(G4AttValue("IKE", G4BestUnit(fInitialKineticEnergy, "Energy"), ""));
180 
181   values->push_back(G4AttValue("IMom", G4BestUnit(fInitialMomentum, "Energy"), ""));
182 
183   values->push_back(G4AttValue("IMag", G4BestUnit(fInitialMomentum.mag(), "Energy"), ""));
184 
185   values->push_back(G4AttValue("NTP", G4UIcommand::ConvertToString(GetPointEntries()), ""));
186 
187 #ifdef G4ATTDEBUG
188   G4cout << G4AttCheck(values, GetAttDefs());
189 #endif
190 
191   return values;
192 }
193 
194 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
195 void RE04Trajectory::AppendStep(const G4Step* aStep)
196 {
197   fPositionRecord->push_back(new RE04TrajectoryPoint(aStep->GetPostStepPoint()->GetPosition(),
198                                                      aStep->GetPreStepPoint()->GetMaterial()));
199 }
200 
201 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
202 G4ParticleDefinition* RE04Trajectory::GetParticleDefinition()
203 {
204   return (G4ParticleTable::GetParticleTable()->FindParticle(fParticleName));
205 }
206 
207 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
208 void RE04Trajectory::MergeTrajectory(G4VTrajectory* secondTrajectory)
209 {
210   if (!secondTrajectory) return;
211 
212   RE04Trajectory* seco = (RE04Trajectory*)secondTrajectory;
213   G4int ent = seco->GetPointEntries();
214   for (G4int i = 1; i < ent; i++)  // initial point of the second trajectory
215                                    // should not be merged
216   {
217     fPositionRecord->push_back((*(seco->fPositionRecord))[i]);
218     //    fPositionRecord->push_back(seco->fPositionRecord->removeAt(1));
219   }
220   delete (*seco->fPositionRecord)[0];
221   seco->fPositionRecord->clear();
222 }
223