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


#include "TH1F.h"
#include "TList.h"
#include "TCanvas.h"
#include "AliAODEvent.h"
#include "AliAODTrack.h"
#include "AliAnalysisTaskPhiCorr.h"
#include "AliMultiAODInputHandler.h"

ClassImp(AliAnalysisTaskPhiCorr)

//________________________________________________________________________
AliAnalysisTaskPhiCorr::AliAnalysisTaskPhiCorr(const char *name) 
    : AliAnalysisTaskME(name), fHists(0), fHistDphiCO(0),  fHistDphiUC(0)
{
  // Constructor
  // Define input and output slots here
  DefineOutput(1, TList::Class());
}


//________________________________________________________________________
void AliAnalysisTaskPhiCorr::UserCreateOutputObjects()
{
  // Create histograms
  // Called once

  fHistDphiCO = new TH1F("fHistDPhiCO", "#Delta #Phi distribution", 64, 0., 3.2);
  fHistDphiCO->GetXaxis()->SetTitle("#Delta#Phi [rad]");
  fHistDphiCO->GetYaxis()->SetTitle("dN/d#Phi");
  fHistDphiCO->SetMarkerStyle(kFullCircle);

  fHistDphiUC = new TH1F("fHistDPhiUC", "#Delta #Phi distribution", 64, 0., 3.2);
  fHistDphiUC->GetXaxis()->SetTitle("#Delta#Phi [rad]");
  fHistDphiUC->GetYaxis()->SetTitle("dN/d#Phi");
  fHistDphiUC->SetMarkerStyle(kOpenCircle);
  
  fHists = new TList();
  fHists->Add(fHistDphiCO);
  fHists->Add(fHistDphiUC);  
}

//________________________________________________________________________
void AliAnalysisTaskPhiCorr::UserExec(Option_t *) 
{
  // Main loop
  // Called for each event
    // Uncorrelated tracks
    Int_t nev = fInputHandler->GetBufferSize();
    Float_t wgt = 1./(nev*(nev-1));
    
    for (Int_t iev = 0; iev < nev; iev++) {
	for (Int_t jev = (iev + 1); jev < nev; jev++) {
	    AliAODEvent* aod1 = GetEvent(iev);
	    AliAODEvent* aod2 = GetEvent(jev);
	    
	    Int_t ntracks1 =  aod1->GetNumberOfTracks();
	    Int_t ntracks2 =  aod2->GetNumberOfTracks();
	    
	    printf("Number of tracks %5d:%5d %5d:%5d\n", iev, ntracks1, jev, ntracks2);
	    
	    for (Int_t iTracks = 0; iTracks < ntracks1; iTracks++) {
		for (Int_t jTracks = 0; jTracks < ntracks2; jTracks++) {
		    
		    
		    AliAODTrack* track1 = aod1->GetTrack(iTracks);
		    AliAODTrack* track2 = aod2->GetTrack(jTracks);
		    
		    Float_t phi1 = track1->Phi();
		    Float_t phi2 = track2->Phi();
		    Float_t dphi = TMath::Abs(phi1 - phi2);
		    if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
		    fHistDphiUC->Fill(dphi, wgt);
		} // tracks
	    } // tracks 
	} // event loop
    } // event loop
    
//    Correlated 
    AliAODEvent* aod = fInputHandler->GetLatestEvent();
    
    Int_t ntracks = aod->GetNumberOfTracks();
    for (Int_t iTracks = 0; iTracks < ntracks; iTracks++) {
	for (Int_t jTracks = (iTracks+1); jTracks < ntracks; jTracks++) {
	      AliAODTrack* track1 = aod->GetTrack(iTracks);
	      AliAODTrack* track2 = aod->GetTrack(jTracks);
	      Float_t phi1 = track1->Phi();
	      Float_t phi2 = track2->Phi();
	      Float_t dphi = TMath::Abs(phi1 - phi2);
	      if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
	      fHistDphiCO->Fill(dphi);
	} // tracks
    }
    
  // Post output data.
  PostData(1, fHists);
}      

