source: trunk/examples/extended/electromagnetic/TestEm3/src/HistoManager.cc @ 1309

Last change on this file since 1309 was 1230, checked in by garnier, 14 years ago

update to geant4.9.3
File size: 8.9 KB
Line 
1//
2// ********************************************************************
3// * License and Disclaimer                                           *
4// *                                                                  *
5// * The  Geant4 software  is  copyright of the Copyright Holders  of *
6// * the Geant4 Collaboration.  It is provided  under  the terms  and *
7// * conditions of the Geant4 Software License,  included in the file *
8// * LICENSE and available at  http://cern.ch/geant4/license .  These *
9// * include a list of copyright holders.                             *
10// *                                                                  *
11// * Neither the authors of this software system, nor their employing *
12// * institutes,nor the agencies providing financial support for this *
13// * work  make  any representation or  warranty, express or implied, *
14// * regarding  this  software system or assume any liability for its *
15// * use.  Please see the license in the file  LICENSE  and URL above *
16// * for the full disclaimer and the limitation of liability.         *
17// *                                                                  *
18// * This  code  implementation is the result of  the  scientific and *
19// * technical work of the GEANT4 collaboration.                      *
20// * By using,  copying,  modifying or  distributing the software (or *
21// * any work based  on the software)  you  agree  to acknowledge its *
22// * use  in  resulting  scientific  publications,  and indicate your *
23// * acceptance of all terms of the Geant4 Software license.          *
24// ********************************************************************
25//
26// $Id: HistoManager.cc,v 1.17 2009/03/04 18:49:17 maire Exp $
27// GEANT4 tag $Name: geant4-09-03-cand-01 $
28//
29//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
30//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
31
32#include "HistoManager.hh"
33#include "HistoMessenger.hh"
34#include "G4UnitsTable.hh"
35
36#ifdef G4ANALYSIS_USE
37#include "AIDA/AIDA.h"
38#endif
39
40//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
41
42HistoManager::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 << "TestEm1::HistoManager::HistoManager :" 
50           << " problem creating the AIDA analysis factory."
51           << G4endl;
52  }       
53#endif
54 
55  fileName[0] = "testem3";
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  histoMessenger = new HistoMessenger(this);
67}
68
69//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
70
71HistoManager::~HistoManager()
72{ 
73  delete histoMessenger;
74 
75#ifdef G4ANALYSIS_USE 
76  delete af;
77#endif     
78}
79
80//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
81
82void HistoManager::book()
83{ 
84#ifdef G4ANALYSIS_USE
85  if(!af) return;           
86 
87 // Creating a tree mapped to an hbook file.
88 fileName[1] = fileName[0] + "." + fileType;
89 G4bool readOnly  = false;
90 G4bool createNew = true;
91 AIDA::ITreeFactory* tf  = af->createTreeFactory(); 
92 tree = tf->create(fileName[1], fileType, readOnly, createNew, fileOption);
93 delete tf;
94 if(!tree) {
95   G4cout << "TestEm1::HistoManager::book :" 
96          << " problem creating the AIDA tree with "
97          << " storeName = " << fileName[1]
98          << " storeType = " << fileType
99          << " readOnly = "  << readOnly
100          << " createNew = " << createNew
101          << " options = "   << fileOption
102          << G4endl;
103   return;
104 }
105 
106 // Creating a histogram factory, whose histograms will be handled by the tree
107 AIDA::IHistogramFactory* hf = af->createHistogramFactory(*tree);
108 
109 // create selected histograms
110 for (G4int k=1; k<MaxHisto; k++) {
111   if (exist[k]) {
112     histo[k] = hf->createHistogram1D( Label[k], Title[k],
113                                                 Nbins[k], Vmin[k], Vmax[k]);
114     factoryOn = true;
115   }                                                                                       
116 }
117 delete hf;
118 if (factoryOn) 
119     G4cout << "\n----> Histogram Tree is opened in " << fileName[1]  << G4endl;
120#endif
121}
122
123//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
124
125void HistoManager::save()
126{ 
127#ifdef G4ANALYSIS_USE
128  if (factoryOn) {
129    saveAscii();          // Write ascii file, if any     
130    tree->commit();       // Writing the histograms to the file
131    tree->close();        // and closing the tree (and the file)
132    G4cout << "\n----> Histogram Tree is saved in " << fileName[1] << G4endl;
133
134    delete tree;
135    tree = 0;
136    factoryOn = false;
137  }
138#endif
139}
140
141//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
142
143void HistoManager::FillHisto(G4int ih, G4double xbin, G4double weight)
144{
145  if (ih >= MaxHisto) {
146    G4cout << "---> warning from HistoManager::FillHisto() : histo " << ih
147           << " xbin= " << xbin << " weight= " << weight << G4endl;
148    return;
149  }
150#ifdef G4ANALYSIS_USE
151  if(exist[ih]) histo[ih]->fill(xbin/Unit[ih], weight);
152#endif
153}
154
155//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
156
157void HistoManager::SetHisto(G4int ih, 
158            G4int nbins, G4double valmin, G4double valmax, const G4String& unit)
159{   
160  if (ih < 1 || ih >= MaxHisto) {
161    G4cout << "---> warning from HistoManager::SetHisto() : histo " << ih
162           << "does not exist" << G4endl;
163    return;
164  }
165
166 // histo 1 : energy deposit in absorber 1
167 // histo 2 : energy deposit in absorber 2
168 // ...etc...........
169 // MaxAbsor = 10 (-1)
170 //
171 // histo 11 : longitudinal profile of energy deposit in absorber 1 (MeV)
172 // histo 12 : longitudinal profile of energy deposit in absorber 2 (MeV) 
173 // ...etc........... 
174 //
175 // histo 21 : energy flow (MeV)
176 // histo 22 : lateral energy leak (MeV) 
177         
178  const G4String id[] = { "0", "1", "2", "3", "4", "5", "6", "7", "8", "9",
179                         "10","11","12","13","14","15","16","17","18","19",
180                         "20","21","22"};
181                         
182  G4String title;
183  G4double vmin = valmin, vmax = valmax;
184                         
185  if (ih < MaxAbsor) {                   
186    title = "Edep in absorber " + id[ih] + " (" + unit + ")";
187    Unit[ih] = G4UnitDefinition::GetValueOf(unit);   
188    vmin = valmin/Unit[ih]; vmax = valmax/Unit[ih]; 
189  } else if (ih > MaxAbsor && ih < 2*MaxAbsor) {
190    title = "longit. profile of Edep (MeV/event) in absorber " 
191           + id[ih-MaxAbsor];
192  } else if (ih == 2*MaxAbsor+1) {
193    title = "energy flow (MeV/event)";
194  } else if (ih == 2*MaxAbsor+2) {
195    title = "lateral energy leak (MeV/event)";
196  } else return;
197       
198  exist[ih] = true;
199  Label[ih] = id[ih];
200  Title[ih] = title;
201  Nbins[ih] = nbins; 
202  Vmin[ih]  = vmin; 
203  Vmax[ih]  = vmax; 
204  Width[ih] = (valmax-valmin)/nbins;
205 
206  G4cout << "----> SetHisto " << ih << ": " << title << ";  "
207         << nbins << " bins from " 
208         << vmin << " " << unit << " to " << vmax << " " << unit << G4endl;
209                 
210}
211
212//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
213
214void HistoManager::Normalize(G4int ih, G4double fac)
215{
216  if (ih >= MaxHisto) {
217    G4cout << "---> warning from HistoManager::Normalize() : histo " << ih
218           << "  fac= " << fac << G4endl;
219    return;
220  }
221#ifdef G4ANALYSIS_USE
222  if(exist[ih]) histo[ih]->scale(fac);
223#endif
224}
225
226//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
227
228void HistoManager::RemoveHisto(G4int ih) 
229{ 
230 if (ih >= MaxHisto) {
231    G4cout << "---> warning from HistoManager::RemoveHisto() : histo " << ih
232           << "does not exist" << G4endl;
233    return;
234  }
235         
236  histo[ih] = 0; exist[ih] = false;                     
237}
238
239//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
240
241void HistoManager::PrintHisto(G4int ih)
242{
243 if (ih < MaxHisto) { ascii[ih] = true; ascii[0] = true; }
244 else
245    G4cout << "---> warning from HistoManager::PrintHisto() : histo " << ih
246           << "does not exist" << G4endl;
247}
248
249//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
250
251#include <fstream>
252
253void HistoManager::saveAscii()
254{
255#ifdef G4ANALYSIS_USE
256
257 if (!ascii[0]) return;
258 
259 G4String name = fileName[0] + ".ascii";
260 std::ofstream File(name, std::ios::out);
261 File.setf( std::ios::scientific, std::ios::floatfield );
262 
263 //write selected histograms
264 for (G4int ih=0; ih<MaxHisto; ih++) {
265    if (exist[ih] && ascii[ih]) {
266      File << "\n  1D histogram " << ih << ": " << Title[ih] 
267           << "\n \n \t     X \t\t     Y" << G4endl;
268     
269      for (G4int iBin=0; iBin<Nbins[ih]; iBin++) {
270         File << "  " << iBin << "\t" 
271              << 0.5*(histo[ih]->axis().binLowerEdge(iBin) +
272                      histo[ih]->axis().binUpperEdge(iBin)) << "\t"
273              << histo[ih]->binHeight(iBin) 
274              << G4endl;
275      } 
276    }
277 }
278#endif
279}
280
281//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
282
283
Note: See TracBrowser for help on using the repository browser.