source: trunk/examples/extended/electromagnetic/TestEm17/src/HistoManager.cc @ 1230

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

update to geant4.9.3
File size: 6.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.4 2008/09/26 20:15:04 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),hf(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] = "testem17";
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  }
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 << "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  hf = af->createHistogramFactory(*tree);
108
109  // create selected histograms
110  for (G4int k=0; 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  if(factoryOn) 
118      G4cout << "\n----> Histogram Tree is opened in " << fileName[1]  << G4endl;
119
120#endif
121}
122
123//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
124
125void HistoManager::save()
126{
127#ifdef G4ANALYSIS_USE
128  if (factoryOn) {
129    tree->commit();       // Writing the histograms to the file
130    tree->close();        // and closing the tree (and the file)
131    G4cout << "\n----> Histogram Tree is saved in " << fileName[1] << G4endl;
132
133    delete hf;
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 e, G4double weight)
144{
145  if (ih > MaxHisto) {
146    G4cout << "---> warning from HistoManager::FillHisto() : histo " << ih
147           << "does not exist; e= " << e << " w= " << weight << G4endl;
148    return;
149  }
150#ifdef G4ANALYSIS_USE
151  if(exist[ih]) histo[ih]->fill(e/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 > MaxHisto) {
161    G4cout << "---> warning from HistoManager::SetHisto() : histo " << ih
162           << "does not exist" << G4endl;
163    return;
164  }
165 
166  const G4String id[] = { "0", "1", "2", "3", "4", "5", "6", "7"};
167  const G4String title[] = 
168                { "dummy",                                              //0
169                  "log10(Etransfert/Emu) Ionization",                   //1
170                  "log10(Etransfert/Emu) Pair",                         //2
171                  "log10(Etransfert/Emu) Brems",                        //3
172                  "log10(Etransfert/Emu) Nuclear",                      //4
173                  "log10(Etransfert/Emu) Ionization",                   //5
174                  "log10(Etransfert/Emu) Pair",                         //6
175                  "log10(Etransfert/Emu) Brems"                         //7
176                 };
177
178  G4String titl = title[ih];
179  G4double vmin = valmin, vmax = valmax;
180  Unit[ih] = 1.;
181
182  if (unit != "none") {
183    titl = title[ih] + " (" + unit + ")";
184    Unit[ih] = G4UnitDefinition::GetValueOf(unit);
185    vmin = valmin/Unit[ih]; vmax = valmax/Unit[ih];
186  }
187
188  exist[ih] = true;
189  Label[ih] = id[ih];
190  Title[ih] = titl;
191  Nbins[ih] = nbins;
192  Vmin[ih]  = vmin;
193  Vmax[ih]  = vmax;
194  Width[ih] = (valmax-valmin)/nbins;
195
196  G4cout << "----> SetHisto " << ih << ": " << titl << ";  "
197         << nbins << " bins from "
198         << vmin << " " << unit << " to " << vmax << " " << unit << G4endl;
199
200}
201
202//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
203
204void HistoManager::RemoveHisto(G4int ih)
205{
206 if (ih > MaxHisto) {
207    G4cout << "---> warning from HistoManager::RemoveHisto() : histo " << ih
208           << "does not exist" << G4endl;
209    return;
210  }
211
212  histo[ih] = 0;  exist[ih] = false;
213}
214
215//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
216
Note: See TracBrowser for help on using the repository browser.