X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=CORRFW%2FAliCFTrackCutPid.h;h=2b4df8dc07a6ca8ba08821c359267a11ec96d17b;hb=3d2662369ff6ab85e620ebc5e98a54015f240769;hp=ef2bf086e8bdf593e7aa3014c308f2d63118f01b;hpb=264ebaacb7bcf8499b20fe6235520fe047a649f5;p=u%2Fmrichter%2FAliRoot.git

diff --git a/CORRFW/AliCFTrackCutPid.h b/CORRFW/AliCFTrackCutPid.h
index ef2bf086e8b..2b4df8dc07a 100644
--- a/CORRFW/AliCFTrackCutPid.h
+++ b/CORRFW/AliCFTrackCutPid.h
@@ -23,6 +23,7 @@
 #include "AliCFCutBase.h"
 #include "AliPID.h"
 #include "AliESDtrack.h"
+#include "AliAODTrack.h"
 #include <TString.h>
 #include <TObject.h>
 #include <TH1F.h>
@@ -58,61 +59,68 @@ class AliCFTrackCutPid : public AliCFCutBase
   void SetANDstatus(TString dets);
   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;}
-  
-  //loads the track detector responses and the track status
-  void TrackInfo(const AliESDtrack *pTrk,ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;  
-  
-  //identifies the track
-  Int_t Identify(Double_t pid[AliPID::kSPECIES]) const;
-  
-  //identifies the track filling the QA histograms                                                          
-  Int_t IdentifyQA(const Double_t pid[AliPID::kSPECIES],Int_t idets) const;
-  
+  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;  
+  //returns the track identification number in caso of an AliAODTrack
+  Int_t GetAODID(AliAODTrack *aodtrack) const;
   
+ 
   //main 
   virtual Bool_t IsSelected(TObject *track); 
   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}  
-  //histo booking                                                    
-  void Init();
-  
   //histos are added to a list
-  void AddQAHistograms(TList *qalist) const;
+  void AddQAHistograms(TList *qalist);
   
   
  private:
+
+  //loads the track detector responses and the track status
+  void TrackInfo(const AliESDtrack *pTrk,ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;
+
+  //identifies the track
+  Int_t Identify(Double_t pid[AliPID::kSPECIES]) const;
+
+  //identifies the track filling the QA histograms
+  Int_t IdentifyQA(const Double_t pid[AliPID::kSPECIES],Int_t idets) const;
+
   void SetPPriors(AliESDtrack *pTrk);                          
-  ULong_t StatusForAND(ULong_t status[kNdets+1]) const; //
+  ULong_t StatusForAND(ULong_t status[kNdets+1]) const; 
   void InitialiseHisto();
-  void DefineHistograms();    
+  void DefineHistograms();                                 // histo booking  
   Bool_t Check(const Double_t *p, Int_t iPsel, Double_t minDiff) const;
   void CombPID(ULong_t status[kNdets+1],Double_t pid[kNdets+1][AliPID::kSPECIES],Double_t *combpid) const;
   
-  Double_t fCut;                                            //probability cut
+  Double_t fCut;                                            // probability cut
   Double_t fMinDiffResponse;                                // minimum difference between detector resposes
   Double_t fMinDiffProbability;                             // minimum difference between probability values
   Int_t fgParticleType;                                     // requested particle type
   Bool_t fgIsComb;                                          // flag for the combined pid
+  Bool_t fgIsAOD;                                           // flag for AOD QA histograms
   Bool_t fCheckResponse;                                    // flag to check the minimum difference of det responsess
   Bool_t fCheckSelection;                                   // flag to check the minimum difference of probabilities
-  Bool_t fIsPpriors;                                        //flag for momentum dependent priors
-  Bool_t fIsDetAND;                                         //flag for AND with multiple detectors
-  Double_t fXmin;                                           //x min QA histo
-  Double_t fXmax;                                           //x max QA histo
-  Int_t fNbins;                                             //n bins QA histo 
-  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
+  Bool_t fIsPpriors;                                        // flag for momentum dependent priors
+  Bool_t fIsDetAND;                                         // flag for AND with multiple detectors
+  Double_t fXmin;                                           // x min QA histo
+  Double_t fXmax;                                           // x max QA histo
+  Int_t fNbins;                                             // n bins QA histo 
+  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
-  Bool_t fDets[kNdets];                                     //boolean(s) corresponding to the chosen detector(s) 
-  Bool_t fDetsInAnd[kNdets];                                //detector to be in AND for the combined PID
-  TH1F *fhResp[kNdets][AliPID::kSPECIES];                   //QA histo
-  TH1F *fhProb[kNdets][AliPID::kSPECIES];                   //QA histo
-  TH1F *fhCombResp[AliPID::kSPECIES];                       //QA histo
-  TH1F *fhCombProb[AliPID::kSPECIES];                       //QA histo
+  Double_t fPriors[AliPID::kSPECIES];                       // a priori concentrations
+  TF1 *fPriorsFunc[AliPID::kSPECIES];                       // momentum dependent priors
+  Bool_t fDets[kNdets];                                     // boolean(s) corresponding to the chosen detector(s) 
+  Bool_t fDetsInAnd[kNdets];                                // detector to be in AND for the combined PID
+  TH1F *fhResp[kNdets][AliPID::kSPECIES];                   // QA histo
+  TH1F *fhProb[kNdets][AliPID::kSPECIES];                   // QA histo
+  TH1F *fhCombResp[AliPID::kSPECIES];                       // QA histo
+  TH1F *fhCombProb[AliPID::kSPECIES];                       // QA histo
   
   ClassDef(AliCFTrackCutPid,1);
 };