PWG2/RESONANCES/AliRsnCutPrimaryVertex.cxx

   1 //
   2 // Class AliRsnCutPrimaryVertex
   3 //
   4 // This cut implementation checks the quality of event primary vertex.
   5 // It currently works only with ESD events (not AOD).
   6 //
   7 // authors: Martin Vala (martin.vala@cern.ch)
   8 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
   9 //
  10
  11 #include "AliRsnCutPrimaryVertex.h"
  12
  13 ClassImp(AliRsnCutPrimaryVertex)
  14
  15 //_________________________________________________________________________________________________
  16 AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex() :
  17   AliRsnCut(AliRsnCut::kEvent),
  18   fAcceptTPC(kFALSE)
  19 {
  20 //
  21 // Default constructor.
  22 //
  23 }
  24
  25 //_________________________________________________________________________________________________
  26 AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex
  27 (const char *name, Double_t maxVz, Int_t nContributors, Bool_t acceptTPC) :
  28   AliRsnCut(name, AliRsnCut::kEvent, 0, nContributors - 1),
  29   fAcceptTPC(acceptTPC)
  30 {
  31 //
  32 // Main constructor.
  33 // Defines the cut range between 0 and
  34 // the minimum required number of contributors.
  35 // The cut will be passed when if the event has a
  36 // primary vertex with number of contributors outside this interval.
  37 // ---
  38 // If the 'acceptTPC' argument is true, events with TPC
  39 // primary vertex will be checked, otherwise they will be
  40 // rejected by default.
  41 // ---
  42 // Since the range check uses the '>=' and '<=', the high edge
  43 // must be decreased by 1 to get the right behaviour, since this is integer.
  44 //
  45
  46   fMinD = 0.0;
  47   fMaxD = maxVz;
  48 }
  49
  50 //_________________________________________________________________________________________________
  51 Bool_t AliRsnCutPrimaryVertex::IsSelected(TObject *obj1, TObject* /*obj2*/)
  52 {
  53 //
  54 // Cut checker
  55 //
  56
  57   // retrieve ESD event
  58   AliRsnEvent *rsn = dynamic_cast<AliRsnEvent*>(obj1);
  59   if (!rsn) return kFALSE;
  60   AliESDEvent *esd = dynamic_cast<AliESDEvent*>(rsn->GetRef());
  61   if (!esd)
  62   {
  63     AliDebug(AliLog::kDebug+2, "NO ESD");
  64     return kFALSE;
  65   }
  66
  67   // get the best primary vertex:
  68   // first try the one with tracks
  69   const AliESDVertex *vTrk  = esd->GetPrimaryVertexTracks();
  70   const AliESDVertex *vSPD  = esd->GetPrimaryVertexSPD();
  71   const AliESDVertex *vTPC  = esd->GetPrimaryVertexTPC();
  72   Double_t            vzTrk = 2.0 * fMaxD;
  73   Double_t            vzSPD = 2.0 * fMaxD;
  74   Double_t            vzTPC = 2.0 * fMaxD;
  75   if (vTrk) vzTrk = TMath::Abs(vTrk->GetZv());
  76   if (vSPD) vzSPD = TMath::Abs(vSPD->GetZv());
  77   if (vTPC) vzTPC = TMath::Abs(vTPC->GetZv());
  78   AliDebug(AliLog::kDebug + 1, Form("Vertex with tracks: contributors = %d, abs(vz) = %f", vTrk->GetNContributors(), vzTrk));
  79   AliDebug(AliLog::kDebug + 1, Form("Vertex with SPD,    contributors = %d, abs(vz) = %f", vSPD->GetNContributors(), vzSPD));
  80   AliDebug(AliLog::kDebug + 1, Form("Vertex with TPC,    contributors = %d, abs(vz) = %f", vTPC->GetNContributors(), vzTPC));
  81   if(vTrk->GetStatus())
  82   {
  83     fCutValueI = vTrk->GetNContributors();
  84     fCutValueD = vzTrk;
  85   }
  86   else if (vSPD->GetStatus())
  87   {
  88     fCutValueI = vSPD->GetNContributors();
  89     fCutValueD = vzSPD;
  90   }
  91   else if (vTPC->GetStatus() && fAcceptTPC)
  92   {
  93     fCutValueI = vTPC->GetNContributors();
  94     fCutValueD = vzTPC;
  95   }
  96   else
  97     return kFALSE;
  98
  99   return OkRangeI() && OkRangeD();
 100 }
 101