Changeset 9821 for trunk/Mars/mhbase/MH3.cc

Timestamp:

08/10/10 12:27:00 (14 years ago)

Author:

tbretz

Message:

Added the possibility to set a conversion function which is applied
before the histogram is displayed.

File:

• trunk/Mars/mhbase/MH3.cc (modified) (14 diffs)

trunk/Mars/mhbase/MH3.cc

@@ -18 +18 @@
 !   Author(s): Thomas Bretz  2002 <mailto:tbretz@astro.uni-wuerzburg.de>
 !
-!   Copyright: MAGIC Software Development, 2000-2008
+!   Copyright: MAGIC Software Development, 2000-2010
 !
 !
@@ -143 +143 @@
 //   + MBinning fBins[3]
 //
+// Class Version 5:
+// ----------------
+//   + TFormula *fConversion
+//
 /////////////////////////////////////////////////////////////////////////////
 #include "MH3.h"
@@ -151 +155 @@
 
 #include <TMath.h>
+#include <TFormula.h>
 
 #include <THashList.h>
@@ -200 +205 @@
     fScale[2] = 1;
 
+    fConversion = NULL;
+
     fName  = gsDefName;
     fTitle = gsDefTitle;
@@ -239 +246 @@
         fHist = new TProfile;
         fHist->SetYTitle("Average");
+        static_cast<TProfile*>(fHist)->SetErrorOption("s");
         break;
     case 2:
@@ -247 +255 @@
         fHist = new TProfile2D;
         fHist->SetZTitle("Average");
+        static_cast<TProfile2D*>(fHist)->SetErrorOption("s");
         break;
     case 3:
@@ -331 +340 @@
     case -1:
         fHist = static_cast<TH1*>(new TProfile);
+        static_cast<TProfile*>(fHist)->SetErrorOption("s");
+
         break;
     }
@@ -352 +363 @@
     : fDimension(type==kHistogram?3:-2)
 {
-    fHist = type&kProfile ? static_cast<TH1*>(new TProfile2D) : static_cast<TH1*>(new TH3D);
+    if (type&kProfile)
+    {
+        fHist = static_cast<TH1*>(new TProfile2D);
+        static_cast<TProfile2D*>(fHist)->SetErrorOption("s");
+    }
+    else
+        fHist = static_cast<TH1*>(new TH3D);
+
 
     fData[0] = new MDataPhrase(memberx);
@@ -370 +388 @@
         delete fHist;
 
+    if (fConversion)
+        delete fConversion;
+
     for (int i=0; i<4; i++)
         if (fData[i])
@@ -390 +411 @@
         delete fData[3];
     fData[3] = new MDataPhrase(phrase);
+}
+
+// --------------------------------------------------------------------------
+//
+// Set a function which is applied to the histogram before it is displayed.
+// Note, that it only effects the displayed histogram.
+//
+//   e.g. SetConversion("sqrt(x)");
+//
+Bool_t MH3::SetConversion(const char *func)
+{
+    if (TString(func).IsNull())
+    {
+        delete fConversion;
+        fConversion = 0;
+        return kTRUE;
+    }
+
+    fConversion = new TFormula;
+
+    // Must have a name otherwise all axis labels disappear like a miracle
+    fConversion->SetName("ConversionFunction");
+    if (fConversion->Compile(func))
+    {
+        *fLog << err << dbginf << "Syntax Error: TFormula::Compile failed for " << func << endl;
+        delete fConversion;
+        fConversion = 0;
+        return kFALSE;
+    }
+
+    gROOT->GetListOfFunctions()->Remove(fConversion);
+
+    return kTRUE;
 }
 
@@ -845 +899 @@
 // --------------------------------------------------------------------------
 //
