ITS/AliITSDetTypeSim.h

   1 #ifndef ALIITSDETTYPESIM_H
   2 #define ALIITSDETTYPESIM_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /*
   7 $Id$
   8 */
   9
  10 /////////////////////////////////////////////////////////////////////////
  11 // * This class contains all of the "external" information needed to do//
  12 // * detector specific simulations for the ITS.                        //
  13 /////////////////////////////////////////////////////////////////////////
  14
  15 #include <TObject.h>
  16 #include <TObjArray.h>
  17 #include <TClonesArray.h>
  18 #include "AliITSCalibration.h"
  19 #include "AliITSSimuParam.h"
  20 #include "AliITSDDLModuleMapSDD.h"
  21 #include "AliITSLoader.h"
  22 #include "AliITSgeom.h"
  23
  24 class TTree;
  25 class AliCDBMetaData;
  26 class AliITSdigit;
  27 class AliITSmodule;
  28 class AliITSpListItem;
  29 class AliITSsimulation;
  30 class AliITSsegmentation;
  31 class AliITSresponse;
  32 class AliITSCalibrationSSD;
  33
  34 class AliITSDetTypeSim : public TObject {
  35  public:
  36
  37     AliITSDetTypeSim();
  38     virtual ~AliITSDetTypeSim();
  39     AliITSDetTypeSim(const AliITSDetTypeSim &source);
  40     AliITSDetTypeSim& operator=(const AliITSDetTypeSim &source);
  41     AliITSgeom *GetITSgeom() const {
  42         if(fLoader)return ((AliITSLoader*)fLoader)->GetITSgeom();
  43         else return 0;}
  44     void SetITSgeom(AliITSgeom *geom);
  45
  46     virtual void SetSimulationModel(Int_t dettype,AliITSsimulation *sim);
  47     virtual AliITSsimulation* GetSimulationModel(Int_t dettype);
  48     virtual AliITSsimulation* GetSimulationModelByModule(Int_t module);
  49
  50     virtual void SetSegmentationModel(Int_t dettype,AliITSsegmentation *seg);
  51     virtual AliITSsegmentation* GetSegmentationModel(Int_t dettype);
  52     virtual AliITSsegmentation* GetSegmentationModelByModule(Int_t module);
  53
  54     virtual void SetCalibrationModel(Int_t iMod,AliITSCalibration *resp);
  55     virtual AliITSCalibration* GetCalibrationModel(Int_t iMod);
  56     virtual AliITSresponse* GetResponse(Int_t dettype) {
  57         return GetCalibrationModel(
  58             GetITSgeom()->GetStartDet(dettype))->GetResponse();}
  59
  60     virtual void SetSimuParam(AliITSSimuParam* spar){
  61       if(fSimuPar) delete fSimuPar;
  62       fSimuPar = new AliITSSimuParam(*spar);
  63     }
  64     virtual AliITSSimuParam* GetSimuParam() const {return fSimuPar;}
  65
  66     virtual AliITSDDLModuleMapSDD* GetDDLModuleMapSDD()const { return fDDLMapSDD;}
  67     TObjArray* GetCalibrationArray() const {return fCalibration;}
  68     TObjArray* GetSegmentation() const {return fSegmentation;}
  69     void ResetCalibrationArray();
  70     void ResetSegmentation();
  71
  72     virtual void SetLoader(AliITSLoader* loader);
  73     AliITSLoader* GetLoader() const {return fLoader;}
  74
  75     virtual void SetDefaults();
  76     virtual void SetDefaultSimulation();
  77     virtual void SetRunNumber(Int_t rn=0){fRunNumber = rn;}
  78     virtual Int_t GetRunNumber() const {return fRunNumber;}
  79     virtual void SetTreeAddressS(TTree* treeS, Char_t* name);
  80     virtual void SetTreeAddressD(TTree* treeD, Char_t* name);
  81
  82     virtual void SetDigits(TObjArray* digits) {fDigits=digits;}
  83     const TClonesArray* GetSDigits() const { return &fSDigits;}
  84     TObjArray*    GetDigits() const {return fDigits;}
  85     Int_t* GetNDigitArray() const {return fNDigits;}
  86     TClonesArray *DigitsAddress(Int_t id) const {
  87         return ((TClonesArray*)(*fDigits)[id]);}
  88     virtual void ResetSDigits(){fNSDigits=0; fSDigits.Clear();}
  89     virtual void ResetDigits();
  90     virtual void ResetDigits(Int_t branch);
  91     virtual void SDigitsToDigits(Option_t *opt,Char_t* name);
  92
  93     virtual void AddSumDigit(AliITSpListItem &sdig);
  94     virtual void AddSimDigit(Int_t branch, AliITSdigit *d);
  95     virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
  96                              Int_t* tracks,Int_t *hits,Float_t* trkcharges,
  97                              Int_t sigexpanded=-1000);
  98     virtual void SetDigitClassName(Int_t i, Char_t* name) {
  99         fDigClassName[i]=name;}
 100     Char_t* GetDigitClassName(Int_t i) const {return fDigClassName[i];}
 101     void StoreCalibration(Int_t firstRun, Int_t lastRun, AliCDBMetaData &md);
 102
 103  protected:
 104     virtual void CreateCalibrationArray();
 105     virtual Bool_t GetCalibration();
 106
 107  private:
 108     void SetDefaultSegmentation(Int_t idet);  // creates def segm.
 109     static const Int_t fgkNdettypes;          // number of different det. types
 110     static const Int_t fgkDefaultNModulesSPD; // Total numbers of SPD modules by default
 111     static const Int_t fgkDefaultNModulesSDD; // Total numbers of SDD modules by default
 112     static const Int_t fgkDefaultNModulesSSD; // Total numbers of SSD modules by default
 113     Int_t fNMod[3];                           // numbers of modules from different types
 114
 115     TObjArray    *fSimulation;   //! [NDet]
 116     TObjArray    *fSegmentation; //! [NDet]
 117     TObjArray    *fCalibration;  //! [NMod]
 118     AliITSCalibrationSSD* fSSDCalibration;  //! SSD calibration object
 119     TObjArray    *fPreProcess;   //! [] e.g. Fill fHitModule with hits
 120     TObjArray    *fPostProcess;  //! [] e.g. Wright Raw data
 121     Int_t         fNSDigits;     //! number of SDigits
 122     TClonesArray  fSDigits;      //! Summable digits
 123     Int_t*        fNDigits;      //! [NDet] number of Digits for det.
 124     Int_t         fRunNumber;    //! run number (to access DB)
 125     TObjArray     *fDigits;      //! [NMod][NDigits]
 126     AliITSSimuParam *fSimuPar;   //! detector simulation parameters
 127     AliITSDDLModuleMapSDD *fDDLMapSDD; //! mapping DDL/module -> SDD module number
 128     TString       fHitClassName; //! String with Hit class name
 129     TString       fSDigClassName;//! String with SDigit class name.
 130     Char_t*       fDigClassName[3]; //! String with digit class name.
 131     AliITSLoader* fLoader;          //! loader
 132     Bool_t        fFirstcall;       //! flag
 133
 134     ClassDef(AliITSDetTypeSim,7) // ITS Simulation structure
 135
 136 };
 137
 138 #endif