Updates and additions: Classes for signal and spectrum extraction; saving of

[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronVarCuts.h

diff --git a/PWG3/dielectron/AliDielectronVarCuts.h b/PWG3/dielectron/AliDielectronVarCuts.h
index a283c02021ab747ed2bef02506d08c45097ba39e..a5c089bfeda246bdc3dcbd65c81ca8b9513c7905 100644 (file)
--- a/PWG3/dielectron/AliDielectronVarCuts.h
+++ b/PWG3/dielectron/AliDielectronVarCuts.h
@@ -22,17 +22,31 @@
 //#############################################################
 
 #include <Rtypes.h>
+
 #include <AliAnalysisCuts.h>
-#include <AliDielectronVarManager.h>
+#include "AliDielectronVarManager.h"
 
 class AliDielectronVarCuts : public AliAnalysisCuts {
 public:
+  // Whether all cut criteria have to be fulfilled of just any
+  enum CutType { kAll=0, kAny };
+  
   AliDielectronVarCuts();
   AliDielectronVarCuts(const char* name, const char* title);
   virtual ~AliDielectronVarCuts();
   //TODO: make copy constructor and assignment operator public
-  void AddCut(Double_t min, Double_t max, AliDielectronVarManager::ValueTypes type);
+  void AddCut(AliDielectronVarManager::ValueTypes type, Double_t min, Double_t max, Bool_t excludeRange=kFALSE);
+  void AddCut(AliDielectronVarManager::ValueTypes type, Double_t value, Bool_t excludeRange=kFALSE);
+  
+  // setters
+  void    SetCutOnMCtruth(Bool_t mc=kTRUE) { fCutOnMCtruth=mc; }
+  void    SetCutType(CutType type)         { fCutType=type;    }
+  
+  // getters
+  Bool_t  GetCutOnMCtruth() const { return fCutOnMCtruth; }
+  CutType GetCutType()      const { return fCutType;      }
 
+  
   //
   //Analysis cuts interface
   //
@@ -46,39 +60,42 @@ public:
   // Cut information
   //
   virtual UInt_t GetSelectedCutsMask() const { return fSelectedCutsMask; }
-  Bool_t IsCutActive(AliDielectronVarManager::ValueTypes cut) { return TESTBIT(fActiveCutsMask,cut); }
 
+  virtual void Print(const Option_t* option = "") const;
+  
 private:
 
-  UChar_t  fActiveCuts[AliDielectronVarManager::kNMaxValues];       // list of activated cuts
-  UChar_t  fNActiveCuts;                      // number of acive cuts
-  UInt_t   fActiveCutsMask;                   // maks of active cuts
+  UShort_t  fActiveCuts[AliDielectronVarManager::kNMaxValues];       // list of activated cuts
+  UShort_t  fNActiveCuts;                      // number of acive cuts
+  UInt_t    fActiveCutsMask;                   // mask of active cuts
   
   UInt_t   fSelectedCutsMask;                 // Maks of selected cuts, is available after calling IsSelected
+
+  Bool_t   fCutOnMCtruth;                     // whether to cut on the MC truth of the particle
+
+  CutType  fCutType;                          // type of the cut: any, all
   
   Double_t fCutMin[AliDielectronVarManager::kNMaxValues];           // minimum values for the cuts
   Double_t fCutMax[AliDielectronVarManager::kNMaxValues];           // maximum values for the cuts
-
-  void ActivateCut(AliDielectronVarManager::ValueTypes cutName);
+  Bool_t fCutExclude[AliDielectronVarManager::kNMaxValues];         // inverse cut logic?
   
   AliDielectronVarCuts(const AliDielectronVarCuts &c);
   AliDielectronVarCuts &operator=(const AliDielectronVarCuts &c);
   
-  ClassDef(AliDielectronVarCuts,1)         //Cut class providing cuts to all infomation available for the AliVParticle interface
+  ClassDef(AliDielectronVarCuts,2)         //Cut class providing cuts to all infomation available for the AliVParticle interface
 };
 
 
 //
 //Inline functions
 //
-inline void AliDielectronVarCuts::AddCut(Double_t min, Double_t max, AliDielectronVarManager::ValueTypes type)
+inline void AliDielectronVarCuts::AddCut(AliDielectronVarManager::ValueTypes type, Double_t value, Bool_t excludeRange)
 {
   //
-  // Set cut range and activate it
+  // Set cut in a small delta around value
   //
-  fCutMin[type]=min;
-  fCutMax[type]=max;
-  ActivateCut(type);
+  const Double_t kDelta=1e-20;
+  AddCut(type,value-kDelta,value+kDelta, excludeRange);
 }
 
 #endif