Warning corrected
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskDiJets.cxx
CommitLineData
bd24e6dc 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16/* $Id$ */
17
18#include "AliAnalysisTaskDiJets.h"
19#include "AliAODEvent.h"
20#include "AliAODJet.h"
21#include "AliAODDiJet.h"
22#include <TClonesArray.h>
23#include <TLorentzVector.h>
4f49cbd0 24#include <TStyle.h>
bd24e6dc 25
26
27ClassImp(AliAnalysisTaskDiJets)
28
29////////////////////////////////////////////////////////////////////////
30
31AliAnalysisTaskDiJets::AliAnalysisTaskDiJets():
32 AliAnalysisTaskSE(),
33 fDiJets(0),
4f49cbd0 34 fDiJetsIn(0),
7bb870ca 35 fUseAODInput(kFALSE),
863dbf3b 36 fFillAOD(kFALSE),
7bb870ca 37 fJetBranch("jets"),
38 fAOD(0),
4f49cbd0 39 fHistList(0),
40 fH1DeltaPt(0),
41 fH1DeltaPhi(0),
42 fH1PhiImbal(0),
43 fH1Asym(0),
44 fH2Pt2vsPt1(0),
45 fH2DifvsSum(0)
bd24e6dc 46{
47 // Default constructor
48}
49
4f49cbd0 50//----------------------------------------------------------------------
bd24e6dc 51AliAnalysisTaskDiJets::AliAnalysisTaskDiJets(const char* name):
52 AliAnalysisTaskSE(name),
53 fDiJets(0),
4f49cbd0 54 fDiJetsIn(0),
7bb870ca 55 fUseAODInput(kFALSE),
863dbf3b 56 fFillAOD(kFALSE),
7bb870ca 57 fJetBranch("jets"),
58 fAOD(0),
4f49cbd0 59 fHistList(0),
60 fH1DeltaPt(0),
61 fH1DeltaPhi(0),
62 fH1PhiImbal(0),
63 fH1Asym(0),
64 fH2Pt2vsPt1(0),
65 fH2DifvsSum(0)
bd24e6dc 66{
67 // Default constructor
4f49cbd0 68 DefineOutput(1, TList::Class());
bd24e6dc 69}
70
4f49cbd0 71//----------------------------------------------------------------------
bd24e6dc 72void AliAnalysisTaskDiJets::UserCreateOutputObjects()
73{
74// Create the output container
75//
7bb870ca 76 if (fDebug) printf("AnalysisTaskDiJets::CreateOutPutData() \n");
bd24e6dc 77 fDiJets = new TClonesArray("AliAODDiJet", 0);
863dbf3b 78 if (fFillAOD){
7bb870ca 79 fDiJets->SetName(Form("dijets_%s",fJetBranch.Data()));
863dbf3b 80 AddAODBranch("TClonesArray", &fDiJets);
81 }
4f49cbd0 82
83 if (!fHistList) fHistList = new TList();
84 Float_t pi=TMath::Pi();
85 gStyle->SetPalette(1);
86
87 fH1DeltaPt = new TH1F("DeltaPt","Difference between the jets' Pt;#Deltap_{T} (GeV/c);Entries",150,0.,150.);
88 fH1DeltaPt->SetMarkerSize(0.6);
89 fH1DeltaPt->SetMarkerColor(4);
90 fH1DeltaPt->SetMarkerStyle(21);
91 fH1DeltaPt->SetOption("E");
92
93 fH1DeltaPhi = new TH1F("DeltaPhi","Difference in the azimuthal angle;#Delta#phi;Entries",100,0.,pi);
94 fH1DeltaPhi->SetMarkerSize(0.6);
95 fH1DeltaPhi->SetMarkerColor(4);
96 fH1DeltaPhi->SetMarkerStyle(21);
97 fH1DeltaPhi->SetOption("E");
98
99 fH1PhiImbal = new TH1F("PhiImb","Phi imbalance;#phi;Entries",100,-pi,pi);
100 fH1PhiImbal->SetMarkerSize(0.6);
101 fH1PhiImbal->SetMarkerColor(4);
102 fH1PhiImbal->SetMarkerStyle(21);
103 fH1PhiImbal->SetOption("E");
104
105 fH1Asym = new TH1F("Asym","Pt asymmetry;#Deltap_{T}/(p_{T,1}+p_{T,2});Entries",50,0.,1.);
106 fH1Asym->SetMarkerSize(0.6);
107 fH1Asym->SetMarkerColor(4);
108 fH1Asym->SetMarkerStyle(21);
109 fH1Asym->SetOption("E");
110
111 fH2Pt2vsPt1 = new TH2F("Pt2vsPt1","Pt2 versus Pt1;p_{T,1} (GeV/c);p_{T,2} (GeV/c)",250,0.,250.,250,0.,250.);
112 fH2Pt2vsPt1->SetOption("cont0");
113
114 fH2DifvsSum = new TH2F("DifvsSum","Pt difference vs Pt sum;p_{T,1}+p_{T,2} (GeV/c);#Deltap_{T} (GeV/c)",400,0.,400.,150,0.,150.);
115 fH2DifvsSum->SetOption("cont0");
116
117 fHistList->Add(fH1DeltaPt);
118 fHistList->Add(fH1DeltaPhi);
119 fHistList->Add(fH1PhiImbal);
120 fHistList->Add(fH1Asym);
121 fHistList->Add(fH2Pt2vsPt1);
122 fHistList->Add(fH2DifvsSum);
bd24e6dc 123}
124
4f49cbd0 125//----------------------------------------------------------------------
bd24e6dc 126void AliAnalysisTaskDiJets::Init()
127{
128 // Initialization
7bb870ca 129 if (fDebug) printf("AnalysisTaskDiJets::Init() \n");
bd24e6dc 130}
131
4f49cbd0 132//----------------------------------------------------------------------
bd24e6dc 133void AliAnalysisTaskDiJets::UserExec(Option_t */*option*/)
134{
135// Execute analysis for current event
136//
7bb870ca 137 if (fDiJets) fDiJets->Delete();
138
139 if(fUseAODInput){
01c15c67 140 fAOD = dynamic_cast<AliAODEvent*> (InputEvent());
ec365e4a 141 if(!fAOD){
7bb870ca 142 // We do not have an input AOD, look in the output
ec365e4a 143 if (fDebug) printf("%s:%d No AOD event in the input\n",(char*)__FILE__,__LINE__);
7bb870ca 144 return;
145 }
146 } else {
147 fAOD = AODEvent();
ec365e4a 148 if(!fAOD){
149 if (fDebug) printf("%s:%d AODEvent not found in the Output",(char*)__FILE__,__LINE__);
4f49cbd0 150 return;
151 }
6c3180e0 152 }
ec365e4a 153
7bb870ca 154 TClonesArray* jets = (TClonesArray*) fAOD->FindListObject(fJetBranch.Data());
6c3180e0 155 // N.B. if we take the aod from the output this is always
156 // empty and since it is the same as fDiJets
7bb870ca 157 fDiJetsIn = (TClonesArray*) (fAOD->GetList()->FindObject("dijets"));
4f49cbd0 158
bd24e6dc 159 if (fDiJetsIn) {
7bb870ca 160 if (fDebug) printf("Found %d dijets in old list \n", fDiJetsIn->GetEntries());
4f49cbd0 161 AliAODJet* jj1, *jj2;
162 AliAODDiJet* testJ;
163
164 if (fDiJetsIn->GetEntries() > 0) {
165 testJ = (AliAODDiJet*) (fDiJetsIn->At(0));
166 jj1 = testJ->Jet(0);
167 jj1->Print("");
168 jj2 = testJ->Jet(1);
169 jj2->Print("");
170 }
bd24e6dc 171 }
4f49cbd0 172
bd24e6dc 173 Int_t nj = jets->GetEntriesFast();
7bb870ca 174 if (fDebug) printf("There are %5d jets in the event \n", nj);
4f49cbd0 175
b3818fc4 176 if (nj < 2){
177 PostData(1, fHistList);
178 return;
179 }
4f49cbd0 180 AliAODJet* jet1 = (AliAODJet*) (jets->At(0));
181 TLorentzVector v1 = *(jet1->MomentumVector());
182 AliAODJet* jet2 = (AliAODJet*) (jets->At(1));
183 TLorentzVector v2 = *(jet2->MomentumVector());
184 TLorentzVector v = v1 + v2;
185 Int_t ndi = fDiJets->GetEntriesFast();
186 TClonesArray &lref = *fDiJets;
187 new(lref[ndi]) AliAODDiJet(v);
188 AliAODDiJet* dijet = (AliAODDiJet*) (fDiJets->At(ndi));
189 dijet->SetJetRefs(jet1, jet2);
190
191 fH1DeltaPt->Fill(jet1->Pt()-jet2->Pt());
192 fH1DeltaPhi->Fill(dijet->DeltaPhi());
193 fH1PhiImbal->Fill(dijet->PhiImbalance());
194 fH1Asym->Fill((jet1->Pt()-jet2->Pt())/(jet1->Pt()+jet2->Pt()));
195 fH2Pt2vsPt1->Fill(jet1->Pt(),jet2->Pt());
196 fH2DifvsSum->Fill(jet1->Pt()+jet2->Pt(),jet1->Pt()-jet2->Pt());
197
198 PostData(1, fHistList);
199 return;
bd24e6dc 200}
201
4f49cbd0 202//----------------------------------------------------------------------
bd24e6dc 203void AliAnalysisTaskDiJets::Terminate(Option_t */*option*/)
204{
205// Terminate analysis
206//
7bb870ca 207 if (fDebug) printf("AnalysisDiJets: Terminate() \n");
bd24e6dc 208}
209