Added some plots

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

diff --git a/TRD/Cal/AliTRDCalPIDNN.h b/TRD/Cal/AliTRDCalPIDNN.h
index 523cdf8a7949e287c3679269c5a5e672a4416f26..85f8fd0217219ae8afbb0787cd857ce5d92167b3 100644 (file)
--- a/TRD/Cal/AliTRDCalPIDNN.h
+++ b/TRD/Cal/AliTRDCalPIDNN.h
@@ -14,21 +14,25 @@
 //                                                                        //
 ////////////////////////////////////////////////////////////////////////////
 
-#ifndef ALITRDCALPID_H
 #include "AliTRDCalPID.h"
-#endif
 
 class AliTRDCalPIDNN : public AliTRDCalPID
 {
-
  public:
 
+  enum{
+    kMLPscale  = 16000            // scaling of the MLP input to be smaller than 1
+  };
+
   AliTRDCalPIDNN();
   AliTRDCalPIDNN(const Text_t *name, const Text_t *title);
   virtual  ~AliTRDCalPIDNN();
   Bool_t    LoadReferences(Char_t *refFile);
   TObject  *GetModel(Int_t ip, Int_t iType, Int_t iPlane) const;
-  Double_t  GetProbability(Int_t spec, Float_t mom, Float_t *dedx, Float_t length, Int_t plane) const;
+  static Int_t GetModelID(Int_t mom, Int_t ii, Int_t plane);
+  Double_t  GetProbability(Int_t spec, Float_t mom
+                         , const Float_t * const dedx
+                         , Float_t length, Int_t plane) const;
 
  private:
 
@@ -36,7 +40,6 @@ class AliTRDCalPIDNN : public AliTRDCalPID
   AliTRDCalPIDNN &operator=(const AliTRDCalPIDNN &c);
            
   void     Init();
-  Int_t    GetModelID(Int_t mom, Int_t , Int_t) const;
 
   ClassDef(AliTRDCalPIDNN, 1) // NN PID reference manager