Adding the possibility to filter for events containing a jet above threshold set...

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnCutPrimaryVertex.cxx

//
// Class AliRsnCutPrimaryVertex
//
// This cut implementation checks the quality of event primary vertex.
// It currently works only with ESD events (not AOD).
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#include "AliRsnCutPrimaryVertex.h"

ClassImp(AliRsnCutPrimaryVertex)

//_________________________________________________________________________________________________
AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex() :
  AliRsnCut(AliRsnCut::kEvent),
  fAcceptTPC(kFALSE)
{
//
// Default constructor.
//
}

//_________________________________________________________________________________________________
AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex
(const char *name, Int_t nContributors, Bool_t acceptTPC) :
  AliRsnCut(name, AliRsnCut::kEvent, 0, nContributors - 1),
  fAcceptTPC(acceptTPC)
{
//
// Main constructor.
// Defines the cut range between 0 and
// the minimum required number of contributors.
// The cut will be passed when if the event has a
// primary vertex with number of contributors outside this interval.
// ---
// If the 'acceptTPC' argument is true, events with TPC
// primary vertex will be checked, otherwise they will be
// rejected by default.
// ---
// Since the range check uses the '>=' and '<=', the high edge
// must be decreased by 1 to get the right behaviour, since this is integer.
//
}

//_________________________________________________________________________________________________
Bool_t AliRsnCutPrimaryVertex::IsSelected(TObject *obj1, TObject* /*obj2*/)
{
//
// Cut checker
//

  // retrieve ESD event
  AliRsnEvent *rsn = dynamic_cast<AliRsnEvent*>(obj1);
  if (!rsn) return kFALSE;
  AliESDEvent *esd = dynamic_cast<AliESDEvent*>(rsn->GetRef());
  if (!esd) {
    AliDebug(AliLog::kDebug+2, "NO ESD");
    return kFALSE;
  }

  // get the three possible primary vertexes of the event:
  // 0 = default one
  // 1 = SPD
  // 2 = TPC
  // then, if the default vertex is TPC, the event is rejected,
  // otherwise, the event is rejected only if its vertex status is 'false'
  // get primary vertexes
  const AliESDVertex *vert[3];
  vert[0] = esd->GetPrimaryVertex();
  vert[1] = esd->GetPrimaryVertexSPD();
  vert[2] = esd->GetPrimaryVertexTPC();

  // if TPC vertexes are rejected by default do this now
  if (!fAcceptTPC && (vert[0] == vert[2])) {
    AliDebug(AliLog::kDebug+2, "Rejecting TPC vertex");
    return kFALSE;
  }

  // if we are here, vert[0] contains the default primary vertex
  // in case it is with tracks or SPD, its status must be OK
  // because otherwise the ESD event returns the lower level vertex
  // then, we can check just the first element in the array
  if (!vert[0]->GetStatus()) {
    AliDebug(AliLog::kDebug+2, "Bad vertex status");
    return kFALSE;
  }

  // if we are here, the status of default vertex is OK
  // and then we check the number of contributors:
  // it must be *outside* the 'allowed' range
  fCutValueI = vert[0]->GetNContributors();
  return /*there is a NOT operator */!/*here*/OkRange();
}