void FitterUtils::fit(bool wantplot, bool constPartReco,
double fracPartReco_const,
ofstream& out, TTree* t, bool update, string plotsfile)
{
//***************Get the PDFs from the workspace
TFile fw(workspacename.c_str());
RooWorkspace* workspace = (RooWorkspace*)fw.Get("workspace");
RooRealVar *B_plus_M = workspace->var("B_plus_M");
RooRealVar *misPT = workspace->var("misPT");
RooRealVar *T = workspace->var("T");
RooRealVar *n = workspace->var("n");
RooRealVar *expoConst = workspace->var("expoConst");
RooRealVar *trueExp = workspace->var("trueExp");
RooRealVar *fractionalErrorJpsiLeak = workspace->var("fractionalErrorJpsiLeak");
cout<<"VALUE OF T IN FIT: "<<T->getVal()<<" +- "<<T->getError()<<endl;
cout<<"VALUE OF n IN FIT: "<<n->getVal()<<" +- "<<n->getError()<<endl;
RooHistPdf *histPdfSignalZeroGamma = (RooHistPdf *) workspace->pdf("histPdfSignalZeroGamma");
RooHistPdf *histPdfSignalOneGamma = (RooHistPdf *) workspace->pdf("histPdfSignalOneGamma");
RooHistPdf *histPdfSignalTwoGamma = (RooHistPdf *) workspace->pdf("histPdfSignalTwoGamma");
RooHistPdf *histPdfPartReco = (RooHistPdf *) workspace->pdf("histPdfPartReco");
RooHistPdf *histPdfJpsiLeak(0);
if(nGenJpsiLeak>0) histPdfJpsiLeak = (RooHistPdf *) workspace->pdf("histPdfJpsiLeak");
RooAbsPdf *combPDF;
if (fit2D)
{
combPDF = new RooPTMVis("combPDF", "combPDF", *misPT, *B_plus_M, *T, *n, *expoConst);
}
else
{
combPDF = new RooExponential("combPDF", "combPDF", *B_plus_M, *expoConst);
}
expoConst->setVal(trueExp->getVal());
RooWorkspace* workspaceGen = (RooWorkspace*)fw.Get("workspaceGen");
RooDataSet* dataGenSignalZeroGamma = (RooDataSet*)workspaceGen->data("dataGenSignalZeroGamma");
RooDataSet* dataGenSignalOneGamma = (RooDataSet*)workspaceGen->data("dataGenSignalOneGamma");
RooDataSet* dataGenSignalTwoGamma = (RooDataSet*)workspaceGen->data("dataGenSignalTwoGamma");
RooDataSet* dataGenPartReco = (RooDataSet*)workspaceGen->data("dataGenPartReco");
RooDataSet* dataGenComb = (RooDataSet*)workspaceGen->data("dataGenComb");
RooDataSet* dataGenJpsiLeak(0);
if(nGenJpsiLeak>0) dataGenJpsiLeak = (RooDataSet*)workspaceGen->data("dataGenJpsiLeak");
if(wantplot)
{
//**************Must get the datasets
RooDataSet* dataSetSignalZeroGamma = (RooDataSet*)workspace->data("dataSetSignalZeroGamma");
RooDataSet* dataSetSignalOneGamma = (RooDataSet*)workspace->data("dataSetSignalOneGamma");
RooDataSet* dataSetSignalTwoGamma = (RooDataSet*)workspace->data("dataSetSignalTwoGamma");
RooDataSet* dataSetPartReco = (RooDataSet*)workspace->data("dataSetPartReco");
RooDataSet* dataSetComb = (RooDataSet*)workspace->data("dataSetComb");
RooDataSet* dataSetJpsiLeak = (RooDataSet*)workspace->data("dataSetJpsiLeak");
//**************Plot all the different components
cout<<"dataGenSignalZeroGamma: "<<dataGenSignalZeroGamma<<endl;
PlotShape(*dataSetSignalZeroGamma, *dataGenSignalZeroGamma, *histPdfSignalZeroGamma, plotsfile, "cSignalZeroGamma", *B_plus_M, *misPT);
PlotShape(*dataSetSignalOneGamma, *dataGenSignalOneGamma, *histPdfSignalOneGamma, plotsfile, "cSignalOneGamma", *B_plus_M, *misPT);
PlotShape(*dataSetSignalTwoGamma, *dataGenSignalTwoGamma, *histPdfSignalTwoGamma, plotsfile, "cSignalTwoGamma", *B_plus_M, *misPT);
PlotShape(*dataSetPartReco, *dataGenPartReco, *histPdfPartReco, plotsfile, "cPartReco", *B_plus_M, *misPT);
PlotShape(*dataSetComb, *dataGenComb, *combPDF, plotsfile, "cComb", *B_plus_M, *misPT);
if(nGenJpsiLeak>1) PlotShape(*dataSetJpsiLeak, *dataGenJpsiLeak, *histPdfJpsiLeak, plotsfile, "cJpsiLeak", *B_plus_M, *misPT);
}
//***************Merge datasets
RooDataSet* dataGenTot(dataGenPartReco);
dataGenTot->append(*dataGenSignalZeroGamma);
dataGenTot->append(*dataGenSignalOneGamma);
dataGenTot->append(*dataGenSignalTwoGamma);
dataGenTot->append(*dataGenComb);
if(nGenJpsiLeak>0) dataGenTot->append(*dataGenJpsiLeak);
//**************Prepare fitting function
RooRealVar nSignal("nSignal", "#signal events", 1.*nGenSignal, nGenSignal-7*sqrt(nGenSignal), nGenSignal+7*sqrt(nGenSignal));
RooRealVar nPartReco("nPartReco", "#nPartReco", 1.*nGenPartReco, nGenPartReco-7*sqrt(nGenPartReco), nGenPartReco+7*sqrt(nGenPartReco));
RooRealVar nComb("nComb", "#nComb", 1.*nGenComb, nGenComb-7*sqrt(nGenComb), nGenComb+7*sqrt(nGenComb));
RooRealVar nJpsiLeak("nJpsiLeak", "#nJpsiLeak", 1.*nGenJpsiLeak, nGenJpsiLeak-7*sqrt(nGenJpsiLeak), nGenJpsiLeak+7*sqrt(nGenJpsiLeak));
RooRealVar fracZero("fracZero", "fracZero",0.5,0,1);
RooRealVar fracOne("fracOne", "fracOne",0.5, 0,1);
RooFormulaVar fracPartReco("fracPartReco", "nPartReco/nSignal", RooArgList(nPartReco,nSignal));
RooFormulaVar fracOneRec("fracOneRec", "(1-fracZero)*fracOne", RooArgList(fracZero, fracOne));
RooAddPdf histPdfSignal("histPdfSignal", "histPdfSignal", RooArgList(*histPdfSignalZeroGamma, *histPdfSignalOneGamma, *histPdfSignalTwoGamma), RooArgList(fracZero, fracOneRec));
RooArgList pdfList(histPdfSignal, *histPdfPartReco, *combPDF);
RooArgList yieldList(nSignal, nPartReco, nComb);
if(nGenJpsiLeak>0)
{
pdfList.add(*histPdfJpsiLeak);
yieldList.add(nJpsiLeak);
}
RooAddPdf totPdf("totPdf", "totPdf", pdfList, yieldList);
//**************** Constrain the fraction of zero and one photon
int nGenSignalZeroGamma(floor(nGenFracZeroGamma*nGenSignal));
int nGenSignalOneGamma(floor(nGenFracOneGamma*nGenSignal));
int nGenSignalTwoGamma(floor(nGenSignal-nGenSignalZeroGamma-nGenSignalOneGamma));
RooRealVar fracZeroConstMean("fracZeroConstMean", "fracZeroConstMean", nGenSignalZeroGamma*1./nGenSignal);
RooRealVar fracZeroConstSigma("fracZeroConstSigma", "fracZeroConstSigma", sqrt(nGenSignalZeroGamma)/nGenSignal);
RooGaussian fracZeroConst("fracZeroConst", "fracZeroConst", fracZero, fracZeroConstMean, fracZeroConstSigma);
RooRealVar fracOneConstMean("fracOneConstMean", "fracOneConstMean", nGenSignalOneGamma*1./nGenSignal/(1-fracZeroConstMean.getVal()));
RooRealVar fracOneConstSigma("fracOneConstSigma", "fracOneConstSigma", sqrt(nGenSignalOneGamma)/nGenSignal/(1-fracZeroConstMean.getVal()));
RooGaussian fracOneConst("fracOneConst", "fracOneConst", fracOne, fracOneConstMean, fracOneConstSigma);
RooRealVar fracPartRecoMean("fracPartRecoMean", "fracPartRecoMean", nGenPartReco/(1.*nGenSignal));
RooRealVar fracPartRecoSigma("fracPartRecoSigma", "fracPartRecoSigma", fracPartReco_const*fracPartRecoMean.getVal());
RooGaussian fracPartRecoConst("fracPartRecoConst", "fracPartRecoConst", fracPartReco, fracPartRecoMean, fracPartRecoSigma);
RooRealVar JpsiLeakMean("JpsiLeakMean", "JpsiLeakMean", nGenJpsiLeak);
RooRealVar JpsiLeakSigma("JpsiLeakSigma", "JpsiLeakSigma", nGenJpsiLeak*fractionalErrorJpsiLeak->getVal());
RooGaussian JpsiLeakConst("JpsiLeakConst", "JpsiLeakConst", nJpsiLeak, JpsiLeakMean, JpsiLeakSigma);
//Extra TEST CONSTRAINT
//RooRealVar combConstMean("combConstMean", "combConstMean", nGenComb);
//RooRealVar combConstSigma("combConstSigma", "combConstSigma", 7.7);
//RooGaussian combConst("combConst", "combConst", nComb, combConstMean, combConstSigma);
//**************** fit
RooAbsReal::defaultIntegratorConfig()->setEpsAbs(1e-8) ;
RooAbsReal::defaultIntegratorConfig()->setEpsRel(1e-8) ;
RooArgSet *par_set = totPdf.getParameters(dataGenTot);
initiateParams(nGenSignalZeroGamma, nGenSignalOneGamma, nGenSignalTwoGamma,
*trueExp, nSignal, nPartReco, nComb, fracZero, fracOne, *expoConst, nJpsiLeak, constPartReco, fracPartRecoSigma);
RooArgSet constraints(fracZeroConst, fracOneConst);
if (constPartReco) constraints.add(fracPartRecoConst);
if(nGenJpsiLeak>0) constraints.add(JpsiLeakConst);
RooAbsReal* nll = totPdf.createNLL(*dataGenTot, Extended(), ExternalConstraints(constraints));
RooMinuit minuit(*nll);
minuit.setStrategy(2);
int migradRes(1);
int hesseRes(4);
vector<int> migradResVec;
vector<int> hesseResVec;
double edm(10);
int nrefit(0);
RooFitResult* fitRes(0);
vector<RooFitResult*> fitResVec;
bool hasConverged(false);
for(int i(0); (i<10) && !hasConverged ; ++i)
{
initiateParams(nGenSignalZeroGamma, nGenSignalOneGamma, nGenSignalTwoGamma,
*trueExp, nSignal, nPartReco, nComb, fracZero, fracOne, *expoConst, nJpsiLeak, constPartReco, fracPartRecoSigma);
cout<<"FITTING: starting with nsignal = "<<nSignal.getValV()<<" refit nbr. "<<i<<endl;
//if(fitRes != NULL && fitRes != 0) delete fitRes;
migradRes = minuit.migrad();
hesseRes = minuit.hesse();
fitRes = minuit.save();
edm = fitRes->edm();
fitResVec.push_back(fitRes);
migradResVec.push_back(migradRes);
hesseResVec.push_back(hesseRes);
if( migradRes == 0 && hesseRes == 0 && edm < 1e-4 ) hasConverged = true;
++nrefit;
cout<<"Fitting nbr "<<i<<" done. Hesse: "<<hesseRes<<" migrad: "<<migradRes<<" edm: "<<edm<<" minNll: "<<fitRes->minNll()<<endl;
}
if(!hasConverged)
{
double minNll(1e20);
int minIndex(-1);
for(unsigned int i(0); i<fitResVec.size(); ++i)
{
if( fitResVec.at(i)->minNll() < minNll)
{
minIndex = i;
minNll = fitResVec[i]->minNll();
}
}
migradRes = migradResVec.at(minIndex);
hesseRes = hesseResVec.at(minIndex);
cout<<"Fit not converged, choose fit "<<minIndex<<". Hesse: "<<hesseRes<<" migrad: "<<migradRes<<" edm: "<<edm<<" minNll: "<<fitRes->minNll()<<endl;
}
fillTreeResult(t, fitRes, update, migradRes, hesseRes, hasConverged);
for(unsigned int i(0); i<fitResVec.size(); ++i) delete fitResVec.at(i);
//totPdf.fitTo(*dataGenTot, Extended(), Save(), Warnings(false));
//*************** output fit status
int w(12);
out<<setw(w)<<migradRes<<setw(w)<<hesseRes<<setw(w)<<edm<<setw(w)<<nrefit<<endl;
if(wantplot) plot_fit_result(plotsfile, totPdf, *dataGenTot);
fw.Close();
//delete and return
delete nll;
delete par_set;
delete workspace;
delete workspaceGen;
delete combPDF;
}
void FitterUtilsSimultaneousExpOfPolyTimesX::fit(bool wantplot, bool constPartReco,
double fracPartReco_const,
ofstream& out, TTree* t, bool update, string plotsfile)
{
//***************Get the PDFs from the workspace
TFile fw(workspacename.c_str());
RooWorkspace* workspace = (RooWorkspace*)fw.Get("workspace");
RooRealVar *B_plus_M = workspace->var("B_plus_M");
RooRealVar *misPT = workspace->var("misPT");
RooRealVar *l1Kee = workspace->var("l1Kee");
RooRealVar *l2Kee = workspace->var("l2Kee");
RooRealVar *l3Kee = workspace->var("l3Kee");
RooRealVar *l4Kee = workspace->var("l4Kee");
RooRealVar *l5Kee = workspace->var("l5Kee");
RooRealVar *l1KeeGen = workspace->var("l1KeeGen");
RooRealVar *l2KeeGen = workspace->var("l2KeeGen");
RooRealVar *l3KeeGen = workspace->var("l3KeeGen");
RooRealVar *l4KeeGen = workspace->var("l4KeeGen");
RooRealVar *l5KeeGen = workspace->var("l5KeeGen");
RooRealVar *fractionalErrorJpsiLeak = workspace->var("fractionalErrorJpsiLeak");
RooRealVar l1Kemu(*l1Kee);
l1Kemu.SetName("l1Kemu"); l1Kemu.SetTitle("l1Kemu");
RooRealVar l2Kemu(*l2Kee);
l2Kemu.SetName("l2Kemu"); l2Kemu.SetTitle("l2Kemu");
RooRealVar l3Kemu(*l3Kee);
l3Kemu.SetName("l3Kemu"); l3Kemu.SetTitle("l3Kemu");
RooRealVar l4Kemu(*l4Kee);
l4Kemu.SetName("l4Kemu"); l4Kemu.SetTitle("l4Kemu");
RooRealVar l5Kemu(*l5Kee);
l5Kemu.SetName("l5Kemu"); l5Kemu.SetTitle("l5Kemu");
RooHistPdf *histPdfSignalZeroGamma = (RooHistPdf *) workspace->pdf("histPdfSignalZeroGamma");
RooHistPdf *histPdfSignalOneGamma = (RooHistPdf *) workspace->pdf("histPdfSignalOneGamma");
RooHistPdf *histPdfSignalTwoGamma = (RooHistPdf *) workspace->pdf("histPdfSignalTwoGamma");
RooHistPdf *histPdfPartReco = (RooHistPdf *) workspace->pdf("histPdfPartReco");
RooHistPdf *histPdfJpsiLeak(0);
if(nGenJpsiLeak>0) histPdfJpsiLeak = (RooHistPdf *) workspace->pdf("histPdfJpsiLeak");
//Here set in the Kemu PDF the parameters that have to be shared
RooExpOfPolyTimesX* combPDF = new RooExpOfPolyTimesX("combPDF", "combPDF", *B_plus_M, *misPT, *l1Kee, *l2Kee, *l3Kee, *l4Kee, *l5Kee);
RooExpOfPolyTimesX* KemuPDF = new RooExpOfPolyTimesX("kemuPDF", "kemuPDF", *B_plus_M, *misPT, l1Kemu, *l2Kee, *l3Kee, *l4Kee, *l5Kee);
RooWorkspace* workspaceGen = (RooWorkspace*)fw.Get("workspaceGen");
RooDataSet* dataGenSignalZeroGamma = (RooDataSet*)workspaceGen->data("dataGenSignalZeroGamma");
RooDataSet* dataGenSignalOneGamma = (RooDataSet*)workspaceGen->data("dataGenSignalOneGamma");
RooDataSet* dataGenSignalTwoGamma = (RooDataSet*)workspaceGen->data("dataGenSignalTwoGamma");
RooDataSet* dataGenPartReco = (RooDataSet*)workspaceGen->data("dataGenPartReco");
RooDataSet* dataGenComb = (RooDataSet*)workspaceGen->data("dataGenComb");
RooDataSet* dataGenKemu = (RooDataSet*)workspaceGen->data("dataGenKemu");
RooDataSet* dataGenJpsiLeak(0);
if(nGenJpsiLeak>0) dataGenJpsiLeak = (RooDataSet*)workspaceGen->data("dataGenJpsiLeak");
if(wantplot)
{
//**************Must get the datasets
RooDataSet* dataSetSignalZeroGamma = (RooDataSet*)workspace->data("dataSetSignalZeroGamma");
RooDataSet* dataSetSignalOneGamma = (RooDataSet*)workspace->data("dataSetSignalOneGamma");
RooDataSet* dataSetSignalTwoGamma = (RooDataSet*)workspace->data("dataSetSignalTwoGamma");
RooDataSet* dataSetPartReco = (RooDataSet*)workspace->data("dataSetPartReco");
RooDataSet* dataSetComb = (RooDataSet*)workspace->data("dataSetComb");
RooDataSet* dataSetJpsiLeak = (RooDataSet*)workspace->data("dataSetJpsiLeak");
//**************Plot all the different components
cout<<"dataGenSignalZeroGamma: "<<dataGenSignalZeroGamma<<endl;
PlotShape(*dataSetSignalZeroGamma, *dataGenSignalZeroGamma, *histPdfSignalZeroGamma, plotsfile, "cSignalZeroGamma", *B_plus_M, *misPT);
PlotShape(*dataSetSignalOneGamma, *dataGenSignalOneGamma, *histPdfSignalOneGamma, plotsfile, "cSignalOneGamma", *B_plus_M, *misPT);
PlotShape(*dataSetSignalTwoGamma, *dataGenSignalTwoGamma, *histPdfSignalTwoGamma, plotsfile, "cSignalTwoGamma", *B_plus_M, *misPT);
PlotShape(*dataSetPartReco, *dataGenPartReco, *histPdfPartReco, plotsfile, "cPartReco", *B_plus_M, *misPT);
PlotShape(*dataSetComb, *dataGenComb, *combPDF, plotsfile, "cComb", *B_plus_M, *misPT);
if(nGenJpsiLeak>1) PlotShape(*dataSetJpsiLeak, *dataGenJpsiLeak, *histPdfJpsiLeak, plotsfile, "cJpsiLeak", *B_plus_M, *misPT);
}
//***************Merge datasets
RooDataSet* dataGenTot(dataGenPartReco);
dataGenTot->append(*dataGenSignalZeroGamma);
dataGenTot->append(*dataGenSignalOneGamma);
dataGenTot->append(*dataGenSignalTwoGamma);
dataGenTot->append(*dataGenComb);
if(nGenJpsiLeak>0) dataGenTot->append(*dataGenJpsiLeak);
//**************Create index category and join samples
RooCategory category("category", "category");
category.defineType("Kee");
category.defineType("Kemu");
RooDataSet dataGenSimultaneous("dataGenSimultaneous", "dataGenSimultaneous", RooArgSet(*B_plus_M, *misPT), Index(category), Import("Kee", *dataGenTot), Import("Kemu", *dataGenKemu));
//**************Prepare fitting function
RooRealVar nSignal("nSignal", "#signal events", 1.*nGenSignal, nGenSignal-7*sqrt(nGenSignal), nGenSignal+7*sqrt(nGenSignal));
RooRealVar nPartReco("nPartReco", "#nPartReco", 1.*nGenPartReco, nGenPartReco-7*sqrt(nGenPartReco), nGenPartReco+7*sqrt(nGenPartReco));
RooRealVar nComb("nComb", "#nComb", 1.*nGenComb, nGenComb-7*sqrt(nGenComb), nGenComb+7*sqrt(nGenComb));
RooRealVar nKemu("nKemu", "#nKemu", 1.*nGenKemu, nGenKemu-7*sqrt(nGenKemu), nGenKemu+7*sqrt(nGenKemu));
RooRealVar nJpsiLeak("nJpsiLeak", "#nJpsiLeak", 1.*nGenJpsiLeak, nGenJpsiLeak-7*sqrt(nGenJpsiLeak), nGenJpsiLeak+7*sqrt(nGenJpsiLeak));
RooRealVar fracZero("fracZero", "fracZero",0.5,0,1);
RooRealVar fracOne("fracOne", "fracOne",0.5, 0,1);
RooFormulaVar fracPartReco("fracPartReco", "nPartReco/nSignal", RooArgList(nPartReco,nSignal));
RooFormulaVar fracOneRec("fracOneRec", "(1-fracZero)*fracOne", RooArgList(fracZero, fracOne));
RooAddPdf histPdfSignal("histPdfSignal", "histPdfSignal", RooArgList(*histPdfSignalZeroGamma, *histPdfSignalOneGamma, *histPdfSignalTwoGamma), RooArgList(fracZero, fracOneRec));
RooArgList pdfList(histPdfSignal, *histPdfPartReco, *combPDF);
RooArgList yieldList(nSignal, nPartReco, nComb);
if(nGenJpsiLeak>0)
{
pdfList.add(*histPdfJpsiLeak);
yieldList.add(nJpsiLeak);
}
RooAddPdf totPdf("totPdf", "totPdf", pdfList, yieldList);
RooExtendPdf totKemuPdf("totKemuPdf", "totKemuPdf", *KemuPDF, nKemu);
//**************** Prepare simultaneous PDF
RooSimultaneous simPdf("simPdf", "simPdf", category);
simPdf.addPdf(totPdf, "Kee");
simPdf.addPdf(totKemuPdf, "Kemu");
//**************** Constrain the fraction of zero and one photon
int nGenSignalZeroGamma(floor(nGenFracZeroGamma*nGenSignal));
int nGenSignalOneGamma(floor(nGenFracOneGamma*nGenSignal));
int nGenSignalTwoGamma(floor(nGenSignal-nGenSignalZeroGamma-nGenSignalOneGamma));
RooRealVar fracZeroConstMean("fracZeroConstMean", "fracZeroConstMean", nGenSignalZeroGamma*1./nGenSignal);
RooRealVar fracZeroConstSigma("fracZeroConstSigma", "fracZeroConstSigma", sqrt(nGenSignalZeroGamma)/nGenSignal);
RooGaussian fracZeroConst("fracZeroConst", "fracZeroConst", fracZero, fracZeroConstMean, fracZeroConstSigma);
RooRealVar fracOneConstMean("fracOneConstMean", "fracOneConstMean", nGenSignalOneGamma*1./nGenSignal/(1-fracZeroConstMean.getVal()));
RooRealVar fracOneConstSigma("fracOneConstSigma", "fracOneConstSigma", sqrt(nGenSignalOneGamma)/nGenSignal/(1-fracZeroConstMean.getVal()));
RooGaussian fracOneConst("fracOneConst", "fracOneConst", fracOne, fracOneConstMean, fracOneConstSigma);
RooRealVar fracPartRecoMean("fracPartRecoMean", "fracPartRecoMean", nGenPartReco/(1.*nGenSignal));
RooRealVar fracPartRecoSigma("fracPartRecoSigma", "fracPartRecoSigma", fracPartReco_const*fracPartRecoMean.getVal());
RooGaussian fracPartRecoConst("fracPartRecoConst", "fracPartRecoConst", fracPartReco, fracPartRecoMean, fracPartRecoSigma);
RooRealVar JpsiLeakMean("JpsiLeakMean", "JpsiLeakMean", nGenJpsiLeak);
RooRealVar JpsiLeakSigma("JpsiLeakSigma", "JpsiLeakSigma", nGenJpsiLeak*fractionalErrorJpsiLeak->getVal());
RooGaussian JpsiLeakConst("JpsiLeakConst", "JpsiLeakConst", nJpsiLeak, JpsiLeakMean, JpsiLeakSigma);
//**************** fit
RooAbsReal::defaultIntegratorConfig()->setEpsAbs(1e-8) ;
RooAbsReal::defaultIntegratorConfig()->setEpsRel(1e-8) ;
initiateParams(nGenSignalZeroGamma, nGenSignalOneGamma, nGenSignalTwoGamma,
nKemu, nSignal, nPartReco, nComb, fracZero, fracOne,
nJpsiLeak, constPartReco, fracPartRecoSigma,
*l1Kee, *l2Kee, *l3Kee, *l4Kee, *l5Kee, l1Kemu, l2Kemu, l3Kemu, l4Kemu, l5Kemu,
*l1KeeGen, *l2KeeGen, *l3KeeGen, *l4KeeGen, *l5KeeGen);
RooArgSet constraints(fracZeroConst, fracOneConst);
if (constPartReco) constraints.add(fracPartRecoConst);
if(nGenJpsiLeak>0) constraints.add(JpsiLeakConst);
RooAbsReal* nll = simPdf.createNLL(dataGenSimultaneous, Extended(), ExternalConstraints(constraints));
RooMinuit minuit(*nll);
minuit.setStrategy(2);
int migradRes(1);
int hesseRes(4);
vector<int> migradResVec;
vector<int> hesseResVec;
double edm(10);
int nrefit(0);
RooFitResult* fitRes(0);
vector<RooFitResult*> fitResVec;
bool hasConverged(false);
for(int i(0); (i<15) && !hasConverged ; ++i)
{
initiateParams(nGenSignalZeroGamma, nGenSignalOneGamma, nGenSignalTwoGamma,
nKemu, nSignal, nPartReco, nComb, fracZero, fracOne,
nJpsiLeak, constPartReco, fracPartRecoSigma,
*l1Kee, *l2Kee, *l3Kee, *l4Kee, *l5Kee, l1Kemu, l2Kemu, l3Kemu, l4Kemu, l5Kemu,
*l1KeeGen, *l2KeeGen, *l3KeeGen, *l4KeeGen, *l5KeeGen);
cout<<"FITTING: starting with nsignal = "<<nSignal.getValV()<<" refit nbr. "<<i<<endl;
//if(fitRes != NULL && fitRes != 0) delete fitRes;
migradRes = minuit.migrad();
hesseRes = minuit.hesse();
fitRes = minuit.save();
edm = fitRes->edm();
fitResVec.push_back(fitRes);
migradResVec.push_back(migradRes);
hesseResVec.push_back(hesseRes);
if( migradRes == 0 && hesseRes == 0 && edm < 1e-3 ) hasConverged = true;
++nrefit;
cout<<"Fitting nbr "<<i<<" done. Hesse: "<<hesseRes<<" migrad: "<<migradRes<<" edm: "<<edm<<" minNll: "<<fitRes->minNll()<<endl;
}
if(!hasConverged)
{
double minNll(1e20);
int minIndex(-1);
for(unsigned int i(0); i<fitResVec.size(); ++i)
{
if( fitResVec.at(i)->minNll() < minNll)
{
minIndex = i;
minNll = fitResVec[i]->minNll();
}
}
migradRes = migradResVec.at(minIndex);
hesseRes = hesseResVec.at(minIndex);
cout<<"Fit not converged, choose fit "<<minIndex<<". Hesse: "<<hesseRes<<" migrad: "<<migradRes<<" edm: "<<edm<<" minNll: "<<fitRes->minNll()<<endl;
}
fillTreeResult(t, fitRes, update, migradRes, hesseRes, hasConverged);
for(unsigned int i(0); i<fitResVec.size(); ++i) delete fitResVec.at(i);
//totPdf.fitTo(*dataGenTot, Extended(), Save(), Warnings(false));
//*************** output fit status
int w(12);
out<<setw(w)<<migradRes<<setw(w)<<hesseRes<<setw(w)<<edm<<setw(w)<<nrefit<<endl;
if(wantplot) plot_fit_result(plotsfile, totPdf, *dataGenTot);
if(wantplot) plot_kemu_fit_result(plotsfile, totKemuPdf, *dataGenKemu);
fw.Close();
//delete and return
delete nll;
delete workspace;
delete workspaceGen;
delete combPDF;
delete KemuPDF;
}
void vertexAssociationFit(TFile* input, TFile* output) {
// style
gStyle->SetOptStat(0);
gStyle->SetOptFit(111);
gStyle->SetOptTitle(1);
// margins
gStyle->SetStatFormat("6.4g");
gStyle->SetFitFormat("5.2g");
gStyle->SetPadTopMargin(0.07);
gStyle->SetPadBottomMargin(0.12);
gStyle->SetPadLeftMargin(0.1);
gStyle->SetPadRightMargin(0.02);
// prepare the output
// * canvas to put the fit per bin
TCanvas* canvas_0 = new TCanvas("FitPerBin","Fit for each vertex bin",1200,900);
canvas_0->Divide(4,4);
// * canvas to put the plots per bin
TCanvas* canvas_1 = new TCanvas("EffPerBin","Efficiency vs purity for each vertex bin",1200,900);
canvas_1->Divide(4,4);
// * canvas with other plots: slope vs nvertices, efficiency and purity for fixed cut, etc.
TCanvas* canvas_2 = new TCanvas("jvAssociation","Efficiency vs purity summary plots",1000,600);
canvas_2->Divide(4,2);
// summary graphs
TGraphErrors* slopeGraph = new TGraphErrors(nbins);
slopeGraph->SetNameTitle("slopeGraph","expo slope in each n_PU bin");
TGraphErrors* constGraph = new TGraphErrors(nbins);
constGraph->SetNameTitle("constGraph","expo normalization in each n_PU bin");
TGraphErrors* cuteffGraph = new TGraphErrors(nbins);
cuteffGraph->SetNameTitle("cuteffGraph","Efficiency in each n_PU bin");
TGraphErrors* cutpurGraph = new TGraphErrors(nbins);
cutpurGraph->SetNameTitle("cutpurGraph","Purity in each n_PU bin");
TGraphErrors* effcutGraph = new TGraphErrors(nbins);
effcutGraph->SetNameTitle("effcutGraph","Cut for fixed efficiency in each n_PU bin");
TGraphErrors* effpurGraph = new TGraphErrors(nbins);
effpurGraph->SetNameTitle("effpurGraph","Purity for fixed efficiency in each n_PU bin");
TGraphErrors* purcutGraph = new TGraphErrors(nbins);
purcutGraph->SetNameTitle("purcutGraph","Cut for fixed purity in each n_PU bin");
TGraphErrors* pureffGraph = new TGraphErrors(nbins);
pureffGraph->SetNameTitle("pureffGraph","Efficiency for fixed purity in each n_PU bin");
// loop over the directories
double slope = 0.;
double slopeError = -999.;
double constnorm = 0.;
double constError = -999.;
TGraphErrors* result = NULL;
TFitResultPtr fitRes(-1);
double x,y, xe,ye;
unsigned int index;
TString histonames[nbins+1];
TString labelnames[nbins+1];
// get the histogram
TH1* evts_per_PUbin = (TH1*)input->Get("finaldistros_ssvhpt/GoodJet/GoodJet_nvertices");
double sumTot(0), effTot(0);
TLatex *latexLabel = new TLatex();
latexLabel->SetTextSize(0.05);
latexLabel->SetTextFont(42);
latexLabel->SetLineWidth(2);
latexLabel->SetNDC();
TLatex *latexLabel2 = new TLatex();
latexLabel2->SetTextSize(0.09);
latexLabel2->SetTextFont(42);
latexLabel2->SetLineWidth(2);
latexLabel2->SetNDC();
for(unsigned int i=0; i<=nbins; ++i) {
// get the histogram
labelnames[i]=Form("n_PU = %d bin",i);
std::cout<<"labelnames["<<i<<"]"<<labelnames[i]<<std::endl;
histonames[i]=Form("finaldistros_ssvhpt/GoodJet/GoodJet_jetvtx_ratio2b_%d_PUvtx",i);
std::cout<<"seizing: histonames["<<i<<"]:"<<histonames[i]<<std::endl;
TH1* fraction = (TH1*)input->Get(histonames[i]);
//fraction->Rebin(2);
// do it
canvas_0->cd(i+1);
fraction->SetTitle("");
MakeNiceHistoStyle(fraction);
if(fraction) vertexAssociationAnalysis(fraction,i,slope,slopeError,constnorm,constError,result,fitRes);
latexLabel2->DrawLatex(0.15, 0.85,labelnames[i]);
latexLabel2->DrawLatex(0.15, 0.75,"CMS #sqrt{s}= 7 TeV");
gPad->Draw();
gPad->SetLogy();
MakeNiceStatBox(fraction);
output->cd();
fraction->Write();
// plot
canvas_1->cd(i+1);
result->SetTitle("");
MakeNiceGraphStyle(result);
result->Draw("alP");
latexLabel->DrawLatex(0.05, 0.95,labelnames[i]);
result->GetXaxis()->SetTitle("Purity");
result->GetYaxis()->SetTitle("Efficiency");
result->GetXaxis()->SetRangeUser(0.8,1.2);
result->GetYaxis()->SetRangeUser(0.85,1.0);
output->cd();
result->Write();
// extract other quantities
slopeGraph->SetPoint(i,i,slope);
slopeGraph->SetPointError(i,0,slopeError);
slopeGraph->GetXaxis()->SetTitle("n_{PU} vertices");
constGraph->SetPoint(i,i,constnorm);
constGraph->SetPointError(i,0,constError);
constGraph->GetXaxis()->SetTitle("n_{PU} vertices");
index = unsigned(((defaultCut-cutmin)/(cutmax-cutmin)*nSteps)+0.5);
findWorkingPoint(result,index,x,y,xe,ye,1);
// to compute the weighted efficiency
int ivtx_bin=evts_per_PUbin->FindBin(i+1);
sumTot+=evts_per_PUbin->GetBinContent(ivtx_bin);
effTot+=y*evts_per_PUbin->GetBinContent(ivtx_bin);
cuteffGraph->SetPoint(i,i,y);
cuteffGraph->SetPointError(i,0,ye);
cuteffGraph->GetXaxis()->SetTitle("n_{PU} vertices");
cutpurGraph->SetPoint(i,i,x);
cutpurGraph->SetPointError(i,0,xe);
cutpurGraph->GetXaxis()->SetTitle("n_{PU} vertices");
x = defaultPur;
findWorkingPoint(result,index,x,y,xe,ye,2);
purcutGraph->SetPoint(i,i,cutmin+(cutmax-cutmin)/nSteps*index);
purcutGraph->GetXaxis()->SetTitle("n_{PU} vertices");
pureffGraph->SetPoint(i,i,y);
pureffGraph->SetPointError(i,0,ye);
pureffGraph->GetXaxis()->SetTitle("n_{PU} vertices");
y = defaultEff;
findWorkingPoint(result,index,x,y,xe,ye,3);
effcutGraph->SetPoint(i,i,cutmin+(cutmax-cutmin)/nSteps*index);
effcutGraph->GetXaxis()->SetTitle("n_{PU} vertices");
effpurGraph->SetPoint(i,i,x);
effpurGraph->SetPointError(i,0,xe);
effpurGraph->GetXaxis()->SetTitle("n_{PU} vertices");
std::cout<<"end of other quantities: bin["<<i<<"]"<<std::endl;
}
std::cout << "Weighted Efficiency" << effTot/sumTot << std::endl;
// plot summary graphs
TF1* f=NULL;
canvas_2->cd(1);
MakeNiceGraphStyle(slopeGraph);
slopeGraph->Draw("alP");
slopeGraph->Fit("pol1");
f = slopeGraph->GetFunction("pol1");
if(f->GetParError(1)>fabs(f->GetParameter(1))) slopeGraph->Fit("pol0");
canvas_2->cd(2);
MakeNiceGraphStyle(constGraph);
constGraph->Draw("alP");
constGraph->Fit("pol2");
f = constGraph->GetFunction("pol2");
canvas_2->cd(3);
MakeNiceGraphStyle(cuteffGraph);
cuteffGraph->Draw("alP");
cuteffGraph->Fit("pol1");
f = cuteffGraph->GetFunction("pol1");
if(f->GetParError(1)>fabs(f->GetParameter(1))) cuteffGraph->Fit("pol0");
latexLabel2->DrawLatex(0.15, 0.15,Form("Jet Fraction cut: %.2f",defaultCut));
canvas_2->cd(4);
MakeNiceGraphStyle(cutpurGraph);
cutpurGraph->Draw("alP");
cutpurGraph->Fit("pol1");
f = cutpurGraph->GetFunction("pol1");
if(f->GetParError(1)>fabs(f->GetParameter(1))) cutpurGraph->Fit("pol0");
latexLabel2->DrawLatex(0.15, 0.15,Form("Jet Fraction cut: %.2f",defaultCut));
canvas_2->cd(5);
MakeNiceGraphStyle(effcutGraph);
effcutGraph->Draw("alP");
effcutGraph->Fit("expo");
latexLabel2->DrawLatex(0.15, 0.15,Form("Fixed efficiency: %.2f",defaultEff));
canvas_2->cd(6);
MakeNiceGraphStyle(effpurGraph);
effpurGraph->Draw("alP");
effpurGraph->Fit("pol1");
f = effpurGraph->GetFunction("pol1");
if(f->GetParError(1)>fabs(f->GetParameter(1))) effpurGraph->Fit("pol0");
latexLabel2->DrawLatex(0.15, 0.15,Form("Fixed efficiency: %.2f",defaultEff));
canvas_2->cd(7);
gPad->SetLeftMargin(0.12);
MakeNiceGraphStyle(purcutGraph);
purcutGraph->Draw("alP");
purcutGraph->Fit("expo");
latexLabel2->DrawLatex(0.15, 0.15,Form("Fixed purity: %.3f",defaultPur));
canvas_2->cd(8);
MakeNiceGraphStyle(pureffGraph);
pureffGraph->Draw("alP");
pureffGraph->Fit("pol1");
f = pureffGraph->GetFunction("pol1");
if(f->GetParError(1)>fabs(f->GetParameter(1))) pureffGraph->Fit("pol0");
latexLabel2->DrawLatex(0.15, 0.15,Form("Fixed purity: %.3f",defaultPur));
// save
output->cd();
canvas_0->Write();
canvas_1->Write();
canvas_2->Write();
canvas_0->SaveAs((TString)(canvas_0->GetName())+".png");
canvas_1->SaveAs((TString)(canvas_1->GetName())+".png");
canvas_2->SaveAs((TString)(canvas_2->GetName())+".png");
}
