FitterBase.cpp File Reference

#include "Algorithms/FitterBase.h"
#include <vector>
#include <memory>
#include "TStopwatch.h"
#include "TTree.h"
#include "TGraphAsymmErrors.h"
#include "TMath.h"

Include dependency graph for FitterBase.cpp:

Go to the source code of this file.

Functions
void	WriteHistograms (TH1 *hist, const std::string &baseName)
	Helper to write histograms. More...
void	WriteHistogramsByMode (SampleHandlerInterface *sample, const std::string &suffix, const bool by_mode, const bool by_channel, const std::vector< TDirectory * > &SampleDir)
	Generic histogram writer - should make main code more palatable. More...

Function Documentation

WriteHistograms()

void WriteHistograms	(	TH1 *	hist,
		const std::string &	baseName
	)

Helper to write histograms.

Definition at line 1460 of file FitterBase.cpp.

                                                           {
// *************************
  if (!hist) return;
  hist->SetTitle(baseName.c_str());
  // Get the class name of the histogram
  TString className = hist->ClassName();
 
  // Set the appropriate axis title based on the histogram type
  if (className.Contains("TH1")) {
    hist->GetYaxis()->SetTitle("Events");
  } else if (className.Contains("TH2")) {
    hist->GetZaxis()->SetTitle("Events");
  }
  hist->Write(baseName.c_str());
}

WriteHistogramsByMode()

void WriteHistogramsByMode	(	SampleHandlerInterface *	sample,
		const std::string &	suffix,
		const bool	by_mode,
		const bool	by_channel,
		const std::vector< TDirectory * > &	SampleDir
	)

Generic histogram writer - should make main code more palatable.

Definition at line 1478 of file FitterBase.cpp.

                                                                    {
// *************************
  MaCh3Modes *modes = sample->GetMaCh3Modes();
  for (int iSample = 0; iSample < sample->GetNSamples(); ++iSample) {
    SampleDir[iSample]->cd();
    const std::string sampleName = sample->GetSampleTitle(iSample);
    for(int iDim1 = 0; iDim1 < sample->GetNDim(iSample); iDim1++) {
      std::string ProjectionName = sample->GetKinVarName(iSample, iDim1);
      std::string ProjectionSuffix = "_1DProj" + std::to_string(iDim1);
 
      // Probably a better way of handling this logic
      if (by_mode) {
        for (int iMode = 0; iMode < modes->GetNModes(); ++iMode) {
          auto modeHist = sample->Get1DVarHistByModeAndChannel(iSample, ProjectionName, iMode);
          WriteHistograms(modeHist.get(), sampleName + "_" + modes->GetMaCh3ModeName(iMode) + ProjectionSuffix + suffix);
        }
      }
 
      if (by_channel) {
        for (int iChan = 0; iChan < sample->GetNOscChannels(iSample); ++iChan) {
          auto chanHist = sample->Get1DVarHistByModeAndChannel(iSample, ProjectionName, -1, iChan); // -1 skips over mode plotting
          WriteHistograms(chanHist.get(), sampleName + "_" + sample->GetFlavourName(iSample, iChan) + ProjectionSuffix + suffix);
        }
      }
 
      if (by_mode && by_channel) {
        for (int iMode = 0; iMode < modes->GetNModes(); ++iMode) {
          for (int iChan = 0; iChan < sample->GetNOscChannels(iSample); ++iChan) {
            auto hist = sample->Get1DVarHistByModeAndChannel(iSample, ProjectionName, iMode, iChan);
            WriteHistograms(hist.get(), sampleName + "_" + modes->GetMaCh3ModeName(iMode) + "_" + sample->GetFlavourName(iSample, iChan) + ProjectionSuffix + suffix);
          }
        }
      }
 
      if (!by_mode && !by_channel) {
        auto hist = sample->Get1DVarHist(iSample, ProjectionName);
        WriteHistograms(hist.get(), sampleName + ProjectionSuffix + suffix);
        // Only for 2D and Beyond
        for (int iDim2 = iDim1 + 1; iDim2 < sample->GetNDim(iSample); ++iDim2) {
          // Get the names for the two dimensions
          std::string XVarName = sample->GetKinVarName(iSample, iDim1);
          std::string YVarName = sample->GetKinVarName(iSample, iDim2);
 
          // Get the 2D histogram for this pair
          auto hist2D = sample->Get2DVarHist(iSample, XVarName, YVarName);
 
          // Write the histogram
          std::string suffix2D = "_2DProj_" + std::to_string(iDim1) + "_vs_" + std::to_string(iDim2) + suffix;
          WriteHistograms(hist2D.get(), sampleName + suffix2D);
        }
      }
    }
  }
}