FitterBase.cpp File Reference

#include "FitterBase.h"
#include "Samples/SampleHandlerFD.h"
#include "TRandom.h"
#include "TStopwatch.h"
#include "TTree.h"
#include "TGraphAsymmErrors.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 (SampleHandlerFD *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 1348 of file FitterBase.cpp.

                                                           {
// *************************
  if (!hist) return;
  hist->SetTitle(baseName.c_str());
  hist->GetYaxis()->SetTitle("Events");
  hist->SetDirectory(nullptr);
  hist->Write(baseName.c_str());
}

WriteHistogramsByMode()

void WriteHistogramsByMode	(	SampleHandlerFD *	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 1359 of file FitterBase.cpp.

                                                                    {
// *************************
  MaCh3Modes *modes = sample->GetMaCh3Modes();
  for (int subSampleIndex = 0; subSampleIndex < sample->GetNsamples(); ++subSampleIndex) {
    SampleDir[subSampleIndex]->cd();
    std::string sampleName = sample->GetTitle();
    // Probably a better way of handling this logic
    if (by_mode) {
      for (int iMode = 0; iMode < modes->GetNModes(); ++iMode) {
        auto modeHist = sample->Get1DVarHistByModeAndChannel(sample->GetXBinVarName(), iMode);
        WriteHistograms(modeHist, sampleName + "_" + modes->GetMaCh3ModeName(iMode) + suffix);
        delete modeHist;
      }
    }
 
    if (by_channel) {
      for (int iChan = 0; iChan < sample->GetNOscChannels(); ++iChan) {
        auto chanHist = sample->Get1DVarHistByModeAndChannel(sample->GetXBinVarName(), -1, iChan); // -1 skips over mode plotting
        WriteHistograms(chanHist, sampleName + "_" + sample->GetFlavourName(iChan) + suffix);
        delete chanHist;
      }
    }
 
    if (by_mode && by_channel) {
      for (int iMode = 0; iMode < modes->GetNModes(); ++iMode) {
        for (int iChan = 0; iChan < sample->GetNOscChannels(); ++iChan) {
          auto hist = sample->Get1DVarHistByModeAndChannel(sample->GetXBinVarName(), iMode, iChan);
          WriteHistograms(hist, sampleName + "_" + modes->GetMaCh3ModeName(iMode) + "_" + sample->GetFlavourName(iChan) + suffix);
          delete hist;
        }
      }
    }
 
    if (!by_mode && !by_channel) {
      auto hist = sample->Get1DVarHistByModeAndChannel(sample->GetXBinVarName());
      WriteHistograms(hist, sampleName + suffix);
      delete hist;
    }
  }
}