]> git.uio.no Git - u/mrichter/AliRoot.git/blob - TRD/Cal/AliTRDCalPIDNN.h
First implementation of neural network PID
[u/mrichter/AliRoot.git] / 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
24  public:
25
26   AliTRDCalPIDNN();
27   AliTRDCalPIDNN(const Text_t *name, const Text_t *title);
28   virtual  ~AliTRDCalPIDNN();
29   Bool_t    LoadReferences(Char_t *refFile);
30   TObject  *GetModel(Int_t ip, Int_t iType, Int_t iPlane) const;
31   Double_t  GetProbability(Int_t spec, Float_t mom, Float_t *dedx, Float_t length, Int_t plane) const;
32
33  private:
34
35   AliTRDCalPIDNN(const AliTRDCalPIDNN &pd);
36   AliTRDCalPIDNN &operator=(const AliTRDCalPIDNN &c);
37            
38   void     Init();
39   Int_t    GetModelID(Int_t mom, Int_t , Int_t) const;
40
41   ClassDef(AliTRDCalPIDNN, 1) // NN PID reference manager
42
43 };
44 #endif