possibility to switch between event vertices
[u/mrichter/AliRoot.git] / PWG1 / TPC / AliPerformanceTPC.cxx
index 0df7089..3b4c792 100644 (file)
@@ -76,6 +76,7 @@ AliPerformanceTPC::AliPerformanceTPC():
 
   // histogram folder 
   fAnalysisFolder(0)
+
 {
   Init();
 }
@@ -93,6 +94,7 @@ AliPerformanceTPC::AliPerformanceTPC(Char_t* name="AliPerformanceTPC", Char_t* t
 
   // histogram folder 
   fAnalysisFolder(0)
+
 {
   // named constructor 
   // 
@@ -200,10 +202,31 @@ void AliPerformanceTPC::Init(){
 }
 
 //_____________________________________________________________________________
-void AliPerformanceTPC::ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack)
+void AliPerformanceTPC::ProcessTPC(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent *const esdEvent)
 {
+//
+// fill TPC QA info
+//
+  if(!esdEvent) return;
   if(!esdTrack) return;
 
+  if( IsUseTrackVertex() ) 
+  { 
+    // Relate TPC inner params to prim. vertex
+    const AliESDVertex *vtxESD = esdEvent->GetPrimaryVertexTracks();
+    Double_t x[3]; esdTrack->GetXYZ(x);
+    Double_t b[3]; AliTracker::GetBxByBz(x,b);
+    Bool_t isOK = esdTrack->RelateToVertexTPCBxByBz(vtxESD, b, kVeryBig);
+    if(!isOK) return;
+
+    /*
+      // JMT -- recaluclate DCA for HLT if not present
+      if ( dca[0] == 0. && dca[1] == 0. ) {
+        track->GetDZ( vtxESD->GetX(), vtxESD->GetY(), vtxESD->GetZ(), esdEvent->GetMagneticField(), dca );
+      }
+    */
+  }
+
   // Fill TPC only resolution comparison information 
   const AliExternalTrackParam *track = esdTrack->GetTPCInnerParam();
   if(!track) return;
@@ -230,7 +253,7 @@ void AliPerformanceTPC::ProcessTPC(AliStack* const stack, AliESDtrack *const esd
   Double_t dcaToVertex = -1;
   if( fCutsRC->GetDCAToVertex2D() ) 
   {
-      dcaToVertex = TMath::Sqrt(dca[0]*dca[0]/fCutsRC->GetMaxDCAToVertexXY()/fCutsRC->GetMaxDCAToVertexXY()                    + dca[1]*dca[1]/fCutsRC->GetMaxDCAToVertexZ()/fCutsRC->GetMaxDCAToVertexZ()); 
+      dcaToVertex = TMath::Sqrt(dca[0]*dca[0]/fCutsRC->GetMaxDCAToVertexXY()/fCutsRC->GetMaxDCAToVertexXY() + dca[1]*dca[1]/fCutsRC->GetMaxDCAToVertexZ()/fCutsRC->GetMaxDCAToVertexZ()); 
   }
   if(fCutsRC->GetDCAToVertex2D() && dcaToVertex > 1) return;
   if(!fCutsRC->GetDCAToVertex2D() && TMath::Abs(dca[0]) > fCutsRC->GetMaxDCAToVertexXY()) return;
@@ -247,14 +270,71 @@ void AliPerformanceTPC::ProcessTPC(AliStack* const stack, AliESDtrack *const esd
 }
 
 //_____________________________________________________________________________
-void AliPerformanceTPC::ProcessTPCITS(AliStack* const /*stack*/, AliESDtrack *const /*esdTrack*/)
+void AliPerformanceTPC::ProcessTPCITS(AliStack* const stack, AliESDtrack *const esdTrack, AliESDEvent* const esdEvent)
 {
   // Fill comparison information (TPC+ITS) 
-  AliDebug(AliLog::kWarning, "Warning: Not implemented");
+  if(!esdTrack) return;
+  if(!esdEvent) return;
+
+  if( IsUseTrackVertex() ) 
+  { 
+    // Relate TPC inner params to prim. vertex
+    const AliESDVertex *vtxESD = esdEvent->GetPrimaryVertexTracks();
+    Double_t x[3]; esdTrack->GetXYZ(x);
+    Double_t b[3]; AliTracker::GetBxByBz(x,b);
+    Bool_t isOK = esdTrack->RelateToVertexBxByBz(vtxESD, b, kVeryBig);
+    if(!isOK) return;
+
+    /*
+      // JMT -- recaluclate DCA for HLT if not present
+      if ( dca[0] == 0. && dca[1] == 0. ) {
+        track->GetDZ( vtxESD->GetX(), vtxESD->GetY(), vtxESD->GetZ(), esdEvent->GetMagneticField(), dca );
+      }
+    */
+  }
+
+  Float_t dca[2], cov[3]; // dca_xy, dca_z, sigma_xy, sigma_xy_z, sigma_z
+  esdTrack->GetImpactParameters(dca,cov);
+
+  if ((esdTrack->GetStatus()&AliESDtrack::kITSrefit)==0) return; // ITS refit
+  if ((esdTrack->GetStatus()&AliESDtrack::kTPCrefit)==0) return; // TPC refit
+
+  Float_t q = esdTrack->Charge();
+  Float_t pt = esdTrack->Pt();
+  Float_t eta = esdTrack->Eta();
+  Float_t phi = esdTrack->Phi();
+  Int_t nClust = esdTrack->GetTPCclusters(0);
+  Int_t nFindableClust = esdTrack->GetTPCNclsF();
+
+  Float_t chi2PerCluster = 0.;
+  if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);
+
+  Float_t clustPerFindClust = 0.;
+  if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;
+  
+  //
+  // select primaries
+  //
+  Double_t dcaToVertex = -1;
+  if( fCutsRC->GetDCAToVertex2D() ) 
+  {
+      dcaToVertex = TMath::Sqrt(dca[0]*dca[0]/fCutsRC->GetMaxDCAToVertexXY()/fCutsRC->GetMaxDCAToVertexXY() + dca[1]*dca[1]/fCutsRC->GetMaxDCAToVertexZ()/fCutsRC->GetMaxDCAToVertexZ()); 
+  }
+  if(fCutsRC->GetDCAToVertex2D() && dcaToVertex > 1) return;
+  if(!fCutsRC->GetDCAToVertex2D() && TMath::Abs(dca[0]) > fCutsRC->GetMaxDCAToVertexXY()) return;
+  if(!fCutsRC->GetDCAToVertex2D() && TMath::Abs(dca[1]) > fCutsRC->GetMaxDCAToVertexZ()) return;
+
+  Double_t vTPCTrackHisto[9] = {nClust,chi2PerCluster,clustPerFindClust,dca[0],dca[1],eta,phi,pt,q};
+  fTPCTrackHisto->Fill(vTPCTrackHisto); 
+  //
+  // Fill rec vs MC information
+  //
+  if(!stack) return;
 }
  
 //_____________________________________________________________________________
-void AliPerformanceTPC::ProcessConstrained(AliStack* const /*stack*/, AliESDtrack *const /*esdTrack*/)
+void AliPerformanceTPC::ProcessConstrained(AliStack* const /*stack*/, AliESDtrack *const /*esdTrack*/, AliESDEvent* const /*esdEvent*/)
 {
   // Fill comparison information (constarained parameters) 
   AliDebug(AliLog::kWarning, "Warning: Not implemented");
@@ -302,14 +382,6 @@ void AliPerformanceTPC::Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEv
     genHeader->PrimaryVertex(vtxMC);
   } 
   
-  // use ESD friends
-  if(bUseESDfriend) {
-    if(!esdFriend) {
-      Error("Exec","esdFriend not available");
-      return;
-    }
-  }
-
   // trigger
   if(!bUseMC &&GetTriggerClass()) {
     Bool_t isEventTriggered = esdEvent->IsTriggerClassFired(GetTriggerClass());
@@ -317,7 +389,13 @@ void AliPerformanceTPC::Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEv
   }
 
   // get TPC event vertex
-  const AliESDVertex *vtxESD = esdEvent->GetPrimaryVertexTPC();
+  const AliESDVertex *vtxESD = NULL; 
+  if(fUseTrackVertex) {
+    vtxESD = esdEvent->GetPrimaryVertexTracks();
+  } else {
+    vtxESD = esdEvent->GetPrimaryVertexTPC();
+  }
+  if(!vtxESD) return;
 
   //  events with rec. vertex
   Int_t mult=0; Int_t multP=0; Int_t multN=0;
@@ -329,16 +407,18 @@ void AliPerformanceTPC::Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEv
     AliESDtrack *track = esdEvent->GetTrack(iTrack);
     if(!track) continue;
 
-    if(bUseESDfriend) {
-    AliESDfriendTrack *friendTrack=esdFriend->GetTrack(iTrack);
-    if(!friendTrack) continue;
-
-      TObject *calibObject=0;
-      AliTPCseed *seed=0;
-      if (!friendTrack) continue;
-        for (Int_t j=0;(calibObject=friendTrack->GetCalibObject(j));++j) {
-         if ((seed=dynamic_cast<AliTPCseed*>(calibObject)))
-         break;
+    if(bUseESDfriend && esdFriend) 
+    {
+      AliESDfriendTrack *friendTrack=esdFriend->GetTrack(iTrack);
+      if(friendTrack) 
+      {
+        TObject *calibObject=0;
+        AliTPCseed *seed=0;
+        if (!friendTrack) continue;
+          for (Int_t j=0;(calibObject=friendTrack->GetCalibObject(j));++j) {
+           if ((seed=dynamic_cast<AliTPCseed*>(calibObject))) {
+           break;
+         }
         }
 
         //AliTPCTransform *transform = AliTPCcalibDB::Instance()->GetTransform() ;
@@ -364,10 +444,12 @@ void AliPerformanceTPC::Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEv
              Double_t vTPCClust[3] = { gclf[0], gclf[1],  TPCside };
              fTPCClustHisto->Fill(vTPCClust);
          }
+       }
     }
-    if(GetAnalysisMode() == 0) ProcessTPC(stack,track);
-    else if(GetAnalysisMode() == 1) ProcessTPCITS(stack,track);
-    else if(GetAnalysisMode() == 2) ProcessConstrained(stack,track);
+
+    if(GetAnalysisMode() == 0) ProcessTPC(stack,track,esdEvent);
+    else if(GetAnalysisMode() == 1) ProcessTPCITS(stack,track,esdEvent);
+    else if(GetAnalysisMode() == 2) ProcessConstrained(stack,track,esdEvent);
     else {
       printf("ERROR: AnalysisMode %d \n",fAnalysisMode);
       return;