]> git.uio.no Git - u/mrichter/AliRoot.git/blob - ITS/AliITSDetTypeSim.h
2471747201e50a504413f086b183f2f8fb33bc79
[u/mrichter/AliRoot.git] / 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 class AliITSGainSSDv2;
34 class AliITSBadChannelsSSDv2;
35 class AliITSNoiseSSDv2;
36
37 class AliITSDetTypeSim : public TObject {
38  public:
39   
40     AliITSDetTypeSim();
41     virtual ~AliITSDetTypeSim(); 
42     AliITSDetTypeSim(const AliITSDetTypeSim &source);
43     AliITSDetTypeSim& operator=(const AliITSDetTypeSim &source);
44     AliITSgeom *GetITSgeom() const {
45         if(fLoader)return ((AliITSLoader*)fLoader)->GetITSgeom();
46         else return 0;}
47     void SetITSgeom(AliITSgeom *geom);
48     
49     virtual void SetSimulationModel(Int_t dettype,AliITSsimulation *sim);
50     virtual AliITSsimulation* GetSimulationModel(Int_t dettype);        
51     virtual AliITSsimulation* GetSimulationModelByModule(Int_t module);
52         
53     virtual void SetSegmentationModel(Int_t dettype,AliITSsegmentation *seg);
54     virtual AliITSsegmentation* GetSegmentationModel(Int_t dettype);
55     virtual AliITSsegmentation* GetSegmentationModelByModule(Int_t module);
56
57     virtual void SetCalibrationModel(Int_t iMod,AliITSCalibration *resp);
58     virtual AliITSCalibration* GetCalibrationModel(Int_t iMod);
59     /* 
60    virtual AliITSresponse* GetResponse(Int_t dettype) {
61         return GetCalibrationModel(
62             GetITSgeom()->GetStartDet(dettype))->GetResponse();}
63     */
64     virtual void SetSimuParam(AliITSSimuParam* spar){
65       if(fSimuPar) delete fSimuPar;
66       fSimuPar = new AliITSSimuParam(*spar);
67     }
68     virtual AliITSSimuParam* GetSimuParam() const {return fSimuPar;}
69
70     virtual AliITSDDLModuleMapSDD* GetDDLModuleMapSDD()const { return fDDLMapSDD;}
71     TObjArray* GetCalibrationArray() const {return fCalibration;}
72     TObjArray* GetSegmentation() const {return fSegmentation;}
73     void ResetCalibrationArray();
74     void ResetSegmentation();
75
76     virtual void SetLoader(AliITSLoader* loader);
77     AliITSLoader* GetLoader() const {return fLoader;}
78
79     virtual void SetDefaults();
80     virtual void SetDefaultSimulation();
81     virtual void SetRunNumber(Int_t rn=0){fRunNumber = rn;}
82     virtual Int_t GetRunNumber() const {return fRunNumber;}
83     virtual void SetTreeAddressS(TTree* treeS, Char_t* name);
84     virtual void SetTreeAddressD(TTree* treeD, Char_t* name);
85
86     virtual void SetDigits(TObjArray* digits) {fDigits=digits;}
87     const TClonesArray* GetSDigits() const { return &fSDigits;}
88     TObjArray*    GetDigits() const {return fDigits;}
89     Int_t* GetNDigitArray() const {return fNDigits;}
90     TClonesArray *DigitsAddress(Int_t id) const {
91         return ((TClonesArray*)(*fDigits)[id]);}
92     virtual void ResetSDigits(){fNSDigits=0; fSDigits.Clear();}
93     virtual void ResetDigits();
94     virtual void ResetDigits(Int_t branch);
95     virtual void SDigitsToDigits(Option_t *opt,Char_t* name);
96
97     virtual void AddSumDigit(AliITSpListItem &sdig);
98     virtual void AddSimDigit(Int_t branch, AliITSdigit *d);
99     virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
100                              Int_t* tracks,Int_t *hits,Float_t* trkcharges,
101                              Int_t sigexpanded=-1000);
102     virtual void SetDigitClassName(Int_t i, Char_t* name) {
103         fDigClassName[i]=name;}
104     Char_t* GetDigitClassName(Int_t i) const {return fDigClassName[i];}
105     void StoreCalibration(Int_t firstRun, Int_t lastRun, AliCDBMetaData &md);
106
107  protected:
108     virtual void CreateCalibrationArray(); 
109     virtual Bool_t GetCalibration();
110     
111  private:
112     void SetDefaultSegmentation(Int_t idet);  // creates def segm.
113
114     //conversion of the old SSD calibration objects tothe new ones
115     void ReadOldSSDNoise(TObjArray *array, 
116                          AliITSNoiseSSDv2 *noiseSSD);
117     void ReadOldSSDBadChannels(TObjArray *array, 
118                                AliITSBadChannelsSSDv2 *badChannelsSSD);
119     void ReadOldSSDGain(TObjArray *array, 
120                         AliITSGainSSDv2 *gainSSD);
121
122     static const Int_t fgkNdettypes;          // number of different det. types
123     static const Int_t fgkDefaultNModulesSPD; // Total numbers of SPD modules by default
124     static const Int_t fgkDefaultNModulesSDD; // Total numbers of SDD modules by default
125     static const Int_t fgkDefaultNModulesSSD; // Total numbers of SSD modules by default
126     Int_t fNMod[3];                           // numbers of modules from different types
127
128     TObjArray    *fSimulation;   //! [NDet]
129     TObjArray    *fSegmentation; //! [NDet]
130     TObjArray    *fCalibration;  //! [NMod]
131     AliITSCalibrationSSD* fSSDCalibration;  //! SSD calibration object
132     TObjArray    *fPreProcess;   //! [] e.g. Fill fHitModule with hits
133     TObjArray    *fPostProcess;  //! [] e.g. Wright Raw data
134     Int_t         fNSDigits;     //! number of SDigits
135     TClonesArray  fSDigits;      //! Summable digits
136     Int_t*        fNDigits;      //! [NDet] number of Digits for det.
137     Int_t         fRunNumber;    //! run number (to access DB)
138     TObjArray     *fDigits;      //! [NMod][NDigits]
139     AliITSSimuParam *fSimuPar;   //! detector simulation parameters
140     AliITSDDLModuleMapSDD *fDDLMapSDD; //! mapping DDL/module -> SDD module number
141     TString       fHitClassName; //! String with Hit class name
142     TString       fSDigClassName;//! String with SDigit class name.
143     Char_t*       fDigClassName[3]; //! String with digit class name.
144     AliITSLoader* fLoader;          //! loader  
145     Bool_t        fFirstcall;       //! flag
146     
147     ClassDef(AliITSDetTypeSim,7) // ITS Simulation structure
148  
149 };
150
151 #endif