Helpers.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Helpers.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include <string>

#include "TEfficiency.h"
#include "TFitResult.h"
#include "TFitResultPtr.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TProfile.h"
#include "TROOT.h"

namespace ActsExamples {

namespace PlotHelpers {
struct Binning {
  Binning(){};

  Binning(std::string bTitle, int bins, float bMin, float bMax)
      : title(bTitle), nBins(bins), min(bMin), max(bMax){};

  std::string title;  
  int nBins;          
  float min;          
  float max;          
};

TH1F* bookHisto(const char* histName, const char* histTitle,
                const Binning& varBinning);

TH2F* bookHisto(const char* histName, const char* histTitle,
                const Binning& varXBinning, const Binning& varYBinning);

void fillHisto(TH1F* hist, float value, float weight = 1.0);

void fillHisto(TH2F* hist, float xValue, float yValue, float weight = 1.0);

void anaHisto(TH1D* inputHist, int j, TH1F* meanHist, TH1F* widthHist);

TEfficiency* bookEff(const char* effName, const char* effTitle,
                     const Binning& varBinning);

TEfficiency* bookEff(const char* effName, const char* effTitle,
                     const Binning& varXBinning, const Binning& varYBinning);

void fillEff(TEfficiency* efficiency, float value, bool status);

void fillEff(TEfficiency* efficiency, float xValue, float yValue, bool status);

TProfile* bookProf(const char* profName, const char* profTitle,
                   const Binning& varXBinning, const Binning& varYBinning);

void fillProf(TProfile* profile, float xValue, float yValue,
              float weight = 1.0);

}  // namespace PlotHelpers

}  // namespace ActsExamples