| 1 | #include "ELYSE/Analysis.hh"
|
|---|
| 2 |
|
|---|
| 3 | //AIDA
|
|---|
| 4 | #include <AIDA/IAnalysisFactory.h>
|
|---|
| 5 | #include <AIDA/ITreeFactory.h>
|
|---|
| 6 | #include <AIDA/ITree.h>
|
|---|
| 7 | #include <AIDA/ITupleFactory.h>
|
|---|
| 8 |
|
|---|
| 9 | //Std
|
|---|
| 10 | #include <iostream>
|
|---|
| 11 | #include <vector>
|
|---|
| 12 | #include <string>
|
|---|
| 13 |
|
|---|
| 14 | ELYSE::Analysis::Analysis(
|
|---|
| 15 | AIDA::IAnalysisFactory* aAIDA
|
|---|
| 16 | ,const std::string& aFormat
|
|---|
| 17 | ,bool aBatch)
|
|---|
| 18 | :fAIDA(aAIDA)
|
|---|
| 19 | ,fBatch(aBatch)
|
|---|
| 20 | ,fTree(0)
|
|---|
| 21 | {
|
|---|
| 22 | if(!fAIDA) return;
|
|---|
| 23 |
|
|---|
| 24 | AIDA::ITreeFactory* treeFactory = fAIDA->createTreeFactory();
|
|---|
| 25 | if(fBatch) {
|
|---|
| 26 | if( (aFormat=="root") || (aFormat=="")) {
|
|---|
| 27 | //ROOT tree :
|
|---|
| 28 | std::string opts = "export=root";
|
|---|
| 29 | fTree = treeFactory->create("ELYSE.root","root",false,true,opts);
|
|---|
| 30 | } else if(aFormat=="hdf5") {
|
|---|
| 31 | //HDF5 tree :
|
|---|
| 32 | std::string opts = "";
|
|---|
| 33 | fTree = treeFactory->create("ELYSE.hdf5","hdf5",false,true,opts);
|
|---|
| 34 | } else if(aFormat=="xml") {
|
|---|
| 35 | // AIDA XML tree :
|
|---|
| 36 | std::string opts = "compress=yes";
|
|---|
| 37 | fTree = treeFactory->create("ELYSE.xml","xml",false,true,opts);
|
|---|
| 38 | } else {
|
|---|
| 39 | std::cout << "Analysis :"
|
|---|
| 40 | << " file format \"" << aFormat << "\""
|
|---|
| 41 | << " not handled."
|
|---|
| 42 | << std::endl;
|
|---|
| 43 | }
|
|---|
| 44 | } else {
|
|---|
| 45 | // Memory tree :
|
|---|
| 46 | fTree = treeFactory->create();
|
|---|
| 47 | }
|
|---|
| 48 | delete treeFactory;
|
|---|
| 49 | if(!fTree) {
|
|---|
| 50 | std::cout << "Analysis : FATAL"
|
|---|
| 51 | << " tree not created." << std::endl;
|
|---|
| 52 | return;
|
|---|
| 53 | }
|
|---|
| 54 |
|
|---|
| 55 | //Booking Event Tuple
|
|---|
| 56 | AIDA::ITupleFactory* tf = fAIDA->createTupleFactory(*fTree);
|
|---|
| 57 | if(fBatch) {
|
|---|
| 58 | std::string column = "int eventId, vtxvol, ";
|
|---|
| 59 | column += "ITuple vtxPos = { double x, y, z }, ";
|
|---|
| 60 | column += "int nPart, ";
|
|---|
| 61 | column += "ITuple track = { int pId, parent, float timeStart, ";
|
|---|
| 62 | column += "ITuple direction = { double dx, dy, dz }, ";
|
|---|
| 63 | column += "double mass, pTot, ETot, ";
|
|---|
| 64 | column += "ITuple momentum = { double px, py, pz }, ";
|
|---|
| 65 | column += "ITuple startPos = { double x, y, z }, ";
|
|---|
| 66 | column += "ITuple stopPos = { double x, y, z } ";
|
|---|
| 67 | column += "int startVol, stopVol ";
|
|---|
| 68 | column += "}, ";
|
|---|
| 69 | column += "int nHits,";
|
|---|
| 70 | column += "ITuple hit = { double time, eTot, x, y, z, trkId, parentId, pdg, edep } ";
|
|---|
| 71 |
|
|---|
| 72 | tf->create("Event","ELYSE Event",column,"");
|
|---|
| 73 | }//batch mode
|
|---|
| 74 |
|
|---|
| 75 |
|
|---|
| 76 | //release the TupleFactory
|
|---|
| 77 | delete tf;
|
|---|
| 78 |
|
|---|
| 79 | }//Ctor
|
|---|
| 80 |
|
|---|
| 81 | //----------------------------------------------------------------------
|
|---|
| 82 | ELYSE::Analysis::~Analysis(){
|
|---|
| 83 | //Not to be changed
|
|---|
| 84 | if(!fTree) return;
|
|---|
| 85 | if(fBatch) {
|
|---|
| 86 | fTree->commit();
|
|---|
| 87 | delete fTree;
|
|---|
| 88 | fTree = 0;
|
|---|
| 89 | }
|
|---|
| 90 | }//Dtor
|
|---|
| 91 |
|
|---|
| 92 | //----------------------------------------------------------------------
|
|---|
| 93 |
|
|---|
| 94 | void ELYSE::Analysis::closeTree(){
|
|---|
| 95 | if(!fTree) return;
|
|---|
| 96 | if(!fBatch) return;
|
|---|
| 97 | fTree->commit();
|
|---|
| 98 | delete fTree;
|
|---|
| 99 | fTree = 0;
|
|---|
| 100 | }
|
|---|
