updated for nonHFE reconstruction study

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskFlowTPCTOFQCSP.cxx

diff --git a/PWGHF/hfe/AliAnalysisTaskFlowTPCTOFQCSP.cxx b/PWGHF/hfe/AliAnalysisTaskFlowTPCTOFQCSP.cxx
index c8b7548c6cd631c471483691a3837ac5f7a5648e..2d9fecef0c6607fada2d1fb571380f54acbc6b19 100644 (file)
--- a/PWGHF/hfe/AliAnalysisTaskFlowTPCTOFQCSP.cxx
+++ b/PWGHF/hfe/AliAnalysisTaskFlowTPCTOFQCSP.cxx
@@ -123,6 +123,7 @@ ClassImp(AliAnalysisTaskFlowTPCTOFQCSP)
 ,fCentralityMin(0)
 ,fCentralityMax(0)
 ,fInvmassCut(0)
+,fpTCut(0)
 ,fTrigger(0)
 ,fPhi(0)
 ,fEta(0)
@@ -164,6 +165,10 @@ ClassImp(AliAnalysisTaskFlowTPCTOFQCSP)
 ,fTPCS(0)
 ,fVz(0)
 ,fPhiCut(0)
+,fOpeningAngleCut(0.1)
+,fOP_angle(0)
+,fOpeningAngleLS(0)
+,fOpeningAngleULS(0)
 {
   //Named constructor
 
@@ -198,6 +203,7 @@ AliAnalysisTaskFlowTPCTOFQCSP::AliAnalysisTaskFlowTPCTOFQCSP()
 ,fCentralityMin(0)
 ,fCentralityMax(0)
 ,fInvmassCut(0)
+,fpTCut(0)
 ,fTrigger(0)
 ,fPhi(0)
 ,fEta(0)
@@ -239,6 +245,10 @@ AliAnalysisTaskFlowTPCTOFQCSP::AliAnalysisTaskFlowTPCTOFQCSP()
 ,fTPCS(0)
 ,fVz(0)
 ,fPhiCut(0)
+,fOpeningAngleCut(0.1)
+,fOP_angle(0)
+,fOpeningAngleLS(0)
+,fOpeningAngleULS(0)
 {
   //Default constructor
 //  fPID = new AliHFEpid("hfePid");
@@ -447,7 +457,7 @@ if(!(((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInput
      Double_t eta = track->Eta();
      Double_t phi = track->Phi();
      Double_t pt = track->Pt();         //pt track after cuts
-     if(pt<1.) continue;
+     if(pt<fpTCut) continue;
 //==========================================================================================================
 //=========================================PID==============================================================
      if(track->GetTPCsignalN() < fTPCS) continue;
@@ -571,6 +581,7 @@ void AliAnalysisTaskFlowTPCTOFQCSP::SelectPhotonicElectron(Int_t itrack,const Al
     if(jTracks == itrack) continue;
     Double_t ptAsso=-999., nsigma=-999.0;
     Double_t mass=-999., width = -999;
+      Double_t openingAngle = -999.;
     Bool_t fFlagLS=kFALSE, fFlagULS=kFALSE;
 
   
@@ -601,6 +612,14 @@ void AliAnalysisTaskFlowTPCTOFQCSP::SelectPhotonicElectron(Int_t itrack,const Al
     if(TMath::Sqrt(TMath::Abs(chi2recg))>3.) continue;
     recg.GetMass(mass,width);
 
+      if(fOP_angle){
+    openingAngle = ge1.GetAngle(ge2);
+    if(fFlagLS) fOpeningAngleLS->Fill(openingAngle);
+    if(fFlagULS) fOpeningAngleULS->Fill(openingAngle);
+    if(openingAngle > fOpeningAngleCut) continue;
+      }
+      
+      
     if(fFlagLS) fInvmassLS1->Fill(mass);
     if(fFlagULS) fInvmassULS1->Fill(mass);
            
@@ -764,6 +783,14 @@ void AliAnalysisTaskFlowTPCTOFQCSP::UserCreateOutputObjects()
   fMultvsCentr = new TH2F("fMultvsCentr", "Multiplicity vs centrality; centrality; Multiplicity", 100, 0., 100, 100, 0, 3000);
   fOutputList->Add(fMultvsCentr);
     
+  fOpeningAngleLS = new TH1F("fOpeningAngleLS","Opening angle for LS pairs",100,0,1);
+  fOutputList->Add(fOpeningAngleLS);
+    
+  fOpeningAngleULS = new TH1F("fOpeningAngleULS","Opening angle for ULS pairs",100,0,1);
+  fOutputList->Add(fOpeningAngleULS);
+  
+    
+    
 //----------------------------------------------------------------------------
     EPVzA = new TH1D("EPVzA", "EPVzA", 80, -2, 2);
     fOutputList->Add(EPVzA);