AliPhysics  abafffd (abafffd)
AliAnalysisTaskEmcalJetEnergySpectrum.cxx
Go to the documentation of this file.
1 /************************************************************************************
2  * Copyright (C) 2017, Copyright Holders of the ALICE Collaboration *
3  * All rights reserved. *
4  * *
5  * Redistribution and use in source and binary forms, with or without *
6  * modification, are permitted provided that the following conditions are met: *
7  * * Redistributions of source code must retain the above copyright *
8  * notice, this list of conditions and the following disclaimer. *
9  * * Redistributions in binary form must reproduce the above copyright *
10  * notice, this list of conditions and the following disclaimer in the *
11  * documentation and/or other materials provided with the distribution. *
12  * * Neither the name of the <organization> nor the *
13  * names of its contributors may be used to endorse or promote products *
14  * derived from this software without specific prior written permission. *
15  * *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND *
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED *
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE *
19  * DISCLAIMED. IN NO EVENT SHALL ALICE COLLABORATION BE LIABLE FOR ANY *
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES *
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; *
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND *
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT *
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS *
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. *
26  ************************************************************************************/
27 #include <algorithm>
28 #include <array>
29 #include <sstream>
30 #include <string>
31 #include <vector>
32 
33 #include <THistManager.h>
34 #include <TLinearBinning.h>
35 #include <TVariableBinning.h>
36 
37 #include "AliAODInputHandler.h"
38 #include "AliAnalysisManager.h"
42 #include "AliEmcalJet.h"
45 #include "AliEventCuts.h"
46 #include "AliInputEventHandler.h"
47 #include "AliJetContainer.h"
48 #include "AliLog.h"
49 #include "AliMultSelection.h"
50 #include "AliVEvent.h"
51 
53 
54 using namespace EmcalTriggerJets;
55 
58  fHistos(nullptr),
59  fEventCuts(nullptr),
60  fIsMC(false),
61  fTriggerSelectionBits(AliVEvent::kAny),
62  fTriggerSelectionString(""),
63  fRequireSubsetMB(false),
64  fMinBiasTrigger(AliVEvent::kAny),
65  fNameTriggerDecisionContainer("EmcalTriggerDecision"),
66  fUseTriggerSelectionForData(false),
67  fUseDownscaleWeight(false),
68  fNameJetContainer("datajets"),
69  fRequestTriggerClusters(true),
70  fRequestCentrality(false),
71  fUseAliEventCuts(false),
72  fCentralityEstimator("V0M")
73 {
74  SetUseAliAnaUtils(true);
75 }
76 
78  AliAnalysisTaskEmcalJet(name, true),
81  fIsMC(false),
82  fTriggerSelectionBits(AliVEvent::kAny),
84  fRequireSubsetMB(false),
85  fMinBiasTrigger(AliVEvent::kAny),
86  fNameTriggerDecisionContainer("EmcalTriggerDecision"),
88  fUseDownscaleWeight(false),
89  fNameJetContainer("datajets"),
91  fRequestCentrality(false),
92  fUseAliEventCuts(false),
94 {
95  SetUseAliAnaUtils(true);
96 }
97 
99  if(fHistos) delete fHistos;
100 }
101 
104 
105  if(!fUserPtBinning.GetSize()) {
106  // binning not set. apply default binning
107  AliInfoStream() << "Using default pt binning";
108  fUserPtBinning.Set(201);
109  double current(0.);
110  for(int istep = 0; istep < 201; istep++) {
111  fUserPtBinning[istep] = current;
112  current += 1;
113  }
114  }
115 
116  TLinearBinning centralitybinning(100, 0., 100.), etabinning(100, -1., 1.), phibinning(100., 0., 7.), nefbinning(100, 0., 1.), trgclusterbinning(kTrgClusterN + 1, -0.5, kTrgClusterN -0.5);
117  TVariableBinning jetptbinning(fUserPtBinning);
118  const TBinning *binnings[6] = {&centralitybinning, &jetptbinning, &etabinning, &phibinning, &nefbinning, &trgclusterbinning};
119  fHistos = new THistManager(Form("Histos_%s", GetName()));
120  fHistos->CreateTH1("hEventCounter", "Event counter histogram", 1, 0.5, 1.5);
121  fHistos->CreateTH1("hEventCounterAbs", "Event counter histogram absolute", 1, 0.5, 1.5);
122  fHistos->CreateTH1("hEventCentrality", "Event centrality", 100., 0., 100.);
123  fHistos->CreateTH1("hEventCentralityAbs", "Event centrality absolute", 100., 0., 100.);
124  fHistos->CreateTH1("hClusterCounter", "Event counter histogram", kTrgClusterN, -0.5, kTrgClusterN - 0.5);
125  fHistos->CreateTH1("hClusterCounterAbs", "Event counter histogram absolute", kTrgClusterN, -0.5, kTrgClusterN - 0.5);
126  fHistos->CreateTHnSparse("hJetTHnSparse", "jet thnsparse", 6, binnings, "s");
127  fHistos->CreateTHnSparse("hMaxJetTHnSparse", "jet thnsparse", 6, binnings, "s");
128 
129  for(auto h : *fHistos->GetListOfHistograms()) fOutput->Add(h);
130 
131  if(fUseAliEventCuts) {
132  fEventCuts = new AliEventCuts(true);
133  // Do not perform trigger selection in the AliEvent cuts but let the task do this before
134  fEventCuts->OverrideAutomaticTriggerSelection(AliVEvent::kAny, true);
135  fOutput->Add(fEventCuts);
136  }
137  PostData(1, fOutput);
138 }
139 
141  if(fEventCuts) {
142  if(!IsTriggerSelected()) return false;
143  if(!fEventCuts->AcceptEvent(fInputEvent)) return false;
144  return true;
145  }
147 }
148 
150  auto datajets = this->GetJetContainer(fNameJetContainer);
151  if(!datajets) {
152  AliErrorStream() << "Jet container " << fNameJetContainer << " not found" << std::endl;
153  return false;
154  }
155 
156  double eventCentrality = 99; // without centrality put everything in the peripheral bin
157  if(fRequestCentrality){
158  AliMultSelection *mult = dynamic_cast<AliMultSelection *>(InputEvent()->FindListObject("MultSelection"));
159  if(!mult){
160  AliErrorStream() << GetName() << ": Centrality selection enabled but no centrality estimator found" << std::endl;
161  return false;
162  }
163  if(mult->IsEventSelected()) return false;
164  eventCentrality = mult->GetEstimator(fCentralityEstimator)->GetPercentile();
165  AliDebugStream(1) << GetName() << ": Centrality " << eventCentrality << std::endl;
166  } else {
167  AliDebugStream(1) << GetName() << ": No centrality selection applied" << std::endl;
168  }
169 
170  auto trgclusters = GetTriggerClusterIndices(fInputEvent->GetFiredTriggerClasses());
171  Double_t weight = 1.;
172  if(fUseDownscaleWeight) {
173  weight = 1./PWG::EMCAL::AliEmcalDownscaleFactorsOCDB::Instance()->GetDownscaleFactorForTriggerClass(MatchTrigger(fInputEvent->GetFiredTriggerClasses().Data()));
174  }
175  fHistos->FillTH1("hEventCounterAbs", 1.);
176  fHistos->FillTH1("hEventCounter", weight);
177  fHistos->FillTH1("hEventCentralityAbs", eventCentrality);
178  fHistos->FillTH1("hEventCentrality", eventCentrality, weight);
179  AliEmcalJet *maxjet(nullptr);
180  for(auto t : trgclusters) {
181  fHistos->FillTH1("hClusterCounterAbs", t);
182  fHistos->FillTH1("hClusterCounter", t, weight);
183  }
184  for(auto j : datajets->accepted()){
185  if(!maxjet || (j->E() > maxjet->E())) maxjet = j;
186  double datapoint[6] = {eventCentrality, j->Pt(), j->Eta(), j->Phi(), j->NEF(), 0.};
187  for(auto t : trgclusters){
188  datapoint[5] = static_cast<double>(t);
189  fHistos->FillTHnSparse("hJetTHnSparse", datapoint, weight);
190  }
191  }
192 
193  double maxdata[6];
194  memset(maxdata, 0., sizeof(double) * 5);
195  maxdata[0] = eventCentrality;
196  if(maxjet){
197  maxdata[1] = maxjet->Pt();
198  maxdata[2] = maxjet->Eta();
199  maxdata[3] = maxjet->Phi();
200  maxdata[4] = maxjet->NEF();
201  }
202  for(auto t : trgclusters){
203  maxdata[5] = static_cast<double>(t);
204  fHistos->FillTHnSparse("hMaxJetTHnSparse", maxdata, weight);
205  }
206  return true;
207 }
208 
210  if(fUseDownscaleWeight) {
211  auto downscalehandler = PWG::EMCAL::AliEmcalDownscaleFactorsOCDB::Instance();
212  if(downscalehandler->GetCurrentRun() != newrun){
213  downscalehandler->SetRun(newrun);
214  }
215  }
216 }
217 
218 std::vector<AliAnalysisTaskEmcalJetEnergySpectrum::TriggerCluster_t> AliAnalysisTaskEmcalJetEnergySpectrum::GetTriggerClusterIndices(const TString &triggerstring) const {
219  // decode trigger string in order to determine the trigger clusters
220  std::vector<TriggerCluster_t> result;
221  result.emplace_back(kTrgClusterANY); // cluster ANY always included
223  // Data - separate trigger clusters
224  std::vector<std::string> clusternames;
225  auto triggerinfos = PWG::EMCAL::Triggerinfo::DecodeTriggerString(triggerstring.Data());
226  for(auto t : triggerinfos) {
227  if(std::find(clusternames.begin(), clusternames.end(), t.Triggercluster()) == clusternames.end()) clusternames.emplace_back(t.Triggercluster());
228  }
229  bool isCENT = (std::find(clusternames.begin(), clusternames.end(), "CENT") != clusternames.end()),
230  isCENTNOTRD = (std::find(clusternames.begin(), clusternames.end(), "CENTNOTRD") != clusternames.end()),
231  isCALO = (std::find(clusternames.begin(), clusternames.end(), "CALO") != clusternames.end()),
232  isCALOFAST = (std::find(clusternames.begin(), clusternames.end(), "CALOFAST") != clusternames.end());
233  if(isCENT || isCENTNOTRD) {
234  if(isCENT) {
235  result.emplace_back(kTrgClusterCENT);
236  if(isCENTNOTRD) {
237  result.emplace_back(kTrgClusterCENTNOTRD);
238  result.emplace_back(kTrgClusterCENTBOTH);
239  } else result.emplace_back(kTrgClusterOnlyCENT);
240  } else {
241  result.emplace_back(kTrgClusterCENTNOTRD);
242  result.emplace_back(kTrgClusterOnlyCENTNOTRD);
243  }
244  }
245  if(isCALO || isCALOFAST) {
246  if(isCALO) {
247  result.emplace_back(kTrgClusterCALO);
248  if(isCALOFAST) {
249  result.emplace_back(kTrgClusterCALOFAST);
250  result.emplace_back(kTrgClusterCALOBOTH);
251  } else result.emplace_back(kTrgClusterOnlyCALO);
252  } else {
253  result.emplace_back(kTrgClusterCALOFAST);
254  result.emplace_back(kTrgClusterOnlyCALOFAST);
255  }
256  }
257  }
258  return result;
259 }
260 
262  if(!fIsMC){
263  // Pure data - do EMCAL trigger selection from selection string
264  if(!(fInputHandler->IsEventSelected() & fTriggerSelectionBits)) return false;
265  if(fTriggerSelectionString.Length()) {
266  if(!fInputEvent->GetFiredTriggerClasses().Contains(fTriggerSelectionString)) return false;
267  if(fRequireSubsetMB && !(fInputHandler->IsEventSelected() & fMinBiasTrigger)) return false; // Require EMCAL trigger to be subset of the min. bias trigger (for efficiency studies)
268  if((fTriggerSelectionString.Contains("EJ") || fTriggerSelectionString.Contains("EG") || fTriggerSelectionString.Contains("DJ") || fTriggerSelectionString.Contains("DG")) && fUseTriggerSelectionForData) {
269  auto trgselresult = static_cast<PWG::EMCAL::AliEmcalTriggerDecisionContainer *>(fInputEvent->FindListObject(fNameTriggerDecisionContainer));
270  AliDebugStream(1) << "Found trigger decision object: " << (trgselresult ? "yes" : "no") << std::endl;
271  if(!trgselresult){
272  AliErrorStream() << "Trigger decision container with name " << fNameTriggerDecisionContainer << " not found in event - not possible to select EMCAL triggers" << std::endl;
273  return false;
274  }
275  if(!trgselresult->IsEventSelected(fTriggerSelectionString)) return false;
276  }
277  }
278  } else {
279  if(!(fInputHandler->IsEventSelected() & AliVEvent::kINT7)) return false;
281  // Simulation - do EMCAL trigger selection from trigger selection object
282  auto mctrigger = static_cast<PWG::EMCAL::AliEmcalTriggerDecisionContainer *>(fInputEvent->FindListObject(fNameTriggerDecisionContainer));
283  AliDebugStream(1) << "Found trigger decision object: " << (mctrigger ? "yes" : "no") << std::endl;
284  if(!mctrigger){
285  AliErrorStream() << "Trigger decision container with name " << fNameTriggerDecisionContainer << " not found in event - not possible to select EMCAL triggers" << std::endl;
286  return false;
287  }
288  if(!mctrigger->IsEventSelected(fTriggerSelectionString)) return false;
289  }
290  }
291  return true;
292 }
293 
294 std::string AliAnalysisTaskEmcalJetEnergySpectrum::MatchTrigger(const std::string &triggerstring){
295  auto triggerclasses = PWG::EMCAL::Triggerinfo::DecodeTriggerString(triggerstring);
296  std::string result;
297  for(const auto &t : triggerclasses) {
298  // Use CENT cluster for downscaling
299  if(t.Triggercluster() != "CENT") continue;
300  if(t.Triggerclass().find(fTriggerSelectionString.Data()) != std::string::npos) continue;
301  result = t.ExpandClassName();
302  break;
303  }
304  return result;
305 }
306 
307 bool AliAnalysisTaskEmcalJetEnergySpectrum::IsSelectEmcalTriggers(const std::string &triggerstring) const {
308  const std::array<std::string, 8> kEMCALTriggers = {
309  "EJ1", "EJ2", "DJ1", "DJ2", "EG1", "EG2", "DG1", "DG2"
310  };
311  bool isEMCAL = false;
312  for(auto emcaltrg : kEMCALTriggers) {
313  if(triggerstring.find(emcaltrg) != std::string::npos) {
314  isEMCAL = true;
315  break;
316  }
317  }
318  return isEMCAL;
319 }
320 
321 
323  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
324  if(!mgr) {
325  std::cerr << "Analysis manager not initialized" << std::endl;
326  return nullptr;
327  }
328 
329  Bool_t isAOD(kFALSE);
330  AliInputEventHandler *inputhandler = static_cast<AliInputEventHandler *>(mgr->GetInputEventHandler());
331  if(inputhandler) {
332  if(inputhandler->IsA() == AliAODInputHandler::Class()){
333  std::cout << "Analysing AOD events\n";
334  isAOD = kTRUE;
335  } else {
336  std::cout << "Analysing ESD events\n";
337  }
338  }
339 
340  std::string jettypestring;
342  switch(jettype){
343  case AliJetContainer::kChargedJet: jettypestring = "ChargedJets"; acctype = AliJetContainer::kTPCfid; break;
344  case AliJetContainer::kFullJet: jettypestring = "FullJets"; acctype = AliJetContainer::kEMCALfid; break;
345  case AliJetContainer::kNeutralJet: jettypestring = "NeutralJets"; acctype = AliJetContainer::kEMCALfid; break;
346  };
347 
348  std::stringstream tag, outfilename;
349  tag << jettypestring << "_R" << std::setw(2) << std::setfill('0') << int(radius * 10.) << "_" << trigger;
350  if(strlen(suffix)) {
351  tag << "_" << suffix;
352  }
353  auto task = new AliAnalysisTaskEmcalJetEnergySpectrum(Form("JetEnergySpectrum_%s", tag.str().data()));
354  task->SetIsMC(isMC);
355  mgr->AddTask(task);
356 
357  auto contains = [](const std::string &str, const std::string &test) {
358  return str.find(test) != std::string::npos;
359  };
360 
361  std::string trgstr(trigger);
362  if(contains(trgstr, "INT7")) task->SetTriggerSelection(AliVEvent::kINT7, "INT7");
363  else if(contains(trgstr, "EJ1")) task->SetTriggerSelection(AliVEvent::kEMCEJE, "EJ1");
364  else if(contains(trgstr, "EJ2")) task->SetTriggerSelection(AliVEvent::kEMCEJE, "EJ2");
365  else if(contains(trgstr, "EG1")) task->SetTriggerSelection(AliVEvent::kEMCEGA, "EG1");
366  else if(contains(trgstr, "EG2")) task->SetTriggerSelection(AliVEvent::kEMCEGA, "EG2");
367 
368  // Connect particle and cluster container
369  AliTrackContainer *tracks(nullptr);
370  AliClusterContainer *clusters(nullptr);
371  if(jettype == AliJetContainer::kChargedJet || jettype == AliJetContainer::kFullJet) {
373  tracks->SetMinPt(0.15);
374  }
375  if(jettype == AliJetContainer::kNeutralJet || jettype == AliJetContainer::kFullJet){
376  clusters = task->AddClusterContainer(EMCalTriggerPtAnalysis::AliEmcalAnalysisFactory::ClusterContainerNameFactory(isAOD));
377  clusters->SetDefaultClusterEnergy(AliVCluster::kHadCorr);
378  clusters->SetClusHadCorrEnergyCut(0.3);
379  }
380 
381 
382  // Create proper jet container
383  auto jetcont = task->AddJetContainer(jettype, AliJetContainer::antikt_algorithm, AliJetContainer::E_scheme, radius, acctype, tracks, clusters);
384  jetcont->SetName("datajets");
385  task->SetNameJetContainer("datajets");
386  std::cout << "Adding jet container with underlying array:" << jetcont->GetArrayName() << std::endl;
387 
388  // Link input and output container
389  outfilename << mgr->GetCommonFileName() << ":JetSpectrum_" << tag.str().data();
390  mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
391  mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("JetSpectrum_%s", tag.str().data()), TList::Class(), AliAnalysisManager::kOutputContainer, outfilename.str().data()));
392 
393  return task;
394 }
EMCal fiducial acceptance (each eta, phi edge narrowed by jet R)
double Double_t
Definition: External.C:58
AliEventCuts * fEventCuts
Event cuts as implemented in AliEventCuts.
Class creating a variable binning, used in the histogram manager.
Class creating a linear binning, used in the histogram manager.
AliJetContainer * GetJetContainer(Int_t i=0) const
Double_t Eta() const
Definition: AliEmcalJet.h:121
Double_t Phi() const
Definition: AliEmcalJet.h:117
Container with name, TClonesArray and cuts for particles.
void SetUseAliAnaUtils(Bool_t b, Bool_t bRejPilup=kTRUE)
Bool_t fRequireSubsetMB
Require for triggers to be a subset of Min. Bias (for efficiency studies)
static AliAnalysisTaskEmcalJetEnergySpectrum * AddTaskJetEnergySpectrum(Bool_t isMC, AliJetContainer::EJetType_t jettype, double radius, const char *trigger, const char *suffix="")
static AliEmcalDownscaleFactorsOCDB * Instance()
std::vector< TriggerCluster_t > GetTriggerClusterIndices(const TString &triggerstring) const
Double_t GetDownscaleFactorForTriggerClass(const TString &trigger) const
Double_t E() const
Definition: AliEmcalJet.h:119
Interface for binnings used by the histogram handler.
Definition: TBinning.h:21
static std::vector< PWG::EMCAL::Triggerinfo > DecodeTriggerString(const std::string &triggerstring)
Decoding trigger string.
void FillTHnSparse(const char *name, const double *x, double weight=1., Option_t *opt="")
virtual bool Run()
Run function. This is the core function of the analysis and contains the user code. Therefore users have to implement this function.
int Int_t
Definition: External.C:63
unsigned int UInt_t
Definition: External.C:33
THashList * GetListOfHistograms() const
Get the list of histograms.
Definition: THistManager.h:671
ULong_t fMinBiasTrigger
Min bias trigger for trigger subset (for efficiency studies)
TH1 * CreateTH1(const char *name, const char *title, int nbins, double xmin, double xmax, Option_t *opt="")
Create a new TH1 within the container.
TPC fiducial acceptance (each eta edge narrowed by jet R)
void FillTH1(const char *hname, double x, double weight=1., Option_t *opt="")
Fill a 1D histogram within the container.
Bool_t fRequestTriggerClusters
Request distinction of trigger clusters.
Double_t Pt() const
Definition: AliEmcalJet.h:109
virtual void RunChanged(Int_t newrun)
Process tasks relevant when a file with a different run number is processed.
virtual Bool_t IsEventSelected()
Performing event selection.
Bool_t fUseTriggerSelectionForData
Use trigger selection on data (require trigger patch in addition to trigger selection string) ...
Bool_t isMC
AliEmcalList * fOutput
!output list
Base task in the EMCAL jet framework.
Represent a jet reconstructed using the EMCal jet framework.
Definition: AliEmcalJet.h:51
Container class for histograms.
Definition: THistManager.h:99
void test(int runnumber=195345)
void UserCreateOutputObjects()
Main initialization function on the worker.
bool Bool_t
Definition: External.C:53
Double_t NEF() const
Definition: AliEmcalJet.h:148
void SetDefaultClusterEnergy(Int_t d)
THnSparse * CreateTHnSparse(const char *name, const char *title, int ndim, const int *nbins, const double *min, const double *max, Option_t *opt="")
Create a new THnSparse within the container.
Container structure for EMCAL clusters.
void SetClusHadCorrEnergyCut(Double_t cut)
static TString ClusterContainerNameFactory(Bool_t isAOD)
Get name of the default cluster container.
static TString TrackContainerNameFactory(Bool_t isAOD)
Get name of the default track container.