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 // $Id: HistoManager.cc,v 1.5 2008/09/26 20:26:01 maire Exp $ >> 27 // GEANT4 tag $Name: geant4-09-02 $ 27 // 28 // 28 //....oooOO0OOooo........oooOO0OOooo........oo 29 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 29 //....oooOO0OOooo........oooOO0OOooo........oo 30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 30 31 31 #include "HistoManager.hh" 32 #include "HistoManager.hh" 32 << 33 #include "HistoMessenger.hh" 33 #include "G4UnitsTable.hh" 34 #include "G4UnitsTable.hh" 34 35 >> 36 #ifdef G4ANALYSIS_USE >> 37 #include "AIDA/AIDA.h" >> 38 #endif >> 39 35 //....oooOO0OOooo........oooOO0OOooo........oo 40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 36 41 37 HistoManager::HistoManager() 42 HistoManager::HistoManager() >> 43 :af(0),tree(0),factoryOn(false) >> 44 { >> 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] = "testem18"; >> 56 fileType = "root"; >> 57 fileOption = "--noErrors export=root uncompress"; >> 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); >> 68 } >> 69 >> 70 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 71 >> 72 HistoManager::~HistoManager() >> 73 { >> 74 delete histoMessenger; >> 75 >> 76 #ifdef G4ANALYSIS_USE >> 77 delete af; >> 78 #endif >> 79 } >> 80 >> 81 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 82 >> 83 void HistoManager::book() >> 84 { >> 85 #ifdef G4ANALYSIS_USE >> 86 if(!af) return; >> 87 >> 88 // Creating a tree mapped to an hbook file. >> 89 fileName[1] = fileName[0] + "." + fileType; >> 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 } >> 106 >> 107 // Creating a histogram factory, whose histograms will be handled by the tree >> 108 AIDA::IHistogramFactory* hf = af->createHistogramFactory(*tree); >> 109 >> 110 // create selected histograms >> 111 for (G4int k=0; k<MaxHisto; k++) { >> 112 if (exist[k]) { >> 113 histo[k] = hf->createHistogram1D( Label[k], Title[k], >> 114 Nbins[k], Vmin[k], Vmax[k]); >> 115 factoryOn = true; >> 116 } >> 117 } >> 118 delete hf; >> 119 if(factoryOn) >> 120 G4cout << "\n----> Histogram Tree is opened in " << fileName[1] << G4endl; >> 121 >> 122 #endif >> 123 } >> 124 >> 125 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 126 >> 127 void HistoManager::save() >> 128 { >> 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; >> 139 } >> 140 #endif >> 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) 38 { 161 { 39 Book(); << 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", "5", "6" }; >> 169 const G4String title[] = >> 170 { "dummy", //0 >> 171 "continuous energy loss along primary track", //1 >> 172 "energy from secondaries", //2 >> 173 "total energy lost by primary track", //3 >> 174 "energy spectrum of e-+", //4 >> 175 "energy spectrum of gamma", //5 >> 176 "step size" //6 >> 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 40 } 201 } 41 202 42 //....oooOO0OOooo........oooOO0OOooo........oo 203 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 43 204 44 void HistoManager::Book() << 205 void HistoManager::RemoveHisto(G4int ih) 45 { 206 { 46 // Create or get analysis manager << 207 if (ih > MaxHisto) { 47 // The choice of analysis technology is done << 208 G4cout << "---> warning from HistoManager::RemoveHisto() : histo " << ih 48 // in HistoManager.hh << 209 << "does not exist" << G4endl; 49 G4AnalysisManager* analysisManager = G4Analy << 210 return; 50 analysisManager->SetDefaultFileType("root"); << 51 analysisManager->SetFileName(fFileName); << 52 analysisManager->SetVerboseLevel(1); << 53 analysisManager->SetActivation(true); // en << 54 << 55 // Define histograms start values << 56 const G4int kMaxHisto = 14; << 57 const G4String id[] = {"0", "1", "2", "3", " << 58 const G4String title[] = { << 59 "dummy", // 0 << 60 "step size of primary track", // 1 << 61 "energy continuously deposited along prima << 62 "energy transfered to secondaries by ionis << 63 "energy transfered to secondaries by Brems << 64 "energy transfered to secondaries by (e+,e << 65 "total energy transfered to secondaries", << 66 "total energy lost by primary track", // << 67 "total energy lost by primary track from e << 68 "energy continuously deposited along secon << 69 "total energy deposited", // 10 << 70 "energy spectrum of gamma", // 11 << 71 "energy spectrum of e-", // 12 << 72 "energy spectrum of e+" // 13 << 73 }; << 74 << 75 // Default values (to be reset via /analysis << 76 G4int nbins = 100; << 77 G4double vmin = 0.; << 78 G4double vmax = 100.; << 79 << 80 // Create all histograms as inactivated << 81 // as we have not yet set nbins, vmin, vmax << 82 for (G4int k = 0; k < kMaxHisto; k++) { << 83 G4int ih = analysisManager->CreateH1(id[k] << 84 analysisManager->SetH1Activation(ih, false << 85 } 211 } >> 212 >> 213 histo[ih] = 0; exist[ih] = false; >> 214 } >> 215 >> 216 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 217 void HistoManager::PrintHisto(G4int ih) >> 218 { >> 219 if (ih < MaxHisto) { ascii[ih] = true; ascii[0] = true; } >> 220 else >> 221 G4cout << "---> warning from HistoManager::PrintHisto() : histo " << ih >> 222 << "does not exist" << G4endl; >> 223 } >> 224 >> 225 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 226 >> 227 #include <fstream> >> 228 >> 229 void HistoManager::saveAscii() >> 230 { >> 231 #ifdef G4ANALYSIS_USE >> 232 >> 233 if (!ascii[0]) return; >> 234 >> 235 G4String name = fileName[0] + ".ascii"; >> 236 std::ofstream File(name, std::ios::out); >> 237 File.setf( std::ios::scientific, std::ios::floatfield ); >> 238 >> 239 //write selected histograms >> 240 for (G4int ih=0; ih<MaxHisto; ih++) { >> 241 if (exist[ih] && ascii[ih]) { >> 242 File << "\n 1D histogram " << ih << ": " << Title[ih] >> 243 << "\n \n \t X \t\t Y" << G4endl; >> 244 >> 245 for (G4int iBin=0; iBin<Nbins[ih]; iBin++) { >> 246 File << " " << iBin << "\t" >> 247 << 0.5*(histo[ih]->axis().binLowerEdge(iBin) + >> 248 histo[ih]->axis().binUpperEdge(iBin)) << "\t" >> 249 << histo[ih]->binHeight(iBin) >> 250 << G4endl; >> 251 } >> 252 } >> 253 } >> 254 #endif 86 } 255 } 87 256 88 //....oooOO0OOooo........oooOO0OOooo........oo 257 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 258 >> 259 89 260