TRD/Cal/AliTRDCalPIDNN.h

   1 #ifndef ALITRDCALPIDNN_H
   2 #define ALITRDCALPIDNN_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ////////////////////////////////////////////////////////////////////////////
   9 //                                                                        //
  10 // PID distributions for the NN method                                    //
  11 //                                                                        //
  12 // Author:                                                                //
  13 // Alex Wilk <wilka@uni-muenster.de>                                      //
  14 //                                                                        //
  15 ////////////////////////////////////////////////////////////////////////////
  16
  17 #ifndef ALITRDCALPID_H
  18 #include "AliTRDCalPID.h"
  19 #endif
  20
  21 class AliTRDCalPIDNN : public AliTRDCalPID
  22 {
  23  public:
  24
  25   enum{
  26     kMLPscale  = 16000            // scaling of the MLP input to be smaller than 1
  27   };
  28
  29   AliTRDCalPIDNN();
  30   AliTRDCalPIDNN(const Text_t *name, const Text_t *title);
  31   virtual  ~AliTRDCalPIDNN();
  32   Bool_t    LoadReferences(Char_t *refFile);
  33   TObject  *GetModel(Int_t ip, Int_t iType, Int_t iPlane) const;
  34   Double_t  GetProbability(Int_t spec, Float_t mom, Float_t *dedx, Float_t length, Int_t plane) const;
  35
  36  private:
  37
  38   AliTRDCalPIDNN(const AliTRDCalPIDNN &pd);
  39   AliTRDCalPIDNN &operator=(const AliTRDCalPIDNN &c);
  40
  41   void     Init();
  42   Int_t    GetModelID(Int_t mom, Int_t , Int_t) const;
  43
  44   ClassDef(AliTRDCalPIDNN, 1) // NN PID reference manager
  45
  46 };
  47 #endif