Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer << 3 // * DISCLAIMER * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th << 5 // * The following disclaimer summarizes all the specific disclaimers * 6 // * the Geant4 Collaboration. It is provided << 6 // * of contributors to this software. The specific disclaimers,which * 7 // * conditions of the Geant4 Software License << 7 // * govern, are listed with their locations in: * 8 // * LICENSE and available at http://cern.ch/ << 8 // * http://cern.ch/geant4/license * 9 // * include a list of copyright holders. << 10 // * 9 // * * 11 // * Neither the authors of this software syst 10 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 11 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 12 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 13 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file << 14 // * use. * 16 // * for the full disclaimer and the limitatio << 17 // * 15 // * * 18 // * This code implementation is the result << 16 // * This code implementation is the intellectual property of the * 19 // * technical work of the GEANT4 collaboratio << 17 // * GEANT4 collaboration. * 20 // * By using, copying, modifying or distri << 18 // * By copying, distributing or modifying the Program (or any work * 21 // * any work based on the software) you ag << 19 // * based on the Program) you indicate your acceptance of this * 22 // * use in resulting scientific publicati << 20 // * statement, and all its terms. * 23 // * acceptance of all terms of the Geant4 Sof << 24 // ******************************************* 21 // ******************************************************************** 25 // 22 // 26 /// \file electromagnetic/TestEm9/src/Histo.cc << 27 /// \brief Implementation of the Histo class << 28 // << 29 // << 30 //-------------------------------------------- 23 //--------------------------------------------------------------------------- 31 // 24 // 32 // ClassName: Histo - Generic histogram/ntup 25 // ClassName: Histo - Generic histogram/ntuple manager class 33 // 26 // 34 // 27 // 35 // Author: V.Ivanchenko 30.10.03 28 // Author: V.Ivanchenko 30.10.03 36 // 29 // 37 //-------------------------------------------- 30 //---------------------------------------------------------------------------- 38 // 31 // 39 32 40 //....oooOO0OOooo........oooOO0OOooo........oo << 33 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 41 //....oooOO0OOooo........oooOO0OOooo........oo << 34 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 42 35 43 #include "Histo.hh" 36 #include "Histo.hh" 44 37 >> 38 #ifdef G4ANALYSIS_USE >> 39 >> 40 #include <memory> // for the auto_ptr(T> >> 41 #include "AIDA/AIDA.h" 45 #include "HistoMessenger.hh" 42 #include "HistoMessenger.hh" 46 43 47 #include "G4RootAnalysisManager.hh" << 44 #endif 48 45 49 //....oooOO0OOooo........oooOO0OOooo........oo << 46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 50 //....oooOO0OOooo........oooOO0OOooo........oo << 47 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 51 48 52 Histo::Histo() : fManager(0), fMessenger(0) << 49 Histo::Histo() 53 { 50 { 54 fMessenger = new HistoMessenger(this); << 51 verbose = 0; 55 << 52 histName = "histo.paw"; 56 fHistName = "test"; << 53 histType = "hbook"; 57 fHistType = "root"; << 54 nHisto = 0; 58 fTupleName = "tuple"; << 55 defaultAct = 1; 59 fTupleTitle = "test"; << 56 tupleName = "tuple.paw"; 60 fNHisto = 0; << 57 tupleId = "100"; 61 fVerbose = 0; << 58 tupleList = ""; 62 fDefaultAct = true; << 59 ntup = 0; 63 fHistoActive = false; << 60 messenger = 0; 64 fNtupleActive = false; << 61 >> 62 #ifdef G4ANALYSIS_USE >> 63 messenger = new HistoMessenger(this); >> 64 #endif 65 } 65 } 66 66 67 //....oooOO0OOooo........oooOO0OOooo........oo << 67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 68 68 69 Histo::~Histo() 69 Histo::~Histo() 70 { 70 { 71 delete fMessenger; << 71 #ifdef G4ANALYSIS_USE >> 72 for(G4int i=0; i<nHisto; i++) { >> 73 if(histo[i]) delete histo[i]; >> 74 } >> 75 delete messenger; >> 76 #endif 72 } 77 } 73 78 74 //....oooOO0OOooo........oooOO0OOooo........oo << 79 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 75 80 76 void Histo::Book() << 81 void Histo::book() 77 { 82 { 78 if (!(fHistoActive || fNtupleActive)) { << 83 #ifdef G4ANALYSIS_USE 79 return; << 84 G4cout << "### Histo books " << nHisto << " histograms " << G4endl; 80 } << 85 // Creating the analysis factory 81 << 86 std::auto_ptr< AIDA::IAnalysisFactory > af( AIDA_createAnalysisFactory() ); 82 // Always creating analysis manager << 87 // Creating the tree factory 83 fManager = G4RootAnalysisManager::Instance() << 88 std::auto_ptr< AIDA::ITreeFactory > tf( af->createTreeFactory() ); 84 89 85 // Creating a tree mapped to a new hbook fil 90 // Creating a tree mapped to a new hbook file. 86 G4String nam = fHistName + "." + fHistType; << 87 91 88 // Open file histogram file << 92 tree = tf->create(histName,histType,false,true,"uncompress"); 89 if (!fManager->OpenFile(nam)) { << 93 if(tree) 90 G4cout << "Histo::Book: ERROR open file <" << 94 G4cout << "Tree store : " << tree->storeName() << G4endl; 91 fHistoActive = false; << 95 else 92 fNtupleActive = false; << 96 G4cout << "ERROR: Tree store " << histName << " is not created!" << G4endl; 93 return; << 97 94 } << 98 // Creating a histogram factory, whose histograms will be handled by the tree 95 G4cout << "### Histo::Save: Opended file <" << 99 std::auto_ptr< AIDA::IHistogramFactory > hf(af->createHistogramFactory( *tree )); 96 << G4endl; << 97 100 98 // Creating an 1-dimensional histograms in t 101 // Creating an 1-dimensional histograms in the root directory of the tree 99 for (G4int i = 0; i < fNHisto; ++i) { << 102 for(G4int i=0; i<nHisto; i++) { 100 if (fActive[i]) { << 103 if(active[i]) { 101 G4String ss = "h" + fIds[i]; << 104 102 fHisto[i] = fManager->CreateH1(ss, fTitl << 105 histo[i] = hf->createHistogram1D(ids[i], titles[i], bins[i], xmin[i], xmax[i]); 103 if (fVerbose > 0) { << 104 G4cout << "Created histogram #" << i < << 105 << fTitles[i] << G4endl; << 106 } << 107 } 106 } 108 } 107 } 109 // Creating a tuple factory, whose tuples wi 108 // Creating a tuple factory, whose tuples will be handled by the tree 110 if (fNtupleActive) { << 109 if(tupleList != "") { 111 fManager->CreateNtuple(fTupleName, fTupleT << 110 std::auto_ptr< AIDA::ITupleFactory > tpf( af->createTupleFactory( *tree ) ); 112 G4int i; << 111 ntup = tpf->create(tupleId, tupleName, tupleList); 113 G4int n = fNtupleI.size(); << 114 for (i = 0; i < n; ++i) { << 115 if (fTupleI[i] == -1) { << 116 fTupleI[i] = fManager->CreateNtupleICo << 117 } << 118 } << 119 n = fNtupleF.size(); << 120 for (i = 0; i < n; ++i) { << 121 if (fTupleF[i] == -1) { << 122 fTupleF[i] = fManager->CreateNtupleFCo << 123 } << 124 } << 125 n = fNtupleD.size(); << 126 for (i = 0; i < n; ++i) { << 127 if (fTupleD[i] == -1) { << 128 fTupleD[i] = fManager->CreateNtupleDCo << 129 } << 130 } << 131 } 112 } 132 } << 113 #endif >> 114 } 133 115 134 //....oooOO0OOooo........oooOO0OOooo........oo << 116 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 135 117 136 void Histo::Save() << 118 void Histo::save() 137 { 119 { 138 if (!(fHistoActive || fNtupleActive)) { << 120 #ifdef G4ANALYSIS_USE 139 return; << 140 } << 141 << 142 // Creating a tree mapped to a new hbook fil << 143 G4String nam = fHistName + "." + fHistType; << 144 << 145 // Write histogram file 121 // Write histogram file 146 if (!fManager->Write()) { << 122 tree->commit(); 147 G4Exception("Histo::Save()", "hist01", Fat << 123 G4cout << "Closing the tree..." << G4endl; 148 } << 124 tree->close(); 149 if (fVerbose > 0) { << 125 G4cout << "Histograms and Ntuples are saved" << G4endl; 150 G4cout << "### Histo::Save: Histograms and << 126 #endif 151 } << 127 } 152 if (fManager->CloseFile() && fVerbose > 0) { << 128 153 G4cout << " File is closed << 129 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... >> 130 >> 131 void Histo::add1D(const G4String& id, const G4String& name, G4int nb, >> 132 G4double x1, G4double x2, G4double u) >> 133 { >> 134 if(verbose > 0) { >> 135 G4cout << "New histogram will be booked: #" << id << " <" << name >> 136 << " " << nb << " " << x1 << " " << x2 << " " << u >> 137 << G4endl; 154 } 138 } 155 fManager->Clear(); << 139 nHisto++; 156 } << 157 << 158 //....oooOO0OOooo........oooOO0OOooo........oo << 159 << 160 void Histo::Add1D(const G4String& id, const G4 << 161 G4double u) << 162 { << 163 if (fVerbose > 0) { << 164 G4cout << "Histo::Add1D: New histogram wil << 165 << " " << x1 << " " << x2 << " " << 166 } << 167 ++fNHisto; << 168 x1 /= u; 140 x1 /= u; 169 x2 /= u; 141 x2 /= u; 170 fActive.push_back(fDefaultAct); << 142 active.push_back(defaultAct); 171 fBins.push_back(nb); << 143 bins.push_back(nb); 172 fXmin.push_back(x1); << 144 xmin.push_back(x1); 173 fXmax.push_back(x2); << 145 xmax.push_back(x2); 174 fUnit.push_back(u); << 146 unit.push_back(u); 175 fIds.push_back(id); << 147 ids.push_back(id); 176 fTitles.push_back(name); << 148 titles.push_back(name); 177 fHisto.push_back(-1); << 149 histo.push_back(0); 178 } << 150 } 179 << 151 180 //....oooOO0OOooo........oooOO0OOooo........oo << 152 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 181 << 153 182 void Histo::SetHisto1D(G4int i, G4int nb, G4do << 154 void Histo::setHisto1D(G4int i, G4int nb, G4double x1, G4double x2, G4double u) 183 { << 155 { 184 if (i >= 0 && i < fNHisto) { << 156 if(i>=0 && i<nHisto) { 185 if (fVerbose > 0) { << 157 if(verbose > 0) { 186 G4cout << "Histo::SetHisto1D: #" << i << << 158 G4cout << "Update histogram: #" << i >> 159 << " " << nb << " " << x1 << " " << x2 << " " << u 187 << G4endl; 160 << G4endl; 188 } 161 } 189 fBins[i] = nb; << 162 bins[i] = nb; 190 fXmin[i] = x1; << 163 xmin[i] = x1; 191 fXmax[i] = x2; << 164 xmax[i] = x2; 192 fUnit[i] = u; << 165 unit[i] = u; 193 fActive[i] = true; << 166 } else { 194 fHistoActive = true; << 167 G4cout << "Histo::setHisto1D: WARNING! wrong histogram index " << i << G4endl; 195 } << 196 else { << 197 G4cout << "Histo::SetHisto1D: WARNING! wro << 198 } 168 } 199 } 169 } 200 170 201 //....oooOO0OOooo........oooOO0OOooo........oo << 171 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 202 172 203 void Histo::Activate(G4int i, G4bool val) << 173 void Histo::fill(G4int i, G4double x, G4double w) 204 { 174 { 205 if (fVerbose > 1) { << 175 if(verbose > 1) { 206 G4cout << "Histo::Activate: Histogram: #" << 176 G4cout << "fill histogram: #" << i << " at x= " << x 207 } << 177 << " weight= " << w 208 if (i >= 0 && i < fNHisto) { << 178 << G4endl; 209 fActive[i] = val; << 210 if (val) { << 211 fHistoActive = true; << 212 } << 213 } << 214 } << 215 << 216 //....oooOO0OOooo........oooOO0OOooo........oo << 217 << 218 void Histo::Fill(G4int i, G4double x, G4double << 219 { << 220 if (!fHistoActive) { << 221 return; << 222 } << 223 if (fVerbose > 1) { << 224 G4cout << "Histo::Fill: Histogram: #" << i << 225 } << 226 if (i >= 0 && i < fNHisto) { << 227 if (fActive[i]) { << 228 fManager->FillH1(fHisto[i], x / fUnit[i] << 229 } << 230 } 179 } 231 else { << 180 #ifdef G4ANALYSIS_USE 232 G4cout << "Histo::Fill: WARNING! wrong his << 181 if(i>=0 && i<nHisto) { >> 182 histo[i]->fill((float)(x/unit[i]), (float)w); >> 183 } else { >> 184 G4cout << "Histo::fill: WARNING! wrong histogram index " << i << G4endl; 233 } 185 } >> 186 #endif 234 } 187 } 235 188 236 //....oooOO0OOooo........oooOO0OOooo........oo << 189 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 237 190 238 void Histo::ScaleH1(G4int i, G4double x) << 191 void Histo::scale(G4int i, G4double x) 239 { 192 { 240 if (!fHistoActive) { << 193 if(verbose > 0) { 241 return; << 194 G4cout << "Scale histogram: #" << i << " by factor " << x << G4endl; 242 } << 243 if (fVerbose > 0) { << 244 G4cout << "Histo::Scale: Histogram: #" << << 245 } 195 } 246 if (i >= 0 && i < fNHisto) { << 196 #ifdef G4ANALYSIS_USE 247 if (fActive[i]) { << 197 if(i>=0 && i<nHisto) { 248 fManager->GetH1(fHisto[i])->scale(x); << 198 histo[i]->scale(x); 249 } << 199 } else { >> 200 G4cout << "Histo::scale: WARNING! wrong histogram index " << i << G4endl; 250 } 201 } 251 else { << 202 #endif 252 G4cout << "Histo::Scale: WARNING! wrong hi << 253 } << 254 } << 255 << 256 //....oooOO0OOooo........oooOO0OOooo........oo << 257 << 258 void Histo::AddTuple(const G4String& w1) << 259 { << 260 fTupleTitle = w1; << 261 } << 262 << 263 //....oooOO0OOooo........oooOO0OOooo........oo << 264 << 265 void Histo::AddTupleI(const G4String& w1) << 266 { << 267 fNtupleActive = true; << 268 fNtupleI.push_back(w1); << 269 fTupleI.push_back(-1); << 270 } << 271 << 272 //....oooOO0OOooo........oooOO0OOooo........oo << 273 << 274 void Histo::AddTupleF(const G4String& w1) << 275 { << 276 fNtupleActive = true; << 277 fNtupleF.push_back(w1); << 278 fTupleF.push_back(-1); << 279 } 203 } 280 204 281 //....oooOO0OOooo........oooOO0OOooo........oo << 205 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 282 206 283 void Histo::AddTupleD(const G4String& w1) << 207 void Histo::addTuple(const G4String& w1, const G4String& w2, const G4String& w3) 284 { 208 { 285 fNtupleActive = true; << 209 tupleId = w1; 286 fNtupleD.push_back(w1); << 210 tupleName = w2; 287 fTupleD.push_back(-1); << 211 tupleList = w3; 288 } 212 } 289 213 290 //....oooOO0OOooo........oooOO0OOooo........oo << 214 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 291 215 292 void Histo::FillTupleI(G4int i, G4int x) << 216 void Histo::fillTuple(const G4String& parname, G4double x) 293 { 217 { 294 if (!fNtupleActive) { << 218 if(verbose > 1) { 295 return; << 219 G4cout << "fill tuple by parameter <" << parname << "> = " << x << G4endl; 296 } << 297 G4int n = fNtupleI.size(); << 298 if (i >= 0 && i < n) { << 299 if (fVerbose > 1) { << 300 G4cout << "Histo::FillTupleI: i= " << i << 301 << "> = " << x << G4endl; << 302 } << 303 fManager->FillNtupleIColumn(fTupleI[i], x) << 304 } << 305 else { << 306 G4cout << "Histo::FillTupleI: WARNING! wro << 307 } 220 } >> 221 #ifdef G4ANALYSIS_USE >> 222 if(ntup) ntup->fill(ntup->findColumn(parname), (float)x); >> 223 #endif 308 } 224 } 309 225 310 //....oooOO0OOooo........oooOO0OOooo........oo << 226 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 311 227 312 void Histo::FillTupleF(G4int i, G4float x) << 228 void Histo::addRow() 313 { 229 { 314 if (!fNtupleActive) { << 230 #ifdef G4ANALYSIS_USE 315 return; << 231 if(ntup) ntup->addRow(); 316 } << 232 #endif 317 G4int n = fNtupleF.size(); << 233 } 318 if (i >= 0 && i < n) { << 319 if (fVerbose > 1) { << 320 G4cout << "Histo::FillTupleF: i= " << i << 321 << "> = " << x << G4endl; << 322 } << 323 fManager->FillNtupleFColumn(fTupleF[i], x) << 324 } << 325 else { << 326 G4cout << "Histo::FillTupleF: WARNING! wro << 327 } << 328 } << 329 234 330 //....oooOO0OOooo........oooOO0OOooo........oo << 235 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 331 236 332 void Histo::FillTupleD(G4int i, G4double x) << 237 void Histo::setFileName(const G4String& nam) 333 { 238 { 334 if (!fNtupleActive) { << 239 histName = nam; 335 return; << 336 } << 337 G4int n = fNtupleD.size(); << 338 if (i >= 0 && i < n) { << 339 if (fVerbose > 1) { << 340 G4cout << "Histo::FillTupleD: i= " << i << 341 << "> = " << x << G4endl; << 342 } << 343 fManager->FillNtupleDColumn(fTupleD[i], x) << 344 } << 345 else { << 346 G4cout << "Histo::FillTupleD: WARNING! wro << 347 } << 348 } 240 } 349 241 350 //....oooOO0OOooo........oooOO0OOooo........oo << 242 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 351 243 352 void Histo::AddRow() << 244 void Histo::setFileType(const G4String& nam) 353 { 245 { 354 if (!fNtupleActive) { << 246 histType = nam; 355 return; << 356 } << 357 fManager->AddNtupleRow(); << 358 } 247 } 359 248 360 //....oooOO0OOooo........oooOO0OOooo........oo << 249 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 361 << 362 void Histo::SetFileName(const G4String& nam) << 363 { << 364 fHistName = nam; << 365 fHistoActive = true; << 366 } << 367 << 368 //....oooOO0OOooo........oooOO0OOooo........oo << 369 << 370 void Histo::SetFileType(const G4String& nam) << 371 { << 372 // format other than ROOT is not tested << 373 if (nam == "root" || nam == "ROOT") { << 374 fHistType = "root"; << 375 } << 376 else if (nam == "xml" || nam == "XML") { << 377 fHistType = "xml"; << 378 } << 379 else if (nam == "ascii" || nam == "ASCII" || << 380 fHistType = "ascii"; << 381 } << 382 } << 383 250 384 //....oooOO0OOooo........oooOO0OOooo........oo << 385 251