1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
17 // TRD dEdx recon utils
24 // lu@physi.uni-heidelberg.de
33 #include "THnSparse.h"
37 #include "TObjArray.h"
39 #include "TStopwatch.h"
42 #include "TTreeStream.h"
45 #include "AliCDBMetaData.h"
46 #include "AliCDBStorage.h"
47 #include "AliESDEvent.h"
48 #include "AliESDfriendTrack.h"
49 #include "AliESDtrack.h"
50 #include "AliTRDcalibDB.h"
51 #include "AliTRDCalROC.h"
52 #include "AliTRDtrackV1.h"
54 #include "AliTRDdEdxBaseUtils.h"
55 #include "AliTRDdEdxReconUtils.h"
59 Int_t AliTRDdEdxReconUtils::ApplyCalib(const Int_t nc0, TVectorD *arrayQ, TVectorD *arrayX, const TObjArray *cobj)
62 //apply calibration on arrayQ
64 if(!cobj){ printf("AliTRDdEdxReconUtils::ApplyCalib error gain array null!!\n"); exit(1);}
66 TVectorD tmpq(arrayQ->GetNrows());
67 TVectorD tmpx(arrayX->GetNrows());
70 const TVectorD * gain = (TVectorD*) cobj->At(0);
71 for(Int_t ii=0; ii<nc0; ii++){
72 const Double_t dq = (*arrayQ)[ii];
73 const Int_t xx = (Int_t)(*arrayX)[ii];
74 const Double_t gg = (*gain)[xx];
91 Double_t AliTRDdEdxReconUtils::ToyCook(const Bool_t kinvq, Int_t &ncluster, TVectorD *arrayQ, TVectorD *arrayX, const TObjArray *cobj)
94 //template for cookdedx
98 ncluster = ApplyCalib(ncluster, arrayQ, arrayX, cobj);
101 printf("AliTRDdEdxReconUtils::ToyCook arrayQ arrayX null, applycalib can not be applied!\n"); exit(1);
105 Double_t lowFrac =-999, highFrac = -999;
107 lowFrac = AliTRDdEdxBaseUtils::Q1Frac(); highFrac = 0.99;
110 lowFrac = 0.01; highFrac = AliTRDdEdxBaseUtils::Q0Frac();
113 Double_t meanQ = AliTRDdEdxBaseUtils::TruncatedMean(ncluster, arrayQ->GetMatrixArray(), lowFrac, highFrac);
123 Double_t AliTRDdEdxReconUtils::CombineddEdx(const Bool_t kinvq, Int_t &concls, TVectorD *coarrayQ, TVectorD *coarrayX, const Int_t tpcncls, const TVectorD *tpcarrayQ, const TVectorD *tpcarrayX, const Int_t trdncls, const TVectorD *trdarrayQ, const TVectorD *trdarrayX)
126 //combine tpc and trd dedx
129 for(Int_t iq=0; iq<tpcncls; iq++){
130 (*coarrayQ)[iq]=(*tpcarrayQ)[iq];
131 if(tpcarrayX && trdarrayX && coarrayX){
132 (*coarrayX)[iq]=(*tpcarrayX)[iq];
135 for(Int_t iq=0; iq<trdncls; iq++){
136 (*coarrayQ)[tpcncls+iq]=(*trdarrayQ)[iq];
137 if(tpcarrayX && trdarrayX && coarrayX){
138 (*coarrayX)[tpcncls+iq]=159+(*trdarrayX)[iq];
142 concls=trdncls+tpcncls;
144 const Double_t coQ = ToyCook(kinvq, concls, coarrayQ, coarrayX);
149 Double_t AliTRDdEdxReconUtils::GetAngularCorrection(const AliTRDseedV1 *seed)
152 //return angular normalization factor
155 return TMath::Sqrt(1+seed->GetYref(1)*seed->GetYref(1)+seed->GetZref(1)*seed->GetZref(1));
158 Double_t AliTRDdEdxReconUtils::GetPadGain(const Int_t det, const Int_t icol, const Int_t irow)
161 //get pad calibration
163 AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
165 printf("AliTRDdEdxReconUtils::GetPadCalib No AliTRDcalibDB instance available\n"); exit(1);
167 AliTRDCalROC * calGainFactorROC = calibration->GetGainFactorROC(det);
168 if(!calGainFactorROC){
169 printf("AliTRDdEdxReconUtils::GetPadCalib no calGainFactorROC!\n"); exit(1);
172 Double_t padgain = -999;
174 icol < calGainFactorROC->GetNcols() &&
176 irow < calGainFactorROC->GetNrows()){
177 padgain = calGainFactorROC->GetValue(icol, irow);
179 printf("AliTRDdEdxReconUtils::GetPadGain padgain 0! %f %f -- %d %d %d -- %d %d\n", padgain, EPSILON, det, icol, irow, calGainFactorROC->GetNcols(), calGainFactorROC->GetNrows()); exit(1);
183 //printf("\nAliTRDdEdxReconUtils::GetPadGain warning!! indices out of range %d %d %d -- %d %d\n\n", det, icol, irow, calGainFactorROC->GetNcols(), calGainFactorROC->GetNrows() );
189 Double_t AliTRDdEdxReconUtils::GetRNDClusterQ(AliTRDcluster *cl)
192 //get cluter q from GetRawQ, apply baseline and Kr pad-calibration
195 const Int_t det = cl->GetDetector();
196 const Int_t pad3col = cl->GetPadCol();
197 const Int_t padrow = cl->GetPadRow();
199 const Double_t baseline = 10;
201 Double_t rndqsum = 0;
202 for(Int_t ii=0; ii<7; ii++){
203 if(cl->GetSignals()[ii] < EPSILON){//bad pad marked by electronics
207 const Int_t icol = pad3col+(ii-3);
208 const Double_t padgain = GetPadGain(det, icol, padrow);
209 if(padgain<0){//indices out of range, pad3col near boundary case
213 const Double_t rndsignal = (cl->GetSignals()[ii] - baseline)/(AliTRDdEdxBaseUtils::IsPadGainOn()? padgain : 1);
215 //sum it anyway even if signal below baseline, as long as the total is positive
216 rndqsum += rndsignal;
222 Double_t AliTRDdEdxReconUtils::GetClusterQ(const Bool_t kinvq, const AliTRDseedV1 * seed, const Int_t itb)
228 AliTRDcluster *cl = 0x0;
230 //GetRNDClusterQ(cl)>0 ensures that the total sum of q is above baseline*NsignalPhysical.
231 cl = seed->GetClusters(itb); if(cl && GetRNDClusterQ(cl)>0 ) dq+= GetRNDClusterQ(cl);//cl->GetRawQ();
232 cl = seed->GetClusters(itb+AliTRDseedV1::kNtb); if(cl && GetRNDClusterQ(cl)>0 ) dq+= GetRNDClusterQ(cl);//cl->GetRawQ();
234 dq /= GetAngularCorrection(seed);
236 dq /= AliTRDdEdxBaseUtils::QScale();
247 Int_t AliTRDdEdxReconUtils::GetArrayClusterQ(const Bool_t kinvq, TVectorD *arrayQ, TVectorD *arrayX, const AliTRDtrackV1 *trdtrack, Int_t timeBin0, Int_t timeBin1, Int_t tstep)
251 //(if kinvq, return 1/q array), size of array must be larger than 31*6
253 if(!arrayQ || arrayQ->GetNrows()< (AliTRDseedV1::kNtb*AliTRDtrackV1::kNplane)){
254 printf("AliTRDdEdxReconUtils::GetArrayClusterQ error arrayQ null or size too small! %d\n", arrayQ? arrayQ->GetNrows() : -999); exit(1);
256 if(!arrayX || arrayX->GetNrows()< (AliTRDseedV1::kNtb*AliTRDtrackV1::kNplane)){
257 printf("AliTRDdEdxReconUtils::GetArrayClusterQ error arrayX null or size too small! %d\n", arrayX? arrayX->GetNrows() : -999); exit(1);
260 const Int_t mintb = 0;
261 const Int_t maxtb = AliTRDseedV1::kNtb-1;
262 if(timeBin0<mintb) timeBin0=mintb;
263 if(timeBin1>maxtb) timeBin1=maxtb;
264 if(tstep<=0) tstep=1;
271 for(Int_t ichamber=0; ichamber < AliTRDtrackV1::kNplane; ichamber++){
272 const AliTRDseedV1 * seed = trdtrack->GetTracklet(ichamber);
276 const Int_t det = seed->GetDetector();
278 for(Int_t itb=timeBin0; itb<=timeBin1; itb+=tstep){
279 const Double_t dq = GetClusterQ(kinvq, seed, itb);
283 const Int_t gtb = det * AliTRDseedV1::kNtb + itb;
292 for(Int_t iq=0; iq<tbN; iq++){
296 (*arrayQ)[ncls] = tbQ[iq];
297 (*arrayX)[ncls] = tbBin[iq];
302 static Int_t kprint = 100;
304 printf("\nAliTRDdEdxReconUtils::GetArrayClusterQ raw cluster-Q\n");
305 for(Int_t iq=0; iq<ncls; iq++){
306 const Int_t ichamber = AliTRDdEdxBaseUtils::ToLayer((*arrayX)[iq]);
307 const AliTRDseedV1 * seed = trdtrack->GetTracklet(ichamber);
309 printf("error seed null!!\n"); exit(1);
311 const Double_t rawq = (*arrayQ)[iq] * 45. * GetAngularCorrection(seed);
312 printf("esdid=%d; chamber=%d; timebin=%d; rawq= %.3f; myq[%d]= %e;\n", trdtrack->GetESDid(), ichamber, AliTRDdEdxBaseUtils::ToTimeBin((*arrayX)[iq]), rawq, iq, (*arrayQ)[iq]);
321 Int_t AliTRDdEdxReconUtils::UpdateArrayX(const Int_t ncls, TVectorD* arrayX)
324 //arrayX det*Ntb+itb -> itb
327 TVectorD countChamber(6);
328 for(Int_t ii = 0; ii<ncls; ii++){
329 const Int_t xx = (Int_t)(*arrayX)[ii];
330 const Int_t idet = AliTRDdEdxBaseUtils::ToDetector(xx);
332 const Double_t ich = AliTRDgeometry::GetLayer(idet);
333 const Double_t itb = AliTRDdEdxBaseUtils::ToTimeBin(xx);
334 (*arrayX)[ii] = ich*AliTRDseedV1::kNtb+itb;
336 countChamber[ich] = 1;
339 const Double_t nch = countChamber.Sum();