Corrected UInt_t <-> Int_t conversion
[u/mrichter/AliRoot.git] / TPC / AliTPCCalibCE.h
1 #ifndef ALITPCCALIBCE_H
2 #define ALITPCCALIBCE_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 ////////////////////////////////////////////////////////////////////////////////////////
7 //                                                                                    //
8 //             Implementation of the TPC Central Electrode calibration                //
9 //                                                                                    //
10 ////////////////////////////////////////////////////////////////////////////////////////
11
12 #include <TVectorT.h>
13 #include <THnSparse.h>
14
15 #include "AliTPCCalibRawBase.h"
16 class TH1S;
17 #include "TObjArray.h"
18 class TH2S;
19 class TH1F;
20 class TTreeSRedirector;
21 class AliTPCCalPad;
22 class AliTPCROC;
23 class AliTPCCalROC;
24 class AliTPCParam;
25 class AliRawReader;
26 class AliTPCRawStream;
27 class TGraph;
28 class TMap;
29 class TCollection;
30
31 struct eventHeaderStruct;
32
33 class AliTPCCalibCE : public AliTPCCalibRawBase {
34   
35 public:
36   AliTPCCalibCE();
37   AliTPCCalibCE(const AliTPCCalibCE &sig);
38   AliTPCCalibCE(const TMap *config);
39   virtual ~AliTPCCalibCE();
40   
41   AliTPCCalibCE& operator = (const  AliTPCCalibCE &source);
42   
43   virtual Int_t Update(const Int_t isector, const Int_t iRow, const Int_t iPad,
44                        const Int_t iTimeBin, const Float_t signal);
45   virtual void ProcessBunch(const Int_t sector, const Int_t row, const Int_t pad,
46                             const Int_t length, const UInt_t startTimeBin, const UShort_t* signal);
47   
48   virtual void Analyse();
49   void AnalyseTrack();
50   
51     //
52   AliTPCCalROC* GetCalRocT0  (Int_t sector, Bool_t force=kFALSE);  // get calibration object - sector
53   AliTPCCalROC* GetCalRocT0Err(Int_t sector, Bool_t force=kFALSE);  // get calibration object - sector
54   AliTPCCalROC* GetCalRocQ   (Int_t sector, Bool_t force=kFALSE);  // get calibration object - sector
55   AliTPCCalROC* GetCalRocRMS(Int_t sector, Bool_t force=kFALSE);  // get calibration object - sector
56   AliTPCCalROC* GetCalRocOutliers(Int_t sector, Bool_t force=kFALSE);  // get calibration object - sector
57   
58   const TObjArray* GetCalPadT0()    const { return &fCalRocArrayT0; }      // get calibration object
59   const TObjArray* GetCalPadT0Err() const { return &fCalRocArrayT0Err; }      // get calibration object
60   const TObjArray* GetCalPadQ()     const { return &fCalRocArrayQ;  }      // get calibration object
61   const TObjArray* GetCalPadRMS()   const { return &fCalRocArrayRMS;}      // get calibration object
62   const TObjArray* GetCalPadOutliers() const { return &fCalRocArrayOutliers;}      // get calibration object
63   
64   TH2S* GetHistoQ  (Int_t sector, Bool_t force=kFALSE);           // get refernce histogram
65   TH2S* GetHistoT0 (Int_t sector, Bool_t force=kFALSE);           // get refernce histogram
66   TH2S* GetHistoRMS(Int_t sector, Bool_t force=kFALSE);           // get refernce histogram
67
68   Float_t GetMeanT0rms() const {return fMeanT0rms;}
69   Float_t GetMeanQrms() const {return fMeanQrms;}
70   Float_t GetMeanRMSrms() const {return fMeanRMSrms;}
71   
72   Int_t   GetPeakDetectionMinus() const {return fPeakDetMinus;}
73   Int_t   GetPeakDetectionPlus()  const {return fPeakDetPlus;}
74   Int_t   GetPeakIntRangeMinus() const {return fPeakIntMinus;}
75   Int_t   GetPeakIntRangePlus()  const {return fPeakIntPlus;}
76   Float_t GetNnoiseThresholdMax() const {return fNoiseThresholdMax;}
77   Float_t GetNnoiseThresholdSum() const {return fNoiseThresholdSum;}
78   
79   TH1S* GetHistoTmean(Int_t sector, Bool_t force=kFALSE);           // get refernce histogram
80   
81     //needed here to merge ClibCE objects
82   TObjArray* GetParamArrayPol1(Int_t sector, Bool_t force=kFALSE);
83   TObjArray* GetParamArrayPol2(Int_t sector, Bool_t force=kFALSE);
84   
85 //    TObjArray*  GetTMeanArrayEvent(){ return &fTMeanArrayEvent; }
86 //    TObjArray*  GetQMeanArrayEvent(){ return &fQMeanArrayEvent; }
87   TVectorF* GetTMeanEvents(Int_t sector, Bool_t force=kFALSE);
88   TVectorF* GetQMeanEvents(Int_t sector, Bool_t force=kFALSE);
89   
90   const TVectorD*   GetEventTimes()  const   { return &fVEventTime;      }
91   const TVectorD*   GetEventIds()    const   { return &fVEventNumber;    }
92   
93   //
94   void  SetRangeRefQ  (Int_t nBins, Float_t xMin, Float_t xMax){ fNbinsQ   = nBins; fXminQ   = xMin; fXmaxQ   = xMax; }   //Set range for Q reference histograms
95   void  SetRangeRefT0 (Int_t nBins, Float_t xMin, Float_t xMax){ fNbinsT0  = nBins; fXminT0  = xMin; fXmaxT0  = xMax; }   //Set range for T0 reference histograms
96   void  SetRangeRefRMS(Int_t nBins, Float_t xMin, Float_t xMax){ fNbinsRMS = nBins; fXminRMS = xMin; fXmaxRMS = xMax; }   //Set range for T0 reference histograms
97   //
98   void  SetRangePeakDetection(Int_t minus, Int_t plus) { fPeakDetMinus=minus; fPeakDetPlus=plus;}
99   void  SetRangePeakIntegral(Int_t minus, Int_t plus) { fPeakIntMinus=minus; fPeakIntPlus=plus;}
100   void  SetNnoiseThresholdMax(Float_t n) {fNoiseThresholdMax=n;}
101   void  SetNnoiseThresholdSum(Float_t n) {fNoiseThresholdSum=n;}
102   //
103   void  SetEventInfo(UInt_t runNumber,UInt_t timestamp, UInt_t eventId){ fRunNumber=runNumber; fTimeStamp=timestamp; fEventId=eventId;}
104   //
105   void  SetPedestalDatabase(AliTPCCalPad * const pedestalTPC, AliTPCCalPad * const padNoiseTPC) {fPedestalTPC = pedestalTPC; fPadNoiseTPC = padNoiseTPC;}
106   void  SetIsZeroSuppressed(Bool_t zs=kTRUE) { fIsZeroSuppressed=zs; }
107   void  SetSecRejectRatio(Float_t ratio) { fSecRejectRatio=ratio; }
108
109   void SetProcessOld(Bool_t process=kTRUE) {fProcessOld=process;}
110   void SetProcessNew(Bool_t process=kTRUE) {fProcessNew=process; if (process&&!fHnDrift) CreateDVhist(); }
111   //Getters
112   Int_t GetNeventsProcessed() const { return fNevents; }
113   
114   Bool_t GetIsZeroSuppressed() const { return fIsZeroSuppressed; }
115   
116   Float_t  GetSecRejectRatio() const { return fSecRejectRatio; }
117
118   const TVectorF *GetTime0Side(Int_t side=0) const {return (side==0)?&fVTime0SideA:&fVTime0SideC;}
119   Float_t GetPeakIntegralMinus() const {return fPeakIntMinus;}
120   Float_t GetPeakIntegralPlus() const {return fPeakIntPlus;}
121   
122   
123   void Merge(AliTPCCalibCE * const ce);
124   virtual Long64_t Merge(TCollection * const list);
125   
126   TGraph *MakeGraphTimeCE(Int_t sector, Int_t xVariable=0, Int_t fitType=0, Int_t fitParameter=0);
127
128   //
129   // New functions using also the laser tracks
130   //
131   Bool_t IsEdgePad(Int_t sector, Int_t row, Int_t pad) const;
132   
133   void FindLocalMaxima(TObjArray * const arrObj, Double_t timestamp, Int_t burst);
134   Int_t FindLaserTrackID(Int_t sector,Int_t row, const Double_t *peakpos,Double_t &mindist, const Double_t *peakposloc, Int_t &itrackMin2);
135   
136   const THnSparseI *GetHnDrift() const {return fHnDrift;}
137   const TObjArray& GetArrHnDrift() const {return fArrHnDrift;}
138   const TVectorD&  GetTimeBursts() const {return fTimeBursts;}
139   const TObjArray  *GetArrFitGraphs() const {return fArrFitGraphs;}
140
141   virtual void DumpToFile(const Char_t *filename, const Char_t *dir="", Bool_t append=kFALSE);
142   
143   static AliTPCCalibCE *ReadFromFile(const Char_t *filename);
144   
145 protected:
146   virtual void EndEvent();
147   virtual void ResetEvent();
148   
149 private:
150     // reference histogram ranges
151   Int_t   fNbinsT0;                 //  Number of bins for T0 reference histogram
152   Float_t fXminT0;                  //  xmin   of T0 reference histogram
153   Float_t fXmaxT0;                  //  xmax   of T0 reference histogram
154   Int_t   fNbinsQ;                  //  Number of bins for T0 reference histogram
155   Float_t fXminQ;                   //  xmin   of T0 reference histogram
156   Float_t fXmaxQ;                   //  xmax   of T0 reference histogram
157   Int_t   fNbinsRMS;                //  Number of bins for T0 reference histogram
158   Float_t fXminRMS;                 //  xmin   of T0 reference histogram
159   Float_t fXmaxRMS;                 //  xmax   of T0 reference histogram
160   Int_t   fPeakDetMinus;               //  Consecutive timebins on rising edge to be regarded as a signal
161   Int_t   fPeakDetPlus;                //  Consecutive timebins on falling edge to be regarded as a signal
162   Int_t   fPeakIntMinus;            //  Peak integral range for COG determination. Bins used before max bin
163   Int_t   fPeakIntPlus;             //  Peak integral range for COG determination. Bins used after max bin
164   Float_t fNoiseThresholdMax;       //  Analysis Treshold for signal finding: Max>fNoiseThresholdMax*PadNoise
165   Float_t fNoiseThresholdSum;       //  Analysis Treshold for signal finding: Sum>fNoiseThresholdSum*PadNoise
166   
167   Bool_t  fIsZeroSuppressed;        //  If data is Zero Suppressed -> Don't subtrakt pedestals!
168   
169   Int_t     fLastSector;            //! Last sector processed
170   
171   Float_t   fSecRejectRatio;        //! Needed percentage of signals in one chamber. Below it will be rejected
172                                       //  This is neede if we do not process a laser event
173   
174   AliTPCParam *fParam;              //! TPC information
175   
176   AliTPCCalPad *fPedestalTPC;       //! Pedestal Information whole TPC
177   AliTPCCalPad *fPadNoiseTPC;       //! Pad noise Information whole TPC
178   AliTPCCalROC *fPedestalROC;       //! Pedestal Information for current ROC
179   AliTPCCalROC *fPadNoiseROC;       //! Pad noise Information for current ROC
180   
181   TObjArray fCalRocArrayT0;         //  Array of AliTPCCalROC class for Time0 calibration
182   TObjArray fCalRocArrayT0Err;      //  Array of AliTPCCalROC class for the error (rms) of Time0 calibration
183   TObjArray fCalRocArrayQ;          //  Array of AliTPCCalROC class for Charge calibration
184   TObjArray fCalRocArrayRMS;        //  Array of AliTPCCalROC class for signal width calibration
185   TObjArray fCalRocArrayOutliers;   //  Array of AliTPCCalROC class for signal outliers
186   
187   TObjArray fHistoQArray;           //  Calibration histograms for Charge distribution
188   TObjArray fHistoT0Array;          //  Calibration histograms for Time0  distribution
189   TObjArray fHistoRMSArray;         //  Calibration histograms for signal width distribution
190   
191   Float_t   fMeanT0rms;             // mean of the rms of all pad T0  fits, used as error estimation of T0 results
192   Float_t   fMeanQrms;              // mean of the rms of all pad Q   fits, used as error estimation of Q results
193   Float_t   fMeanRMSrms;            // mean of the rms of all pad TMS fits, used as error estimation of RMS results
194   
195   TObjArray fHistoTmean;            //! Calibration histograms of the mean CE position for all sectors
196   
197   TObjArray fParamArrayEventPol1;   //  Store mean arrival time parameters for each sector event by event from global plane fit
198   TObjArray fParamArrayEventPol2;   //  Store mean arrival time parameters for each sector event by event from global parabola fit
199   TObjArray fTMeanArrayEvent;       //  Store mean arrival time for each sector event by event
200   TObjArray fQMeanArrayEvent;       //  Store mean arrival Charge for each sector event by event
201   TVectorD  fVEventTime;            //  Timestamps of the events
202   TVectorD  fVEventNumber;          //  Eventnumbers of the events
203   TVectorF  fVTime0SideA;           //  Mean Time0 for side A for all events
204   TVectorF  fVTime0SideC;           //  Mean Time0 for side C for all events
205   Double_t  fEventId;               //! Event Id of the current event
206   UInt_t  fOldRunNumber;          //! Old Run Number
207   
208   TObjArray fPadTimesArrayEvent;    //! Pad Times for the event, before mean Time0 corrections
209   TObjArray fPadQArrayEvent;        //! Charge for the event, only needed for debugging streamer
210   TObjArray fPadRMSArrayEvent;      //! Signal width for the event, only needed for debugging streamer
211   TObjArray fPadPedestalArrayEvent; //! Signal width for the event, only needed for debugging streamer
212   
213   Int_t     fCurrentChannel;        //! current channel processed
214   Int_t     fCurrentSector;         //! current sector processed
215   Int_t     fCurrentRow;            //! current row processed
216   Float_t   fMaxPadSignal;          //! maximum bin of current pad
217   Int_t     fMaxTimeBin;            //! time bin with maximum value
218   Float_t   fPadSignal[1024];       //! signal of current Pad
219   Float_t   fPadPedestal;           //! Pedestal Value of current pad
220   Float_t   fPadNoise;              //! Noise Value of current pad
221   
222   TVectorD  fVTime0Offset;          //!  Time0 Offset for each sector;
223   TVectorD  fVTime0OffsetCounter;   //!  Time0 Offset counter for each sector;
224   TVectorD  fVMeanQ;                //!  Mean Q for each sector;
225   TVectorD  fVMeanQCounter;         //!  Mean Q counter for each sector;
226   
227   Float_t   fCurrentCETimeRef;      //! Time refernce of the current sector
228   
229   // new part of the algorithm
230   Bool_t      fProcessOld;             // Whether to use the old algorithm
231   Bool_t      fProcessNew;             // Whether to use the new algorithm
232   Bool_t      fAnalyseNew;             //! Whether to analyse the new part of the algorithm.
233                                        //In the DA this needs to be switched off, in the Preprocessor on...
234   enum {kHnBinsDV=5};
235   THnSparseI *fHnDrift;                //! Histogram digits for each pad and timebin for several timestamps
236   TObjArray   fArrHnDrift;             // array of sparse histograms for each burst
237   TVectorD    fTimeBursts;             //  time stamps of bursts
238   UInt_t      fBinsLastAna[100];       // number of bin in the THnSparse during the last analysis
239   UShort_t    fPeaks[5];               //! Peak position: 4 laser layers and CE
240   UShort_t    fPeakWidths[5];          //! Peak window widths
241   TObjArray  *fArrFitGraphs;           // Fit resut graphs for each parameter
242   
243   
244   //
245   void   FindPedestal(Float_t part=.6);
246   void   UpdateCETimeRef(); //Get the time reference of the last valid measurement in sector
247   void   FindCESignal(TVectorD &param, Float_t &qSum, const TVectorF maxima);
248   void   FindLocalMaxima(TVectorF &maxima);
249   Bool_t IsPeak(Int_t pos, Int_t tminus, Int_t tplus) const;
250   
251   TH2S* GetHisto(Int_t sector, TObjArray *arr,
252                  Int_t nbinsY, Float_t ymin, Float_t ymax,
253                  const Char_t *type, Bool_t force);
254   TH1S* GetHisto(Int_t sector, TObjArray *arr,
255                  const Char_t *type, Bool_t force);
256   
257   AliTPCCalROC* GetCalRoc(Int_t sector, TObjArray* arr, Bool_t force) const;
258   
259   TVectorF* GetVectSector(Int_t sector, TObjArray *arr, UInt_t size, Bool_t force=kFALSE) const;
260   TVectorF* GetPadTimesEvent(Int_t sector, Bool_t force=kFALSE);
261   
262   TObjArray* GetParamArray(Int_t sector, TObjArray *arr, Bool_t force=kFALSE) const;
263   
264   void ResetPad();
265   void ProcessPad();
266
267   // new part of the algorithm
268   void CreateDVhist();
269   
270   void   FindLaserLayers();
271   Bool_t IsPeakInRange(UShort_t timebin) const;
272
273   TObjArray *SetupMeasured();
274   void ResetMeasured(TObjArray * const arr);
275   
276   void AddCEtoIdeal(TObjArray *arr);
277
278   void CalculateDV(TObjArray * const arrIdeal, TObjArray * const arrMeasured, Int_t burst);
279   Double_t SetBurstHnDrift();
280   //debug
281   TVectorF* GetPadQEvent(Int_t sector, Bool_t force=kFALSE);
282   TVectorF* GetPadRMSEvent(Int_t sector, Bool_t force=kFALSE);
283   TVectorF* GetPadPedestalEvent(Int_t sector, Bool_t force=kFALSE);
284   
285   ClassDef(AliTPCCalibCE,9)  //Implementation of the TPC Central Electrode calibration
286 };
287
288 //Inline functions
289 //_____________________________________________________________________
290 inline Bool_t AliTPCCalibCE::IsPeakInRange(UShort_t timebin) const
291 {
292   //
293   // Check whether timebin is in the range of a laser layer
294   //
295 //   return kTRUE;
296   if (fPeaks[4]<2) return kTRUE; //not determined yet
297   for (Int_t i=0; i<5; ++i){
298     if (TMath::Abs((Short_t)timebin-(Short_t)fPeaks[i])<(Short_t)fPeakWidths[i]) return kTRUE;
299   }
300   return kFALSE;
301 }
302
303 #endif