| Geant4 Cross Reference |
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 // INCL++ intra-nuclear cascade model
27 // Alain Boudard, CEA-Saclay, France
28 // Joseph Cugnon, University of Liege, Belgium
29 // Jean-Christophe David, CEA-Saclay, France
30 // Pekka Kaitaniemi, CEA-Saclay, France, and Helsinki Institute of Physics, Finland
31 // Sylvie Leray, CEA-Saclay, France
32 // Davide Mancusi, CEA-Saclay, France
33 //
34 #define INCLXX_IN_GEANT4_MODE 1
35
36 #include "globals.hh"
37
38 #include "G4INCLAvatarDumpAction.hh"
39 #include <sstream>
40 #include <string>
41
42 namespace G4INCL {
43
44 AvatarDumpAction::AvatarDumpAction() :
45 oFile(0),
46 eventCounter(0)
47 {
48 }
49
50 AvatarDumpAction::~AvatarDumpAction() {}
51
52 void AvatarDumpAction::beforeCascadeUserAction(IPropagationModel * /*pm*/) {
53 std::stringstream ss;
54 ss << "avatar-dump-" << eventCounter << ".dat";
55 oFile = new std::ofstream(ss.str().c_str());
56 }
57
58 void AvatarDumpAction::afterAvatarUserAction(IAvatar *avatar, Nucleus *nucleus, FinalState *finalState) {
59 ParticleList particles = nucleus->getStore()->getParticles();
60 ParticleList highlight;
61 if(finalState) {
62 ParticleList const &modified = finalState->getModifiedParticles();
63 highlight.insert(highlight.end(), modified.begin(), modified.end());
64 ParticleList const &outgoing = finalState->getOutgoingParticles();
65 highlight.insert(highlight.end(), outgoing.begin(), outgoing.end());
66 ParticleList const &destroyed = finalState->getDestroyedParticles();
67 highlight.insert(highlight.end(), destroyed.begin(), destroyed.end());
68 ParticleList const &created = finalState->getCreatedParticles();
69 highlight.insert(highlight.end(), created.begin(), created.end());
70 ParticleList const &entering = finalState->getEnteringParticles();
71 highlight.insert(highlight.end(), entering.begin(), entering.end());
72 particles.insert(particles.end(), created.begin(), created.end());
73 particles.insert(particles.end(), entering.begin(), entering.end());
74 }
75
76 (*oFile) << avatar->getTime() << '\t' << avatar->getType() << '\t' << particles.size() << '\n';
77 for(ParticleIter p=particles.begin(), e=particles.end(); p!=e; ++p) {
78 ThreeVector const &pos = (*p)->getPosition();
79 ThreeVector const &vel = (*p)->getPropagationVelocity();
80 G4int highlightIt = highlight.contains(*p);
81 (*oFile)
82 << (*p)->getID() << '\t'
83 << (*p)->getParticipantType() << '\t'
84 << (*p)->getType() << '\t'
85 << pos.getX() << '\t'
86 << pos.getY() << '\t'
87 << pos.getZ() << '\t'
88 << vel.getX() << '\t'
89 << vel.getY() << '\t'
90 << vel.getZ() << '\t'
91 << (*p)->getKineticEnergy() << '\t'
92 << (*p)->getPotentialEnergy() << '\t'
93 << highlightIt << '\n';
94 }
95 }
96
97 void AvatarDumpAction::afterCascadeUserAction(Nucleus * /*nucleus*/) {
98 oFile->close();
99 delete oFile;
100 ++eventCounter;
101 }
102
103 }
104