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 // $Id: HistoManager.cc,v 1.7 2008/09/24 18:53:56 maire Exp $ >> 27 // GEANT4 tag $Name: geant4-09-03-patch-02 $ 26 // 28 // 27 //....oooOO0OOooo........oooOO0OOooo........oo 29 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 28 //....oooOO0OOooo........oooOO0OOooo........oo 30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 29 31 30 #include "HistoManager.hh" 32 #include "HistoManager.hh" 31 << 33 #include "HistoMessenger.hh" 32 #include "G4UnitsTable.hh" 34 #include "G4UnitsTable.hh" 33 35 >> 36 #ifdef G4ANALYSIS_USE >> 37 #include "AIDA/AIDA.h" >> 38 #endif >> 39 34 //....oooOO0OOooo........oooOO0OOooo........oo 40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 35 41 36 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] = "testem14"; >> 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() 37 { 84 { 38 Book(); << 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 " << fileName[1] << G4endl; >> 121 #endif 39 } 122 } 40 123 41 //....oooOO0OOooo........oooOO0OOooo........oo 124 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 42 125 43 void HistoManager::Book() << 126 void HistoManager::save() 44 { 127 { 45 // Create or get analysis manager << 128 #ifdef G4ANALYSIS_USE 46 // The choice of analysis technology is done << 129 if (factoryOn) { 47 // in HistoManager.hh << 130 saveAscii(); // Write ascii file, if any 48 G4AnalysisManager* analysisManager = G4Analy << 131 tree->commit(); // Writing the histograms to the file 49 analysisManager->SetDefaultFileType("root"); << 132 tree->close(); // and closing the tree (and the file) 50 analysisManager->SetFileName(fFileName); << 133 G4cout << "\n----> Histogram Tree is saved in " << fileName[1] << G4endl; 51 analysisManager->SetVerboseLevel(1); << 134 52 analysisManager->SetActivation(true); // en << 135 delete tree; 53 << 136 tree = 0; 54 // Define histograms start values << 137 factoryOn = false; 55 const G4int kMaxHisto = 7; << 56 const G4String id[] = {"0", "1", "2", "3", " << 57 const G4String title[] = { << 58 "dummy", // 0 << 59 "scattered primary particle: energy spectr << 60 "scattered primary particle: costheta dist << 61 "charged secondaries: energy spectrum", / << 62 "charged secondaries: costheta distributio << 63 "neutral secondaries: energy spectrum", / << 64 "neutral secondaries: costheta distributio << 65 }; << 66 << 67 // Default values (to be reset via /analysis << 68 G4int nbins = 100; << 69 G4double vmin = 0.; << 70 G4double vmax = 100.; << 71 << 72 // Create all histograms as inactivated << 73 // as we have not yet set nbins, vmin, vmax << 74 for (G4int k = 0; k < kMaxHisto; k++) { << 75 G4int ih = analysisManager->CreateH1(id[k] << 76 analysisManager->SetH1Activation(ih, false << 77 } 138 } >> 139 #endif 78 } 140 } 79 141 80 //....oooOO0OOooo........oooOO0OOooo........oo 142 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 143 >> 144 void HistoManager::FillHisto(G4int ih, G4double e, G4double weight) >> 145 { >> 146 if (ih > MaxHisto) { >> 147 G4cout << "---> warning from HistoManager::FillHisto() : histo " << ih >> 148 << "does not exist; e= " << e << " w= " << weight << G4endl; >> 149 return; >> 150 } >> 151 #ifdef G4ANALYSIS_USE >> 152 if(exist[ih]) histo[ih]->fill(e/Unit[ih], weight); >> 153 #endif >> 154 } >> 155 >> 156 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 157 >> 158 void HistoManager::SetHisto(G4int ih, >> 159 G4int nbins, G4double valmin, G4double valmax, const G4String& unit) >> 160 { >> 161 if (ih > MaxHisto) { >> 162 G4cout << "---> warning from HistoManager::SetHisto() : histo " << ih >> 163 << "does not exist" << G4endl; >> 164 return; >> 165 } >> 166 >> 167 const G4String id[] = { "0", "1", "2", "3", "4", "5", "6"}; >> 168 const G4String title[] = >> 169 { "dummy", //0 >> 170 "scattered primary particle: energy spectrum", //1 >> 171 "scattered primary particle: costheta distribution", //2 >> 172 "charged secondaries: energy spectrum", //3 >> 173 "charged secondaries: costheta distribution", //4 >> 174 "neutral secondaries: energy spectrum", //5 >> 175 "neutral secondaries: costheta distribution" //6 >> 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::PrintHisto(G4int ih) >> 219 { >> 220 if (ih < MaxHisto) { ascii[ih] = true; ascii[0] = true; } >> 221 else >> 222 G4cout << "---> warning from HistoManager::PrintHisto() : histo " << ih >> 223 << "does not exist" << G4endl; >> 224 } >> 225 >> 226 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 227 >> 228 #include <fstream> >> 229 >> 230 void HistoManager::saveAscii() >> 231 { >> 232 #ifdef G4ANALYSIS_USE >> 233 >> 234 if (!ascii[0]) return; >> 235 >> 236 G4String name = fileName[0] + ".ascii"; >> 237 std::ofstream File(name, std::ios::out); >> 238 File.setf( std::ios::scientific, std::ios::floatfield ); >> 239 >> 240 //write selected histograms >> 241 for (G4int ih=0; ih<MaxHisto; ih++) { >> 242 if (exist[ih] && ascii[ih]) { >> 243 File << "\n 1D histogram " << ih << ": " << Title[ih] >> 244 << "\n \n \t X \t\t Y" << G4endl; >> 245 >> 246 for (G4int iBin=0; iBin<Nbins[ih]; iBin++) { >> 247 File << " " << iBin << "\t" >> 248 << histo[ih]->binMean(iBin) << "\t" >> 249 << histo[ih]->binHeight(iBin) >> 250 << G4endl; >> 251 } >> 252 } >> 253 } >> 254 #endif >> 255 } >> 256 >> 257 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 258 81 259