ITS/AliITSLoader.h

   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     //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
  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();}
  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();}
  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
  86     AliDataLoader fV0DataLoader;
  87     static const TString fgkDefaultV0ContainerName;
  88
  89     AliDataLoader fCascadeDataLoader;
  90     static const TString fgkDefaultCascadeContainerName;
  91
  92    public:
  93      ClassDef(AliITSLoader,2)
  94  };
  95
  96 #endif
  97
  98