//________________________________________________________________________
void AliAnalysisTaskPhiCorr::Terminate(Option_t *) 
{
  // Draw result to the screen
  // Called once at the end of the query

  TCanvas *c1 = new TCanvas("AliAnalysisTaskPt","Pt",10,10,510,510);
  c1->cd(1)->SetLogy();
  fHistDphiUC->DrawCopy("E");
  fHistDphiCO->DrawCopy("Esame");
}
Commit	Line	Data
1d24927a	1
	2	#include "TH1F.h"
	3	#include "TList.h"
	4	#include "TCanvas.h"
	5	#include "AliAODEvent.h"
	6	#include "AliAODTrack.h"
	7	#include "AliAnalysisTaskPhiCorr.h"
	8	#include "AliMultiAODInputHandler.h"
	9
	10	ClassImp(AliAnalysisTaskPhiCorr)
	11
	12	//________________________________________________________________________
	13	AliAnalysisTaskPhiCorr::AliAnalysisTaskPhiCorr(const char *name)
	14	: AliAnalysisTaskME(name), fHists(0), fHistDphiCO(0), fHistDphiUC(0)
	15	{
	16	// Constructor
	17	// Define input and output slots here
	18	DefineOutput(1, TList::Class());
	19	}
	20
	21
	22	//________________________________________________________________________
	23	void AliAnalysisTaskPhiCorr::UserCreateOutputObjects()
	24	{
	25	// Create histograms
	26	// Called once
	27
	28	fHistDphiCO = new TH1F("fHistDPhiCO", "#Delta #Phi distribution", 64, 0., 3.2);
	29	fHistDphiCO->GetXaxis()->SetTitle("#Delta#Phi [rad]");
	30	fHistDphiCO->GetYaxis()->SetTitle("dN/d#Phi");
	31	fHistDphiCO->SetMarkerStyle(kFullCircle);
	32
	33	fHistDphiUC = new TH1F("fHistDPhiUC", "#Delta #Phi distribution", 64, 0., 3.2);
	34	fHistDphiUC->GetXaxis()->SetTitle("#Delta#Phi [rad]");
	35	fHistDphiUC->GetYaxis()->SetTitle("dN/d#Phi");
	36	fHistDphiUC->SetMarkerStyle(kOpenCircle);
	37
	38	fHists = new TList();
	39	fHists->Add(fHistDphiCO);
	40	fHists->Add(fHistDphiUC);
	41	}
	42
	43	//________________________________________________________________________
	44	void AliAnalysisTaskPhiCorr::UserExec(Option_t *)
	45	{
	46	// Main loop
	47	// Called for each event
	48	// Uncorrelated tracks
	49	Int_t nev = fInputHandler->GetBufferSize();
	50	Float_t wgt = 1./(nev*(nev-1));
	51
	52	for (Int_t iev = 0; iev < nev; iev++) {
	53	for (Int_t jev = (iev + 1); jev < nev; jev++) {
	54	AliAODEvent* aod1 = GetEvent(iev);
	55	AliAODEvent* aod2 = GetEvent(jev);
	56
	57	Int_t ntracks1 = aod1->GetNumberOfTracks();
	58	Int_t ntracks2 = aod2->GetNumberOfTracks();
	59
	60	printf("Number of tracks %5d:%5d %5d:%5d\n", iev, ntracks1, jev, ntracks2);
	61
	62	for (Int_t iTracks = 0; iTracks < ntracks1; iTracks++) {
	63	for (Int_t jTracks = 0; jTracks < ntracks2; jTracks++) {
	64
65
66	AliAODTrack* track1 = aod1->GetTrack(iTracks);
67	AliAODTrack* track2 = aod2->GetTrack(jTracks);
68
69	Float_t phi1 = track1->Phi();
70	Float_t phi2 = track2->Phi();
71	Float_t dphi = TMath::Abs(phi1 - phi2);
72	if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
73	fHistDphiUC->Fill(dphi, wgt);
74	} // tracks
75	} // tracks
76	} // event loop
77	} // event loop
78
79	// Correlated
cb6876bb	80	AliAODEvent* aod = fInputHandler->GetLatestEvent();
cb6876bb	81
1d24927a	82	Int_t ntracks = aod->GetNumberOfTracks();
	83	for (Int_t iTracks = 0; iTracks < ntracks; iTracks++) {
	84	for (Int_t jTracks = (iTracks+1); jTracks < ntracks; jTracks++) {
	85	AliAODTrack* track1 = aod->GetTrack(iTracks);
	86	AliAODTrack* track2 = aod->GetTrack(jTracks);
	87	Float_t phi1 = track1->Phi();
	88	Float_t phi2 = track2->Phi();
	89	Float_t dphi = TMath::Abs(phi1 - phi2);
	90	if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
	91	fHistDphiCO->Fill(dphi);
	92	} // tracks
	93	}
	94
	95	// Post output data.
	96	PostData(1, fHists);
	97	}
	98
	99	//________________________________________________________________________
	100	void AliAnalysisTaskPhiCorr::Terminate(Option_t *)
	101	{
	102	// Draw result to the screen
	103	// Called once at the end of the query
	104
	105	TCanvas *c1 = new TCanvas("AliAnalysisTaskPt","Pt",10,10,510,510);
	106	c1->cd(1)->SetLogy();
	107	fHistDphiUC->DrawCopy("E");
	108	fHistDphiCO->DrawCopy("Esame");
	109	}