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 /// \file medical/electronScattering/src/Histo << 26 // $Id: HistoManager.cc,v 1.2 2010-11-09 20:01:12 asaim Exp $ 27 /// \brief Implementation of the HistoManager << 27 // GEANT4 tag $Name: not supported by cvs2svn $ 28 // << 29 // << 30 // 28 // 31 //....oooOO0OOooo........oooOO0OOooo........oo 29 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 32 //....oooOO0OOooo........oooOO0OOooo........oo 30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 33 31 34 #include "HistoManager.hh" 32 #include "HistoManager.hh" 35 << 33 #include "HistoMessenger.hh" 36 #include "G4UnitsTable.hh" 34 #include "G4UnitsTable.hh" 37 35 >> 36 #ifdef G4ANALYSIS_USE >> 37 #include "AIDA/AIDA.h" >> 38 #endif >> 39 38 //....oooOO0OOooo........oooOO0OOooo........oo 40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 39 41 40 HistoManager::HistoManager() : fFileName("nrcc << 42 HistoManager::HistoManager() >> 43 :af(0),tree(0),factoryOn(false) 41 { 44 { 42 Book(); << 45 #ifdef G4ANALYSIS_USE >> 46 // Creating the analysis factory >> 47 af = AIDA_createAnalysisFactory(); >> 48 if(!af) { >> 49 G4cout << " HistoManager::HistoManager() :" >> 50 << " problem creating the AIDA analysis factory." >> 51 << G4endl; >> 52 } >> 53 #endif >> 54 >> 55 fileName[0] = "nrccBenchmark"; >> 56 fileType = "root"; >> 57 fileOption = ""; >> 58 // histograms >> 59 for (G4int k=0; k<MaxHisto; k++) { >> 60 histo[k] = 0; >> 61 exist[k] = false; >> 62 Unit[k] = 1.0; >> 63 Width[k] = 1.0; >> 64 ascii[k] = false; >> 65 } >> 66 >> 67 histoMessenger = new HistoMessenger(this); 43 } 68 } 44 69 45 //....oooOO0OOooo........oooOO0OOooo........oo 70 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 46 71 47 HistoManager::~HistoManager() {} << 72 HistoManager::~HistoManager() >> 73 { >> 74 delete histoMessenger; >> 75 >> 76 #ifdef G4ANALYSIS_USE >> 77 delete af; >> 78 #endif >> 79 } 48 80 49 //....oooOO0OOooo........oooOO0OOooo........oo 81 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 50 82 51 void HistoManager::Book() << 83 void HistoManager::book() 52 { 84 { 53 // Create or get analysis manager << 85 #ifdef G4ANALYSIS_USE 54 G4AnalysisManager* analysisManager = G4Analy << 86 if(!af) return; 55 analysisManager->SetDefaultFileType("root"); << 87 56 analysisManager->SetFileName(fFileName); << 88 // Creating a tree mapped to an hbook file. 57 analysisManager->SetVerboseLevel(1); << 89 fileName[1] = fileName[0] + "." + fileType; 58 analysisManager->SetActivation(true); // en << 90 G4bool readOnly = false; >> 91 G4bool createNew = true; >> 92 AIDA::ITreeFactory* tf = af->createTreeFactory(); >> 93 tree = tf->create(fileName[1], fileType, readOnly, createNew, fileOption); >> 94 delete tf; >> 95 if(!tree) { >> 96 G4cout << "HistoManager::book() :" >> 97 << " problem creating the AIDA tree with " >> 98 << " storeName = " << fileName[1] >> 99 << " storeType = " << fileType >> 100 << " readOnly = " << readOnly >> 101 << " createNew = " << createNew >> 102 << " options = " << fileOption >> 103 << G4endl; >> 104 return; >> 105 } 59 106 60 // Define histograms start values << 107 // Creating a histogram factory, whose histograms will be handled by the tree 61 const G4int kMaxHisto = 7; << 108 AIDA::IHistogramFactory* hf = af->createHistogramFactory(*tree); 62 const G4String id[] = {"0", "1", "2", "3", " << 63 109 64 const G4String title[] = { << 110 // create selected histograms 65 "dummy", // 0 << 111 for (G4int k=0; k<MaxHisto; k++) { 66 "projected angle at exit", // 1 << 112 if (exist[k]) { 67 "dN/dS = f(r) at exit", // 2 << 113 histo[k] = hf->createHistogram1D( Label[k], Title[k], 68 "d(N/cost)/dS = f(r) at exit", // 3 << 114 Nbins[k], Vmin[k], Vmax[k]); 69 "normalized d(N/cost)/dS = f(r) at exit", << 115 factoryOn = true; 70 "scattering angle theta at exit", // 5 << 116 } 71 "measured scattering angle theta at exit" << 117 } 72 }; << 118 delete hf; >> 119 if(factoryOn) >> 120 G4cout << "\n----> Histogram Tree is opened in " << fileName[1] << G4endl; 73 121 74 // Default values (to be reset via /analysis << 122 #endif 75 G4int nbins = 100; << 123 } 76 G4double vmin = 0.; << 77 G4double vmax = 100.; << 78 124 79 // Create all histograms as inactivated << 125 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 80 // as we have not yet set nbins, vmin, vmax << 126 81 for (G4int k = 0; k < kMaxHisto; k++) { << 127 void HistoManager::save() 82 G4int ih = analysisManager->CreateH1(id[k] << 128 { 83 analysisManager->SetH1Activation(ih, false << 129 #ifdef G4ANALYSIS_USE >> 130 if (factoryOn) { >> 131 saveAscii(); // Write ascii file, if any >> 132 tree->commit(); // Writing the histograms to the file >> 133 tree->close(); // and closing the tree (and the file) >> 134 G4cout << "\n----> Histogram Tree is saved in " << fileName[1] << G4endl; >> 135 >> 136 delete tree; >> 137 tree = 0; >> 138 factoryOn = false; 84 } 139 } >> 140 #endif 85 } 141 } >> 142 >> 143 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 144 >> 145 void HistoManager::FillHisto(G4int ih, G4double e, G4double weight) >> 146 { >> 147 if (ih > MaxHisto) { >> 148 G4cout << "---> warning from HistoManager::FillHisto() : histo " << ih >> 149 << "does not exist; e= " << e << " w= " << weight << G4endl; >> 150 return; >> 151 } >> 152 #ifdef G4ANALYSIS_USE >> 153 if(exist[ih]) histo[ih]->fill(e/Unit[ih], weight); >> 154 #endif >> 155 } >> 156 >> 157 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 158 >> 159 void HistoManager::SetHisto(G4int ih, >> 160 G4int nbins, G4double valmin, G4double valmax, const G4String& unit) >> 161 { >> 162 if (ih > MaxHisto) { >> 163 G4cout << "---> warning from HistoManager::SetHisto() : histo " << ih >> 164 << "does not exist" << G4endl; >> 165 return; >> 166 } >> 167 >> 168 const G4String id[] = { "0", "1", "2", "3", "4" }; >> 169 >> 170 const G4String title[] = >> 171 { "dummy", //0 >> 172 "projected angle at exit", //1 >> 173 "dN/dS = f(r) at exit", //2 >> 174 "d(N/cost)/dS = f(r) at exit", //3 >> 175 "normalized d(N/cost)/dS = f(r) at exit" //4 >> 176 >> 177 }; >> 178 >> 179 G4String titl = title[ih]; >> 180 G4double vmin = valmin, vmax = valmax; >> 181 Unit[ih] = 1.; >> 182 >> 183 if (unit != "none") { >> 184 titl = title[ih] + " (" + unit + ")"; >> 185 Unit[ih] = G4UnitDefinition::GetValueOf(unit); >> 186 vmin = valmin/Unit[ih]; vmax = valmax/Unit[ih]; >> 187 } >> 188 >> 189 exist[ih] = true; >> 190 Label[ih] = id[ih]; >> 191 Title[ih] = titl; >> 192 Nbins[ih] = nbins; >> 193 Vmin[ih] = vmin; >> 194 Vmax[ih] = vmax; >> 195 Width[ih] = (valmax-valmin)/nbins; >> 196 >> 197 G4cout << "----> SetHisto " << ih << ": " << titl << "; " >> 198 << nbins << " bins from " >> 199 << vmin << " " << unit << " to " << vmax << " " << unit << G4endl; >> 200 >> 201 } >> 202 >> 203 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 204 >> 205 void HistoManager::RemoveHisto(G4int ih) >> 206 { >> 207 if (ih > MaxHisto) { >> 208 G4cout << "---> warning from HistoManager::RemoveHisto() : histo " << ih >> 209 << "does not exist" << G4endl; >> 210 return; >> 211 } >> 212 >> 213 histo[ih] = 0; exist[ih] = false; >> 214 } >> 215 >> 216 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 217 >> 218 void HistoManager::Scale(G4int ih, G4double fac) >> 219 { >> 220 if (ih > MaxHisto) { >> 221 G4cout << "---> warning from HistoManager::Scale() : histo " << ih >> 222 << "does not exist. (fac = " << fac << ")" << G4endl; >> 223 return; >> 224 } >> 225 #ifdef G4ANALYSIS_USE >> 226 if(exist[ih]) histo[ih]->scale(fac); >> 227 #endif >> 228 } >> 229 >> 230 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 231 >> 232 void HistoManager::PrintHisto(G4int ih) >> 233 { >> 234 if (ih < MaxHisto) { ascii[ih] = true; ascii[0] = true; } >> 235 else >> 236 G4cout << "---> warning from HistoManager::PrintHisto() : histo " << ih >> 237 << "does not exist" << G4endl; >> 238 } >> 239 >> 240 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 241 >> 242 #include <fstream> >> 243 >> 244 void HistoManager::saveAscii() >> 245 { >> 246 #ifdef G4ANALYSIS_USE >> 247 >> 248 if (!ascii[0]) return; >> 249 >> 250 G4String name = fileName[0] + ".ascii"; >> 251 std::ofstream File(name, std::ios::out); >> 252 File.setf( std::ios::scientific, std::ios::floatfield ); >> 253 >> 254 //write selected histograms >> 255 for (G4int ih=0; ih<MaxHisto; ih++) { >> 256 if (exist[ih] && ascii[ih]) { >> 257 File << "\n 1D histogram " << ih << ": " << Title[ih] >> 258 << "\n \n \t X \t\t Y" << G4endl; >> 259 >> 260 for (G4int iBin=0; iBin<Nbins[ih]; iBin++) { >> 261 File << " " << iBin << "\t" >> 262 << 0.5*(histo[ih]->axis().binLowerEdge(iBin) + >> 263 histo[ih]->axis().binUpperEdge(iBin)) << "\t" >> 264 << histo[ih]->binHeight(iBin) >> 265 << G4endl; >> 266 } >> 267 } >> 268 } >> 269 #endif >> 270 } >> 271 >> 272 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 86 273