]>
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 | ||
36 | //Back Propagated Tracks | |
37 | ||
38 | virtual void CleanBackTracks() {fBackTracksDataLoader.GetBaseLoader(0)->Clean();} | |
39 | Int_t LoadBackTracks(Option_t* opt=""){return fBackTracksDataLoader.GetBaseLoader(0)->Load(opt);} | |
40 | void SetBackTracksFileName(const TString& fname){fBackTracksDataLoader.SetFileName(fname);} | |
41 | TTree* TreeB(){ return fBackTracksDataLoader.Tree();} // returns a pointer to the tree of BackTracks | |
42 | void UnloadBackTracks(){fBackTracksDataLoader.GetBaseLoader(0)->Unload();} | |
43 | virtual Int_t WriteBackTracks(Option_t* opt=""){return fBackTracksDataLoader.GetBaseLoader(0)->WriteData(opt);} | |
44 | ||
45 | ||
46 | protected: | |
47 | ||
48 | // METHODS | |
49 | virtual void MakeRawClustersContainer() {fRawClustersDataLoader.MakeTree();} | |
50 | Int_t PostRawClusters(){return fRawClustersDataLoader.GetBaseLoader(0)->Post();} | |
51 | ||
52 | virtual void MakeBackTracksContainer() {fBackTracksDataLoader.MakeTree();} | |
53 | Int_t PostBackTracks(){return fBackTracksDataLoader.GetBaseLoader(0)->Post();} | |
54 | ||
55 | // DATA | |
56 | AliDataLoader fRawClustersDataLoader; | |
57 | static const TString fgkDefaultRawClustersContainerName; | |
58 | ||
59 | AliDataLoader fBackTracksDataLoader; | |
60 | static const TString fgkDefaultBackTracksContainerName; | |
61 | ||
62 | AliDataLoader fVertexDataLoader; | |
63 | static const TString fgkDefaultVerticesContainerName; | |
64 | ||
65 | public: | |
66 | ClassDef(AliITSLoader,2) | |
67 | }; | |
68 | ||
69 | #endif | |
70 | ||
71 |