Skip to content

Commit ec5f91f

Browse files
committed
DRY sideband ranges
1 parent b59dc9e commit ec5f91f

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

PWGHF/D2H/Macros/HFInvMassFitter.cxx

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -241,17 +241,19 @@ void HFInvMassFitter::doFit()
241241
const ParameterRanges rooNBkgParamRanges{0., 1.2 * integralHisto, 0.3 * integralHisto};
242242
mRooNBkg = new RooRealVar("mRooNBkg", "number of background", randomizeInitialParameter(rooNBkgParamRanges), rooNBkgParamRanges.lower, rooNBkgParamRanges.upper); // background yield
243243
mBkgPdf = new RooAddPdf("mBkgPdf", "background fit function", RooArgList(*bkgPdf), RooArgList(*mRooNBkg));
244+
std::string sbRanges{"SBL,SBR"};
244245
if (mTypeOfSgnPdf == GausSec) { // two peak fit
246+
sbRanges.append(",SEC");
245247
if (strcmp(mFitOption.c_str(), "Chi2") == 0) {
246-
mBkgPdf->chi2FitTo(dataHistogram, Range("SBL,SBR,SEC"), Save());
248+
mBkgPdf->chi2FitTo(dataHistogram, Range(sbRanges.c_str()), Save());
247249
} else {
248-
mBkgPdf->fitTo(dataHistogram, Range("SBL,SBR,SEC"), Save());
250+
mBkgPdf->fitTo(dataHistogram, Range(sbRanges.c_str()), Save());
249251
}
250252
} else { // single peak fit
251253
if (strcmp(mFitOption.c_str(), "Chi2") == 0) {
252-
mBkgPdf->chi2FitTo(dataHistogram, Range("SBL,SBR"), Save());
254+
mBkgPdf->chi2FitTo(dataHistogram, Range(sbRanges.c_str()), Save());
253255
} else {
254-
mBkgPdf->fitTo(dataHistogram, Range("SBL,SBR"), Save());
256+
mBkgPdf->fitTo(dataHistogram, Range(sbRanges.c_str()), Save());
255257
}
256258
}
257259
// define the frame to evaluate background sidebands chi2 (bg pdf needs to be plotted within sideband ranges)

0 commit comments

Comments
 (0)