#include <ThinSliceDataSet.h>

Public Member Functions
	ThinSliceDataSet ()

	ThinSliceDataSet (const std::vector< double > &incident_bins, const std::vector< fhicl::ParameterSet > &selections)

	~ThinSliceDataSet ()

const std::map< int, std::string > &	GetSelectionNames () const

std::string &	GetSelectionName (int id)

std::map< int, TH1 * > &	GetSelectionHists ()

std::map< int, TH1 * > &	GetRebinnedSelectionHists ()

TH1 *	GetSelectionHist (int id)

TH1D &	GetIncidentHist ()

TH1D &	GetRebinnedIncidentHist ()

TH1 *	GetRebinnedSelectionHist (int id)

void	FillIncidentHist (const std::vector< double > &vals)

void	FillSelectionHist (int id, double val)

template<size_t N>
void	FillSelectionHist (int id, const double(&vals)[N])

void	MakeRebinnedHists ()

void	Refill1DRebinned ()

void	GetCumulatives ()

void	GenerateStatFluctuation ()

void	FillHistsFromSamples (const std::map< int, std::vector< std::vector< ThinSliceSample >>> &samples, double &flux)

Private Member Functions
void	Rebin1D (TH1 sel_hist, TH1 rebinned)

void	Rebin2D (TH1 sel_hist, TH1 rebinned)

void	Rebin3D (TH1 sel_hist, TH1 rebinned)

Private Attributes
std::map< int, TH1 * >	fSelectionHists

TH1D	fIncidentHist

std::map< int, TH1 * >	fSelectionHistsRebinned

TH1D	fIncidentHistRebinned

bool	fMadeRebinned = false

std::map< int, std::string >	fSelectionNames

std::vector< std::pair< std::pair< int, int >, double > >	fCumulatives

TRandom3	fRNG = TRandom3(0)

double	fTotal

Detailed Description

Definition at line 16 of file ThinSliceDataSet.h.

Constructor & Destructor Documentation

protoana::ThinSliceDataSet::ThinSliceDataSet ( )

inline

Definition at line 18 of file ThinSliceDataSet.h.

18 {};

protoana::ThinSliceDataSet::ThinSliceDataSet	(	const std::vector< double > &	incident_bins,
		const std::vector< fhicl::ParameterSet > &	selections
	)

Definition at line 4 of file ThinSliceDataSet.cxx.

                                                      {
   fIncidentHist = TH1D("Data_incident_hist",
                            "Data;Reconstructed KE (MeV)",
                            incident_bins.size() - 1,
                            &incident_bins[0]);
   for (auto it = selections.begin(); it != selections.end(); ++it) {
     fSelectionNames[it->get<int>("ID")] = it->get<std::string>("Name");
     std::string sel_name = "Data_selected_" + it->get<std::string>("Name") +
                            "_hist";
     std::vector<std::vector<double>> selected_bins =
         it->get<std::vector<std::vector<double>>>("RecoBins");
 
     std::vector<std::string> titles =
         it->get<std::vector<std::string>>("AxisTitles");
     TString title = "Data";
     for (auto & t : titles) {
       title += ";" + t; 
     }
 
     if (selected_bins.size() == 1) {
       fSelectionHists[it->get<int>("ID")] = new TH1D(
           sel_name.c_str(), title/*.c_str()"Data;Reconstructed KE (MeV)"*/,
           selected_bins[0].size() - 1, &selected_bins[0][0]);
     }
     else if (selected_bins.size() == 2) {
       fSelectionHists[it->get<int>("ID")] = new TH2D(
           sel_name.c_str(), title/*.c_str()"Data;Reconstructed KE (MeV)"*/,
           selected_bins[0].size() - 1, &selected_bins[0][0],
           selected_bins[1].size() - 1, &selected_bins[1][0]);
     }
     else if (selected_bins.size() == 3) {
       fSelectionHists[it->get<int>("ID")] = new TH3D(
           sel_name.c_str(), title/*.c_str()"Data;Reconstructed KE (MeV)"*/,
           selected_bins[0].size() - 1, &selected_bins[0][0],
           selected_bins[1].size() - 1, &selected_bins[1][0],
           selected_bins[2].size() - 1, &selected_bins[2][0]);
     }
     /*else {
      * throw
      * }*/
   }
 }

