PWG4/JetTasks/AliAnalysisHelperJetTasks.cxx

   1
   2 #include "TROOT.h"
   3 #include "TDirectory.h"
   4 #include "TKey.h"
   5 #include "TList.h"
   6 #include "TSystem.h"
   7 #include "TH1F.h"
   8 #include "TProfile.h"
   9 #include "THnSparse.h"
  10 #include "TSeqCollection.h"
  11 #include "TMethodCall.h"
  12 #include "TFile.h"
  13 #include "TString.h"
  14 #include "AliMCEvent.h"
  15 #include "AliLog.h"
  16 #include "AliESDEvent.h"
  17 #include "AliAODJet.h"
  18 #include "AliAODEvent.h"
  19 #include "AliStack.h"
  20 #include "AliGenEventHeader.h"
  21 #include "AliGenCocktailEventHeader.h"
  22 #include <AliGenDPMjetEventHeader.h>
  23 #include "AliGenPythiaEventHeader.h"
  24 #include <fstream>
  25 #include <iostream>
  26 #include "AliAnalysisHelperJetTasks.h"
  27 #include "TMatrixDSym.h"
  28 #include "TMatrixDSymEigen.h"
  29 #include "TVector.h"
  30
  31 ClassImp(AliAnalysisHelperJetTasks)
  32
  33 Int_t AliAnalysisHelperJetTasks::fgLastProcessType = -1;
  34
  35
  36 AliGenPythiaEventHeader*  AliAnalysisHelperJetTasks::GetPythiaEventHeader(AliMCEvent *mcEvent){
  37
  38   if(!mcEvent)return 0;
  39   AliGenEventHeader* genHeader = mcEvent->GenEventHeader();
  40   AliGenPythiaEventHeader* pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(genHeader);
  41   if(!pythiaGenHeader){
  42     // cocktail ??
  43     AliGenCocktailEventHeader* genCocktailHeader = dynamic_cast<AliGenCocktailEventHeader*>(genHeader);
  44
  45     if (!genCocktailHeader) {
  46       AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Unknown header type (not Pythia or Cocktail)");
  47       //      AliWarning(Form("%s %d: Unknown header type (not Pythia or Cocktail)",(char*)__FILE__,__LINE__));
  48       return 0;
  49     }
  50     TList* headerList = genCocktailHeader->GetHeaders();
  51     for (Int_t i=0; i<headerList->GetEntries(); i++) {
  52       pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(headerList->At(i));
  53       if (pythiaGenHeader)
  54         break;
  55     }
  56     if(!pythiaGenHeader){
  57       AliWarningGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),"Pythia event header not found");
  58       return 0;
  59     }
  60   }
  61   return pythiaGenHeader;
  62
  63 }
  64
  65
  66 void AliAnalysisHelperJetTasks::PrintStack(AliMCEvent *mcEvent,Int_t iFirst,Int_t iLast,Int_t iMaxPrint){
  67
  68   AliStack *stack = mcEvent->Stack();
  69   if(!stack){
  70     Printf("%s%d No Stack available",(char*)__FILE__,__LINE__);
  71     return;
  72   }
  73
  74   static Int_t iCount = 0;
  75   if(iCount>iMaxPrint)return;
  76   Int_t nStack = stack->GetNtrack();
  77   if(iLast == 0)iLast = nStack;
  78   else if(iLast > nStack)iLast = nStack;
  79
  80
  81   Printf("####################################################################");
  82   for(Int_t np = iFirst;np<iLast;++np){
  83     TParticle *p = stack->Particle(np);
  84     Printf("Nr.%d --- Status %d ---- Mother1 %d Mother2 %d Daughter1 %d Daughter2 %d ",
  85            np,p->GetStatusCode(),p->GetMother(0),p->GetMother(1),p->GetDaughter(0),p->GetDaughter(1));
  86     Printf("Eta %3.3f Phi %3.3f ",p->Eta(),p->Phi());
  87     p->Print();
  88     Printf("---------------------------------------");
  89   }
  90   iCount++;
  91 }
  92
  93
  94
  95
  96 void AliAnalysisHelperJetTasks::GetClosestJets(AliAODJet *genJets,const Int_t &kGenJets,
  97                                                AliAODJet *recJets,const Int_t &kRecJets,
  98                                                Int_t *iGenIndex,Int_t *iRecIndex,
  99                                                Int_t iDebug,Float_t maxDist){
 100
 101   //
 102   // Relate the two input jet Arrays
 103   //
 104
 105   //
 106   // The association has to be unique
 107   // So check in two directions
 108   // find the closest rec to a gen
 109   // and check if there is no other rec which is closer
 110   // Caveat: Close low energy/split jets may disturb this correlation
 111
 112
 113   // Idea: search in two directions generated e.g (a--e) and rec (1--3)
 114   // Fill a matrix with Flags (1 for closest rec jet, 2 for closest rec jet
 115   // in the end we have something like this
 116   //    1   2   3
 117   // ------------
 118   // a| 3   2   0
 119   // b| 0   1   0
 120   // c| 0   0   3
 121   // d| 0   0   1
 122   // e| 0   0   1
 123   // Topology
 124   //   1     2
 125   //     a         b
 126   //
 127   //  d      c
 128   //        3     e
 129   // Only entries with "3" match from both sides
 130
 131   // In case we have more jets than kmaxjets only the
 132   // first kmaxjets are searched
 133   // all other are -1
 134   // use kMaxJets for a test not to fragemnt the memory...
 135
 136   for(int i = 0;i < kGenJets;++i)iGenIndex[i] = -1;
 137   for(int j = 0;j < kRecJets;++j)iRecIndex[j] = -1;
 138
 139
 140
 141   const int kMode = 3;
 142
 143   const Int_t nGenJets = TMath::Min(kMaxJets,kGenJets);
 144   const Int_t nRecJets = TMath::Min(kMaxJets,kRecJets);
 145
 146   if(nRecJets==0||nGenJets==0)return;
 147
 148   // UShort_t *iFlag = new UShort_t[nGenJets*nRecJets];
 149   UShort_t iFlag[kMaxJets*kMaxJets];
 150   for(int i = 0;i < nGenJets;++i){
 151     for(int j = 0;j < nRecJets;++j){
 152       iFlag[i*nGenJets+j] = 0;
 153     }
 154   }
 155
 156
 157
 158   // find the closest distance to the generated
 159   for(int ig = 0;ig<nGenJets;++ig){
 160     Float_t dist = maxDist;
 161     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());
 162     for(int ir = 0;ir<nRecJets;++ir){
 163       Double_t dR = genJets[ig].DeltaR(&recJets[ir]);
 164       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());
 165       if(iDebug>1)Printf("Distance (%d)--(%d) %3.3f ",ig,ir,dR);
 166       if(dR<dist){
 167         iRecIndex[ig] = ir;
 168         dist = dR;
 169       }
 170     }
 171     if(iRecIndex[ig]>=0)iFlag[ig*nGenJets+iRecIndex[ig]]+=1;
 172     // reset...
 173     iRecIndex[ig] = -1;
 174   }
 175   // other way around
 176   for(int ir = 0;ir<nRecJets;++ir){
 177     Float_t dist = maxDist;
 178     for(int ig = 0;ig<nGenJets;++ig){
 179       Double_t dR = genJets[ig].DeltaR(&recJets[ir]);
 180       if(dR<dist){
 181         iGenIndex[ir] = ig;
 182         dist = dR;
 183       }
 184     }
 185     if(iGenIndex[ir]>=0)iFlag[iGenIndex[ir]*nGenJets+ir]+=2;
 186     // reset...
 187     iGenIndex[ir] = -1;
 188   }
 189
 190   // check for "true" correlations
 191
 192   if(iDebug>1)Printf(">>>>>> Matrix");
 193
 194   for(int ig = 0;ig<nGenJets;++ig){
 195     for(int ir = 0;ir<nRecJets;++ir){
 196       // Print
 197       if(iDebug>1)printf("Flag[%d][%d] %d ",ig,ir,iFlag[ig*nGenJets+ir]);
 198
 199       if(kMode==3){
 200         // we have a uniqie correlation
 201         if(iFlag[ig*nGenJets+ir]==3){
 202           iGenIndex[ir] = ig;
 203           iRecIndex[ig] = ir;
 204         }
 205       }
 206       else{
 207         // we just take the correlation from on side
 208         if((iFlag[ig*nGenJets+ir]&2)==2){
 209           iGenIndex[ir] = ig;
 210         }
 211         if((iFlag[ig*nGenJets+ir]&1)==1){
 212           iRecIndex[ig] = ir;
 213         }
 214       }
 215     }
 216     if(iDebug>1)printf("\n");
 217   }
 218 }
 219
 220
 221 void  AliAnalysisHelperJetTasks::MergeOutputDirs(const char* cFiles,const char* cPattern,char *cOutFile){
 222   // Routine to merge only directories containing the pattern
 223   //
 224   TString outFile(cOutFile);
 225   if(outFile.Length()==0)outFile = Form("%s.root",cPattern);
 226   ifstream in1;
 227   in1.open(cFiles);
 228   // open all files
 229   TList fileList;
 230   char cFile[240];
 231   if(in1>>cFile){// only open the first file
 232     fileList.Add(TFile::Open(cFile));
 233   }
 234
 235   TFile *fOut = 0;
 236   if(fileList.GetEntries()){// open the first file
 237     TFile* fIn = dynamic_cast<TFile*>(fileList.At(0));
 238     if(fIn){
 239       Printf("Input File not Found");
 240     }
 241     // fetch the keys for the directories
 242     TList *ldKeys = fIn->GetListOfKeys();
 243     for(int id = 0;id < ldKeys->GetEntries();id++){
 244       // check if it is a directory
 245       TKey *dKey = (TKey*)ldKeys->At(id);
 246       TDirectory *dir = dynamic_cast<TDirectory*>(dKey->ReadObj());
 247       if(dir){
 248         TString dName(dir->GetName());
 249         if(dName.Contains(cPattern)){
 250           // open new file if first match
 251           if(!fOut){
 252             fOut = new TFile(outFile.Data(),"RECREATE");
 253           }
 254           // merge all the stuff that is in this directory
 255           TList *llKeys = dir->GetListOfKeys();
 256           TList *tmplist;
 257           TMethodCall callEnv;
 258           for(int il = 0;il < llKeys->GetEntries();il++){
 259             TKey *lKey = (TKey*)llKeys->At(id);
 260             TObject *object = dynamic_cast<TObject*>(lKey->ReadObj());
 261             //  from TSeqCollections::Merge
 262             if(!object)continue;
 263             // If current object is not mergeable just skip it
 264             if (!object->IsA()) {
 265               continue;
 266             }
 267             callEnv.InitWithPrototype(object->IsA(), "Merge", "TCollection*");
 268             if (!callEnv.IsValid()) {
 269               continue;
 270             }
 271             // Current object mergeable - get corresponding objects in input lists
 272             tmplist = new TList();
 273             for(int i = 1; i < fileList.GetEntries();i++){
 274               TDirectory *fInTmp = dynamic_cast<TDirectory*>(fileList.At(i));
 275               if(!fInTmp){
 276                 Printf("File %d not found",i);
 277                 continue;
 278               }
 279               TDirectory *dTmp = (TDirectory*)fInTmp->Get(dName.Data());
 280               if(!dTmp){
 281                 Printf("Directory %s not found",dName.Data());
 282                 continue;
 283               }
 284               TObject* oTmp = (TObject*)dTmp->Get(object->GetName());
 285               if(!oTmp){
 286                 Printf("Object %s not found",object->GetName());
 287                 continue;
 288               }
 289               tmplist->Add(oTmp);
 290             }
 291             callEnv.SetParam((Long_t) tmplist);
 292             callEnv.Execute(object);
 293             delete tmplist;tmplist = 0;
 294             fOut->cd();
 295             object->Write();
 296           }
 297         }
 298       }
 299     }
 300     if(fOut){
 301       fOut->Close();
 302     }
 303   }
 304 }
 305
 306 void  AliAnalysisHelperJetTasks::MergeOutput(char* cFiles, char* cDir, char *cList,char *cOutFile,Bool_t bUpdate){
 307
 308   // This is used to merge the analysis-output from different
 309   // data samples/pt_hard bins
 310   // in case the eventweigth was set to xsection/ntrials already, this
 311   // is not needed. Both methods only work in case we do not mix different
 312   // pt_hard bins, and do not have overlapping bins
 313
 314   const Int_t nMaxBins = 12;
 315   // LHC08q jetjet100: Mean = 1.42483e-03, RMS = 6.642e-05
 316   // LHC08r jetjet50: Mean = 2.44068e-02, RMS = 1.144e-03
 317   // LHC08v jetjet15-50: Mean = 2.168291 , RMS = 7.119e-02
 318   // const Float_t xsection[nBins] = {2.168291,2.44068e-02};
 319
 320   Float_t xsection[nMaxBins];
 321   Float_t nTrials[nMaxBins];
 322   Float_t sf[nMaxBins];
 323   TList *lIn[nMaxBins];
 324   TDirectory *dIn[nMaxBins];
 325   TFile *fIn[nMaxBins];
 326
 327   ifstream in1;
 328   in1.open(cFiles);
 329
 330   char cFile[120];
 331   Int_t ibTotal = 0;
 332   while(in1>>cFile){
 333     fIn[ibTotal] = TFile::Open(cFile);
 334     if(strlen(cDir)==0){
 335       dIn[ibTotal] = gDirectory;
 336     }
 337     else{
 338       dIn[ibTotal] = (TDirectory*)fIn[ibTotal]->Get(cDir);
 339     }
 340     if(!dIn[ibTotal]){
 341       Printf("%s:%d No directory %s found, exiting...",__FILE__,__LINE__,cDir);
 342       fIn[ibTotal]->ls();
 343       return;
 344     }
 345
 346     lIn[ibTotal] = (TList*)dIn[ibTotal]->Get(cList);
 347     Printf("Merging file %s %s",cFile, cDir);
 348     if(!lIn[ibTotal]){
 349       Printf("%s:%d No list %s found, exiting...",__FILE__,__LINE__,cList);
 350       fIn[ibTotal]->ls();
 351       return;
 352     }
 353     TH1* hTrials = (TH1F*)lIn[ibTotal]->FindObject("fh1Trials");
 354     if(!hTrials){
 355       Printf("%s:%d fh1PtHard_Trials not found in list, exiting...",__FILE__,__LINE__);
 356       return;
 357     }
 358     TProfile* hXsec = (TProfile*)lIn[ibTotal]->FindObject("fh1Xsec");
 359     if(!hXsec){
 360       Printf("%s:%d fh1Xsec  not found in list, exiting...",__FILE__,__LINE__);
 361       return;
 362     }
 363     xsection[ibTotal] = hXsec->GetBinContent(1);
 364     nTrials[ibTotal] = hTrials->Integral();
 365     sf[ibTotal] = xsection[ibTotal]/ nTrials[ibTotal];
 366     ibTotal++;
 367   }
 368
 369   if(ibTotal==0){
 370     Printf("%s:%d No files found for mergin, exiting",__FILE__,__LINE__);
 371     return;
 372   }
 373
 374   TFile *fOut = 0;
 375   if(bUpdate)fOut = new TFile(cOutFile,"UPDATE");
 376   else fOut = new TFile(cOutFile,"RECREATE");
 377   TDirectory *dOut = fOut->mkdir(dIn[0]->GetName());
 378   dOut->cd();
 379   TList *lOut = new TList();
 380   lOut->SetName(lIn[0]->GetName());
 381
 382   // for the start scale all...
 383   for(int ie = 0; ie < lIn[0]->GetEntries();++ie){
 384     TH1 *h1Add = 0;
 385     THnSparse *hnAdd = 0;
 386     for(int ib = 0;ib < ibTotal;++ib){
 387       // dynamic cast does not work with cint
 388       TObject *h = lIn[ib]->At(ie);
 389       if(h->InheritsFrom("TH1")){
 390         TH1 *h1 = (TH1*)h;
 391         if(ib==0){
 392           h1Add = (TH1*)h1->Clone(h1->GetName());
 393           h1Add->Scale(sf[ib]);
 394         }
 395         else{
 396           h1Add->Add(h1,sf[ib]);
 397         }
 398       }
 399       else if(h->InheritsFrom("THnSparse")){
 400         THnSparse *hn = (THnSparse*)h;
 401         if(ib==0){
 402           hnAdd = (THnSparse*)hn->Clone(hn->GetName());
 403           hnAdd->Scale(sf[ib]);
 404         }
 405         else{
 406           hnAdd->Add(hn,sf[ib]);
 407         }
 408       }
 409
 410
 411     }// ib
 412     if(h1Add)lOut->Add(h1Add);
 413     else if(hnAdd)lOut->Add(hnAdd);
 414   }
 415   dOut->cd();
 416   lOut->Write(lOut->GetName(),TObject::kSingleKey);
 417   fOut->Close();
 418 }
 419
 420 Bool_t AliAnalysisHelperJetTasks::PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials){
 421   //
 422   // get the cross section and the trails either from pyxsec.root or from pysec_hists.root
 423   // This is to called in Notify and should provide the path to the AOD/ESD file
 424
 425   TString file(currFile);
 426   fXsec = 0;
 427   fTrials = 1;
 428
 429   if(file.Contains("root_archive.zip#")){
 430     Ssiz_t pos1 = file.Index("root_archive",12,TString::kExact);
 431     Ssiz_t pos = file.Index("#",1,pos1,TString::kExact);
 432     file.Replace(pos+1,20,"");
 433   }
 434   else {
 435     // not an archive take the basename....
 436     file.ReplaceAll(gSystem->BaseName(file.Data()),"");
 437   }
 438   Printf("%s",file.Data());
 439
 440
 441
 442
 443   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
 444   if(!fxsec){
 445     // next trial fetch the histgram file
 446     fxsec = TFile::Open(Form("%s%s",file.Data(),"pyxsec_hists.root"));
 447     if(!fxsec){
 448         // not a severe condition but inciate that we have no information
 449       return kFALSE;
 450     }
 451     else{
 452       // find the tlist we want to be independtent of the name so use the Tkey
 453       TKey* key = (TKey*)fxsec->GetListOfKeys()->At(0);
 454       if(!key){
 455         fxsec->Close();
 456         return kFALSE;
 457       }
 458       TList *list = dynamic_cast<TList*>(key->ReadObj());
 459       if(!list){
 460         fxsec->Close();
 461         return kFALSE;
 462       }
 463       fXsec = ((TProfile*)list->FindObject("h1Xsec"))->GetBinContent(1);
 464       fTrials  = ((TH1F*)list->FindObject("h1Trials"))->GetBinContent(1);
 465       fxsec->Close();
 466     }
 467   } // no tree pyxsec.root
 468   else {
 469     TTree *xtree = (TTree*)fxsec->Get("Xsection");
 470     if(!xtree){
 471       fxsec->Close();
 472       return kFALSE;
 473     }
 474     UInt_t   ntrials  = 0;
 475     Double_t  xsection  = 0;
 476     xtree->SetBranchAddress("xsection",&xsection);
 477     xtree->SetBranchAddress("ntrials",&ntrials);
 478     xtree->GetEntry(0);
 479     fTrials = ntrials;
 480     fXsec = xsection;
 481     fxsec->Close();
 482   }
 483   return kTRUE;
 484 }
 485
 486 Bool_t AliAnalysisHelperJetTasks::PrintDirectorySize(const char* currFile){
 487
 488   TFile *fIn = TFile::Open(currFile);
 489   if(!fIn){
 490     // not a severe condition but inciate that we have no information
 491     return kFALSE;
 492   }
 493   // find the tlists we want to be independtent of the name so use the Tkey
 494   TList* keyList = fIn->GetListOfKeys();
 495   Float_t memorySize = 0;
 496   Float_t diskSize = 0;
 497
 498   for(int i = 0;i < keyList->GetEntries();i++){
 499     TKey* ikey = (TKey*)keyList->At(i);
 500
 501     //    TList *list = dynamic_cast<TList*>(key->ReadObj());
 502     //    TNamed *name = dynamic_cast<TNamed*>(ikey->ReadObj());
 503     TDirectory *dir =  dynamic_cast<TDirectory*>(ikey->ReadObj());
 504
 505
 506
 507
 508     if(dir){
 509       Printf("%03d    : %60s %8d %8d ",i,dir->GetName(),ikey->GetObjlen(),ikey->GetNbytes());
 510       TList * dirKeyList = dir->GetListOfKeys();
 511       for(int j = 0;j<dirKeyList->GetEntries();j++){
 512         TKey* jkey = (TKey*)dirKeyList->At(j);
 513         TList *list =  dynamic_cast<TList*>(jkey->ReadObj());
 514
 515         memorySize += (Float_t)jkey->GetObjlen()/1024./1024.;
 516         diskSize +=  (Float_t)jkey->GetNbytes()/1024./1024.;
 517         if(list){
 518           Printf("%03d/%03d: %60s %5.2f MB %5.2f MB",i,j,list->GetName(),(Float_t)jkey->GetObjlen()/1024./1024.,(Float_t)jkey->GetNbytes()/1024./1024.);
 519         }
 520         else{
 521           Printf("%03d/%03d: %60s %5.2f MB %5.2f MB",i,j,jkey->GetName(),(Float_t)jkey->GetObjlen()/1024./1024.,(Float_t)jkey->GetNbytes()/1024./1024.);
 522         }
 523       }
 524     }
 525   }
 526   Printf("Total %5.2f MB %5.2f MB",memorySize,diskSize);
 527   return kTRUE;
 528 }
 529
 530
 531 Bool_t  AliAnalysisHelperJetTasks::Selected(Bool_t bSet,Bool_t bNew){
 532   static Bool_t bSelected = kTRUE; // if service task is not run we acccpet all
 533   if(bSet){
 534     bSelected = bNew;
 535   }
 536   return bSelected;
 537 }
 538
 539 Bool_t  AliAnalysisHelperJetTasks::IsCosmic(){
 540   return ((SelectInfo()&kIsCosmic)==kIsCosmic);
 541 }
 542
 543 Bool_t  AliAnalysisHelperJetTasks::IsPileUp(){
 544   return ((SelectInfo()&kIsPileUp)==kIsPileUp);
 545 }
 546
 547
 548 Bool_t  AliAnalysisHelperJetTasks::TestSelectInfo(UInt_t iMask){
 549   return ((SelectInfo()&iMask)==iMask);
 550 }
 551
 552
 553 UInt_t  AliAnalysisHelperJetTasks::SelectInfo(Bool_t bSet,UInt_t iNew){
 554   static UInt_t iSelectInfo = 0; //
 555   if(bSet){
 556     iSelectInfo = iNew;
 557   }
 558   return iSelectInfo;
 559 }
 560
 561
 562 //___________________________________________________________________________________________________________
 563
 564 Bool_t AliAnalysisHelperJetTasks::GetEventShapes(TVector3 &n01, TVector3 * pTrack, Int_t nTracks, Double_t * eventShapes)
 565 {
 566   // ***
 567   // Event shape calculation
 568   // sona.pochybova@cern.ch
 569
 570   const Int_t kTracks = 1000;
 571   if(nTracks>kTracks)return kFALSE;
 572
 573   //variables for thrust calculation
 574   TVector3 pTrackPerp[kTracks];
 575   Double_t psum2 = 0;
 576
 577   TVector3 psum;
 578   TVector3 psum02;
 579   TVector3 psum03;
 580
 581   Double_t psum1 = 0;
 582   Double_t psum102 = 0;
 583   Double_t psum103 = 0;
 584
 585   Double_t thrust[kTracks];
 586   Double_t th = -3;
 587   Double_t thrust02[kTracks];
 588   Double_t th02 = -4;
 589   Double_t thrust03[kTracks];
 590   Double_t th03 = -5;
 591
 592   //Sphericity calculation variables
 593   TMatrixDSym m(3);
 594   Double_t s00 = 0;
 595   Double_t s01 = 0;
 596   Double_t s02 = 0;
 597
 598   Double_t s10 = 0;
 599   Double_t s11 = 0;
 600   Double_t s12 = 0;
 601
 602   Double_t s20 = 0;
 603   Double_t s21 = 0;
 604   Double_t s22 = 0;
 605
 606   Double_t ptot = 0;
 607
 608   Double_t c = -10;
 609
 610 //
 611 //loop for thrust calculation
 612 //
 613
 614   for(Int_t i = 0; i < nTracks; i++)
 615     {
 616       pTrackPerp[i].SetXYZ(pTrack[i].X(), pTrack[i].Y(), 0);
 617       psum2 += pTrackPerp[i].Mag();
 618     }
 619
 620   //additional starting axis
 621   TVector3 n02;
 622   n02 = pTrack[1].Unit();
 623   n02.SetZ(0.);
 624   TVector3 n03;
 625   n03 = pTrack[2].Unit();
 626   n03.SetZ(0.);
 627
 628   //switches for calculating thrust for different starting points
 629   Int_t switch1 = 1;
 630   Int_t switch2 = 1;
 631   Int_t switch3 = 1;
 632
 633   //indexes for iteration of different starting points
 634   Int_t l1 = 0;
 635   Int_t l2 = 0;
 636   Int_t l3 = 0;
 637
 638   //maximal number of iterations
 639   //  Int_t nMaxIter = 100;
 640
 641   for(Int_t k = 0; k < nTracks; k++)
 642     {
 643
 644       if(switch1 == 1){
 645         psum.SetXYZ(0., 0., 0.);
 646         psum1 = 0;
 647         for(Int_t i = 0; i < nTracks; i++)
 648           {
 649             psum1 += (TMath::Abs(n01.Dot(pTrackPerp[i])));
 650             if (n01.Dot(pTrackPerp[i]) > 0) psum += pTrackPerp[i];
 651             if (n01.Dot(pTrackPerp[i]) < 0) psum -= pTrackPerp[i];
 652           }
 653         thrust[l1] = psum1/psum2;
 654       }
 655
 656       if(switch2 == 1){
 657         psum02.SetXYZ(0., 0., 0.);
 658         psum102 = 0;
 659         for(Int_t i = 0; i < nTracks; i++)
 660           {
 661             psum102 += (TMath::Abs(n02.Dot(pTrackPerp[i])));
 662             if (n02.Dot(pTrackPerp[i]) > 0) psum02 += pTrackPerp[i];
 663             if (n02.Dot(pTrackPerp[i]) < 0) psum02 -= pTrackPerp[i];
 664           }
 665         thrust02[l2] = psum102/psum2;
 666       }
 667
 668       if(switch3 == 1){
 669         psum03.SetXYZ(0., 0., 0.);
 670         psum103 = 0;
 671         for(Int_t i = 0; i < nTracks; i++)
 672           {
 673             psum103 += (TMath::Abs(n03.Dot(pTrackPerp[i])));
 674             if (n03.Dot(pTrackPerp[i]) > 0) psum03 += pTrackPerp[i];
 675             if (n03.Dot(pTrackPerp[i]) < 0) psum03 -= pTrackPerp[i];
 676           }
 677         thrust03[l3] = psum103/psum2;
 678       }
 679
 680       //check whether thrust value converged
 681       if(TMath::Abs(th-thrust[l1]) < 10e-7){
 682         switch1 = 0;
 683       }
 684
 685       if(TMath::Abs(th02-thrust02[l2]) < 10e-7){
 686         switch2 = 0;
 687       }
 688
 689       if(TMath::Abs(th03-thrust03[l3]) < 10e-7){
 690         switch3 = 0;
 691       }
 692
 693       //if it didn't, continue with the calculation
 694       if(switch1 == 1){
 695         th = thrust[l1];
 696         n01 = psum.Unit();
 697         l1++;
 698       }
 699
 700       if(switch2 == 1){
 701         th02 = thrust02[l2];
 702         n02 = psum02.Unit();
 703         l2++;
 704       }
 705
 706       if(switch3 == 1){
 707         th03 = thrust03[l3];
 708         n03 = psum03.Unit();
 709         l3++;
 710       }
 711
 712       //if thrust values for all starting direction converged check if to the same value
 713       if(switch2 == 0 && switch1 == 0 && switch3 == 0){
 714         if(TMath::Abs(th-th02) < 10e-7 && TMath::Abs(th-th03) < 10e-7 && TMath::Abs(th02-th03) < 10e-7){
 715           eventShapes[0] = th;
 716           AliInfoGeneral(Form(" %s:%d",(char*)__FILE__,__LINE__),Form("===== THRUST VALUE FOUND AT %d :: %f\n", k, th));
 717           break;
 718         }
 719         //if they did not, reset switches
 720         else{
 721           switch1 = 1;
 722           //      th = -1.;
 723           switch2 = 1;
 724           //      th02 = -2.;
 725           switch3 = 1;
 726           //      th03 = -4.;
 727         }
 728       }
 729
 730       //      Printf("========== %d +++ th :: %f=============\n", l1, th);
 731       //      Printf("========== %d +++ th2 :: %f=============\n", l2, th02);
 732       //      Printf("========== %d +++ th3 :: %f=============\n", l3, th03);
 733
 734     }
 735
 736   //if no common limitng value was found, take the maximum and take the corresponding thrust axis
 737   if(switch1 == 1 && switch2 == 1 && switch3 == 1){
 738     eventShapes[0] = TMath::Max(thrust[l1-1], thrust02[l2-1]);
 739     eventShapes[0] = TMath::Max(eventShapes[0], thrust03[l3-1]);
 740     if(TMath::Abs(eventShapes[0]-thrust[l1-1]) < 10e-7)
 741       n01 = n01;
 742     if(TMath::Abs(eventShapes[0]-thrust02[l2-1]) < 10e-7)
 743       n01 = n02;
 744     if(TMath::Abs(eventShapes[0]-thrust03[l3-1]) < 10e-7)
 745       n01 = n03;
 746     Printf("NO LIMITING VALUE FOUND :: MAXIMUM = %f\n", eventShapes[0]);
 747   }
 748
 749 //
 750 //other event shapes variables
 751 //
 752   for(Int_t j = 0; j < nTracks; j++)
 753     {
 754       s00 = s00 + (pTrack[j].Px()*pTrack[j].Px())/pTrack[j].Mag();
 755       s01 = s01 + (pTrack[j].Px()*pTrack[j].Py())/pTrack[j].Mag();
 756       s02 = s02 + (pTrack[j].Px()*pTrack[j].Pz())/pTrack[j].Mag();
 757
 758       s10 = s10 + (pTrack[j].Py()*pTrack[j].Px())/pTrack[j].Mag();
 759       s11 = s11 + (pTrack[j].Py()*pTrack[j].Py())/pTrack[j].Mag();
 760       s12 = s12 + (pTrack[j].Py()*pTrack[j].Pz())/pTrack[j].Mag();
 761
 762       s20 = s20 + (pTrack[j].Pz()*pTrack[j].Px())/pTrack[j].Mag();
 763       s21 = s21 + (pTrack[j].Pz()*pTrack[j].Py())/pTrack[j].Mag();
 764       s22 = s22 + (pTrack[j].Pz()*pTrack[j].Pz())/pTrack[j].Mag();
 765
 766       ptot += pTrack[j].Mag();
 767     }
 768
 769   if(ptot > 0.)
 770     {
 771       m(0,0) = s00/ptot;
 772       m(0,1) = s01/ptot;
 773       m(0,2) = s02/ptot;
 774
 775       m(1,0) = s10/ptot;
 776       m(1,1) = s11/ptot;
 777       m(1,2) = s12/ptot;
 778
 779       m(2,0) = s20/ptot;
 780       m(2,1) = s21/ptot;
 781       m(2,2) = s22/ptot;
 782
 783       TMatrixDSymEigen eigen(m);
 784       TVectorD eigenVal = eigen.GetEigenValues();
 785
 786       Double_t sphericity = (3/2)*(eigenVal(2)+eigenVal(1));
 787       eventShapes[1] = sphericity;
 788
 789       Double_t aplanarity = (3/2)*(eigenVal(2));
 790       eventShapes[2] = aplanarity;
 791
 792       c = 3*(eigenVal(0)*eigenVal(1)+eigenVal(0)*eigenVal(2)+eigenVal(1)*eigenVal(2));
 793       eventShapes[3] = c;
 794     }
 795   return kTRUE;
 796 }
 797
 798
 799
 800  //__________________________________________________________________________________________________________________________
 801 // Trigger Decisions copid from PWG0/AliTriggerAnalysis
 802
 803
 804 Bool_t AliAnalysisHelperJetTasks::IsTriggerFired(const AliVEvent* aEv, Trigger trigger)
 805 {
 806   // checks if an event has been triggered
 807   // no usage of ofline trigger here yet
 808
 809   // here we do a dirty hack to take also into account the
 810   // missing trigger bits and Bunch crossing paatern for real data
 811
 812
 813   if(aEv->InheritsFrom("AliESDEvent")){
 814     const AliESDEvent *esd = (AliESDEvent*)aEv;
 815     switch (trigger)
 816       {
 817       case kAcceptAll:
 818         {
 819           return kTRUE;
 820           break;
 821         }
 822       case kMB1:
 823         {
 824           if(esd->GetFiredTriggerClasses().Contains("CINT1B-"))return kTRUE;
 825           // does the same but without or'ed V0s
 826           if(esd->GetFiredTriggerClasses().Contains("CSMBB"))return kTRUE;
 827           if(esd->GetFiredTriggerClasses().Contains("CINT6B-"))return kTRUE;
 828           // this is for simulated data
 829           if(esd->GetFiredTriggerClasses().Contains("MB1"))return kTRUE;
 830           break;
 831         }
 832       case kMB2:
 833         {
 834           if(esd->GetFiredTriggerClasses().Contains("MB2"))return kTRUE;
 835           break;
 836         }
 837       case kMB3:
 838         {
 839           if(esd->GetFiredTriggerClasses().Contains("MB3"))return kTRUE;
 840           break;
 841         }
 842       case kSPDGFO:
 843         {
 844           if(esd->GetFiredTriggerClasses().Contains("CSMBB"))return kTRUE;
 845           if(esd->GetFiredTriggerClasses().Contains("CINT6B-"))return kTRUE;
 846           if(esd->GetFiredTriggerClasses().Contains("GFO"))return kTRUE;
 847           break;
 848         }
 849       default:
 850         {
 851           Printf("IsEventTriggered: ERROR: Trigger type %d not implemented in this method", (Int_t) trigger);
 852           break;
 853         }
 854       }
 855   }
 856   else if(aEv->InheritsFrom("AliAODEvent")){
 857     const AliAODEvent *aod = (AliAODEvent*)aEv;
 858     switch (trigger)
 859       {
 860       case kAcceptAll:
 861         {
 862           return kTRUE;
 863           break;
 864         }
 865       case kMB1:
 866         {
 867           if(aod->GetFiredTriggerClasses().Contains("CINT1B"))return kTRUE;
 868           // does the same but without or'ed V0s
 869           if(aod->GetFiredTriggerClasses().Contains("CSMBB"))return kTRUE;
 870           // for sim data
 871           if(aod->GetFiredTriggerClasses().Contains("MB1"))return kTRUE;
 872           break;
 873         }
 874       case kMB2:
 875         {
 876           if(aod->GetFiredTriggerClasses().Contains("MB2"))return kTRUE;
 877           break;
 878         }
 879       case kMB3:
 880         {
 881           if(aod->GetFiredTriggerClasses().Contains("MB3"))return kTRUE;
 882           break;
 883         }
 884       case kSPDGFO:
 885         {
 886           if(aod->GetFiredTriggerClasses().Contains("CSMBB"))return kTRUE;
 887           if(aod->GetFiredTriggerClasses().Contains("GFO"))return kTRUE;
 888           break;
 889         }
 890       default:
 891         {
 892           Printf("IsEventTriggered: ERROR: Trigger type %d not implemented in this method", (Int_t) trigger);
 893           break;
 894         }
 895       }
 896   }
 897     return kFALSE;
 898 }
 899
 900
 901  AliAnalysisHelperJetTasks::MCProcessType  AliAnalysisHelperJetTasks::GetPythiaEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug) {
 902
 903   AliGenPythiaEventHeader* pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(aHeader);
 904
 905   if (!pythiaGenHeader) {
 906     //    printf(" AliAnalysisHelperJetTasks::GetProcessType : Unknown gen Header type). \n");
 907     return kInvalidProcess;
 908   }
 909
 910
 911   Int_t pythiaType = pythiaGenHeader->ProcessType();
 912   fgLastProcessType = pythiaType;
 913   MCProcessType globalType = kInvalidProcess;
 914
 915
 916   if (adebug) {
 917     printf(" AliAnalysisHelperJetTasks::GetProcessType : Pythia process type found: %d \n",pythiaType);
 918   }
 919
 920
 921   if(pythiaType==92||pythiaType==93){
 922     globalType = kSD;
 923   }
 924   else if(pythiaType==94){
 925     globalType = kDD;
 926   }
 927   //else if(pythiaType != 91){ // also exclude elastic to be sure... CKB??
 928   else {
 929     globalType = kND;
 930   }
 931   return globalType;
 932 }
 933
 934
 935  AliAnalysisHelperJetTasks::MCProcessType  AliAnalysisHelperJetTasks::GetDPMjetEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug) {
 936   //
 937   // get the process type of the event.
 938   //
 939
 940   // can only read pythia headers, either directly or from cocktalil header
 941   AliGenDPMjetEventHeader* dpmJetGenHeader = dynamic_cast<AliGenDPMjetEventHeader*>(aHeader);
 942
 943   if (!dpmJetGenHeader) {
 944     printf(" AliAnalysisHelperJetTasks::GetDPMjetProcessType : Unknown header type (not DPMjet or). \n");
 945     return kInvalidProcess;
 946   }
 947
 948   Int_t dpmJetType = dpmJetGenHeader->ProcessType();
 949   fgLastProcessType = dpmJetType;
 950   MCProcessType globalType = kInvalidProcess;
 951
 952
 953   if (adebug) {
 954     printf(" AliAnalysisHelperJetTasks::GetDPMJetProcessType : DPMJet process type found: %d \n",dpmJetType);
 955   }
 956
 957
 958   if (dpmJetType == 1 || dpmJetType == 4) { // explicitly inelastic plus central diffraction
 959     globalType = kND;
 960   }
 961   else if (dpmJetType==5 || dpmJetType==6) {
 962     globalType = kSD;
 963   }
 964   else if (dpmJetType==7) {
 965     globalType = kDD;
 966   }
 967   return globalType;
 968 }
 969