HistContainer.cxx

Go to the documentation of this file.

#include "TCanvas.h"
#include "TClass.h"
#include "TColor.h"
#include "TPaletteAxis.h"
#include "TROOT.h"

#include "travex/utils.h"
#include "travex/HistContainer.h"

using namespace tvx;


HistContainer::HistContainer(const std::string name, TDirectory* motherDir, const std::string option) :
   TDirectoryFile(name.c_str(), name.c_str(), option.c_str(), motherDir),
   fHs()
{
   cd();
}


void HistContainer::Add(TH1* hist)
{
   if (!hist || std::string(hist->GetName()).empty() ) {
      TVX_WARNING("Cannot add invalid histogram");
      return;
   }

   std::string hist_name(hist->GetName());

   if ( FindHist(hist_name) )
      TVX_WARNING("Replacing existing histogram %s", hist_name.c_str());

   fHs[hist_name].reset(hist);
}


void HistContainer::SaveAllAs(std::string prefix, std::string img_format)
{
   static TCanvas canvas("canvas", "canvas", 1200, 600);
   canvas.UseCurrentStyle();
   canvas.SetGridx(true);
   canvas.SetGridy(true);

   for (auto& iHist : fHs)
   {
      std::string histName = iHist.first;
      std::unique_ptr<TH1>& hist = iHist.second;

      char* opts = (char*) hist->GetOption();

      if (strstr(opts, "XX")) canvas.SetLogx(true);
      else canvas.SetLogx(false);

      if (strstr(opts, "XY")) canvas.SetLogy(true);
      else canvas.SetLogy(false);

      if (strstr(opts, "XZ")) canvas.SetLogz(true);
      else canvas.SetLogz(false);

      hist->Draw();

      TColor *color = nullptr;
      float r, g, b;

      if (strstr(opts, "whit_zro")) {
         hist->GetZaxis()->SetRangeUser(-1,1);
         hist->SetContour(11);
         gPad->Update();
         TPaletteAxis *palette = (TPaletteAxis*) hist->GetListOfFunctions()->FindObject("palette");
         color = gROOT->GetColor( palette->GetValueColor(0) );
         color->GetRGB(r, g, b);
         color->SetRGB(255, 255, 255);
      }

      // Now check if there are other associated objects like functions and graphs
      TList* sub_list = hist->GetListOfFunctions();
      TIter  next(sub_list);

      while ( TObject *iObj = (TObject*) next() )
      {
         if ( !iObj ) continue;
         iObj->Draw();
      }

      std::string sFileName = prefix + "/" + histName + "." + img_format;
      canvas.SaveAs(sFileName.c_str());

      // Restore modified color
      if (color) color->SetRGB(r, g, b);
   }
}