Skip to content

[HS3] Align RooFit HS3 JSON I/O with the current HS3 naming and HistFactory conventions. #22735

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Phmonski wants to merge 2 commits into
base: master
Choose a base branch
from
+430 −46
Open

[HS3] Align RooFit HS3 JSON I/O with the current HS3 naming and HistFactory conventions. #22735

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
9cf05ab
[HS3] Minor patch to further align HS3 with the spec
Phmonski Jul 1, 2026
98c0bd8
[HS3] fix bug with importing legacy workspaces
Phmonski Jul 3, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
111 changes: 85 additions & 26 deletions roofit/hs3/src/JSONFactories_HistFactory.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -191,6 +191,43 @@ std::string constraintName(std::string const &paramName)
return paramName + "Constraint";
}

bool isLegacyConstraintType(std::string const &value)
{
return value == "Gauss" || value == "Poisson" || value == "Const" || value == "Lognormal";
}

RooAbsPdf *findNamedConstraint(RooJSONFactoryWSTool &tool, std::string const &constraintName,
std::string const &sample)
{
if (auto *constraint = tool.workspace()->pdf(constraintName)) {
return constraint;
}

try {
return tool.request<RooAbsPdf>(constraintName, sample);
} catch (RooJSONFactoryWSTool::DependencyMissingError const &err) {
if (err.child() != constraintName) {
throw;
}
}

return nullptr;
}

RooAbsPdf &createLegacyConstraint(RooJSONFactoryWSTool &tool, const JSONNode &mod, RooRealVar &param,
std::string const &constraintType)
{
if (constraintType == "Gauss") {
param.setError(1.0);
return getOrCreate<RooGaussian>(*tool.workspace(), constraintName(param.GetName()), param,
*tool.workspace()->var(std::string("nom_") + param.GetName()), 1.);
}

RooJSONFactoryWSTool::error("legacy constraint value '" + constraintType + "' for modifier '" +
RooJSONFactoryWSTool::name(mod) +
"' is a known constraint type, but it cannot be resolved in this context");
}

