]>
Commit | Line | Data |
---|---|---|
ab9f7002 | 1 | #ifndef ALITRDMCMSIM_H |
2 | #define ALITRDMCMSIM_H | |
dfd03fc3 | 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 | // // | |
10 | // Multi Chip Module Simulation Class // | |
11 | // // | |
12 | /////////////////////////////////////////////////////// | |
13 | ||
ce4786b9 | 14 | #include <iostream> |
ab9f7002 | 15 | |
ce4786b9 | 16 | class TObject; |
ab9f7002 | 17 | class TClonesArray; |
4ff7ed2b | 18 | class TH2F; |
dfd03fc3 | 19 | |
b0a41e80 | 20 | class AliRunLoader; |
dfd03fc3 | 21 | class AliTRDfeeParam; |
b0a41e80 | 22 | class AliTRDtrapConfig; |
b65e5048 | 23 | class AliTRDarrayADC; |
ce4786b9 | 24 | class AliTRDarrayDictionary; |
40bd6ee4 | 25 | class AliTRDdigitsManager; |
dfd03fc3 | 26 | |
27 | class AliTRDmcmSim : public TObject { | |
dfd03fc3 | 28 | public: |
ecf39416 | 29 | AliTRDmcmSim(); |
30 | virtual ~AliTRDmcmSim(); | |
dfd03fc3 | 31 | |
5f006bd7 | 32 | void Init(Int_t det, Int_t rob, Int_t mcm, Bool_t newEvent = kFALSE); |
ce4786b9 | 33 | // Initialize MCM by the position parameters |
34 | ||
5f006bd7 | 35 | void Reset(); |
ce4786b9 | 36 | // clears filter registers and internal data |
b0a41e80 | 37 | |
ab9f7002 | 38 | Bool_t LoadMCM(AliRunLoader* const runloader, Int_t det, Int_t rob, Int_t mcm); |
ce4786b9 | 39 | |
b0a41e80 | 40 | void NoiseTest(Int_t nsamples, Int_t mean, Int_t sigma, Int_t inputGain = 1, Int_t inputTail = 2); |
dfd03fc3 | 41 | |
5f006bd7 | 42 | Int_t GetDataRaw(Int_t iadc, Int_t timebin) const { return (fADCR[iadc][timebin] >> 2); } |
43 | // Get unfiltered ADC data | |
44 | Int_t GetDataFiltered(Int_t iadc, Int_t timebin) const { return (fADCF[iadc][timebin] >> 2); } | |
ce4786b9 | 45 | // Get filtered ADC data |
46 | ||
6419bebb | 47 | void SetData(Int_t iadc, const Int_t* const adc); // Set ADC data with array |
ce4786b9 | 48 | void SetData(Int_t iadc, Int_t it, Int_t adc); // Set ADC data |
5f006bd7 | 49 | void SetData(AliTRDarrayADC * const adcArray, |
6b094867 | 50 | AliTRDdigitsManager * const digitsManager = 0x0); // Set ADC data from adcArray |
6419bebb | 51 | void SetDataByPad(const AliTRDarrayADC *const adcArray, |
6b094867 | 52 | AliTRDdigitsManager * const digitsManager = 0x0); // Set ADC data from adcArray |
ce4786b9 | 53 | void SetDataPedestal(Int_t iadc); // Fill ADC data with pedestal values |
dfd03fc3 | 54 | |
ce4786b9 | 55 | static Bool_t GetApplyCut() { return fgApplyCut; } |
56 | static void SetApplyCut(Bool_t applyCut) { fgApplyCut = applyCut; } | |
4ff7ed2b | 57 | |
5f006bd7 | 58 | static Int_t GetAddBaseline() { return fgAddBaseline; } |
59 | static void SetAddBaseline(Int_t baseline) { fgAddBaseline = baseline; } | |
60 | // Additional baseline which is added for the processing | |
61 | // in the TRAP and removed when writing back the data. | |
62 | // This is needed to run with TRAP parameters set for a | |
63 | // different baseline but it will not change the baseline | |
64 | // of the output. | |
4ff7ed2b | 65 | |
dc1f2681 | 66 | static void SetStoreClusters(Bool_t storeClusters) { fgStoreClusters = storeClusters; } |
c1309b7e | 67 | static Bool_t GetStoreClusters() { return fgStoreClusters; } |
dc1f2681 | 68 | |
b0a41e80 | 69 | Int_t GetDetector() const { return fDetector; }; // Returns Chamber ID (0-539) |
ce4786b9 | 70 | Int_t GetRobPos() const { return fRobPos; }; // Returns ROB position (0-7) |
71 | Int_t GetMcmPos() const { return fMcmPos; }; // Returns MCM position (0-17) (16,17 are mergers) | |
72 | Int_t GetRow() const { return fRow; }; // Returns Row number on chamber where the MCM is sitting | |
73 | Int_t GetCol( Int_t iadc ); // Get corresponding column (0-143) from for ADC channel iadc = [0:20] | |
ecf39416 | 74 | // for the ADC/Col mapping, see: http://wiki.kip.uni-heidelberg.de/ti/TRD/index.php/Image:ROB_MCM_numbering.pdf |
dfd03fc3 | 75 | |
4ff7ed2b | 76 | void WriteData(AliTRDarrayADC *digits); |
77 | Bool_t StoreTracklets(); // Stores tracklets via runloader | |
6419bebb | 78 | TString GetTrklBranchName() const { return fTrklBranchName; } |
6b094867 | 79 | void SetTrklBranchName(TString name) { fTrklBranchName = name; } |
1d93b218 | 80 | |
ce4786b9 | 81 | Int_t ProduceRawStream( UInt_t *buf, Int_t bufsize, UInt_t iEv = 0 ) const; // Produce raw data stream - Real data format |
1d93b218 | 82 | Int_t ProduceTrackletStream( UInt_t *buf, Int_t bufsize ); // produce the tracklet stream for this MCM |
5f006bd7 | 83 | |
b0a41e80 | 84 | // different stages of processing in the TRAP |
85 | void Filter(); // Apply digital filters for existing data (according to configuration) | |
ecf39416 | 86 | void ZSMapping(); // Do ZS mapping for existing data |
b0a41e80 | 87 | void Tracklet(); // Run tracklet preprocessor and perform tracklet fit |
dfd03fc3 | 88 | |
b0a41e80 | 89 | // apply individual filters to all channels and timebins |
90 | void FilterPedestal(); // Apply pedestal filter | |
91 | void FilterGain(); // Apply gain filter | |
92 | void FilterTail(); // Apply tail filter | |
dfd03fc3 | 93 | |
b0a41e80 | 94 | // filter initialization (resets internal registers) |
ce4786b9 | 95 | void FilterPedestalInit(Int_t baseline = 10); |
b0a41e80 | 96 | void FilterGainInit(); |
5f006bd7 | 97 | void FilterTailInit(Int_t baseline = -1); |
ecf39416 | 98 | |
b0a41e80 | 99 | // feed single sample to individual filter |
100 | // this changes the internal registers | |
ce4786b9 | 101 | // all filters operate on (10+2)-bit values! |
b0a41e80 | 102 | UShort_t FilterPedestalNextSample(Int_t adc, Int_t timebin, UShort_t value); |
103 | UShort_t FilterGainNextSample(Int_t adc, UShort_t value); | |
104 | UShort_t FilterTailNextSample(Int_t adc, UShort_t value); | |
1d93b218 | 105 | |
b0a41e80 | 106 | // tracklet calculation |
25b41f6f | 107 | void AddHitToFitreg(Int_t adc, UShort_t timebin, UShort_t qtot, Short_t ypos, Int_t label[]); |
b0a41e80 | 108 | void CalcFitreg(); |
109 | void TrackletSelection(); | |
110 | void FitTracklet(); | |
1d93b218 | 111 | |
ce4786b9 | 112 | Int_t GetNHits() const { return fNHits; } |
113 | Bool_t GetHit(Int_t index, Int_t &channel, Int_t &timebin, Int_t &qtot, Int_t &ypos, Float_t &y, Int_t &label) const; | |
ab9f7002 | 114 | TClonesArray* GetTrackletArray() const { return fTrackletArray; } |
dfd03fc3 | 115 | |
b0a41e80 | 116 | // data display |
ab9f7002 | 117 | void Print(Option_t* const option="") const; // print stored data to stdout |
118 | void Draw(Option_t* const option =""); // draw data (ADC data, hits and tracklets) | |
ce4786b9 | 119 | |
120 | friend std::ostream& operator<<(std::ostream &os, const AliTRDmcmSim &mcm); // data output using ostream (e.g. cout << mcm;) | |
59f78ad5 | 121 | static ostream& Cfdat(ostream &os); // manipulator to activate cfdat output |
122 | static ostream& Raw (ostream &os); // manipulator to activate raw output | |
123 | static ostream& Text (ostream &os); // manipulator to activate text output | |
dfd03fc3 | 124 | |
8ea391e3 | 125 | // PID |
126 | Int_t GetPID(Int_t q0, Int_t q1); | |
127 | void PrintPidLutHuman(); | |
8ea391e3 | 128 | |
129 | // I/O | |
130 | void PrintFitRegXml(ostream& os) const; | |
131 | void PrintTrackletsXml(ostream& os) const; | |
6d976847 | 132 | void PrintAdcDatTxt(ostream& os) const; |
8ea391e3 | 133 | void PrintAdcDatHuman(ostream& os) const; |
134 | void PrintAdcDatXml(ostream& os) const; | |
a10c6981 | 135 | void PrintAdcDatDatx(ostream& os, Bool_t broadcast=kFALSE, Int_t timeBinOffset = -1) const; |
8ea391e3 | 136 | |
43573bcf | 137 | static Bool_t ReadPackedConfig(AliTRDtrapConfig *cfg, Int_t hc, UInt_t *data, Int_t size); |
2b2b540f | 138 | |
139 | // DMEM addresses | |
140 | static const Int_t fgkDmemAddrLUTcor0 = 0xC02A; | |
141 | static const Int_t fgkDmemAddrLUTcor1 = 0xC028; | |
142 | static const Int_t fgkDmemAddrLUTnbins = 0xC029; | |
143 | ||
144 | static const Int_t fgkDmemAddrLUTStart = 0xC100; // LUT start address | |
145 | static const Int_t fgkDmemAddrLUTEnd = 0xC3FF; // maximum possible end address for the LUT table | |
146 | static const Int_t fgkDmemAddrLUTLength = 0xC02B; // address where real size of the LUT table is stored | |
147 | ||
148 | static const Int_t fgkDmemAddrTrackletStart = 0xC0E0; // Storage area for tracklets, start address | |
149 | static const Int_t fgkDmemAddrTrackletEnd = 0xC0E3; // Storage area for tracklets, end address | |
150 | ||
151 | static const Int_t fgkDmemAddrDeflCorr = 0xc022; // DMEM address of deflection correction | |
152 | static const Int_t fgkDmemAddrNdrift = 0xc025; // DMEM address of Ndrift | |
153 | static const Int_t fgkDmemAddrDeflCutStart = 0xc030; // DMEM start address of deflection cut | |
154 | static const Int_t fgkDmemAddrDeflCutEnd = 0xc055; // DMEM end address of deflection cut | |
155 | ||
b0a41e80 | 156 | protected: |
ce4786b9 | 157 | Bool_t CheckInitialized() const; // Check whether the class is initialized |
5f006bd7 | 158 | |
ce4786b9 | 159 | void SetNTimebins(Int_t ntimebins); // allocate data arrays corr. to the no. of timebins |
160 | ||
161 | static const Int_t fgkFormatIndex; // index for format settings in stream | |
162 | ||
ce4786b9 | 163 | static const Int_t fgkMaxTracklets = 4; // maximum number of tracklet-words submitted per MCM (one per CPU) |
164 | static const Int_t fgkAddDigits = 2; // additional digits used for internal representation of ADC data | |
165 | // all internal data as after data control block (i.e. 12 bit), s. TRAP manual | |
166 | static const Int_t fgkNCPU = 4; // Number of CPUs in the TRAP | |
167 | static const Int_t fgkNHitsMC = 100; // maximum number of hits for which MC information is kept | |
168 | ||
169 | static const UShort_t fgkFPshifts[4]; // shifts for pedestal filter | |
170 | ||
ce4786b9 | 171 | Bool_t fInitialized; // memory is allocated if initialized |
172 | Int_t fDetector; // Chamber ID | |
5f006bd7 | 173 | Int_t fRobPos; // ROB Position on chamber |
174 | Int_t fMcmPos; // MCM Position on chamber | |
b0a41e80 | 175 | Int_t fRow; // Pad row number (0-11 or 0-15) of the MCM on chamber |
ce4786b9 | 176 | Int_t fNTimeBin; // Number of timebins currently allocated |
b0a41e80 | 177 | Int_t **fADCR; // Array with MCM ADC values (Raw, 12 bit) |
178 | Int_t **fADCF; // Array with MCM ADC values (Filtered, 12 bit) | |
5f006bd7 | 179 | UInt_t *fMCMT; // tracklet word for one mcm/trap-chip |
b0a41e80 | 180 | TClonesArray *fTrackletArray; // Array of AliTRDtrackletMCM which contains MC information in addition to the tracklet word |
ce4786b9 | 181 | Int_t *fZSMap; // Zero suppression map (1 dimensional projection) |
dfd03fc3 | 182 | |
b0a41e80 | 183 | Int_t fFitPtr[fgkNCPU]; // pointer to the tracklet to be calculated by CPU i |
dfd03fc3 | 184 | |
6b094867 | 185 | TString fTrklBranchName; // name of the tracklet branch to right to |
186 | ||
b0a41e80 | 187 | // Parameter classes |
188 | AliTRDfeeParam *fFeeParam; // FEE parameters | |
189 | AliTRDtrapConfig *fTrapConfig; // TRAP config | |
b0a41e80 | 190 | |
40bd6ee4 | 191 | AliTRDdigitsManager *fDigitsManager; // pointer to digits manager used for MC label calculation |
ce4786b9 | 192 | AliTRDarrayDictionary* fDict[3]; // pointers to label dictionaries |
193 | ||
40bd6ee4 | 194 | |
b0a41e80 | 195 | // internal filter registers |
196 | UInt_t* fPedAcc; // Accumulator for pedestal filter | |
197 | UInt_t* fGainCounterA; // Counter for values above FGTA in the gain filter | |
198 | UInt_t* fGainCounterB; // Counter for values above FGTB in the gain filter | |
199 | UShort_t* fTailAmplLong; // Amplitude of the long component in the tail filter | |
200 | UShort_t* fTailAmplShort; // Amplitude of the short component in the tail filter | |
201 | ||
202 | // hit detection | |
203 | // individual hits can be stored as MC info | |
ab9f7002 | 204 | struct Hit_t { // Array of detected hits (only available in MC) |
54d34aac | 205 | Hit_t() : fChannel(0), fTimebin(0), fQtot(0), fYpos(0) { fLabel[0] = 0; fLabel[1] = 0; fLabel[2] = 0; } |
ab9f7002 | 206 | Int_t fChannel; // ADC channel of the hit |
207 | Int_t fTimebin; // timebin of the hit | |
208 | Int_t fQtot; // total charge of the hit | |
209 | Int_t fYpos; // calculated y-position | |
25b41f6f | 210 | Int_t fLabel[3]; // up to 3 labels (only in MC) |
ab9f7002 | 211 | } fHits[fgkNHitsMC]; |
b0a41e80 | 212 | Int_t fNHits; // Number of detected hits |
213 | ||
214 | // tracklet calculation | |
5f006bd7 | 215 | struct FitReg_t { // pointer to the 18 fit registers |
ce4786b9 | 216 | Int_t fNhits; // number of hits |
ab9f7002 | 217 | UInt_t fQ0; // charge accumulated in first window |
218 | UInt_t fQ1; // charge accumulated in second window | |
219 | UInt_t fSumX; // sum x | |
5f006bd7 | 220 | Int_t fSumY; // sum y |
ab9f7002 | 221 | UInt_t fSumX2; // sum x**2 |
222 | UInt_t fSumY2; // sum y**2 | |
ce4786b9 | 223 | Int_t fSumXY; // sum x*y |
ab9f7002 | 224 | } *fFitReg; |
b0a41e80 | 225 | |
ce4786b9 | 226 | // Sort functions as in TRAP |
5f006bd7 | 227 | void Sort2(UShort_t idx1i, UShort_t idx2i, UShort_t val1i, UShort_t val2i, |
6b094867 | 228 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const val1o, UShort_t * const val2o) const; |
5f006bd7 | 229 | void Sort3(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, |
230 | UShort_t val1i, UShort_t val2i, UShort_t val3i, | |
231 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, | |
6b094867 | 232 | UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o); |
5f006bd7 | 233 | void Sort6To4(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, |
234 | UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, | |
235 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, UShort_t * const idx4o, | |
6b094867 | 236 | UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o, UShort_t * const val4o); |
5f006bd7 | 237 | void Sort6To2Worst(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, |
238 | UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, | |
6b094867 | 239 | UShort_t * const idx5o, UShort_t * const idx6o); |
b0a41e80 | 240 | |
5f006bd7 | 241 | UInt_t AddUintClipping(UInt_t a, UInt_t b, UInt_t nbits) const; |
ce4786b9 | 242 | // Add a and b (unsigned) with clipping to the maximum value representable by nbits |
b0a41e80 | 243 | |
244 | private: | |
245 | AliTRDmcmSim(const AliTRDmcmSim &m); // not implemented | |
246 | AliTRDmcmSim &operator=(const AliTRDmcmSim &m); // not implemented | |
247 | ||
36dc3337 | 248 | static Bool_t fgApplyCut; // apply cut on deflection length |
4ff7ed2b | 249 | |
5f006bd7 | 250 | static Int_t fgAddBaseline; // add baseline to the ADC values |
4ff7ed2b | 251 | |
dc1f2681 | 252 | static Bool_t fgStoreClusters; // whether to store all clusters in the tracklets |
253 | ||
6170d369 | 254 | ClassDef(AliTRDmcmSim,7) |
dfd03fc3 | 255 | }; |
256 | ||
ce4786b9 | 257 | std::ostream& operator<<(std::ostream& os, const AliTRDmcmSim& mcm); |
258 | ||
dfd03fc3 | 259 | #endif |