]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/AliITS.h
Modifications associated with remerging the Ba/Sa and Dubna pixel simulations,
[u/mrichter/AliRoot.git] / ITS / AliITS.h
index 39f50e15934877e0fd20fcff46cc6d89c8832a1b..29aec99036f91604d814639dd896f3410d7a1b49 100644 (file)
@@ -12,7 +12,7 @@
 #include <TObjArray.h> // used in inline function GetModule.
 #include <TBranch.h>   // used in inline function SetHitsAddressBranch
 
-#include "AliRun.h"
+#include "AliLoader.h"
 #include "AliDetector.h"
 #include "AliITSDetType.h"
 
@@ -23,6 +23,7 @@ class TFile;
 //class AliITSDetType;
 class AliITSsimulation;
 class AliITSClusterFinder;
+class AliITSLoader;
 class AliITSsegmentation;
 class AliITSresponse;
 class AliITShit;
@@ -32,6 +33,9 @@ class AliITSdigit;
 class AliITSRecPoint;
 class AliITSRawCluster;
 class AliITSmodule;
+class AliVertexer;
+class AliDigitizer;
+class AliRunDigitizer;
 
 const Int_t kNTYPES=3;
 
@@ -42,18 +46,19 @@ class AliITS : public AliDetector {
     AliITS();  // Default creator.
     AliITS(const char *name, const char *title); // standard Creator
     virtual ~AliITS(); // destructor
-    AliITS(AliITS &source); // copy constructor. Not to be used!
+    AliITS(const AliITS &source); // copy constructor. Not to be used!
     AliITS& operator=(AliITS &source); // = operator. Not to be used!
     virtual Int_t IsVersion() const {return 1;}
-    virtual Int_t DistancetoPrimitive(Int_t px, Int_t py);
+    virtual Int_t DistancetoPrimitive(Int_t,Int_t) const{return 999;};
 
     //===================== Simulation Geometry ========================
     // get geometry version - detailed (major) or coarse (minor)
-    virtual Int_t GetMajorVersion(){return -1;}
-    virtual Int_t GetMinorVersion(){return -1;}
+    virtual Int_t GetMajorVersion() const {return -1;}
+    virtual Int_t GetMinorVersion() const {return -1;}
     virtual void  GetGeometryVersion(Int_t &a,Int_t &b) 
                           {a = GetMajorVersion();b=GetMinorVersion();return;}
-    virtual void  SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;}
+    virtual void  SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
+    virtual Bool_t GetEUCLID()const {return fEuclidOut;}
     //-------------------- Geometry Transformations --------------------
     // ITS geometry functions
     AliITSgeom   *GetITSgeom() const {return fITSgeom;}
@@ -65,17 +70,19 @@ class AliITS : public AliDetector {
     AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
                                              (fITSmodules->At(index));}
 
-    //================ Nessesary general Classes =======================
+    //================ Necessary general Classes =======================
     virtual void Init();
+    virtual AliLoader* MakeLoader(const char* topfoldername);
     virtual void SetDefaults();
     virtual void SetDefaultSimulation();
     virtual void SetDefaultClusterFinders();
-    virtual void MakeBranch(Option_t *opt=" ", const char *file=0);
+    virtual void MakeBranch(Option_t *opt=" ");
     virtual void SetTreeAddress();
-    // For a give branch from the treeH sets the TClonesArray address.
-    virtual void SetHitsAddressBranch(TBranch *b){b->SetAddress(&fHits);}
+    // For a given branch from the treeH sets the TClonesArray address.
+    virtual void SetHitsAddressBranch(TBranch *b) {b->SetAddress(&fHits);}
     // Return pointer to DetType #id
-    AliITSDetType *DetType(Int_t id);
+    AliITSDetType *DetType(Int_t id){
+        return ((AliITSDetType*) fDetTypes->At(id));};
     //Int_t           NDetTypes() {return fNDetTypes;}
     //---------- Configuration Methods (per detector type) -------------
     // Determines which ITS subdetectors will be processed. Effects
