Geant4 Cross Reference |
1 // 1 2 // ******************************************* 3 // * License and Disclaimer 4 // * 5 // * The Geant4 software is copyright of th 6 // * the Geant4 Collaboration. It is provided 7 // * conditions of the Geant4 Software License 8 // * LICENSE and available at http://cern.ch/ 9 // * include a list of copyright holders. 10 // * 11 // * Neither the authors of this software syst 12 // * institutes,nor the agencies providing fin 13 // * work make any representation or warran 14 // * regarding this software system or assum 15 // * use. Please see the license in the file 16 // * for the full disclaimer and the limitatio 17 // * 18 // * This code implementation is the result 19 // * technical work of the GEANT4 collaboratio 20 // * By using, copying, modifying or distri 21 // * any work based on the software) you ag 22 // * use in resulting scientific publicati 23 // * acceptance of all terms of the Geant4 Sof 24 // ******************************************* 25 // 26 // 27 // ------------------------------------------- 28 // GEANT 4 - Underground Dark Matter Detecto 29 // 30 // For information related to this code c 31 // e-mail: alexander.howard@cern.ch 32 // ------------------------------------------- 33 // Comments 34 // 35 // Underground Advanced 36 // by A. Howard and H. Araujo 37 // (27th November 2001) 38 // 39 // History: 40 // 21 Feb 2002 AH: Added Analysis 41 // 42 // SteppingAction program 43 // ------------------------------------------- 44 45 #include "DMXSteppingAction.hh" 46 #include "DMXSteppingActionMessenger.hh" 47 48 #include "DMXEventAction.hh" 49 50 #include "G4AnalysisManager.hh" 51 #include "G4RunManager.hh" 52 #include "G4Track.hh" 53 #include "G4Step.hh" 54 #include "G4StepPoint.hh" 55 #include "G4TrackStatus.hh" 56 #include "G4ParticleDefinition.hh" 57 #include "G4ParticleTypes.hh" 58 #include "G4VVisManager.hh" 59 #include "G4Colour.hh" 60 #include "G4Polyline.hh" 61 #include "G4VisAttributes.hh" 62 #include "globals.hh" 63 #include "G4ios.hh" 64 65 66 //....oooOO0OOooo........oooOO0OOooo........oo 67 68 DMXSteppingAction::DMXSteppingAction() 69 : evtAction(0) { 70 71 steppingMessenger = new DMXSteppingActionMes 72 73 // defaults for messenger 74 colourNeutronFlag = "magenta"; 75 colourGammaFlag = "cyan"; 76 colourOpticalFlag = "white"; 77 colourChargedPlusFlag = "red"; 78 colourChargedMinusFlag = "blue"; 79 80 } 81 82 //....oooOO0OOooo........oooOO0OOooo........oo 83 84 DMXSteppingAction::~DMXSteppingAction() 85 { 86 87 delete steppingMessenger; 88 89 } 90 91 //....oooOO0OOooo........oooOO0OOooo........oo 92 93 void DMXSteppingAction::UserSteppingAction(con 94 { 95 if (!evtAction) 96 evtAction = 97 dynamic_cast<const DMXEventAction*> 98 (G4RunManager::GetRunManager()->GetUserE 99 100 101 // removed 28/11/01 - unnecessary unless pro 102 // kill track if too many steps 103 // NB: This is set to DBL_MAX - therefore ma 104 // G4int MaxNoSteps = DBL_MAX; 105 // G4int StepNo = fStep->GetTrack()->GetCur 106 // if(StepNo >= MaxNoSteps) fStep->GetTrack 107 108 G4int StepNo = fStep->GetTrack()->GetCurrent 109 if(StepNo == 1) 110 { 111 G4double partEnergy = fStep->GetPreStepP 112 G4ParticleDefinition* particleType = fSt 113 114 G4AnalysisManager* man = G4AnalysisManag 115 if (particleType == G4Gamma::Definition( 116 man->FillH1(8,partEnergy); 117 else if (particleType == G4Neutron::Defi 118 man->FillH1(9,partEnergy); 119 else if (particleType == G4Electron::Def 120 man->FillH1(10,partEnergy); 121 else if (particleType == G4Positron::Def 122 man->FillH1(11,partEnergy); 123 else 124 man->FillH1(12,partEnergy); 125 } 126 127 128 // check what is to be drawn from EventActio 129 G4String drawColsFlag = evtAction->GetDrawCo 130 G4String drawTrksFlag = evtAction->GetDrawTr 131 132 // draw by step (here) instead of by event ( 133 if (drawColsFlag=="custom" && drawTrksFlag!= 134 135 // check that VisManager exists 136 G4VVisManager* pVVisManager = G4VVisManage 137 if(pVVisManager) { 138 139 // particle colour in a string 140 G4String name = fStep->GetTrack()->GetDe 141 G4String strColour; 142 if(name=="neutron") { 143 if(drawTrksFlag=="charged") return; 144 strColour = colourNeutronFlag; 145 } else if (name=="gamma") { 146 if(drawTrksFlag=="charged") return; 147 strColour = colourGammaFlag; 148 } else if (name=="opticalphoton") { 149 if(drawTrksFlag!="all") return; 150 strColour = colourOpticalFlag; 151 } 152 else if (name=="alpha" || name=="e+") 153 strColour = colourChargedPlusFlag; 154 else 155 strColour = colourChargedMinusFlag; 156 157 // convert string to G4Colour 158 G4Colour colour; 159 if (strColour=="white") colour=G4 160 else if(strColour=="grey" ) colour=G4 161 else if(strColour=="lgrey") colour=G4 162 else if(strColour=="black") colour=G4 163 else if(strColour=="red") colour=G4 164 else if(strColour=="green") colour=G4 165 else if(strColour=="blue") colour=G4 166 else if(strColour=="cyan") colour=G4 167 else if(strColour=="magenta") colour=G4 168 else if(strColour=="yellow") colour=G4 169 else if(strColour=="lgreen") colour=G4 170 else if(strColour=="lblue") colour=G4 171 else colour=G4 172 173 // create line with colour 174 G4VisAttributes attribs(colour); 175 G4Polyline polyline; 176 polyline.SetVisAttributes(attribs); 177 178 // draw line 179 G4Point3D start(fStep->GetPreStepPoint() 180 G4Point3D end(fStep->GetPostStepPoint()- 181 polyline.push_back(start); 182 polyline.push_back(end); 183 pVVisManager->Draw(polyline); 184 } 185 186 } 187 188 } 189 190 191