Changeset 8704 for trunk/MagicSoft/Mars/mjtrain/MJTrainEnergy.h

Timestamp:

08/24/07 09:37:31 (17 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mjtrain/MJTrainEnergy.h (modified) (1 diff)

trunk/MagicSoft/Mars/mjtrain/MJTrainEnergy.h

@@ -10 +10 @@
 class MJTrainEnergy : public MJTrainRanForest
 {
+private:
+    TString fTrainParameter;
+    TString fResultFunction;
+
 public:
-    MJTrainEnergy() { }
+    MJTrainEnergy() { SetTrainLin(); }
+
+    void SetTrainLog() { SetTrainFunc("log10(MMcEvt.fEnergy)", "pow(10, x)"); }
+    void SetTrainLin() { SetTrainFunc("MMcEvt.fEnergy", "x"); }
+
+    void SetTrainFunc(const char *par, const char *res)
+    {
+        fTrainParameter = par;
+        fResultFunction = res;
+    }
+
     Bool_t Train(const char *out, const MDataSet &set, Int_t num);