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