[RF][HS3] Added globs tracking

root-project · Jan 10, 2025 · 590c79b · 590c79b
1 parent 017c63b
commit 590c79b
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/roofit/hs3/src/RooJSONFactoryWSTool.cxx b/roofit/hs3/src/RooJSONFactoryWSTool.cxx
@@ -686,7 +686,7 @@ void importAnalysis(const JSONNode &rootnode, const JSONNode &analysisNode, cons
    for (const auto &p : pars) {
       if (mc->GetParametersOfInterest()->find(*p))
          continue;
-      if (p->isConstant() && !mainPars.find(*p)) {
+      if (p->isConstant() && !mainPars.find(*p) && domainPars.find(*p)) {
          globs.add(*p);
       } else if (domainPars.find(*p)) {
          nps.add(*p);
@@ -1792,6 +1792,16 @@ void RooJSONFactoryWSTool::exportSingleModelConfig(JSONNode &rootnode, RooStats:
       npDomain.writeJSON(appendNamedChild(domainsNode, npDomainName));
    }
 
+   if (mc.GetGlobalObservables()) {
+      std::string globDomainName = analysisName + "_global_observables";
+      domains.append_child() << globDomainName;
+      RooFit::JSONIO::Detail::Domains::ProductDomain globDomain;
+      for (auto *glob : static_range_cast<const RooRealVar *>(*mc.GetGlobalObservables())) {
+         globDomain.readVariable(*glob);
+      }
+      globDomain.writeJSON(appendNamedChild(domainsNode, globDomainName));
+   }
+
    if (mc.GetParametersOfInterest()) {
       std::string poiDomainName = analysisName + "_parameters_of_interest";
       domains.append_child() << poiDomainName;