@@ -84,27 +91,33 @@ class AliITS : public AliDetector {
     // Returns the list of ITS subdetectors that will be processed.
     Option_t* GetDetectors(){return fOpt;}
     // Set response 
-    virtual void SetResponseModel(Int_t id, AliITSresponse *response);
+    virtual void SetResponseModel(Int_t id, AliITSresponse *response){
+        ((AliITSDetType*) fDetTypes->At(id))->ResponseModel(response);};
     // Set segmentation 
-    virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
+    virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg){
+        ((AliITSDetType*) fDetTypes->At(id))->SegmentationModel(seg);};
     // Set simulation - temporary 
-    virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim);
+    virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim){
+        ((AliITSDetType*) fDetTypes->At(id))->SimulationModel(sim);};
     // Set simulation - temporary 
     virtual AliITSsimulation* GetSimulationModel(Int_t id){
        return ((AliITSDetType*)(fDetTypes->At(id)))->GetSimulationModel();}
     // Set reconstruction 
-    virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
+    virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec){
+        ((AliITSDetType*) fDetTypes->At(id))->ReconstructionModel(rec);};
     // Set class names for digit and rec point 
-    virtual void SetClasses(Int_t id, const char *digit, const char *cluster);
+    virtual void SetClasses(Int_t id, const char *digit, const char *cluster){
+        ((AliITSDetType*) fDetTypes->At(id))->ClassNames(digit,cluster);};
 
     //=================== Hits =========================================
     virtual void StepManager() {} // See Step Manager for specific geometry.
     virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits);
     //------------ sort hits by module for Digitisation ----------------
     virtual void InitModules(Int_t size,Int_t &nmodules);  
+    virtual void FillModules(TTree *treeH, Int_t mask = 0);
     virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
-                            Option_t *opt,Text_t *filename);
-    virtual void ClearModules();
+                            Option_t *opt, const char *filename);
+    virtual void ClearModules(){if(fITSmodules) fITSmodules->Delete();};
 
     //===================== Digitisation ===============================
     void MakeBranchS(const char *file);
@@ -112,24 +125,30 @@ class AliITS : public AliDetector {
     TClonesArray * GetSDigits() { return fSDigits; }
     void MakeBranchInTreeD(TTree *treeD,const char *file=0);
     void MakeBranchD(const char *file){
-       MakeBranchInTreeD(gAlice->TreeD(),file);}
+       MakeBranchInTreeD(GetLoader()->TreeD(),file);}
     void SetTreeAddressD(TTree *treeD);
     void Hits2SDigits(); // Turn hits into SDigits
-    void Hits2PreDigits(); // Turn hits into SDigits
+    void Hits2PreDigits(){ // Turn hits into SDigits
+        HitsToPreDigits(fLoader->GetEvent(),
+            /*fLoader->GetRunLoader()->GetHeader()->GetEvent(),*/
+                        0,-1," ",fOpt," ");};
+    AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const;
     void SDigits2Digits(){SDigitsToDigits("All");} // Turn SDigits to Digits
     void SDigitsToDigits(Option_t *opt="All"); // Turn SDigits to Digits
     void Hits2Digits(); // Turn hits straight into Digits.
     //------------------ Internal functions ----------------------------
     // Standard Hits To SDigits function
     void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
-                 Option_t *add, Option_t *det, Text_t *filename);
+                       Option_t *add, Option_t *det, const char *filename)
+        {HitsToPreDigits(evNumber,bgrev,size,add,det,filename);};
     // Standard Hits To SDigits function
     void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
-                 Option_t *add, Option_t *det, Text_t *filename);
+                 Option_t *add, Option_t *det, const char *filename);
     // Standard Hits To Digits function
     void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
