]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDtrackV1.cxx
re-activate contrib code
[u/mrichter/AliRoot.git] / TRD / AliTRDtrackV1.cxx
index 156ca2621a24c018b326791d01307207fd027a47..4921d25bf9caf2268b70c5848f4fbfad72e80733 100644 (file)
 #include "AliTRDReconstructor.h"
 #include "AliTRDPIDResponse.h"
 #include "AliTRDrecoParam.h"
-#include "AliTRDdEdxUtils.h"
+#include "AliTRDdEdxBaseUtils.h"
+#include "AliTRDdEdxCalibHistArray.h"
+#include "AliTRDdEdxCalibUtils.h"
+#include "AliTRDdEdxReconUtils.h"
 
 ClassImp(AliTRDtrackV1)
 
@@ -48,6 +51,9 @@ AliTRDtrackV1::AliTRDtrackV1() : AliKalmanTrack()
   ,fESDid(0)
   ,fDE(0.)
   ,fTruncatedMean(0)
+  ,fNchamberdEdx(0)
+  ,fNclusterdEdx(0)
+  ,fNdEdxSlices(0)
   ,fkReconstructor(NULL)
   ,fBackupTrack(NULL)
   ,fTrackLow(NULL)
@@ -67,6 +73,7 @@ AliTRDtrackV1::AliTRDtrackV1() : AliKalmanTrack()
     fTrackletIndex[ip] = -1;
     fTracklet[ip]      = NULL;
   }
+  SetLabel(-123456789); // reset label
 }
 
 //_______________________________________________________________
@@ -75,6 +82,9 @@ AliTRDtrackV1::AliTRDtrackV1(const AliTRDtrackV1 &ref) : AliKalmanTrack(ref)
   ,fESDid(ref.fESDid)
   ,fDE(ref.fDE)
   ,fTruncatedMean(ref.fTruncatedMean)
+  ,fNchamberdEdx(ref.fNchamberdEdx)
+  ,fNclusterdEdx(ref.fNclusterdEdx)
+  ,fNdEdxSlices(ref.fNdEdxSlices)
   ,fkReconstructor(ref.fkReconstructor)
   ,fBackupTrack(NULL)
   ,fTrackLow(NULL)
@@ -106,6 +116,9 @@ AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack()
   ,fESDid(0)
   ,fDE(0.)
   ,fTruncatedMean(0)
+  ,fNchamberdEdx(0)                                                 
+  ,fNclusterdEdx(0)
+  ,fNdEdxSlices(0)
   ,fkReconstructor(NULL)
   ,fBackupTrack(NULL)
   ,fTrackLow(NULL)
@@ -119,7 +132,7 @@ AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack()
   SetLabel(t.GetLabel());
   SetChi2(0.0);
 
-  SetMass(t.GetMass(kTRUE));
+  SetMass(t.GetMassForTracking());
   AliKalmanTrack::SetNumberOfClusters(t.GetTRDncls()); 
   Int_t ti[]={-1, -1, -1, -1, -1, -1}; t.GetTRDtracklets(&ti[0]);
   for(int ip=0; ip<kNplane; ip++){ 
@@ -132,13 +145,15 @@ AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack()
   for(int is =0; is<AliPID::kSPECIES; is++) fPID[is] = pid;
 
   const AliExternalTrackParam *par = &t;
-  if (t.GetStatus() & AliESDtrack::kTRDbackup) { 
+  /* RS
+  if (t.GetStatus() & AliESDtrack::kTRDbackup) {
     par = t.GetOuterParam();
     if (!par) {
       AliError("No backup info!"); 
       par = &t;
     }
   }
+  */
   Set(par->GetX() 
      ,par->GetAlpha()
      ,par->GetParameter()
@@ -146,8 +161,8 @@ AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack()
 
   if(t.GetStatus() & AliESDtrack::kTIME) {
     StartTimeIntegral();
-    Double_t times[10]; 
-    t.GetIntegratedTimes(times); 
+    Double_t times[AliPID::kSPECIESC]; 
+    t.GetIntegratedTimes(times,AliPID::kSPECIESC); 
     SetIntegratedTimes(times);
     SetIntegratedLength(t.GetIntegratedLength());
   }
@@ -160,6 +175,9 @@ AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], c
   ,fESDid(0)
   ,fDE(0.)
   ,fTruncatedMean(0)
+  ,fNchamberdEdx(0)                                                 
+  ,fNclusterdEdx(0)
+  ,fNdEdxSlices(0)
   ,fkReconstructor(NULL)
   ,fBackupTrack(NULL)
   ,fTrackLow(NULL)
@@ -217,6 +235,7 @@ AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], c
   
   Float_t pid = 1./AliPID::kSPECIES;
   for(int is =0; is<AliPID::kSPECIES; is++) fPID[is] = pid;
+  SetLabel(-123456789); // reset label
 }
 
 //_______________________________________________________________
@@ -367,38 +386,36 @@ Bool_t AliTRDtrackV1::CookPID()
     AliError("PID Response not available");
     return kFALSE;
   }
