Skip to content

Commit 9abe2e9

Browse files
committed
[RF][HS3] Replace RooStats::ModelConfig with RooFit::ModelConfig
This avoids some unneeded RooStats dependency.
1 parent 75a3583 commit 9abe2e9

3 files changed

Lines changed: 13 additions & 13 deletions

File tree

roofit/hs3/src/RooJSONFactoryWSTool.cxx

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,7 @@ std::unique_ptr<RooAbsData> loadData(const JSONNode &p, RooWorkspace &workspace)
544544
*
545545
* This function imports an analysis, represented by the provided JSONNodes 'analysisNode' and 'likelihoodsNode',
546546
* into the workspace represented by the provided RooWorkspace. The analysis information is read from the JSONNodes
547-
* and added to the workspace as one or more RooStats::ModelConfig objects.
547+
* and added to the workspace as one or more RooFit::ModelConfig objects.
548548
*
549549
* @param rootnode The root JSONNode representing the entire JSON file.
550550
* @param analysisNode The JSONNode representing the analysis to be imported.
@@ -567,8 +567,8 @@ void importAnalysis(const JSONNode &rootnode, const JSONNode &analysisNode, cons
567567
if (workspace.obj(mcname))
568568
return;
569569

570-
workspace.import(RooStats::ModelConfig{mcname.c_str(), mcname.c_str()});
571-
auto *mc = static_cast<RooStats::ModelConfig *>(workspace.obj(mcname));
570+
workspace.import(RooFit::ModelConfig{mcname.c_str(), mcname.c_str()});
571+
auto *mc = static_cast<RooFit::ModelConfig *>(workspace.obj(mcname));
572572
mc->SetWS(workspace);
573573

574574
auto *nllNode = RooJSONFactoryWSTool::findNamedChild(likelihoodsNode, analysisNode["likelihood"].val());
@@ -1738,7 +1738,7 @@ void RooJSONFactoryWSTool::importDependants(const JSONNode &n)
17381738
}
17391739
}
17401740

1741-
void RooJSONFactoryWSTool::exportModelConfig(JSONNode &rootnode, RooStats::ModelConfig const &mc,
1741+
void RooJSONFactoryWSTool::exportModelConfig(JSONNode &rootnode, RooFit::ModelConfig const &mc,
17421742
const std::vector<CombinedData> &combDataSets,
17431743
const std::vector<RooAbsData *> &singleDataSets)
17441744
{
@@ -1773,7 +1773,7 @@ void RooJSONFactoryWSTool::exportModelConfig(JSONNode &rootnode, RooStats::Model
17731773
}
17741774
}
17751775

1776-
void RooJSONFactoryWSTool::exportSingleModelConfig(JSONNode &rootnode, RooStats::ModelConfig const &mc,
1776+
void RooJSONFactoryWSTool::exportSingleModelConfig(JSONNode &rootnode, RooFit::ModelConfig const &mc,
17771777
std::string const &analysisName,
17781778
std::map<std::string, std::string> const *dataComponents)
17791779
{
@@ -1936,7 +1936,7 @@ void RooJSONFactoryWSTool::exportAllObjects(JSONNode &n)
19361936

19371937
// export all ModelConfig objects and attached Pdfs
19381938
for (TObject *obj : _workspace.allGenericObjects()) {
1939-
if (auto mc = dynamic_cast<RooStats::ModelConfig *>(obj)) {
1939+
if (auto mc = dynamic_cast<RooFit::ModelConfig *>(obj)) {
19401940
exportModelConfig(n, *mc, combData, singleData);
19411941
}
19421942
}
@@ -2446,7 +2446,7 @@ RooWorkspace RooJSONFactoryWSTool::cleanWS(const RooWorkspace &ws, bool onlyMode
24462446
RooWorkspace tmpWS = RooWorkspace();
24472447
if (onlyModelConfig) {
24482448
for (auto *obj : ws.allGenericObjects()) {
2449-
if (auto *mc = dynamic_cast<RooStats::ModelConfig *>(obj)) {
2449+
if (auto *mc = dynamic_cast<RooFit::ModelConfig *>(obj)) {
24502450
tmpWS.import(*mc->GetPdf(), RooFit::RecycleConflictNodes(true));
24512451
}
24522452
}
@@ -2479,7 +2479,7 @@ RooWorkspace RooJSONFactoryWSTool::cleanWS(const RooWorkspace &ws, bool onlyMode
24792479
}
24802480

24812481
/*
2482-
if (auto* mc = dynamic_cast<RooStats::ModelConfig*>(obj)) {
2482+
if (auto* mc = dynamic_cast<RooFit::ModelConfig*>(obj)) {
24832483
// Import the PDF
24842484
tmpWS.import(*mc->GetPdf());
24852485
@@ -2500,7 +2500,7 @@ RooWorkspace RooJSONFactoryWSTool::cleanWS(const RooWorkspace &ws, bool onlyMode
25002500
tmpWS.import(*nuis);
25012501
25022502
2503-
RooStats::ModelConfig* mc_new = new RooStats::ModelConfig(mc->GetName(), mc->GetName());
2503+
RooFit::ModelConfig* mc_new = new RooFit::ModelConfig(mc->GetName(), mc->GetName());
25042504
25052505
mc_new->SetPdf(*tmpWS.pdf(mc->GetPdf()->GetName()));
25062506
mc_new->SetObservables(*tmpWS.set(obs->GetName()));
@@ -2627,7 +2627,7 @@ RooWorkspace RooJSONFactoryWSTool::sanitizeWS(const RooWorkspace &ws)
26272627
}
26282628
}
26292629

2630-
if (auto *mc = dynamic_cast<RooStats::ModelConfig *>(obj)) {
2630+
if (auto *mc = dynamic_cast<RooFit::ModelConfig *>(obj)) {
26312631
// Sanitize ModelConfig name
26322632
if (!isValidName(mc->GetName())) {
26332633
mc->SetName(sanitizeName(mc->GetName()).c_str());

roofit/hs3/test/testHS3SimultaneousFit.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ std::unique_ptr<RooFitResult> writeJSONAndFitModel(std::string &jsonStr)
4646
// Simultaneous PDF and model config
4747
ws.factory("SIMUL::simPdf(channelCat[channel_1=0, channel_2=1], channel_1=model_1, channel_2=model_2)");
4848

49-
RooStats::ModelConfig modelConfig{"ModelConfig"};
49+
RooFit::ModelConfig modelConfig{"ModelConfig"};
5050

5151
modelConfig.SetWS(ws);
5252
modelConfig.SetPdf("simPdf");

roofit/hs3/test/testRooFitHS3.cxx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
#include <RooSimultaneous.h>
2626
#include <RooWorkspace.h>
2727
#include <RooFormulaVar.h>
28-
#include <RooStats/ModelConfig.h>
28+
#include <RooFit/ModelConfig.h>
2929

3030
#include <TROOT.h>
3131

@@ -524,7 +524,7 @@ TEST(RooFitHS3, ModelConfigWithMultiVarGaussian)
524524
ws1.import(mv, RooFit::Silence(), RooFit::RecycleConflictNodes());
525525

526526
// Build a ModelConfig referencing the pdf and its observables
527-
RooStats::ModelConfig mc{"mc", &ws1};
527+
RooFit::ModelConfig mc{"mc", &ws1};
528528
mc.SetPdf(*ws1.pdf("mvgauss"));
529529
mc.SetObservables("x,y");
530530
ws1.import(mc);

0 commit comments

Comments
 (0)