updated pPb tasks

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEpidTOF.h

diff --git a/PWGHF/hfe/AliHFEpidTOF.h b/PWGHF/hfe/AliHFEpidTOF.h
index 185b45e12b64906c357caca5310585d9356efde7..a093b82d072c8d4b5cfc5cebd40cdd005636cfc7 100644 (file)
--- a/PWGHF/hfe/AliHFEpidTOF.h
+++ b/PWGHF/hfe/AliHFEpidTOF.h
@@ -14,6 +14,7 @@
 #endif
 
 class AliVParticle;
+class AliVTrack;
 class AliPID;
 
 class AliHFEpidQAmanager;
@@ -30,11 +31,15 @@ class AliHFEpidTOF : public AliHFEpidBase{
     virtual Int_t     IsSelected(const AliHFEpidObject *track, AliHFEpidQAmanager *piqa) const;
   
     void SetTOFnSigma(Float_t nSigma) { fNsigmaTOF = nSigma; };
-    void SetTOFnSigmaBand(Float_t lower, Float_t upper) { fSigmaBordersTOFLower[0] = lower; fSigmaBordersTOFUpper[0] = upper; SetBit(kSigmaBand, kTRUE); }
+    void SetTOFnSigmaBand(Float_t lower, Float_t upper);
     void SetTOFnSigmaBandCentrality(Float_t lower, Float_t upper, Int_t centralityBin); 
+    void UseTOFonlyIfAvailable() { fUseOnlyIfAvailable = kTRUE; }
+    void SetRejectTOFmismatch() { fRejectMismatch = kTRUE; }
 
   protected:
     void Copy(TObject &ref) const;
+    Bool_t IsMismatch(const AliVTrack *const track) const;
+
   private:
     enum {
       kSigmaBand = BIT(15)
@@ -42,6 +47,8 @@ class AliHFEpidTOF : public AliHFEpidBase{
     Float_t    fNsigmaTOF;          // TOF sigma band
     Float_t    fSigmaBordersTOFLower[12]; // Min.  sigma cut
     Float_t    fSigmaBordersTOFUpper[12]; // Max.  sigma cut
+    Bool_t     fUseOnlyIfAvailable;       // Use TOF obly if available
+    Bool_t     fRejectMismatch;           // Reject TOF mismatch
 
     ClassDef(AliHFEpidTOF, 1)
 };