Changeset 5869 for trunk/MagicSoft/Mars/manalysis

Timestamp:

01/17/05 11:54:30 (20 years ago)

Author:

tbretz

Message:

*** empty log message ***

Location:

trunk/MagicSoft/Mars/manalysis

Files:

• MEnergyEstimate.cc (modified) (3 diffs)
• MEnergyEstimate.h (modified) (2 diffs)

trunk/MagicSoft/Mars/manalysis/MEnergyEstimate.cc

@@ -25 +25 @@
 
 /////////////////////////////////////////////////////////////////////////////
-//                                                                         //
-// MEnergyEstimate                                                         //
-//                                                                         //
-// Task to estimate the energy                                             //
-//                                                                         //
-//                                                                         //
+//
+// MEnergyEstimate
+//
+// Task to estimate the energy by a rule, eg:
+//
+// MEnergyEstimate est;
+// est.SetRule("0.5 + (1.1*MHillas.fLength) + (2.2*MHillasSrc.fDist) + (3.3*MHillas.fSize) +"
+//             "(4.4*MHillas.fSize*MHillas.fLength) + (5.5*MHillasSrc.fDist*MHillas.fLength)");
+//
+// For description of rules, see MDataChain.
+//
+// The default rule is "MMcEvt.fEnergy"
+//
+// Output:
+//   MEnergyEst
+//
 /////////////////////////////////////////////////////////////////////////////
 #include "MEnergyEstimate.h"
@@ -36 +46 @@
 #include "MParList.h"
 
-#include "MMcEvt.hxx"
-#include "MHillas.h"
+#include "MDataChain.h"
 #include "MEnergyEst.h"
 
@@ -47 +56 @@
 using namespace std;
 
+// --------------------------------------------------------------------------
+//
+// Default constructor. Initialize fData with default rule "MMcEvt.fEnergy"
+//
+MEnergyEstimate::MEnergyEstimate(const char *name, const char *title)
+    : fData(0), fEnergy(0)
+{
+    fName  = name  ? name  : "MEnergyEstimate";
+    fTitle = title ? title : "Task to estimate the energy by a rule";
+
+    fData = new MDataChain("MMcEvt.fEnergy");
+}
 
 // --------------------------------------------------------------------------
 //
-// Default constructor.
+// delete fData
 //
-MEnergyEstimate::MEnergyEstimate(const char *name, const char *title)
+MEnergyEstimate::~MEnergyEstimate()
 {
-    fName  = name  ? name  : "MEnergyEstimate";
-    fTitle = title ? title : "Task to estimate the energy";
-
-    AddToBranchList("MMcEvt.fEnergy");
+    delete fData;
 }
 
+// --------------------------------------------------------------------------
+//
+// Delete fData. Initialize a new MDataChain with rule.
+// Returns if fData->IsValid()
+//
+Bool_t MEnergyEstimate::SetRule(const char *rule)
+{
+    delete fData;
+    fData = new MDataChain(rule);
+
+    return fData->IsValid();
+}
+
+// --------------------------------------------------------------------------
+//
+// Forwards SetVariables to fData to allow optimizations.
+//
+void MEnergyEstimate::SetVariables(const TArrayD &arr)
+{
+    fData->SetVariables(arr);
+}
+
+// --------------------------------------------------------------------------
+//
+// FindCreate "MEnergyEst"
+// PreProcess fData.
+//
 Int_t MEnergyEstimate::PreProcess(MParList *plist)
 {
-   fHillas = (MHillas*)plist->FindObject("MHillas");
-   if (!fHillas)
-   {
-       *fLog << err << dbginf << "MHillas not found... aborting." << endl;
-       return kFALSE;
-   }
+    fEnergy = (MEnergyEst*)plist->FindCreateObj("MEnergyEst");
+    if (!fEnergy)
+        return kFALSE;
 
-   fMcEvt = (MMcEvt*)plist->FindObject("MMcEvt");
-   if (!fMcEvt)
-   {
-       *fLog << err << dbginf << "MMcEvt not found... aborting." << endl;
-       return kFALSE;
-   }
+    if (!fData->PreProcess(plist))
+        return kFALSE;
 
-   fEnergy = (MEnergyEst*)plist->FindCreateObj("MEnergyEst");
-   if (!fEnergy)
-      return kFALSE;
+    *fLog << inf << "Rule for energy estimation: " << fData->GetRule() << endl;
 
-   return kTRUE;    
+    return kTRUE;
 }
 
+// --------------------------------------------------------------------------
+//
+// Get value from fData and set it to fEnergy. SetReadyToSave for fEnergy.
+// Return kCONTINUE if value is NaN (Not a Number)
+//
 Int_t MEnergyEstimate::Process()
 {
-  //fEnergy->SetEnergy(fHillas->GetSize());
-  fEnergy->SetEnergy(fMcEvt->GetEnergy());
-  return kTRUE;
+    const Double_t val = fData->GetValue();
+    if (TMath::IsNaN(val))
+        return kCONTINUE;
+
+    fEnergy->SetEnergy(val);
+    fEnergy->SetReadyToSave();
+    return kTRUE;
 }

trunk/MagicSoft/Mars/manalysis/MEnergyEstimate.h

@@ -6 +6 @@
 #endif
 
-class MMcEvt;
-class MHillas;
+class MData;
 class MEnergyEst;
 
@@ -13 +12 @@
 {
 private:
-    MMcEvt     *fMcEvt;
-    MHillas    *fHillas;
+    MData      *fData;    //->
     MEnergyEst *fEnergy;
 
 public:
     MEnergyEstimate(const char *name=NULL, const char *title=NULL);
+    ~MEnergyEstimate();
+
+    Bool_t SetRule(const char *rule);
 
     Int_t PreProcess(MParList *plist);
     Int_t Process();
 
-    ClassDef(MEnergyEstimate, 0) // Task to copy the MC energy (preliminary)
+    void SetVariables(const TArrayD &);
+
+    ClassDef(MEnergyEstimate, 0) // Task to estimate the energy by a rule
 };