[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 "AliMultiEventInputHandler.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));
    fMixedEvent.Reset();
    
    for (Int_t iev = 0; iev < nev; iev++) {
	AliAODEvent* aod = (AliAODEvent*) GetEvent(iev);
	fMixedEvent.AddEvent(aod);
    }
    fMixedEvent.Init();
    Int_t ntrack = fMixedEvent.GetNumberOfTracks();
    if (ntrack > 1) {
      for (Int_t itr = 0; itr < ntrack -1; itr++) {
	for (Int_t jtr = itr+1; jtr < ntrack; jtr++) {
	  AliVParticle* track1 = fMixedEvent.GetTrack(itr);
	  AliVParticle* track2 = fMixedEvent.GetTrack(jtr);
	  Int_t iev1 =  fMixedEvent.EventIndex(itr);
	  Int_t iev2 =  fMixedEvent.EventIndex(jtr);

	  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;
	  if (iev1 != iev2) {
	    fHistDphiUC->Fill(dphi, wgt);
	  } else {
	    fHistDphiCO->Fill(dphi, 0.5);	    
	  }
	} // tarcks
      } // tracks
    } // more than 1 
  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();
  fHistDphiCO->DrawCopy("E");
  fHistDphiUC->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"
cd17c2ba	8	#include "AliMultiEventInputHandler.h"
1d24927a	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));
976f2793	51	fMixedEvent.Reset();
1d24927a	52
1d24927a	53	for (Int_t iev = 0; iev < nev; iev++) {
976f2793	54	AliAODEvent* aod = (AliAODEvent*) GetEvent(iev);
976f2793	55	fMixedEvent.AddEvent(aod);
1d24927a	56	}
976f2793	57	fMixedEvent.Init();
	58	Int_t ntrack = fMixedEvent.GetNumberOfTracks();
	59	if (ntrack > 1) {
	60	for (Int_t itr = 0; itr < ntrack -1; itr++) {
	61	for (Int_t jtr = itr+1; jtr < ntrack; jtr++) {
	62	AliVParticle* track1 = fMixedEvent.GetTrack(itr);
	63	AliVParticle* track2 = fMixedEvent.GetTrack(jtr);
	64	Int_t iev1 = fMixedEvent.EventIndex(itr);
	65	Int_t iev2 = fMixedEvent.EventIndex(jtr);
cd17c2ba	66
976f2793	67	Float_t phi1 = track1->Phi();
	68	Float_t phi2 = track2->Phi();
	69	Float_t dphi = TMath::Abs(phi1 - phi2);
	70	if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
	71	if (iev1 != iev2) {
	72	fHistDphiUC->Fill(dphi, wgt);
	73	} else {
	74	fHistDphiCO->Fill(dphi, 0.5);
	75	}
	76	} // tarcks
	77	} // tracks
	78	} // more than 1
1d24927a	79	PostData(1, fHists);
	80	}
	81
	82	//________________________________________________________________________
	83	void AliAnalysisTaskPhiCorr::Terminate(Option_t *)
	84	{
	85	// Draw result to the screen
	86	// Called once at the end of the query
	87
976f2793	88	TCanvas *c1 = new TCanvas("AliAnalysisTaskPt","Pt",10,10,510,510);
1d24927a	89	c1->cd(1)->SetLogy();
976f2793	90	fHistDphiCO->DrawCopy("E");
976f2793	91	fHistDphiUC->DrawCopy("Esame");
1d24927a	92	}