Skip to content
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

Jump to bottom

Add optional range for channels in ChannelCompatibilityCheck #760

Merged
merged 2 commits into from
Oct 10, 2023
Merged

Add optional range for channels in ChannelCompatibilityCheck #760

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
12296cc
add optional ranges for channel group in ChannelCompatibilityCheck
IzaakWN May 23, 2022
4a99418
add groupRanges_ to header file
IzaakWN May 23, 2022
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
1 change: 1 addition & 0 deletions interface/ChannelCompatibilityCheck.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ class ChannelCompatibilityCheck : public FitterAlgoBase {
static bool saveFitResult_;

static std::vector<std::string> groups_;
static std::map<TString,std::pair<double,double>> groupRanges_;

virtual bool runSpecific(RooWorkspace *w, RooStats::ModelConfig *mc_s, RooStats::ModelConfig *mc_b, RooAbsData &data, double &limit, double &limitErr, const double *hint);
};
Expand Down
30 changes: 25 additions & 5 deletions src/ChannelCompatibilityCheck.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,14 +25,15 @@ bool ChannelCompatibilityCheck::fixedMu_ = false;
bool ChannelCompatibilityCheck::saveFitResult_ = true;
bool ChannelCompatibilityCheck::runMinos_ = true;
std::vector<std::string> ChannelCompatibilityCheck::groups_;
std::map<TString,std::pair<double,double>> ChannelCompatibilityCheck::groupRanges_;

ChannelCompatibilityCheck::ChannelCompatibilityCheck() :
FitterAlgoBase("ChannelCompatibilityCheck specific options")
{
options_.add_options()
("fixedSignalStrength", boost::program_options::value<float>(&mu_)->default_value(mu_), "Compute the compatibility for a fixed signal strength. If not specified, it's left floating")
("saveFitResult", "Save fit results in output file")
("group,g", boost::program_options::value<std::vector<std::string> >(&groups_), "Group together channels that contain a given name. Can be used multiple times.")
("group,g", boost::program_options::value<std::vector<std::string> >(&groups_), "Group together channels that contain a given name. Can be used multiple times. Optionally, set range as name=rMin,rMax")
("runMinos", boost::program_options::value<bool>(&runMinos_)->default_value(runMinos_), "Compute also uncertainties using profile likeilhood (MINOS or robust variants of it)")
;
}
Expand All @@ -42,6 +43,19 @@ void ChannelCompatibilityCheck::applyOptions(const boost::program_options::varia
applyOptionsBase(vm);
fixedMu_ = !vm["fixedSignalStrength"].defaulted();
saveFitResult_ = vm.count("saveFitResult");
for(unsigned int i = 0; i < groups_.size(); i++) {
std::vector<std::string> groupExpr;
boost::split(groupExpr, groups_[i], boost::is_any_of("=,")); // "-g channel=rMin,rMax" or just "-g channel"
if (groupExpr.size() == 3) {
groups_[i] = groupExpr[0];
groupRanges_[groupExpr[0]] = {atof(groupExpr[1].c_str()), atof(groupExpr[2].c_str())};
if (verbose>=0) std::cout << "Will set range of channel " << groupExpr[0] << " to [" << groupExpr[1] << ", " << groupExpr[2] << "]" << std::endl;
} else if (groupExpr.size() == 1) {
if (verbose>=1) std::cout << "No range to parse for channel " << groups_[i] << std::endl;
} else {
std::cout << "Error parsing group expression : " << groups_[i] << std::endl;
}
}
}

bool ChannelCompatibilityCheck::runSpecific(RooWorkspace *w, RooStats::ModelConfig *mc_s, RooStats::ModelConfig *mc_b, RooAbsData &data, double &limit, double &limitErr, const double *hint) {
Expand All @@ -64,10 +78,13 @@ bool ChannelCompatibilityCheck::runSpecific(RooWorkspace *w, RooStats::ModelConf
RooAbsPdf *pdfi = sim->getPdf(cat->getLabel());
if (pdfi == 0) continue;
RooCustomizer customizer(*pdfi, "freeform");
TString riName = TString::Format("_ChannelCompatibilityCheck_%s_%s", r->GetName(), nameForLabel(cat->getLabel()).c_str());
rs.insert(std::pair<std::string,std::string>(nameForLabel(cat->getLabel()), riName.Data()));
std::string label = nameForLabel(cat->getLabel());
TString riName = TString::Format("_ChannelCompatibilityCheck_%s_%s", r->GetName(), label.c_str());
rs.insert(std::pair<std::string,std::string>(label, riName.Data()));
std::pair<double,double> range = {r->getMin(),r->getMax()};
if (groupRanges_.find(TString(label)) != groupRanges_.end()) range = groupRanges_[label];
if (w->var(riName) == 0) {
w->factory(TString::Format("%s[%g,%g]", riName.Data(), r->getMin(), r->getMax()));
w->factory(TString::Format("%s[%g,%g]", riName.Data(), range.first, range.second));
}
customizer.replaceArg(*r, *w->var(riName));
newsim->addPdf((RooAbsPdf&)*customizer.build(), cat->getLabel());
Expand Down Expand Up @@ -136,7 +153,10 @@ std::string ChannelCompatibilityCheck::nameForLabel(const char *label)
{
std::string ret(label);
for (std::vector<std::string>::const_iterator it = groups_.begin(), ed = groups_.end(); it != ed; ++it) {
if (ret.find(*it) != std::string::npos) { ret = *it; break; }
if (ret.find(*it) != std::string::npos) {
if (verbose>=1) std::cout << "Grouping channel" << label << " with " << *it << std::endl;
ret = *it; break;
}
}
return ret;
}
Expand Down