Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // 26 // 27 // ------------------------------------------- 27 // ------------------------------------------------------------------- 28 // 28 // 29 // GEANT4 Class header file 29 // GEANT4 Class header file 30 // 30 // 31 // 31 // 32 // File name: G4EmElementSelector 32 // File name: G4EmElementSelector 33 // 33 // 34 // Author: Vladimir Ivanchenko 34 // Author: Vladimir Ivanchenko 35 // 35 // 36 // Creation date: 29.05.2008 36 // Creation date: 29.05.2008 37 // 37 // 38 // Modifications: 38 // Modifications: 39 // 39 // 40 // 40 // 41 // Class Description: 41 // Class Description: 42 // 42 // 43 // Generic helper class for the random selecti 43 // Generic helper class for the random selection of an element 44 44 45 // ------------------------------------------- 45 // ------------------------------------------------------------------- 46 // 46 // 47 47 48 #ifndef G4EmElementSelector_h 48 #ifndef G4EmElementSelector_h 49 #define G4EmElementSelector_h 1 49 #define G4EmElementSelector_h 1 50 50 51 #include "globals.hh" 51 #include "globals.hh" 52 #include "G4ParticleDefinition.hh" 52 #include "G4ParticleDefinition.hh" 53 #include "G4Material.hh" 53 #include "G4Material.hh" 54 #include "G4Element.hh" 54 #include "G4Element.hh" 55 #include "G4ElementVector.hh" 55 #include "G4ElementVector.hh" 56 #include "G4PhysicsLogVector.hh" 56 #include "G4PhysicsLogVector.hh" 57 #include "Randomize.hh" 57 #include "Randomize.hh" 58 #include <vector> 58 #include <vector> 59 59 60 class G4VEmModel; 60 class G4VEmModel; 61 61 62 class G4EmElementSelector 62 class G4EmElementSelector 63 { 63 { 64 64 65 public: 65 public: 66 66 67 G4EmElementSelector(G4VEmModel*, const G4Mat 67 G4EmElementSelector(G4VEmModel*, const G4Material*, G4int bins, 68 G4double emin, G4double 68 G4double emin, G4double emax, 69 G4bool spline = true); 69 G4bool spline = true); 70 70 71 ~G4EmElementSelector(); 71 ~G4EmElementSelector(); 72 72 73 void Initialise(const G4ParticleDefinition*, 73 void Initialise(const G4ParticleDefinition*, G4double cut = 0.0); 74 74 75 void Dump(const G4ParticleDefinition* p = nu 75 void Dump(const G4ParticleDefinition* p = nullptr); 76 76 77 const G4Element* SelectRandomAtom(const G4do << 78 const G4do << 79 << 80 inline const G4Element* SelectRandomAtom(G4d 77 inline const G4Element* SelectRandomAtom(G4double kineticEnergy) const; >> 78 inline const G4Element* SelectRandomAtom(const G4double kineticEnergy, >> 79 const G4double logEKin) const; 81 80 82 inline const G4Material* GetMaterial() const 81 inline const G4Material* GetMaterial() const; 83 82 >> 83 private: >> 84 84 // hide assignment operator 85 // hide assignment operator 85 G4EmElementSelector & operator=(const G4EmE 86 G4EmElementSelector & operator=(const G4EmElementSelector &right) = delete; 86 G4EmElementSelector(const G4EmElementSelect 87 G4EmElementSelector(const G4EmElementSelector&) = delete; 87 88 88 private: << 89 << 90 G4VEmModel* model; 89 G4VEmModel* model; 91 const G4Material* material; 90 const G4Material* material; 92 const G4ElementVector* theElementVector; 91 const G4ElementVector* theElementVector; 93 92 94 G4int nElmMinusOne; 93 G4int nElmMinusOne; 95 G4int nbins; 94 G4int nbins; 96 95 97 G4double cutEnergy; 96 G4double cutEnergy; 98 G4double lowEnergy; 97 G4double lowEnergy; 99 G4double highEnergy; 98 G4double highEnergy; 100 99 101 std::vector<G4PhysicsLogVector*> xSections; 100 std::vector<G4PhysicsLogVector*> xSections; 102 101 103 }; 102 }; 104 103 105 //....oooOO0OOooo........oooOO0OOooo........oo 104 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo..... 106 //....oooOO0OOooo........oooOO0OOooo........oo 105 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo..... 107 106 108 inline const G4Element* G4EmElementSelector::S 107 inline const G4Element* G4EmElementSelector::SelectRandomAtom(G4double e) const 109 { 108 { 110 const G4Element* element = (*theElementVecto 109 const G4Element* element = (*theElementVector)[nElmMinusOne]; 111 if (nElmMinusOne > 0) { 110 if (nElmMinusOne > 0) { 112 G4double x = G4UniformRand(); 111 G4double x = G4UniformRand(); 113 size_t idx(0); 112 size_t idx(0); 114 for(G4int i=0; i<nElmMinusOne; ++i) { 113 for(G4int i=0; i<nElmMinusOne; ++i) { 115 if (x <= (xSections[i])->Value(e, idx)) 114 if (x <= (xSections[i])->Value(e, idx)) { >> 115 element = (*theElementVector)[i]; >> 116 break; >> 117 } >> 118 } >> 119 } >> 120 return element; >> 121 } >> 122 >> 123 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo..... >> 124 inline const G4Element* >> 125 G4EmElementSelector::SelectRandomAtom(const G4double e,const G4double loge)const >> 126 { >> 127 const G4Element* element = (*theElementVector)[nElmMinusOne]; >> 128 if (nElmMinusOne > 0) { >> 129 // 1. Determine energy index (only once) >> 130 const size_t nBins = (xSections[0])->GetVectorLength(); >> 131 // handle cases below/above the enrgy grid (by ekin, idx that gives a=0/1) >> 132 // ekin = x[0] if e<=x[0] and idx will be 0 ^ a=0 => so y=y0 >> 133 // ekin = x[N-1] if e>=x[N-1] and idx will be N-2 ^ a=1 => so y=y_{N-1} >> 134 const G4double ekin = std::max((xSections[0])->Energy(0), >> 135 std::min((xSections[0])->Energy(nBins-1),e)); >> 136 // compute the lower index of the bin (idx \in [0,N-2] will be guaranted) >> 137 const size_t idx = (xSections[0])->ComputeLogVectorBin(loge); >> 138 // 2. Do the linear interp.(robust for corner cases through ekin, idx and a) >> 139 const G4double x1 = (xSections[0])->Energy(idx); >> 140 const G4double x2 = (xSections[0])->Energy(idx+1); >> 141 // note: all corner cases of the previous methods are covered and eventually >> 142 // gives a=0/1 that results in y=y0\y_{N-1} if e<=x[0]/e>=x[N-1] or >> 143 // y=y_i/y_{i+1} if e<x[i]/e>=x[i+1] due to small numerical errors >> 144 const G4double a = std::max(0., std::min(1., (ekin - x1)/(x2 - x1))); >> 145 const G4double urnd = G4UniformRand(); >> 146 for (G4int i = 0; i < nElmMinusOne; ++i) { >> 147 const G4double y1 = (*xSections[i])[idx]; >> 148 const G4double y2 = (*xSections[i])[idx+1]; >> 149 if (urnd <= y1 + a*(y2-y1)) { 116 element = (*theElementVector)[i]; 150 element = (*theElementVector)[i]; 117 break; 151 break; 118 } 152 } 119 } 153 } 120 } 154 } 121 return element; 155 return element; 122 } 156 } 123 157 124 //....oooOO0OOooo........oooOO0OOooo........oo 158 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo..... 125 159 126 inline const G4Material* G4EmElementSelector:: 160 inline const G4Material* G4EmElementSelector::GetMaterial() const 127 { 161 { 128 return material; 162 return material; 129 } 163 } 130 164 131 //....oooOO0OOooo........oooOO0OOooo........oo 165 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo..... 132 166 133 #endif 167 #endif 134 168 135 169