-  Int_t nslices = pidResponse->GetNumberOfSlices();
-  Double_t dEdx[kNplane * (Int_t)AliTRDPIDResponse::kNslicesNN];
-  Float_t trackletP[kNplane];
-  memset(dEdx, 0, sizeof(Double_t) * kNplane * (Int_t)AliTRDPIDResponse::kNslicesNN);
-  memset(trackletP, 0, sizeof(Float_t)*kNplane);
+  Double_t dEdx[kNplane * (Int_t)AliTRDseedV1::kNdEdxSlices] = {0.};
+  Float_t trackletP[kNplane] = {0.};
+
+  fNdEdxSlices = pidResponse->GetNumberOfSlices();
   for(Int_t iseed = 0; iseed < kNplane; iseed++){
     if(!fTracklet[iseed]) continue;
     trackletP[iseed] = fTracklet[iseed]->GetMomentum();
-    fTracklet[iseed]->SetPID();
-    if(pidResponse->GetPIDmethod() == AliTRDPIDResponse::kLQ1D){
+//    if(pidResponse->GetPIDmethod() == AliTRDPIDResponse::kLQ1D){
       dEdx[iseed] = fTracklet[iseed]->GetdQdl();
-    } else {
-      fTracklet[iseed]->CookdEdx(nslices);
+/*    } else {
+      fTracklet[iseed]->CookdEdx(fNdEdxSlices);
       const Float_t *trackletdEdx = fTracklet[iseed]->GetdEdx();
-      for(Int_t islice = 0; islice < nslices; islice++){
-        dEdx[iseed*nslices + islice] = trackletdEdx[islice];
+      for(Int_t islice = 0; islice < fNdEdxSlices; islice++){
+        dEdx[iseed*fNdEdxSlices + islice] = trackletdEdx[islice];
       }
-    }
+    }*/
   }
-  pidResponse->GetResponse(nslices, dEdx, trackletP, fPID);
+  pidResponse->GetResponse(fNdEdxSlices, dEdx, trackletP, fPID);
 
   static Int_t nprint = 0;
   if(!nprint){
-    AliTRDdEdxUtils::PrintControl();
+    AliTRDdEdxBaseUtils::PrintControl();
     nprint++;
   }
 
   //do truncated mean
-  AliTRDdEdxUtils::SetObjPHQ(AliTRDcalibDB::Instance()->GetPHQ());
-  const Double_t mag    = AliTRDdEdxUtils::IsExBOn() ? GetBz()  : -1;
-  const Double_t charge = AliTRDdEdxUtils::IsExBOn() ? Charge() : -1;
-  fTruncatedMean = CookTruncatedMean(0, mag, charge, kTRUE);
+  AliTRDdEdxCalibUtils::SetObjArray(AliTRDcalibDB::Instance()->GetPHQ());
+  const Double_t mag    = AliTRDdEdxBaseUtils::IsExBOn() ? GetBz()  : -1;
+  const Double_t charge = AliTRDdEdxBaseUtils::IsExBOn() ? Charge() : -1;
+  fTruncatedMean = CookTruncatedMean(0, mag, charge, kTRUE, fNchamberdEdx, fNclusterdEdx);
 
   return kTRUE;
 }
@@ -654,6 +671,7 @@ Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho)
 
     // Energy losses
     Double_t p2    = (1.0 + GetTgl()*GetTgl()) / (GetSigned1Pt()*GetSigned1Pt());
+    if (fMass<0) p2 *= 4; // q=2
     Double_t beta2 = p2 / (p2 + fMass*fMass);
     if ((beta2 < 1.0e-10) || 
         ((5940.0 * beta2/(1.0 - beta2 + 1.0e-10) - beta2) < 0.0)) {
@@ -663,6 +681,7 @@ Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho)
     Double_t dE    = 0.153e-3 / beta2 
                    * (TMath::Log(5940.0 * beta2/(1.0 - beta2 + 1.0e-10)) - beta2)
                    * xrho;
