Files
geant4/examples/advanced/cosmicray_charging/src/LISAAnalysisManager.cc
T
2016-06-09 10:56:29 +02:00

175 lines
5.8 KiB
C++

//
// ********************************************************************
// * DISCLAIMER *
// * *
// * The following disclaimer summarizes all the specific disclaimers *
// * of contributors to this software. The specific disclaimers,which *
// * govern, are listed with their locations in: *
// * http://cern.ch/geant4/license *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. *
// * *
// * This code implementation is the intellectual property of the *
// * GEANT4 collaboration. *
// * By copying, distributing or modifying the Program (or any work *
// * based on the Program) you indicate your acceptance of this *
// * statement, and all its terms. *
// ********************************************************************
//
// ********************************************************************
// * *
// * cosmicray_charging advanced example for Geant4 *
// * (adapted simulation of test-mass charging in the LISA mission) *
// * *
// * Henrique Araujo (h.araujo@imperial.ac.uk) & Peter Wass *
// * Imperial College London *
// * *
// * LISAAnalysisManager class *
// * *
// ********************************************************************
//
// HISTORY
// 22/02/2004: migrated from LISA-V04
//
// ********************************************************************
#ifdef G4ANALYSIS_USE
#include "LISAAnalysisManager.hh"
#include <strstream>
#include "globals.hh"
LISAAnalysisManager* LISAAnalysisManager::instance = 0;
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
LISAAnalysisManager::LISAAnalysisManager() :
af(0),
tf(0),
run_tree(0),
run_tpf(0),
run_tuple(0)
{;}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
LISAAnalysisManager::~LISAAnalysisManager() {
if(af) {
delete tf;
G4cout << " LISAAnalysis -- deleted tree factory" << G4endl;
delete af;
G4cout << " LISAAnalysis -- deleted analysis factory" << G4endl;
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
LISAAnalysisManager* LISAAnalysisManager::getInstance() {
if (!instance) instance = new LISAAnalysisManager();
return instance;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
void LISAAnalysisManager::Dispose() {
if(instance) {
delete instance;
instance = 0;
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
void LISAAnalysisManager::Init() {
G4cout << G4endl << "******* Analysis with AIDA 3.0 *********" << G4endl;
// create analysis factory
if( (af = AIDA_createAnalysisFactory()) )
G4cout << " LISAAnalysis -- created analysis factory" << G4endl;
// create tree factory
if( (tf = af->createTreeFactory()) )
G4cout << " LISAAnalysis -- created tree factory" << G4endl;
}
//
// RUN ANALYSIS **************************************************************
//
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
void LISAAnalysisManager::bookRun(G4String hbookfile) {
// create RunTree
G4bool fileExists = false;
G4bool readOnly = false;
run_tree = tf->create(hbookfile, "hbook", readOnly, fileExists);
G4cout << " LISAAnalysis -- tree store: " << run_tree->storeName()<<G4endl;
// create TupleFactory
run_tpf = af->createTupleFactory(*run_tree );
G4cout << " LISAAnalysis -- created NTuple factory" << G4endl;
// Run Information
run_tuple = run_tpf->create( "1", "Run Tuple",
"float evt,tm,energy,charge,in,out,seed1,seed2");
assert(run_tuple);
G4cout << " LISAAnalysis -- created Run NTuple" << G4endl;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
void LISAAnalysisManager::FinishRun() {
// Committing transaction with the tree
G4cout << " LISAAnalysis -- committing run_tree..." << G4endl;
run_tree->commit();
run_tree->close();
delete run_tpf;
delete run_tree;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
void LISAAnalysisManager::analyseRun(G4int evt, G4int tm, G4double energy,
G4int charge, G4int in, G4int out, long seed1, long seed2) {
AIDA::ITuple* ntuple = dynamic_cast<AIDA::ITuple *> ( run_tree->find("1") );
// Fill the ntuple
ntuple->fill( ntuple->findColumn( "evt" ), (float) evt );
ntuple->fill( ntuple->findColumn( "tm" ), (float) tm );
ntuple->fill( ntuple->findColumn( "energy" ), (float) energy );
ntuple->fill( ntuple->findColumn( "charge" ), (float) charge );
ntuple->fill( ntuple->findColumn( "in" ), (float) in );
ntuple->fill( ntuple->findColumn( "out" ), (float) out );
ntuple->fill( ntuple->findColumn( "seed1" ), (float) seed1 );
ntuple->fill( ntuple->findColumn( "seed2" ), (float) seed2 );
// Values of attributes are prepared; store them to the nTuple:
ntuple->addRow();
}
#endif