PWGHF/hfe/AliHFEpidBayes.h

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15 //
  16 // Class for Bayes PID
  17 // electron selection via combined PID
  18 // For more information please check the implementation file
  19 //
  20 #ifndef ALIHFEPIDBAYES_H
  21 #define ALIHFEPIDBAYES_H
  22
  23 #ifndef ALIHFEPIDBASE_H
  24 #include "AliHFEpidBase.h"
  25 #endif
  26
  27 //#ifndef ALIPID_H
  28 //#include "AliPID.h"
  29 //#endif
  30
  31 class TList;
  32 class AliAODTrack;
  33 class AliAODMCParticle;
  34 class AliESDtrack;
  35 class AliMCParticle;
  36 class AliVParticle;
  37 class AliHFEcollection;
  38 class AliHFEpidQAmanager;
  39 class AliPIDCombined;
  40 class AliPID;
  41
  42 class AliHFEpidBayes : public AliHFEpidBase{
  43   public:
  44     AliHFEpidBayes();
  45     AliHFEpidBayes(const Char_t *name);
  46     AliHFEpidBayes(const AliHFEpidBayes &ref);
  47     AliHFEpidBayes &operator=(const AliHFEpidBayes &ref);
  48     void Copy(TObject &ref) const;
  49     virtual ~AliHFEpidBayes();
  50
  51     virtual Bool_t InitializePID(Int_t /*run*/);
  52     virtual Int_t IsSelected(const AliHFEpidObject *track, AliHFEpidQAmanager *pidqa) const;
  53     void CalcCombProb(const AliHFEpidObject *track,const AliPIDResponse *fkPIDResponse, Double_t* probTPCTOF) const;
  54     void SetBayesDetectorMask(Int_t detmask) { fDetMask = detmask; };
  55     void SetBayesPIDThreshold(Float_t pidthres) {fpidthres=pidthres;};
  56
  57 protected:
  58
  59   private:
  60   AliPIDCombined       *fPIDCombined;            //! combined PID object
  61   Int_t                fDetMask;                 // detector mask
  62   Int_t                fDetMaskDefault;          // default detector mask
  63   Int_t                fDetMaskDefaultandTRD;    // default detector mask +TRD
  64   Float_t              fpidthres;                // pid threshold
  65   ClassDef(AliHFEpidBayes, 1)                    // Bayes Electron ID class
  66 };
  67
  68
  69 #endif