8cfd1c088e23dade1642f9a85194755a472ffd2d
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisHelperJetTasks.cxx
1
2 #include "TROOT.h"
3 #include "TKey.h"
4 #include "TList.h"
5 #include "TSystem.h"
6 #include "TH1F.h"
7 #include "TProfile.h"
8 #include "THnSparse.h"
9 #include "TFile.h"
10 #include "TString.h"
11 #include "AliMCEvent.h"
12 #include "AliLog.h"
13 #include "AliESDEvent.h"
14 #include "AliAODJet.h"
15 #include "AliStack.h"
16 #include "AliGenEventHeader.h"
17 #include "AliGenCocktailEventHeader.h"
18 #include "AliGenPythiaEventHeader.h"
19 #include <fstream>
20 #include <iostream>
21 #include "AliAnalysisHelperJetTasks.h"
22 #include "TMatrixDSym.h"
23 #include "TMatrixDSymEigen.h"
24 #include "TVector.h"
25
26 ClassImp(AliAnalysisHelperJetTasks)
27
28
29
30  
31 AliGenPythiaEventHeader*  AliAnalysisHelperJetTasks::GetPythiaEventHeader(AliMCEvent *mcEvent){
32   
33   AliGenEventHeader* genHeader = mcEvent->GenEventHeader();
34   AliGenPythiaEventHeader* pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(genHeader);
35   if(!pythiaGenHeader){
36     // cocktail ??
37     AliGenCocktailEventHeader* genCocktailHeader = dynamic_cast<AliGenCocktailEventHeader*>(genHeader);
38     
39     if (!genCocktailHeader) {
40       AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Unknown header type (not Pythia or Cocktail)");
41       //      AliWarning(Form("%s %d: Unknown header type (not Pythia or Cocktail)",(char*)__FILE__,__LINE__));
42       return 0;
43     }
44     TList* headerList = genCocktailHeader->GetHeaders();
45     for (Int_t i=0; i<headerList->GetEntries(); i++) {
46       pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(headerList->At(i));
47       if (pythiaGenHeader)
48         break;
49     }
50     if(!pythiaGenHeader){
51       AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Pythia event header not found");
52       return 0;
53     }
54   }
55   return pythiaGenHeader;
56
57 }
58
59
60 void AliAnalysisHelperJetTasks::PrintStack(AliMCEvent *mcEvent,Int_t iFirst,Int_t iLast,Int_t iMaxPrint){
61
62   AliStack *stack = mcEvent->Stack();
63   if(!stack){
64     Printf("%s%d No Stack available",(char*)__FILE__,__LINE__);
65     return;
66   }
67
68   static Int_t iCount = 0;
69   if(iCount>iMaxPrint)return;
70   Int_t nStack = stack->GetNtrack();
71   if(iLast == 0)iLast = nStack;
72   else if(iLast > nStack)iLast = nStack;
73
74
75   Printf("####################################################################");
76   for(Int_t np = iFirst;np<iLast;++np){
77     TParticle *p = stack->Particle(np);
78     Printf("Nr.%d --- Status %d ---- Mother1 %d Mother2 %d Daughter1 %d Daughter2 %d ",
79            np,p->GetStatusCode(),p->GetMother(0),p->GetMother(1),p->GetDaughter(0),p->GetDaughter(1));
80     Printf("Eta %3.3f Phi %3.3f ",p->Eta(),p->Phi()); 
81     p->Print();    
82     Printf("---------------------------------------");
83   }
84   iCount++;
85 }
86
87
88
89
90 void AliAnalysisHelperJetTasks::GetClosestJets(AliAODJet *genJets,const Int_t &kGenJets,
91                                                AliAODJet *recJets,const Int_t &kRecJets,
92                                                Int_t *iGenIndex,Int_t *iRecIndex,
93                                                Int_t iDebug,Float_t maxDist){
94
95   //
96   // Relate the two input jet Arrays
97   //
98
99   //
100   // The association has to be unique
101   // So check in two directions
102   // find the closest rec to a gen
103   // and check if there is no other rec which is closer
104   // Caveat: Close low energy/split jets may disturb this correlation
105
106
107   // Idea: search in two directions generated e.g (a--e) and rec (1--3)
108   // Fill a matrix with Flags (1 for closest rec jet, 2 for closest rec jet
109   // in the end we have something like this
110   //    1   2   3
111   // ------------
112   // a| 3   2   0
113   // b| 0   1   0
114   // c| 0   0   3
115   // d| 0   0   1
116   // e| 0   0   1
117   // Topology
118   //   1     2
119   //     a         b        
120   //
121   //  d      c
122   //        3     e
123   // Only entries with "3" match from both sides
124
125   // In case we have more jets than kmaxjets only the 
126   // first kmaxjets are searched
127   // all other are -1
128   // use kMaxJets for a test not to fragemnt the memory...
129
130   for(int i = 0;i < kGenJets;++i)iGenIndex[i] = -1;
131   for(int j = 0;j < kRecJets;++j)iRecIndex[j] = -1;
132
133
134   
135   const int kMode = 3;
136
137   const Int_t nGenJets = TMath::Min(kMaxJets,kGenJets);
138   const Int_t nRecJets = TMath::Min(kMaxJets,kRecJets);
139
140   if(nRecJets==0||nGenJets==0)return;
141
142   // UShort_t *iFlag = new UShort_t[nGenJets*nRecJets];
143   UShort_t iFlag[kMaxJets*kMaxJets];
144   for(int i = 0;i < nGenJets;++i){
145     for(int j = 0;j < nRecJets;++j){
146       iFlag[i*nGenJets+j] = 0;
147     }
148   }
149
150
151
152   // find the closest distance to the generated
153   for(int ig = 0;ig<nGenJets;++ig){
154     Float_t dist = maxDist;
155     if(iDebug>1)Printf("Gen (%d) p_T %3.3f eta %3.3f ph %3.3f ",ig,genJets[ig].Pt(),genJets[ig].Eta(),genJets[ig].Phi());
156     for(int ir = 0;ir<nRecJets;++ir){
157       Double_t dR = genJets[ig].DeltaR(&recJets[ir]);
158       if(iDebug>1)Printf("Rec (%d) p_T %3.3f eta %3.3f ph %3.3f ",ir,recJets[ir].Pt(),recJets[ir].Eta(),recJets[ir].Phi());
159       if(iDebug>1)Printf("Distance (%d)--(%d) %3.3f ",ig,ir,dR);
160       if(dR<dist){
161         iRecIndex[ig] = ir;
162         dist = dR;
163       } 
164     }
165     if(iRecIndex[ig]>=0)iFlag[ig*nGenJets+iRecIndex[ig]]+=1;
166     // reset...
167     iRecIndex[ig] = -1;
168   }
169   // other way around
170   for(int ir = 0;ir<nRecJets;++ir){
171     Float_t dist = maxDist;
172     for(int ig = 0;ig<nGenJets;++ig){
173       Double_t dR = genJets[ig].DeltaR(&recJets[ir]);
174       if(dR<dist){
175         iGenIndex[ir] = ig;
176         dist = dR;
177       } 
178     }
179     if(iGenIndex[ir]>=0)iFlag[iGenIndex[ir]*nGenJets+ir]+=2;
180     // reset...
181     iGenIndex[ir] = -1;
182   }
183
184   // check for "true" correlations
185
186   if(iDebug>1)Printf(">>>>>> Matrix");
187
188   for(int ig = 0;ig<nGenJets;++ig){
189     for(int ir = 0;ir<nRecJets;++ir){
190       // Print
191       if(iDebug>1)printf("Flag[%d][%d] %d ",ig,ir,iFlag[ig*nGenJets+ir]);
192
193       if(kMode==3){
194         // we have a uniqie correlation
195         if(iFlag[ig*nGenJets+ir]==3){
196           iGenIndex[ir] = ig;
197           iRecIndex[ig] = ir;
198         }
199       }
200       else{
201         // we just take the correlation from on side
202         if((iFlag[ig*nGenJets+ir]&2)==2){
203           iGenIndex[ir] = ig;
204         }
205         if((iFlag[ig*nGenJets+ir]&1)==1){
206           iRecIndex[ig] = ir;
207         }
208       }
209     }
210     if(iDebug>1)printf("\n");
211   }
212 }
213
214
215
216 void  AliAnalysisHelperJetTasks::MergeOutput(char* cFiles, char* cList){
217
218   // This is used to merge the analysis-output from different 
219   // data samples/pt_hard bins
220   // in case the eventweigth was set to xsection/ntrials already, this
221   // is not needed. Both methods only work in case we do not mix different 
222   // pt_hard bins, and do not have overlapping bins
223
224   const Int_t nMaxBins = 12;
225   // LHC08q jetjet100: Mean = 1.42483e-03, RMS = 6.642e-05
226   // LHC08r jetjet50: Mean = 2.44068e-02, RMS = 1.144e-03
227   // LHC08v jetjet15-50: Mean = 2.168291 , RMS = 7.119e-02
228   // const Float_t xsection[nBins] = {2.168291,2.44068e-02};
229
230   Float_t xsection[nMaxBins];
231   Float_t nTrials[nMaxBins];
232   Float_t sf[nMaxBins];
233   TList *lIn[nMaxBins];
234   TFile *fIn[nMaxBins];
235
236   ifstream in1;
237   in1.open(cFiles);
238
239   char cFile[120];
240   Int_t ibTotal = 0;
241   while(in1>>cFile){
242     fIn[ibTotal] = TFile::Open(cFile);
243     lIn[ibTotal] = (TList*)fIn[ibTotal]->Get(cList);
244     Printf("Merging file %s",cFile);
245     if(!lIn[ibTotal]){
246       Printf("%s:%d No list %s found, exiting...",__FILE__,__LINE__,cList);
247       fIn[ibTotal]->ls();
248       return;
249     }
250     TH1* hTrials = (TH1F*)lIn[ibTotal]->FindObject("fh1Trials");
251     if(!hTrials){
252       Printf("%s:%d fh1PtHard_Trials not found in list, exiting...",__FILE__,__LINE__);
253       return;
254     }
255     TProfile* hXsec = (TProfile*)lIn[ibTotal]->FindObject("fh1Xsec");
256     if(!hXsec){
257       Printf("%s:%d fh1Xsec  not found in list, exiting...",__FILE__,__LINE__);
258       return;
259     }
260     xsection[ibTotal] = hXsec->GetBinContent(1);
261     nTrials[ibTotal] = hTrials->Integral();
262     sf[ibTotal] = xsection[ibTotal]/ nTrials[ibTotal];
263     ibTotal++;
264   }
265
266   if(ibTotal==0){
267     Printf("%s:%d No files found for mergin, exiting",__FILE__,__LINE__);
268     return;
269   }
270
271   TFile *fOut = new TFile("allpt.root","RECREATE");
272   TList *lOut = new TList();
273   lOut->SetName(lIn[0]->GetName());
274   // for the start scale all...
275   for(int ie = 0; ie < lIn[0]->GetEntries();++ie){
276     TH1 *h1Add = 0;
277     THnSparse *hnAdd = 0;
278     for(int ib = 0;ib < ibTotal;++ib){
279       // dynamic cast does not work with cint
280       TObject *h = lIn[ib]->At(ie);
281       if(h->InheritsFrom("TH1")){
282         TH1 *h1 = (TH1*)h;
283         if(ib==0){
284           h1Add = (TH1*)h1->Clone(h1->GetName());
285           h1Add->Scale(sf[ib]);
286         }
287         else{
288           h1Add->Add(h1,sf[ib]);
289         }
290       }
291       else if(h->InheritsFrom("THnSparse")){
292         THnSparse *hn = (THnSparse*)h;
293         if(ib==0){
294           hnAdd = (THnSparse*)hn->Clone(hn->GetName());
295           hnAdd->Scale(sf[ib]);
296         }
297         else{
298           hnAdd->Add(hn,sf[ib]);
299         }
300       }
301       
302
303     }// ib
304     if(h1Add)lOut->Add(h1Add);
305     else if(hnAdd)lOut->Add(hnAdd);
306   }
307   fOut->cd();
308   lOut->Write(lOut->GetName(),TObject::kSingleKey);
309   fOut->Close();
310 }
311
312 Bool_t AliAnalysisHelperJetTasks::PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials){
313   //
314   // get the cross section and the trails either from pyxsec.root or from pysec_hists.root
315   // This is to called in Notify and should provide the path to the AOD/ESD file
316
317   TString file(currFile);  
318   fXsec = 0;
319   fTrials = 1;
320
321   if(file.Contains("root_archive.zip#")){
322     Ssiz_t pos1 = file.Index("root_archive",12,TString::kExact);
323     Ssiz_t pos = file.Index("#",1,pos1,TString::kExact);
324     file.Replace(pos+1,20,"");
325   }
326   else {
327     // not an archive take the basename....
328     file.ReplaceAll(gSystem->BaseName(file.Data()),"");
329   }
330   Printf("%s",file.Data());
331   
332  
333    
334
335   TFile *fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec.root")); // problem that we cannot really test the existance of a file in a archive so we have to lvie with open error message from root
336   if(!fxsec){
337     // next trial fetch the histgram file
338     fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec_hists.root"));
339     if(!fxsec){
340         // not a severe condition but inciate that we have no information
341       return kFALSE;
342     }
343     else{
344       // find the tlist we want to be independtent of the name so use the Tkey
345       TKey* key = (TKey*)fxsec->GetListOfKeys()->At(0); 
346       if(!key){
347         fxsec->Close();
348         return kFALSE;
349       }
350       TList *list = dynamic_cast<TList*>(key->ReadObj());
351       if(!list){
352         fxsec->Close();
353         return kFALSE;
354       }
355       fXsec = ((TProfile*)list->FindObject("h1Xsec"))->GetBinContent(1);
356       fTrials  = ((TH1F*)list->FindObject("h1Trials"))->GetBinContent(1);
357       fxsec->Close();
358     }
359   } // no tree pyxsec.root
360   else {
361     TTree *xtree = (TTree*)fxsec->Get("Xsection");
362     if(!xtree){
363       fxsec->Close();
364       return kFALSE;
365     }
366     UInt_t   ntrials  = 0;
367     Double_t  xsection  = 0;
368     xtree->SetBranchAddress("xsection",&xsection);
369     xtree->SetBranchAddress("ntrials",&ntrials);
370     xtree->GetEntry(0);
371     fTrials = ntrials;
372     fXsec = xsection;
373     fxsec->Close();
374   }
375   return kTRUE;
376 }
377
378 Bool_t  AliAnalysisHelperJetTasks::Selected(Bool_t bSet,Bool_t bNew){
379   static Bool_t bSelected = kTRUE; // if service task is not run we acccpet all
380   if(bSet){
381     bSelected = bNew;
382   }
383   return bSelected;
384 }
385
386 //___________________________________________________________________________________________________________
387
388 Bool_t AliAnalysisHelperJetTasks::GetEventShapes(TVector3 &n01, TVector3 * pTrack, Int_t nTracks, Double_t * eventShapes)
389 {       
390   // ***
391   // Event shape calculation
392   // sona.pochybova@cern.ch
393
394   const Int_t kTracks = 1000;
395   if(nTracks>kTracks)return kFALSE;
396
397   //variables for thrust calculation
398   TVector3 pTrackPerp[kTracks];
399   Double_t psum2 = 0;
400
401   TVector3 psum;
402   TVector3 psum02;
403   TVector3 psum03;
404
405   Double_t psum1 = 0;
406   Double_t psum102 = 0;
407   Double_t psum103 = 0;
408
409   Double_t thrust[kTracks];
410   Double_t th = -3;
411   Double_t thrust02[kTracks];
412   Double_t th02 = -4;
413   Double_t thrust03[kTracks];
414   Double_t th03 = -5;
415
416   //Sphericity calculation variables
417   TMatrixDSym m(3);
418   Double_t s00 = 0;
419   Double_t s01 = 0;
420   Double_t s02 = 0;
421   
422   Double_t s10 = 0;
423   Double_t s11 = 0;
424   Double_t s12 = 0;
425   
426   Double_t s20 = 0;
427   Double_t s21 = 0;
428   Double_t s22 = 0;
429   
430   Double_t ptot = 0;
431   
432   Double_t c = -10;
433
434 //
435 //loop for thrust calculation  
436 //
437     
438   for(Int_t i = 0; i < nTracks; i++)
439     {
440       pTrackPerp[i].SetXYZ(pTrack[i].X(), pTrack[i].Y(), 0);
441       psum2 += pTrackPerp[i].Mag();
442     }
443
444   //additional starting axis    
445   TVector3 n02;
446   n02 = pTrack[1].Unit();
447   n02.SetZ(0.);   
448   TVector3 n03;
449   n03 = pTrack[2].Unit();
450   n03.SetZ(0.);
451
452   //switches for calculating thrust for different starting points
453   Int_t switch1 = 1;
454   Int_t switch2 = 1;
455   Int_t switch3 = 1;
456
457   //indexes for iteration of different starting points
458   Int_t l1 = 0;
459   Int_t l2 = 0;
460   Int_t l3 = 0;
461
462   //maximal number of iterations
463   //  Int_t nMaxIter = 100;
464  
465   for(Int_t k = 0; k < nTracks; k++)
466     {  
467       
468       if(switch1 == 1){
469         psum.SetXYZ(0., 0., 0.);
470         psum1 = 0;
471         for(Int_t i = 0; i < nTracks; i++)
472           {
473             psum1 += (TMath::Abs(n01.Dot(pTrackPerp[i])));
474             if (n01.Dot(pTrackPerp[i]) > 0) psum += pTrackPerp[i];
475             if (n01.Dot(pTrackPerp[i]) < 0) psum -= pTrackPerp[i];
476           }
477         thrust[l1] = psum1/psum2;
478       }
479
480       if(switch2 == 1){
481         psum02.SetXYZ(0., 0., 0.);
482         psum102 = 0;
483         for(Int_t i = 0; i < nTracks; i++)
484           {
485             psum102 += (TMath::Abs(n02.Dot(pTrackPerp[i])));
486             if (n02.Dot(pTrackPerp[i]) > 0) psum02 += pTrackPerp[i];
487             if (n02.Dot(pTrackPerp[i]) < 0) psum02 -= pTrackPerp[i];
488           }
489         thrust02[l2] = psum102/psum2;
490       }
491       
492       if(switch3 == 1){
493         psum03.SetXYZ(0., 0., 0.);
494         psum103 = 0;
495         for(Int_t i = 0; i < nTracks; i++)
496           {
497             psum103 += (TMath::Abs(n03.Dot(pTrackPerp[i])));
498             if (n03.Dot(pTrackPerp[i]) > 0) psum03 += pTrackPerp[i];
499             if (n03.Dot(pTrackPerp[i]) < 0) psum03 -= pTrackPerp[i];
500           }
501         thrust03[l3] = psum103/psum2;
502       }
503
504       //check whether thrust value converged    
505       if(TMath::Abs(th-thrust[l1]) < 10e-7){
506         switch1 = 0;
507       }
508       
509       if(TMath::Abs(th02-thrust02[l2]) < 10e-7){
510         switch2 = 0;
511       }
512
513       if(TMath::Abs(th03-thrust03[l3]) < 10e-7){
514         switch3 = 0;
515       }
516
517       //if it didn't, continue with the calculation
518       if(switch1 == 1){
519         th = thrust[l1]; 
520         n01 = psum.Unit();
521         l1++;
522       }
523
524       if(switch2 == 1){
525         th02 = thrust02[l2];
526         n02 = psum02.Unit();
527         l2++;
528       }
529
530       if(switch3 == 1){
531         th03 = thrust03[l3];
532         n03 = psum03.Unit();
533         l3++;
534       }
535
536       //if thrust values for all starting direction converged check if to the same value
537       if(switch2 == 0 && switch1 == 0 && switch3 == 0){
538         if(TMath::Abs(th-th02) < 10e-7 && TMath::Abs(th-th03) < 10e-7 && TMath::Abs(th02-th03) < 10e-7){
539           eventShapes[0] = th;
540           AliInfoGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),Form("===== THRUST VALUE FOUND AT %d :: %f\n", k, th));
541           break;
542         }
543         //if they did not, reset switches
544         else{
545           switch1 = 1;
546           //      th = -1.;
547           switch2 = 1;
548           //      th02 = -2.;
549           switch3 = 1;
550           //      th03 = -4.;
551         }
552       }
553
554       //      Printf("========== %d +++ th :: %f=============\n", l1, th);
555       //      Printf("========== %d +++ th2 :: %f=============\n", l2, th02);
556       //      Printf("========== %d +++ th3 :: %f=============\n", l3, th03);
557       
558     }
559
560   //if no common limitng value was found, take the maximum and take the corresponding thrust axis
561   if(switch1 == 1 && switch2 == 1 && switch3 == 1){
562     eventShapes[0] = TMath::Max(thrust[l1-1], thrust02[l2-1]);
563     eventShapes[0] = TMath::Max(eventShapes[0], thrust03[l3-1]);
564     if(TMath::Abs(eventShapes[0]-thrust[l1-1]) < 10e-7)
565       n01 = n01;
566     if(TMath::Abs(eventShapes[0]-thrust02[l2-1]) < 10e-7)
567       n01 = n02;
568     if(TMath::Abs(eventShapes[0]-thrust03[l3-1]) < 10e-7)
569       n01 = n03;
570     Printf("NO LIMITING VALUE FOUND :: MAXIMUM = %f\n", eventShapes[0]);
571   }
572
573 //
574 //other event shapes variables
575 //
576   for(Int_t j = 0; j < nTracks; j++)
577     {  
578       s00 = s00 + (pTrack[j].Px()*pTrack[j].Px())/pTrack[j].Mag();
579       s01 = s01 + (pTrack[j].Px()*pTrack[j].Py())/pTrack[j].Mag();
580       s02 = s02 + (pTrack[j].Px()*pTrack[j].Pz())/pTrack[j].Mag();
581       
582       s10 = s10 + (pTrack[j].Py()*pTrack[j].Px())/pTrack[j].Mag();
583       s11 = s11 + (pTrack[j].Py()*pTrack[j].Py())/pTrack[j].Mag();
584       s12 = s12 + (pTrack[j].Py()*pTrack[j].Pz())/pTrack[j].Mag();
585       
586       s20 = s20 + (pTrack[j].Pz()*pTrack[j].Px())/pTrack[j].Mag();
587       s21 = s21 + (pTrack[j].Pz()*pTrack[j].Py())/pTrack[j].Mag();
588       s22 = s22 + (pTrack[j].Pz()*pTrack[j].Pz())/pTrack[j].Mag();
589       
590       ptot += pTrack[j].Mag();
591     }
592
593   if(ptot > 0.)
594     {
595       m(0,0) = s00/ptot;
596       m(0,1) = s01/ptot;
597       m(0,2) = s02/ptot;
598
599       m(1,0) = s10/ptot;
600       m(1,1) = s11/ptot;
601       m(1,2) = s12/ptot;
602
603       m(2,0) = s20/ptot;
604       m(2,1) = s21/ptot;
605       m(2,2) = s22/ptot;
606
607       TMatrixDSymEigen eigen(m);
608       TVectorD eigenVal = eigen.GetEigenValues();
609
610       Double_t sphericity = (3/2)*(eigenVal(2)+eigenVal(1));
611       eventShapes[1] = sphericity;
612
613       Double_t aplanarity = (3/2)*(eigenVal(2));
614       eventShapes[2] = aplanarity;
615
616       c = 3*(eigenVal(0)*eigenVal(1)+eigenVal(0)*eigenVal(2)+eigenVal(1)*eigenVal(2));
617       eventShapes[3] = c;
618     }
619   return kTRUE;
620 }
621   
622
623
624  //__________________________________________________________________________________________________________________________
625 // Trigger Decisions copid from PWG0/AliTriggerAnalysis
626
627
628 Bool_t AliAnalysisHelperJetTasks::IsTriggerFired(const AliVEvent* aEv, Trigger trigger)
629 {
630   // checks if an event has been triggered
631   // no usage of ofline trigger here yet
632   return IsTriggerBitFired(aEv, trigger);
633 }
634
635 Bool_t AliAnalysisHelperJetTasks::IsTriggerBitFired(const AliVEvent* aEv, Trigger trigger)
636 {
637   // checks if an event is fired using the trigger bits
638
639   return IsTriggerBitFired(aEv->GetTriggerMask(), trigger);
640 }
641
642 Bool_t AliAnalysisHelperJetTasks::IsTriggerBitFired(ULong64_t triggerMask, Trigger trigger)
643 {
644   // checks if an event is fired using the trigger bits
645   //
646   // this function needs the branch TriggerMask in the ESD
647
648   // definitions from p-p.cfg
649   ULong64_t spdFO = (1 << 14);
650   ULong64_t v0left = (1 << 10);
651   ULong64_t v0right = (1 << 11);
652
653   switch (trigger)
654   {
655     case kAcceptAll:
656     {
657       return kTRUE;
658       break;
659     }
660     case kMB1:
661     {
662       if (triggerMask & spdFO || ((triggerMask & v0left) || (triggerMask & v0right)))
663         return kTRUE;
664       break;
665     }
666     case kMB2:
667     {
668       if (triggerMask & spdFO && ((triggerMask & v0left) || (triggerMask & v0right)))
669         return kTRUE;
670       break;
671     }
672     case kMB3:
673     {
674       if (triggerMask & spdFO && (triggerMask & v0left) && (triggerMask & v0right))
675         return kTRUE;
676       break;
677     }
678     case kSPDGFO:
679     {
680       if (triggerMask & spdFO)
681         return kTRUE;
682       break;
683     }
684     default:
685       Printf("IsEventTriggered: ERROR: Trigger type %d not implemented in this method", (Int_t) trigger);
686       break;
687   }
688
689   return kFALSE;
690 }
691
692 Bool_t AliAnalysisHelperJetTasks::IsTriggerBitFired(const AliVEvent* aEv, ULong64_t tclass)
693 {
694   // Checks if corresponding bit in mask is on
695   
696   ULong64_t trigmask = aEv->GetTriggerMask();
697   return (trigmask & (1ull << (tclass-1)));
698 }