Fixes for #89479: change in AliPIDResponse and AliPIDCombined (STEERBase). F.Noferini

[u/mrichter/AliRoot.git] / STEER / STEERBase / AliPIDCombined.h

diff --git a/STEER/STEERBase/AliPIDCombined.h b/STEER/STEERBase/AliPIDCombined.h
index 446449fd03ee39afe98be5e72b32d36136c2f4c5..ce1a52b2ec7844d115348a7620b2d2cd9f916bb8 100644 (file)
--- a/STEER/STEERBase/AliPIDCombined.h
+++ b/STEER/STEERBase/AliPIDCombined.h
@@ -18,6 +18,7 @@
 #include <AliPID.h>\r
 #include <AliPIDResponse.h>\r
 #include <TH1F.h>\r
+#include <TH2F.h>\r
 \r
 //class TH1;\r
 class AliPIDResponse;\r
@@ -37,13 +38,17 @@ public:
   void SetPriorDistribution(AliPID::EParticleType type,TH1F *prior);\r
   //  const TH1* GetPriorDistribution(AliPID::EParticleType type) const {return (TH1*)fPriorsDistributions[type];}\r
   TH1* GetPriorDistribution(AliPID::EParticleType type)  const {return (TH1*)fPriorsDistributions[type];}\r
+  \r
+  void GetPriors(const AliVTrack *track,Double_t* p,const AliPIDResponse *response,UInt_t detUsed) const;\r
+  \r
+  void SetDefaultTPCPriors();\r
        \r
   UInt_t ComputeProbabilities(const AliVTrack *track, const AliPIDResponse *response, Double_t* bayesProbabilities) const;\r
   void SetSelectedSpecies(Int_t selectedSpecies) {fSelectedSpecies = selectedSpecies;}\r
   Int_t GetSelectedSpecies() const {return fSelectedSpecies;}\r
 \r
 protected:\r
-  void GetPriors(const AliVTrack *track,Double_t* priors) const;\r
+  void GetPriors(const AliVTrack *track,Double_t* priors,Float_t centrality=-1) const;\r
   void ComputeBayesProbabilities(Double_t* bayesProbabilities,const Double_t* probDensity, const Double_t* priors) const;\r
   void SetCombinedStatus(const AliPIDResponse::EDetPidStatus status,UInt_t *mask, const AliPIDResponse::EDetCode bit, Double_t* p) const;\r
 \r
@@ -56,8 +61,10 @@ private:
   Bool_t fEnablePriors;      // Enable bayesian PID (if kFALSE priors set flat)\r
   Int_t fSelectedSpecies;    // Number of selected species to study\r
   TH1F *fPriorsDistributions[AliPID::kSPECIES+AliPID::kSPECIESLN]; // priors\r
+  Bool_t fUseDefaultTPCPriors; // switch to use Defaul TPC Priors\r
+  static TH2F *fDefaultPriorsTPC[5]; // Default priors for TPC tracks\r
 \r
-  ClassDef(AliPIDCombined,1);\r
+  ClassDef(AliPIDCombined,2);\r
 };\r
 \r
 #endif\r