Added two missing includes to allow macro compilation (thanks to Laurent for remarkin...
[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 "AliITSLoader.h"
17 #include "AliITSSimuParam.h"
18 #include "AliITSFOGeneratorSPD.h"
19
20 class TObjArray;
21 class TClonesArray;
22 class TTree;
23 class AliCDBMetaData;
24 class AliITSdigit;
25 class AliITSmodule;
26 class AliITSpListItem;
27 class AliITSsimulation;
28 class AliITSsegmentation;
29 class AliITSresponse;
30 class AliITSCalibrationSSD;
31 class AliITSGainSSDv2;
32 class AliITSBadChannelsSSDv2;
33 class AliITSNoiseSSDv2;
34 class AliITSDDLModuleMapSDD;
35 class AliITSCalibration;
36 class AliITSgeom;
37 class AliITSFOSignalsSPD;
38 class AliITSTriggerConditions;
39
40 class AliITSDetTypeSim : public TObject {
41  public:
42   
43     AliITSDetTypeSim();
44     virtual ~AliITSDetTypeSim(); 
45     AliITSDetTypeSim(const AliITSDetTypeSim &source);
46     AliITSDetTypeSim& operator=(const AliITSDetTypeSim &source);
47     AliITSgeom *GetITSgeom() const {
48         if(fLoader)return ((AliITSLoader*)fLoader)->GetITSgeom();
49         else return 0;}
50     void SetITSgeom(AliITSgeom *geom);
51     
52     virtual void SetSimulationModel(Int_t dettype,AliITSsimulation *sim);
53     virtual AliITSsimulation* GetSimulationModel(Int_t dettype) const;        
54     virtual AliITSsimulation* GetSimulationModelByModule(Int_t module) const;
55         
56     virtual void SetSegmentationModel(Int_t dettype,AliITSsegmentation *seg);
57     virtual AliITSsegmentation* GetSegmentationModel(Int_t dettype) const;
58     virtual AliITSsegmentation* GetSegmentationModelByModule(Int_t module) const;
59
60     virtual void SetCalibrationModel(Int_t iMod,AliITSCalibration *resp);
61    virtual void SetSPDNoisyModel(Int_t iMod, AliITSCalibration *cal);
62
63     virtual AliITSCalibration* GetCalibrationModel(Int_t iMod) const;
64     virtual AliITSCalibration* GetSPDNoisyModel(Int_t iMod) const;
65     virtual AliITSTriggerConditions* GetTriggerConditions();
66
67     virtual void SetSimuParam(const AliITSSimuParam* spar){
68       if(fSimuPar) delete fSimuPar;
69       fSimuPar = new AliITSSimuParam(*spar);
70     }
71     virtual AliITSSimuParam* GetSimuParam() const {return fSimuPar;}
72
73     virtual AliITSDDLModuleMapSDD* GetDDLModuleMapSDD()const { return fDDLMapSDD;}
74     TObjArray* GetCalibrationArray() const {return fCalibration;}
75     TObjArray* GetSegmentation() const {return fSegmentation;}
76     void ResetCalibrationArray();
77     void ResetSegmentation();
78
79     virtual void SetLoader(AliITSLoader* loader);
80     AliITSLoader* GetLoader() const {return fLoader;}
81
82     virtual void SetDefaults();
83     virtual void SetDefaultSimulation();
84     virtual void SetRunNumber(Int_t rn=0){fRunNumber = rn;}
85     virtual Int_t GetRunNumber() const {return fRunNumber;}
86     virtual void SetTreeAddressS(TTree* treeS, const Char_t* name);
87     virtual void SetTreeAddressD(TTree* treeD, const Char_t* name);
88
89     virtual void SetDigits(TObjArray* digits) {fDigits=digits;}
90     const TClonesArray* GetSDigits() const { return &fSDigits;}
91     TObjArray*    GetDigits() const {return fDigits;}
92     Int_t* GetNDigitArray() const {return fNDigits;}
93     TClonesArray *DigitsAddress(Int_t id) const {
94         return ((TClonesArray*)(*fDigits)[id]);}
95     virtual void ResetSDigits(){fNSDigits=0; fSDigits.Clear();}
96     virtual void ResetDigits();
97     virtual void ResetDigits(Int_t branch);
98     virtual void SDigitsToDigits(Option_t *opt, Char_t* name);
99
100     virtual void AddSumDigit(AliITSpListItem &sdig);
101     virtual void AddSimDigit(Int_t branch, const AliITSdigit *d);
102     virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
103                              Int_t* tracks,Int_t *hits,Float_t* trkcharges,
104                              Int_t sigexpanded=-1000);
105     virtual void SetDigitClassName(Int_t i, const Char_t* name) {
106         fkDigClassName[i]=name;}
107     const Char_t* GetDigitClassName(Int_t i) const {return fkDigClassName[i];}
108
109     virtual void ResetFOSignals() {fFOGenerator.ResetSignals();}
110     virtual void ProcessSPDDigitForFastOr(UInt_t module, UInt_t colM, UInt_t rowM);
111     virtual void ProcessNoiseForFastOr() {fFOGenerator.ProcessNoise();}
112     virtual AliITSFOSignalsSPD* GetFOSignals() {return fFOGenerator.GetFOSignals();}
113     virtual void WriteFOSignals();
114
115
116  protected:
117     virtual void CreateCalibrationArray(); 
118     virtual Bool_t GetCalibration();
119     
120  private:
121     void SetDefaultSegmentation(Int_t idet);  // creates def segm.
122
123     //conversion of the old SSD calibration objects tothe new ones
124     void ReadOldSSDNoise(const TObjArray *array, 
125                          AliITSNoiseSSDv2 *noiseSSD);
126     void ReadOldSSDBadChannels(const TObjArray *array, 
127                                AliITSBadChannelsSSDv2 *badChannelsSSD);
128     void ReadOldSSDGain(const TObjArray *array, 
129                         AliITSGainSSDv2 *gainSSD);
130
131     static const Int_t fgkNdettypes;          // number of different det. types
132     static const Int_t fgkDefaultNModulesSPD; // Total numbers of SPD modules by default
133     static const Int_t fgkDefaultNModulesSDD; // Total numbers of SDD modules by default
134     static const Int_t fgkDefaultNModulesSSD; // Total numbers of SSD modules by default
135     Int_t fNMod[3];                           // numbers of modules from different types
136
137     TObjArray    *fSimulation;   //! [NDet]
138     TObjArray    *fSegmentation; //! [NDet]
139     TObjArray    *fCalibration;  //! [NMod]
140     AliITSCalibrationSSD* fSSDCalibration;  //! SSD calibration object
141     TObjArray    *fSPDNoisy;     //! [fgkDefaultNModulesSPD]
142     Int_t         fNSDigits;     //! number of SDigits
143     TClonesArray  fSDigits;      //! Summable digits
144     Int_t*        fNDigits;      //! [NDet] number of Digits for det.
145     Int_t         fRunNumber;    //! run number (to access DB)
146     TObjArray     *fDigits;      //! [NMod][NDigits]
147     AliITSSimuParam *fSimuPar;   //! detector simulation parameters
148     AliITSDDLModuleMapSDD *fDDLMapSDD; //! mapping DDL/module -> SDD module number
149     const Char_t*       fkDigClassName[3]; //! String with digit class name.
150     AliITSLoader* fLoader;          //! loader  
151     Bool_t        fFirstcall;       //! flag
152     AliITSFOGeneratorSPD fFOGenerator; //! Fast-OR generator object
153     AliITSTriggerConditions* fTriggerConditions; //! Trigger conditions 
154        
155     ClassDef(AliITSDetTypeSim,11) // ITS Simulation structure
156  
157 };
158
159 #endif