Bug fix for poly fit (ChiaraB)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliRDHFCutsD0toKpi.h

diff --git a/PWG3/vertexingHF/AliRDHFCutsD0toKpi.h b/PWG3/vertexingHF/AliRDHFCutsD0toKpi.h
index a48bd44272f2db2ed25ad95f61c50bca8d34c4a6..d8021472376b7c60ff5e7233a0e4aa31c732941f 100644 (file)
--- a/PWG3/vertexingHF/AliRDHFCutsD0toKpi.h
+++ b/PWG3/vertexingHF/AliRDHFCutsD0toKpi.h
@@ -11,6 +11,9 @@
 
 #include "AliRDHFCuts.h"
 
+class AliAODEvent;
+class AliAODRecoDecayHF;
+
 class AliRDHFCutsD0toKpi : public AliRDHFCuts 
 {
  public:
@@ -25,21 +28,32 @@ class AliRDHFCutsD0toKpi : public AliRDHFCuts
   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters);
 
   using AliRDHFCuts::IsSelected;
-  virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel);
-  
+  virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel) 
+                         {return IsSelected(obj,selectionLevel,0);}
+  virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent* aod);
+
   Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
   Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(1,iPtBin)] : 1.e6);}
-
+  Int_t CombineSelectionLevels(Int_t selectionvalTrack,Int_t selectionvalCand,Int_t selectionvalPID)const;
   virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;
 
-  virtual Int_t IsSelectedPID(AliAODRecoDecayHF *rd) const;
-
-
+  virtual Int_t IsSelectedPID(AliAODRecoDecayHF *rd);
+  Int_t IsSelectedPIDdefault(AliAODRecoDecayHF *rd);
+  Int_t IsSelectedSpecialCuts(AliAODRecoDecayHF *d) const;
+  void SetUseSpecialCuts(Bool_t useSpecialCuts) {fUseSpecialCuts=useSpecialCuts;}
+  void SetLowPt(Bool_t lowpt) {fLowPt=lowpt;}
+  Bool_t GetUseSpecialCuts() const {return fUseSpecialCuts;}
+  void SetUseDefaultPID(Bool_t defPID){fDefaultPID=defPID;}
+  Bool_t GetIsUsedDefPID(){return fDefaultPID;}
+  
+  
  protected:
-
-
-  ClassDef(AliRDHFCutsD0toKpi,1);  // class for cuts on AOD reconstructed D0->Kpi
+  Bool_t fUseSpecialCuts;           // flag to switch on/off special cuts
+  Bool_t fLowPt;           // flag to switch on/off different pid for low pt D0
+  Bool_t fDefaultPID;      // flag to switch on/off the default pid
+  ClassDef(AliRDHFCutsD0toKpi,2);  // class for cuts on AOD reconstructed D0->Kpi
 };
 
 #endif
 
+