]>
Commit | Line | Data |
---|---|---|
88cb7938 | 1 | #ifndef ALIITSGETTER_H |
2 | #define ALIITSGETTER_H | |
3 | ||
4 | #include <AliLoader.h> | |
5 | #include <AliITSVertex.h> | |
6 | ||
7 | class AliITSLoader: public AliLoader | |
8 | { | |
9 | public: | |
10 | AliITSLoader(){}; | |
11 | AliITSLoader(const Char_t *name,const Char_t *topfoldername); | |
12 | AliITSLoader(const Char_t *name,TFolder *topfolder); | |
13 | ||
14 | virtual ~AliITSLoader(); | |
15 | ||
16 | void MakeTree(Option_t* opt); | |
17 | ||
18 | //Raw Clusters | |
19 | virtual void CleanRawClusters() {fRawClustersDataLoader.GetBaseLoader(0)->Clean();} | |
20 | Int_t LoadRawClusters(Option_t* opt=""){return fRawClustersDataLoader.GetBaseLoader(0)->Load(opt);} | |
21 | void SetRawClustersFileName(const TString& fname){fRawClustersDataLoader.SetFileName(fname);} | |
22 | TTree* TreeC(){ return fRawClustersDataLoader.Tree();} // returns a pointer to the tree of RawClusters | |
23 | void UnloadRawClusters(){fRawClustersDataLoader.GetBaseLoader(0)->Unload();} | |
24 | virtual Int_t WriteRawClusters(Option_t* opt=""){return fRawClustersDataLoader.GetBaseLoader(0)->WriteData(opt);} | |
25 | ||
26 | //Vertices | |
27 | virtual void CleanVertices() {fVertexDataLoader.GetBaseLoader(0)->Clean();} | |
28 | Int_t LoadVertices(Option_t* opt=""){return fVertexDataLoader.GetBaseLoader(0)->Load(opt);} | |
29 | void SetVerticesFileName(const TString& fname){fVertexDataLoader.SetFileName(fname);} | |
30 | void UnloadVertices(){fVertexDataLoader.GetBaseLoader(0)->Unload();} | |
31 | virtual Int_t WriteVertices(Option_t* opt=""){return fVertexDataLoader.GetBaseLoader(0)->WriteData(opt);} | |
32 | virtual Int_t PostVertex(AliITSVertex *ptr){return fVertexDataLoader.GetBaseLoader(0)->Post(ptr);} | |
33 | // virtual void SetVerticesContName(const char *name){fVertexDataLoader.GetBaseLoader(0)->SetName(name);} | |
34 | AliITSVertex *GetVertex(){return static_cast <AliITSVertex*>(fVertexDataLoader.GetBaseLoader(0)->Get());} | |
35 | ||
566abf75 | 36 | //V0s |
37 | virtual void CleanV0s() {fV0DataLoader.GetBaseLoader(0)->Clean();} | |
38 | Int_t LoadV0s(Option_t* opt=""){return fV0DataLoader.GetBaseLoader(0)->Load(opt);} | |
39 | void SetV0FileName(const TString& fname){fV0DataLoader.SetFileName(fname);} | |
40 | void UnloadV0s(){fV0DataLoader.GetBaseLoader(0)->Unload();} | |
41 | virtual Int_t WriteV0s(Option_t* opt=""){return fV0DataLoader.GetBaseLoader(0)->WriteData(opt);} | |
42 | TTree* TreeV0(){ return fV0DataLoader.Tree();} | |
43 | ||
44 | //Cascades | |
45 | virtual void CleanCascades() {fCascadeDataLoader.GetBaseLoader(0)->Clean();} | |
46 | Int_t LoadCascades(Option_t* opt=""){return fCascadeDataLoader.GetBaseLoader(0)->Load(opt);} | |
47 | void SetCascadeFileName(const TString& fname){fCascadeDataLoader.SetFileName(fname);} | |
48 | void UnloadCascades(){fCascadeDataLoader.GetBaseLoader(0)->Unload();} | |
49 | virtual Int_t WriteCascades(Option_t* opt=""){return fCascadeDataLoader.GetBaseLoader(0)->WriteData(opt);} | |
50 | TTree* TreeX(){ return fCascadeDataLoader.Tree();} | |
51 | ||
88cb7938 | 52 | //Back Propagated Tracks |
53 | ||
54 | virtual void CleanBackTracks() {fBackTracksDataLoader.GetBaseLoader(0)->Clean();} | |
55 | Int_t LoadBackTracks(Option_t* opt=""){return fBackTracksDataLoader.GetBaseLoader(0)->Load(opt);} | |
56 | void SetBackTracksFileName(const TString& fname){fBackTracksDataLoader.SetFileName(fname);} | |
57 | TTree* TreeB(){ return fBackTracksDataLoader.Tree();} // returns a pointer to the tree of BackTracks | |
58 | void UnloadBackTracks(){fBackTracksDataLoader.GetBaseLoader(0)->Unload();} | |
59 | virtual Int_t WriteBackTracks(Option_t* opt=""){return fBackTracksDataLoader.GetBaseLoader(0)->WriteData(opt);} | |
60 | ||
61 | ||
62 | protected: | |
63 | ||
64 | // METHODS | |
65 | virtual void MakeRawClustersContainer() {fRawClustersDataLoader.MakeTree();} | |
66 | Int_t PostRawClusters(){return fRawClustersDataLoader.GetBaseLoader(0)->Post();} | |
67 | ||
68 | virtual void MakeBackTracksContainer() {fBackTracksDataLoader.MakeTree();} | |
69 | Int_t PostBackTracks(){return fBackTracksDataLoader.GetBaseLoader(0)->Post();} | |
566abf75 | 70 | virtual void MakeV0Container() {fV0DataLoader.MakeTree();} |
71 | Int_t PostV0s(){return fV0DataLoader.GetBaseLoader(0)->Post();} | |
72 | ||
73 | virtual void MakeCascadeContainer() {fCascadeDataLoader.MakeTree();} | |
74 | Int_t PostCascades(){return fCascadeDataLoader.GetBaseLoader(0)->Post();} | |
88cb7938 | 75 | |
76 | // DATA | |
77 | AliDataLoader fRawClustersDataLoader; | |
78 | static const TString fgkDefaultRawClustersContainerName; | |
79 | ||
80 | AliDataLoader fBackTracksDataLoader; | |
81 | static const TString fgkDefaultBackTracksContainerName; | |
82 | ||
83 | AliDataLoader fVertexDataLoader; | |
84 | static const TString fgkDefaultVerticesContainerName; | |
85 | ||
566abf75 | 86 | AliDataLoader fV0DataLoader; |
87 | static const TString fgkDefaultV0ContainerName; | |
88 | ||
89 | AliDataLoader fCascadeDataLoader; | |
90 | static const TString fgkDefaultCascadeContainerName; | |
91 | ||
88cb7938 | 92 | public: |
93 | ClassDef(AliITSLoader,2) | |
94 | }; | |
95 | ||
96 | #endif | |
97 | ||
98 |