Adding 2 triggers to list of known unsupported triggers

[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDpidRefMakerNN.h

diff --git a/PWG1/TRD/AliTRDpidRefMakerNN.h b/PWG1/TRD/AliTRDpidRefMakerNN.h
index fbf805533b376399e6fdbe59ff5954f5f5c4c3ee..fea1044be8923c914ea63a04ba8d29bdcb78ed3d 100644 (file)
--- a/PWG1/TRD/AliTRDpidRefMakerNN.h
+++ b/PWG1/TRD/AliTRDpidRefMakerNN.h
@@ -18,6 +18,9 @@
 #ifndef ALITRDPIDREFMAKER_H
 #include "AliTRDpidRefMaker.h"
 #endif
+#ifndef ALITRDCALPID_H
+#include "Cal/AliTRDCalPID.h"
+#endif
 
 class TEventList;
 class TMultiLayerPerceptron;
@@ -39,10 +42,9 @@ public:
    ,kMinStat    = 50     // minimum statistics/PID bin
   };
   AliTRDpidRefMakerNN();
-
+  AliTRDpidRefMakerNN(const char *n);
   virtual ~AliTRDpidRefMakerNN();
   
-  void    CreateOutputObjects();
   Int_t   GetEpochs() const {return fEpochs;};
   Int_t   GetMinTrain() const {return fMinTrain;};
   Int_t   GetTrainMomBin() const {return fTrainMomBin;};
@@ -56,11 +58,12 @@ public:
   void    SetDoTraining(Bool_t train) {fDoTraining = train;};
   void    SetContinueTraining(Bool_t continTrain) {fContinueTraining = continTrain;};
   void    SetTrainPath(Int_t path) {fTrainPath = path;};
-  Bool_t    LoadFile(const Char_t *InFileNN);
+  Bool_t  LoadFile(const Char_t *InFileNN);
   void    SetScaledEdx(Float_t s) {fScale = s;};
 
-  void    MakeTrainingLists(Int_t mombin = 0);                                 // build the training and the test list
-  void    MonitorTraining(Int_t mombin);                       // monitor training process
+  Bool_t  MakeTrainingSample();                     // convert AnalysisResults.root to training file
+  void    MakeTrainingLists(Int_t mombin = 0);      // build the training and the test list
+  void    MonitorTraining(Int_t mombin);            // monitor training process
 
 protected:
   void MakeRefs(Int_t mombin);                         // train the neural networks for a given momentum bin
@@ -68,6 +71,7 @@ protected:
 private:
   AliTRDpidRefMakerNN(const AliTRDpidRefMakerNN&);              // not implemented
   AliTRDpidRefMakerNN& operator=(const AliTRDpidRefMakerNN&);   // not implemented
+  void    MakeTrainTestTrees();
 
   TEventList *fTrain[AliTRDCalPID::kNMom];          // Training list for each momentum
   TEventList *fTest[AliTRDCalPID::kNMom];           // Test list for each momentum
@@ -81,13 +85,13 @@ private:
   Bool_t        fContinueTraining;         // checks if training from an older run should be continued
   Int_t         fTrainPath;                // sets the path for continuing the training
 
-  Float_t       fScale;
+  Float_t       fScale;                    // scaling factor
   Int_t         fLy;                       // TRD layer
   Int_t         fNtrkl;                    // No. tracklets
-  TTree         *fTrainData[AliTRDCalPID::kNMom]; 
-  TFile         *fRef;
+  TTree         *fTrainData[AliTRDCalPID::kNMom];//Tree for reference data for all momentum bins 
+  TFile         *fRef;                     //file containing reference data
 
-  ClassDef(AliTRDpidRefMakerNN, 3); // TRD reference  maker for NN
+  ClassDef(AliTRDpidRefMakerNN, 3);        // TRD reference  maker for NN
 };
 
 #endif