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.9 2010-11-09 21:06:39 asaim Exp $ >> 27 // GEANT4 tag $Name: geant4-09-04-patch-01 $ 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 >> 40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 41 >> 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] = "testem15"; >> 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); >> 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 " << fileName[1] << G4endl; >> 121 #endif >> 122 } >> 123 >> 124 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 125 >> 126 void HistoManager::save() >> 127 { >> 128 #ifdef G4ANALYSIS_USE >> 129 if (factoryOn) { >> 130 saveAscii(); // Write ascii file, if any >> 131 tree->commit(); // Writing the histograms to the file >> 132 tree->close(); // and closing the tree (and the file) >> 133 G4cout << "\n----> Histogram Tree is saved in " << fileName[1] << G4endl; >> 134 >> 135 delete tree; >> 136 tree = 0; >> 137 factoryOn = false; >> 138 } >> 139 #endif >> 140 } >> 141 35 //....oooOO0OOooo........oooOO0OOooo........oo 142 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 36 143 37 HistoManager::HistoManager() : fFileName("test << 144 void HistoManager::FillHisto(G4int ih, G4double e, G4double weight) 38 { 145 { 39 Book(); << 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 40 } 154 } 41 155 42 //....oooOO0OOooo........oooOO0OOooo........oo 156 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 43 157 44 HistoManager::~HistoManager() {} << 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", "7", "8", "9"}; >> 168 const G4String title[] = >> 169 { "dummy", //0 >> 170 "Multiple Scattering. True step length", //1 >> 171 "Multiple Scattering. Geom step length", //2 >> 172 "Multiple Scattering. Ratio geomSl/trueSl", //3 >> 173 "Multiple Scattering. Lateral displacement: radius", //4 >> 174 "Multiple Scattering. Lateral displac: psi_space", //5 >> 175 "Multiple Scattering. Angular distrib: theta_plane", //6 >> 176 "Multiple Scattering. Phi-position angle", //7 >> 177 "Multiple Scattering. Phi-direction angle", //8 >> 178 "Multiple Scattering. Correlation: std::cos(phiPos-phiDir)"//9 >> 179 }; >> 180 >> 181 >> 182 G4String titl = title[ih]; >> 183 G4double vmin = valmin, vmax = valmax; >> 184 Unit[ih] = 1.; >> 185 >> 186 if (unit != "none") { >> 187 titl = title[ih] + " (" + unit + ")"; >> 188 Unit[ih] = G4UnitDefinition::GetValueOf(unit); >> 189 vmin = valmin/Unit[ih]; vmax = valmax/Unit[ih]; >> 190 } >> 191 >> 192 exist[ih] = true; >> 193 Label[ih] = id[ih]; >> 194 Title[ih] = titl; >> 195 Nbins[ih] = nbins; >> 196 Vmin[ih] = vmin; >> 197 Vmax[ih] = vmax; >> 198 Width[ih] = (valmax-valmin)/nbins; >> 199 >> 200 G4cout << "----> SetHisto " << ih << ": " << titl << "; " >> 201 << nbins << " bins from " >> 202 << vmin << " " << unit << " to " << vmax << " " << unit << G4endl; >> 203 >> 204 } 45 205 46 //....oooOO0OOooo........oooOO0OOooo........oo 206 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 47 207 48 void HistoManager::Book() << 208 void HistoManager::RemoveHisto(G4int ih) 49 { << 209 { 50 // Create or get analysis manager << 210 if (ih > MaxHisto) { 51 // The choice of analysis technology is done << 211 G4cout << "---> warning from HistoManager::RemoveHisto() : histo " << ih 52 // in HistoManager.hh << 212 << "does not exist" << G4endl; 53 G4AnalysisManager* analysisManager = G4Analy << 213 return; 54 analysisManager->SetDefaultFileType("root"); << 55 analysisManager->SetFileName(fFileName); << 56 analysisManager->SetVerboseLevel(1); << 57 analysisManager->SetActivation(true); // en << 58 << 59 // Define histograms start values << 60 const G4int kMaxHisto = 17; << 61 const G4String id[] = {"0", "1", "2", "3", << 62 "9", "10", "11", "12" << 63 const G4String title[] = { << 64 "dummy", // 0 << 65 "Multiple Scattering. True step length", << 66 "Multiple Scattering. Geom step length", << 67 "Multiple Scattering. Ratio geomSl/trueSl" << 68 "Multiple Scattering. Lateral displacement << 69 "Multiple Scattering. Lateral displac: psi << 70 "Multiple Scattering. Angular distrib: the << 71 "Multiple Scattering. Phi-position angle", << 72 "Multiple Scattering. Phi-direction angle" << 73 "Multiple Scattering. Correlation: std::co << 74 "Gamma Conversion. Open Angle * Egamma", << 75 "Gamma Conversion. Log10(P recoil)", // 1 << 76 "Gamma Conversion. Phi P recoil angle", / << 77 "Gamma Conversion. Phi P plus angle", // << 78 "Gamma Conversion. 2 * cos(phiplus + phimi << 79 "Gamma Conversion. E plus / E gamma", // << 80 "Gamma Conversion. Phi of Gamma Polarizati << 81 }; << 82 << 83 // Default values (to be reset via /analysis << 84 G4int nbins = 100; << 85 G4double vmin = 0.; << 86 G4double vmax = 100.; << 87 << 88 // Create all histograms as inactivated << 89 // as we have not yet set nbins, vmin, vmax << 90 for (G4int k = 0; k < kMaxHisto; k++) { << 91 G4int ih = analysisManager->CreateH1(id[k] << 92 analysisManager->SetH1Activation(ih, false << 93 } 214 } >> 215 >> 216 histo[ih] = 0; exist[ih] = false; >> 217 } >> 218 >> 219 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 220 >> 221 void HistoManager::PrintHisto(G4int ih) >> 222 { >> 223 if (ih < MaxHisto) { ascii[ih] = true; ascii[0] = true; } >> 224 else >> 225 G4cout << "---> warning from HistoManager::PrintHisto() : histo " << ih >> 226 << "does not exist" << G4endl; 94 } 227 } 95 228 96 //....oooOO0OOooo........oooOO0OOooo........oo 229 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 230 >> 231 #include <fstream> >> 232 >> 233 void HistoManager::saveAscii() >> 234 { >> 235 #ifdef G4ANALYSIS_USE >> 236 >> 237 if (!ascii[0]) return; >> 238 >> 239 G4String name = fileName[0] + ".ascii"; >> 240 std::ofstream File(name, std::ios::out); >> 241 File.setf( std::ios::scientific, std::ios::floatfield ); >> 242 >> 243 //write selected histograms >> 244 for (G4int ih=0; ih<MaxHisto; ih++) { >> 245 if (exist[ih] && ascii[ih]) { >> 246 File << "\n 1D histogram " << ih << ": " << Title[ih] >> 247 << "\n \n \t X \t\t Y" << G4endl; >> 248 >> 249 for (G4int iBin=0; iBin<Nbins[ih]; iBin++) { >> 250 File << " " << iBin << "\t" >> 251 << 0.5*(histo[ih]->axis().binLowerEdge(iBin) + >> 252 histo[ih]->axis().binUpperEdge(iBin)) << "\t" >> 253 << histo[ih]->binHeight(iBin) >> 254 << G4endl; >> 255 } >> 256 } >> 257 } >> 258 #endif >> 259 } >> 260 >> 261 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 262 >> 263 97 264