ITS UPGRADE

[u/mrichter/AliRoot.git] / ITS / UPGRADE / ITSUpgradeBase / AliITSULoader.h

diff --git a/ITS/UPGRADE/ITSUpgradeBase/AliITSULoader.h b/ITS/UPGRADE/ITSUpgradeBase/AliITSULoader.h
new file mode 100644 (file)
index 0000000..182ac32
--- /dev/null
+++ b/ITS/UPGRADE/ITSUpgradeBase/AliITSULoader.h
@@ -0,0 +1,136 @@
+#ifndef ALIITSULOADER_H
+#define ALIITSULOADER_H
+//////////////////////////////////////////////////////////
+// Loader class for ITS Upgrade                         //
+//////////////////////////////////////////////////////////
+#include <AliLoader.h>
+#include <AliESDVertex.h>
+class AliITSpidESD;
+class AliITSdigit;
+class TObjArray;
+
+class AliITSULoader: public AliLoader{
+  public:
+    AliITSULoader();
+    AliITSULoader(const Char_t *name,const Char_t *topfoldername);
+    AliITSULoader(const Char_t *name,TFolder *topfolder);
+
+    virtual ~AliITSULoader();
+
+    void           MakeTree(Option_t* opt);
+    virtual void   SetupDigits(TObjArray *digPerDet,Int_t n,
+                                const Char_t **digclass); // Sets up digits
+    // Gets the AliITSdigit for a given chip and a specific digit in that
+    // chip. Array of digits stored in AliITS (must use 
+    // SetupDigits(AliITS *its)).
+    // virtual AliITSdigit* GetDigit(AliITS *its,Int_t chip,Int_t digit);
+    // Gets the AliITSdigit for a given chip and a specific digit in that
+    // chip. Array of digits stored in a user defined TObjArray digPerDet
+    virtual AliITSdigit* GetDigit(TObjArray *digPerDet,Int_t chip,Int_t digit);
+
+    //Raw Clusters
+    AliDataLoader* GetRawClLoader() {return GetDataLoader("Raw Clusters");}
+    virtual void   CleanRawClusters() {
+        GetRawClLoader()->GetBaseLoader(0)->Clean();}
+    Int_t          LoadRawClusters(Option_t* opt=""){
+        return GetRawClLoader()->GetBaseLoader(0)->Load(opt);}
+    void           SetRawClustersFileName(const TString& fname){
+        GetRawClLoader()->SetFileName(fname);}
+    // returns a pointer to the tree of  RawClusters
+    TTree*         TreeC(){ return GetRawClLoader()->Tree();} 
+    void           UnloadRawClusters(){
+        GetRawClLoader()->GetBaseLoader(0)->Unload();}
+    virtual Int_t  WriteRawClusters(Option_t* opt=""){
+        return GetRawClLoader()->GetBaseLoader(0)->WriteData(opt);}
+
+    //Vertices
+    AliDataLoader* GetVertexDataLoader() {
+        return GetDataLoader("Primary Vertices");}
+    virtual void   CleanVertices() {
+        GetVertexDataLoader()->GetBaseLoader(0)->Clean();}
+    Int_t          LoadVertices(Option_t* opt=""){
+        return GetVertexDataLoader()->GetBaseLoader(0)->Load(opt);}
+    void           SetVerticesFileName(const TString& fname){
+        GetVertexDataLoader()->SetFileName(fname);}
+    void           UnloadVertices(){
+        GetVertexDataLoader()->GetBaseLoader(0)->Unload();}
+    virtual Int_t  WriteVertices(Option_t* opt=""){
+        return GetVertexDataLoader()->GetBaseLoader(0)->WriteData(opt);}
+    virtual Int_t PostVertex(AliESDVertex *ptr){
+        return GetVertexDataLoader()->GetBaseLoader(0)->Post(ptr);}
+    //    virtual void SetVerticesContName(const char *name){
+    //       GetVertexDataLoader()->GetBaseLoader(0)->SetName(name);}
+    AliESDVertex *GetVertex(){
+        return static_cast <AliESDVertex*>(GetVertexDataLoader()->
+                                           GetBaseLoader(0)->Get());}
+    //V0s
+    AliDataLoader* GetV0DataLoader() {return GetDataLoader("V0 Vertices");}
+    virtual void   CleanV0s() {GetV0DataLoader()->GetBaseLoader(0)->Clean();}
+    Int_t          LoadV0s(Option_t* opt=""){
+        return GetV0DataLoader()->GetBaseLoader(0)->Load(opt);}
+    void           SetV0FileName(const TString& fname){
+        GetV0DataLoader()->SetFileName(fname);}
+    void           UnloadV0s(){GetV0DataLoader()->GetBaseLoader(0)->Unload();}
+    virtual Int_t  WriteV0s(Option_t* opt=""){
+        return GetV0DataLoader()->GetBaseLoader(0)->WriteData(opt);}
+    TTree*         TreeV0(){ return GetV0DataLoader()->Tree();}
+
+    //Cascades
+    AliDataLoader* GetCascadeDataLoader() {return GetDataLoader("Cascades");}
+    virtual void   CleanCascades() {
+        GetCascadeDataLoader()->GetBaseLoader(0)->Clean();}
+    Int_t          LoadCascades(Option_t* opt=""){
+        return GetCascadeDataLoader()->GetBaseLoader(0)->Load(opt);}
+    void           SetCascadeFileName(const TString& fname){
+        GetCascadeDataLoader()->SetFileName(fname);}
+    void           UnloadCascades(){
+        GetCascadeDataLoader()->GetBaseLoader(0)->Unload();}
+    virtual Int_t  WriteCascades(Option_t* opt=""){
+        return GetCascadeDataLoader()->GetBaseLoader(0)->WriteData(opt);}
+    TTree*         TreeX(){ return GetCascadeDataLoader()->Tree();}
+
+    //Back Propagated Tracks
+    AliDataLoader* GetBackTracksDataLoader() {
+        return GetDataLoader("Back Propagated Tracks");}
+    virtual void   CleanBackTracks() {
+        GetBackTracksDataLoader()->GetBaseLoader(0)->Clean();}
+    Int_t          LoadBackTracks(Option_t* opt=""){
+        return GetBackTracksDataLoader()->GetBaseLoader(0)->Load(opt);}
+    void           SetBackTracksFileName(const TString& fname){
+        GetBackTracksDataLoader()->SetFileName(fname);}
+     // returns a pointer to the tree of  BackTracks
+    TTree*         TreeB(){ return GetBackTracksDataLoader()->Tree();}
+    void           UnloadBackTracks(){
+        GetBackTracksDataLoader()->GetBaseLoader(0)->Unload();}
+    virtual Int_t  WriteBackTracks(Option_t* opt=""){
+        return GetBackTracksDataLoader()->GetBaseLoader(0)->WriteData(opt);}
+
+    // Geometry. Geom is read from file, unless already loaded
+    // readout from file can be forced if force=kTRUE
+  protected:
+
+    AliITSULoader(const AliITSULoader &ob); // copy constructor
+    AliITSULoader& operator=(const AliITSULoader & /* source */); // ass.
+
+    // METHODS
+    virtual void   MakeRawClustersContainer() {GetRawClLoader()->MakeTree();}
+    Int_t          PostRawClusters(){
+        return GetRawClLoader()->GetBaseLoader(0)->Post();}
+
+    virtual void   MakeBackTracksContainer() {
+        GetBackTracksDataLoader()->MakeTree();}
+    Int_t          PostBackTracks(){
+        return GetBackTracksDataLoader()->GetBaseLoader(0)->Post();}
+    virtual void   MakeV0Container() {GetV0DataLoader()->MakeTree();}
+    Int_t          PostV0s(){
+        return GetV0DataLoader()->GetBaseLoader(0)->Post();}
+
+    virtual void   MakeCascadeContainer() {GetCascadeDataLoader()->MakeTree();}
+    Int_t          PostCascades(){
+        return GetCascadeDataLoader()->GetBaseLoader(0)->Post();}
+
+
+    ClassDef(AliITSULoader,1) // Loader for additional ITS specific trees.
+};
+ 
+#endif