+    if (fMass<0) dE *= 4; // q=2
     fBudget[0] += xrho;
 
     /*
@@ -891,19 +910,28 @@ void AliTRDtrackV1::UpdateESDtrack(AliESDtrack *track)
   // pack the two numbers together and store them in the ESD
   track->SetTRDntracklets(nPID | (nTrk<<3));
   // allocate space to store raw PID signals dEdx & momentum
-  track->SetNumberOfTRDslices((AliTRDPIDResponse::kNslicesNN+3)*AliTRDgeometry::kNlayer);
+  // independent of the method used to calculate PID (see below AliTRDPIDResponse)
+  track->SetNumberOfTRDslices((AliTRDseedV1::kNdEdxSlices+2)*AliTRDgeometry::kNlayer);
   // store raw signals
   Float_t p, sp; Double_t spd;
   for (Int_t ip = 0; ip < kNplane; ip++) {
     if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue;
-    if(!fTracklet[ip]->HasPID()) continue;
+    // Fill TRD dEdx info into ESD track
+    // a. Set Summed dEdx into the first slice
+    track->SetTRDslice(fTracklet[ip]->GetdQdl(), ip, 0); 
+    // b. Set NN dEdx slices
     fTracklet[ip]->CookdEdx(AliTRDPIDResponse::kNslicesNN);
     const Float_t *dedx = fTracklet[ip]->GetdEdx();
-    for (Int_t js = 0; js < AliTRDPIDResponse::kNslicesNN; js++, dedx++){
-      track->SetTRDslice(*dedx, ip, js+1);
+    for (Int_t js(0), ks(1); js < AliTRDPIDResponse::kNslicesNN; js++, ks++, dedx++){
+      if(ks>=AliTRDseedV1::kNdEdxSlices){
+        AliError(Form("Exceed allocated space for dEdx slices."));
+        break;
+      }
+      track->SetTRDslice(*dedx, ip, ks);
     }
-    p = fTracklet[ip]->GetMomentum(&sp);
-    spd = sp; track->SetTRDmomentum(p, ip, &spd);
+    // fill TRD momentum info into ESD track
+    p = fTracklet[ip]->GetMomentum(&sp); spd = sp;
+    track->SetTRDmomentum(p, ip, &spd);
     // store global quality per tracklet instead of momentum error
     // 26.09.11 A.Bercuci
     // first implementation store no. of time bins filled in tracklet (5bits  see "y" bits) and
@@ -922,36 +950,53 @@ void AliTRDtrackV1::UpdateESDtrack(AliESDtrack *track)
     Int_t nCross(fTracklet[ip]->IsRowCross()?fTracklet[ip]->GetTBcross():0); if(nCross>3) nCross = 3;
     Char_t trackletQ = Char_t(fTracklet[ip]->GetTBoccupancy() | (nCross<<5) | (fTracklet[ip]->IsChmbGood()<<7));
     track->SetTRDTimBin(trackletQ, ip);
-    track->SetTRDslice(fTracklet[ip]->GetdQdl(), ip, 0); // Set Summed dEdx into the first slice
   }
   // store PID probabilities
   track->SetTRDpid(fPID);
 
   //store truncated mean
   track->SetTRDsignal(fTruncatedMean);
+  track->SetTRDNchamberdEdx(fNchamberdEdx);
+  track->SetTRDNclusterdEdx(fNclusterdEdx);
 }
 
 //_______________________________________________________________
-Double_t  AliTRDtrackV1::CookTruncatedMean(const Bool_t kinvq, const Double_t mag, const Int_t charge, const Int_t kcalib, TVectorD *Qs, TVectorD *Xs, Int_t timeBin0, Int_t timeBin1, Int_t tstep) const
+Double_t  AliTRDtrackV1::CookTruncatedMean(const Bool_t kinvq, const Double_t mag, const Int_t charge, const Int_t kcalib, Int_t &nch, Int_t &ncls, TVectorD *Qs, TVectorD *Xs, Int_t timeBin0, Int_t timeBin1, Int_t tstep) const
 {
   //
   //Origin: Xianguo Lu <xianguo.lu@cern.ch>, Marian Ivanov <marian.ivanov@cern.ch>
   //
 
   TVectorD arrayQ(200), arrayX(200);
-  Int_t ncls = AliTRDdEdxUtils::GetArrayClusterQ(kinvq, &arrayQ, &arrayX, this, timeBin0, timeBin1, tstep);
+  ncls = AliTRDdEdxReconUtils::GetArrayClusterQ(kinvq, &arrayQ, &arrayX, this, timeBin0, timeBin1, tstep);
 
-  const TObjArray *cobj = kcalib ? AliTRDdEdxUtils::GetObjPHQ(kinvq, mag, charge) : NULL;
+  const TObjArray *cobj = kcalib ? AliTRDdEdxCalibUtils::GetObj(kinvq, mag, charge) : NULL;
   
-  const Double_t tmean = AliTRDdEdxUtils::ToyCook(kinvq, ncls, &arrayQ, &arrayX, cobj);
+  const Double_t tmean = AliTRDdEdxReconUtils::ToyCook(kinvq, ncls, &arrayQ, &arrayX, cobj);
 
-  const Int_t nch = AliTRDdEdxUtils::UpdateArrayX(ncls, &arrayX);
+  nch = AliTRDdEdxReconUtils::UpdateArrayX(ncls, &arrayX);
 
   if(Qs && Xs){
     (*Qs)=arrayQ;
     (*Xs)=arrayX;
   }
 
-  return AliTRDdEdxUtils::GetSignal(nch, ncls, tmean);
+  //printf("\ntest %.10f %d %d\n", tmean, nch, ncls);
+
+  return tmean;
 }
 
+//_______________________________________________________________
+TObject* AliTRDtrackV1::Clone(const char* newname) const
+{
+  // temporary override TObject::Clone to avoid crashes in reco
+  AliTRDtrackV1* src = (AliTRDtrackV1*)this;
+  Bool_t isown = src->IsOwner();
+  AliInfo(Form("src_owner %d",isown));
+  AliTRDtrackV1* dst = new AliTRDtrackV1(*src);
+  if (isown) {
+    src->SetBit(kOwner);
+    dst->SetOwner();
+  }
+  return dst;
+}