switch to generated q-vector

diff --git a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.cxx b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.cxx
index fdc934cf42f9380fded304ff41e08d8b2c828634..5e9529b1535c0cf707e480e160350d7b984a5053 100644 (file)
--- a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.cxx
+++ b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.cxx
@@ -71,7 +71,8 @@ AliAnalysisTaskSpectraAllChAOD::AliAnalysisTaskSpectraAllChAOD(const char *name)
   fnDCABins(60),
   fDCAmin(-3),
   fDCAmax(3),
   fnDCABins(60),
   fDCAmin(-3),
   fDCAmax(3),

-  fDCAzCut(999999.)
+  fDCAzCut(999999.),
+  fQvecGen(0)

 {
   // Default constructor
   DefineInput(0, TChain::Class());
 {
   // Default constructor
   DefineInput(0, TChain::Class());


@@ -206,7 +207,7 @@ void AliAnalysisTaskSpectraAllChAOD::UserExec(Option_t *)
     if(fIsQvecCalibMode){
       QvecMC = fEventCuts->CalculateQVectorMC(fVZEROside);
     }
     if(fIsQvecCalibMode){
       QvecMC = fEventCuts->CalculateQVectorMC(fVZEROside);
     }

-  else QvecMC = fEventCuts->GetQvecPercentileMC(fVZEROside);
+    else QvecMC = fEventCuts->GetQvecPercentileMC(fVZEROside);

   }
   
   Double_t Cent=fEventCuts->GetCent();
   }
   
   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;
       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;
       varTrk[3]=(Double_t)IDrec;
       varTrk[4]=(Double_t)IDgen;
       varTrk[5]=(Double_t)isph;

diff --git a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h
index 905744de1a2e20591b71014a68de5df6c0c36317..4b969b5d6cf240a92ea22f015ebc9b3afe9fa8ae 100644 (file)
--- a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h
+++ b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h
@@ -43,7 +43,8 @@ class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE
     fnDCABins(60),
     fDCAmin(-3),
     fDCAmax(3),
     fnDCABins(60),
     fDCAmin(-3),
     fDCAmax(3),

-    fDCAzCut(0)
+    fDCAzCut(0),
+    fQvecGen(0)

       {}
   AliAnalysisTaskSpectraAllChAOD(const char *name);
   virtual ~AliAnalysisTaskSpectraAllChAOD() {
       {}
   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 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:
   
 
  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
   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&);
   
   AliAnalysisTaskSpectraAllChAOD(const AliAnalysisTaskSpectraAllChAOD&);
   AliAnalysisTaskSpectraAllChAOD& operator=(const AliAnalysisTaskSpectraAllChAOD&);
   

-  ClassDef(AliAnalysisTaskSpectraAllChAOD, 9);
+  ClassDef(AliAnalysisTaskSpectraAllChAOD, 10);

 };
 
 #endif
 };
 
 #endif