From 95a1cff1c1b223b912c1831b6a8bb20026853cfd Mon Sep 17 00:00:00 2001
From: dgangadh <dhevan.raja.gangadharan@cern.ch>
Date: Mon, 13 Jan 2014 15:48:45 +0100
Subject: [PATCH 1/1] K0s analysis update (Matt Steinpreis)

---
 .../K0Analysis/AliFemtoK0Analysis.cxx         | 29 ++++++++++++++++++-
 .../K0Analysis/AliFemtoK0EventCollection.cxx  |  3 ++
 .../K0Analysis/AliFemtoK0EventCollection.h    |  1 +
 3 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0Analysis.cxx b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0Analysis.cxx
index daae5813f60..7282188c7f7 100644
--- a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0Analysis.cxx
+++ b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0Analysis.cxx
@@ -374,8 +374,20 @@ void AliFemtoK0Analysis::UserCreateOutputObjects()
 
   TH1F *fHistPsi = new TH1F("fHistPsi","",90,-PI/2,PI/2);
   fOutputList->Add(fHistPsi);
+  TH2F *fHistPhi = new TH2F("fHistPhi","",kCentBins,.5,kCentBins+.5,180,0,2*PI);
+  fOutputList->Add(fHistPhi);
   TH2F *fHistPhiPsi = new TH2F("fHistPhiPsi","",kCentBins,.5,kCentBins+.5,180,0,2*PI);
   fOutputList->Add(fHistPhiPsi);
+  
+
+  TH3F *fHistDPhi = new TH3F("fHistDPhi","",kCentBins,.5,kCentBins+.5,200,0.,2.,90,0,PI);
+  TH3F *fHistDPhiBkg = new TH3F("fHistDPhiBkg","",kCentBins,.5,kCentBins+.5,200,0.,2.,90,0,PI);
+  TH3F *fHistDPhiPsi = new TH3F("fHistDPhiPsi","",kCentBins,.5,kCentBins+.5,200,0.,2.,90,0,PI);
+  TH3F *fHistDPhiPsiBkg = new TH3F("fHistDPhiPsiBkg","",kCentBins,.5,kCentBins+.5,200,0.,2.,90,0,PI);
+  fOutputList->Add(fHistDPhi);
+  fOutputList->Add(fHistDPhiBkg);
+  fOutputList->Add(fHistDPhiPsi);
+  fOutputList->Add(fHistDPhiPsiBkg);
 
 /////////Pair Distributions///////////////////
 
@@ -830,6 +842,7 @@ void AliFemtoK0Analysis::Exec(Option_t *)
         if(v0PhiPsi < 0) v0PhiPsi += 2.*PI;
         else if (v0PhiPsi > 2.*PI) v0PhiPsi -= 2.*PI;
 		else{};
+        tempK0[v0Count].fPhi	= v0Phi;
         tempK0[v0Count].fPhiPsi = v0PhiPsi; 
         
         //for separation
@@ -918,6 +931,7 @@ void AliFemtoK0Analysis::Exec(Option_t *)
     ((TH1F*)fOutputList->FindObject("fHistPy"))		->Fill((fEvt)->fK0Particle[i].fMomentum[1]);
     ((TH1F*)fOutputList->FindObject("fHistPz"))		->Fill((fEvt)->fK0Particle[i].fMomentum[2]);
 
+    ((TH2F*)fOutputList->FindObject("fHistPhi"))	->Fill(centBin+1,(fEvt)->fK0Particle[i].fPhi);
     ((TH2F*)fOutputList->FindObject("fHistPhiPsi"))	->Fill(centBin+1,(fEvt)->fK0Particle[i].fPhiPsi);
 
     for(int evnum=0; evnum<kEventsToMix+1; evnum++)// Event buffer loop: evnum=0 is the current event, all other evnum's are past events
@@ -1144,7 +1158,12 @@ void AliFemtoK0Analysis::Exec(Option_t *)
         if(phipsi2 > PI) phipsi2 = phipsi2-PI;
         if(phipsi1 < 0.25*PI || phipsi1 > 0.75*PI) inPlane1 = kTRUE;
         if(phipsi2 < 0.25*PI || phipsi2 > 0.75*PI) inPlane2 = kTRUE;
