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
  14 #include "AliDetector.h"
  15 #include "AliITSsimulationSPD.h"
  16 #include "AliITSsimulationSDD.h"
  17 #include "AliITSsimulationSSD.h"
  18
  19 class TString;
  20 class TTree;
  21 class TFile;
  22
  23 class AliITSDetType;
  24 class AliITSsimulation;
  25 class AliITSClusterFinder;
  26 class AliITSsegmentation;
  27 class AliITSresponse;
  28 class AliITShit;
  29 class AliITSgeom;
  30 class AliITSdigit;
  31 class AliITSRecPoint;
  32 class AliITSRawCluster;
  33 class AliITSmodule;
  34 class AliITStrack;
  35
  36
  37
  38 class AliITS : public AliDetector {
  39
  40   public:
  41                     AliITS();
  42                     AliITS(const char *name, const char *title);
  43     virtual        ~AliITS();
  44     AliITS(AliITS &source);
  45     AliITS & operator=(AliITS &source);
  46
  47     virtual void   AddHit(Int_t track, Int_t *vol, Float_t *hits);
  48     virtual void   AddRealDigit(Int_t branch, Int_t *digits);
  49     virtual void   AddSimDigit(Int_t branch, AliITSdigit *d);
  50     virtual void   AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,Int_t* tracks,Int_t *hits,Float_t* trkcharges);
  51     //
  52     virtual void   AddCluster(Int_t branch, AliITSRawCluster *c);
  53     virtual void   AddRecPoint(const AliITSRecPoint &p);
  54
  55     virtual void   ResetDigits();                   // depending on how the
  56     virtual void   ResetDigits(Int_t branch);       // tree will be filled only
  57     virtual void   ResetClusters();                 // one of the methods in
  58     virtual void   ResetClusters(Int_t branch);     // the pair will be kept
  59     virtual void   ResetRecPoints();
  60
  61     // get geometry version - detailed (major) or coarse (minor)
  62     void GetGeometryVersion(Int_t &a,Int_t &b) const
  63                    {a = fMajorVersion;b=fMinorVersion;return;}
  64     virtual Int_t  IsVersion() const {return 1;}
  65     virtual Int_t  DistancetoPrimitive(Int_t px, Int_t py);
  66     virtual void   Init();
  67     virtual void   SetDefaults();
  68     virtual void   SetDefaultSimulation();
  69     virtual void   SetDefaultClusterFinders();
  70     // create separate tree for clusters - declustering refining
  71     virtual  void  MakeTreeC(Option_t *option="C");
  72     void           GetTreeC(Int_t event);
  73     virtual void   MakeBranch(Option_t *opt=" ", char *file=0);
  74     void           SetTreeAddress();
  75     virtual void   SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;}
  76     virtual void   StepManager() {}
  77     // sort hits by module
  78     virtual void   InitModules(Int_t size,Int_t &nmodules);
  79     virtual void   FillModules(Int_t evnt,Int_t bgrev,
  80                        Int_t nmodules,Option_t *opt,Text_t *filename);
  81     virtual void   ClearModules();
  82     // Digitisation
  83     virtual void   SDigits2Digits();
  84     void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
  85                  Option_t *add, Option_t *det, Text_t *filename);
  86     // Reconstruct hits
  87     void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
  88     // Fast simulation of space points from hits
  89     void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
  90                  Option_t *add, Option_t *det, Text_t *filename);
  91
  92     // Write digits into raw data format
  93     virtual void Digits2RawData() {}
  94     // Decode raw data and store digits
  95     virtual void RawData2Digits() {}
  96
  97     // Configuration Methods (per detector type )
  98     // Set response
  99     virtual void   SetResponseModel(Int_t id, AliITSresponse *response);
 100     // Set segmentation
 101     virtual void   SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
 102     // Set simulation - temporary
 103     virtual void   SetSimulationModel(Int_t id, AliITSsimulation *sim);
 104     // Set reconstruction
 105     virtual void   SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
 106     // Set class names for digit and rec point
 107     virtual void   SetClasses(Int_t id, const char *digit, const char *cluster);
 108
 109
 110     // Getters
 111     // ITS geometry functions
 112     virtual AliITSgeom  *GetITSgeom() const {return fITSgeom;}
 113     // return pointer to the array of modules
 114     virtual TObjArray   *GetModules() const {return fITSmodules;}
 115     // return pointer to a particular module
 116     AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
 117                                               (fITSmodules->At(index));}
 118
 119     // Return pointers to digits
 120     TObjArray            *Dtype() {return fDtype;}
 121     Int_t                *Ndtype() {return fNdtype;}
 122     virtual TClonesArray *DigitsAddress(Int_t id)
 123                    {return ((TClonesArray *) (*fDtype)[id]);}
 124     // Return pointers to clusters
 125     TObjArray            *Ctype() {return fCtype;}
 126     Int_t                *Nctype() {return fNctype;}
 127     virtual TClonesArray *ClustersAddress(Int_t id)
 128                    {return ((TClonesArray *) (*fCtype)[id]);}
 129
 130     // Return pointer to rec points
 131     TClonesArray  *RecPoints()   {return fRecPoints;}
 132
 133     // Return pointer to DetType #id
 134     AliITSDetType  *DetType(Int_t id);
 135     //Int_t           NDetTypes() {return fNDetTypes;}
 136
 137     // Return pointer to the tree of clusters
 138     TTree          *TreeC() {return fTreeC;}
 139
 140
 141     // tracking
 142
 143     AliITStrack Tracking(AliITStrack &track, AliITStrack *reference, TObjArray *fpoints, Int_t **vettid,
 144          Bool_t flagvert );
 145
 146     void DoTracking(Int_t evNumber, Int_t min_t, Int_t max_t, TFile *file, Bool_t flagvert);
 147
 148  protected:
 149
 150     static const Int_t fgkNTYPES;  // Number of detector types
 151     AliITSgeom *fITSgeom;      // Pointer to ITS geometry
 152     TObjArray  *fITSmodules;   //! Pointer to ITS modules
 153     Bool_t      fEuclidOut;    // Flag to write geometry in euclid format
 154     Int_t       fIdN;          // the number of layers
 155     Int_t      *fIdSens;       //[fIdN] layer identifier
 156 //    TObjArray  *fIdName;       // array of volume Id names
 157     TString  *fIdName;       //[fIdN] layer identifier
 158     // Geometry and Stepmanager version numbers used.
 159     Int_t fMajorVersion;      // detailed and coarse(minor) versions
 160     Int_t fMinorVersion;      // detailed and coarse(minor) versions
 161     //
 162     // Int_t       fNDetTypes;   // Number of detector types
 163     TObjArray     *fDetTypes;    // List of detector types
 164
 165     TObjArray     *fDtype;       //[fgkNTYPES]  List of digits
 166     Int_t         *fNdtype;      //[fgkNTYPES] Num. of digits per type of det.
 167     TObjArray     *fCtype;       //[fgkNTYPES] List of clusters
 168     Int_t         *fNctype;      //[fgkNTYPES] Num. of clust. per type of det.
 169
 170     TClonesArray  *fRecPoints;   // List of reconstructed points
 171     Int_t          fNRecPoints;  // Number of rec points
 172     TTree         *fTreeC;       // Tree for raw clusters
 173
 174
 175     ClassDef(AliITS,1) // Base class for ITS
 176 };
 177
 178 #endif