CbmQaHist.h

Go to the documentation of this file.

/* Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov [committer] */
 
 
#ifndef CbmQaHist_H
#define CbmQaHist_H
 
#include "CbmQaCanvas.h"
 
#include <Logger.h>
 
#include <TFitResultPtr.h>
#include <TPaveStats.h>
#include <TStyle.h>
#include <TVirtualPad.h>
 
// The TH* headers are needed here for the ROOT linker
 
#include <TF1.h>
#include <TH1D.h>
#include <TH1F.h>
#include <TH1I.h>
#include <TProfile.h>
#include <TProfile2D.h>
 
template<class HistTypeT>
class CbmQaHist : public HistTypeT {
 public:
  CbmQaHist() : HistTypeT()
  {
    // we don't call SetOptStatFit() here since it will call Clone()
    // which calls this default constructor in an endless recursion
    if (gStyle) {
      fOptStat = gStyle->GetOptStat();
      fOptFit  = gStyle->GetOptFit();
    }
    this->SetLineWidth(2);
  }
 
  CbmQaHist(const CbmQaHist& h) : HistTypeT(h) { SetOptStatFit(h.fOptStat, h.fOptFit); }
 
  template<typename... Types>
  CbmQaHist(Types... args) : HistTypeT(args...)
  {
    if (gStyle) {
      SetOptStatFit(gStyle->GetOptStat(), gStyle->GetOptFit());
    }
    this->SetLineWidth(2);
  }
 
  ~CbmQaHist() {}
 
  template<typename... Types>
  TFitResultPtr Fit(Types... args)
  {
    TVirtualPad* padsav = gPad;
    CbmQaCanvas::GetDummyCanvas().cd();
    this->Sumw2();
    auto ret = HistTypeT::Fit(args...);
    CbmQaCanvas::GetDummyCanvas().Clear();
 
    // make the output look nice
 
    if (padsav) padsav->cd();
    auto* f = this->GetFunction("gaus");
    if (f) {
      f->SetParNames("Peak", "#mu", "#sigma");
      f->SetLineColor(kRed);
      f->SetLineWidth(3);
      TPaveStats* st = (TPaveStats*) this->FindObject("stats");
      if (!st) {
        LOG(fatal) << "CbmQaHist: can not access statistics of histogram with name \"" << this->GetName() << '\"';
      }
      else {
        st->SetX1NDC(0.6);
        st->SetX2NDC(0.940);
        st->SetY1NDC(0.5);
        st->SetY2NDC(0.930);
        st->SetOptStat(111110);
        st->SetOptFit(10001);
      }
    }
    return ret;
  }
 
  void SetOptStat(Int_t stat = 1) { SetOptStatFit(stat, fOptFit); }
 
  void SetOptFit(Int_t fit = 1) { SetOptStatFit(fOptStat, fit); }
 
  void SetOptStatFit(int stat, int fit)
  {
    // the only way to create and auto-size the stat window is to draw the histogram
 
    fOptStat = stat;
    fOptFit  = fit;
    if (!gStyle) {
      return;
    }  // should not happen
 
    TVirtualPad* savePad = gPad;
    int saveStat         = gStyle->GetOptStat();
    int saveFit          = gStyle->GetOptFit();
 
    CbmQaCanvas::GetDummyCanvas().cd();
    gStyle->SetOptStat(fOptStat);
    gStyle->SetOptFit(fOptFit);
 
    this->SetStats(0);  // remove the old stat window
    this->SetStats(1);  // set the flag to create the stat window during Draw()
    this->Draw();
    CbmQaCanvas::GetDummyCanvas().Update();
    CbmQaCanvas::GetDummyCanvas().Clear();
 
    // restore the environment
    gStyle->SetOptStat(saveStat);
    gStyle->SetOptFit(saveFit);
    if (savePad) savePad->cd();
  }
 
 private:
  int fOptStat = 1;
  int fOptFit  = 0;
 
  ClassDefNV(CbmQaHist, 1);
};
 
#endif