Script to draw occupancies

[u/mrichter/AliRoot.git] / ITS / AliITSLoader.h

diff --git a/ITS/AliITSLoader.h b/ITS/AliITSLoader.h
index ebba0445e9ecf1317d127e0dcfe36516ee6e65c2..cbdd340ad82f11e2780fe3592160dce8b096454c 100644 (file)
--- a/ITS/AliITSLoader.h
+++ b/ITS/AliITSLoader.h
@@ -1,10 +1,12 @@
 #ifndef ALIITSLOADER_H
 #define ALIITSLOADER_H
-
+//////////////////////////////////////////////////////////
+// Loader class for ITS                                 //
+//////////////////////////////////////////////////////////
 #include <AliLoader.h>
 #include <AliESDVertex.h>
-
-class AliITS;
+#include <AliITSgeom.h>
+class AliITSpidESD;
 class AliITSdigit;
 class TObjArray;
 
@@ -17,16 +19,12 @@ class AliITSLoader: public AliLoader{
     virtual ~AliITSLoader();
 
     void           MakeTree(Option_t* opt);
-
-    // General ITS shortcuts
-    virtual AliITS* GetITS(); // Return pointer to the ITS.
-    virtual void    SetupDigits(AliITS *its); // Sets up digit using AliITS
-    virtual void    SetupDigits(TObjArray *digPerDet,Int_t n,
+    virtual void   SetupDigits(TObjArray *digPerDet,Int_t n,
                                 const Char_t **digclass); // Sets up digits
     // Gets the AliITSdigit for a given module and a specific digit in that
     // module. Array of digits stored in AliITS (must use 
     // SetupDigits(AliITS *its)).
-    virtual AliITSdigit* GetDigit(AliITS *its,Int_t module,Int_t digit);
+    // virtual AliITSdigit* GetDigit(AliITS *its,Int_t module,Int_t digit);
     // Gets the AliITSdigit for a given module and a specific digit in that
     // module. Array of digits stored in a user defined TObjArray digPerDet
     virtual AliITSdigit* GetDigit(TObjArray *digPerDet,Int_t module,Int_t digit);
@@ -66,7 +64,6 @@ class AliITSLoader: public AliLoader{
     AliESDVertex *GetVertex(){
         return static_cast <AliESDVertex*>(GetVertexDataLoader()->
                                            GetBaseLoader(0)->Get());}
-
     //V0s
     AliDataLoader* GetV0DataLoader() {return GetDataLoader("V0 Vertices");}
     virtual void   CleanV0s() {GetV0DataLoader()->GetBaseLoader(0)->Clean();}
@@ -109,8 +106,15 @@ class AliITSLoader: public AliLoader{
     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
+    AliITSgeom* GetITSgeom(Bool_t force=kFALSE); 
+    void SetITSgeom(AliITSgeom* g);
   protected:
 
+    AliITSLoader(const AliITSLoader &ob); // copy constructor
+    AliITSLoader& operator=(const AliITSLoader & /* source */); // ass.
+
     // METHODS
     virtual void   MakeRawClustersContainer() {GetRawClLoader()->MakeTree();}
     Int_t          PostRawClusters(){
@@ -134,8 +138,10 @@ class AliITSLoader: public AliLoader{
     static const TString fgkDefaultVerticesContainerName;     //default for primary vertices container name
     static const TString fgkDefaultV0ContainerName;           //default for V0 container name
     static const TString fgkDefaultCascadeContainerName;      //default fo cascade container name
+    AliITSgeom *fGeom;     //! pointer to the ITS geometry class
+
 
-    ClassDef(AliITSLoader,3) // Loader for additional ITS specific trees.
+    ClassDef(AliITSLoader,6) // Loader for additional ITS specific trees.
 };
  
 #endif