protoana::ThinSliceDataSet::~ThinSliceDataSet ( )

inline

Definition at line 21 of file ThinSliceDataSet.h.

21 {};

Member Function Documentation

void protoana::ThinSliceDataSet::FillHistsFromSamples	(	const std::map< int, std::vector< std::vector< ThinSliceSample >>> &	samples,
		double &	flux
	)

Definition at line 250 of file ThinSliceDataSet.cxx.

                    {
 
   flux = 0.;
   for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
     it->second->Reset();
   }
 
   for (auto it = samples.begin(); it != samples.end(); ++it) {
     for (size_t i = 0; i < it->second.size(); ++i) {
       for (size_t j = 0; j < it->second[i].size(); ++j) {
         const auto & hists = it->second[i][j].GetSelectionHists();
         for (auto it2 = hists.begin(); it2 != hists.end(); ++it2) {
           fSelectionHists[it2->first]->Add(it2->second);
           flux += it2->second->Integral();
         }
       }
     }
   }
 }

void protoana::ThinSliceDataSet::FillIncidentHist ( const std::vector< double > & vals )

inline

Definition at line 55 of file ThinSliceDataSet.h.

                                                         {
     for (size_t i = 0; i < vals.size(); ++i) {
       fIncidentHist.Fill(vals.at(i));
     }
   };

void protoana::ThinSliceDataSet::FillSelectionHist	(	int	id,
		double	val
	)

inline

Definition at line 61 of file ThinSliceDataSet.h.

                                              {
     if (fSelectionHists.find(id) != fSelectionHists.end()) {
       fSelectionHists.at(id)->Fill(val);
     }
   };

template<size_t N>

void protoana::ThinSliceDataSet::FillSelectionHist	(	int	id,
		const double(&)	vals[N]
	)

inline

Definition at line 67 of file ThinSliceDataSet.h.

                                                                                {
     if (N < 1 || N > 3) {
       std::string message = "Error: trying to fill hists with too many values";
       message += std::to_string(N);
       throw std::runtime_error(message);
     }
 
     if (fSelectionHists.find(id) != fSelectionHists.end()) {
       if (N == 1) {
         fSelectionHists.at(id)->Fill(vals[0]);
       }
       else if (N == 2) {
         ((TH2D*)fSelectionHists.at(id))->Fill(vals[0], vals[1]);
       }
       else if (N == 3) {
         ((TH3D*)fSelectionHists.at(id))->Fill(vals[0], vals[1], vals[2]);
       }
     }
   }

void protoana::ThinSliceDataSet::GenerateStatFluctuation ( )

Definition at line 192 of file ThinSliceDataSet.cxx.

                                                        {
 
   //bool retry = true;
   //while (retry) {
     for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
       it->second->Reset();
       //fSelectionHistsRebinned[it->first]->Reset();
     }
 
     for (int i = 0; i < fTotal; ++i) {
       double r = fRNG.Uniform();
       std::pair<int, int> bin;
       for (size_t j = 0; j < fCumulatives.size(); ++j) {
         //std::cout << fCumulatives[j].second << " " <<  r <<
         //             " "  << fCumulatives[j].second - r << std::endl;
         if ((fCumulatives[j].second - r) > 0.) {
           bin = fCumulatives[j].first;
         }
         else {
           break;
         }
       }
       //std::cout << "Found bin: " << bin.first << " " << bin.second << std::endl;
       fSelectionHists[bin.first]->AddBinContent(bin.second); 
       //fSelectionHistsRebinned[bin.first]->AddBinContent(bin.second); 
     }
 
     //bool good = true;
     /*
     double new_total = 0.;
     for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
       for (int i = 1; i <= it->second->GetNbinsX(); ++i) {
         if (it->second->GetBinContent(i) < 1.) {
           std::cout << "DataSet fluctuation: Found bin with 0 content. Adding" << std::endl;
           it->second->AddBinContent(i);
           //fSelectionHistsRebinned[it->first]->AddBinContent(i);
           //good = false;
         }
         new_total += it->second->GetBinContent(i);
       }
     }
 
     for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
       it->second->Scale(new_total/fTotal);
       //fSelectionHistsRebinned[it->first]->Scale(new_total/fTotal);
     }
     */
 
     Refill1DRebinned();
 
     //retry = !good;
   //}
 
  // std::cout << "Bin vals: ";
  // std::cout << bin.second << " ";
  // std::cout << std::endl;
 }

