switch to generated q-vector
authorlmilano <lmilano@cern.ch>
Sat, 4 Oct 2014 11:35:48 +0000 (13:35 +0200)
committerlmilano <lmilano@cern.ch>
Sat, 4 Oct 2014 11:35:48 +0000 (13:35 +0200)
PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.cxx
PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h

index fdc934c..5e9529b 100644 (file)
@@ -71,7 +71,8 @@ AliAnalysisTaskSpectraAllChAOD::AliAnalysisTaskSpectraAllChAOD(const char *name)
   fnDCABins(60),
   fDCAmin(-3),
   fDCAmax(3),
-  fDCAzCut(999999.)
+  fDCAzCut(999999.),
+  fQvecGen(0)
 {
   // Default constructor
   DefineInput(0, TChain::Class());
@@ -206,7 +207,7 @@ void AliAnalysisTaskSpectraAllChAOD::UserExec(Option_t *)
     if(fIsQvecCalibMode){
       QvecMC = fEventCuts->CalculateQVectorMC(fVZEROside);
     }
-  else QvecMC = fEventCuts->GetQvecPercentileMC(fVZEROside);
+    else QvecMC = fEventCuts->GetQvecPercentileMC(fVZEROside);
   }
   
   Double_t Cent=fEventCuts->GetCent();
@@ -291,7 +292,8 @@ void AliAnalysisTaskSpectraAllChAOD::UserExec(Option_t *)
       Double_t varTrk[8];
       varTrk[0]=track->Pt();
       varTrk[1]=Cent;
-      varTrk[2]=Qvec;
+      if(fIsMC && fQvecGen) varTrk[2]=QvecMC;
+        else varTrk[2]=Qvec;
       varTrk[3]=(Double_t)IDrec;
       varTrk[4]=(Double_t)IDgen;
       varTrk[5]=(Double_t)isph;
index 905744d..4b969b5 100644 (file)
@@ -43,7 +43,8 @@ class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE
     fnDCABins(60),
     fDCAmin(-3),
     fDCAmax(3),
-    fDCAzCut(0)
+    fDCAzCut(0),
+    fQvecGen(0)
       {}
   AliAnalysisTaskSpectraAllChAOD(const char *name);
   virtual ~AliAnalysisTaskSpectraAllChAOD() {
@@ -89,6 +90,8 @@ class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE
   void SetDCAmax(Double_t val)                        { fDCAmax = val; }
   Bool_t GetDCA(const AliAODTrack* trk, Double_t * p);
   void SetDCAzCut(Double_t val)                        { fDCAzCut = val; }
+  
+  void GetQvecGen(Bool_t val) { fQvecGen = val; } //enable Qvec from generated
 
  private:
   
@@ -112,10 +115,13 @@ class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE
   Double_t                         fDCAmin;                // min DCA value
   Double_t                         fDCAmax;                // max DCA value
   Double_t                         fDCAzCut;               //cut on DCA z
+  
+  Bool_t fQvecGen; //enable Qvec from generated
+  
   AliAnalysisTaskSpectraAllChAOD(const AliAnalysisTaskSpectraAllChAOD&);
   AliAnalysisTaskSpectraAllChAOD& operator=(const AliAnalysisTaskSpectraAllChAOD&);
   
-  ClassDef(AliAnalysisTaskSpectraAllChAOD, 9);
+  ClassDef(AliAnalysisTaskSpectraAllChAOD, 10);
 };
 
 #endif