-                 Option_t *add, Option_t *det, Text_t *filename);
-    void ResetSDigits();                  // Resets the Summable digits.
+                 Option_t *add, Option_t *det, const char *filename);
+    // Resets the Summable digits.
+    void ResetSDigits(){if(fSDigits) fSDigits->Clear();fNSDigits = 0;};
     void ResetDigits();                   // depending on how the
     void ResetDigits(Int_t branch);       // tree will be filled only
     void AddSumDigit(AliITSpListItem &sdig);
@@ -142,22 +161,20 @@ class AliITS : public AliDetector {
     Int_t        *Ndtype() {return fNdtype;}
     TClonesArray *DigitsAddress(Int_t id)
        {return ((TClonesArray *) (*fDtype)[id]);}
+    void SelectVertexer(TString sel=" "){fSelectedVertexer = sel;}
 
     //===================== Raw Data IO ================================
     // Write digits into raw data format
-    virtual void Digits2RawData() {}
-    // Decode raw data and store digits
-    virtual void RawData2Digits() {}
+    virtual void Digits2Raw();
 
     //==================== Clusterization ==============================
     // create separate tree for clusters - declustering refining
     void MakeTreeC(Option_t *option="C");
     void GetTreeC(Int_t event);
     void AddCluster(Int_t branch, AliITSRawCluster *c);
-    void ResetClusters();                 // one of the methods in 
+    // one of the methods in
+    void ResetClusters(){for(Int_t i=0;i<kNTYPES;i++ ) ResetClusters(i);}; 
     void ResetClusters(Int_t branch);     // the pair will be kept
-    // Return pointer to the tree of clusters
-    TTree        *TreeC() {return fTreeC;}
     // Return pointers to clusters 
     TObjArray    *Ctype() {return fCtype;}
     Int_t        *Nctype() {return fNctype;}
@@ -167,13 +184,14 @@ class AliITS : public AliDetector {
     //=================== Reconstruction ===============================
     void MakeBranchR(const char *file, Option_t *opt=" ");
     void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");}
+    void MakeBranchC();
     void SetTreeAddressR(TTree *treeR);
     void AddRecPoint(const AliITSRecPoint &p);
     void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
-                 Option_t *add, Option_t *det, Text_t *filename);
-    void Digits2Reco();
+                 Option_t *add, Option_t *det, const char *filename);
+    void Digits2Reco(){DigitsToRecPoints(fLoader->GetEvent(),0,fOpt);};
     void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
-    void ResetRecPoints();
+    void ResetRecPoints(){if(fRecPoints) fRecPoints->Clear();fNRecPoints = 0;};
     // Return pointer to rec points 
     TClonesArray  *RecPoints()   {return fRecPoints;}
 
@@ -191,20 +209,20 @@ class AliITS : public AliDetector {
     Int_t         fNDetTypes;  // Number of detector types
     TObjArray    *fDetTypes;   // List of detector types
 
-    TClonesArray  *fSDigits;    // List of Summable digits.
+    TClonesArray  *fSDigits;    //! List of Summable digits.
     Int_t         fNSDigits;   // Number of Summable Digits.
 
-    TObjArray    *fDtype;      // List of digits
+    TObjArray    *fDtype;      //! List of digits
     Int_t        *fNdtype;     //[fNDetTypes] Num. of digits per type of det. 
 
-    TObjArray    *fCtype;      // List of clusters
+    TObjArray    *fCtype;      //! List of clusters
     Int_t        *fNctype;     //[fNDetTypes] Num. of clust. per type of det.
-    TTree        *fTreeC;      // Tree for raw clusters
 
-    TClonesArray *fRecPoints;  // List of reconstructed points
+    TClonesArray *fRecPoints;  //! List of reconstructed points
     Int_t         fNRecPoints; // Number of rec points
+    TString fSelectedVertexer; // Vertexer selected in CreateVertexer
 
-    ClassDef(AliITS,1) // Base class for ITS
+    ClassDef(AliITS,4) // Base class for ITS
 };
 
 #endif