X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=CORRFW%2FAliCFTrackCutPid.h;h=2b4df8dc07a6ca8ba08821c359267a11ec96d17b;hb=ae0752da81fa0d3818cc3d3e8a9445a22f1c7ab8;hp=df94600680cc671e548dd2c1c554e7ca17852252;hpb=56143784e24339e4bf356dc0d9ed341ee651d460;p=u%2Fmrichter%2FAliRoot.git

diff --git a/CORRFW/AliCFTrackCutPid.h b/CORRFW/AliCFTrackCutPid.h
index df94600680c..2b4df8dc07a 100644
--- a/CORRFW/AliCFTrackCutPid.h
+++ b/CORRFW/AliCFTrackCutPid.h
@@ -60,6 +60,7 @@ class AliCFTrackCutPid : public AliCFCutBase
   void SetDetectorProbabilityRestriction(TString det, Int_t iPart, Double_t upperprob); 
   void SetHistogramAxis(Int_t nbins, Double_t xmin, Double_t xmax) {fNbins=nbins; fXmin = xmin; fXmax = xmax;}
   void SetAODmode(Bool_t isaod = kFALSE) {fgIsAOD=isaod;}  
+  void SetProbThreshold(Double_t value) {fProbThreshold=value;}
  
   //returns the track identification number  
   Int_t GetID(ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;  
@@ -108,6 +109,9 @@ class AliCFTrackCutPid : public AliCFCutBase
   Int_t fDetRestr;                                          // id of the detector for the restriction
   Int_t fiPartRestr;                                        // id of the particle for the restriction
   Double_t fDetProbRestr;                                   // probability restriction value
+  Double_t fProbThreshold;                                  // if different from 0, the assigned PID will be set to 
+                                                            // fgParticleType if the probability is larger than this threshold,
+                                                            // regardless it is the highest or not (!)
 
   Double_t fPriors[AliPID::kSPECIES];                       // a priori concentrations
   TF1 *fPriorsFunc[AliPID::kSPECIES];                       // momentum dependent priors