void protoana::ThinSliceDataSet::GetCumulatives ( )

inline

Definition at line 90 of file ThinSliceDataSet.h.

                         {
     fTotal = 0;
     std::cout << "Making cumulatives" << std::endl;
     for (auto it = fSelectionHists.begin();
          it != fSelectionHists.end(); ++it) {
       for (int i = 1; i <= it->second->GetNbinsX(); ++i) {
         double val = it->second->GetBinContent(i);
         if (fCumulatives.size()) {
           val += fCumulatives.back().second;
         }
         fCumulatives.push_back({{it->first, i}, val});
         fTotal += it->second->GetBinContent(i);
         std::cout << fCumulatives.back().second << " " << fTotal << std::endl;
       }
     }
 
     for (auto it = fCumulatives.begin(); it != fCumulatives.end(); ++it) {
       it->second /= fTotal;
     }
 
     std::sort(fCumulatives.begin(), fCumulatives.end(),
               [](auto a, auto b){return (a.second > b.second);});
     std::cout << "N Cumulatives: " << fCumulatives.size() << std::endl;
     for (auto c : fCumulatives) {
       std::cout << c.second << std::endl;
     }
   };

TH1D& protoana::ThinSliceDataSet::GetIncidentHist ( )

inline

Definition at line 43 of file ThinSliceDataSet.h.

                            {
     return fIncidentHist;
   };

TH1D& protoana::ThinSliceDataSet::GetRebinnedIncidentHist ( )

inline

Definition at line 47 of file ThinSliceDataSet.h.

                                    {
     return fIncidentHistRebinned;
   };

TH1* protoana::ThinSliceDataSet::GetRebinnedSelectionHist ( int id )

inline

Definition at line 51 of file ThinSliceDataSet.h.

                                          {
     return fSelectionHistsRebinned.at(id);
   };

std::map<int, TH1 *>& protoana::ThinSliceDataSet::GetRebinnedSelectionHists ( )

inline

Definition at line 35 of file ThinSliceDataSet.h.

                                                    {
     return fSelectionHistsRebinned;
   };

TH1* protoana::ThinSliceDataSet::GetSelectionHist ( int id )

inline

Definition at line 39 of file ThinSliceDataSet.h.

                                  {
     return fSelectionHists.at(id);
   };

std::map<int, TH1 *>& protoana::ThinSliceDataSet::GetSelectionHists ( )

inline

Definition at line 31 of file ThinSliceDataSet.h.

                                            {
     return fSelectionHists;
   };

std::string& protoana::ThinSliceDataSet::GetSelectionName ( int id )

inline

Definition at line 27 of file ThinSliceDataSet.h.

                                        {
     return fSelectionNames.at(id);
   }

const std::map<int, std::string>& protoana::ThinSliceDataSet::GetSelectionNames ( ) const

inline

Definition at line 23 of file ThinSliceDataSet.h.

                                                            {
     return fSelectionNames;
   };

void protoana::ThinSliceDataSet::MakeRebinnedHists ( )

