Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/digits_hits/scorer/src/G4PSTrackCounter.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 //
 27 //
 28 // G4PSTrackCounter
 29 #include "G4PSTrackCounter.hh"
 30 #include "G4UnitsTable.hh"
 31 #include "G4VScoreHistFiller.hh"
 32 
 33 ///////////////////////////////////////////////////////////////////////////////
 34 // (Description)
 35 //   This is a primitive scorer class for scoring number of tracks in a cell.
 36 //
 37 // Created: 2007-02-02  Tsukasa ASO, Akinori Kimura.
 38 // 2010-07-22   Introduce Unit specification.
 39 //
 40 ///////////////////////////////////////////////////////////////////////////////
 41 
 42 G4PSTrackCounter::G4PSTrackCounter(const G4String& name, G4int direction, G4int depth)
 43   : G4VPrimitivePlotter(name, depth)
 44   , fDirection(direction)
 45 {
 46   SetUnit("");
 47 }
 48 
 49 G4bool G4PSTrackCounter::ProcessHits(G4Step* aStep, G4TouchableHistory*)
 50 {
 51   G4StepPoint* preStep = aStep->GetPreStepPoint();
 52   G4StepPoint* posStep = aStep->GetPostStepPoint();
 53   G4bool IsEnter       = preStep->GetStepStatus() == fGeomBoundary;
 54   G4bool IsExit        = posStep->GetStepStatus() == fGeomBoundary;
 55 
 56   // Regular : count in prestep volume.
 57   G4int index = GetIndex(aStep);
 58 
 59   //  G4cout << " trk " << aStep->GetTrack()->GetTrackID()
 60   //   << " index " << index << " In " << IsEnter << " Out " <<IsExit
 61   //   << " PreStatus " << preStep->GetStepStatus()
 62   //   << " PostStatus " << posStep->GetStepStatus()
 63   //   << " w " << aStep->GetPreStepPoint()->GetWeight()
 64   // <<    G4endl;
 65 
 66   G4bool flag = false;
 67 
 68   if(IsEnter && fDirection == fCurrent_In)
 69     flag = true;
 70   else if(IsExit && fDirection == fCurrent_Out)
 71     flag = true;
 72   else if((IsExit || IsEnter) && fDirection == fCurrent_InOut)
 73     flag = true;
 74 
 75   if(flag)
 76   {
 77     // G4cout << " Count " << G4endl;
 78     G4double val = 1.0;
 79     if(weighted)
 80       val *= aStep->GetPreStepPoint()->GetWeight();
 81     EvtMap->add(index, val);
 82 
 83     if(!hitIDMap.empty() && hitIDMap.find(index) != hitIDMap.cend())
 84     {
 85       auto filler = G4VScoreHistFiller::Instance();
 86       if(filler == nullptr)
 87       {
 88         G4Exception(
 89           "G4PSTrackCounter::ProcessHits", "SCORER0123", JustWarning,
 90           "G4TScoreHistFiller is not instantiated!! Histogram is not filled.");
 91       }
 92       else
 93       {
 94         filler->FillH1(hitIDMap[index],
 95                        aStep->GetPreStepPoint()->GetKineticEnergy(), val);
 96       }
 97     }
 98   }
 99 
100   return true;
101 }
102 
103 void G4PSTrackCounter::Initialize(G4HCofThisEvent* HCE)
104 {
105   EvtMap = new G4THitsMap<G4double>(detector->GetName(), GetName());
106   if(HCID < 0)
107   {
108     HCID = GetCollectionID(0);
109   }
110   HCE->AddHitsCollection(HCID, (G4VHitsCollection*) EvtMap);
111   // G4cout <<   "----- PS Initialize " << G4endl;
112 }
113 
114 void G4PSTrackCounter::clear() { EvtMap->clear(); }
115 
116 void G4PSTrackCounter::PrintAll()
117 {
118   G4cout << " MultiFunctionalDet  " << detector->GetName() << G4endl;
119   G4cout << " PrimitiveScorer " << GetName() << G4endl;
120   G4cout << " Number of entries " << EvtMap->entries() << G4endl;
121   for(const auto& [copy, count] : *(EvtMap->GetMap()))
122   {
123     G4cout << "  copy no.: " << copy
124            << "  track count: " << *(count) << " [tracks] " << G4endl;
125   }
126 }
127 
128 void G4PSTrackCounter::SetUnit(const G4String& unit)
129 {
130   if(unit.empty())
131   {
132     unitName  = unit;
133     unitValue = 1.0;
134   }
135   else
136   {
137     G4String msg = "Invalid unit [" + unit + "] (Current  unit is [" +
138                    GetUnit() + "] ) for " + GetName();
139     G4Exception("G4PSTrackCounter::SetUnit", "DetPS0018", JustWarning, msg);
140   }
141 }
142