00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030 #include "G4PSPopulation.hh"
00031
00032
00034
00035
00036
00037
00038
00039
00041
00042 G4PSPopulation::G4PSPopulation(G4String name, G4int depth)
00043 :G4VPrimitiveScorer(name,depth),HCID(-1),weighted(false)
00044 {
00045 SetUnit("");
00046 }
00047
00048 G4PSPopulation::~G4PSPopulation()
00049 {;}
00050
00051 G4bool G4PSPopulation::ProcessHits(G4Step* aStep,G4TouchableHistory*)
00052 {
00053
00054 G4int index = GetIndex(aStep);
00055 G4TrackLogger& tlog = fCellTrackLogger[index];
00056 if (tlog.FirstEnterance(aStep->GetTrack()->GetTrackID())) {
00057 G4double val = 1.0;
00058 if(weighted) val *= aStep->GetPreStepPoint()->GetWeight();
00059 EvtMap->add(index,val);
00060 }
00061
00062 return TRUE;
00063 }
00064
00065 void G4PSPopulation::Initialize(G4HCofThisEvent* HCE)
00066 {
00067 EvtMap = new G4THitsMap<G4double>(detector->GetName(),GetName());
00068 if(HCID < 0) {HCID = GetCollectionID(0);}
00069 HCE->AddHitsCollection(HCID, (G4VHitsCollection*)EvtMap);
00070 }
00071
00072 void G4PSPopulation::EndOfEvent(G4HCofThisEvent*)
00073 {
00074 fCellTrackLogger.clear();
00075 }
00076
00077 void G4PSPopulation::clear(){
00078 EvtMap->clear();
00079 fCellTrackLogger.clear();
00080 }
00081
00082 void G4PSPopulation::DrawAll()
00083 {;}
00084
00085 void G4PSPopulation::PrintAll()
00086 {
00087 G4cout << " MultiFunctionalDet " << detector->GetName() << G4endl;
00088 G4cout << " PrimitiveScorer " << GetName() << G4endl;
00089 G4cout << " Number of entries " << EvtMap->entries() << G4endl;
00090 std::map<G4int,G4double*>::iterator itr = EvtMap->GetMap()->begin();
00091 for(; itr != EvtMap->GetMap()->end(); itr++) {
00092 G4cout << " copy no.: " << itr->first
00093 << " population: " << *(itr->second)/GetUnitValue()
00094 << " [tracks]"
00095 << G4endl;
00096 }
00097 }
00098
00099 void G4PSPopulation::SetUnit(const G4String& unit)
00100 {
00101 if (unit == "" ){
00102 unitName = unit;
00103 unitValue = 1.0;
00104 }else{
00105 G4String msg = "Invalid unit ["+unit+"] (Current unit is [" +GetUnit()+"] ) for " + GetName();
00106 G4Exception("G4PSPopulation::SetUnit","DetPS0014",JustWarning,msg);
00107 }
00108
00109 }