Geant4  10.02.p01
G4PSTrackCounter.cc
Go to the documentation of this file.
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 // $Id: G4PSTrackCounter.cc 81087 2014-05-20 15:44:27Z gcosmo $
28 //
29 // G4PSTrackCounter
30 #include "G4PSTrackCounter.hh"
31 #include "G4UnitsTable.hh"
32 
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 //
41 
43  :G4VPrimitiveScorer(name,depth),HCID(-1),fDirection(direction),
44  EvtMap(0),weighted(false)
45 {
46  SetUnit("");
47 }
48 
50 {;}
51 
53 {
54 
55  G4StepPoint* preStep = aStep->GetPreStepPoint();
56  G4StepPoint* posStep = aStep->GetPostStepPoint();
57  G4bool IsEnter = preStep->GetStepStatus()==fGeomBoundary;
58  G4bool IsExit = posStep->GetStepStatus()==fGeomBoundary;
59 
60  // Regular : count in prestep volume.
61  G4int index = GetIndex(aStep);
62 
63  // G4cout << " trk " << aStep->GetTrack()->GetTrackID()
64  // << " index " << index << " In " << IsEnter << " Out " <<IsExit
65  // << " PreStatus " << preStep->GetStepStatus()
66  // << " PostStatus " << posStep->GetStepStatus()
67  // << " w " << aStep->GetPreStepPoint()->GetWeight()
68  // << G4endl;
69 
70  G4bool flag = FALSE;
71 
72  if ( IsEnter && fDirection == fCurrent_In ) flag = TRUE;
73  else if ( IsExit && fDirection == fCurrent_Out ) flag = TRUE;
74  else if ( (IsExit||IsEnter) && fDirection == fCurrent_InOut ) flag = TRUE;
75 
76  if ( flag ){
77  //G4cout << " Count " << G4endl;
78  G4double val = 1.0;
79  if(weighted) val *= aStep->GetPreStepPoint()->GetWeight();
80  EvtMap->add(index,val);
81  }
82 
83  return TRUE;
84 }
85 
87 {
89  if(HCID < 0) {HCID = GetCollectionID(0);}
91  //G4cout << "----- PS Initialize " << G4endl;
92 }
93 
95 {;}
96 
98  EvtMap->clear();
99 }
100 
102 {;}
103 
105 {
106  G4cout << " MultiFunctionalDet " << detector->GetName() << G4endl;
107  G4cout << " PrimitiveScorer " << GetName() << G4endl;
108  G4cout << " Number of entries " << EvtMap->entries() << G4endl;
109  std::map<G4int,G4double*>::iterator itr = EvtMap->GetMap()->begin();
110  for(; itr != EvtMap->GetMap()->end(); itr++) {
111  G4cout << " copy no.: " << itr->first
112  << " track count: " << *(itr->second)
113  << " [tracks] "
114  << G4endl;
115  }
116 }
117 
119 {
120  if (unit == "" ){
121  unitName = unit;
122  unitValue = 1.0;
123  }else{
124  G4String msg = "Invalid unit ["+unit+"] (Current unit is [" +GetUnit()+"] ) for " + GetName();
125  G4Exception("G4PSTrackCounter::SetUnit","DetPS0018",JustWarning,msg);
126  }
127 
128 }
129 
virtual void SetUnit(const G4String &unit)
void clear()
Definition: G4THitsMap.hh:209
G4double GetWeight() const
virtual void PrintAll()
G4String GetName() const
G4THitsMap< G4double > * EvtMap
G4String name
Definition: TRTMaterials.hh:40
G4StepStatus GetStepStatus() const
virtual void clear()
int G4int
Definition: G4Types.hh:78
virtual ~G4PSTrackCounter()
G4StepPoint * GetPreStepPoint() const
virtual void EndOfEvent(G4HCofThisEvent *)
G4GLOB_DLL std::ostream G4cout
bool G4bool
Definition: G4Types.hh:79
G4int add(const G4int &key, T *&aHit) const
Definition: G4THitsMap.hh:138
#define FALSE
Definition: globals.hh:52
virtual G4int GetIndex(G4Step *)
#define TRUE
Definition: globals.hh:55
Definition: G4Step.hh:76
virtual G4bool ProcessHits(G4Step *, G4TouchableHistory *)
G4int GetCollectionID(G4int)
void AddHitsCollection(G4int HCID, G4VHitsCollection *aHC)
void G4Exception(const char *originOfException, const char *exceptionCode, G4ExceptionSeverity severity, const char *comments)
Definition: G4Exception.cc:41
G4int entries() const
Definition: G4THitsMap.hh:79
G4PSTrackCounter(G4String name, G4int direction, G4int depth=0)
G4StepPoint * GetPostStepPoint() const
std::map< G4int, T * > * GetMap() const
Definition: G4THitsMap.hh:68
#define G4endl
Definition: G4ios.hh:61
G4MultiFunctionalDetector * detector
double G4double
Definition: G4Types.hh:76
const G4String & GetUnit() const
virtual void DrawAll()
virtual void Initialize(G4HCofThisEvent *)