[AD] Add translation methods for RooParametricHist

Author: runtingt

interface/CombineCodegenImpl.h

@@ -8,6 +8,7 @@ class FastVerticalInterpHistPdf2;
 class FastVerticalInterpHistPdf2D2;
 class ProcessNormalization;
 class VerticalInterpPdf;
+class RooParametricHist;
 
 namespace RooFit::Experimental {
 
@@ -18,8 +19,10 @@ namespace RooFit::Experimental {
   void codegenImpl(FastVerticalInterpHistPdf2D2& arg, CodegenContext& ctx);
   void codegenImpl(ProcessNormalization& arg, CodegenContext& ctx);
   void codegenImpl(VerticalInterpPdf& arg, CodegenContext& ctx);
+  void codegenImpl(RooParametricHist& arg, CodegenContext& ctx);
 
   std::string codegenIntegralImpl(VerticalInterpPdf& arg, int code, const char* rangeName, CodegenContext& ctx);
+  std::string codegenIntegralImpl(RooParametricHist& arg, int code, const char* rangeName, CodegenContext& ctx);
 
 }  // namespace RooFit::Experimental
 

src/CombineCodegenImpl.cxx

@@ -9,6 +9,7 @@
 #include "../interface/VerticalInterpHistPdf.h"
 #include "../interface/VerticalInterpPdf.h"
 #include "../interface/CombineMathFuncs.h"
+#include "../interface/RooParametricHist.h"
 
 #include <RooUniformBinning.h>
 
@@ -242,6 +243,33 @@ void RooFit::Experimental::codegenImpl(VerticalInterpPdf& arg, CodegenContext& c
                               arg.quadraticAlgo()));
 }
 
+void RooFit::Experimental::codegenImpl(RooParametricHist& arg, CodegenContext& ctx) {
+  std::vector<double> diffs_flat;
+  std::vector<double> sums_flat;
+  if (arg.hasMorphs()) {
+    arg.getFlattenedMorphs(diffs_flat, sums_flat);
+  }
+
+  std::string xName = ctx.getResult(arg.observable());
+  std::stringstream bin_i;
+  bin_i << ctx.buildCall("RooFit::Detail::MathFuncs::parametricHistFindBin", arg.getNBins(), arg.getBins(), xName);
+  std::stringstream code;
+  code << ctx.buildCall("RooFit::Detail::MathFuncs::parametricHistEvaluate",
+                        bin_i.str(),
+                        arg.getPars(),
+                        arg.getBins(),
+                        arg.getNBins(),
+                        arg.getCoeffList(),
+                        static_cast<int>(arg.getCoeffList().size()),
+                        arg.hasMorphs() ? diffs_flat : std::vector<double>{},
+                        arg.hasMorphs() ? sums_flat : std::vector<double>{},
+                        arg.getWidths(),
+                        arg.getSmoothRegion()) +
+              ";\n";
+  ctx.addToCodeBody(code.str());
+  ctx.addResult(&arg, code.str());
+}
+
 std::string RooFit::Experimental::codegenIntegralImpl(VerticalInterpPdf& arg,
                                                       int code,
                                                       const char* rangeName,
@@ -257,4 +285,31 @@ std::string RooFit::Experimental::codegenIntegralImpl(VerticalInterpPdf& arg,
                        arg.quadraticAlgo());
 }
 
+std::string RooFit::Experimental::codegenIntegralImpl(RooParametricHist& arg,
+                                                      int code,
+                                                      const char* rangeName,
+                                                      CodegenContext& ctx) {
+  std::vector<double> diffs_flat;
+  std::vector<double> sums_flat;
+  if (arg.hasMorphs()) {
+    arg.getFlattenedMorphs(diffs_flat, sums_flat);
+  }
+  double xMin = rangeName ? arg.getObs().getMin(rangeName) : -1;
+  double xMax = rangeName ? arg.getObs().getMax(rangeName) : -1;  // if rangeName is null, these values won't be used
+
+  return ctx.buildCall("RooFit::Detail::MathFuncs::parametricHistIntegral",
+                       arg.getPars(),
+                       arg.getBins(),
+                       arg.getNBins(),
+                       arg.getCoeffList(),
+                       static_cast<int>(arg.getCoeffList().size()),
+                       arg.hasMorphs() ? diffs_flat : std::vector<double>{},
+                       arg.hasMorphs() ? sums_flat : std::vector<double>{},
+                       arg.getWidths(),
+                       arg.getSmoothRegion(),
+                       rangeName ? std::string("\"") + rangeName + "\"" : std::string("nullptr"),
+                       xMin,
+                       xMax);
+}
+
 #endif  // ROOT_VERSION_CODE >= ROOT_VERSION(6,36,0)