Patch for the correction framework

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpidTOF.h

#ifndef ALIHFEPIDTOF_H
#define ALIHFEPIDTOF_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice   */   

//
// Class for TOF PID
// Rejects protons and kaons at the TPC dE/dx line crossings
// For more information please check the implementation file
//
#ifndef ALIHFEPIDBASE_H
#include "AliHFEpidBase.h"
#endif

class TList;
class TH2F;

class AliAODTrack;
class AliAODMCParticle;
class AliESDtrack;
class AliMCParticle;
class AliESDpid;
class AliLog;

class AliHFEcollection;

class AliHFEpidTOF : public AliHFEpidBase{
  public:
    AliHFEpidTOF(const Char_t *name);
    virtual ~AliHFEpidTOF();
    AliHFEpidTOF(const AliHFEpidTOF &c);
    AliHFEpidTOF &operator=(const AliHFEpidTOF &c);
  
    virtual Bool_t    InitializePID();
    virtual Int_t     IsSelected(AliHFEpidObject *track);
    virtual Bool_t    HasQAhistos() const { return kTRUE; };
  
    void SetTOFnSigma(Short_t nSigma) { fNsigmaTOF = nSigma; };

    Double_t Likelihood(const AliESDtrack *track, Int_t species, Float_t rsig = 2.); 
 
  protected:
    void Copy(TObject &ref) const;
    void AddQAhistograms(TList *qaHist);
    Int_t MakePIDesd(AliESDtrack *esdTrack, AliMCParticle *mcTrack);
    Int_t MakePIDesdV2(AliESDtrack *esdTrack, AliMCParticle *mcTrack);
    Int_t MakePIDesdV3(AliESDtrack *esdTrack, AliMCParticle *mcTrack);
    Int_t MakePIDaod(AliAODTrack *aodTrack, AliAODMCParticle *mcTrack);
  
  private:
    typedef enum{
      kHistTOFpidFlags = 0,
      kHistTOFpidBetavP = 1,
      kHistTOFsignal = 2,
      kHistTOFlength =3,
      kHistTOFpid0 = 4,
      kHistTOFpid1 = 5,
      kHistTOFpid2 = 6,
      kHistTOFpid3 = 7,
      kHistTOFpid4 = 8
    } QAHist_t;
  
    AliPID        *fPID;           //! PID Object
    AliHFEcollection *fQAList;     //! QA histograms

    Short_t    fNsigmaTOF;         // TOF sigma band

    ClassDef(AliHFEpidTOF, 1)
};

#endif