-	
+
+		//for dphi, dphipsi
+		float dPhi = fabs((fEvt)->fK0Particle[i].fPhi - (fEvt+evnum)->fK0Particle[j].fPhi);
+        if(dPhi > PI) dPhi = 2*PI-dPhi;
+        float dPhiPsi = fabs((fEvt)->fK0Particle[i].fPhiPsi - (fEvt+evnum)->fK0Particle[j].fPhiPsi);
+        if(dPhiPsi > PI) dPhiPsi = 2*PI-dPhiPsi;
 
         if(evnum==0) //Same Event
         {     
@@ -1166,6 +1185,11 @@ void AliFemtoK0Analysis::Exec(Option_t *)
              ((TH3F*)fOutputList->FindObject("fHistQinvSignalEPIn"))->Fill(centBin+1, pairKt, qinv);
            else if(!inPlane1 && !inPlane2)
              ((TH3F*)fOutputList->FindObject("fHistQinvSignalEPOut"))->Fill(centBin+1, pairKt, qinv);
+
+           //dPhi,dPhiPsi
+		   ((TH3F*)fOutputList->FindObject("fHistDPhi"))->Fill(centBin+1,pairKt,dPhi);
+		   ((TH3F*)fOutputList->FindObject("fHistDPhiPsi"))->Fill(centBin+1,pairKt,dPhiPsi);
+
            
            //for mass bin study
            //if(CL1 && CL2) ((TH3F*)fOutputList->FindObject("fHistCLCLSignal"))->Fill(centBin+1, pairKt, qinv);	   
@@ -1223,6 +1247,9 @@ void AliFemtoK0Analysis::Exec(Option_t *)
            else if(!inPlane1 && !inPlane2)
              ((TH3F*)fOutputList->FindObject("fHistQinvBkgEPOut"))->Fill(centBin+1, pairKt, qinv);
 
+		   ((TH3F*)fOutputList->FindObject("fHistDPhiBkg"))->Fill(centBin+1,pairKt,dPhi);
+		   ((TH3F*)fOutputList->FindObject("fHistDPhiPsiBkg"))->Fill(centBin+1,pairKt,dPhiPsi);
+
            //for mass bin study
            //if(CL1 && CL2) ((TH3F*)fOutputList->FindObject("fHistCLCLBkg"))->Fill(centBin+1, pairKt, qinv);	   
            //else if ((CL1 && CR2) || (CR1 && CL2)) ((TH3F*)fOutputList->FindObject("fHistCLCRBkg"))->Fill(centBin+1, pairKt, qinv);
diff --git a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.cxx b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.cxx
index a963694445d..4e83d14627b 100644
--- a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.cxx
+++ b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.cxx
@@ -47,6 +47,7 @@ AliFemtoK0Particle::AliFemtoK0Particle() :
  fPNeg(),
  fPosXYZ(),
  fNegXYZ(),
+ fPhi(0),
  fPhiPsi(0)
 {
   //Default constructor
@@ -75,6 +76,7 @@ AliFemtoK0Particle::AliFemtoK0Particle(const AliFemtoK0Particle &obj) :
  fPNeg(),
  fPosXYZ(),
  fNegXYZ(),
+ fPhi(),
  fPhiPsi()
 
 {
@@ -114,6 +116,7 @@ AliFemtoK0Particle &AliFemtoK0Particle::operator=(const AliFemtoK0Particle &obj)
    fNegXYZ[j][i] = obj.fNegXYZ[j][i];
   }
  }
+ fPhi = obj.fPhi;
  fPhiPsi = obj.fPhiPsi;
  return (*this);
 }
diff --git a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.h b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.h
index 860144e4ccc..a48268c53b1 100644
--- a/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.h
+++ b/PWGCF/FEMTOSCOPY/K0Analysis/AliFemtoK0EventCollection.h
@@ -59,6 +59,7 @@ class AliFemtoK0Particle  // Reconstructed K0s parameters needed for correlation
   float fNegXYZ[9][3];	//corrected daughter TPC positions
 
   //for event plane
+  double fPhi;
   double fPhiPsi;
 
   ClassDef(AliFemtoK0Particle, 1);
-- 
2.39.3