Adde cut selection for primary vertex quality

diff --git a/PWG2/RESONANCES/AliRsnCutPrimaryVertex.cxx b/PWG2/RESONANCES/AliRsnCutPrimaryVertex.cxx
new file mode 100644 (file)
index 0000000..5927b06
--- /dev/null
+++ b/PWG2/RESONANCES/AliRsnCutPrimaryVertex.cxx
@@ -0,0 +1,112 @@
+//
+// Class AliRsnCutPrimaryVertex
+//
+// General implementation of a single cut strategy, which can be:
+// - a value contained in a given interval  [--> IsBetween()   ]
+// - a value equal to a given reference     [--> MatchesValue()]
+//
+// In all cases, the reference value(s) is (are) given as data members
+// and each kind of cut requires a given value type (Int, UInt, Double),
+// but the cut check procedure is then automatized and chosen thanks to
+// an enumeration of the implemented cut types.
+// At the end, the user (or any other point which uses this object) has
+// to use the method IsSelected() to check if this cut has been passed.
+//
+// authors: Martin Vala (martin.vala@cern.ch)
+//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
+//
+#include "Riostream.h"
+#include "TMath.h"
+
+#include "AliLog.h"
+#include "AliESDEvent.h"
+#include "AliESDVertex.h"
+
+#include "AliRsnEvent.h"
+#include "AliRsnCutPrimaryVertex.h"
+
+ClassImp(AliRsnCutPrimaryVertex)
+
+//_________________________________________________________________________________________________
+AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex() :
+  AliRsnCut()
+{
+//
+// Default constructor.
+//
+}
+
+//_________________________________________________________________________________________________
+AliRsnCutPrimaryVertex::AliRsnCutPrimaryVertex
+(const char *name, Int_t nContributors) :
+  AliRsnCut(name, 0, nContributors)
+{
+//
+// Main constructor.
+// the cut range is outside the interval 0 - min number of contributors.
+//
+}
+
+//_________________________________________________________________________________________________
+Bool_t AliRsnCutPrimaryVertex::IsSelected(ETarget tgt, AliRsnDaughter *track)
+{
+//
+// Cut checker.
+//
+
+  AliWarning("Cannot apply this cut to particles");
+  return kTRUE;
+}
+
+//_________________________________________________________________________________________________
+Bool_t AliRsnCutPrimaryVertex::IsSelected(ETarget /*tgt*/, AliRsnPairParticle* /*pair*/)
+{
+//
+// Cut checker
+//
+
+  AliWarning("Cannot apply this cut to pairs");
+  return kTRUE;
+}
+
+//_________________________________________________________________________________________________
+Bool_t AliRsnCutPrimaryVertex::IsSelected(ETarget /*tgt*/, AliRsnEvent *event)
+{
+//
+// Cut checker
+//
+
+  // retrieve ESD event
+  AliESDEvent *esd = dynamic_cast<AliESDEvent*>(event->GetRef());
+  if (!esd) {
+    AliDebug(AliLog::kDebug+2, "NO ESD");
+    return kTRUE;
+  }
+
+  // check primary vertex and eventually fill step 1
+  // if a vertex with tracks was successfully reconstructed,
+  // it is used for computing DCA;
+  // otherwise, the one computed with SPD is used.
+  // This is known from the "Status" parameter of the vertex itself.
+  const AliESDVertex *v = esd->GetPrimaryVertex();
+  if (!v->GetStatus()) v = esd->GetPrimaryVertexSPD();
+  if (!v->GetStatus()) {
+    AliDebug(AliLog::kDebug+2, "Bad vertex status");
+    return kFALSE;
+  }
+
+  fCutValueI = v->GetNContributors();
+
+  return !OkRange();
+}
+
+//_________________________________________________________________________________________________
+Bool_t AliRsnCutPrimaryVertex::IsSelected(ETarget /*tgt*/, AliRsnEvent* /*ev1*/, AliRsnEvent* /*ev2*/)
+{
+//
+// Cut checker
+//
+
+  AliWarning("Cannot apply this cut to event mixing");
+  return kTRUE;
+}

diff --git a/PWG2/RESONANCES/AliRsnCutPrimaryVertex.h b/PWG2/RESONANCES/AliRsnCutPrimaryVertex.h
new file mode 100644 (file)
index 0000000..ab96260
--- /dev/null
+++ b/PWG2/RESONANCES/AliRsnCutPrimaryVertex.h
@@ -0,0 +1,35 @@
+//
+// Class AliRsnCutRange
+//
+// General implementation of cuts which check a value inside a range.
+// This range can be defined by two integers or two doubles.
+// A user-friendly enumeration allows to define what is checked.
+//
+// authors: Martin Vala (martin.vala@cern.ch)
+//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
+//
+
+#ifndef ALIRSNCUTPRIMARYVERTEX_H
+#define ALIRSNCUTPRIMARYVERTEX_H
+
+#include "AliPID.h"
+#include "AliRsnCut.h"
+
+class AliRsnCutPrimaryVertex : public AliRsnCut
+{
+ public:
+
+  AliRsnCutPrimaryVertex();
+  AliRsnCutPrimaryVertex(const char *name, Int_t minContributors);
+
+  virtual Bool_t IsSelected(ETarget tgt, AliRsnDaughter *daughter);
+  virtual Bool_t IsSelected(ETarget tgt, AliRsnPairParticle *pair);
+  virtual Bool_t IsSelected(ETarget tgt, AliRsnEvent *event);
+  virtual Bool_t IsSelected(ETarget tgt, AliRsnEvent *ev1, AliRsnEvent *ev2);
+
+protected:
+
+  ClassDef(AliRsnCutPrimaryVertex, 1)
+};
+
+#endif