Include informations from rstate module

[u/mrichter/AliRoot.git] / TRD / Cal / AliTRDCalPID.h

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// Authors:                                                                  //
//                                                                           //
//  Alex Bercuci <A.Bercuci@gsi.de>                                          //
//  Alex Wilk <wilka@uni-muenster.de>                                        //
//  Prashant Shukla <shukla@pi0.physi.uni-heidelberg.de>                     //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#ifndef ROOT_TNamed
#include <TNamed.h>
#endif

class AliTRDCalPID : public TNamed
{

 public:

  enum {
    kNMom   = 11,
    kNPlane = 6
  };

  AliTRDCalPID();
  AliTRDCalPID(const Text_t *name, const Text_t *title);
  virtual         ~AliTRDCalPID();

  virtual Bool_t   LoadReferences(Char_t *refFile) = 0;
  static  Double_t GetMomentum(Int_t ip)            
                                                      { return (ip<0 || ip>=kNMom) 
                                                        ? -1.0 
                                                        : fTrackMomentum[ip]; 
                                                      }
  virtual TObject *GetModel(Int_t ip, Int_t iType, Int_t iPlane) const = 0;
  virtual Double_t GetProbability(Int_t spec, Float_t mom, Float_t *dedx
                                , Float_t length, Int_t plane) const = 0;
  static  Char_t  *GetPartName(Int_t i)               { return fPartName[i]; }
  static  Char_t  *GetPartSymb(Int_t i)               { return fPartSymb[i]; }

          void     SetPartName(Int_t i, Char_t *name) { fPartName[i] = name; }
          void     SetPartSymb(Int_t i, Char_t *symb) { fPartSymb[i] = symb; }

 protected:

  virtual void     Init() = 0;
  virtual Int_t    GetModelID(Int_t mom, Int_t spec, Int_t plane) const = 0;

 private:

  AliTRDCalPID(const AliTRDCalPID& pd);
  AliTRDCalPID    &operator=(const AliTRDCalPID &c);

 protected:

  static  Char_t   *fPartName[5];          //! Names of particle species
  static  Char_t   *fPartSymb[5];          //! Symbols of particle species

  static  Float_t   fTrackMomentum[kNMom]; //  Track momenta for which response functions are available
  TObjArray        *fModel;                //  Model for probability estimate

  ClassDef(AliTRDCalPID, 3)                //  Base class for TRD PID methods

};
#endif