ITS/AliITS.h

   1 #ifndef ALIITS_H
   2 #define ALIITS_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ////////////////////////////////////////////////////////////////////////
   9 //           Manager class for set: ITS                               //
  10 ////////////////////////////////////////////////////////////////////////
  11
  12 #include <TObjArray.h> // used in inline function GetModule.
  13 #include <TBranch.h>   // used in inline function SetHitsAddressBranch
  14
  15 #include "AliLoader.h"
  16 #include "AliDetector.h"
  17 #include "AliITSDetType.h"
  18
  19 class TString;
  20 class TTree;
  21 class TFile;
  22
  23 //class AliITSDetType;
  24 class AliITSsimulation;
  25 class AliITSClusterFinder;
  26 class AliITSLoader;
  27 class AliITSsegmentation;
  28 class AliITSresponse;
  29 class AliITShit;
  30 class AliITSgeom;
  31 class AliITSpListItem;
  32 class AliITSdigit;
  33 class AliITSRecPoint;
  34 class AliITSRawCluster;
  35 class AliITSmodule;
  36 class AliVertexer;
  37
  38 const Int_t kNTYPES=3;
  39
  40 class AliITS : public AliDetector {
  41
  42  public:
  43     //================= Standard Classes ===============================
  44     AliITS();  // Default creator.
  45     AliITS(const char *name, const char *title); // standard Creator
  46     virtual ~AliITS(); // destructor
  47     AliITS(const AliITS &source); // copy constructor. Not to be used!
  48     AliITS& operator=(AliITS &source); // = operator. Not to be used!
  49     virtual Int_t IsVersion() const {return 1;}
  50     virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) const;
  51
  52     //===================== Simulation Geometry ========================
  53     // get geometry version - detailed (major) or coarse (minor)
  54     virtual Int_t GetMajorVersion() const {return -1;}
  55     virtual Int_t GetMinorVersion() const {return -1;}
  56     virtual void  GetGeometryVersion(Int_t &a,Int_t &b)
  57                            {a = GetMajorVersion();b=GetMinorVersion();return;}
  58     virtual void  SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
  59     virtual Bool_t GetEUCLID()const {return fEuclidOut;}
  60     //-------------------- Geometry Transformations --------------------
  61     // ITS geometry functions
  62     AliITSgeom   *GetITSgeom() const {return fITSgeom;}
  63     // Sets ITS geometry ! be very careful using this function.
  64     void   SetITSgeom(AliITSgeom *geom) {fITSgeom = geom;}
  65     // return pointer to the array of modules
  66     TObjArray    *GetModules() const {return fITSmodules;}
  67     // return pointer to a particular module
  68     AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
  69                                               (fITSmodules->At(index));}
  70
  71     //================ Necessary general Classes =======================
  72     virtual void Init();
  73     virtual AliLoader* MakeLoader(const char* topfoldername);
  74     virtual void SetDefaults();
  75     virtual void SetDefaultSimulation();
  76     virtual void SetDefaultClusterFinders();
  77     virtual void MakeBranch(Option_t *opt=" ");
  78     virtual void SetTreeAddress();
  79     // For a given branch from the treeH sets the TClonesArray address.
  80     virtual void SetHitsAddressBranch(TBranch *b) {b->SetAddress(&fHits);}
  81     // Return pointer to DetType #id
  82     AliITSDetType *DetType(Int_t id);
  83     //Int_t           NDetTypes() {return fNDetTypes;}
  84     //---------- Configuration Methods (per detector type) -------------
  85     // Determines which ITS subdetectors will be processed. Effects
  86     // digitization, and Reconstruction only.
  87     void SetDetectors(Option_t *opt="All"){fOpt = opt;}
  88     // Returns the list of ITS subdetectors that will be processed.
  89     Option_t* GetDetectors(){return fOpt;}
  90     // Set response
  91     virtual void SetResponseModel(Int_t id, AliITSresponse *response);
  92     // Set segmentation
  93     virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
  94     // Set simulation - temporary
  95     virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim);
  96     // Set simulation - temporary
  97     virtual AliITSsimulation* GetSimulationModel(Int_t id){
  98         return ((AliITSDetType*)(fDetTypes->At(id)))->GetSimulationModel();}
  99     // Set reconstruction
 100     virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
 101     // Set class names for digit and rec point
 102     virtual void SetClasses(Int_t id, const char *digit, const char *cluster);
 103
 104     //=================== Hits =========================================
 105     virtual void StepManager() {} // See Step Manager for specific geometry.
 106     virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits);
 107     //------------ sort hits by module for Digitisation ----------------
 108     virtual void InitModules(Int_t size,Int_t &nmodules);
 109     virtual void FillModules(TTree *treeH, Int_t mask = 0);
 110     virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
 111                              Option_t *opt, const char *filename);
 112     virtual void ClearModules();
 113
 114     //===================== Digitisation ===============================
 115     void MakeBranchS(const char *file);
 116     void SetTreeAddressS(TTree *treeS);
 117     TClonesArray * GetSDigits() { return fSDigits; }
 118     void MakeBranchInTreeD(TTree *treeD,const char *file=0);
 119     void MakeBranchD(const char *file){
 120         MakeBranchInTreeD(GetLoader()->TreeD(),file);}
 121     void SetTreeAddressD(TTree *treeD);
 122     void Hits2SDigits(); // Turn hits into SDigits
 123     void Hits2PreDigits(); // Turn hits into SDigits
 124     AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const;
 125     void SDigits2Digits(){SDigitsToDigits("All");} // Turn SDigits to Digits
 126     void SDigitsToDigits(Option_t *opt="All"); // Turn SDigits to Digits
 127     void Hits2Digits(); // Turn hits straight into Digits.
 128     //------------------ Internal functions ----------------------------
 129     // Standard Hits To SDigits function
 130     void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 131                  Option_t *add, Option_t *det, const char *filename);
 132     // Standard Hits To SDigits function
 133     void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 134                  Option_t *add, Option_t *det, const char *filename);
 135     // Standard Hits To Digits function
 136     void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 137                  Option_t *add, Option_t *det, const char *filename);
 138     void ResetSDigits();                  // Resets the Summable digits.
 139     void ResetDigits();                   // depending on how the
 140     void ResetDigits(Int_t branch);       // tree will be filled only
 141     void AddSumDigit(AliITSpListItem &sdig);
 142     void AddRealDigit(Int_t branch, Int_t *digits);
 143     void AddSimDigit(Int_t branch, AliITSdigit *d);
 144     void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
 145                      Int_t* tracks,Int_t *hits,Float_t* trkcharges);
 146     // Return pointers to digits
 147     TObjArray    *Dtype() {return fDtype;}
 148     Int_t        *Ndtype() {return fNdtype;}
 149     TClonesArray *DigitsAddress(Int_t id)
 150         {return ((TClonesArray *) (*fDtype)[id]);}
 151     void SelectVertexer(TString sel=" "){fSelectedVertexer = sel;}
 152
 153     //===================== Raw Data IO ================================
 154     // Write digits into raw data format
 155     virtual void Digits2Raw();
 156
 157     //==================== Clusterization ==============================
 158     // create separate tree for clusters - declustering refining
 159     void MakeTreeC(Option_t *option="C");
 160     void GetTreeC(Int_t event);
 161     void AddCluster(Int_t branch, AliITSRawCluster *c);
 162     void ResetClusters();                 // one of the methods in
 163     void ResetClusters(Int_t branch);     // the pair will be kept
 164     // Return pointers to clusters
 165     TObjArray    *Ctype() {return fCtype;}
 166     Int_t        *Nctype() {return fNctype;}
 167     TClonesArray *ClustersAddress(Int_t id)
 168                    {return ((TClonesArray *) (*fCtype)[id]);}
 169
 170     //=================== Reconstruction ===============================
 171     void MakeBranchR(const char *file, Option_t *opt=" ");
 172     void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");}
 173     void MakeBranchC();
 174     void SetTreeAddressR(TTree *treeR);
 175     void AddRecPoint(const AliITSRecPoint &p);
 176     void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
 177                  Option_t *add, Option_t *det, const char *filename);
 178     void Digits2Reco();
 179     void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
 180     void ResetRecPoints();
 181     // Return pointer to rec points
 182     TClonesArray  *RecPoints()   {return fRecPoints;}
 183
 184  protected:
 185     //================== Data Members ==================================
 186     AliITSgeom   *fITSgeom;    // Pointer to ITS geometry
 187     Bool_t        fEuclidOut;  // Flag to write geometry in euclid format
 188     TObjArray    *fITSmodules; //! Pointer to ITS modules
 189     Option_t     *fOpt;        //! Detector option ="All" unless changed.
 190
 191     Int_t         fIdN;        // the number of layers
 192     Int_t        *fIdSens;     //[fIdN] layer identifier
 193     TString      *fIdName;     //[fIdN] layer identifier
 194
 195     Int_t         fNDetTypes;  // Number of detector types
 196     TObjArray    *fDetTypes;   // List of detector types
 197
 198     TClonesArray  *fSDigits;    //! List of Summable digits.
 199     Int_t         fNSDigits;   // Number of Summable Digits.
 200
 201     TObjArray    *fDtype;      //! List of digits
 202     Int_t        *fNdtype;     //[fNDetTypes] Num. of digits per type of det.
 203
 204     TObjArray    *fCtype;      //! List of clusters
 205     Int_t        *fNctype;     //[fNDetTypes] Num. of clust. per type of det.
 206
 207     TClonesArray *fRecPoints;  //! List of reconstructed points
 208     Int_t         fNRecPoints; // Number of rec points
 209     TString fSelectedVertexer; // Vertexer selected in CreateVertexer
 210
 211     ClassDef(AliITS,4) // Base class for ITS
 212 };
 213
 214 #endif