PWG2/DIFFRACTIVE/AliAnalysisTaskDDMeson.cxx

   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 // Reconstruct mesons
  17 // in central diffractive event
  18 // in pp collisions
  19 //
  20 // Authors:
  21 //  Xianguo Lu <lu@physi.uni-heidelberg.de>
  22 //
  23
  24 #include <TH1I.h>
  25 #include <TH2I.h>
  26 #include <TH2D.h>
  27 #include <TList.h>
  28 #include <TLorentzVector.h>
  29 #include <TRandom3.h>
  30 #include <THnSparse.h>
  31
  32 #include "AliCDBManager.h"
  33 #include "AliCDBEntry.h"
  34 #include "AliESDInputHandler.h"
  35 #include "AliESDtrackCuts.h"
  36 #include "AliGeomManager.h"
  37 #include "AliITSAlignMille2Module.h"
  38 #include "AliITSsegmentationSPD.h"
  39 #include "AliKFParticle.h"
  40 #include "AliMultiplicity.h"
  41 #include "AliPID.h"
  42 #include "AliSPDUtils.h"
  43 #include "AliTriggerAnalysis.h"
  44
  45 #include "AliAnalysisTaskDDMeson.h"
  46
  47 void AliAnalysisTaskDDMeson::IniTask()
  48 {
  49   //
  50   //initialize values used in many places
  51   //
  52   fnmass = 512;
  53   fmass1 = 0.01 * fnmass;
  54
  55   fnptv = 10;
  56   fptv1 = 0.2 * fnptv;
  57
  58   fnpta = 51*2;
  59   fpta1 = 0.05 * fnpta;
  60
  61   fneta = 64;
  62   feta = 0.04 * fneta/2.;
  63
  64   fnnsel = 2;
  65   fnsel1 = 2+fnnsel;
  66
  67   fncts = 10;
  68   fnctlab = 20;
  69
  70   fCHECKVBA = 2;
  71   fCHECKVBC = 4;
  72
  73   //------------------
  74   fOpt.ToUpper();
  75 }
  76
  77 AliAnalysisTaskDDMeson::AliAnalysisTaskDDMeson(const TString opt):
  78   AliAnalysisTaskSE("DDMeson")
  79   , fOpt(opt)
  80   , fESD(0x0)
  81
  82   , fnmass(-999)
  83   , fmass1(-999)
  84   , fnptv(-999)
  85   , fptv1(-999)
  86   , fnpta(-999)
  87   , fpta1(-999)
  88   , fneta(-999)
  89   , feta(-999)
  90   , fnnsel(-999)
  91   , fnsel1(-999)
  92   , fncts(-999)
  93   , fnctlab(-999)
  94   , fCHECKVBA(-999)
  95   , fCHECKVBC(-999)
  96
  97   , fHBIT(0x0)
  98   , fBitcg(0)
  99
 100   , fRun(-999)
 101   , fat(0x0)
 102   , fct(0x0)
 103   , fbt(0x0)
 104   , fnt(0x0)
 105   , ftt(0x0)
 106
 107   , fv0ntrk(0x0)
 108   , frsntrk(0x0)
 109
 110   , fhps(0x0)
 111   , fhfo(0x0)
 112   , fhspd(0x0)
 113   , fhv0fmd(0x0)
 114   , fhpriv(0x0)
 115   , fhntrk(0x0)
 116
 117   , fHist(0x0)
 118   , fThnMass(0x0)
 119   , fThnDPt(0x0)
 120   , fThnDEta(0x0)
 121   , fThnKF(0x0)
 122 {
 123   //
 124   // Dummy constructor
 125   //
 126   //slot in TaskSE must start from 1
 127   DefineOutput(1, TList::Class());
 128   DefineOutput(2, THnSparse::Class());
 129   DefineOutput(3, THnSparse::Class());
 130   DefineOutput(4, THnSparse::Class());
 131   DefineOutput(5, THnSparse::Class());
 132
 133   IniTask();
 134 }
 135
 136 AliAnalysisTaskDDMeson::AliAnalysisTaskDDMeson(const AliAnalysisTaskDDMeson  &p):
 137   AliAnalysisTaskSE("DDMeson")
 138   , fOpt(p.fOpt)
 139   , fESD(p.fESD)
 140
 141   , fnmass(p.fnmass)
 142   , fmass1(p.fmass1)
 143   , fnptv(p.fnptv)
 144   , fptv1(p.fptv1)
 145   , fnpta(p.fnpta)
 146   , fpta1(p.fpta1)
 147   , fneta(p.fneta)
 148   , feta(p.feta)
 149   , fnnsel(p.fnnsel)
 150   , fnsel1(p.fnsel1)
 151   , fncts(p.fncts)
 152   , fnctlab(p.fnctlab)
 153   , fCHECKVBA(p.fCHECKVBA)
 154   , fCHECKVBC(p.fCHECKVBC)
 155
 156   , fHBIT(p.fHBIT)
 157   , fBitcg(p.fBitcg)
 158
 159   , fRun(p.fRun)
 160   , fat(p.fat)
 161   , fct(p.fct)
 162   , fbt(p.fbt)
 163   , fnt(p.fnt)
 164   , ftt(p.ftt)
 165
 166   , fv0ntrk(p.fv0ntrk)
 167   , frsntrk(p.frsntrk)
 168
 169   , fhps(p.fhps)
 170   , fhfo(p.fhfo)
 171   , fhspd(p.fhspd)
 172   , fhv0fmd(p.fhv0fmd)
 173   , fhpriv(p.fhpriv)
 174   , fhntrk(p.fhntrk)
 175
 176   , fHist(p.fHist)
 177   , fThnMass(p.fThnMass)
 178   , fThnDPt(p.fThnDPt)
 179   , fThnDEta(p.fThnDEta)
 180   , fThnKF(p.fThnKF)
 181 {
 182   //
 183   // Copy constructor
 184   //
 185
 186   IniTask();
 187 }
 188
 189 AliAnalysisTaskDDMeson & AliAnalysisTaskDDMeson::operator=(const AliAnalysisTaskDDMeson  &p)
 190 {
 191   //
 192   // overload =
 193   //
 194   fESD = p.fESD;
 195
 196   fnmass = p.fnmass;
 197   fmass1 = p.fmass1;
 198   fnptv = p.fnptv;
 199   fptv1 = p.fptv1;
 200   fnpta = p.fnpta;
 201   fpta1 = p.fpta1;
 202   fneta = p.fneta;
 203   feta = p.feta;
 204   fnnsel = p.fnnsel;
 205   fnsel1 = p.fnsel1;
 206   fncts = p.fncts;
 207   fnctlab = p.fnctlab;
 208   fCHECKVBA = p.fCHECKVBA;
 209   fCHECKVBC = p.fCHECKVBC;
 210
 211   fHBIT = p.fHBIT;
 212   fBitcg = p.fBitcg;
 213
 214   fRun = p.fRun;
 215
 216   fat = p.fat;
 217   fct = p.fct;
 218   fbt = p.fbt;
 219   fnt = p.fnt;
 220   ftt = p.ftt;
 221
 222   fv0ntrk = p.fv0ntrk;
 223   frsntrk = p.frsntrk;
 224
 225   fhps = p.fhps;
 226   fhfo = p.fhfo;
 227   fhspd = p.fhspd;
 228   fhv0fmd = p.fhv0fmd;
 229   fhpriv = p.fhpriv;
 230   fhntrk = p.fhntrk;
 231
 232   fHist = p.fHist;
 233   fThnMass = p.fThnMass;
 234   fThnDPt = p.fThnDPt;
 235   fThnDEta = p.fThnDEta;
 236   fThnKF = p.fThnKF;
 237
 238   IniTask();
 239
 240   return *this;
 241 }
 242
 243 AliAnalysisTaskDDMeson::~AliAnalysisTaskDDMeson()
 244 {
 245   //
 246   //Destructor
 247   //
 248   if(fESD) delete fESD;
 249
 250   delete fHBIT;
 251   delete fat;
 252   delete fct;
 253   delete fbt;
 254   delete fnt;
 255   delete ftt;
 256
 257   delete fv0ntrk;
 258   delete frsntrk;
 259
 260   delete fhps;
 261   delete fhfo;
 262   delete fhspd;
 263   delete fhv0fmd;
 264   delete fhpriv;
 265   delete fhntrk;
 266
 267   delete fThnMass;
 268   delete fThnDPt;
 269   delete fThnDEta;
 270   delete fThnKF;
 271
 272   fHist->Clear();
 273   delete fHist;
 274
 275 }
 276
 277 void AliAnalysisTaskDDMeson::CheckRange(Double_t &ptv, Double_t &pta, Double_t &etaa
 278                                         , Double_t &mpi
 279                                         ) const
 280 {
 281   //
 282   //save over/under flow bins
 283   //
 284
 285   const Double_t eps = 1e-6;
 286   if(ptv>=fptv1) ptv = fptv1 - eps;
 287   if(pta>=fpta1) pta = fpta1 - eps;
 288
 289   if(etaa >= feta) etaa = feta - eps;
 290
 291   if(etaa <= -feta) etaa = -feta + eps;
 292
 293   if(mpi >= fmass1) mpi = fmass1 - eps;
 294
 295 }
 296
 297 void AliAnalysisTaskDDMeson::UserCreateOutputObjects()
 298 {
 299   //
 300   //CreateOutputObjects
 301   //
 302   //=======================================
 303   const Double_t kvz0=1, kvz1=5;
 304   const Int_t nkvz=(Int_t)(kvz1-kvz0);
 305
 306   const Double_t charge0=1, charge1=4;
 307   const Int_t ncharge=(Int_t)(charge1-charge0);
 308
 309   const Double_t mass0 = 0;
 310   //=======================================
 311   //total momentum and pt in lab frame:
 312   const Double_t p0=0;
 313   //=======================================
 314   const Double_t nsel0 = 2;
 315
 316   //=======================================
 317   const Double_t cts0=0, cts1=1;
 318
 319   //=======================================
 320   const Double_t ctlab0 = -1, ctlab1 = 1;
 321
 322   //=======================================
 323   //------- Mass
 324   const Int_t    nparPtMass = 7;
 325   const Int_t    nbinPtMass[]     ={fnnsel, nkvz, ncharge,  fnmass, fnptv, fncts, fnctlab};
 326   const Double_t xminPtMass[]     ={nsel0,  kvz0, charge0,   mass0,    p0,  cts0,  ctlab0};
 327   const Double_t xmaxPtMass[]     ={fnsel1, kvz1, charge1,  fmass1, fptv1,  cts1,  ctlab1};
 328   fThnMass = new THnSparseD("DDMeson_Mass","nsel, kvz, charge, Mpipi, ptv, cts, ctlab", nparPtMass, nbinPtMass, xminPtMass, xmaxPtMass);
 329
 330   //------- DPt
 331   const Int_t    nparDPt = 4;
 332   const Int_t    nbinDPt[] ={fnnsel, nkvz, ncharge,  fnpta};
 333   const Double_t xminDPt[] ={nsel0, kvz0, charge0,  p0};
 334   const Double_t xmaxDPt[] ={fnsel1, kvz1, charge1,  fpta1};
 335   fThnDPt = new THnSparseD("DDMeson_DPt","nsel, kvz, charge, pt1", nparDPt, nbinDPt, xminDPt, xmaxDPt);
 336   //------- DEta
 337   const Int_t    nparDEta = 4;
 338   const Int_t    nbinDEta[] ={fnnsel, nkvz, ncharge,  fneta};
 339   const Double_t xminDEta[] ={nsel0, kvz0, charge0,  -feta};
 340   const Double_t xmaxDEta[] ={fnsel1, kvz1, charge1,  feta};
 341   fThnDEta = new THnSparseD("DDMeson_DEta"," nsel, kvz, charge, eta1", nparDEta, nbinDEta, xminDEta, xmaxDEta);
 342
 343   //------- KF
 344   const Double_t mks = 0.494;
 345   const Double_t dks = 0.024;
 346   const Int_t nparKF = 4;
 347   const Int_t nbinKF[] = {nkvz, 200, 200, 200};
 348   const Double_t xminKF[] ={kvz0, mks-dks, mks-dks, 0};
 349   const Double_t xmaxKF[] ={kvz1, mks+dks, mks+dks, 10};
 350   fThnKF = new THnSparseD("DDMeson_KF","kvz, rawks, kfks, chi2", nparKF, nbinKF, xminKF, xmaxKF);
 351   //=======================================
 352   //=======================================
 353   fHBIT = new TH1I("HBIT","",32,1,33);
 354
 355   const Int_t runb0 = 114650; //runa1=114649, runb1=117630, runc0=117631, runc1=121526, rund0=121527, rund1=126460, rune0=126461, rune1 = 130930; //runf0=130931, all checked on elog
 356   const Int_t runf1 = 133900;//02OCT2010
 357   const Int_t run0 = runb0-1, run1 = runf1+1;
 358   const Int_t nrun = (Int_t)(run1-run0);
 359
 360   fat = new TH1I("at","",nrun,run0,run1);
 361   fct = new TH1I("ct","",nrun,run0,run1);
 362   fbt = new TH1I("bt","",nrun,run0,run1);
 363   fnt = new TH1I("nt","",nrun,run0,run1);
 364   ftt = new TH1I("tt","",nrun,run0,run1);
 365
 366   fv0ntrk = new TH2D("c_v0ntrk","",80,0,80, 4, 1, 5);//x: ntrk; y: GetV0
 367   frsntrk = new TH2D("c_rsntrk","",1000,0,1000, 200, 0,200);
 368
 369   fhps =    new TH1I("a000_ps",    "", 20, 0,20);
 370   fhfo =    new TH2I("a010_fo",    "", 50, 0, 50, 50, 0, 50);
 371   fhspd =   new TH1I("a011_spd",   "", 50, 0, 50);
 372   fhv0fmd = new TH2I("a020_v0fmd", "", 4, 0, 4, 4, 0, 4);
 373   fhpriv =  new TH1I("a100_priv",  "", 2, 0,2);
 374   fhntrk =  new TH1I("a110_ntrk",  "", (Int_t)fnsel1, 0, (Int_t)fnsel1);
 375
 376   //------
 377
 378   fHist = new TList;
 379
 380   //------>>> Very important!!!!!!!
 381   fHist->SetOwner();
 382
 383   fHist->Add(fat);
 384   fHist->Add(fct);
 385   fHist->Add(fbt);
 386   fHist->Add(fnt);
 387   fHist->Add(ftt);
 388
 389   fHist->Add(fv0ntrk);
 390   fHist->Add(frsntrk);
 391   fHist->Add(fhps);
 392   fHist->Add(fhfo);
 393   fHist->Add(fhspd);
 394   fHist->Add(fhv0fmd);
 395   fHist->Add(fhpriv);
 396   fHist->Add(fhntrk);
 397 }
 398
 399 void AliAnalysisTaskDDMeson::UserExec(Option_t *)
 400 {
 401   //
 402   //Execute
 403   //
 404   //----------------------------------- general protection
 405   if(!CheckESD())
 406     return;
 407
 408   //----------------------------------- trigger bits
 409   if(!CheckBit())
 410     return;
 411
 412   //----------------------------------- track cuts
 413   const Int_t ntrk0 = fESD->GetNumberOfTracks();
 414   const AliESDtrack * trks[ntrk0];
 415   for(Int_t ii=0; ii<ntrk0; ii++){
 416     trks[ii] = 0x0;
 417   }
 418   const Int_t nsel = CutESD(trks);
 419   if(nsel<2)
 420     return;
 421
 422   for(Int_t ii=0; ii<nsel; ii++){
 423     for(Int_t jj=ii+1; jj<nsel; jj++){
 424       const AliESDtrack * trkpair[2]={trks[ii], trks[jj]};
 425
 426       //----------------------------------- tracks initailization
 427       TRandom3 tmprd(0);
 428       if(tmprd.Rndm()>0.5){
 429         SwapTrack(trkpair);
 430       }
 431
 432       //------------------------------------------------------------------------
 433       const Int_t pipdg = 211;
 434       const AliKFParticle daughter0(*(trkpair[0]), pipdg);
 435       const AliKFParticle daughter1(*(trkpair[1]), pipdg);
 436
 437       const AliKFParticle parent(daughter0, daughter1);
 438       const Double_t kfchi2 = parent.GetChi2();
 439       const Double_t kfmass = parent.GetMass();
 440
 441       //------------------------------------------------------------------------
 442       const Int_t kvz = GetV0();
 443
 444       const AliESDtrack * trk1=trkpair[0];
 445       const AliESDtrack * trk2=trkpair[1];
 446
 447       const Double_t ch1 = trk1->GetSign();
 448       const Double_t ch2 = trk2->GetSign();
 449       const Int_t combch = GetCombCh(ch1, ch2);
 450
 451       Double_t pta  = trk1->Pt();
 452       Double_t etaa = trk1->Eta();
 453
 454       Double_t tmpp1[3], tmpp2[3];
 455       trk1->GetPxPyPz(tmpp1);
 456       trk2->GetPxPyPz(tmpp2);
 457
 458       //------------------------------------------------------------------------
 459       //------------------------------------------------------------------------
 460       const Double_t ctlab = GetCtlab(tmpp1, tmpp2);
 461
 462       Double_t cts = -999;
 463       const Double_t masspi = AliPID::ParticleMass(AliPID::kPion);
 464       const TLorentzVector sumv = GetKinematics(tmpp1, tmpp2, masspi, masspi, cts);
 465       Double_t pipi = sumv.Mag();
 466       Double_t ptv = sumv.Pt();
 467
 468       CheckRange(ptv, pta, etaa, pipi);
 469
 470       const Double_t varMass[]  ={nsel, kvz, combch
 471                                        , pipi, ptv, cts
 472                                        , ctlab
 473       };
 474       const Double_t varDPt[]        ={nsel, kvz, combch, pta};
 475       const Double_t varDEta[]       ={nsel, kvz, combch, etaa};
 476       fThnMass->Fill(varMass);
 477       fThnDPt->Fill(varDPt);
 478       fThnDEta->Fill(varDEta);
 479
 480       const Double_t varKF[] = {kvz, pipi, kfmass, kfchi2};
 481       if(nsel==2 && combch==3)
 482         fThnKF->Fill(varKF);
 483     }
 484   }
 485   //------------------------------------------------------------------------
 486   //------------------------------------------------------------------------
 487
 488   PostData(1, fHist);
 489   PostData(2, fThnMass);
 490   PostData(3, fThnDPt);
 491   PostData(4, fThnDEta);
 492   PostData(5, fThnKF);
 493 }
 494 //=======================================================================
 495 //=======================================================================
 496
 497 void AliAnalysisTaskDDMeson::Terminate(Option_t *)
 498 {
 499   //
 500   //fillbit before leaving
 501   //
 502   FillBit();
 503 }
 504
 505 void AliAnalysisTaskDDMeson::FillBit()
 506 {
 507   //
 508   //save the bit count in f*t
 509   //
 510   Double_t tot[5];
 511   CalcBit(fHBIT, tot);
 512
 513   const Int_t bin = fat->FindBin(fRun);
 514   fat->AddBinContent(bin, tot[0]);
 515   fct->AddBinContent(bin, tot[1]);
 516   fbt->AddBinContent(bin, tot[2]);
 517   fnt->AddBinContent(bin, tot[3]);
 518   ftt->AddBinContent(bin, tot[4]);
 519 }
 520
 521 void AliAnalysisTaskDDMeson::CalcBit(TH1I *hc, Double_t tot[]) const
 522 {
 523   //
 524   //calculate the bit count in f*t
 525   //
 526   if(hc->GetBinLowEdge(1)!=1){
 527     printf("xlulog hc defined incorrectly!! %f\n",hc->GetBinLowEdge(1));
 528     return;
 529   }
 530   if(hc->GetBinContent(0)){
 531     printf("\nxlulog hc !!!!!!!!!!!!!!!!!!! underflow!!!!!!!!!!!!!!!!! %f\n\n", hc->GetBinContent(0));
 532     return;
 533   }
 534   if(hc->GetBinContent(hc->GetNbinsX())){
 535     printf("\nxlulog hc !!!!!!!!!!!!!!!!! OVERflow!!!!!!!!!!!!!! %f\n\n", hc->GetBinContent(hc->GetNbinsX()));
 536     return;
 537   }
 538
 539
 540   Double_t atot=0, ctot=0, btot=0, ntot=0;
 541   for(Int_t ib=1; ib<=hc->GetNbinsX(); ib++){
 542     const Double_t cont = hc->GetBinContent(ib);
 543     if(       (ib & fCHECKVBA) && !(ib & fCHECKVBC) ){
 544       atot += cont;
 545     }
 546     else if( !(ib & fCHECKVBA) &&  (ib & fCHECKVBC) ){
 547       ctot += cont;
 548     }
 549     else if(  (ib & fCHECKVBA) &&  (ib & fCHECKVBC) ){
 550       btot += cont;
 551     }
 552     else{
 553       ntot += cont;
 554     }
 555   }
 556
 557   const Double_t ttot   = atot + ctot + btot + ntot;
 558
 559   tot[0]=atot;
 560   tot[1]=ctot;
 561   tot[2]=btot;
 562   tot[3]=ntot;
 563   tot[4]=ttot;
 564
 565   //---
 566   char fullname[5][20];
 567   strncpy(fullname[0],"V0A-ONLY",19);
 568   strncpy(fullname[1],"V0C-ONLY",19);
 569   strncpy(fullname[2],"V0A & V0C",19);
 570   strncpy(fullname[3],"!V0A & !V0C",19);
 571   strncpy(fullname[4],"TOTAL",19);
 572   for (Int_t ii=0;ii<5;ii++){
 573     fullname[ii][19] = '\0';
 574   }
 575
 576   for(Int_t ii=0;ii<5;ii++){
 577     printf("xlulog %s CTP-L0: %s\tTOT: %10.0f\n",fOpt.Data(), fullname[ii],tot[ii]);
 578   }
 579
 580   //---
 581
 582   const Int_t nb=hc->GetNbinsX();
 583   for(Int_t ib=0; ib<=nb+1; ib++){
 584     hc->SetBinContent(ib, 0);
 585   }
 586 }
 587
 588 //====================================================================================
 589
 590 Bool_t AliAnalysisTaskDDMeson::CheckESD()
 591 {
 592   //
 593   //general protection
 594   //
 595   const AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler *>(fInputHandler);
 596   if(esdH)
 597     fESD = esdH->GetEvent();
 598   if(!fESD){
 599     AliError("xlulog No ESD Event");
 600     return 0;
 601   }
 602
 603   if(fabs(fESD->GetMagneticField())<1){
 604     printf("xlulog strange Bfield! %f\n", fESD->GetMagneticField());
 605     return 0;
 606   }
 607
 608   const Int_t tmprun = fESD->GetRunNumber();
 609
 610   if(fRun!=tmprun){
 611     if(fRun>0){
 612       printf("xlulog run changed during exec!! %d %d\n", tmprun, fRun);
 613       FillBit();
 614     }
 615     fRun = tmprun;
 616     if(fOpt.Contains("SPD")){
 617       SPDLoadGeom();
 618     }
 619   }
 620
 621   return 1;
 622 }
 623
 624 void AliAnalysisTaskDDMeson::SPDLoadGeom() const
 625 {
 626   //method to get the gGeomanager
 627   // it is called at the CreatedOutputObject stage
 628   // to comply with the CAF environment
 629   AliCDBManager *man = AliCDBManager::Instance();
 630   TString cdbpath("local:///lustre/alice/alien/alice/data/2010/OCDB");
 631   man->SetDefaultStorage(cdbpath);
 632   man->SetRun(fRun);
 633   man->SetSpecificStorage("ITS/Align/Data",cdbpath);
 634   man->SetSpecificStorage("GRP/Geometry/Data",cdbpath);
 635
 636   AliCDBEntry* obj = man->Get(AliCDBPath("GRP", "Geometry", "Data"));
 637   if (obj != NULL) {
 638     AliGeomManager::SetGeometry((TGeoManager*)obj->GetObject());
 639     AliGeomManager::ApplyAlignObjsFromCDB("ITS");
 640     printf("xlulog DDMeson %s Geom Loaded for run %d !\n", fOpt.Data(), fRun);
 641   }
 642   else {
 643     printf("xlulog DDMeson %s was unable to load Geom for run %d !\n", fOpt.Data(), fRun);
 644   }
 645 }
 646
 647 Bool_t AliAnalysisTaskDDMeson::SPDLoc2Glo(const Int_t id, const Double_t *loc, Double_t *glo) const
 648 {
 649   //
 650   //SPDLoc2Glo
 651   //
 652
 653   static TGeoHMatrix mat;
 654   Int_t vid = AliITSAlignMille2Module::GetVolumeIDFromIndex(id);
 655   if (vid<0) {
 656     printf("xlulog Did not find module with such ID %d\n",id);
 657     return kFALSE;
 658   }
 659   AliITSAlignMille2Module::SensVolMatrix(vid,&mat);
 660   mat.LocalToMaster(loc,glo);
 661   return kTRUE;
 662 }
 663
 664 Bool_t AliAnalysisTaskDDMeson::CheckChipEta(const Int_t chipKey) const
 665 {
 666   //
 667   //CheckChipEta
 668   //
 669
 670   //no eta cut
 671   if(fOpt.Contains("SPD0"))
 672     return kTRUE;
 673
 674   const Double_t etacut = 1.0;
 675
 676   UInt_t module=999, offchip=999;
 677   AliSPDUtils::GetOfflineFromOfflineChipKey(chipKey,module,offchip);
 678   UInt_t hs = AliSPDUtils::GetOnlineHSFromOffline(module);
 679   if(hs<2) offchip = 4 - offchip; // inversion  in the inner layer...
 680
 681   const Int_t col[]={
 682     hs<2? 0 : 31,
 683     hs<2? 31 : 0,
 684     hs<2? 31 : 0,
 685     hs<2? 0 : 31};
 686   const Int_t aa[]={0, 0, 255, 255};
 687   const AliITSsegmentationSPD seg;
 688
 689   for(Int_t ic=0; ic<4; ic++){
 690     Float_t localchip[3]={0.,0.,0.};
 691     seg.DetToLocal(aa[ic],col[ic]+32*offchip,localchip[0],localchip[2]); // local coordinate of the chip center
 692     //printf("local coordinates %d %d: %f %f \n",chipKey, ic, localchip[0],localchip[2]);
 693     const Double_t local[3] = {localchip[0],localchip[1],localchip[2]};
 694     Double_t glochip[3]={0.,0.,0.};
 695     if(!SPDLoc2Glo(module,local,glochip)){
 696       return kFALSE;
 697     }
 698
 699     const TVector3 pos(glochip[0],glochip[1],glochip[2]);
 700     //pos.Print();
 701
 702     if(fabs(pos.Eta()) > etacut)
 703       return kFALSE;
 704   }
 705
 706   return kTRUE;
 707 }
 708
 709 void AliAnalysisTaskDDMeson::GetNFO(Int_t &ni, Int_t &no) const
 710 {
 711   //
 712   //GetNFO
 713   //
 714
 715   const AliMultiplicity *mult = fESD->GetMultiplicity();
 716   Int_t nFOinner=0;
 717   Int_t nFOouter=0;
 718   for(Int_t iChipKey=0; iChipKey < 1200; iChipKey++){
 719     if(mult->TestFastOrFiredChips(iChipKey) && CheckChipEta(iChipKey)){ // here you check if the FastOr bit is 1 or 0
 720       if(iChipKey<400)
 721         nFOinner++;  // here you count the FastOr bits in the inner layer
 722       else
 723         nFOouter++;  // here you count the FastOr bits in the outer layer
 724     }
 725   }
 726
 727   ni = nFOinner;
 728   no = nFOouter;
 729 }
 730
 731 Bool_t AliAnalysisTaskDDMeson::CheckBit()
 732 {
 733   //
 734   //get offline trigger
 735   //
 736   fhps->Fill(1);
 737
 738   AliTriggerAnalysis triggerAnalysis;
 739
 740   //hardware 1: hw 0: offline
 741   const Bool_t khw = kFALSE;
 742   const Bool_t v0A       = (triggerAnalysis.V0Trigger(fESD, AliTriggerAnalysis::kASide, khw) == AliTriggerAnalysis::kV0BB);
 743   const Bool_t v0C       = (triggerAnalysis.V0Trigger(fESD, AliTriggerAnalysis::kCSide, khw) == AliTriggerAnalysis::kV0BB);
 744
 745   /*
 746     const Bool_t v0ABG     = (triggerAnalysis.V0Trigger(fESD, AliTriggerAnalysis::kASide, khw) == AliTriggerAnalysis::kV0BG);
 747     const Bool_t v0CBG     = (triggerAnalysis.V0Trigger(fESD, AliTriggerAnalysis::kCSide, khw) == AliTriggerAnalysis::kV0BG);
 748     const Bool_t v0BG      = v0ABG || v0CBG;
 749     const Int_t fastOROffline = triggerAnalysis.SPDFiredChips(fESD, 0); // SPD number of chips from clusters (!)
 750
 751     //http://alisoft.cern.ch/viewvc/trunk/ANALYSIS/AliPhysicsSelection.cxx?view=markup&root=AliRoot
 752     //Bool_t mb1 = (fastOROffline > 0 || v0A || v0C) && (!v0BG);
 753    */
 754
 755   const Bool_t fmdA = triggerAnalysis.FMDTrigger(fESD, AliTriggerAnalysis::kASide);
 756   const Bool_t fmdC = triggerAnalysis.FMDTrigger(fESD, AliTriggerAnalysis::kCSide);
 757
 758   Bool_t bitA = kFALSE, bitC = kFALSE;
 759   if(fOpt.Contains("V0")){
 760     bitA = bitA || v0A;
 761     bitC = bitC || v0C;
 762   }
 763   if(fOpt.Contains("FMD")){
 764     bitA = bitA || fmdA;
 765     bitC = bitC || fmdC;
 766   }
 767
 768   //---------------------------------------------------------------------------------------------
 769   //---------------------------------------------------------------------------------------------
 770   //considering the efficiency per chip, nfo and fastORhw is required only to have non-0 count!!
 771
 772   //simulate SPD _hardware_ trigger in eta range
 773   if(fOpt.Contains("SPD")){
 774     Int_t ni=-999, no = -999;
 775     GetNFO(ni, no);
 776     if(!bitA && !bitC){
 777       fhfo->Fill(ni, no);
 778     }
 779     if(ni<2 || no<2)
 780       return 0;
 781   }
 782
 783   //simulate MB condition, since for double gap SPD is definitely fired, fastORhw>0 to reduce GA, GC, NG
 784   //http://alisoft.cern.ch/viewvc/trunk/ANALYSIS/AliTriggerAnalysis.cxx?view=markup&root=AliRoot
 785   //Int_t AliTriggerAnalysis::SPDFiredChips(const AliESDEvent* aEsd, Int_t origin, Bool_t fillHists, Int_t layer)
 786   // returns the number of fired chips in the SPD
 787   // origin = 0 --> aEsd->GetMultiplicity()->GetNumberOfFiredChips() (filled from clusters)
 788   // origin = 1 --> aEsd->GetMultiplicity()->TestFastOrFiredChips() (from hardware bits)
 789   const Int_t fastORhw = triggerAnalysis.SPDFiredChips(fESD, 1);
 790   fhspd->Fill(fastORhw);
 791   if(fastORhw<1)
 792     return 0;
 793
 794   //---------------------------------------------------------------------------------------------
 795   //---------------------------------------------------------------------------------------------
 796
 797   const Int_t iv0 = v0A + 2*v0C;
 798   const Int_t ifmd = fmdA + 2*fmdC;
 799   fhv0fmd->Fill(iv0, ifmd);
 800
 801   //---------------------------------------------------------------------------------------------
 802   //---------------------------------------------------------------------------------------------
 803
 804   //need a base number to distringuish from null entry; fHBit starts from 1; important!!
 805   fBitcg=1;
 806
 807   if(bitA)
 808     fBitcg += fCHECKVBA;
 809   if(bitC)
 810     fBitcg += fCHECKVBC;
 811
 812   fHBIT->Fill( fBitcg );
 813
 814   return 1;
 815 }
 816
 817 Int_t AliAnalysisTaskDDMeson::CutESD(const AliESDtrack *outtrk[])
 818 {
 819   //
 820   //track cuts
 821   //
 822
 823   //collision vertex cut
 824   //A cut in XY is implicitly done during the reconstruction by constraining the vertex to the beam diamond.
 825
 826   // Primary vertex
 827   Bool_t kpr0 = kTRUE;
 828   const AliESDVertex *vertex = fESD->GetPrimaryVertexTracks();
 829   if(vertex->GetNContributors()<1) {
 830     // SPD vertex
 831     vertex = fESD->GetPrimaryVertexSPD();
 832     if(vertex->GetNContributors()<1) {
 833       // NO GOOD VERTEX, SKIP EVENT
 834       kpr0 = kFALSE;
 835     }
 836   }
 837   const Bool_t kpriv = kpr0 && ( fabs(fESD->GetPrimaryVertex()->GetZ())<10 );
 838   fhpriv->Fill(kpriv);
 839   if(!kpriv)
 840     return 0;
 841
 842   AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;//::GetStandardITSTPCTrackCuts2010(kTRUE);
 843
 844   //--->
 845   // TPC
 846   esdTrackCuts->SetMinNClustersTPC(70);
 847   esdTrackCuts->SetMaxChi2PerClusterTPC(4);
 848   esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
 849   esdTrackCuts->SetRequireTPCRefit(kTRUE);
 850   // ITS
 851   esdTrackCuts->SetRequireITSRefit(kTRUE);
 852   esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
 853                                          AliESDtrackCuts::kAny);
 854   // 7*(0.0026+0.0050/pt^1.01)
 855   esdTrackCuts->SetMaxDCAToVertexXYPtDep("0.0182+0.0350/pt^1.01");
 856
 857   esdTrackCuts->SetMaxDCAToVertexZ(2);
 858   esdTrackCuts->SetDCAToVertex2D(kFALSE);
 859   esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
 860   //---<
 861
 862   esdTrackCuts->SetEtaRange(-0.9, 0.9);
 863
 864   const TObjArray* seltrk = esdTrackCuts->GetAcceptedTracks(fESD);
 865   const Int_t n0=seltrk->GetEntries();
 866   const AliESDtrack * trks[n0];
 867   Int_t nsel=0;
 868   for(Int_t ii=0; ii<n0; ii++){
 869     trks[ii]=0x0;
 870     const AliESDtrack *esdtrack = (AliESDtrack *)(seltrk->At(ii));
 871
 872     /*
 873     if(!CutTrack(esdtrack))
 874       continue;
 875     */
 876
 877     trks[nsel++]=esdtrack;
 878   }
 879
 880   delete seltrk;
 881   delete esdTrackCuts;
 882
 883   fv0ntrk->Fill(nsel,GetV0());
 884
 885   fhntrk->Fill(nsel);
 886
 887   frsntrk->Fill(fESD->GetNumberOfTracks(), nsel);
 888
 889   if(nsel>=fnsel1)
 890     return 0;
 891
 892   for(Int_t ii=0; ii<nsel; ii++){
 893     outtrk[ii]=trks[ii];
 894   }
 895
 896   return nsel;
 897 }
 898
 899 /*
 900 Bool_t AliAnalysisTaskDDMeson::CutTrack(const AliESDtrack *  esdtrack) const
 901 {
 902   //
 903   //track cut
 904   //
 905
 906   return kTRUE;
 907 }
 908 */
 909
 910 //=================================================================================
 911
 912 void AliAnalysisTaskDDMeson::SwapTrack(const AliESDtrack * trks[]) const
 913 {
 914   //
 915   //swap two esdtracks
 916   //
 917   const AliESDtrack * tmpt = trks[0];
 918   trks[0]=trks[1];
 919   trks[1]=tmpt;
 920 }
 921
 922 Int_t AliAnalysisTaskDDMeson::GetV0() const
 923 {
 924   //
 925   //V0 bit combination
 926   //
 927
 928   const Bool_t ka = (fBitcg & fCHECKVBA);
 929   const Bool_t kc = (fBitcg & fCHECKVBC);
 930
 931
 932   //a1c0 a0c1 must be adjacent so that a cut for signal gas can be eaily applied
 933   const Int_t a0c0 = 1;
 934   const Int_t a1c0 = 2;
 935   const Int_t a0c1 = 3;
 936   const Int_t a1c1 = 4;
 937
 938   if(!ka && !kc)
 939     return a0c0;
 940   else{
 941     if(ka && kc)
 942       return a1c1;
 943     else if(ka)
 944       return a1c0;
 945     else
 946       return a0c1;
 947   }
 948 }
 949
 950 Int_t AliAnalysisTaskDDMeson::GetCombCh(const Double_t s1, const Double_t s2) const
 951 {
 952   //
 953   //get combination of charges
 954   //
 955   const Int_t combPP = 1;
 956   const Int_t combMM = 2;
 957   const Int_t combPM = 3;
 958
 959   if(s1*s2<0){
 960     return combPM;
 961   }
 962   else if(s1>0)
 963     return combPP;
 964   else
 965     return combMM;
 966 }
 967
 968 //==========================================================================
 969
 970 TLorentzVector AliAnalysisTaskDDMeson::GetKinematics(const Double_t *pa, const Double_t *pb, const Double_t ma, const Double_t mb, Double_t & cts) const
 971 {
 972   //
 973   //get kinematics, cts = cos(theta_{#star})
 974   //
 975   TLorentzVector va, vb, sumv;
 976   va.SetXYZM(pa[0], pa[1], pa[2], ma);
 977   vb.SetXYZM(pb[0], pb[1], pb[2], mb);
 978   sumv = va+vb;
 979
 980   const TVector3 bv = -sumv.BoostVector();
 981
 982   va.Boost(bv);
 983   vb.Boost(bv);
 984   //printf("test a %f %f %f -- %f %f\n", va.X(), va.Y(), va.Z(), va.Theta(), va.Phi());
 985   //printf("test b %f %f %f -- %f %f\n", vb.X(), vb.Y(), vb.Z(), vb.Theta(), vb.Phi());
 986
 987   cts = fabs(cos(va.Theta()));
 988
 989   return sumv;
 990 }
 991
 992 Double_t AliAnalysisTaskDDMeson::GetCtlab(const Double_t *pa, const Double_t *pb) const
 993 {
 994   //
 995   //cos(theat_lab)
 996   //
 997
 998   TVector3 va, vb;
 999   va.SetXYZ(pa[0], pa[1], pa[2]);
1000   vb.SetXYZ(pb[0], pb[1], pb[2]);
1001
1002   const TVector3 ua = va.Unit();
1003   const TVector3 ub = vb.Unit();
1004
1005   return ua.Dot(ub);
1006 }