Changes for

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

#ifndef ALIPIDCOMBINED_H\r
#define ALIPIDCOMBINED_H\r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
//---------------------------------------------------------------//\r
//        Base class for combining PID response of               //\r
//        of different detectors                                 //\r
//        and compute Bayesian probabilities                     //\r
//                                                               //\r
//   Origin: Pietro Antonioli, INFN-BO, pietro.antonioli@cern.ch //\r
//                                                               //\r
//---------------------------------------------------------------//\r
\r
\r
\r
#include <TNamed.h>\r
#include "AliPID.h"\r
#include "AliPIDResponse.h"\r
#include <TH1F.h>\r
#include <TH2F.h>\r
\r
//class TH1;\r
class AliPIDResponse;\r
\r
class AliPIDCombined : public TNamed {\r
public:\r
  AliPIDCombined();\r
  AliPIDCombined(const TString& name, const TString& title);\r
  virtual ~AliPIDCombined();\r
\r
  void SetDetectorMask(Int_t mask) {fDetectorMask=mask;}\r
  Int_t GetDetectorMask() const {return fDetectorMask;}\r
  void SetRejectMismatchMask(Int_t mask) {fRejectMismatchMask=mask;}\r
  Int_t GetRejectMismatchMask() const {return fRejectMismatchMask;}\r
  void SetEnablePriors(Bool_t flag) {fEnablePriors=flag;}\r
  Bool_t GetEnablePriors() const {return fEnablePriors;}\r
  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,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) const;\r
  void SetCombinedStatus(const AliPIDResponse::EDetPidStatus status,UInt_t *mask, const AliPIDResponse::EDetCode bit, Double_t* p,const Float_t probMis) const;\r
\r
private:\r
  AliPIDCombined(const AliPIDCombined&);\r
  AliPIDCombined &operator=(const AliPIDCombined&);\r
\r
  Int_t fDetectorMask;       // Detectors included in combined pid\r
  Int_t fRejectMismatchMask; // Detectors set return flat prob. if mismatch detected \r
  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::kSPECIESC]; // priors\r
  Bool_t fUseDefaultTPCPriors; // switch to use Defaul TPC Priors\r
  static TH2F *fDefaultPriorsTPC[AliPID::kSPECIES]; // Default priors for TPC tracks\r
\r
  ClassDef(AliPIDCombined, 3);   // Combined PID using priors\r
};\r
\r
#endif\r