Changeset 9821 for trunk/Mars/mhbase

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.

Location:

trunk/Mars/mhbase

Files:

• MH3.cc (modified) (14 diffs)
• MH3.h (modified) (4 diffs)
• MHn.cc (modified) (2 diffs)
• MHn.h (modified) (1 diff)

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");
     }

trunk/Mars/mhbase/MH3.h

@@ -15 +15 @@
 class MData;
 class MBinning;
+class TFormula;
 
 class MH3 : public MH
@@ -55 +56 @@
     Byte_t      fStyleBits;      // Set the range of a histogram automatically in Finalize
 
-//    TH1        *fHistDraw;       //!
+    TFormula   *fConversion;     // Conversion function for displaying the histogram
 
     void HandleLogAxis(TAxis &axe) const;
+    void Convert(TH1 &h) const;
 
     void StreamPrimitive(ostream &out) const;
+
 
     enum {
@@ -122 +125 @@
     void SetWeight(const char *phrase);
 
+    Bool_t SetConversion(const char *func="");
+
     // Getter
     Int_t GetDimension() const { return TMath::Abs(fDimension); }
@@ -160 +165 @@
     void RecursiveRemove(TObject *obj);
 
-    ClassDef(MH3, 4) // Generalized 1/2/3D-histogram for Mars variables
+    ClassDef(MH3, 5) // Generalized 1/2/3D-histogram for Mars variables
 };
 

trunk/Mars/mhbase/MHn.cc

@@ -18 +18 @@
 !   Author(s): Thomas Bretz  2002 <mailto:tbretz@astro.uni-wuerzburg.de>
 !
-!   Copyright: MAGIC Software Development, 2000-2007
+!   Copyright: MAGIC Software Development, 2000-2010
 !
 !
@@ -485 +485 @@
 // --------------------------------------------------------------------------
 //
+// call MH3::SetConversion for the current histogram
+//
+void MHn::SetConversion(const char *func)
+{
+    if (fHist[fNum-1])
+        fHist[fNum-1]->SetConversion(func);
+}
+
+// --------------------------------------------------------------------------
+//
 // Call SetupFill for all initialized histograms
 //

trunk/Mars/mhbase/MHn.h

@@ -65 +65 @@
     void SetWeight(const char *phrase);
 
+    // Set a conversion function for displaying the histogram
+    void SetConversion(const char *func="");
+
     // MH
     Bool_t SetupFill(const MParList *pList);