QVector<double> SeriesSmoother::smooth( const QVector<double> & rawData)
{
// New smooth data
QVector<double> smData(rawData.size());
const int midPoint = (m_pointCount-1)/2;
// Coefficient index
int coeffIdx = 0;
// Index of first data point
int startIdx = 0;
for (int i = 0; i < rawData.size(); ++i) {
// Determine which row of coefficients to use
if ( i < midPoint ) {
// Start of the data
coeffIdx = i;
startIdx = 0;
} else if ( rawData.size() < i + midPoint ) {
// End of the data
coeffIdx = rawData.size() - i;
startIdx = rawData.size() - m_pointCount;
} else {
// Middle of the data
coeffIdx = midPoint;
startIdx = i - midPoint;
}
double sum = 0;
for (int j = 0; j < m_pointCount; ++j) {
sum += m_coeffs.at(coeffIdx).at(j) * rawData.at(j+startIdx);
}
smData[i] = sum;
}
return smData;
}
double RunHypoTest(char *smwwFileName, char *ttbarFileName, char *wp3jetsFileName, char *wp4jetsFileName, char *opsFileName, char *outputFileName, double lambda) {
TFile *smwwFile = new TFile(smwwFileName);
TFile *ttbarFile = new TFile(ttbarFileName);
TFile *wp3jetsFile = new TFile(wp3jetsFileName);
TFile *wp4jetsFile = new TFile(wp4jetsFileName);
TFile *opsFile = new TFile(opsFileName);
TFile *outputFile = new TFile(outputFileName, "UPDATE");
TH1F *smww = (TH1F*)smwwFile->Get(WW_MASS_HISTOGRAM_NAME);
TH1F *ttbar = (TH1F*)ttbarFile->Get(WW_MASS_HISTOGRAM_NAME);
TH1F *wp3jets = (TH1F*)wp3jetsFile->Get(WW_MASS_HISTOGRAM_NAME);
TH1F *wp4jets = (TH1F*)wp4jetsFile->Get(WW_MASS_HISTOGRAM_NAME);
//Histogram of ww-scattering with effective operator contributions
TH1F *ops = (TH1F*)opsFile->Get(WW_MASS_HISTOGRAM_NAME);
RooRealVar *mww = new RooRealVar("mww", "M_{WW}", 600, 2500, "GeV");
RooDataHist smData("smData", "smData", RooArgList(*mww), smww);
RooDataHist opsData("opsData", "opsData", RooArgList(*mww), ops);
RooDataHist ttbarData("ttbarData", "ttbarData", RooArgList(*mww), ttbar);
RooDataHist wp3jetsData("wp3jetsData", "wp3jetsData", RooArgList(*mww), wp3jets);
RooDataHist wp4jetsData("wp4jetsData", "wp4jetsData", RooArgList(*mww), wp4jets);
/*
RooAbsPdf *opsModel;
if (lambda == 400) {
opsModel = SpecialCaseModel(&opsData, mww, (char*)"ops");
}
else {
opsModel = MakeModel(&opsData, mww, (char*)"ops");
}*/
RooAbsPdf *opsModel = MakeModel(&opsData, mww, (char*)"ops");
//RooPlot *xframe = mww->frame();
//opsData.plotOn(xframe);
//opsModel->plotOn(xframe);
//printf("Chi-squared for lambda = %f: = %f\n", lambda, xframe->chiSquare("opsModel", "opsData", 3));
RooAbsPdf *smModel = MakeModelNoSignal(&smData, mww, (char*)"sm");
RooAbsPdf *ttbarModel = MakeModelNoSignal(&ttbarData, mww, (char*)"ttbar");
RooAbsPdf *wp3jetsModel = MakeModelNoSignal(&wp3jetsData, mww, (char*)"wp3jets");
RooAbsPdf *wp4jetsModel = MakeModelNoSignal(&wp4jetsData, mww, (char*)"wp4jets");
TCanvas *canvas = new TCanvas(opsFileName);
RooPlot *frame = mww->frame();
frame->SetTitle("");
//smData.plotOn(frame, RooFit::LineColor(kBlack), RooFit::Name("smData"));
//smModel->plotOn(frame, RooFit::LineColor(kBlue), RooFit::Name("smModel"));
//ttbarModel->plotOn(frame, RooFit::LineColor(kRed), RooFit::Name("ttbarModel"));
//wp3jetsModel->plotOn(frame, RooFit::LineColor(kYellow), RooFit::Name("wpjetsModel"));
opsData.plotOn(frame);
opsModel->plotOn(frame, RooFit::LineColor(kBlue), RooFit::Name("opsModel"));
//leg->AddEntry(frame->findObject("smModel"), "SM Model", "lep");
//leg->AddEntry(frame->findObject("ttbarModel"), "TTBar Model", "lep");
//leg->AddEntry(frame->findObject("wp3jetsModel"), "WP3Jets Model", "lep");
//leg->AddEntry(frame->findObject("opsModel"), "Effective Operator Model", "lep");
frame->Draw();
canvas->Write();
Double_t ww_x = WW_CROSS_SECTION * smww->GetEntries();
Double_t ttbar_x = TTBAR_CROSS_SECTION * ttbar->GetEntries();
Double_t wp3jets_x = WP3JETS_CROSS_SECTION * wp3jets->GetEntries();
Double_t wp4jets_x = WP4JETS_CROSS_SECTION * wp4jets->GetEntries();
Double_t ttbar_weight = ttbar_x/(ttbar_x + wp3jets_x + wp4jets_x + ww_x);
Double_t wp3jets_weight = wp3jets_x/(wp3jets_x + ttbar_x + ww_x);
Double_t wp4jets_weight = wp4jets_x/(wp4jets_x + ttbar_x + ww_x);
RooRealVar *ttbarWeight = new RooRealVar("ttbarWeight", "ttbarWeight", 0.0, 1.0, ttbar_weight);
RooRealVar *wp3jetsWeight = new RooRealVar("wp3jetsWeight", "wp3jetsWeight", 0.0, 1.0, wp3jets_weight);
RooRealVar *wp4jetsWeight = new RooRealVar("wp4jetsWeight", "wp4jetsWeight", 0.0, 1.0, wp4jets_weight);
ttbarWeight->setConstant();
wp3jetsWeight->setConstant();
wp4jetsWeight->setConstant();
RooRealVar *mu = new RooRealVar("mu", "mu", 0.0, 1.0, "");
RooAddPdf *wwModel = new RooAddPdf("wwModel", "u*effective_ww + (1-u)*SM_WW",
RooArgList(*opsModel, *smModel), RooArgList(*mu), kTRUE);
RooAddPdf *model = new RooAddPdf("model", "Full model",
RooArgList(*ttbarModel, *wp3jetsModel, *wp4jetsModel, *wwModel),
RooArgList(*ttbarWeight, *wp3jetsWeight, *wp4jetsWeight), kTRUE);
//Generate data under the alternate hypothesis
mu->setVal(1.0);
int nTestSetEvents = WW_CROSS_SECTION * TOTAL_INTEGRATED_LUMINOSITY;
RooAbsData *generatedData = model->generate(*mww, nTestSetEvents);
TCanvas *canvas2 = new TCanvas("CombinedModels");
RooPlot *frame2 = mww->frame();
//wwModel->plotOn(frame2, RooFit::LineColor(kRed), RooFit::Name("wwModel"));
//generatedData->plotOn(frame2);
mu->setVal(0.0);
model->plotOn(frame2, RooFit::LineColor(kBlue), RooFit::Name("nullModel"));
mu->setVal(1.0);
model->plotOn(frame2, RooFit::LineColor(kRed), RooFit::Name("altModel"));
TLegend *leg2 = new TLegend(0.65,0.73,0.86,0.87);
//leg->AddEntry(frame2->findObject("wwModel"), "SM WW Scattering model with background",
// "lep");
leg2->AddEntry(frame2->findObject("nullModel"),
"SM + Background", "lep");
leg2->AddEntry(frame2->findObject("altModel"),
"Effective Operator + Background", "lep");
frame2->SetTitle("");
frame2->GetXaxis()->SetTitle("M_{WW} (GeV)");
frame2->GetYaxis()->SetTitle("");
frame2->Draw();
leg2->Draw();
canvas2->Write();
outputFile->Close();
RooArgSet poi(*mu);
RooArgSet *nullParams = (RooArgSet*) poi.snapshot();
nullParams->setRealValue("mu", 0.0);
RooStats::ProfileLikelihoodCalculator plc(*generatedData, *model, poi, 0.05, nullParams);
RooStats::HypoTestResult* htr = plc.GetHypoTest();
std::cerr << "P Value = " << htr->NullPValue() << "\n";
return htr->Significance();
}
