From fd40e2fece3df6031a062ca9cd4ac2cfaf46aafd Mon Sep 17 00:00:00 2001
From: Julian Book <Julian>
Date: Wed, 11 Dec 2013 15:08:04 +0100
Subject: [PATCH] -add run selection in event cuts

---
 PWGDQ/dielectron/AliDielectronEventCuts.cxx | 11 +++++++++++
 PWGDQ/dielectron/AliDielectronEventCuts.h   |  6 ++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/PWGDQ/dielectron/AliDielectronEventCuts.cxx b/PWGDQ/dielectron/AliDielectronEventCuts.cxx
index b5c1c3f8337..d3dc0c9bf2d 100644
--- a/PWGDQ/dielectron/AliDielectronEventCuts.cxx
+++ b/PWGDQ/dielectron/AliDielectronEventCuts.cxx
@@ -45,6 +45,7 @@ const char* AliDielectronEventCuts::fgkVtxNames[AliDielectronEventCuts::kVtxTrac
 
 AliDielectronEventCuts::AliDielectronEventCuts() :
   AliAnalysisCuts(),
+  fRun(),
   fVtxZmin(0.),
   fVtxZmax(0.),
   fRequireVtx(kFALSE),
@@ -74,6 +75,7 @@ AliDielectronEventCuts::AliDielectronEventCuts() :
 //______________________________________________
 AliDielectronEventCuts::AliDielectronEventCuts(const char* name, const char* title) :
   AliAnalysisCuts(name, title),
+  fRun(),
   fVtxZmin(0.),
   fVtxZmax(0.),
   fRequireVtx(kFALSE),
@@ -232,6 +234,15 @@ Bool_t AliDielectronEventCuts::IsSelectedAOD(TObject* event)
   AliAODEvent *ev=dynamic_cast<AliAODEvent*>(event);
   if (!ev) return kFALSE;
 
+  // run rejection
+  Int_t run = ev->GetRunNumber();
+  if(fRun.GetNrows()) {
+    for(Int_t irun=0; irun<fRun.GetNrows(); irun++) {
+      printf("check run %.0f \n ",fRun(irun));
+      if(fRun(irun)==run) return kFALSE;
+    }
+  }
+
   if (fCentMin<fCentMax){
     AliCentrality *centrality=ev->GetCentrality();
     Double_t centralityF=-1;
diff --git a/PWGDQ/dielectron/AliDielectronEventCuts.h b/PWGDQ/dielectron/AliDielectronEventCuts.h
index be0d0954f1f..fe844baa296 100644
--- a/PWGDQ/dielectron/AliDielectronEventCuts.h
+++ b/PWGDQ/dielectron/AliDielectronEventCuts.h
@@ -20,6 +20,7 @@
 //#############################################################
 
 #include <TF1.h>
+#include <TVectorD.h>
 
 #include <AliAnalysisUtils.h>
 #include <AliAnalysisCuts.h>
@@ -38,7 +39,7 @@ public:
 
   virtual ~AliDielectronEventCuts();
 
-
+  void SetRunRejection(const TVectorD * vec)    { fRun.Use(vec->GetNrows(),vec->GetMatrixArray()); }
   void SetVertexType(EVtxType type)             { fVtxType=type;                }
   void SetVertexZ(Double_t zmin, Double_t zmax) { fVtxZmin=zmin; fVtxZmax=zmax; }
   void SetRequireVertex(Bool_t req=kTRUE)       { fRequireVtx=req;              }
@@ -61,6 +62,7 @@ public:
 
 private:
   static const char* fgkVtxNames[AliDielectronEventCuts::kVtxTracksOrSPD+1];  //vertex names
+  TVectorD fRun;                    // run rejection vector
   Double_t fVtxZmin;                // minimum z vertex position
   Double_t fVtxZmax;                // maximum z vertex position
   Bool_t   fRequireVtx;             // require a vertex
@@ -89,7 +91,7 @@ private:
   AliDielectronEventCuts &operator=(const AliDielectronEventCuts &c);
 
   
-  ClassDef(AliDielectronEventCuts,2)         // Dielectron EventCuts
+  ClassDef(AliDielectronEventCuts,3)         // Dielectron EventCuts
 };
 
 
-- 
2.39.3