updated pPb tasks

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

diff --git a/PWGHF/hfe/AliHFEpidTOF.h b/PWGHF/hfe/AliHFEpidTOF.h
index ca2c843d46458c9312279f1b3a626fab34313ce5..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,16 +31,24 @@ 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) { fSigmaBordersTOF[0] = lower; fSigmaBordersTOF[1] = 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)
     };
     Float_t    fNsigmaTOF;          // TOF sigma band
-    Float_t    fSigmaBordersTOF[2]; // Min. and Max TOF sigma cut
+    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)
 };