--- /dev/null
+// $Id: AliAnalysisTaskEmcalJetSample.cxx $
+//
+// Jet sample analysis task.
+//
+// Author: S.Aiola
+
+#include <TClonesArray.h>
+#include <TH1F.h>
+#include <TH2F.h>
+#include <TList.h>
+#include <TLorentzVector.h>
+
+#include "AliVCluster.h"
+#include "AliVTrack.h"
+#include "AliEmcalJet.h"
+#include "AliRhoParameter.h"
+#include "AliLog.h"
+
+#include "AliAnalysisTaskEmcalJetSample.h"
+
+ClassImp(AliAnalysisTaskEmcalJetSample)
+
+//________________________________________________________________________
+AliAnalysisTaskEmcalJetSample::AliAnalysisTaskEmcalJetSample() :
+ AliAnalysisTaskEmcalJet("AliAnalysisTaskEmcalJetSample", kTRUE)
+
+{
+ // Default constructor.
+
+ for (Int_t i = 0; i < 4; i++) {
+ fHistTracksPt[i] = 0;
+ fHistClustersPt[i] = 0;
+ fHistLeadingJetPt[i] = 0;
+ fHistJetsPhiEta[i] = 0;
+ fHistJetsPtArea[i] = 0;
+ fHistJetsPtLeadHad[i] = 0;
+ fHistJetsCorrPtArea[i] = 0;
+ }
+
+ SetMakeGeneralHistograms(kTRUE);
+}
+
+//________________________________________________________________________
+AliAnalysisTaskEmcalJetSample::AliAnalysisTaskEmcalJetSample(const char *name) :
+ AliAnalysisTaskEmcalJet(name, kTRUE)
+{
+ // Standard constructor.
+
+ for (Int_t i = 0; i < 4; i++) {
+ fHistTracksPt[i] = 0;
+ fHistClustersPt[i] = 0;
+ fHistLeadingJetPt[i] = 0;
+ fHistJetsPhiEta[i] = 0;
+ fHistJetsPtArea[i] = 0;
+ fHistJetsPtLeadHad[i] = 0;
+ fHistJetsCorrPtArea[i] = 0;
+ }
+
+ SetMakeGeneralHistograms(kTRUE);
+}
+
+//________________________________________________________________________
+AliAnalysisTaskEmcalJetSample::~AliAnalysisTaskEmcalJetSample()
+{
+ // Destructor.
+}
+
+//________________________________________________________________________
+void AliAnalysisTaskEmcalJetSample::UserCreateOutputObjects()
+{
+ // Create user output.
+
+ AliAnalysisTaskEmcalJet::UserCreateOutputObjects();
+
+ TString histname;
+
+ for (Int_t i = 0; i < 4; i++) {
+ if (!fTracksName.IsNull()) {
+ histname = "fHistTracksPt_";
+ histname += i;
+ fHistTracksPt[i] = new TH1F(histname.Data(), histname.Data(), fNbins / 2, fMinBinPt, fMaxBinPt / 2);
+ fHistTracksPt[i]->GetXaxis()->SetTitle("p_{T,track} (GeV/c)");
+ fHistTracksPt[i]->GetYaxis()->SetTitle("counts");
+ fOutput->Add(fHistTracksPt[i]);
+ }
+
+ if (!fCaloName.IsNull()) {
+ histname = "fHistClustersPt_";
+ histname += i;
+ fHistClustersPt[i] = new TH1F(histname.Data(), histname.Data(), fNbins / 2, fMinBinPt, fMaxBinPt / 2);
+ fHistClustersPt[i]->GetXaxis()->SetTitle("p_{T,clus} (GeV/c)");
+ fHistClustersPt[i]->GetYaxis()->SetTitle("counts");
+ fOutput->Add(fHistClustersPt[i]);
+ }
+
+ if (!fJetsName.IsNull()) {
+ histname = "fHistLeadingJetPt_";
+ histname += i;
+ fHistLeadingJetPt[i] = new TH1F(histname.Data(), histname.Data(), fNbins, fMinBinPt, fMaxBinPt);
+ fHistLeadingJetPt[i]->GetXaxis()->SetTitle("p_{T}^{raw} (GeV/c)");
+ fHistLeadingJetPt[i]->GetYaxis()->SetTitle("counts");
+ fOutput->Add(fHistLeadingJetPt[i]);
+
+ histname = "fHistJetsPhiEta_";
+ histname += i;
+ fHistJetsPhiEta[i] = new TH2F(histname.Data(), histname.Data(), 50, -1, 1, 101, 0, TMath::Pi()*2 + TMath::Pi()/200);
+ fHistJetsPhiEta[i]->GetXaxis()->SetTitle("#eta");
+ fHistJetsPhiEta[i]->GetYaxis()->SetTitle("#phi");
+ fOutput->Add(fHistJetsPhiEta[i]);
+
+ histname = "fHistJetsPtArea_";
+ histname += i;
+ fHistJetsPtArea[i] = new TH2F(histname.Data(), histname.Data(), fNbins, fMinBinPt, fMaxBinPt, 30, 0, 3);
+ fHistJetsPtArea[i]->GetXaxis()->SetTitle("p_{T}^{raw} (GeV/c)");
+ fHistJetsPtArea[i]->GetYaxis()->SetTitle("area");
+ fOutput->Add(fHistJetsPtArea[i]);
+
+ histname = "fHistJetsPtLeadHad_";
+ histname += i;
+ fHistJetsPtLeadHad[i] = new TH2F(histname.Data(), histname.Data(), fNbins, fMinBinPt, fMaxBinPt, fNbins / 2, fMinBinPt, fMaxBinPt / 2);
+ fHistJetsPtLeadHad[i]->GetXaxis()->SetTitle("p_{T}^{raw} (GeV/c)");
+ fHistJetsPtLeadHad[i]->GetYaxis()->SetTitle("p_{T,lead} (GeV/c)");
+ fHistJetsPtLeadHad[i]->GetZaxis()->SetTitle("counts");
+ fOutput->Add(fHistJetsPtLeadHad[i]);
+
+ if (!fRhoName.IsNull()) {
+ histname = "fHistJetsCorrPtArea_";
+ histname += i;
+ fHistJetsCorrPtArea[i] = new TH2F(histname.Data(), histname.Data(), fNbins*2, -fMaxBinPt, fMaxBinPt, 30, 0, 3);
+ fHistJetsCorrPtArea[i]->GetXaxis()->SetTitle("p_{T}^{corr} [GeV/c]");
+ fHistJetsCorrPtArea[i]->GetYaxis()->SetTitle("area");
+ fOutput->Add(fHistJetsCorrPtArea[i]);
+ }
+ }
+ }
+ PostData(1, fOutput); // Post data for ALL output slots > 0 here.
+}
+
+//________________________________________________________________________
+Bool_t AliAnalysisTaskEmcalJetSample::FillHistograms()
+{
+ // Fill histograms.
+
+ if (fTracks) {
+ const Int_t ntracks = fTracks->GetEntriesFast();
+
+ for (Int_t it = 0; it < ntracks; it++) {
+ AliVTrack *track = static_cast<AliVTrack*>(fTracks->At(it));
+
+ if (!track) {
+ AliError(Form("Could not receive track %d", it));
+ continue;
+ }
+
+ if (!AcceptTrack(track))
+ continue;
+
+ fHistTracksPt[fCentBin]->Fill(track->Pt());
+ }
+ }
+
+ if (fCaloClusters) {
+ const Int_t nclusters = fCaloClusters->GetEntriesFast();
+
+ for (Int_t ic = 0; ic < nclusters; ic++) {
+ AliVCluster *cluster = static_cast<AliVCluster*>(fCaloClusters->At(ic));
+
+ if (!cluster) {
+ AliError(Form("Could not receive cluster %d", ic));
+ continue;
+ }
+
+ TLorentzVector nPart;
+ cluster->GetMomentum(nPart, fVertex);
+ fHistClustersPt[fCentBin]->Fill(nPart.Pt());
+ }
+ }
+
+ if (fJets) {
+ Int_t *sortedJets = GetSortedArray(fJets);
+
+ if (sortedJets) {
+ AliEmcalJet* leadJet = static_cast<AliEmcalJet*>(fJets->At(sortedJets[0]));
+ if (leadJet)
+ fHistLeadingJetPt[fCentBin]->Fill(leadJet->Pt());
+ else
+ AliError("Could not retrieve leading jet!");
+ }
+
+ const Int_t njets = fJets->GetEntriesFast();
+ for (Int_t ij = 0; ij < njets; ij++) {
+
+ AliEmcalJet* jet = static_cast<AliEmcalJet*>(fJets->At(ij));
+ if (!jet) {
+ AliError(Form("Could not receive jet %d", ij));
+ continue;
+ }
+
+ if (!AcceptJet(jet))
+ continue;
+
+ fHistJetsPtArea[fCentBin]->Fill(jet->Pt(), jet->Area());
+ fHistJetsPhiEta[fCentBin]->Fill(jet->Eta(), jet->Phi());
+
+ Float_t ptLeading = GetLeadingHadronPt(jet);
+ fHistJetsPtLeadHad[fCentBin]->Fill(jet->Pt(), ptLeading);
+
+ if (fRho) {
+ Float_t corrPt = jet->Pt() - fRhoVal * jet->Area();
+ fHistJetsCorrPtArea[fCentBin]->Fill(corrPt, jet->Area());
+ }
+ }
+ }
+
+ return kTRUE;
+}
+
+//________________________________________________________________________
+Bool_t AliAnalysisTaskEmcalJetSample::Run()
+{
+ // Run analysis code here, if needed. It will be executed before FillHistograms().
+
+ return kTRUE; // If return kFALSE FillHistogram() will NOT be executed.
+}
+
+//________________________________________________________________________
+void AliAnalysisTaskEmcalJetSample::Terminate(Option_t *)
+{
+ // Called once at the end of the analysis.
+}
--- /dev/null
+// $Id: AddTaskEmcalJetSample.C $
+
+AliAnalysisTaskEmcalJetSample* AddTaskEmcalJetSample(
+ const char *ntracks = "Tracks",
+ const char *nclusters = "CaloClusters",
+ const char *njets = "Jets",
+ const char *nrho = "Rho",
+ Double_t jetradius = 0.2,
+ Double_t jetptcut = 1,
+ Double_t jetareacut = 0.557,
+ UInt_t type = AliAnalysisTaskEmcal::kEMCAL,
+ Int_t leadhadtype = 0,
+ const char *taskname = "AliAnalysisTaskEmcalJetSample"
+)
+{
+ // Get the pointer to the existing analysis manager via the static access method.
+ //==============================================================================
+ AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+ if (!mgr)
+ {
+ ::Error("AddTaskEmcalJetSample", "No analysis manager to connect to.");
+ return NULL;
+ }
+
+ // Check the analysis type using the event handlers connected to the analysis manager.
+ //==============================================================================
+ if (!mgr->GetInputEventHandler())
+ {
+ ::Error("AddTaskEmcalJetSample", "This task requires an input event handler");
+ return NULL;
+ }
+
+ //-------------------------------------------------------
+ // Init the task and do settings
+ //-------------------------------------------------------
+
+ TString name(taskname);
+ if (strcmp(njets,"")) {
+ name += "_";
+ name += njets;
+ }
+ if (strcmp(nrho,"")) {
+ name += "_";
+ name += nrho;
+ }
+ if (type == AliAnalysisTaskEmcal::kTPC)
+ name += "_TPC";
+ else if (type == AliAnalysisTaskEmcal::kEMCAL)
+ name += "_EMCAL";
+ else if (type == AliAnalysisTaskEmcal::kUser)
+ name += "_USER";
+
+ AliAnalysisTaskEmcalJetSample* jetTask = new AliAnalysisTaskEmcalJetSample(name);
+ jetTask->SetAnaType(type);
+ jetTask->SetTracksName(ntracks);
+ jetTask->SetClusName(nclusters);
+ jetTask->SetJetsName(njets);
+ jetTask->SetRhoName(nrho);
+ jetTask->SetJetRadius(jetradius);
+ jetTask->SetJetPtCut(jetptcut);
+ jetTask->SetPercAreaCut(jetareacut);
+ jetTask->SetLeadingHadronType(leadhadtype);
+
+ //-------------------------------------------------------
+ // Final settings, pass to manager and set the containers
+ //-------------------------------------------------------
+
+ mgr->AddTask(jetTask);
+
+ // Create containers for input/output
+ AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer() ;
+ TString contname(name);
+ contname += "_histos";
+ AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contname.Data(),
+ TList::Class(),AliAnalysisManager::kOutputContainer,
+ Form("%s", AliAnalysisManager::GetCommonFileName()));
+ mgr->ConnectInput (jetTask, 0, cinput1 );
+ mgr->ConnectOutput (jetTask, 1, coutput1 );
+
+ return jetTask;
+}
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff