PWGCF/Correlations/DPhi/FourierDecomposition/AliMuonEffMC.cxx

   1 // MUON track QA referring AliMuonEffMC.cxx
   2 // Author : Saehanseul Oh
   3
   4 #include "AliMuonEffMC.h"
   5
   6 #include <TList.h>
   7 #include <TH1D.h>
   8 #include <TH2F.h>
   9 #include <TH3F.h>
  10 #include <THn.h>
  11 #include <TChain.h>
  12 #include <TFile.h>
  13
  14 #include "AliAnalysisManager.h"
  15 #include "AliAnalysisTask.h"
  16 #include "AliESDVertex.h"
  17 #include "AliESDEvent.h"
  18 #include "AliAODEvent.h"
  19 #include "AliESDMuonTrack.h"
  20 #include "AliAODTrack.h"
  21 #include "AliESDVertex.h"
  22 #include "AliAODVertex.h"
  23 #include "AliCentrality.h"
  24 #include "AliVParticle.h"
  25 #include "AliMCParticle.h"
  26 #include "AliAnalysisHelperJetTasks.h"
  27 #include "AliMCEvent.h"
  28 #include "AliAODMCParticle.h"
  29
  30 using std::cout;
  31 using std::endl;
  32
  33 ClassImp(AliMuonEffMC)
  34
  35 //________________________________________________________________________
  36 AliMuonEffMC::AliMuonEffMC() :
  37   AliAnalysisTaskSE(), fESD(0), fAOD(0), fMC(0), fCentrality(99), fZVertex(99), fOutputList(0x0),
  38   fHEventStat(0), fHEvt(0x0), fIsMc(kTRUE), fMDProcess(kTRUE), fCentralityEstimator("V0M"),
  39   fNEtaBins(15), fNpTBins(100), fNCentBins(5), fNZvtxBins(10), fNPhiBins(24),
  40   fHMuonParGen(0x0), fHMuonDetGen(0x0), fHMuonDetRec(0x0), fHEtcDetRec(0x0)
  41 {
  42   // Constructor
  43
  44 }
  45
  46 //________________________________________________________________________
  47 AliMuonEffMC::AliMuonEffMC(const char *name) :
  48   AliAnalysisTaskSE(name), fESD(0), fAOD(0), fMC(0), fCentrality(99), fZVertex(99), fOutputList(0x0),
  49   fHEventStat(0), fHEvt(0x0),  fIsMc(kTRUE), fMDProcess(kTRUE), fCentralityEstimator("V0M"),
  50   fNEtaBins(15), fNpTBins(100), fNCentBins(5), fNZvtxBins(10), fNPhiBins(24),
  51   fHMuonParGen(0x0), fHMuonDetGen(0x0), fHMuonDetRec(0x0), fHEtcDetRec(0x0)
  52 {
  53   // Constructor
  54   DefineInput(0, TChain::Class());
  55   DefineOutput(1, TList::Class());
  56
  57 }
  58
  59 //________________________________________________________________________
  60 AliMuonEffMC::~AliMuonEffMC()
  61 {
  62   //Destructor
  63   if(fOutputList) delete fOutputList;
  64 }
  65
  66 //________________________________________________________________________
  67 void AliMuonEffMC::UserCreateOutputObjects()
  68 {
  69   // Create histograms
  70   // Called once (per slave on PROOF!)
  71
  72   fOutputList = new TList();
  73   fOutputList->SetOwner(1);
  74
  75   for(Int_t i=0; i<4; i++)
  76   {
  77     fHMuMotherGenPt[i] = 0x0;
  78     fHMuMotherRecPt[i] = 0x0;
  79     fHMuMotherGenPhi[i] = 0x0;
  80     fHMuMotherRecPhi[i] = 0x0;
  81     fHMuMotherGenEta[i] = 0x0;
  82     fHMuMotherRecEta[i] = 0x0;
  83     fHMuDCA[i] = 0x0;
  84   }
  85
  86   fHEventStat = new TH1D("fHEventStat","Event statistics for analysis",18,0,18);
  87   fHEventStat->GetXaxis()->SetBinLabel(1,"Event");
  88   fHEventStat->GetXaxis()->SetBinLabel(2,"SelectedEvent");
  89   fHEventStat->GetXaxis()->SetBinLabel(3,"File");
  90   fHEventStat->GetXaxis()->SetBinLabel(4,"fSPHighpt");  //!Global Trigger Single plus High p_T
  91   fHEventStat->GetXaxis()->SetBinLabel(5,"fSPAllpt");   //!Global Trigger Single plus All p_T
  92   fHEventStat->GetXaxis()->SetBinLabel(6,"fSMLowpt");   //!Global Trigger Single minus Low p_T
  93   fHEventStat->GetXaxis()->SetBinLabel(7,"fSMHighpt");  //!Global Trigger Single minus High p_T
  94   fHEventStat->GetXaxis()->SetBinLabel(8,"fSMAllpt");   //!Global Trigger Single minus All p_T
  95   fHEventStat->GetXaxis()->SetBinLabel(9,"fSULowpt");   //!Global Trigger Single undefined Low p_T
  96   fHEventStat->GetXaxis()->SetBinLabel(10,"fSUHighpt"); //!Global Trigger Single undefined High p_T
  97   fHEventStat->GetXaxis()->SetBinLabel(11,"fSUAllpt");  //!Global Trigger Single undefined All p_T
  98   fHEventStat->GetXaxis()->SetBinLabel(12,"fUSLowpt");  //!Global Trigger UnlikeSign pair Low p_T
  99   fHEventStat->GetXaxis()->SetBinLabel(13,"fUSHighpt"); //!Global Trigger UnlikeSign pair High p_T
 100   fHEventStat->GetXaxis()->SetBinLabel(14,"fUSAllpt");  //!Global Trigger UnlikeSign pair All p_T
 101   fHEventStat->GetXaxis()->SetBinLabel(15,"fLSLowpt");  //!Global Trigger LikeSign pair pair Low p_T
 102   fHEventStat->GetXaxis()->SetBinLabel(16,"fLSHighpt"); //!Global Trigger LikeSign pair pair High p_T
 103   fHEventStat->GetXaxis()->SetBinLabel(17,"fLSAllpt");  //!Global Trigger LikeSign pair pair All p_T
 104   fHEventStat->GetXaxis()->SetBinLabel(18,"fSPLowpt");   //!Global Trigger Single plus Low p_T
 105   fOutputList->Add(fHEventStat);
 106
 107   fHEvt = new TH2F("fHEvt", "Event-level variables; Zvtx; Cent", 30, -15, 15, 103, -2, 101);
 108   fOutputList->Add(fHEvt);
 109
 110
 111   Int_t iTrackBin[5];
 112   Double_t* trackBins[5];
 113   const char* trackAxisTitle[5];
 114
 115   // eta
 116   Double_t etaBins[fNEtaBins+1];
 117   for (Int_t i=0; i<=fNEtaBins; i++) { etaBins[i] = (Double_t)(-4.0 + 1.5/fNEtaBins*i); }
 118   iTrackBin[0] = fNEtaBins;
 119   trackBins[0] = etaBins;
 120   trackAxisTitle[0] = "#eta";
 121
 122   // p_T
 123   Double_t pTBins[fNpTBins+1];
 124   for (Int_t i=0; i<=fNpTBins; i++) { pTBins[i] = (Double_t)(10.0/fNpTBins*i); }
 125   iTrackBin[1] = fNpTBins;
 126   trackBins[1] = pTBins;
 127   trackAxisTitle[1] = "p_{T} (GeV/c)";
 128
 129   // centrality
 130   Double_t CentBins[fNCentBins+1];
 131   for (Int_t i=0; i<=fNCentBins; i++) { CentBins[i] = (Double_t)(100.0/fNCentBins*i); }
 132   iTrackBin[2] = fNCentBins;
 133   trackBins[2] = CentBins;
 134   trackAxisTitle[2] = "Cent";
 135
 136   // Z-vertex
 137   Double_t ZvtxBins[fNZvtxBins+1];
 138   for (Int_t i=0; i<=fNZvtxBins; i++) { ZvtxBins[i] = (Double_t)(-10.0 + 20.0/fNZvtxBins*i); }
 139   iTrackBin[3] = fNZvtxBins;
 140   trackBins[3] = ZvtxBins;
 141   trackAxisTitle[3] = "Zvtx";
 142
 143   // phi
 144   Double_t phiBins[fNPhiBins+1];
 145   for (Int_t i=0; i<=fNPhiBins; i++) { phiBins[i] = (Double_t)(TMath::TwoPi()/fNPhiBins * i); }
 146   iTrackBin[4] = fNPhiBins;
 147   trackBins[4] = phiBins;
 148   trackAxisTitle[4] = "#phi";
 149
 150   // THn for tracking efficiency
 151   fHMuonParGen = new THnF("fHMuonParGen", "", 5, iTrackBin, 0, 0);
 152   for (Int_t i=0; i<5; i++)
 153   {
 154     fHMuonParGen->SetBinEdges(i, trackBins[i]);
 155     fHMuonParGen->GetAxis(i)->SetTitle(trackAxisTitle[i]);
 156   }
 157   fHMuonParGen->Sumw2();
 158   fOutputList->Add(fHMuonParGen);
 159
 160   fHMuonDetGen = (THnF*) fHMuonParGen->Clone("fHMuonDetGen");
 161   fHMuonDetGen->Sumw2();
 162   fOutputList->Add(fHMuonDetGen);
 163
 164   fHMuonDetRec = (THnF*) fHMuonParGen->Clone("fHMuonDetRec");
 165   fHMuonDetRec->Sumw2();
 166   fOutputList->Add(fHMuonDetRec);
 167
 168   fHEtcDetRec = (THnF*) fHMuonParGen->Clone("fHEtcDetRec");
 169   fHEtcDetRec->Sumw2();
 170   fOutputList->Add(fHEtcDetRec);
 171
 172   if(fMDProcess)
 173   {
 174     const char* MotherSpecies[4] = {"Pion","Kaon","D", "Etc"};
 175
 176     for(Int_t i=0; i<4; i++)
 177     {
 178       fHMuMotherGenPt[i] = new TH2F(Form("fHMuMotherGenPt_%s",MotherSpecies[i]),";p_{T,muon}^{gen} (GeV/c);p_{T,mother}^{Truth} (GeV/c);",500, 0, 50, 500, 0, 50);
 179       fOutputList->Add(fHMuMotherGenPt[i]);
 180
 181       fHMuMotherRecPt[i] = new TH2F(Form("fHMuMotherRecPt_%s",MotherSpecies[i]),";p_{T,muon}^{rec} (GeV/c);p_{T,mother}^{Truth} (GeV/c);",500, 0, 50, 500, 0, 50);
 182       fOutputList->Add(fHMuMotherRecPt[i]);
 183
 184       fHMuMotherGenPhi[i] = new TH2F(Form("fHMuMotherGenPhi_%s",MotherSpecies[i]),";#phi_{gen};mother #phi;",100, 0, TMath::TwoPi(), 100, 0, TMath::TwoPi());
 185       fOutputList->Add(fHMuMotherGenPhi[i]);
 186
 187       fHMuMotherRecPhi[i] = new TH2F(Form("fHMuMotherRecPhi_%s",MotherSpecies[i]),";#phi_{rec};mother #phi;",100, 0, TMath::TwoPi(), 100, 0, TMath::TwoPi());
 188       fOutputList->Add(fHMuMotherRecPhi[i]);
 189
 190       fHMuMotherGenEta[i] = new TH2F(Form("fHMuMotherGenEta_%s",MotherSpecies[i]),";#eta_{gen};mother #eta;",100, -5., -1., 100, -5., -1.);
 191       fOutputList->Add(fHMuMotherGenEta[i]);
 192
 193       fHMuMotherRecEta[i] = new TH2F(Form("fHMuMotherRecEta_%s",MotherSpecies[i]),";#eta_{rec};mother #eta;",100, -5., -1., 100, -5., -1.);
 194       fOutputList->Add(fHMuMotherRecEta[i]);
 195
 196       fHMuDCA[i] =  new TH1F(Form("fHMuDCA_%s",MotherSpecies[i]), ";DCA", 100, 0, 50);
 197       fOutputList->Add(fHMuDCA[i]);
 198     }
 199   }
 200   PostData(1, fOutputList);
 201 }
 202
 203 //________________________________________________________________________
 204 void AliMuonEffMC::UserExec(Option_t *)
 205 {
 206   // Main loop, Called for each event
 207   Int_t ntrks = 0; // number of tracks in an event
 208
 209   if(((TString)InputEvent()->IsA()->GetName())=="AliAODEvent")
 210   {
 211     fAOD = dynamic_cast<AliAODEvent*>(InputEvent());
 212     if (!fAOD) { AliError("AOD event not found. Nothing done!"); return; }
 213     ntrks = fAOD->GetNTracks();
 214   }
 215   else
 216   {
 217     fESD = dynamic_cast<AliESDEvent*>(InputEvent());
 218     if (!fESD) { AliError("ESD event not found. Nothing done!"); return; }
 219     ntrks = fESD->GetNumberOfMuonTracks();
 220   }
 221
 222   fHEventStat->Fill(0.5);
 223   if(fIsMc)
 224   {
 225     fMC = MCEvent();
 226     if (!fMC) { AliError("MC event not avaliable."); return; }
 227   }
 228
 229   // Centrality, vertex, other event variables...
 230   if(fAOD)
 231   {
 232     const AliAODVertex* vertex = fAOD->GetPrimaryVertex();
 233     fZVertex = vertex->GetZ();
 234     if(fAOD->GetCentrality())  fCentrality = fAOD->GetCentrality()->GetCentralityPercentile(fCentralityEstimator);
 235   }
 236   else if(fESD)
 237   {
 238     const AliESDVertex* vertex = fESD->GetPrimaryVertex();
 239     fZVertex = vertex->GetZ();
 240     if(fESD->GetCentrality()) fCentrality = fESD->GetCentrality()->GetCentralityPercentile(fCentralityEstimator);
 241   }
 242
 243   if ((fESD && !VertexOk(fESD)) || (fAOD && !VertexOk(fAOD))) { AliInfo(Form("Event REJECTED. z = %.1f", fZVertex)); return; }
 244   if (fCentrality > 100. || fCentrality < -1.5) { AliInfo(Form("Event REJECTED. fCentrality = %.1f", fCentrality)); return; }
 245
 246   if(fCentrality < 0) fCentrality = 0.5; //ad hoc centrality for pp
 247   // Fill Event histogram
 248   fHEvt->Fill(fZVertex, fCentrality);
 249   fHEventStat->Fill(1.5);
 250
 251   ULong64_t trigword = 0;
 252   if(fAOD) trigword=fAOD->GetTriggerMask();
 253   else if(fESD) trigword=fESD->GetTriggerMask();
 254
 255   if (trigword & 0x01) fHEventStat->Fill(17.5);
 256   if (trigword & 0x02) fHEventStat->Fill(3.5);
 257   if (trigword & 0x04) fHEventStat->Fill(4.5);
 258   if (trigword & 0x08) fHEventStat->Fill(5.5);
 259   if (trigword & 0x010) fHEventStat->Fill(6.5);
 260   if (trigword & 0x020) fHEventStat->Fill(7.5);
 261   if (trigword & 0x040) fHEventStat->Fill(8.5);
 262   if (trigword & 0x080) fHEventStat->Fill(9.5);
 263   if (trigword & 0x100) fHEventStat->Fill(10.5);
 264   if (trigword & 0x200) fHEventStat->Fill(11.5);
 265   if (trigword & 0x400) fHEventStat->Fill(12.5);
 266   if (trigword & 0x800) fHEventStat->Fill(13.5);
 267   if (trigword & 0x1000) fHEventStat->Fill(14.5);
 268   if (trigword & 0x2000) fHEventStat->Fill(15.5);
 269   if (trigword & 0x4000) fHEventStat->Fill(16.5);
 270
 271   // generated level loop
 272   for (Int_t ipart=0; ipart<fMC->GetNumberOfTracks(); ipart++)
 273   {
 274     if(fAOD)
 275     {
 276       AliAODMCParticle *AodMcParticle  = (AliAODMCParticle*)fMC->GetTrack(ipart);
 277       if(TMath::Abs(AodMcParticle->PdgCode())==13)
 278       {
 279         Double_t fillArrayParGen[5] = { AodMcParticle->Eta(), AodMcParticle->Pt(), fCentrality, fZVertex, AodMcParticle->Phi() };
 280         fHMuonParGen->Fill(fillArrayParGen);
 281       }
 282     }
 283     else if(fESD)
 284     {
 285       AliMCParticle *McParticle  = (AliMCParticle*)fMC->GetTrack(ipart);
 286       if(TMath::Abs(McParticle->PdgCode())==13)
 287       {
 288         Double_t fillArrayParGen[5] = { McParticle->Eta(), McParticle->Pt(), fCentrality, fZVertex, McParticle->Phi() };
 289         fHMuonParGen->Fill(fillArrayParGen);
 290       }
 291     }
 292   }
 293
 294   // reconstructed level loop
 295   for (Int_t iTrack = 0; iTrack<ntrks; iTrack++)
 296   {
 297     Int_t label = 0;
 298     Double_t trackpt = 0;
 299     Double_t tracketa = 0;
 300     Double_t trackphi = 0;
 301     Double_t dcavalue = 0;
 302
 303     Double_t mcpt = 0;
 304     Double_t mceta = 0;
 305     Double_t mcphi = 0;
 306
 307     Int_t motherlabel =0;
 308     Int_t motherpdg = 0;
 309     Double_t motherpt = 0;
 310     Double_t mothereta = 0;
 311     Double_t motherphi = 0;
 312
 313     if(fAOD)
 314     {
 315       AliAODTrack* muonTrack = (AliAODTrack*)fAOD->GetTrack(iTrack);
 316       if(muonTrack)
 317       {
 318         if(!(IsGoodMUONtrack(*muonTrack)) || !(muonTrack->IsMuonTrack())) continue;
 319         trackpt = muonTrack->Pt();
 320         tracketa = muonTrack->Eta();
 321         trackphi = muonTrack->Phi();
 322         label =  TMath::Abs(muonTrack->GetLabel());
 323       }
 324     }
 325     else if(fESD)
 326     {
 327       AliESDMuonTrack* muonTrack = fESD->GetMuonTrack(iTrack);
 328       if(muonTrack)
 329       {
 330         if(!IsGoodMUONtrack(*muonTrack)) continue;
 331         trackpt = muonTrack->Pt();
 332         tracketa = muonTrack->Eta();
 333         trackphi = muonTrack->Phi();
 334         label =  TMath::Abs(muonTrack->GetLabel());
 335         dcavalue = muonTrack->GetDCA();
 336       }
 337     }
 338
 339     Double_t fillArrayDetRec[5] = { tracketa, trackpt, fCentrality, fZVertex, trackphi };
 340     fHMuonDetRec->Fill(fillArrayDetRec);
 341
 342     if(fAOD)
 343     {
 344       AliAODMCParticle *aodMcParticle  = (AliAODMCParticle*)fMC->GetTrack(label);
 345       if(TMath::Abs(aodMcParticle->PdgCode()) != 13)
 346       {
 347         fHEtcDetRec->Fill(fillArrayDetRec);
 348         continue;
 349       }
 350       mcpt = aodMcParticle->Pt();
 351       mceta = aodMcParticle->Eta();
 352       mcphi = aodMcParticle->Phi();
 353       motherlabel = aodMcParticle->GetMother();
 354       if(motherlabel > 0)
 355       {
 356         AliAODMCParticle *aodMotherParticle  = (AliAODMCParticle*)fMC->GetTrack(motherlabel);
 357         motherpdg = TMath::Abs(aodMotherParticle->PdgCode());
 358         motherpt = aodMotherParticle->Pt();
 359         mothereta = aodMotherParticle->Eta();
 360         motherphi = aodMotherParticle->Phi();
 361       }
 362     }
 363     else if(fESD)
 364     {
 365       AliMCParticle *McParticle  = (AliMCParticle*)fMC->GetTrack(label);
 366       if(TMath::Abs(McParticle->PdgCode()) != 13)
 367       {
 368         fHEtcDetRec->Fill(fillArrayDetRec);
 369         continue;
 370       }
 371       mcpt = McParticle->Pt();
 372       mceta = McParticle->Eta();
 373       mcphi = McParticle->Phi();
 374       motherlabel = McParticle->GetMother();
 375       if(motherlabel > 0)
 376       {
 377         AliMCParticle *MotherParticle  = (AliMCParticle*)fMC->GetTrack(motherlabel);
 378         motherpdg = TMath::Abs(MotherParticle->PdgCode());
 379         motherpt = MotherParticle->Pt();
 380         mothereta = MotherParticle->Eta();
 381         motherphi = MotherParticle->Phi();
 382       }
 383     }
 384     Double_t fillArrayDetGen[5] = { mceta, mcpt, fCentrality, fZVertex, mcphi };
 385     fHMuonDetGen->Fill(fillArrayDetGen);
 386
 387     // mother-daughter kinematic relation
 388     if(fMDProcess)
 389     {
 390       if(motherlabel > 0)
 391       {
 392         if(motherpdg==411 || motherpdg==413 || motherpdg==421 || motherpdg==423 || motherpdg==431 || motherpdg==433 || motherpdg==10413 || motherpdg==10411 || motherpdg==10423 || motherpdg==10421 || motherpdg==10433 || motherpdg==10431 || motherpdg==20413 || motherpdg==415 || motherpdg==20423 || motherpdg==425 || motherpdg==20433 || motherpdg==435)
 393         {
 394           fHMuMotherGenPt[2]->Fill(mcpt, motherpt);
 395           fHMuMotherRecPt[2]->Fill(trackpt, motherpt);
 396           fHMuMotherGenPhi[2]->Fill(mcphi, motherphi);
 397           fHMuMotherRecPhi[2]->Fill(trackphi, motherphi);
 398           fHMuMotherGenEta[2]->Fill(mceta, mothereta);
 399           fHMuMotherRecEta[2]->Fill(tracketa, mothereta);
 400           if(fESD) fHMuDCA[2]->Fill(dcavalue);
 401         }
 402         else if(motherpdg==211)
 403         {
 404           fHMuMotherGenPt[0]->Fill(mcpt, motherpt);
 405           fHMuMotherRecPt[0]->Fill(trackpt, motherpt);
 406           fHMuMotherGenPhi[0]->Fill(mcphi, motherphi);
 407           fHMuMotherRecPhi[0]->Fill(trackphi, motherphi);
 408           fHMuMotherGenEta[0]->Fill(mceta, mothereta);
 409           fHMuMotherRecEta[0]->Fill(tracketa, mothereta);
 410           if(fESD) fHMuDCA[0]->Fill(dcavalue);
 411         }
 412         else if(motherpdg==321)
 413         {
 414           fHMuMotherGenPt[1]->Fill(mcpt, motherpt);
 415           fHMuMotherRecPt[1]->Fill(trackpt, motherpt);
 416           fHMuMotherGenPhi[1]->Fill(mcphi, motherphi);
 417           fHMuMotherRecPhi[1]->Fill(trackphi, motherphi);
 418           fHMuMotherGenEta[1]->Fill(mceta, mothereta);
 419           fHMuMotherRecEta[1]->Fill(tracketa, mothereta);
 420           if(fESD) fHMuDCA[1]->Fill(dcavalue);
 421         }
 422         else
 423         {
 424           fHMuMotherGenPt[3]->Fill(mcpt, motherpt);
 425           fHMuMotherRecPt[3]->Fill(trackpt, motherpt);
 426           fHMuMotherGenPhi[3]->Fill(mcphi, motherphi);
 427           fHMuMotherRecPhi[3]->Fill(trackphi, motherphi);
 428           fHMuMotherGenEta[3]->Fill(mceta, mothereta);
 429           fHMuMotherRecEta[3]->Fill(tracketa, mothereta);
 430           if(fESD) fHMuDCA[3]->Fill(dcavalue);
 431         }
 432       }
 433     }    // (mother hadron) : (daughter muon) QA */
 434   }
 435
 436   PostData(1, fOutputList);
 437   return;
 438 }
 439
 440 //________________________________________________________________________
 441 void AliMuonEffMC::Terminate(Option_t *)
 442 {
 443   // Draw result to the screen
 444   // Called once at the end of the query
 445
 446 }
 447
 448 //________________________________________________________________________
 449 Bool_t AliMuonEffMC::VertexOk(TObject* obj) const
 450 {
 451   // Modified from AliAnalyseLeadingTrackUE::VertexSelection()
 452
 453   Int_t nContributors  = 0;
 454   Double_t zVertex     = 999;
 455   TString name("");
 456
 457   if (obj->InheritsFrom("AliESDEvent")) {
 458     AliESDEvent* esdevt = (AliESDEvent*) obj;
 459     const AliESDVertex* vtx = esdevt->GetPrimaryVertex();
 460     if (!vtx)
 461       return 0;
 462     nContributors = vtx->GetNContributors();
 463     zVertex       = vtx->GetZ();
 464     name          = vtx->GetName();
 465   }
 466   else if (obj->InheritsFrom("AliAODEvent")) {
 467     AliAODEvent* aodevt = (AliAODEvent*) obj;
 468     if (aodevt->GetNumberOfVertices() < 1)
 469       return 0;
 470     const AliAODVertex* vtx = aodevt->GetPrimaryVertex();
 471     nContributors = vtx->GetNContributors();
 472     zVertex       = vtx->GetZ();
 473     name          = vtx->GetName();
 474   }
 475
 476   // Reject if TPC-only vertex
 477   if (name.CompareTo("TPCVertex")==0)
 478     return kFALSE;
 479
 480   // Check # contributors and range...
 481   if( nContributors < 1 || TMath::Abs(zVertex) > 10 ) {
 482     return kFALSE;
 483   }
 484
 485   return kTRUE;
 486 }
 487
 488
 489 //________________________________________________________________________
 490 Bool_t AliMuonEffMC::IsGoodMUONtrack(AliESDMuonTrack &track)
 491 {
 492   // Applying track cuts for MUON tracks
 493   if(!track.ContainTrackerData()) return kFALSE;
 494
 495   Double_t thetaTrackAbsEnd = TMath::ATan(track.GetRAtAbsorberEnd()/505.) * TMath::RadToDeg();
 496   Double_t eta = track.Eta();
 497
 498   // Theta cut at absorber end
 499   if(thetaTrackAbsEnd <= 2. || thetaTrackAbsEnd >= 10.) return kFALSE;
 500   // Eta cut
 501   if(eta <= -4. || eta >= -2.5) return kFALSE;
 502   if(track.GetMatchTrigger() <= 0) return kFALSE;
 503
 504   return kTRUE;
 505 }
 506
 507 //________________________________________________________________________
 508 Bool_t AliMuonEffMC::IsGoodMUONtrack(AliAODTrack &track)
 509 {
 510   if (!track.IsMuonTrack()) return kFALSE;
 511
 512   Double_t dThetaAbs = TMath::ATan(track.GetRAtAbsorberEnd()/505.)
 513                      * TMath::RadToDeg();
 514   if ((dThetaAbs<2.) || (dThetaAbs>10.)) return kFALSE;
 515
 516   Double_t dEta = track.Eta();
 517   if ((dEta<-4.) || (dEta>2.5)) return kFALSE;
 518
 519   if (track.GetMatchTrigger()<0.5) return kFALSE;
 520
 521   return kTRUE;
 522 }