including switch to set on/off iso-track core removal, cleaning and bug fix

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronV0Cuts.h

diff --git a/PWGDQ/dielectron/AliDielectronV0Cuts.h b/PWGDQ/dielectron/AliDielectronV0Cuts.h
index 4a369ae1bdb6e71de4ac480c262db346c5396d05..5f59b08c990331e5e9ccc9b5224e28520bef5f1d 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronV0Cuts.h
+++ b/PWGDQ/dielectron/AliDielectronV0Cuts.h
@@ -32,6 +32,7 @@ public:
     kOffline,
     kOnTheFly
   };
+  enum PIDCutType { kBoth=0, kAny };
 
   AliDielectronV0Cuts();
   AliDielectronV0Cuts(const char* name, const char* title);
@@ -48,9 +49,12 @@ public:
   void SetV0finder(EV0finder finder) {fV0finder=finder;}
   void SetPdgCodes(Int_t mother, Int_t negDaughter, Int_t posDaughter) {fMotherPdg=mother; fNegPdg=negDaughter; fPosPdg=posDaughter;}
   void SetExcludeTracks(Bool_t exclude) {fExcludeTracks=exclude;}
-  void SetDefaultPID(Int_t def) {fPID=def;}
+  void SetDefaultPID(Int_t def, PIDCutType type=kBoth) {fPID=def; fPIDCutType=type; }
   void ResetUniqueEventNumbers() { fOrbit=0; fPeriod=0; fBunchCross=0; }
 
+  // cut information
+  virtual void Print(const Option_t* option = "") const;
+
 private:
 
   TBits fV0TrackArr;                        // array with booleans where TrackID corresponds to bitnumber
@@ -61,6 +65,7 @@ private:
   Int_t fNegPdg;                            // target pdg code of the negative daughter
   Int_t fPosPdg;                            // target pdg code of the positive daughter
   Int_t fPID;                               // default PID usage (see AliDielectronPID)
+  PIDCutType fPIDCutType;                   // apply PIDcuts to one or both daughter tracks
 
   // memebers needed to identify an event
   UInt_t fOrbit;                            // orbit number
@@ -70,7 +75,7 @@ private:
   AliDielectronV0Cuts(const AliDielectronV0Cuts &c);
   AliDielectronV0Cuts &operator=(const AliDielectronV0Cuts &c);
 
-  ClassDef(AliDielectronV0Cuts,1)
+  ClassDef(AliDielectronV0Cuts,3)          // cut class for V0 candidates
 };
 
 #endif