Changeset 8892 for trunk/MagicSoft/Mars/mhbase

Timestamp:

05/19/08 15:04:36 (17 years ago)

Author:

tbretz

Message:

*** empty log message ***

Location:

trunk/MagicSoft/Mars/mhbase

Files:

• MH.cc (modified) (17 diffs)
• MH.h (modified) (1 diff)

trunk/MagicSoft/Mars/mhbase/MH.cc

@@ -1 +1 @@
 /* ======================================================================== *\
-! $Name: not supported by cvs2svn $:$Id: MH.cc,v 1.36 2007-08-25 15:30:24 tbretz Exp $
+! $Name: not supported by cvs2svn $:$Id: MH.cc,v 1.37 2008-05-19 14:04:12 tbretz Exp $
 ! --------------------------------------------------------------------------
 !
@@ -212 +212 @@
 // --------------------------------------------------------------------------
 //
-// Applies a given binning to a 1D-histogram
+// If labels are set for this axis the correct MBinning corresponding
+// to the existing label range is returned (this is necessary to
+// maintain the correct number of bins in the histogram)
+// otherwise the given binning is returned.
+//
+MBinning MH::GetBinningForLabels(TAxis &x, const MBinning *bins)
+{
+    if (!x.GetLabels())
+        return *bins;
+
+    const Int_t n = TMath::Max(x.GetLabels()->GetEntries(), 1);
+    return MBinning(n, 0, n);
+}
+
+// --------------------------------------------------------------------------
+//
+// If Labels are set this function deletes the fXbins Array from
+// the axis (which makes the axis a variable bin-size axis)
+// and sets the Nbins, Xmin and Xmax according to the number of labels.
+//
+void MH::RestoreBinningForLabels(TAxis &x)
+{
+    if (!x.GetLabels())
+        return;
+
+    const Int_t n = TMath::Max(x.GetLabels()->GetEntries(), 1);
+    x.Set(n, 0, n);
+
+    const_cast<TArrayD*>(x.GetXbins())->Set(0);
+}
+
+// --------------------------------------------------------------------------
+//
+// Applies a given binning to a 1D-histogram. In case the axis has labels
+// (e.g. GetXaxis()->GetLabels()) the binning is set according to the
+// labels.
 //
 void MH::SetBinning(TH1 *h, const MBinning *binsx)
@@ -225 +260 @@
 #endif
 
+#if ROOT_VERSION_CODE < ROOT_VERSION(5,12,00)
     // All this is reset by TAxis::Set
     const TAttAxis att(x);
@@ -247 +283 @@
     x.SetTimeDisplay(tm);
     x.SetTimeFormat(tf);
+#else
+    if (!x.GetLabels())
+        h->SetBins(binsx->GetNumBins(), binsx->GetEdges());
+#endif
+
 
 #if ROOT_VERSION_CODE < ROOT_VERSION(3,03,03)
@@ -255 +296 @@
 // --------------------------------------------------------------------------
 //
-// Applies given binnings to the two axis of a 2D-histogram
+// Applies given binnings to the two axis of a 2D-histogram.
+// In case the axis has labels (e.g. GetXaxis()->GetLabels())
+// the binning is set according to the labels.
 //
 void MH::SetBinning(TH2 *h, const MBinning *binsx, const MBinning *binsy)
@@ -261 +304 @@
     TAxis &x = *h->GetXaxis();
     TAxis &y = *h->GetYaxis();
+
+    const MBinning bx(GetBinningForLabels(x, binsx));
+    const MBinning by(GetBinningForLabels(y, binsy));
 
     //
@@ -270 +316 @@
 #endif
 
+#if ROOT_VERSION_CODE < ROOT_VERSION(5,12,00)
     // All this is reset by TAxis::Set
     const TAttAxis attx(x);
@@ -283 +330 @@
     // TH1D::fNcells must be set correctly.
     //
-    h->SetBins(binsx->GetNumBins(), 0, 1,
-               binsy->GetNumBins(), 0, 1);
+    h->SetBins(bx.GetNumBins(), 0, 1,
+               by.GetNumBins(), 0, 1);
 
     //
@@ -290 +337 @@
     // in one of the two given histograms
     //
-    x.Set(binsx->GetNumBins(), binsx->GetEdges());
-    y.Set(binsy->GetNumBins(), binsy->GetEdges());
+    x.Set(bx.GetNumBins(), bx.GetEdges());
+    y.Set(by.GetNumBins(), by.GetEdges());
 
     // All this is reset by TAxis::Set
@@ -300 +347 @@
     x.SetTimeFormat(tfx);
     y.SetTimeFormat(tfy);
+#else
+    h->SetBins(bx.GetNumBins(), bx.GetEdges(),
+               by.GetNumBins(), by.GetEdges());
+#endif
+
+    RestoreBinningForLabels(x);
+    RestoreBinningForLabels(y);
 
 #if ROOT_VERSION_CODE < ROOT_VERSION(3,03,03)
@@ -310 +364 @@
 //
 // Applies given binnings to the three axis of a 3D-histogram
+// In case the axis has labels (e.g. GetXaxis()->GetLabels())
+// the binning is set according to the labels.
 //
 void MH::SetBinning(TH3 *h, const MBinning *binsx, const MBinning *binsy, const MBinning *binsz)
@@ -319 +375 @@
     TAxis &y = *h->GetYaxis();
     TAxis &z = *h->GetZaxis();
+
+    const MBinning bx(GetBinningForLabels(x, binsx));
+    const MBinning by(GetBinningForLabels(y, binsy));
+    const MBinning bz(GetBinningForLabels(z, binsz));
 
 #if ROOT_VERSION_CODE < ROOT_VERSION(3,03,03)
@@ -326 +386 @@
 #endif
 
+#if ROOT_VERSION_CODE < ROOT_VERSION(5,12,00)
     // All this is reset by TAxis::Set
     const TAttAxis attx(x);
@@ -336 +397 @@
     const TString  tfy(y.GetTimeFormat());
     const TString  tfz(z.GetTimeFormat());
+#endif
 
     //
@@ -342 +404 @@
     // TH1D::fNcells must be set correctly.
     //
-    h->SetBins(binsx->GetNumBins(), 0, 1,
-               binsy->GetNumBins(), 0, 1,
-               binsz->GetNumBins(), 0, 1);
+    h->SetBins(bx.GetNumBins(), 0, 1,
+               by.GetNumBins(), 0, 1,
+               bz.GetNumBins(), 0, 1);
 
     //
@@ -350 +412 @@
     // in one of the two given histograms
     //
-    x.Set(binsx->GetNumBins(), binsx->GetEdges());
-    y.Set(binsy->GetNumBins(), binsy->GetEdges());
-    z.Set(binsz->GetNumBins(), binsz->GetEdges());
-
+    x.Set(bx.GetNumBins(), bx.GetEdges());
+    y.Set(by.GetNumBins(), by.GetEdges());
+    z.Set(bz.GetNumBins(), bz.GetEdges());
+
+    RestoreBinningForLabels(x);
+    RestoreBinningForLabels(y);
+    RestoreBinningForLabels(z);
+
+#if ROOT_VERSION_CODE < ROOT_VERSION(5,12,00)
     // All this is reset by TAxis::Set
     attx.Copy(x);
@@ -364 +431 @@
     y.SetTimeFormat(tfy);
     z.SetTimeFormat(tfz);
+#endif
 
 #if ROOT_VERSION_CODE < ROOT_VERSION(3,03,03)

trunk/MagicSoft/Mars/mhbase/MH.h

@@ -26 +26 @@
     Byte_t fSerialNumber;   // Serial number (eg of telecope)
     UInt_t fNumExecutions;  // Number of calls to Fill function
+
+    static MBinning GetBinningForLabels(TAxis &x, const MBinning *bins);
+    static void RestoreBinningForLabels(TAxis &x);
 
 public: