Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/event/src/G4SmartTrackStack.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 // G4SmartTrackStack class implementation
 27 //
 28 // Author: S.Kamperis - 4 October 2012
 29 // --------------------------------------------------------------------
 30 
 31 #include "G4SmartTrackStack.hh"
 32 #include "G4VTrajectory.hh"
 33 #include "G4Track.hh"
 34 
 35 void G4SmartTrackStack::dumpStatistics()
 36 {
 37   // Print to stderr so that we can split stats output from normal
 38   // output of Geant4 which is typically being printed to stdout
 39   for (G4int i=0; i<nTurn; ++i)
 40   {
 41     G4cerr << stacks[i]->GetNTrack() << " ";
 42     G4cerr << stacks[i]->getTotalEnergy() << " ";
 43   }
 44   G4cerr << G4endl;
 45 }
 46 
 47 G4SmartTrackStack::G4SmartTrackStack() 
 48 {
 49   for(G4int i=0; i<nTurn; ++i)
 50   {
 51     stacks[i] = new G4TrackStack(5000);
 52     energies[i] = 0.;
 53   }
 54 }
 55 
 56 G4SmartTrackStack::~G4SmartTrackStack()
 57 {
 58   for (auto* sp : stacks)
 59   {
 60     delete sp;
 61   }
 62 }
 63 
 64 void G4SmartTrackStack::TransferTo(G4TrackStack* aStack)
 65 {
 66   for (auto* sp : stacks)
 67   {
 68     sp->TransferTo(aStack);
 69   }
 70   nTracks = 0;
 71 }
 72 
 73 G4StackedTrack G4SmartTrackStack::PopFromStack()
 74 {
 75   G4StackedTrack aStackedTrack;
 76 
 77   if (nTracks != 0)
 78   {
 79     while (true)
 80     {
 81       if (stacks[fTurn]->GetNTrack() != 0u)
 82       {
 83         aStackedTrack = stacks[fTurn]->PopFromStack();
 84         energies[fTurn] -= aStackedTrack.GetTrack()->GetDynamicParticle()->GetTotalEnergy();
 85         --nTracks;
 86         break;
 87       }
 88       
 89       fTurn = (fTurn+1) % nTurn;
 90     }
 91   }
 92 
 93   return aStackedTrack;
 94 }
 95 
 96 enum
 97 {
 98   electronCode = 11, positronCode = -11, gammaCode = 22, neutronCode = 2112
 99 };
100 
101 void G4SmartTrackStack::PushToStack( const G4StackedTrack& aStackedTrack )
102 {
103 
104   G4int iDest = 0;
105   if (aStackedTrack.GetTrack()->GetParentID() != 0)
106   {
107     G4int code = aStackedTrack.GetTrack()->GetDynamicParticle()->GetPDGcode();
108     if (code == electronCode)
109       iDest = 2;
110     else if (code == gammaCode)
111       iDest = 3;
112     else if (code == positronCode)
113       iDest = 4;
114     else if (code == neutronCode)
115       iDest = 1;
116   }
117   else
118   {
119     // We have a primary track, which should go first.
120     fTurn = 0; // reseting the turn
121   }
122   stacks[iDest]->PushToStack(aStackedTrack);
123   energies[iDest] += aStackedTrack.GetTrack()->GetDynamicParticle()->GetTotalEnergy();
124   ++nTracks;
125   
126   G4long dy1 = stacks[iDest]->GetNTrack() - stacks[iDest]->GetSafetyValue1();
127   G4long dy2 = stacks[fTurn]->GetNTrack() - stacks[fTurn]->GetSafetyValue2();
128   
129   if (dy1 > 0 || dy1 > dy2 ||
130       (iDest == 2 &&
131        stacks[iDest]->GetNTrack() < 50 && energies[iDest] < energies[fTurn]))
132   {
133     fTurn = iDest;
134   }
135   
136   if (nTracks > maxNTracks) maxNTracks = nTracks;
137 }
138 
139 void G4SmartTrackStack::clear()
140 {
141   for (G4int i = 0; i < nTurn; ++i)
142   {
143     stacks[i]->clear();
144     energies[i] = 0.0;
145     fTurn = 0;
146   }
147   nTracks = 0;
148 }
149 
150 void G4SmartTrackStack::clearAndDestroy()
151 {
152   for (G4int i = 0; i < nTurn; ++i)
153   {
154     stacks[i]->clearAndDestroy();
155     energies[i] = 0.0;
156     fTurn = 0;
157   }
158   nTracks = 0;
159 }
160