| 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 // Author: Maria Grazia Pia (Maria.Grazia.Pia@ 28 // Author: Maria Grazia Pia (Maria.Grazia.Pia@cern.ch)
29 // 29 //
30 // History: 30 // History:
31 // ----------- 31 // -----------
32 // 1 Aug 2001 MGP Created 32 // 1 Aug 2001 MGP Created
33 // 31 Jul 2008 MGP Revised and rename 33 // 31 Jul 2008 MGP Revised and renamed to G4CompositeDataSet
34 // 34 //
35 // 35 //
36 // ------------------------------------------- 36 // -------------------------------------------------------------------
37 37
38 #include "G4CompositeDataSet.hh" 38 #include "G4CompositeDataSet.hh"
39 #include "G4DataSet.hh" 39 #include "G4DataSet.hh"
40 #include "G4IInterpolator.hh" 40 #include "G4IInterpolator.hh"
41 #include <fstream> 41 #include <fstream>
42 #include <sstream> 42 #include <sstream>
43 43
44 G4CompositeDataSet::G4CompositeDataSet(G4IInte 44 G4CompositeDataSet::G4CompositeDataSet(G4IInterpolator* algo,
45 G4double eUnit, 45 G4double eUnit,
46 G4double dataUnit, 46 G4double dataUnit,
47 G4int zMin, 47 G4int zMin,
48 G4int zMax) 48 G4int zMax)
49 : 49 :
50 algorithm(algo), 50 algorithm(algo),
51 unitEnergies(eUnit), 51 unitEnergies(eUnit),
52 unitData(dataUnit), 52 unitData(dataUnit),
53 minZ(zMin), 53 minZ(zMin),
54 maxZ(zMax) 54 maxZ(zMax)
55 { 55 {
56 if (algorithm == 0) 56 if (algorithm == 0)
57 G4Exception("G4CompositeDataSet::G4Composi 57 G4Exception("G4CompositeDataSet::G4CompositeDataSet",
58 "pii00000001", 58 "pii00000001",
59 FatalException, 59 FatalException,
60 "Interpolation == 0"); 60 "Interpolation == 0");
61 } 61 }
62 62
63 63
64 64
65 G4CompositeDataSet::~G4CompositeDataSet() 65 G4CompositeDataSet::~G4CompositeDataSet()
66 { 66 {
67 CleanUpComponents(); 67 CleanUpComponents();
68 if (algorithm) delete algorithm; 68 if (algorithm) delete algorithm;
69 } 69 }
70 70
71 71
72 G4double G4CompositeDataSet::FindValue(G4doubl 72 G4double G4CompositeDataSet::FindValue(G4double energy, G4int componentId) const
73 { 73 {
74 const G4IDataSet* component(GetComponent(com 74 const G4IDataSet* component(GetComponent(componentId));
75 75
76 if (component) return component->FindValue(e 76 if (component) return component->FindValue(energy);
77 77
78 std::ostringstream message; 78 std::ostringstream message;
79 message << "G4CompositeDataSet::FindValue - 79 message << "G4CompositeDataSet::FindValue - component " << componentId << " not found";
80 80
81 G4Exception("G4CompositeDataSet::FindValue" 81 G4Exception("G4CompositeDataSet::FindValue",
82 "pii00000010", 82 "pii00000010",
83 FatalException, 83 FatalException,
84 message.str().c_str()); 84 message.str().c_str());
85 85
86 return 0.; 86 return 0.;
87 } 87 }
88 88
89 void G4CompositeDataSet::PrintData(void) const 89 void G4CompositeDataSet::PrintData(void) const
90 { 90 {
91 const G4int n = (G4int)NumberOfComponents(); << 91 const size_t n(NumberOfComponents());
92 92
93 G4cout << "The data set has " << n << " comp 93 G4cout << "The data set has " << n << " components" << G4endl;
94 G4cout << G4endl; 94 G4cout << G4endl;
95 95
96 G4int i(0); << 96 size_t i(0);
97 97
98 while (i<n) 98 while (i<n)
99 { 99 {
100 G4cout << "--- Component " << i << " --- 100 G4cout << "--- Component " << i << " ---" << G4endl;
101 GetComponent(i)->PrintData(); 101 GetComponent(i)->PrintData();
102 ++i; << 102 i++;
103 } 103 }
104 } 104 }
105 105
106 void G4CompositeDataSet::SetEnergiesData(G4Dat 106 void G4CompositeDataSet::SetEnergiesData(G4DataVector* energies, G4DataVector* data, G4int componentId)
107 { 107 {
108 G4IDataSet * component(components[componentI 108 G4IDataSet * component(components[componentId]);
109 109
110 if (component) 110 if (component)
111 { 111 {
112 component->SetEnergiesData(energies, dat 112 component->SetEnergiesData(energies, data, 0);
113 return; 113 return;
114 } 114 }
115 115
116 std::ostringstream message; 116 std::ostringstream message;
117 message << "G4CompositeDataSet::SetEnergiesD 117 message << "G4CompositeDataSet::SetEnergiesData - component " << componentId << " not found";
118 118
119 G4Exception("G4CompositeDataSet::SetEnergies 119 G4Exception("G4CompositeDataSet::SetEnergiesData",
120 "pii00000020", 120 "pii00000020",
121 FatalException, 121 FatalException,
122 message.str().c_str()); 122 message.str().c_str());
123 123
124 } 124 }
125 125
126 G4bool G4CompositeDataSet::LoadData(const G4St 126 G4bool G4CompositeDataSet::LoadData(const G4String& argFileName)
127 { 127 {
128 CleanUpComponents(); 128 CleanUpComponents();
129 129
130 for (G4int z(minZ); z<maxZ; ++z) << 130 for (G4int z(minZ); z<maxZ; z++)
131 { 131 {
132 G4IDataSet* component = new G4DataSet(z, 132 G4IDataSet* component = new G4DataSet(z, algorithm->Clone(), unitEnergies, unitData);
133 if (!component->LoadData(argFileName)) 133 if (!component->LoadData(argFileName))
134 { 134 {
135 delete component; 135 delete component;
136 return false; 136 return false;
137 } 137 }
138 AddComponent(component); 138 AddComponent(component);
139 } 139 }
140 return true; 140 return true;
141 } 141 }
142 142
143 143
144 144
145 G4bool G4CompositeDataSet::SaveData(const G4St 145 G4bool G4CompositeDataSet::SaveData(const G4String& argFileName) const
146 { 146 {
147 for (G4int z=minZ; z<maxZ; ++z) << 147 for (G4int z=minZ; z<maxZ; z++)
148 { 148 {
149 const G4IDataSet* component(GetComponent 149 const G4IDataSet* component(GetComponent(z-minZ));
150 150
151 if (!component) 151 if (!component)
152 { 152 {
153 std::ostringstream message; 153 std::ostringstream message;
154 message << "G4CompositeDataSet::SaveData - 154 message << "G4CompositeDataSet::SaveData - component " << (z-minZ) << " not found";
155 G4Exception("G4CompositeDataSet::SaveData" 155 G4Exception("G4CompositeDataSet::SaveData",
156 "pii00000030", 156 "pii00000030",
157 FatalException, 157 FatalException,
158 message.str().c_str()); 158 message.str().c_str());
159 } 159 }
160 160
161 if (!component->SaveData(argFileName)) 161 if (!component->SaveData(argFileName))
162 return false; 162 return false;
163 } 163 }
164 164
165 return true; 165 return true;
166 } 166 }
167 167
168 void G4CompositeDataSet::CleanUpComponents(voi 168 void G4CompositeDataSet::CleanUpComponents(void)
169 { 169 {
170 while (!components.empty()) 170 while (!components.empty())
171 { 171 {
172 if (components.back()) 172 if (components.back())
173 delete components.back(); 173 delete components.back();
174 components.pop_back(); 174 components.pop_back();
175 } 175 }
176 } 176 }
177 177
178 178
179 G4double G4CompositeDataSet::RandomSelect(G4in 179 G4double G4CompositeDataSet::RandomSelect(G4int componentId) const
180 { 180 {
181 G4double value = 0.; 181 G4double value = 0.;
182 if (componentId >= 0 && componentId < (G4int 182 if (componentId >= 0 && componentId < (G4int)components.size())
183 { 183 {
184 const G4IDataSet* dataSet = GetComponent 184 const G4IDataSet* dataSet = GetComponent(componentId);
185 value = dataSet->RandomSelect(); 185 value = dataSet->RandomSelect();
186 } 186 }
187 return value; 187 return value;
188 } 188 }
189 189