+// Apply the conversion function to the contents stored in fHist and
+// store the result in h.
+//
+void MH3::Convert(TH1 &h) const
+{
+    for (Int_t z=0; z<=h.GetNbinsZ()+1; z++)
+        for (Int_t y=0; y<=h.GetNbinsY()+1; y++)
+            for (Int_t x=0; x<=h.GetNbinsX()+1; x++)
+            {
+                h.SetBinContent(x, y, z, fConversion->Eval(fHist->GetBinContent(x, y, z)));
+                h.SetBinError(  x, y, z, fConversion->Eval(fHist->GetBinError(  x, y, z)));
+            }
+
+    if (h.InheritsFrom(TProfile::Class()))
+        for (Int_t x=0; x<=h.GetNbinsX()+1; x++)
+            static_cast<TProfile&>(h).SetBinEntries(x, 1);
+
+    if (h.InheritsFrom(TProfile2D::Class()))
+        for (Int_t x=0; x<=h.GetNbinsX()+1; x++)
+            static_cast<TProfile2D&>(h).SetBinEntries(x, 1);
+}
+
+// --------------------------------------------------------------------------
+//
 // FIXME
 //
@@ -853 +931 @@
     if (TMath::Abs(fDimension)==2)
         MH::SetPalette("pretty");
+
+    if (fConversion)
+    {
+        TH1 *h = 0;
+        if ((h=dynamic_cast<TH1*>(gPad->FindObject(fHist->GetName()))))
+            Convert(*h);
+    }
 
     const TString pfx(MString::Format("%sProfX", fHist->GetName()));
@@ -963 +1048 @@
     }
 
+
+    TH1 *h = fHist;
+
+    if (fConversion)
+    {
+        h = static_cast<TH1*>(fHist->Clone());
+        h->SetDirectory(0);
+        h->SetBit(kCanDelete);
+
+        Convert(*h);
+    }
+
     // FIXME: We may have to remove all our own options from str!
     if (!only)
-        fHist->Draw(str);
+        h->Draw(str);
 
     AppendPad();
@@ -972 +1069 @@
     if (profx)
     {
-        const TString pfx(MString::Format("%sProfX", fHist->GetName()));
+        const TString pfx(MString::Format("%sProfX", h->GetName()));
 
         if (same && (p=dynamic_cast<TProfile*>(gPad->FindObject(pfx))))
             *fLog << warn << "TProfile " << pfx << " already in pad." << endl;
 
-        p = ((TH2*)fHist)->ProfileX(pfx, -1, -1, "s");
+        p = ((TH2*)h)->ProfileX(pfx, -1, -1, "s");
         p->UseCurrentStyle();
-        p->SetLineColor(blue ? kBlue : fHist->GetLineColor());
+        p->SetLineColor(blue ? kBlue : h->GetLineColor());
         p->SetBit(kCanDelete);
         p->SetDirectory(NULL);
-        p->SetXTitle(fHist->GetXaxis()->GetTitle());
-        p->SetYTitle(fHist->GetYaxis()->GetTitle());
+        p->SetXTitle(h->GetXaxis()->GetTitle());
+        p->SetYTitle(h->GetYaxis()->GetTitle());
         p->Draw(only&&!same?"":"same");
     }
     if (profy)
     {
-        const TString pfy(MString::Format("%sProfY", fHist->GetName()));
+        const TString pfy(MString::Format("%sProfY", h->GetName()));
 
         if (same && (p=dynamic_cast<TProfile*>(gPad->FindObject(pfy))))
             *fLog << warn << "TProfile " << pfy << " already in pad." << endl;
 
-        p = ((TH2*)fHist)->ProfileY(pfy, -1, -1, "s");
+        p = ((TH2*)h)->ProfileY(pfy, -1, -1, "s");
         p->UseCurrentStyle();
-        p->SetLineColor(blue ? kBlue : fHist->GetLineColor());
+        p->SetLineColor(blue ? kBlue : h->GetLineColor());
         p->SetBit(kCanDelete);
         p->SetDirectory(NULL);
-        p->SetYTitle(fHist->GetXaxis()->GetTitle());
-        p->SetXTitle(fHist->GetYaxis()->GetTitle());
+        p->SetYTitle(h->GetXaxis()->GetTitle());
+        p->SetXTitle(h->GetYaxis()->GetTitle());
         p->Draw(only&&!same?"":"same");
     }