]>
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 | |
ce4786b9 | 32 | void Init(Int_t det, Int_t rob, Int_t mcm, Bool_t newEvent = kFALSE); |
33 | // Initialize MCM by the position parameters | |
34 | ||
35 | void Reset(); | |
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 | |
ce4786b9 | 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); } | |
45 | // Get filtered ADC data | |
46 | ||
dfd03fc3 | 47 | void SetData(Int_t iadc, Int_t *adc); // Set ADC data with array |
ce4786b9 | 48 | void SetData(Int_t iadc, Int_t it, Int_t adc); // Set ADC data |
6b094867 | 49 | void SetData(AliTRDarrayADC * const adcArray, |
50 | AliTRDdigitsManager * const digitsManager = 0x0); // Set ADC data from adcArray | |
51 | void SetDataByPad(AliTRDarrayADC *const adcArray, | |
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 | |
ce4786b9 | 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 | |
b0a41e80 | 66 | Int_t GetDetector() const { return fDetector; }; // Returns Chamber ID (0-539) |
ce4786b9 | 67 | Int_t GetRobPos() const { return fRobPos; }; // Returns ROB position (0-7) |
68 | Int_t GetMcmPos() const { return fMcmPos; }; // Returns MCM position (0-17) (16,17 are mergers) | |
69 | Int_t GetRow() const { return fRow; }; // Returns Row number on chamber where the MCM is sitting | |
70 | Int_t GetCol( Int_t iadc ); // Get corresponding column (0-143) from for ADC channel iadc = [0:20] | |
ecf39416 | 71 | // for the ADC/Col mapping, see: http://wiki.kip.uni-heidelberg.de/ti/TRD/index.php/Image:ROB_MCM_numbering.pdf |
dfd03fc3 | 72 | |
4ff7ed2b | 73 | void WriteData(AliTRDarrayADC *digits); |
74 | Bool_t StoreTracklets(); // Stores tracklets via runloader | |
6b094867 | 75 | TString GetTrklBranchName() { return fTrklBranchName; } |
76 | void SetTrklBranchName(TString name) { fTrklBranchName = name; } | |
1d93b218 | 77 | |
ce4786b9 | 78 | Int_t ProduceRawStream( UInt_t *buf, Int_t bufsize, UInt_t iEv = 0 ) const; // Produce raw data stream - Real data format |
1d93b218 | 79 | Int_t ProduceTrackletStream( UInt_t *buf, Int_t bufsize ); // produce the tracklet stream for this MCM |
b0a41e80 | 80 | |
81 | // different stages of processing in the TRAP | |
82 | void Filter(); // Apply digital filters for existing data (according to configuration) | |
ecf39416 | 83 | void ZSMapping(); // Do ZS mapping for existing data |
b0a41e80 | 84 | void Tracklet(); // Run tracklet preprocessor and perform tracklet fit |
dfd03fc3 | 85 | |
b0a41e80 | 86 | // apply individual filters to all channels and timebins |
87 | void FilterPedestal(); // Apply pedestal filter | |
88 | void FilterGain(); // Apply gain filter | |
89 | void FilterTail(); // Apply tail filter | |
dfd03fc3 | 90 | |
b0a41e80 | 91 | // filter initialization (resets internal registers) |
ce4786b9 | 92 | void FilterPedestalInit(Int_t baseline = 10); |
b0a41e80 | 93 | void FilterGainInit(); |
ce4786b9 | 94 | void FilterTailInit(Int_t baseline = -1); |
ecf39416 | 95 | |
b0a41e80 | 96 | // feed single sample to individual filter |
97 | // this changes the internal registers | |
ce4786b9 | 98 | // all filters operate on (10+2)-bit values! |
b0a41e80 | 99 | UShort_t FilterPedestalNextSample(Int_t adc, Int_t timebin, UShort_t value); |
100 | UShort_t FilterGainNextSample(Int_t adc, UShort_t value); | |
101 | UShort_t FilterTailNextSample(Int_t adc, UShort_t value); | |
1d93b218 | 102 | |
b0a41e80 | 103 | // tracklet calculation |
104 | void AddHitToFitreg(Int_t adc, UShort_t timebin, UShort_t qtot, Short_t ypos, Int_t label); | |
105 | void CalcFitreg(); | |
106 | void TrackletSelection(); | |
107 | void FitTracklet(); | |
1d93b218 | 108 | |
ce4786b9 | 109 | Int_t GetNHits() const { return fNHits; } |
110 | 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 | 111 | TClonesArray* GetTrackletArray() const { return fTrackletArray; } |
dfd03fc3 | 112 | |
b0a41e80 | 113 | // data display |
ab9f7002 | 114 | void Print(Option_t* const option="") const; // print stored data to stdout |
115 | void Draw(Option_t* const option =""); // draw data (ADC data, hits and tracklets) | |
ce4786b9 | 116 | |
117 | friend std::ostream& operator<<(std::ostream &os, const AliTRDmcmSim &mcm); // data output using ostream (e.g. cout << mcm;) | |
59f78ad5 | 118 | static ostream& Cfdat(ostream &os); // manipulator to activate cfdat output |
119 | static ostream& Raw (ostream &os); // manipulator to activate raw output | |
120 | static ostream& Text (ostream &os); // manipulator to activate text output | |
dfd03fc3 | 121 | |
8ea391e3 | 122 | // PID |
123 | Int_t GetPID(Int_t q0, Int_t q1); | |
124 | void PrintPidLutHuman(); | |
125 | void PrintPidLutDatx(ostream& os) const; | |
126 | void SetPIDLutScaleDMEM(); | |
6b094867 | 127 | void SetPIDLut(TH2F * const lut); |
8ea391e3 | 128 | void SetPIDLut(Int_t *lut, Int_t nbinsq0, Int_t nbinsq1); |
129 | ||
130 | // I/O | |
131 | void PrintFitRegXml(ostream& os) const; | |
132 | void PrintTrackletsXml(ostream& os) const; | |
133 | void PrintAdcDatHuman(ostream& os) const; | |
134 | void PrintAdcDatXml(ostream& os) const; | |
135 | void PrintAdcDatDatx(ostream& os, Bool_t broadcast=kFALSE) const; | |
136 | ||
b0a41e80 | 137 | protected: |
ce4786b9 | 138 | Bool_t CheckInitialized() const; // Check whether the class is initialized |
b0a41e80 | 139 | |
ce4786b9 | 140 | void SetNTimebins(Int_t ntimebins); // allocate data arrays corr. to the no. of timebins |
141 | ||
142 | static const Int_t fgkFormatIndex; // index for format settings in stream | |
143 | ||
144 | static const Int_t fgkNADC; // Number of ADC | |
145 | static const Int_t fgkMaxTracklets = 4; // maximum number of tracklet-words submitted per MCM (one per CPU) | |
146 | static const Int_t fgkAddDigits = 2; // additional digits used for internal representation of ADC data | |
147 | // all internal data as after data control block (i.e. 12 bit), s. TRAP manual | |
148 | static const Int_t fgkNCPU = 4; // Number of CPUs in the TRAP | |
149 | static const Int_t fgkNHitsMC = 100; // maximum number of hits for which MC information is kept | |
150 | ||
151 | static const UShort_t fgkFPshifts[4]; // shifts for pedestal filter | |
152 | ||
153 | ||
154 | Bool_t fInitialized; // memory is allocated if initialized | |
155 | Int_t fDetector; // Chamber ID | |
b0a41e80 | 156 | Int_t fRobPos; // ROB Position on chamber |
157 | Int_t fMcmPos; // MCM Position on chamber | |
158 | Int_t fRow; // Pad row number (0-11 or 0-15) of the MCM on chamber | |
ce4786b9 | 159 | Int_t fNTimeBin; // Number of timebins currently allocated |
b0a41e80 | 160 | Int_t **fADCR; // Array with MCM ADC values (Raw, 12 bit) |
161 | Int_t **fADCF; // Array with MCM ADC values (Filtered, 12 bit) | |
ce4786b9 | 162 | UInt_t *fMCMT; // tracklet word for one mcm/trap-chip |
b0a41e80 | 163 | TClonesArray *fTrackletArray; // Array of AliTRDtrackletMCM which contains MC information in addition to the tracklet word |
ce4786b9 | 164 | Int_t *fZSMap; // Zero suppression map (1 dimensional projection) |
dfd03fc3 | 165 | |
b0a41e80 | 166 | Int_t fFitPtr[fgkNCPU]; // pointer to the tracklet to be calculated by CPU i |
dfd03fc3 | 167 | |
6b094867 | 168 | TString fTrklBranchName; // name of the tracklet branch to right to |
169 | ||
b0a41e80 | 170 | // Parameter classes |
171 | AliTRDfeeParam *fFeeParam; // FEE parameters | |
172 | AliTRDtrapConfig *fTrapConfig; // TRAP config | |
b0a41e80 | 173 | |
40bd6ee4 | 174 | AliTRDdigitsManager *fDigitsManager; // pointer to digits manager used for MC label calculation |
ce4786b9 | 175 | AliTRDarrayDictionary* fDict[3]; // pointers to label dictionaries |
176 | ||
40bd6ee4 | 177 | |
b0a41e80 | 178 | // internal filter registers |
179 | UInt_t* fPedAcc; // Accumulator for pedestal filter | |
180 | UInt_t* fGainCounterA; // Counter for values above FGTA in the gain filter | |
181 | UInt_t* fGainCounterB; // Counter for values above FGTB in the gain filter | |
182 | UShort_t* fTailAmplLong; // Amplitude of the long component in the tail filter | |
183 | UShort_t* fTailAmplShort; // Amplitude of the short component in the tail filter | |
184 | ||
185 | // hit detection | |
186 | // individual hits can be stored as MC info | |
ab9f7002 | 187 | struct Hit_t { // Array of detected hits (only available in MC) |
188 | Int_t fChannel; // ADC channel of the hit | |
189 | Int_t fTimebin; // timebin of the hit | |
190 | Int_t fQtot; // total charge of the hit | |
191 | Int_t fYpos; // calculated y-position | |
192 | Int_t fLabel; // label (only in MC) | |
193 | } fHits[fgkNHitsMC]; | |
b0a41e80 | 194 | Int_t fNHits; // Number of detected hits |
195 | ||
196 | // tracklet calculation | |
ab9f7002 | 197 | struct FitReg_t { // pointer to the 18 fit registers |
ce4786b9 | 198 | Int_t fNhits; // number of hits |
ab9f7002 | 199 | UInt_t fQ0; // charge accumulated in first window |
200 | UInt_t fQ1; // charge accumulated in second window | |
201 | UInt_t fSumX; // sum x | |
ce4786b9 | 202 | Int_t fSumY; // sum y |
ab9f7002 | 203 | UInt_t fSumX2; // sum x**2 |
204 | UInt_t fSumY2; // sum y**2 | |
ce4786b9 | 205 | Int_t fSumXY; // sum x*y |
ab9f7002 | 206 | } *fFitReg; |
b0a41e80 | 207 | |
ce4786b9 | 208 | // Sort functions as in TRAP |
b0a41e80 | 209 | void Sort2(UShort_t idx1i, UShort_t idx2i, UShort_t val1i, UShort_t val2i, |
6b094867 | 210 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const val1o, UShort_t * const val2o) const; |
b0a41e80 | 211 | void Sort3(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, |
212 | UShort_t val1i, UShort_t val2i, UShort_t val3i, | |
6b094867 | 213 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, |
214 | UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o); | |
b0a41e80 | 215 | void Sort6To4(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, |
216 | UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, | |
6b094867 | 217 | UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, UShort_t * const idx4o, |
218 | UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o, UShort_t * const val4o); | |
b0a41e80 | 219 | void Sort6To2Worst(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, |
220 | UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, | |
6b094867 | 221 | UShort_t * const idx5o, UShort_t * const idx6o); |
b0a41e80 | 222 | |
ce4786b9 | 223 | UInt_t AddUintClipping(UInt_t a, UInt_t b, UInt_t nbits) const; |
224 | // Add a and b (unsigned) with clipping to the maximum value representable by nbits | |
b0a41e80 | 225 | |
226 | private: | |
227 | AliTRDmcmSim(const AliTRDmcmSim &m); // not implemented | |
228 | AliTRDmcmSim &operator=(const AliTRDmcmSim &m); // not implemented | |
229 | ||
36dc3337 | 230 | static Bool_t fgApplyCut; // apply cut on deflection length |
4ff7ed2b | 231 | |
36dc3337 | 232 | static Int_t fgAddBaseline; // add baseline to the ADC values |
4ff7ed2b | 233 | |
ce4786b9 | 234 | ClassDef(AliTRDmcmSim,6) |
dfd03fc3 | 235 | }; |
236 | ||
ce4786b9 | 237 | std::ostream& operator<<(std::ostream& os, const AliTRDmcmSim& mcm); |
238 | ||
dfd03fc3 | 239 | #endif |