PWG1/global/AliAnalysisTaskVertexESD.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2009, 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 //*************************************************************************
  17 // Class AliAnalysisTaskVertexESD
  18 // AliAnalysisTask to extract from ESD the information for the analysis
  19 // of the primary vertex reconstruction efficiency and resolution
  20 // (for MC events) and distributions (for real data). Three vertices:
  21 // - SPD tracklets
  22 // - ITS+TPC tracks
  23 // - TPC-only tracks
  24 //
  25 // Author: A.Dainese, andrea.dainese@pd.infn.it
  26 //*************************************************************************
  27
  28 #include <TChain.h>
  29 #include <TTree.h>
  30 #include <TNtuple.h>
  31 #include <TBranch.h>
  32 #include <TClonesArray.h>
  33 #include <TObjArray.h>
  34 #include <TH1F.h>
  35 #include <TH2F.h>
  36 #include <TCanvas.h>
  37 #include <TGraphAsymmErrors.h>
  38 #include <TFile.h>
  39
  40 #include "AliAnalysisTask.h"
  41 #include "AliAnalysisManager.h"
  42
  43 #include "AliMultiplicity.h"
  44 #include "AliVertexerTracks.h"
  45 #include "AliESDtrack.h"
  46 #include "AliExternalTrackParam.h"
  47 #include "AliESDVertex.h"
  48 #include "AliESDEvent.h"
  49 #include "AliESDInputHandler.h"
  50 #include "AliTrackReference.h"
  51 //#include "AliTriggerAnalysis.h"
  52
  53 #include "AliMCEventHandler.h"
  54 #include "AliMCEvent.h"
  55 #include "AliStack.h"
  56 #include "AliLog.h"
  57
  58 #include "AliGenEventHeader.h"
  59 #include "AliAnalysisTaskVertexESD.h"
  60
  61
  62 ClassImp(AliAnalysisTaskVertexESD)
  63
  64 //________________________________________________________________________
  65 AliAnalysisTaskVertexESD::AliAnalysisTaskVertexESD(const char *name) :
  66 AliAnalysisTaskSE(name),
  67   fCheckEventType(kTRUE),
  68   fReadMC(kFALSE),
  69   fRecoVtxTPC(kFALSE),
  70   fRecoVtxITSTPC(kTRUE),
  71   fRecoVtxITSTPCHalfEvent(kFALSE),
  72   fOnlyITSTPCTracks(kFALSE),
  73   fOnlyITSSATracks(kFALSE),
  74   fFillNtuple(kFALSE),
  75   fFillTreeBeamSpot(kFALSE),
  76   fESD(0),
  77   fOutput(0),
  78   fNtupleVertexESD(0),
  79   fhSPDVertexX(0),
  80   fhSPDVertexY(0),
  81   fhSPDVertexZ(0),
  82   fhTRKVertexX(0),
  83   fhTRKVertexY(0),
  84   fhTRKVertexZ(0),
  85   fhTPCVertexX(0),
  86   fhTPCVertexY(0),
  87   fhTPCVertexZ(0),
  88   fhTrackRefs(0),
  89   fTreeBeamSpot(0),
  90   fhTriggeredTrklets(0),
  91   fhSPD3DTrklets(0),
  92   fhSPDZTrklets(0),
  93   fhTRKTrklets(0),
  94   fhTRKcTrklets(0),
  95   fhTRKncTrklets(0),
  96   fhSPD3DZreco(0),
  97   fhSPDZZreco(0),
  98   fhSPDVertexXPile(0),
  99   fhSPDVertexYPile(0),
 100   fhSPDVertexZPile(0),
 101   fhSPDVertexDiffZPile(0),
 102   fhSPDContributorsPile(0),
 103   fhSPDDispContributors(0)
 104 {
 105   // Constructor
 106
 107   // Define input and output slots here
 108   // Output slot #0 writes into a TList container
 109   DefineOutput(1, TList::Class());  //My private output
 110 }
 111 //________________________________________________________________________
 112 AliAnalysisTaskVertexESD::~AliAnalysisTaskVertexESD()
 113 {
 114   // Destructor
 115
 116   // histograms are in the output list and deleted when the output
 117   // list is deleted by the TSelector dtor
 118
 119   if (fOutput) {
 120     delete fOutput;
 121     fOutput = 0;
 122   }
 123 }
 124
 125
 126 //________________________________________________________________________
 127 void AliAnalysisTaskVertexESD::UserCreateOutputObjects()
 128 {
 129   // Create histograms
 130   // Called once
 131
 132   // Several histograms are more conveniently managed in a TList
 133   fOutput = new TList;
 134   fOutput->SetOwner();
 135
 136   fNtupleVertexESD = new TNtuple("fNtupleVertexESD","vertices","run:tstamp:bunchcross:triggered:dndygen:xtrue:ytrue:ztrue:xSPD:xerrSPD:ySPD:yerrSPD:zSPD:zerrSPD:ntrksSPD:SPD3D:dphiSPD:xTPC:xerrTPC:yTPC:yerrTPC:zTPC:zerrTPC:ntrksTPC:constrTPC:xTRK:xerrTRK:yTRK:yerrTRK:zTRK:zerrTRK:ntrksTRK:constrTRK:ntrklets:nESDtracks:nITSrefit5or6:nTPCin:nTPCinEta09:SPD0cls:xSPDp:xerrSPDp:ySPDp:yerrSPDp:zSPDp:zerrSPDp:ntrksSPDp:xTPCnc:xerrTPCnc:yTPCnc:yerrTPCnc:zTPCnc:zerrTPCnc:ntrksTPCnc:xTPCc:xerrTPCc:yTPCc:yerrTPCc:zTPCc:zerrTPCc:ntrksTPCc:xTRKnc:xerrTRKnc:yTRKnc:yerrTRKnc:zTRKnc:zerrTRKnc:ntrksTRKnc:xTRKc:xerrTRKc:yTRKc:yerrTRKc:zTRKc:zerrTRKc:ntrksTRKc:deltaxTRKnc:deltayTRKnc:deltazTRKnc:deltaxerrTRKnc:deltayerrTRKnc:deltazerrTRKnc:ntrksEvenTRKnc:ntrksOddTRKnc");
 137
 138   fOutput->Add(fNtupleVertexESD);
 139
 140   fhSPDVertexX = new TH1F("fhSPDVertexX","SPDVertex x; x vertex [cm]; events",200,-1,1);
 141   fOutput->Add(fhSPDVertexX);
 142   fhSPDVertexY = new TH1F("fhSPDVertexY","SPDVertex y; y vertex [cm]; events",200,-1,1);
 143   fOutput->Add(fhSPDVertexY);
 144   fhSPDVertexZ = new TH1F("fhSPDVertexZ","SPDVertex z; z vertex [cm]; events",200,-20,20);
 145   fOutput->Add(fhSPDVertexZ);
 146   fhTRKVertexX = new TH1F("fhTRKVertexX","TRKVertex x; x vertex [cm]; events",200,-1,1);
 147   fOutput->Add(fhTRKVertexX);
 148   fhTRKVertexY = new TH1F("fhTRKVertexY","TRKVertex y; y vertex [cm]; events",200,-1,1);
 149   fOutput->Add(fhTRKVertexY);
 150   fhTRKVertexZ = new TH1F("fhTRKVertexZ","TRKVertex z; z vertex [cm]; events",200,-20,20);
 151   fOutput->Add(fhTRKVertexZ);
 152   fhTPCVertexX = new TH1F("fhTPCVertexX","TPCVertex x; x vertex [cm]; events",200,-3,3);
 153   fOutput->Add(fhTPCVertexX);
 154   fhTPCVertexY = new TH1F("fhTPCVertexY","TPCVertex y; y vertex [cm]; events",200,-3,3);
 155   fOutput->Add(fhTPCVertexY);
 156   fhTPCVertexZ = new TH1F("fhTPCVertexZ","TPCVertex z; z vertex [cm]; events",200,-20,20);
 157   fOutput->Add(fhTPCVertexZ);
 158
 159
 160   fhSPDVertexXPile = new TH1F("fhSPDVertexXPile","SPDVertexPile x; x vertex [cm]; events",200,-20,20);
 161   fOutput->Add(fhSPDVertexXPile);
 162   fhSPDVertexYPile = new TH1F("fhSPDVertexYPile","SPDVertexPile y; y vertex [cm]; events",200,-20,20);
 163   fOutput->Add(fhSPDVertexYPile);
 164   fhSPDVertexZPile = new TH1F("fhSPDVertexZPile","SPDVertexPile z; z vertex [cm]; events",200,-40,40);
 165   fOutput->Add(fhSPDVertexZPile);
 166   fhSPDVertexDiffZPile = new TH1F("fhSPDVertexDiffZPile","SPDVertexDiff z; zmain - zdiff [cm]; events",200,-80,80);
 167   fOutput->Add(fhSPDVertexDiffZPile);
 168
 169   fhTrackRefs = new TH2F("fhTrackRefs","Track references; x; y",1000,-4,4,1000,-4,4);
 170   fOutput->Add(fhTrackRefs);
 171
 172   fTreeBeamSpot = new TTree("fTreeBeamSpot", "BeamSpotTree");
 173   UShort_t triggered, ntrkletsS, ntrksTRKnc;
 174   UInt_t run, bx;
 175   Float_t cetTimeLHC,xTRKnc, yTRKnc, zTRKnc;
 176   fTreeBeamSpot->Branch("run", &run, "run/i");
 177   fTreeBeamSpot->Branch("cetTimeLHC", &cetTimeLHC, "cetTimeLHC/F");
 178   fTreeBeamSpot->Branch("bx", &bx, "bx/i");
 179   fTreeBeamSpot->Branch("triggered", &triggered, "triggered/s");
 180   fTreeBeamSpot->Branch("ntrkletsS", &ntrkletsS, "ntrkletsS/s");
 181   fTreeBeamSpot->Branch("xTRKnc", &xTRKnc, "xTRKnc/F");
 182   fTreeBeamSpot->Branch("yTRKnc", &yTRKnc, "yTRKnc/F");
 183   fTreeBeamSpot->Branch("zTRKnc", &zTRKnc, "zTRKnc/F");
 184   fTreeBeamSpot->Branch("ntrksTRKnc", &ntrksTRKnc, "ntrksTRKnc/s");
 185   fOutput->Add(fTreeBeamSpot);
 186
 187   Int_t nbinTrklets=11;
 188   Float_t lowTrklets[12]={-0.5,0.5,1.5,2.5,3.5,4.5,5.5,7.5,10.5,50.5,100.5,200.};
 189   fhTriggeredTrklets = new TH1F("fhTriggeredTrklets","trklets dist for triggered ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 190   fOutput->Add(fhTriggeredTrklets);
 191   fhSPD3DTrklets = new TH1F("fhSPD3DTrklets","trklets dist for SPD3D ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 192   fOutput->Add(fhSPD3DTrklets);
 193   fhSPDZTrklets = new TH1F("fhSPDZTrklets","trklets dist for SPDZ ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 194   fOutput->Add(fhSPDZTrklets);
 195   fhTRKTrklets = new TH1F("fhTRKTrklets","trklets dist for TRK ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 196   fOutput->Add(fhTRKTrklets);
 197   fhTRKcTrklets = new TH1F("fhTRKcTrklets","trklets dist for TRKc ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 198   fOutput->Add(fhTRKcTrklets);
 199   fhTRKncTrklets = new TH1F("fhTRKncTrklets","trklets dist for TRKnc ev.; ntrklets; entries",nbinTrklets,lowTrklets);
 200   fOutput->Add(fhTRKncTrklets);
 201
 202   Int_t nbinZreco = 16;
 203   Double_t lowZreco[17]={-15.0,-10.0,-7.0,-5,-4,-3,-2,-1,0,1,2,3,4,5,7,10,15};
 204   fhSPD3DZreco = new TH1F("fhSPD3DZreco","Zreco dist for SPD3D ev.; Zreco [cm]; entries",nbinZreco,lowZreco);
 205   fOutput->Add(fhSPD3DZreco);
 206   fhSPDZZreco = new TH1F("fhSPDZZreco","Zreco dist for SPDZ ev.; Zreco [cm]; entries",nbinZreco,lowZreco);
 207   fOutput->Add(fhSPDZZreco);
 208
 209   fhSPDContributorsPile = new TH1F("fhSPDContributorsPile","ncontributors pile up vertex; ncontributors; entries",200,-0.5,199.5);
 210   fOutput->Add(fhSPDContributorsPile);
 211
 212   fhSPDDispContributors = new TH2F("fhSPDDispContributors","ncontributors main-pile; ncontributors main; ncontributors pile",200,-0.5,199.5,200,-0.5,199.5);
 213   fOutput->Add(fhSPDDispContributors);
 214
 215   PostData(1, fOutput);
 216
 217   return;
 218 }
 219
 220 //________________________________________________________________________
 221 void AliAnalysisTaskVertexESD::UserExec(Option_t *)
 222 {
 223   // Main loop
 224   // Called for each event
 225
 226   if (!InputEvent()) {
 227     Printf("ERROR: fESD not available");
 228     return;
 229   }
 230   AliESDEvent* esdE = (AliESDEvent*) InputEvent();
 231
 232   // Select PHYSICS events (type=7, for data) and MC events (type=0)
 233   // fCheckEventType is kFALSE if fReadMC is kTRUE, hence check is skipped
 234   if(fCheckEventType) {
 235     if(esdE->GetEventType()!=7 && esdE->GetEventType()!=0) return;
 236   }
 237
 238
 239   TArrayF mcVertex(3);
 240   mcVertex[0]=9999.; mcVertex[1]=9999.; mcVertex[2]=9999.;
 241   Float_t dNchdy=-999.;
 242   // ***********  MC info ***************
 243   if (fReadMC) {
 244     AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
 245     if (!eventHandler) {
 246       Printf("ERROR: Could not retrieve MC event handler");
 247       return;
 248     }
 249
 250     AliMCEvent* mcEvent = eventHandler->MCEvent();
 251     if (!mcEvent) {
 252       Printf("ERROR: Could not retrieve MC event");
 253       return;
 254     }
 255
 256     AliStack* stack = mcEvent->Stack();
 257     if (!stack) {
 258       AliDebug(AliLog::kError, "Stack not available");
 259       return;
 260     }
 261
 262     AliHeader* header = mcEvent->Header();
 263     if (!header) {
 264       AliDebug(AliLog::kError, "Header not available");
 265       return;
 266     }
 267     AliGenEventHeader* genHeader = header->GenEventHeader();
 268     genHeader->PrimaryVertex(mcVertex);
 269
 270     Int_t ngenpart = (Int_t)stack->GetNtrack();
 271     //printf("# generated particles = %d\n",ngenpart);
 272     dNchdy=0;
 273     for(Int_t ip=0; ip<ngenpart; ip++) {
 274       TParticle* part = (TParticle*)stack->Particle(ip);
 275       // keep only electorns, muons, pions, kaons and protons
 276       Int_t apdg = TMath::Abs(part->GetPdgCode());
 277       if(apdg!=11 && apdg!=13 && apdg!=211 && apdg!=321 && apdg!=2212) continue;
 278       // reject secondaries
 279       if(TMath::Sqrt((part->Vx()-mcVertex[0])*(part->Vx()-mcVertex[0])+(part->Vy()-mcVertex[1])*(part->Vy()-mcVertex[1]))>0.0010) continue;
 280       // reject incoming protons
 281       Double_t energy  = part->Energy();
 282       if(energy>900.) continue;
 283       Double_t pz = part->Pz();
 284       Double_t y = 0.5*TMath::Log((energy+pz+1.e-13)/(energy-pz+1.e-13));
 285       if(TMath::Abs(y)<1.0) dNchdy += 0.5; // count 1/2 of particles in |y|<1
 286       // tracks refs
 287       TClonesArray *trefs=0;
 288       Int_t ntrefs = mcEvent->GetParticleAndTR(ip,part,trefs);
 289       if(ntrefs<0) continue;
 290       for(Int_t iref=0; iref<ntrefs; iref++) {
 291         AliTrackReference *tref = (AliTrackReference*)trefs->At(iref);
 292         if(tref->R()>10.) continue;
 293         fhTrackRefs->Fill(tref->X(),tref->Y());
 294       }
 295     }
 296     //printf("# primary particles = %7.1f\n",dNchdy);
 297   }
 298   // ***********  MC info ***************
 299
 300
 301   // Trigger
 302   //ULong64_t triggerMask;
 303   //ULong64_t spdFO = (1 << 14);
 304   //ULong64_t v0left = (1 << 10);
 305   //ULong64_t v0right = (1 << 11);
 306
 307   //triggerMask=esdE->GetTriggerMask();
 308   // MB1: SPDFO || V0L || V0R
 309   //Bool_t eventTriggered = (triggerMask & spdFO || ((triggerMask & v0left) || (triggerMask & v0right)));
 310   //MB2: GFO && V0R
 311   //triggerMask & spdFO && ((triggerMask&v0left) || (triggerMask&v0right))
 312   //Bool_t eventTriggered = (triggerMask & spdFO);
 313
 314   //static AliTriggerAnalysis* triggerAnalysis = new AliTriggerAnalysis();
 315   Bool_t eventTriggered = 0;//triggerAnalysis->IsTriggerFired(esdE, AliTriggerAnalysis::kSPDGFO /*| AliTriggerAnalysis::kOfflineFlag*/);
 316
 317   // use response of AliPhysicsSelection
 318   eventTriggered = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();
 319
 320
 321   Int_t nInFile = esdE->GetEventNumberInFile();
 322
 323  const AliMultiplicity *alimult = esdE->GetMultiplicity();
 324   Int_t ntrklets=0,spd0cls=0;
 325   if(alimult) {
 326     ntrklets = alimult->GetNumberOfTracklets();
 327
 328     for(Int_t l=0;l<alimult->GetNumberOfTracklets();l++){
 329       if(alimult->GetDeltaPhi(l)<-9998.) ntrklets--;
 330     }
 331     spd0cls = alimult->GetNumberOfSingleClusters()+ntrklets;
 332   }
 333
 334
 335   UShort_t triggered, ntrkletsS, ntrksTRKnc;
 336   UInt_t run, bx;
 337   Float_t cetTimeLHC,xTRKnc, yTRKnc, zTRKnc;
 338   fTreeBeamSpot->SetBranchAddress("run", &run);
 339   fTreeBeamSpot->SetBranchAddress("cetTimeLHC", &cetTimeLHC);
 340   fTreeBeamSpot->SetBranchAddress("bx", &bx);
 341   fTreeBeamSpot->SetBranchAddress("triggered", &triggered);
 342   fTreeBeamSpot->SetBranchAddress("ntrkletsS", &ntrkletsS);
 343   fTreeBeamSpot->SetBranchAddress("xTRKnc", &xTRKnc);
 344   fTreeBeamSpot->SetBranchAddress("yTRKnc", &yTRKnc);
 345   fTreeBeamSpot->SetBranchAddress("zTRKnc", &zTRKnc);
 346   fTreeBeamSpot->SetBranchAddress("ntrksTRKnc", &ntrksTRKnc);
 347
 348
 349   Double_t tstamp = esdE->GetTimeStamp();
 350   Float_t cetTime =(tstamp-1262304000.)+7200.;
 351
 352   Float_t cetTime1h =(tstamp-1262304000.)+3600.;
 353
 354   cetTimeLHC = (Float_t)cetTime1h;
 355
 356   Int_t ntracks = esdE->GetNumberOfTracks();
 357   Int_t nITS5or6=0,nTPCin=0,nTPCinEta09=0;
 358   //printf("Tracks # = %d\n",esdE->GetNumberOfTracks());
 359   for(Int_t itr=0; itr<ntracks; itr++) {
 360     AliESDtrack *t = esdE->GetTrack(itr);
 361     if(t->GetNcls(0)>=5) nITS5or6++;
 362     Double_t z0; t->GetZAt(0,esdE->GetMagneticField(),z0);
 363     if(t->GetNcls(1)>0 && TMath::Abs(t->GetD(0,0,esdE->GetMagneticField()))<2.8 && TMath::Abs(z0)<20) {
 364       nTPCin++;
 365       if(TMath::Abs(t->GetTgl())<1.5) nTPCinEta09++;
 366     }
 367   }
 368
 369
 370   const AliESDVertex *spdv=esdE->GetPrimaryVertexSPD();
 371   const AliESDVertex *spdvp=esdE->GetPileupVertexSPD(0);
 372   const AliESDVertex *tpcv=esdE->GetPrimaryVertexTPC();
 373   const AliESDVertex *trkv=esdE->GetPrimaryVertexTracks();
 374
 375   // fill histos
 376
 377   if(spdv) {
 378     if(spdv->GetNContributors()>0) {
 379       TString title=spdv->GetTitle();
 380       if(title.Contains("3D")) {
 381         fhSPDVertexX->Fill(spdv->GetXv());
 382         fhSPDVertexY->Fill(spdv->GetYv());
 383       }
 384       fhSPDVertexZ->Fill(spdv->GetZv());
 385     }
 386   }
 387
 388   if(trkv) {
 389     if(trkv->GetNContributors()>0) {
 390       fhTRKVertexX->Fill(trkv->GetXv());
 391       fhTRKVertexY->Fill(trkv->GetYv());
 392       fhTRKVertexZ->Fill(trkv->GetZv());
 393     }
 394   }
 395
 396   if(tpcv) {
 397     if(tpcv->GetNContributors()>0) {
 398       fhTPCVertexX->Fill(tpcv->GetXv());
 399       fhTPCVertexY->Fill(tpcv->GetYv());
 400       fhTPCVertexZ->Fill(tpcv->GetZv());
 401     }
 402   }
 403
 404
 405
 406    /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 407
 408   // filling Vertex reco efficiency plots
 409   if(eventTriggered ? 1. : 0.){
 410     fhTriggeredTrklets->Fill(ntrklets);
 411     if(spdv->GetNContributors()>0.5){
 412       TString spdtitle = spdv->GetTitle();
 413       if(spdtitle.Contains("vertexer: 3D") ? 1. : 0.){
 414         fhSPD3DTrklets->Fill(ntrklets);
 415         fhSPD3DZreco->Fill(spdv->GetZv());
 416       }
 417       else{
 418         fhSPDZTrklets->Fill(ntrklets);
 419         fhSPDZZreco->Fill(spdv->GetZv());
 420       }
 421     }
 422     if(trkv->GetNContributors()>0.5)fhTRKTrklets->Fill(ntrklets);
 423     if(fRecoVtxITSTPC) {
 424       AliESDVertex *trkvc = ReconstructPrimaryVertexITSTPC(kTRUE);
 425       if(trkvc->GetNContributors()>0.5)fhTRKcTrklets->Fill(ntrklets);
 426       delete trkvc; trkvc=0;
 427       AliESDVertex *trkvnc = ReconstructPrimaryVertexITSTPC(kFALSE);
 428       if(trkvnc->GetNContributors()>0.5)fhTRKncTrklets->Fill(ntrklets);
 429       delete trkvnc; trkvnc=0;
 430     }
 431   }
 432   /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 433
 434   Float_t xpile=-999.;
 435   Float_t ypile=-999.;
 436   Float_t zpile=-999.;
 437   Float_t expile=-999.;
 438   Float_t eypile=-999.;
 439   Float_t ezpile=-999.;
 440   Int_t ntrkspile=-1;
 441
 442   if(esdE->GetNumberOfPileupVerticesSPD()>0 && spdvp){
 443
 444     if(spdvp->GetNContributors()>0) {
 445
 446           fhSPDVertexXPile->Fill(spdvp->GetXv());
 447           fhSPDVertexYPile->Fill(spdvp->GetYv());
 448       fhSPDContributorsPile->Fill(spdvp->GetNContributors());
 449       fhSPDDispContributors->Fill(spdv->GetNContributors(),spdvp->GetNContributors());
 450       fhSPDVertexZPile->Fill(spdvp->GetZv());
 451       fhSPDVertexDiffZPile->Fill(spdv->GetZv()-spdvp->GetZv());
 452     }
 453
 454     xpile=spdvp->GetXv();
 455     expile=spdvp->GetXRes();
 456     ypile=spdvp->GetYv();
 457     eypile=spdvp->GetYRes();
 458     zpile=spdvp->GetZv();
 459     ezpile=spdvp->GetZRes();
 460     ntrkspile=spdvp->GetNContributors();
 461   }
 462
 463   // fill ntuple
 464   Int_t isize=82;
 465   Float_t xnt[82]; for(Int_t iii=0;iii<isize;iii++) xnt[iii]=0.;
 466
 467   Int_t isizeSecNt=9;
 468   Float_t secnt[9]; for(Int_t iii=0;iii<isizeSecNt;iii++) secnt[iii]=0.;
 469
 470   Int_t index=0;
 471   Int_t indexSecNt=0;
 472
 473   xnt[index++]=(Float_t)esdE->GetRunNumber();
 474   secnt[indexSecNt++]=(Float_t)esdE->GetRunNumber();
 475   run = (Int_t)esdE->GetRunNumber();
 476
 477   xnt[index++]=cetTime; //(Float_t)esdE->GetTimeStamp();
 478   //secnt[indexSecNt++]=cetTime;
 479   secnt[indexSecNt++]=cetTime1h;
 480
 481   xnt[index++]=(Float_t)esdE->GetBunchCrossNumber();
 482   secnt[indexSecNt++]=(Float_t)esdE->GetBunchCrossNumber();
 483   bx = (Int_t)esdE->GetBunchCrossNumber();
 484
 485   xnt[index++]=(eventTriggered ? 1. : 0.);
 486   secnt[indexSecNt++]=(eventTriggered ? 1. : 0.);
 487   triggered = (UShort_t)(eventTriggered ? 1 : 0);
 488
 489   xnt[index++]=(Float_t)dNchdy;
 490
 491   xnt[index++]=mcVertex[0];
 492   xnt[index++]=mcVertex[1];
 493   xnt[index++]=mcVertex[2];
 494
 495   xnt[index++]=spdv->GetXv();
 496   xnt[index++]=spdv->GetXRes();
 497   xnt[index++]=spdv->GetYv();
 498   xnt[index++]=spdv->GetYRes();
 499   xnt[index++]=spdv->GetZv();
 500   xnt[index++]=spdv->GetZRes();
 501   xnt[index++]=spdv->GetNContributors();
 502   TString spdtitle = spdv->GetTitle();
 503   xnt[index++]=(spdtitle.Contains("vertexer: 3D") ? 1. : 0.);
 504   xnt[index++]=spdv->GetDispersion();
 505
 506   xnt[index++]=tpcv->GetXv();
 507   xnt[index++]=tpcv->GetXRes();
 508   xnt[index++]=tpcv->GetYv();
 509   xnt[index++]=tpcv->GetYRes();
 510   xnt[index++]=tpcv->GetZv();
 511   xnt[index++]=tpcv->GetZRes();
 512   xnt[index++]=tpcv->GetNContributors();
 513   TString tpctitle = tpcv->GetTitle();
 514   xnt[index++]=(tpctitle.Contains("WithConstraint") ? 1. : 0.);
 515
 516   xnt[index++]=trkv->GetXv();
 517   xnt[index++]=trkv->GetXRes();
 518   xnt[index++]=trkv->GetYv();
 519   xnt[index++]=trkv->GetYRes();
 520   xnt[index++]=trkv->GetZv();
 521   xnt[index++]=trkv->GetZRes();
 522   xnt[index++]=trkv->GetNContributors();// tpccontrorig;
 523   TString trktitle = trkv->GetTitle();
 524   xnt[index++]=(trktitle.Contains("WithConstraint") ? 1. : 0.);
 525
 526   xnt[index++]=float(ntrklets);
 527   secnt[indexSecNt++]=float(ntrklets);
 528   ntrkletsS = (UShort_t)ntrklets;
 529
 530   xnt[index++]=float(ntracks);
 531   xnt[index++]=float(nITS5or6);
 532
 533   xnt[index++]=float(nTPCin);
 534   xnt[index++]=float(nTPCinEta09);
 535
 536   xnt[index++]=spd0cls;
 537
 538   xnt[index++]=xpile;
 539   xnt[index++]=expile;
 540   xnt[index++]=ypile;
 541   xnt[index++]=eypile;
 542   xnt[index++]=zpile;
 543   xnt[index++]=ezpile;
 544   xnt[index++]=(Float_t)ntrkspile;
 545
 546
 547
 548   // add recalculated vertices TRK and TPC
 549
 550   if(fRecoVtxTPC) {
 551     AliESDVertex *tpcvnc = ReconstructPrimaryVertexTPC(kFALSE);
 552     xnt[index++]=tpcvnc->GetXv();
 553     xnt[index++]=tpcvnc->GetXRes();
 554     xnt[index++]=tpcvnc->GetYv();
 555     xnt[index++]=tpcvnc->GetYRes();
 556     xnt[index++]=tpcvnc->GetZv();
 557     xnt[index++]=tpcvnc->GetZRes();
 558     xnt[index++]=tpcvnc->GetNContributors();
 559     delete tpcvnc; tpcvnc=0;
 560
 561     AliESDVertex *tpcvc = ReconstructPrimaryVertexTPC(kTRUE);
 562     xnt[index++]=tpcvc->GetXv();
 563     xnt[index++]=tpcvc->GetXRes();
 564     xnt[index++]=tpcvc->GetYv();
 565     xnt[index++]=tpcvc->GetYRes();
 566     xnt[index++]=tpcvc->GetZv();
 567     xnt[index++]=tpcvc->GetZRes();
 568     xnt[index++]=tpcvc->GetNContributors();
 569     delete tpcvc; tpcvc=0;
 570   } else index+=14;
 571
 572   if(fRecoVtxITSTPC) {
 573     AliESDVertex *trkvnc = ReconstructPrimaryVertexITSTPC(kFALSE);
 574     xnt[index++]=trkvnc->GetXv();
 575     xnt[index++]=trkvnc->GetXRes();
 576     xnt[index++]=trkvnc->GetYv();
 577     xnt[index++]=trkvnc->GetYRes();
 578     xnt[index++]=trkvnc->GetZv();
 579     xnt[index++]=trkvnc->GetZRes();
 580     xnt[index++]=trkvnc->GetNContributors();
 581
 582     secnt[indexSecNt++]=trkvnc->GetXv();
 583     secnt[indexSecNt++]=trkvnc->GetYv();
 584     secnt[indexSecNt++]=trkvnc->GetZv();
 585     secnt[indexSecNt++]=trkvnc->GetNContributors();
 586
 587     xTRKnc = (Float_t)trkvnc->GetXv();
 588     yTRKnc = (Float_t)trkvnc->GetYv();
 589     zTRKnc = (Float_t)trkvnc->GetZv();
 590     ntrksTRKnc = (trkvnc->GetNContributors()<0 ? 0 : (UShort_t)trkvnc->GetNContributors());
 591
 592     delete trkvnc; trkvnc=0;
 593
 594
 595     AliESDVertex *trkvc = ReconstructPrimaryVertexITSTPC(kTRUE);
 596     xnt[index++]=trkvc->GetXv();
 597     xnt[index++]=trkvc->GetXRes();
 598     xnt[index++]=trkvc->GetYv();
 599     xnt[index++]=trkvc->GetYRes();
 600     xnt[index++]=trkvc->GetZv();
 601     xnt[index++]=trkvc->GetZRes();
 602     xnt[index++]=trkvc->GetNContributors();
 603     delete trkvc; trkvc=0;
 604   } else index+=14;
 605
 606   if(fRecoVtxITSTPCHalfEvent) {
 607     AliESDVertex *trkvncodd  = ReconstructPrimaryVertexITSTPC(kFALSE,1);
 608     AliESDVertex *trkvnceven = ReconstructPrimaryVertexITSTPC(kFALSE,2);
 609     if(trkvncodd->GetNContributors()>0 &&
 610        trkvnceven->GetNContributors()>0) {
 611       xnt[index++]=trkvncodd->GetXv()-trkvnceven->GetXv();
 612       xnt[index++]=trkvncodd->GetYv()-trkvnceven->GetYv();
 613       xnt[index++]=trkvncodd->GetZv()-trkvnceven->GetZv();
 614       xnt[index++]=TMath::Sqrt(trkvncodd->GetXRes()*trkvncodd->GetXRes()+trkvnceven->GetXRes()*trkvnceven->GetXRes());
 615       xnt[index++]=TMath::Sqrt(trkvncodd->GetYRes()*trkvncodd->GetYRes()+trkvnceven->GetYRes()*trkvnceven->GetYRes());
 616       xnt[index++]=TMath::Sqrt(trkvncodd->GetZRes()*trkvncodd->GetZRes()+trkvnceven->GetZRes()*trkvnceven->GetZRes());
 617       xnt[index++]=trkvnceven->GetNContributors();
 618       xnt[index++]=trkvncodd->GetNContributors();
 619     } else {
 620       xnt[index++]=0.;
 621       xnt[index++]=0.;
 622       xnt[index++]=0.;
 623       xnt[index++]=0.;
 624       xnt[index++]=0.;
 625       xnt[index++]=0.;
 626       xnt[index++]=-1;
 627       xnt[index++]=-1;
 628     }
 629     delete trkvncodd; trkvncodd=0;
 630     delete trkvnceven; trkvnceven=0;
 631   } else index+=8;
 632
 633
 634   if(index>isize) printf("AliAnalysisTaskVertexESD: ERROR, index!=isize\n");
 635   if(fFillNtuple) fNtupleVertexESD->Fill(xnt);
 636
 637   if(indexSecNt>isizeSecNt) printf("AliAnalysisTaskVertexESD: ERROR, indexSecNt!=isizeSecNt\n");
 638
 639   // if(indexTree>isizeTree) printf("AliAnalysisTaskVertexESD: ERROR, indexTree!=isizeTree\n");
 640   // only every second event (to reduce output size)
 641   if(fFillTreeBeamSpot && (nInFile%2 == 0)) fTreeBeamSpot->Fill();
 642
 643
 644   // Post the data already here
 645   PostData(1, fOutput);
 646
 647   return;
 648 }
 649
 650 //________________________________________________________________________
 651 void AliAnalysisTaskVertexESD::Terminate(Option_t *)
 652 {
 653   // Draw result to the screen
 654   // Called once at the end of the query
 655   fOutput = dynamic_cast<TList*> (GetOutputData(1));
 656   if (!fOutput) {
 657     Printf("ERROR: fOutput not available");
 658     return;
 659   }
 660
 661   //////////////////////////////////////////////////////
 662   /*
 663   TH1F *fhTriggeredTrklets=(TH1F*)fOutput->FindObject("fhTriggeredTrklets");
 664   TH1F *fhSPDZTrklets=(TH1F*)fOutput->FindObject("fhSPDZTrklets");
 665   TH1F *fhSPD3DTrklets=(TH1F*)fOutput->FindObject("fhSPD3DTrklets");
 666   TH1F *fhTRKTrklets=(TH1F*)fOutput->FindObject("fhTRKTrklets");
 667   TH1F *fhTRKcTrklets=(TH1F*)fOutput->FindObject("fhTRKcTrklets");
 668   TH1F *fhTRKncTrklets=(TH1F*)fOutput->FindObject("fhTRKncTrklets");
 669   TH1F *fhSPDZZreco=(TH1F*)fOutput->FindObject("fhSPDZZreco");
 670   TH1F *fhSPD3DZreco=(TH1F*)fOutput->FindObject("fhSPD3DZreco");
 671
 672   TGraphAsymmErrors *fhSPDZEffTrklets=new TGraphAsymmErrors(fhSPDZTrklets,fhTriggeredTrklets,"w");
 673   fhSPDZEffTrklets->SetName("fhSPDZEffTrklets");
 674   fhSPDZEffTrklets->SetDrawOption("AP");
 675   TGraphAsymmErrors *fhSPD3DEffTrklets=new TGraphAsymmErrors(fhSPD3DTrklets,fhTriggeredTrklets,"w");
 676   fhSPD3DEffTrklets->SetName("fhSPD3DEffTrklets");
 677   TH1F * fhSPDOverallTrklets=(TH1F*)fhSPDZTrklets->Clone("fhSPDOverallTrklets");
 678   fhSPDOverallTrklets->Add(fhSPD3DTrklets);
 679   TGraphAsymmErrors *fhSPDOverallEffTrklets=new TGraphAsymmErrors(fhSPDOverallTrklets,fhTriggeredTrklets,"w");
 680   fhSPDOverallEffTrklets->SetName("fhSPDOverallEffTrklets");
 681   TGraphAsymmErrors *fhTRKEffTrklets=new TGraphAsymmErrors(fhTRKTrklets,fhTriggeredTrklets,"w");
 682   fhTRKEffTrklets->SetName("fhTRKEffTrklets");
 683   TGraphAsymmErrors *fhTRKcEffTrklets=new TGraphAsymmErrors(fhTRKcTrklets,fhTriggeredTrklets,"w");
 684   fhTRKcEffTrklets->SetName("fhTRKcEffTrklets");
 685   TGraphAsymmErrors *fhTRKncEffTrklets=new TGraphAsymmErrors(fhTRKncTrklets,fhTriggeredTrklets,"w");
 686   fhTRKncEffTrklets->SetName("fhTRKncEffTrklets");
 687   TH1F * fhSPDOverallZreco=(TH1F*)fhSPDZZreco->Clone("fhSPDOverallZreco");
 688   fhSPDOverallZreco->Add(fhSPD3DZreco);
 689   TGraphAsymmErrors *fhSPDEffZreco=new TGraphAsymmErrors(fhSPD3DZreco,fhSPDOverallZreco,"w");
 690   fhSPDEffZreco->SetName("fhSPDEffZreco");
 691
 692   TH1F *fhEff = new TH1F("hEff","hEff",6,0.5,6.5);
 693   Int_t count=1;
 694   if(fhSPDZTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 695     fhEff->Fill(count,fhSPDZTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 696     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPDZTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 697   }
 698   fhEff->GetXaxis()->SetBinLabel(count,"SPDZ");
 699
 700   count++;
 701   if(fhSPD3DTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 702     fhEff->Fill(count,fhSPD3DTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 703     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPD3DTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 704   }
 705   fhEff->GetXaxis()->SetBinLabel(count,"SPD3D");
 706
 707   count++;
 708   if(fhSPDOverallTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 709     fhEff->Fill(count,fhSPDOverallTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 710     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPDOverallTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 711   }
 712   fhEff->GetXaxis()->SetBinLabel(count,"SPD Overall");
 713
 714   count++;
 715   if(fhTRKTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 716     fhEff->Fill(count,fhTRKTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 717     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 718   }
 719   fhEff->GetXaxis()->SetBinLabel(count,"TRK");
 720
 721   count++;
 722   if(fhTRKcTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 723     fhEff->Fill(count,fhTRKcTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 724     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKcTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 725   }
 726   fhEff->GetXaxis()->SetBinLabel(count,"TRKc");
 727
 728   count++;
 729   if(fhTRKncTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
 730     fhEff->Fill(count,fhTRKncTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
 731     fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKncTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
 732   }
 733   fhEff->GetXaxis()->SetBinLabel(count,"TRKnc");
 734
 735   count++;
 736   fhEff->Print("all");
 737
 738   TFile* fileEff = new TFile("VtxEff.root","recreate");
 739   fhSPDZEffTrklets->Write();
 740   fhSPD3DEffTrklets->Write();
 741   fhSPDOverallEffTrklets->Write();
 742   fhTRKEffTrklets->Write();
 743   fhTRKcEffTrklets->Write();
 744   fhTRKncEffTrklets->Write();
 745   fhSPDEffZreco->Write();
 746   fhEff->Write();
 747   fileEff->Close();
 748   delete fileEff;
 749
 750   /////////////////////////////////////////
 751   */
 752
 753
 754   if (!fNtupleVertexESD){
 755     Printf("ERROR: fNtuple not available");
 756     return;
 757   }
 758
 759   fNtupleVertexESD = dynamic_cast<TNtuple*>(fOutput->FindObject("fNtupleVertexESD"));
 760
 761
 762   return;
 763 }
 764
 765 //_________________________________________________________________________
 766 AliESDVertex* AliAnalysisTaskVertexESD::ReconstructPrimaryVertexTPC(Bool_t constr) const {
 767   // On the fly reco of TPC vertex from ESD
 768   AliESDEvent* evt = (AliESDEvent*) fInputEvent;
 769   AliVertexerTracks vertexer(evt->GetMagneticField());
 770   if(evt->GetNumberOfTracks()<500) {
 771     vertexer.SetTPCMode(); // defaults
 772   } else {
 773     vertexer.SetTPCMode(0.1,1.0,5.0,10,1,3.,0.1,1.5,3.,30.,1,1);// PbPb
 774   }
 775   Float_t diamondcovxy[3]; evt->GetDiamondCovXY(diamondcovxy);
 776   Double_t pos[3]={evt->GetDiamondX(),evt->GetDiamondY(),0};
 777   Double_t cov[6]={diamondcovxy[0],diamondcovxy[1],diamondcovxy[2],0.,0.,10.*10.};
 778   AliESDVertex *initVertex = new AliESDVertex(pos,cov,1.,1);
 779   vertexer.SetVtxStart(initVertex);
 780   delete initVertex;
 781   if(!constr) vertexer.SetConstraintOff();
 782
 783   return vertexer.FindPrimaryVertex(evt);
 784 }
 785
 786 //_________________________________________________________________________
 787 AliESDVertex* AliAnalysisTaskVertexESD::ReconstructPrimaryVertexITSTPC(Bool_t constr,Int_t mode) const {
 788   // On the fly reco of ITS+TPC vertex from ESD
 789   // mode=0 use all tracks
 790   // mode=1 use odd-number tracks
 791   // mode=2 use even-number tracks
 792
 793   AliESDEvent* evt = (AliESDEvent*) fInputEvent;
 794   AliVertexerTracks vertexer(evt->GetMagneticField());
 795   if(evt->GetNumberOfTracks()<500) {
 796     vertexer.SetITSMode(); // defaults
 797     vertexer.SetMinClusters(4); // default is 5
 798   } else {
 799     vertexer.SetITSMode(0.1,0.1,0.5,5,1,3.,100.,1000.,3.,30.,1,1);// PbPb
 800   }
 801   //vertexer.SetITSpureSA();
 802   Float_t diamondcovxy[3]; evt->GetDiamondCovXY(diamondcovxy);
 803   Double_t pos[3]={evt->GetDiamondX(),evt->GetDiamondY(),0};
 804   Double_t cov[6]={diamondcovxy[0],diamondcovxy[1],diamondcovxy[2],0.,0.,10.*10.};
 805   AliESDVertex *initVertex = new AliESDVertex(pos,cov,1.,1);
 806   vertexer.SetVtxStart(initVertex);
 807   delete initVertex;
 808   if(!constr) vertexer.SetConstraintOff();
 809
 810   // use only ITS-TPC or only ITS-SA tracks
 811   if(fOnlyITSTPCTracks || fOnlyITSSATracks || mode>0) {
 812     Int_t iskip=0;
 813     Int_t *skip = new Int_t[evt->GetNumberOfTracks()];
 814     for(Int_t itr=0;itr<evt->GetNumberOfTracks(); itr++) {
 815       AliESDtrack* track = evt->GetTrack(itr);
 816       if(fOnlyITSTPCTracks && track->GetNcls(1)==0) { // skip ITSSA
 817         skip[iskip++]=itr;
 818         continue;
 819       }
 820       if(fOnlyITSSATracks && track->GetNcls(1)>0) { // skip ITSTPC
 821         skip[iskip++]=itr;
 822         continue;
 823       }
 824       if(mode==1 && itr%2==0) skip[iskip++]=itr;
 825       if(mode==2 && itr%2==1) skip[iskip++]=itr;
 826     }
 827     vertexer.SetSkipTracks(iskip,skip);
 828     delete [] skip; skip=NULL;
 829   }
 830
 831   return vertexer.FindPrimaryVertex(evt);
 832 }