ParamHistFunc &createPHF(const std::string &phfname, std::string const &sysname,
const std::vector<std::string> &parnames, const std::vector<double> &vals,
RooJSONFactoryWSTool &tool, RooAbsCollection &constraints, const RooArgSet &observables,
Expand Down Expand Up @@ -255,12 +292,10 @@ const JSONNode &findStaterror(const JSONNode &comp)
RooAbsPdf &
getOrCreateConstraint(RooJSONFactoryWSTool &tool, const JSONNode &mod, RooRealVar &param, const std::string &sample)
{
if (auto constrName = mod.find("constraint_name")) {
JSONNode const *constrName = mod.find("constraint_name");
if (constrName) {
auto constraint_name = constrName->val();
auto constraint = tool.workspace()->pdf(constraint_name);
if (!constraint) {
constraint = tool.request<RooAbsPdf>(constrName->val(), sample);
}
auto constraint = findNamedConstraint(tool, constraint_name, sample);
if (!constraint) {
RooJSONFactoryWSTool::error("unable to find definition of of constraint '" + constraint_name +
"' for modifier '" + RooJSONFactoryWSTool::name(mod) + "'");
Expand All @@ -269,19 +304,35 @@ getOrCreateConstraint(RooJSONFactoryWSTool &tool, const JSONNode &mod, RooRealVa
param.setError(gauss->getSigma().getVal());
}
return *constraint;
} else {
std::string constraint_type = "Gauss";
if (auto constrType = mod.find("constraint_type")) {
constraint_type = constrType->val();
}

if (auto constr = mod.find("constraint")) {
std::string constraintValue = constr->val();
if (auto *constraint = findNamedConstraint(tool, constraintValue, sample)) {
if (auto gauss = dynamic_cast<RooGaussian *const>(constraint)) {
param.setError(gauss->getSigma().getVal());
}
return *constraint;
}
if (constraint_type == "Gauss") {
param.setError(1.0);
return getOrCreate<RooGaussian>(*tool.workspace(), constraintName(param.GetName()), param,
*tool.workspace()->var(std::string("nom_") + param.GetName()), 1.);

if (isLegacyConstraintType(constraintValue)) {
return createLegacyConstraint(tool, mod, param, constraintValue);
}
RooJSONFactoryWSTool::error("unknown or invalid constraint for modifier '" + RooJSONFactoryWSTool::name(mod) +
"'");

RooJSONFactoryWSTool::error("unable to resolve constraint value '" + constraintValue + "' for modifier '" +
RooJSONFactoryWSTool::name(mod) +
"': this looks like a legacy workspace where the 'constraint' field is neither a "
"constraint pdf name nor a supported legacy constraint type");
}

std::string constraint_type = "Gauss";
if (auto constrType = mod.find("constraint_type")) {
constraint_type = constrType->val();
}
if (isLegacyConstraintType(constraint_type)) {
return createLegacyConstraint(tool, mod, param, constraint_type);
}
RooJSONFactoryWSTool::error("unknown or invalid constraint for modifier '" + RooJSONFactoryWSTool::name(mod) + "'");
}
double poissonTau(RooPoisson const &constraint, RooAbsArg const &gamma)
{
Expand Down Expand Up @@ -366,7 +417,7 @@ bool importHistSample(RooJSONFactoryWSTool &tool, RooDataHist &dh, RooArgSet con
RooRealVar &constrParam = getOrCreate<RooRealVar>(ws, sysname, 1., -3, 5);
constrParam.setError(0.0);
normElems.add(constrParam);
if (mod.has_child("constraint_name") || mod.has_child("constraint_type")) {
if (mod.has_child("constraint") || mod.has_child("constraint_name") || mod.has_child("constraint_type")) {
// for norm factors, constraints are optional
constraints.add(getOrCreateConstraint(tool, mod, constrParam, sampleName));
}
Expand Down Expand Up @@ -429,9 +480,20 @@ bool importHistSample(RooJSONFactoryWSTool &tool, RooDataHist &dh, RooArgSet con
RooJSONFactoryWSTool::error("unable to instantiate shapesys '" + sysname +
"' with neither values nor parameters!");
}
std::string constraint(mod.has_child("constraint_type") ? mod["constraint_type"].val()
: mod.has_child("constraint") ? mod["constraint"].val()
: "unknown");
std::string constraint = "unknown";
if (mod.has_child("constraint_type")) {
constraint = mod["constraint_type"].val();
} else if (mod.has_child("constraint")) {
std::string constraintValue = mod["constraint"].val();
if (isLegacyConstraintType(constraintValue)) {
constraint = constraintValue;
} else {
RooJSONFactoryWSTool::error("unable to resolve constraint value '" + constraintValue +
"' for modifier '" + RooJSONFactoryWSTool::name(mod) +
"': this looks like a legacy workspace where the 'constraint' field is "
"not a supported legacy constraint type");
}
}
shapeElems.add(createPHF(funcName, sysname, parnames, vals, tool, constraints, varlist, constraint,
defaultGammaMin, defaultShapeSysGammaMax, minShapeUncertainty));
} else if (modtype == "custom") {
Expand Down Expand Up @@ -1165,13 +1227,11 @@ void configureStatError(Channel &channel)

bool exportChannel(RooJSONFactoryWSTool *tool, const Channel &channel, JSONNode &elem)
{
// Write the constraint reference (either by name or by type) for any
// modifier that supports an external Gaussian/Poisson/etc. constraint.
// Write the constraint reference for any modifier that supports an
// external Gaussian/Poisson/etc. constraint.
auto writeConstraint = [](JSONNode &mod, auto const &sys) {
if (sys.constraint) {
mod["constraint_name"] << sys.constraint->GetName();
} else if (sys.constraintType) {
mod["constraint_type"] << toString(sys.constraintType);
mod["constraint"] << sys.constraint->GetName();
}
};

Expand All @@ -1192,7 +1252,7 @@ bool exportChannel(RooJSONFactoryWSTool *tool, const Channel &channel, JSONNode
mod["parameter"] << nf.param->GetName();
mod["type"] << "normfactor";
if (nf.constraint) {
mod["constraint_name"] << nf.constraint->GetName();
mod["constraint"] << nf.constraint->GetName();
tool->queueExport(*nf.constraint);
}
}
Expand Down Expand Up @@ -1264,7 +1324,6 @@ bool exportChannel(RooJSONFactoryWSTool *tool, const Channel &channel, JSONNode
mod["name"] << ::Literals::staterror;
mod["type"] << ::Literals::staterror;
optionallyExportGammaParameters(mod, "stat_" + channel.name, sample.staterrorParameters);
mod["constraint_type"] << toString(sample.barlowBeestonLightConstraintType);
}

if (!observablesWritten) {
Expand Down
Loading