Changeset 893 for trunk/MagicSoft/Mars/mmontecarlo

Timestamp:

07/23/01 13:54:06 (23 years ago)

Author:

tbretz

Message:

*** empty log message ***

Location:

trunk/MagicSoft/Mars/mmontecarlo

Files:

• MMcThresholdCalc.cc (modified) (5 diffs)
• MMcThresholdCalc.h (modified) (2 diffs)

trunk/MagicSoft/Mars/mmontecarlo/MMcThresholdCalc.cc

@@ -58 +58 @@
 // Default Constructor.
 //
-MMcThresholdCalc::MMcThresholdCalc(const UInt_t dim, const char* name,
+// Specify the number of trigger conditions you want to use.
+// The default is 0.
+//
+// dim < 0: use only condition number dim (eg "MMcTrig;3")
+// dim = 0: use only condition without a number ("MMcTrig")
+// dim > 0: use conditions up to dim
+//
+MMcThresholdCalc::MMcThresholdCalc(const Int_t dim, const char* name,
                                    const char* title) : fDimension(dim)
 {
     *fName  = name  ? name  : "MMcThresholdCalc";
     *fTitle = title ? title : "Task to calculate the energy threshold from Monte Carlo";
-
-    // Arrays of MMcTrig* and MHMcEnergy* are created in order to be 
-    // able to work with root files with several trigger conditions.
-    fMcTrig     = new MMcTrig*[fDimension];
-    fHMcEnergy  = new MHMcEnergy*[fDimension];
-    fMustDelete = new Bool_t[fDimension];
-
-    for (unsigned int i=0; i<fDimension; i++)
-        fMustDelete[i]=kFALSE;
 }
 
 // -------------------------------------------------------------------------
 //
-// Default Destructor.
+// Destructor.
 //
 MMcThresholdCalc::~MMcThresholdCalc()
 {
-    for (unsigned int i=0; i<fDimension; i++)
-        if (fMustDelete[i])
-            delete fHMcEnergy[i];
+    if (fMcTrig)
+        delete fMcTrig;
 
-    delete fMcTrig;
-    delete fHMcEnergy;
-    delete fMustDelete;
+    if (fEnergy)
+        delete fEnergy;
 }
 
@@ -95 +91 @@
 Bool_t MMcThresholdCalc::PreProcess(MParList* pList)
 {
+    //
     // This task takes into accout if the root file has one trigger
     // condition (MMcTrig) or severl of them (MMcTrig;#.)
@@ -105 +102 @@
     }
 
-    char auxname[15]="MMcTrig"; // string to write container names
 
-    for (unsigned int i=0; i<fDimension; i++)
+    const UInt_t from = fDimension<=0 ? -fDimension : 0;
+
+    if (fDimension<=0)
+        fDimension = -fDimension;
+
+    fMcTrig = new TObjArray(pList->FindObjectList("MMcTrig", from, fDimension));
+    if (fMcTrig->GetEntriesFast() != fDimension)
     {
-        if (fDimension>1)
-            sprintf(auxname+7, ";%i", i+1);
-
-        fMcTrig[i] = (MMcTrig*)pList->FindObject(auxname);
-        if (fMcTrig[i])
-            continue;
-
-        *fLog << dbginf << "'MMcTrig";
-        if (fDimension>1)
-            *fLog << ";" << i+1;
-        *fLog << "' not found... aborting." << endl;
-
+        *fLog << dbginf << "Error: Not all requested MMcTrig objects are available...aborting." << endl;
         return kFALSE;
     }
 
-    strcpy(auxname, "MHMcEnergy");
-    for (unsigned int i=0; i<fDimension; i++)
-    {
-        if (fDimension>1&&i!=0)
-            sprintf(auxname+10, ";%i", i);
-
-        fHMcEnergy[i] = (MHMcEnergy*)pList->FindObject(auxname);
-        if (fHMcEnergy[i])
-            continue;
-
-        *fLog << dbginf << "'" << auxname << "' not found in list... creating." << endl;
-
-        fHMcEnergy[i] = new MHMcEnergy(fDimension>1&&i!=0 ? i : 0);
-        fMustDelete[i] = kTRUE;
-        pList->AddToList(fHMcEnergy[i]);
-    }
+    fEnergy = new TObjArray(pList->FindCreateObjList("MHMcEnergy", from, fDimension));
+    if (fMcTrig->GetEntriesFast() != fDimension)
+        return kFALSE;
 
     return kTRUE;
@@ -156 +134 @@
     const Float_t reciproc = 1./energy;
 
-    for (unsigned int i=0; i<fDimension; i++)
-        if (fMcTrig[i]->GetFirstLevel()>0)
-            fHMcEnergy[i]->Fill(lg10, reciproc);
+    for (Int_t i=0; i<fDimension; i++)
+        if (GetTrig(i)->GetFirstLevel()>0)
+            GetHEnergy(i)->Fill(lg10, reciproc);
 
     return kTRUE;
@@ -170 +148 @@
 Bool_t MMcThresholdCalc::PostProcess()
 {
-    for (unsigned int i=0; i<fDimension; i++)
+    for (Int_t i=0; i<fDimension; i++)
     {
-      Float_t peak;
-      Float_t sigma;
+        MHMcEnergy &hist = *GetHEnergy(i);
 
-      fHMcEnergy[i]->Fit(1, 3);
-      
-      peak  = fHMcEnergy[i]->GetGaussPeak();
-      sigma = fHMcEnergy[i]->GetGaussSigma();
-      fHMcEnergy[i]->Fit(peak - 2.   *sigma, peak + 2.   *sigma);
-        
-      peak  = fHMcEnergy[i]->GetGaussPeak();
-      sigma = fHMcEnergy[i]->GetGaussSigma();
-      fHMcEnergy[i]->Fit(peak - fSqrt2*sigma, peak + fSqrt2*sigma);
+        Float_t peak;
+        Float_t sigma;
+
+        hist.Fit(1, 3);
+
+        peak  = hist.GetGaussPeak();
+        sigma = hist.GetGaussSigma();
+        hist.Fit(peak - 2.   *sigma, peak + 2.   *sigma);
+
+        peak  = hist.GetGaussPeak();
+        sigma = hist.GetGaussSigma();
+        hist.Fit(peak - fSqrt2*sigma, peak + fSqrt2*sigma);
     }
     return kTRUE;

trunk/MagicSoft/Mars/mmontecarlo/MMcThresholdCalc.h

@@ -13 +13 @@
 #include "MTask.h"
 #endif
+#ifndef ROOT_TObjArray
+#include "TObjArray.h"
+#endif
 
 class MMcEvt;
@@ -21 +24 @@
 {
 private:
-    UInt_t fDimension;
+    Int_t fDimension;
 
     const MMcEvt  *fMcEvt;       // Container with Monte Carlo information
-    MMcTrig      **fMcTrig;      // Container with Monte Carlo trigger information
-    MHMcEnergy   **fHMcEnergy;   // Container where we save the energy (threshold)
 
-    Bool_t        *fMustDelete;  // delete indicator for selfcreated energy hists
+    TObjArray    *fMcTrig;       // Container with Monte Carlo trigger information
+    TObjArray    *fEnergy;       // Container where we save the energy (threshold)
 
     const static Float_t fSqrt2; // sqrt(2)
 
+    MMcTrig    *GetTrig   (UInt_t i) { return (MMcTrig*)   (*fMcTrig)[i]; }
+    MHMcEnergy *GetHEnergy(UInt_t i) { return (MHMcEnergy*)(*fEnergy)[i]; }
+
 public:
 
-    MMcThresholdCalc(const UInt_t dim = 1,
+    MMcThresholdCalc(const Int_t dim = 0,
                      const char* name = NULL, const char* title = NULL);
     ~MMcThresholdCalc();