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
  13 #include <TObjArray.h> // used in inline function GetModule.
  14 #include "AliDetector.h"
  15 #include "AliITSTrigger.h"
  16 #include "AliITSDetTypeSim.h"
  17
  18
  19 class TString;
  20 class TTree;
  21 class AliITSpListItem;
  22 class AliITSsimulation;
  23 class AliITSsegmentation;
  24
  25 class AliITSCalibration;
  26 class AliITShit;
  27 class AliITSgeom;
  28 class AliITSdigit;
  29 class AliITSmodule;
  30 class AliRunDigitizer;
  31
  32
  33 class AliITS : public AliDetector {
  34
  35  public:
  36     //================= Standard Classes ===============================
  37     AliITS();  // Default creator.
  38     AliITS(const Char_t *title); // standard Creator
  39     AliITS(const char *name, const char *title); // extended standard Creator
  40     virtual ~AliITS(); // destructor
  41     virtual Int_t IsVersion() const {return 1;}
  42
  43     //===================== Simulation Geometry ========================
  44     // get geometry version - detailed (major) or coarse (minor)
  45     virtual Int_t GetMajorVersion() const {return -1;}
  46     virtual Int_t GetMinorVersion() const {return -1;}
  47     virtual void  GetGeometryVersion(Int_t &a,Int_t &b) const
  48                            {a = GetMajorVersion();b=GetMinorVersion();return;}
  49     virtual void  SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
  50     virtual Bool_t GetEUCLID()const {return fEuclidOut;}
  51     //-------------------- Geometry Transformations --------------------
  52
  53     // ITS geometry functions From Simulation
  54     AliITSgeom* GetITSgeom() const {return fDetTypeSim->GetITSgeom();}
  55     void   SetITSgeom(AliITSgeom *geom) {fDetTypeSim->SetITSgeom(geom);}
  56     // return pointer to the array of modules
  57     TObjArray *GetModules(){return fITSmodules;}
  58
  59     AliITSmodule *GetModule(Int_t index){
  60         return (AliITSmodule*)(fITSmodules->At(index));}
  61     virtual void SetSimuParam(AliITSSimuParam *sp){
  62       fSimuParam=sp;
  63       fDetTypeSim->SetSimuParam(sp);
  64     }
  65     AliITSSimuParam* GetSimuParam() const {return fSimuParam;}
  66
  67     virtual void SetDetTypeSim(AliITSDetTypeSim* dts) {fDetTypeSim=dts;}
  68     AliITSDetTypeSim* GetDetTypeSim() const {return fDetTypeSim;}
  69     //================ Necessary general Classes =======================
  70     virtual void Init();
  71     virtual AliLoader* MakeLoader(const char* topfoldername);
  72     virtual void SetDefaults();
  73     virtual void SetDefaultSimulation();
  74     virtual void MakeBranch(Option_t *opt=" ");
  75     virtual void MakeBranchS(const char* fl);
  76     virtual void MakeBranchD(const char* file);
  77     virtual void MakeBranchInTreeD(TTree* treeD, const char* file=0);
  78     virtual void SetTreeAddress();
  79      //---------- Configuration Methods (per detector type) -------------
  80     // Determines which ITS subdetectors will be processed. Effects
  81     // digitization, and Reconstruction only.
  82     void SetDetectors(Option_t *opt="All"){fOpt = opt;}
  83     // Returns the list of ITS subdetectors that will be processed.
  84     Option_t* GetDetectors(){return fOpt;}
  85
  86     // Set calibration
  87     virtual void SetCalibrationModel(Int_t dettype, AliITSCalibration *cal){
  88         fDetTypeSim->SetCalibrationModel(dettype,cal);}
  89     // Set segmentation for Simulation
  90     virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg){
  91         fDetTypeSim->SetSegmentationModel(id,seg);}
  92     // Set simulation
  93     virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim){
  94         fDetTypeSim->SetSimulationModel(id,sim);}
  95     // Set simulation
  96     virtual AliITSsimulation* GetSimulationModel(Int_t id){
  97         return fDetTypeSim->GetSimulationModel(id);}
  98     //=================== Hits =========================================
  99     virtual void StepManager() {} // See Step Manager for specific geometry.
 100     //------------ sort hits by module for Digitisation ----------------
 101     virtual void FillModules(Int_t /* evnt */,Int_t bgrev,Int_t /* nmodules */,
 102                              Option_t *opt, const char *filename);
 103     virtual Bool_t InitModules(Int_t size,Int_t &nmodules);
 104     virtual void FillModules(TTree *treeH, Int_t mask = 0);
 105     virtual void ClearModules(){if(fITSmodules) fITSmodules->Delete();}
 106     virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits);
 107    // Trigger
 108     virtual AliTriggerDetector* CreateTriggerDetector() const;
 109
 110     virtual void UpdateInternalGeometry();
 111     virtual void SDigitsToDigits(Option_t *opt="All");
 112     virtual void SDigits2Digits(){SDigitsToDigits("All");}
 113     virtual void Hits2Digits();
 114     virtual void Hits2SDigits();
 115     virtual void Hits2PreDigits();
 116     virtual void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 117                           Option_t *option,Option_t *opt,
 118                           const char *filename);
 119     virtual void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 120                              Option_t *option,Option_t *opt,
 121                              const char *filename);
 122     void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
 123                        Option_t *add, Option_t *det, const char *filename)
 124       {HitsToPreDigits(evNumber,bgrev,size,add,det,filename);}
 125
 126     virtual void ResetDigits();
 127     virtual void ResetDigits(Int_t branch);
 128     virtual void AddSumDigit(AliITSpListItem &sdig);
 129     virtual void AddSimDigit(Int_t branch, AliITSdigit *d);
 130     virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
 131                      Int_t* tracks,Int_t *hits,Float_t* trkcharges,
 132                      Int_t sigexpanded=-1000);
 133     TObjArray* GetDigits()  const {return fDetTypeSim->GetDigits();}
 134     Int_t* GetNDigitArray() const {return fDetTypeSim->GetNDigitArray();}
 135     TClonesArray *DigitsAddress(Int_t id) {
 136         return fDetTypeSim->DigitsAddress(id);}
 137     //Fast simulation
 138     virtual void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
 139                  Option_t *add, Option_t *det, const char *filename);
 140     virtual Int_t Hits2Clusters(TTree *in, TTree *out);
 141     virtual void CheckLabels(Int_t lab[3]) const;
 142
 143     //===================== Raw Data IO ================================
 144     // Write digits into raw data format
 145     virtual void   Digits2Raw();
 146     virtual Bool_t Raw2SDigits(AliRawReader*);
 147
 148
 149  protected:
 150     static const Int_t fgkNTYPES=3; //number of detector types
 151     //================== Data Members ==================================
 152     AliITSDetTypeSim *fDetTypeSim; //detector type for simulation
 153     Bool_t        fEuclidOut;  // Flag to write geometry in euclid format
 154     Option_t     *fOpt;        //! Detector option ="All" unless changed.
 155     Int_t         fIdN;        // the number of layers
 156     Int_t        *fIdSens;     //[fIdN] layer identifier
 157     TString      *fIdName;     //[fIdN] layer identifier
 158     TObjArray    *fITSmodules; //! Pointer to ITS modules
 159     Bool_t        fTiming;     // flag to turn on/off timers.
 160     AliITSSimuParam* fSimuParam; //simulation parameters
 161     TClonesArray** fModA;      //! Used by Raw2SDigits (one TC per module)
 162     TClonesArray* fpSDigits;   //! Branch address to build SD from raw data
 163
 164  private:
 165     AliITS(const AliITS &source); // copy constructor. Not to be used!
 166     AliITS& operator=(const AliITS &source); // = operator. Not to be used!
 167     ClassDef(AliITS,9) // Base class for ITS
 168
 169 };
 170
 171 #endif