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 | } |
---|