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

/*
 Simple use case for mixed event analysis
 based on ESD or AOD
 Delta_phi correlation analysis is performed on charged tracks 
 for same and mixed events
 Author: andreas.morsch@cern.ch 
*/


#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
92502f31	1	/*
	2	Simple use case for mixed event analysis
	3	based on ESD or AOD
	4	Delta_phi correlation analysis is performed on charged tracks
	5	for same and mixed events
	6	Author: andreas.morsch@cern.ch
	7	*/
	8
1d24927a	9
	10	#include "TH1F.h"
	11	#include "TList.h"
	12	#include "TCanvas.h"
	13	#include "AliAODEvent.h"
	14	#include "AliAODTrack.h"
	15	#include "AliAnalysisTaskPhiCorr.h"
cd17c2ba	16	#include "AliMultiEventInputHandler.h"
1d24927a	17
	18	ClassImp(AliAnalysisTaskPhiCorr)
	19
	20	//________________________________________________________________________
	21	AliAnalysisTaskPhiCorr::AliAnalysisTaskPhiCorr(const char *name)
	22	: AliAnalysisTaskME(name), fHists(0), fHistDphiCO(0), fHistDphiUC(0)
	23	{
	24	// Constructor
	25	// Define input and output slots here
	26	DefineOutput(1, TList::Class());
	27	}
	28
	29
	30	//________________________________________________________________________
	31	void AliAnalysisTaskPhiCorr::UserCreateOutputObjects()
	32	{
	33	// Create histograms
	34	// Called once
	35
	36	fHistDphiCO = new TH1F("fHistDPhiCO", "#Delta #Phi distribution", 64, 0., 3.2);
	37	fHistDphiCO->GetXaxis()->SetTitle("#Delta#Phi [rad]");
	38	fHistDphiCO->GetYaxis()->SetTitle("dN/d#Phi");
	39	fHistDphiCO->SetMarkerStyle(kFullCircle);
	40
	41	fHistDphiUC = new TH1F("fHistDPhiUC", "#Delta #Phi distribution", 64, 0., 3.2);
	42	fHistDphiUC->GetXaxis()->SetTitle("#Delta#Phi [rad]");
	43	fHistDphiUC->GetYaxis()->SetTitle("dN/d#Phi");
	44	fHistDphiUC->SetMarkerStyle(kOpenCircle);
	45
	46	fHists = new TList();
	47	fHists->Add(fHistDphiCO);
	48	fHists->Add(fHistDphiUC);
	49	}
	50
	51	//________________________________________________________________________
	52	void AliAnalysisTaskPhiCorr::UserExec(Option_t *)
	53	{
	54	// Main loop
	55	// Called for each event
	56	// Uncorrelated tracks
	57	Int_t nev = fInputHandler->GetBufferSize();
	58	Float_t wgt = 1./(nev*(nev-1));
976f2793	59	fMixedEvent.Reset();
a9764b55	60
a9764b55	61
1d24927a	62	for (Int_t iev = 0; iev < nev; iev++) {
976f2793	63	AliAODEvent* aod = (AliAODEvent*) GetEvent(iev);
976f2793	64	fMixedEvent.AddEvent(aod);
1d24927a	65	}
976f2793	66	fMixedEvent.Init();
	67	Int_t ntrack = fMixedEvent.GetNumberOfTracks();
	68	if (ntrack > 1) {
	69	for (Int_t itr = 0; itr < ntrack -1; itr++) {
	70	for (Int_t jtr = itr+1; jtr < ntrack; jtr++) {
	71	AliVParticle* track1 = fMixedEvent.GetTrack(itr);
	72	AliVParticle* track2 = fMixedEvent.GetTrack(jtr);
	73	Int_t iev1 = fMixedEvent.EventIndex(itr);
	74	Int_t iev2 = fMixedEvent.EventIndex(jtr);
cd17c2ba	75
976f2793	76	Float_t phi1 = track1->Phi();
	77	Float_t phi2 = track2->Phi();
	78	Float_t dphi = TMath::Abs(phi1 - phi2);
	79	if (dphi > TMath::Pi()) dphi = 2. * TMath::Pi() - dphi;
	80	if (iev1 != iev2) {
	81	fHistDphiUC->Fill(dphi, wgt);
	82	} else {
	83	fHistDphiCO->Fill(dphi, 0.5);
	84	}
	85	} // tarcks
	86	} // tracks
	87	} // more than 1
a9764b55	88
1d24927a	89	PostData(1, fHists);
	90	}
	91
	92	//________________________________________________________________________
	93	void AliAnalysisTaskPhiCorr::Terminate(Option_t *)
	94	{
	95	// Draw result to the screen
	96	// Called once at the end of the query
	97
976f2793	98	TCanvas *c1 = new TCanvas("AliAnalysisTaskPt","Pt",10,10,510,510);
1d24927a	99	c1->cd(1)->SetLogy();
976f2793	100	fHistDphiCO->DrawCopy("E");
976f2793	101	fHistDphiUC->DrawCopy("Esame");
1d24927a	102	}