up from Rosi

[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskPIDCombined.cxx

diff --git a/ANALYSIS/AliAnalysisTaskPIDCombined.cxx b/ANALYSIS/AliAnalysisTaskPIDCombined.cxx
index 24ff6780595da04df7667aaa91bd9a56918218fa..465b1cec426fe9d785e73ed2fd0be2d70ce0408f 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskPIDCombined.cxx
+++ b/ANALYSIS/AliAnalysisTaskPIDCombined.cxx
@@ -64,7 +64,9 @@ AliAnalysisTaskPIDCombined::AliAnalysisTaskPIDCombined() :
   fPIDResponse(0x0),\r
   fPIDCombined(0x0),\r
   fTrackCuts(0x0),\r
-  fTrackFilter(0x0)\r
+  fTrackFilter(0x0),\r
+  fDeDx(NULL),
+  fDeDxTuned(NULL)
 {\r
   //\r
   // Constructor\r
@@ -89,7 +91,9 @@ AliAnalysisTaskPIDCombined::AliAnalysisTaskPIDCombined(const char *name) :
   fPIDResponse(0x0),\r
   fPIDCombined(0x0),\r
   fTrackCuts(0x0),\r
-  fTrackFilter(0x0)\r
+  fTrackFilter(0x0),\r
+  fDeDx(NULL),
+  fDeDxTuned(NULL)
 {\r
   //\r
   // Constructor\r
@@ -126,13 +130,13 @@ void AliAnalysisTaskPIDCombined::UserCreateOutputObjects()
   fPIDCombined->SetDefaultTPCPriors();\r
   fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTPC+AliPIDResponse::kDetTOF);\r
 \r
-  // no light nuclei\r
-  fPIDCombined->SetSelectedSpecies(AliPID::kSPECIES);\r
+  // no light nuclei - no need to call it, this is default\r
+  //  fPIDCombined->SetSelectedSpecies(AliPID::kSPECIES);\r
 \r
 \r
   fHistList.Add(new TH1D("nEvents","Number of Evnets;Selection",2,0,2));\r
 \r
-  for (Int_t ispec=0; ispec<AliPID::kSPECIES; ++ispec){\r
+  for (Int_t ispec=0; ispec<5; ++ispec){\r
 \r
  \r
     fProbTPC[ispec]=new TH2D(Form("prob%s_mom_TPC",AliPID::ParticleName(ispec)),\r
@@ -221,7 +225,11 @@ void AliAnalysisTaskPIDCombined::UserCreateOutputObjects()
   }\r
 \r
 \r
-\r
+  fDeDx = new TH2D("hDeDx",";p_{TPC};dE/dx (a.u.)",500,0,5,500,0,500);
+  fHistList.Add(fDeDx);
+  fDeDxTuned = new TH2D("hDeDxTuned",";p_{TPC};dE/dx (a.u.)",500,0,5,500,0,500);
+  fHistList.Add(fDeDxTuned);
+
   fHistList.SetOwner();\r
   PostData(1,&fHistList);\r
 \r
@@ -301,6 +309,10 @@ void AliAnalysisTaskPIDCombined::UserExec(Option_t *)
 \r
       }\r
 \r
+      fPIDResponse->GetTPCsignalTunedOnData(track);
+
+      fDeDx->Fill(mom,track->GetTPCsignal());
+      fDeDxTuned->Fill(mom,track->GetTPCsignalTunedOnData());
 \r
     }\r
   }\r