Definition at line 49 of file ThinSliceDataSet.cxx.

                                                  {
   if (!fMadeRebinned) {
     std::string inc_name = fIncidentHist.GetName();
     inc_name += "Rebinned";
     fIncidentHistRebinned = TH1D(inc_name.c_str(), fIncidentHist.GetTitle(),
                                  fIncidentHist.GetNbinsX(), 0, fIncidentHist.GetNbinsX());
     for (int i = 1; i <= fIncidentHist.GetNbinsX(); ++i) {
       fIncidentHistRebinned.SetBinContent(i, fIncidentHist.GetBinContent(i));
 
       double low_edge = fIncidentHist.GetXaxis()->GetBinLowEdge(i);
       double up_edge = fIncidentHist.GetXaxis()->GetBinUpEdge(i);
       std::string bin_label = (low_edge < 0. ? "< 0." :
                                (protoana::PreciseToString(low_edge, 0) + " - " +
                                 protoana::PreciseToString(up_edge, 0)));
       fIncidentHistRebinned.GetXaxis()->SetBinLabel(i, bin_label.c_str());
     }
 
     for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
       TH1 * sel_hist = (TH1 *)it->second;
       std::string name = sel_hist->GetName();
       name += "Rebinned";
       
       size_t nAxes = 1;
       if (sel_hist->GetNbinsY() > 1) ++nAxes;
       if (sel_hist->GetNbinsZ() > 1) ++nAxes;
 
       if (nAxes == 1) {
         TString title = sel_hist->GetTitle();
         title += ";";
         title += sel_hist->GetXaxis()->GetTitle();
         fSelectionHistsRebinned[it->first] = new TH1D(
             name.c_str(), title/*.c_str()sel_hist->GetTitle()*/,
             sel_hist->GetNbinsX(), 0, sel_hist->GetNbinsX());
         Rebin1D(sel_hist, fSelectionHistsRebinned[it->first]);
       }
       else if (nAxes == 2) {
         std::string title = sel_hist->GetTitle();
         title += ";";
         title += sel_hist->GetXaxis()->GetTitle();
         title += ";";
         title += sel_hist->GetYaxis()->GetTitle();
 
         fSelectionHistsRebinned[it->first] = new TH2D(
             name.c_str(), title.c_str()/*sel_hist->GetTitle()*/,
             sel_hist->GetNbinsX(), 0, sel_hist->GetNbinsX(),
             sel_hist->GetNbinsY(), 0, sel_hist->GetNbinsY());
         Rebin2D(sel_hist, fSelectionHistsRebinned[it->first]);
       }
       else if (nAxes == 3) {
         std::string title = sel_hist->GetTitle();
         title += ";";
         title += sel_hist->GetXaxis()->GetTitle();
         title += ";";
         title += sel_hist->GetYaxis()->GetTitle();
         title += ";";
         title += sel_hist->GetZaxis()->GetTitle();
 
         fSelectionHistsRebinned[it->first] = new TH3D(
             name.c_str(), title.c_str()/*sel_hist->GetTitle()*/,
             sel_hist->GetNbinsX(), 0, sel_hist->GetNbinsX(),
             sel_hist->GetNbinsY(), 0, sel_hist->GetNbinsY(),
             sel_hist->GetNbinsZ(), 0, sel_hist->GetNbinsZ());
         Rebin3D(sel_hist, fSelectionHistsRebinned[it->first]);
       }
     }
 
     fMadeRebinned = true;
   }
 }

void protoana::ThinSliceDataSet::Rebin1D	(	TH1 *	sel_hist,
		TH1 *	rebinned
	)

private

Definition at line 128 of file ThinSliceDataSet.cxx.

                                                                      {
   for (int i = 1; i <= sel_hist->GetNbinsX(); ++i) {
     double low_x = sel_hist->GetXaxis()->GetBinLowEdge(i);
     double up_x = sel_hist->GetXaxis()->GetBinUpEdge(i);
     std::string bin_label = (low_x < 0. ? "< 0." :
                              (protoana::PreciseToString(low_x, 0) + " - " +
                               protoana::PreciseToString(up_x, 0)));
     rebinned->GetXaxis()->SetBinLabel(i, bin_label.c_str());
 
     rebinned->SetBinContent(i, sel_hist->GetBinContent(i));
   }
 }

void protoana::ThinSliceDataSet::Rebin2D	(	TH1 *	sel_hist,
		TH1 *	rebinned
	)

private

Definition at line 141 of file ThinSliceDataSet.cxx.

                                                                      {
   for (int i = 1; i <= sel_hist->GetNbinsX(); ++i) {
     double low_x = sel_hist->GetXaxis()->GetBinLowEdge(i);
     double up_x = sel_hist->GetXaxis()->GetBinUpEdge(i);
     std::string bin_label = (low_x < 0. ? "< 0." :
                              (protoana::PreciseToString(low_x, 0) + " - " +
                               protoana::PreciseToString(up_x, 0)));
     rebinned->GetXaxis()->SetBinLabel(
         i, bin_label.c_str());
     for (int j = 1; j <= sel_hist->GetNbinsY(); ++j) {
       double low_y = sel_hist->GetYaxis()->GetBinLowEdge(j);
       double up_y = sel_hist->GetYaxis()->GetBinUpEdge(j);
       std::string y_label = (low_y < 0. ? "< 0." :
                              (protoana::PreciseToString(low_y, 0) + " - " +
                               protoana::PreciseToString(up_y, 0)));
       rebinned->GetYaxis()->SetBinLabel(j, bin_label.c_str());
       rebinned->SetBinContent(i, j, sel_hist->GetBinContent(i, j));
     }
   }
 }

