fix bug for variable sized bins

Author: viktorklochkov

src/EntryConfig.cpp

@@ -54,39 +54,37 @@ EntryConfig::EntryConfig(const Axis& x, Cuts* cuts_x, const Axis& y, Cuts* cuts_
   InitPlot();
 }
 
-
-
 TH1* EntryConfig::CreateHisto1D() const {
-  auto* ret = new TH1F(name_.c_str(), title_.c_str(),
-                       axes_.at(0).GetNbins(), axes_.at(0).GetXmin(), axes_.at(0).GetXmax());
-  ret->SetXTitle(axes_.at(0).GetTitle());
+  auto x = axes_.at(0);
+  TH1* ret = x.IsVariableBinSize() ?
+     new TH1F(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXbins()->GetArray()) :
+     new TH1F(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXmin(), x.GetXmax());
+  ret->SetXTitle(x.GetTitle());
   ret->SetYTitle("Entries");
   return ret;
 }
-// TODO fix axes
+
 TProfile* EntryConfig::CreateProfile() const {
+  auto x{axes_.at(0)}, y{axes_.at(1)};
+  auto ret = x.IsVariableBinSize() ?
+        new TProfile(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXbins()->GetArray()) :
+        new TProfile(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXmin(), x.GetXmax());
 
-  TProfile* ret{nullptr};
-  if (axes_[1].GetNbins() == 1 && axes_[1].GetXmax() == 1. && axes_[1].GetXmin() == 0.) {// Not init by user
-    ret = new TProfile(name_.c_str(), title_.c_str(),
-                       axes_.at(0).GetNbins(), axes_.at(0).GetXmin(), axes_.at(0).GetXmax());
-  } else if (axes_.size() == 2) {
-    ret = new TProfile(name_.c_str(), title_.c_str(),
-                       axes_.at(0).GetNbins(), axes_.at(0).GetXmin(), axes_.at(0).GetXmax(),
-                       axes_.at(1).GetXmin(), axes_.at(1).GetXmax());
-  }
-  ret->SetYTitle(axes_.at(1).GetTitle());
-  ret->SetXTitle(axes_.at(0).GetTitle());
+  ret->SetXTitle(x.GetTitle());
+  ret->SetYTitle(y.GetTitle());
   return ret;
 }
 
 TH2* EntryConfig::CreateHisto2D() const {
+  auto x{axes_.at(0)}, y{axes_.at(1)};
+  assert(x.IsVariableBinSize() && y.IsVariableBinSize() || !x.IsVariableBinSize() && !y.IsVariableBinSize());
+
+  auto* ret = x.IsVariableBinSize() ?
+      new TH2F(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXbins()->GetArray(), y.GetNbins(), y.GetXbins()->GetArray()) :
+      new TH2F(name_.c_str(), title_.c_str(), x.GetNbins(), x.GetXmin(), x.GetXmax(), y.GetNbins(), y.GetXmin(), y.GetXmax());
 
-  auto* ret = new TH2F(name_.c_str(), title_.c_str(),
-                       axes_.at(0).GetNbins(), axes_.at(0).GetXmin(), axes_.at(0).GetXmax(),
-                       axes_.at(1).GetNbins(), axes_.at(1).GetXmin(), axes_.at(1).GetXmax());
-  ret->SetXTitle(axes_.at(0).GetTitle());
-  ret->SetYTitle(axes_.at(1).GetTitle());
+  ret->SetXTitle(x.GetTitle());
+  ret->SetYTitle(y.GetTitle());
   ret->SetZTitle("Entries");
   ret->SetMinimum(1);
   return ret;