7ed6c6290f1f7d0d62d94a1526f494e18e99c933
[u/mrichter/AliRoot.git] / ITS / AliITSDetTypeRec.h
1 #ifndef ALIITSDETTYPEREC_H
2 #define ALIITSDETTYPEREC_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 reconstruction for the ITS.                      //
13 ////////////////////////////////////////////////////////////////////////
14 #include <TObject.h>
15 #include <TObjArray.h>
16 #include <TClonesArray.h>
17 #include "AliITSDDLModuleMapSDD.h"
18 #include "AliITSresponseSDD.h"
19 #include "AliITSgeom.h"
20 class TTree;
21 class TBranch;
22
23 //#include "AliITSLoader.h"
24 //#include "AliRunLoader.h"
25
26 class AliITSsegmentation;
27 class AliITSCalibration;
28 class AliITSCalibrationSSD;
29 class AliITSresponseSDD;
30 class AliITSClusterFinder;
31 class AliITSRawCluster;
32 class AliITSRecPoint;
33 class AliRawReader;
34 class AliITSGainSSDv2;
35 class AliITSBadChannelsSSDv2;
36 class AliITSNoiseSSDv2;
37
38 class AliITSDetTypeRec : public TObject {
39   public:
40     AliITSDetTypeRec(); // Default constructor
41  
42     virtual ~AliITSDetTypeRec(); // Proper Destructor
43
44     virtual AliITSgeom* GetITSgeom() const { return fITSgeom; }
45     virtual void SetITSgeom(AliITSgeom *geom) { fITSgeom = geom; }
46     virtual void SetDefaults();
47     virtual void SetDefaultClusterFinders();
48     virtual void SetDefaultClusterFindersV2(Bool_t rawdata=kFALSE);
49     virtual void MakeBranch(TTree *tree,Option_t *opt);
50     virtual void SetTreeAddressD(TTree* treeD);
51
52     virtual void SetSegmentationModel(Int_t dettype, AliITSsegmentation *seg);
53     virtual void SetCalibrationModel(Int_t iMod, AliITSCalibration *cal);
54     virtual void SetSPDDeadModel(Int_t iMod, AliITSCalibration *cal);
55     virtual void SetReconstructionModel(Int_t dettype, AliITSClusterFinder *rec);
56     virtual Bool_t GetCalibration();
57     virtual Bool_t GetCalibrationSPD(Bool_t cacheStatus);
58     virtual Bool_t GetCalibrationSDD(Bool_t cacheStatus);
59     virtual Bool_t GetCalibrationSSD(Bool_t cacheStatus);
60     virtual AliITSsegmentation* GetSegmentationModel(Int_t dettype);
61     virtual AliITSCalibration* GetCalibrationModel(Int_t iMod);
62     virtual AliITSCalibration* GetSPDDeadModel(Int_t iMod);
63     virtual AliITSClusterFinder* GetReconstructionModel(Int_t dettype);
64     virtual AliITSDDLModuleMapSDD* GetDDLModuleMapSDD() const { return fDDLMapSDD;}
65     virtual AliITSresponseSDD* GetResponseSDD() const { return fRespSDD;}
66     virtual Bool_t IsHLTmodeC() const {return fIsHLTmodeC;}
67     virtual void SetHLTmodeC(Bool_t ishltc){fIsHLTmodeC=ishltc;}
68     virtual Float_t GetAverageGainSDD() const {
69       if(fAveGainSDD>0.) return fAveGainSDD;
70       else return 1.;
71     }
72
73     virtual void SetDigitClassName(Int_t i,Char_t *digit) 
74       {fDigClassName[i]=digit;}
75     virtual void SetClusterClassName(Int_t i,Char_t *cluster)
76       {fClusterClassName[i]=cluster;}
77     virtual void SetRecPointClassName(Int_t i,Char_t *recpoint) 
78       {fRecPointClassName[i]=recpoint;}
79     
80     virtual void SetLoadOnlySPDCalib(Bool_t opt=kFALSE)
81       {fLoadOnlySPDCalib=opt;}
82
83     Char_t* GetDigitClassName(Int_t i) const {return fDigClassName[i];}
84     Char_t* GetClusterClassName(Int_t i) const {return fClusterClassName[i];}
85     Char_t* GetRecPointClassName(Int_t i) const {return fRecPointClassName[i];}
86     
87     TObjArray* GetDigits() const {return fDigits;} 
88     Int_t *Ndtype() {return fNdtype;}
89     TClonesArray *DigitsAddress(Int_t id) const {return ((TClonesArray*)(*fDigits)[id]);}
90     virtual void SelectVertexer(TString sel=" "){fSelectedVertexer = sel;}
91     //
92     virtual void AddCluster(Int_t branch, AliITSRawCluster *c);
93     virtual void ResetClusters(); 
94     virtual void ResetClusters(Int_t branch);
95     TBranch* MakeBranchInTree(TTree *tree, const char* name, const char *classname, void* address,Int_t size, Int_t splitlevel);
96
97     TObjArray    *Ctype()  {return fCtype;}
98     Int_t        *Nctype() {return fNctype;}
99     TClonesArray *ClustersAddress(Int_t id) const {return ((TClonesArray*)(*fCtype)[id]);}
100     virtual void ResetDigits();
101     virtual void ResetDigits(Int_t branch);
102
103     void MakeBranchR(TTree *treeR,Option_t *opt=" ");
104     void SetTreeAddressR(TTree *treeR);
105     void AddRecPoint(const AliITSRecPoint &p);
106     void ResetRecPoints(){if(fRecPoints) fRecPoints->Clear();fNRecPoints = 0;};
107     // Return pointer to rec points 
108     TClonesArray  *RecPoints()   {return fRecPoints;}
109     void MakeBranchRF(TTree *treeR){MakeBranchR(treeR,"Fast");}
110     void DigitsToRecPoints(TTree *treeD,TTree *treeR,Int_t lastEntry,Option_t *det,Bool_t v2=kFALSE);
111     void DigitsToRecPoints(AliRawReader* rawReader,TTree *treeR,Option_t *det="All");
112
113   private:
114     // private methods
115     AliITSDetTypeRec(const AliITSDetTypeRec& rec);
116     AliITSDetTypeRec& operator=(const AliITSDetTypeRec &source);
117  
118     //conversion of the old SSD calibration objects tothe new ones
119     void ReadOldSSDNoise(TObjArray *array, 
120                          AliITSNoiseSSDv2 *noiseSSD);
121     void ReadOldSSDBadChannels(TObjArray *array, 
122                                AliITSBadChannelsSSDv2 *badChannelsSSD);
123     void ReadOldSSDGain(TObjArray *array, 
124                         AliITSGainSSDv2 *gainSSD);
125
126     //    virtual void SetLoader(AliITSLoader* loader) {fLoader=loader;}
127     static const Int_t fgkNdettypes;          // number of det. types
128     static const Int_t fgkDefaultNModulesSPD; // Total numbers of SPD modules by default
129     static const Int_t fgkDefaultNModulesSDD; // Total numbers of SDD modules by default
130     static const Int_t fgkDefaultNModulesSSD; // Total numbers of SSD modules by default
131     Int_t *fNMod;     // numbers of modules from different types
132
133     AliITSgeom   *fITSgeom;       //! ITS geometry
134
135     TObjArray    *fReconstruction;//! [NDet]
136     TObjArray    *fSegmentation;  //! [NDet]
137     TObjArray    *fCalibration;   //! [NMod]
138     AliITSCalibrationSSD* fSSDCalibration;  //! SSD calibration object
139     TObjArray    *fSPDDead;       //! [fgkDefaultNModulesSPD]
140     TObjArray    *fPreProcess;    //! [] e.g. Find Calibration values
141     TObjArray    *fPostProcess;   //! [] e.g. find primary vertex
142     TObjArray    *fDigits;        //! [NMod][NDigits]
143     AliITSDDLModuleMapSDD *fDDLMapSDD; //! mapping DDL/module -> SDD module number
144     AliITSresponseSDD *fRespSDD;  //! SDD response parameters 
145     Float_t       fAveGainSDD;    //! Average gain of SDD good anodes
146     Bool_t        fIsHLTmodeC;    //! flag for HLT mode C status (used by SDD)
147     Int_t        *fNdtype;        //! detector types  
148     Char_t*       fClusterClassName[3]; //! String with Cluster class name
149     Char_t*       fDigClassName[3];     //! String with digit class name.
150     Char_t*       fRecPointClassName[3];//! String with RecPoint class name
151
152     TObjArray    *fCtype;      //! List of clusters
153     Int_t        *fNctype;     //[fNDetTypes] Num. of clust. per type of det.
154
155     TClonesArray *fRecPoints;  //! List of reconstructed points
156     Int_t         fNRecPoints; // Number of rec points
157
158     TString fSelectedVertexer; // Vertexer selected in CreateVertexer
159     Bool_t fFirstcall;         //! flag
160     Bool_t fLoadOnlySPDCalib;  //! flag for loading calibrations only fr SPD
161
162     ClassDef(AliITSDetTypeRec,13) // ITS Reconstruction structure
163 };
164
165 #endif