void protoana::ThinSliceDataSet::Rebin3D	(	TH1 *	sel_hist,
		TH1 *	rebinned
	)

private

Definition at line 162 of file ThinSliceDataSet.cxx.

                                                                      {
   for (int i = 1; i <= sel_hist->GetNbinsX(); ++i) {
     double low_x = sel_hist->GetXaxis()->GetBinLowEdge(i);
     double up_x = sel_hist->GetXaxis()->GetBinUpEdge(i);
     std::string bin_label = (low_x < 0. ? "< 0." :
                              (protoana::PreciseToString(low_x, 0) + " - " +
                               protoana::PreciseToString(up_x, 0)));
     rebinned->GetXaxis()->SetBinLabel(i, bin_label.c_str());
     for (int j = 1; j <= sel_hist->GetNbinsY(); ++j) {
       double low_y = sel_hist->GetYaxis()->GetBinLowEdge(j);
       double up_y = sel_hist->GetYaxis()->GetBinUpEdge(j);
       std::string y_label = (low_y < 0. ? "< 0." :
                              (protoana::PreciseToString(low_y, 0) + " - " +
                               protoana::PreciseToString(up_y, 0)));
       rebinned->GetYaxis()->SetBinLabel(j, bin_label.c_str());
 
       for (int k = 1; k <= sel_hist->GetNbinsY(); ++k) {
         double low_z = sel_hist->GetYaxis()->GetBinLowEdge(k);
         double up_z = sel_hist->GetYaxis()->GetBinUpEdge(k);
         std::string y_label = (low_z < 0. ? "< 0." :
                                (protoana::PreciseToString(low_z, 0) + " - " +
                                 protoana::PreciseToString(up_z, 0)));
         rebinned->GetZaxis()->SetBinLabel(k, bin_label.c_str());
 
         rebinned->SetBinContent(i, j, k, sel_hist->GetBinContent(i, j, k));
       }
     }
   }
 }

void protoana::ThinSliceDataSet::Refill1DRebinned ( )

Definition at line 119 of file ThinSliceDataSet.cxx.

                                                 {
   for (auto it = fSelectionHists.begin(); it != fSelectionHists.end(); ++it) {
     for (int i = 1; i <= it->second->GetNbinsX(); ++i) {
       fSelectionHistsRebinned[it->first]->SetBinContent(
           i, it->second->GetBinContent(i));
     }
   }
 }

Member Data Documentation

std::vector<std::pair<std::pair<int, int>, double> > protoana::ThinSliceDataSet::fCumulatives

private

Definition at line 134 of file ThinSliceDataSet.h.

TH1D protoana::ThinSliceDataSet::fIncidentHist

private

Definition at line 129 of file ThinSliceDataSet.h.

TH1D protoana::ThinSliceDataSet::fIncidentHistRebinned

private

Definition at line 131 of file ThinSliceDataSet.h.

bool protoana::ThinSliceDataSet::fMadeRebinned = false

private

Definition at line 132 of file ThinSliceDataSet.h.

TRandom3 protoana::ThinSliceDataSet::fRNG = TRandom3(0)

private

Definition at line 135 of file ThinSliceDataSet.h.

std::map<int, TH1 *> protoana::ThinSliceDataSet::fSelectionHists

private

Definition at line 128 of file ThinSliceDataSet.h.

std::map<int, TH1 *> protoana::ThinSliceDataSet::fSelectionHistsRebinned

private

Definition at line 130 of file ThinSliceDataSet.h.

std::map<int, std::string> protoana::ThinSliceDataSet::fSelectionNames

private

Definition at line 133 of file ThinSliceDataSet.h.

double protoana::ThinSliceDataSet::fTotal

private

Definition at line 136 of file ThinSliceDataSet.h.

The documentation for this class was generated from the following files:

protoduneana/protoduneana/Utilities/FitUtils/ThinSliceDataSet.h
protoduneana/protoduneana/Utilities/FitUtils/ThinSliceDataSet.cxx

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation