| 106 |
|
double y = (static_cast<double>(j) + 0.5)*ystep + ylo; |
| 107 |
|
double sig_cut = 1.0e-5; |
| 108 |
|
double nu = XbinSize * YbinSize * fit_fcn(x, y, xval); |
| 109 |
< |
double sigma = mdef->chisquare_error(nu) < sig_cut |
| 110 |
< |
? sig_cut : mdef->chisquare_error(nu); |
| 109 |
> |
double sigma = fabs(mdef->chisquare_error(0.0)) < sig_cut |
| 110 |
> |
? sig_cut : mdef->chisquare_error(0.0); |
| 111 |
|
|
| 112 |
< |
fsum += xstep * ystep * mdef->get_formula()->Eval(x, y) |
| 112 |
> |
fsum += pow(xstep * ystep * mdef->get_formula()->Eval(x, y), 2.0) |
| 113 |
|
/ sigma / sigma; |
| 114 |
|
} |
| 115 |
|
} |
| 130 |
|
// add errors in data points in histogram |
| 131 |
|
for (int i = 0; i < energy.bins.size(); i++) { |
| 132 |
|
for (int j = 0; j < energy.bins.at(i).size(); j++) { |
| 133 |
< |
double x = (static_cast<double>(i) + 0.5) * (energy.Xhi - energy.Xlo) |
| 133 |
> |
double x = (static_cast<double>(i) + 0.5) * (energy.Xhi - energy.Xlo) |
| 134 |
|
/ static_cast<double>(energy.bins.size()) + energy.Xlo; |
| 135 |
|
double y = (static_cast<double>(j) + 0.5) * (energy.Yhi - energy.Ylo) |
| 136 |
|
/ static_cast<double>(energy.bins.at(i).size()) + energy.Ylo; |
| 137 |
|
double nu = fit_fcn(x, y, xval) * XbinSize * YbinSize; |
| 138 |
|
double sig_cut = 1.0e-5; |
| 139 |
< |
if (mdef->chisquare_error(nu) > sig_cut || !ignorezero) |
| 139 |
> |
if (fabs(mdef->chisquare_error(energy.bins.at(i).at(j))) > sig_cut || !ignorezero) { |
| 140 |
|
chisquare += pow(energy.bins.at(i).at(j) - nu, 2.0) |
| 141 |
< |
/ pow(mdef->chisquare_error(nu), 2.0); |
| 142 |
< |
else |
| 141 |
> |
/ pow(mdef->chisquare_error(energy.bins.at(i).at(j)), 2.0); |
| 142 |
> |
} |
| 143 |
> |
else { |
| 144 |
|
chisquare += pow(energy.bins.at(i).at(j) - nu, 2.0) |
| 145 |
|
/ pow(sig_cut, 2.0); |
| 146 |
+ |
} |
| 147 |
|
} |
| 148 |
|
} |
| 149 |
< |
|
| 149 |
> |
|
| 150 |
|
// add errors due to Gaussians outside histogram |
| 151 |
< |
double eps = 0.1; // accuracy set for this function |
| 151 |
> |
double eps = 0.01; // accuracy set for this function |
| 152 |
|
for (int i = 0; i < ngauss; i++) { |
| 153 |
< |
double *pval = xval+i*mdef->get_formula()->GetNpar(); |
| 153 |
> |
double *pval = xval+i*(mdef->get_formula()->GetNpar()); |
| 154 |
|
double par_x = pval[mdef->get_indiv_max_x()]; |
| 155 |
|
double par_y = pval[mdef->get_indiv_max_y()]; |
| 156 |
|
double par_sig = pval[3]; |
| 157 |
< |
double cutoff_rad = -log(eps * 2.0 * PI * par_sig * par_sig |
| 158 |
< |
/ pval[0]); |
| 157 |
> |
double cutoff_rad = sqrt(-log(eps * 2.0 * PI * par_sig * par_sig |
| 158 |
> |
/ pval[0]) * 2.0 * par_sig * par_sig); |
| 159 |
|
|
| 160 |
|
bool left = par_x < energy.Xlo + cutoff_rad; |
| 161 |
|
bool right = par_x > energy.Xhi - cutoff_rad; |
| 162 |
|
bool top = par_y > energy.Yhi - cutoff_rad; |
| 163 |
< |
bool bottom = par_y > energy.Ylo + cutoff_rad; |
| 163 |
> |
bool bottom = par_y < energy.Ylo + cutoff_rad; |
| 164 |
|
|
| 165 |
|
if (left) { |
| 166 |
|
double xlo = par_x - cutoff_rad; |
| 167 |
|
double xhi = energy.Xlo; |
| 168 |
|
double ylo = energy.Ylo; |
| 169 |
|
double yhi = energy.Yhi; |
| 170 |
< |
chisquare += formula_int(xlo, xhi, ylo, yhi, pval, |
| 170 |
> |
chisquare += 2.0*formula_int(xlo, xhi, ylo, yhi, pval, |
| 171 |
|
XbinSize, YbinSize, xval); |
| 172 |
|
} |
| 173 |
|
if (right) { |
| 175 |
|
double xhi = par_x + cutoff_rad; |
| 176 |
|
double ylo = energy.Ylo; |
| 177 |
|
double yhi = energy.Yhi; |
| 178 |
< |
chisquare += formula_int(xlo, xhi, ylo, yhi, pval, |
| 178 |
> |
chisquare += 2.0*formula_int(xlo, xhi, ylo, yhi, pval, |
| 179 |
|
XbinSize, YbinSize, xval); |
| 180 |
|
} |
| 181 |
|
if (top) { |
| 183 |
|
double xhi = right ? par_x + cutoff_rad : energy.Xhi; |
| 184 |
|
double ylo = energy.Yhi; |
| 185 |
|
double yhi = par_y + cutoff_rad; |
| 186 |
< |
chisquare += formula_int(xlo, xhi, ylo, yhi, pval, |
| 186 |
> |
chisquare += 2.0*formula_int(xlo, xhi, ylo, yhi, pval, |
| 187 |
|
XbinSize, YbinSize, xval); |
| 188 |
|
} |
| 189 |
|
if (bottom) { |
| 191 |
|
double xhi = right ? par_x + cutoff_rad : energy.Xhi; |
| 192 |
|
double ylo = par_y - cutoff_rad; |
| 193 |
|
double yhi = energy.Ylo; |
| 194 |
< |
chisquare += formula_int(xlo, xhi, ylo, yhi, pval, |
| 194 |
> |
chisquare += 2.0*formula_int(xlo, xhi, ylo, yhi, pval, |
| 195 |
|
XbinSize, YbinSize, xval); |
| 196 |
|
} |
| 197 |
|
} |
| 198 |
< |
|
| 198 |
> |
|
| 199 |
|
fcnval = chisquare; |
| 200 |
|
} |
| 201 |
|
catch (exception ex) { |
| 483 |
|
} |
| 484 |
|
} |
| 485 |
|
|
| 486 |
+ |
void fill_histo_with_vec(TH2 *hist, vector< vector<double> > &vec) { |
| 487 |
+ |
for (int i = 0; i < vec.size(); i++) { |
| 488 |
+ |
for (int j = 0; j < vec[i].size(); j++) { |
| 489 |
+ |
hist->SetBinContent(i+1, j+1, vec[i][j]); |
| 490 |
+ |
} |
| 491 |
+ |
} |
| 492 |
+ |
} |
| 493 |
+ |
|
| 494 |
|
void par_init(TH2 *hist, TMinuit *gMinuit, vector<TString> &pars, |
| 495 |
|
double *pval, double *perr, double *plo, double *phi, |
| 496 |
|
int npar, results r) { |
| 499 |
|
if (ngauss <= mdef->get_n_special_par_sets()) { |
| 500 |
|
init_spec_pars = true; |
| 501 |
|
for (int k = 0; k < ngauss; k++) { |
| 502 |
< |
int ipar0 = k*mdef->get_formula()->GetNpar(); // index of par 0 |
| 502 |
> |
int ipar0 = k*mdef->get_formula()->GetNpar(); // index of indiv par 0 |
| 503 |
|
for (int l = 0; l < mdef->get_formula()->GetNpar(); l++) { |
| 504 |
|
int ipar = ipar0 + l; |
| 505 |
|
if (ipar < npar) |
| 519 |
|
vector< vector<double> > sub_energy(energy.bins.size()); |
| 520 |
|
double max_sub, max_x = 0.0, max_y = 0.0; |
| 521 |
|
double pval_other[256]; |
| 522 |
< |
max_sub = -(numeric_limits<double>::infinity()); |
| 522 |
> |
max_sub = -(numeric_limits<double>::max()); |
| 523 |
|
for (int i = 0; i < energy.bins.size(); i++) { |
| 524 |
|
sub_energy[i].resize(energy.bins[i].size()); |
| 525 |
|
for (int j = 0; j < energy.bins[i].size(); j++) { |
| 528 |
|
double y = static_cast<double>(j)*(energy.Yhi - energy.Ylo) |
| 529 |
|
/ static_cast<double>(energy.bins[i].size()) + energy.Ylo; |
| 530 |
|
if (ngauss > 1) { |
| 531 |
< |
// subtract 4*sigma plus integral of fit function |
| 531 |
> |
// subtract integral of fit function |
| 532 |
|
try { |
| 533 |
|
int npval_other = r.pval.at(r.pval.size()-1).size(); |
| 534 |
|
if (npval_other > 256) { |
| 541 |
|
ngauss--; |
| 542 |
|
double nu = fit_fcn(x, y, pval_other) * XbinSize * YbinSize; |
| 543 |
|
ngauss++; |
| 544 |
< |
sub_energy[i][j] = energy.bins[i][j] - nu |
| 535 |
< |
- 4.0*mdef->chisquare_error(nu); |
| 544 |
> |
sub_energy[i][j] = energy.bins[i][j] - nu; |
| 545 |
|
} |
| 546 |
|
catch (exception ex) { |
| 547 |
|
cerr << "Exception in par_init" << endl; |
| 550 |
|
else { |
| 551 |
|
sub_energy[i][j] = energy.bins[i][j]; |
| 552 |
|
} |
| 544 |
– |
if (sub_energy[i][j] > max_sub) { |
| 545 |
– |
max_sub = sub_energy[i][j]; |
| 546 |
– |
max_x = x; |
| 547 |
– |
max_y = y; |
| 548 |
– |
} |
| 553 |
|
} |
| 554 |
|
} |
| 555 |
+ |
TH2D *hist_sub = new TH2D("hist_sub", "Subtracted histo", |
| 556 |
+ |
sub_energy.size(), energy.Xlo, |
| 557 |
+ |
energy.Xhi, |
| 558 |
+ |
sub_energy[0].size(), energy.Ylo, |
| 559 |
+ |
energy.Yhi); |
| 560 |
+ |
fill_histo_with_vec(hist_sub, sub_energy); |
| 561 |
+ |
max_sub = hist_sub->GetSumOfWeights(); |
| 562 |
+ |
max_x = hist_sub->GetMean(1); |
| 563 |
+ |
max_y = hist_sub->GetMean(2); |
| 564 |
|
|
| 565 |
|
for (int i = npar1; i < npar; i++) { |
| 566 |
|
if (mdef->get_indiv_max_E() == i - npar1) { |
| 567 |
|
double nu = 0.0; |
| 568 |
|
if (ngauss > 1) { |
| 569 |
+ |
ngauss--; |
| 570 |
|
nu = fit_fcn(max_x, max_y, pval_other) * XbinSize * YbinSize; |
| 571 |
+ |
ngauss++; |
| 572 |
|
} |
| 573 |
< |
pval[i] = max_sub + (ngauss > 1 ? 4.0*mdef->chisquare_error(nu) |
| 574 |
< |
: 0.0); |
| 560 |
< |
perr[i] = mdef->chisquare_error(pval[i])*0.1; |
| 573 |
> |
pval[i] = max_sub; |
| 574 |
> |
perr[i] = mdef->chisquare_error(pval[i])*0.5; |
| 575 |
|
plo[i] = 0.0; |
| 576 |
|
phi[i] = 1.0e6; |
| 577 |
|
} |
| 679 |
|
|
| 680 |
|
// initialize parameters |
| 681 |
|
par_init(hist, gMinuit, pars, pval, perr, plo, phi, npar, r); |
| 682 |
< |
|
| 682 |
> |
|
| 683 |
|
// minimize |
| 684 |
|
double chisquare, edm, errdef; |
| 685 |
|
int nvpar, nparx; |
| 686 |
|
trouble t; |
| 687 |
|
t.occ = T_NULL; |
| 688 |
|
t.istat = 3; |
| 689 |
< |
// fix the N values and sigmas of the Gaussians |
| 689 |
> |
// fix sigmas of the Gaussians |
| 690 |
|
for (int i = 0; i < ngauss; i++) { |
| 691 |
|
double parno[2]; |
| 692 |
< |
parno[0] = i*npar_indiv + 1; |
| 693 |
< |
parno[1] = i*npar_indiv + 4; |
| 680 |
< |
gMinuit->mnexcm("FIX", parno, 2, erflg); |
| 692 |
> |
parno[0] = i*npar_indiv + 4; |
| 693 |
> |
gMinuit->mnexcm("FIX", parno, 1, erflg); |
| 694 |
|
} |
| 695 |
|
gMinuit->mnexcm("SIMPLEX", 0, 0, erflg); |
| 696 |
|
gMinuit->mnexcm("MIGRAD", 0, 0, erflg); |
| 684 |
– |
// release N values, fix mu_x and mu_y |
| 685 |
– |
for (int i = 0; i < ngauss; i++) { |
| 686 |
– |
double parno[4]; |
| 687 |
– |
parno[0] = i*npar_indiv + 1; |
| 688 |
– |
parno[1] = i*npar_indiv + 2; |
| 689 |
– |
parno[2] = i*npar_indiv + 3; |
| 690 |
– |
parno[3] = i*npar_indiv + 4; |
| 691 |
– |
gMinuit->mnexcm("RELEASE", parno, 1, erflg); |
| 692 |
– |
gMinuit->mnexcm("FIX", parno+1, 2, erflg); |
| 693 |
– |
} |
| 694 |
– |
gMinuit->mnexcm("MIGRAD", 0, 0, erflg); |
| 695 |
– |
// release mu_x and mu_y |
| 696 |
– |
for (int i = 0; i < ngauss; i++) { |
| 697 |
– |
double parno[4]; |
| 698 |
– |
parno[0] = i*npar_indiv + 1; |
| 699 |
– |
parno[1] = i*npar_indiv + 2; |
| 700 |
– |
parno[2] = i*npar_indiv + 3; |
| 701 |
– |
parno[3] = i*npar_indiv + 4; |
| 702 |
– |
gMinuit->mnexcm("RELEASE", parno+1, 2, erflg); |
| 703 |
– |
} |
| 704 |
– |
gMinuit->mnexcm("MIGRAD", 0, 0, erflg); |
| 697 |
|
gMinuit->mnstat(chisquare, edm, errdef, nvpar, nparx, istat); |
| 698 |
|
if (istat == 3) { |
| 699 |
|
/* we're not concerned about MINOS errors right now |
| 758 |
|
} |
| 759 |
|
ndof -= npar; |
| 760 |
|
|
| 761 |
+ |
r.chisquare.push_back(chisquare); |
| 762 |
|
double P = prob(chisquare, ndof); |
| 770 |
– |
cout << "P = "<<P << endl; |
| 763 |
|
if (P > P_cutoff_val) { |
| 764 |
|
break; |
| 765 |
|
} |
