21#include "TrackUtil/Lpav.h"
31static double err_dis_inv(
double x,
double y,
double w,
double a,
double b ) {
32 if ( a == 0 && b == 0 ) {
return w; }
35 double f = x * b - y * a;
36 double rsq = x * x + y * y;
79 if ( m_wsum <= 0 )
return;
80 m_wsum_temp = m_wsum + wi;
81 double rri( xi * xi + yi * yi );
82 double wrri( wi * rri );
83 double wsum_inv( 1 / m_wsum_temp );
84 m_xav = ( m_xsum + wi * xi ) * wsum_inv;
85 m_yav = ( m_ysum + wi * yi ) * wsum_inv;
87 double xxav( ( m_xxsum + wi * xi * xi ) * wsum_inv );
88 double yyav( ( m_yysum + wi * yi * yi ) * wsum_inv );
89 double xyav( ( m_xysum + wi * xi * yi ) * wsum_inv );
90 double xrrav( ( m_xrrsum + xi * wrri ) * wsum_inv );
91 double yrrav( ( m_yrrsum + yi * wrri ) * wsum_inv );
92 double rrrrav( ( m_rrrrsum + wrri * rri ) * wsum_inv );
94 calculate_average_n( xxav, yyav, xyav, xrrav, yrrav, rrrrav );
98 if ( m_wsum <= 0 )
return;
100 double wsum_inv( 1 / m_wsum_temp );
101 m_xav = m_xsum * wsum_inv;
102 m_yav = m_ysum * wsum_inv;
104 double xxav( m_xxsum * wsum_inv );
105 double yyav( m_yysum * wsum_inv );
106 double xyav( m_xysum * wsum_inv );
107 double xrrav( m_xrrsum * wsum_inv );
108 double yrrav( m_yrrsum * wsum_inv );
109 double rrrrav( m_rrrrsum * wsum_inv );
111 calculate_average_n( xxav, yyav, xyav, xrrav, yrrav, rrrrav );
114void Lpav::calculate_average_n(
double xxav,
double yyav,
double xyav,
double xrrav,
115 double yrrav,
double rrrrav ) {
116 double xxav_p = xxav - m_xav * m_xav;
117 double yyav_p = yyav - m_yav * m_yav;
118 double xyav_p = xyav - m_xav * m_yav;
119 double rrav_p = xxav_p + yyav_p;
121 double a = std::fabs( xxav_p - yyav_p );
122 double b = 4 * xyav_p * xyav_p;
123 double asqpb = a * a + b;
124 double rasqpb = std::sqrt( asqpb );
125 double splus = 1 + a / rasqpb;
126 double sminus = b / ( asqpb * splus );
127 splus = std::sqrt( 0.5 * splus );
128 sminus = std::sqrt( 0.5 * sminus );
132 if ( xxav_p <= yyav_p )
145 if ( xyav_p < 0 ) m_sinrot = -m_sinrot;
155 if ( m_cosrot * m_xav + m_sinrot * m_yav <= 0 )
157 m_cosrot = -m_cosrot;
158 m_sinrot = -m_sinrot;
160 m_rscale = std::sqrt( rrav_p );
161 double cos2 = m_cosrot * m_cosrot;
162 double sin2 = m_sinrot * m_sinrot;
163 double cs2 = 2 * m_sinrot * m_cosrot;
164 double rrav_p_inv( 1 / rrav_p );
165 m_xxavp = ( cos2 * xxav_p + cs2 * xyav_p + sin2 * yyav_p ) * rrav_p_inv;
166 m_yyavp = ( cos2 * yyav_p - cs2 * xyav_p + sin2 * xxav_p ) * rrav_p_inv;
168 double xav2 = m_xav * m_xav;
169 double yav2 = m_yav * m_yav;
171 ( xrrav - 2 * xxav * m_xav + xav2 * m_xav - 2 * xyav * m_yav + m_xav * yav2 ) -
174 ( yrrav - 2 * yyav * m_yav + yav2 * m_yav - 2 * xyav * m_xav + m_yav * xav2 ) -
176 m_xrravp = ( m_cosrot * xrrav_p + m_sinrot * yrrav_p ) * rrav_p_inv / m_rscale;
177 m_yrravp = ( -m_sinrot * xrrav_p + m_cosrot * yrrav_p ) * rrav_p_inv / m_rscale;
179 double rrav = xxav + yyav;
180 double rrrrav_p = rrrrav - 2 * m_yav * yrrav - 2 * m_xav * xrrav + rrav * ( xav2 + yav2 ) -
181 2 * m_xav * xrrav_p - xav2 * rrav_p - 2 * m_yav * yrrav_p - yav2 * rrav_p;
182 m_rrrravp = rrrrav_p * rrav_p_inv * rrav_p_inv;
187 if ( m_wsum <= 0 )
return;
188 m_wsum_temp = m_wsum + wi;
189 double wsum_inv( 1 / m_wsum_temp );
190 double rri( xi * xi + yi * yi );
191 m_xav = ( m_xsum + wi * xi ) * wsum_inv;
192 m_yav = ( m_ysum + wi * yi ) * wsum_inv;
197 m_xxavp = ( m_xxsum + wi * xi * xi ) * wsum_inv;
198 m_xyavp = ( m_xysum + wi * xi * yi ) * wsum_inv;
199 m_yyavp = ( m_yysum + wi * yi * yi ) * wsum_inv;
200 double wrri( wi * rri );
201 m_xrravp = ( m_xrrsum + xi * wrri ) * wsum_inv;
202 m_yrravp = ( m_yrrsum + yi * wrri ) * wsum_inv;
203 m_rrrravp = ( m_rrrrsum + rri * wrri ) * wsum_inv;
207 if ( m_wsum <= 0 )
return;
208 m_wsum_temp = m_wsum;
209 double wsum_inv( 1 / m_wsum_temp );
210 m_xav = m_xsum * wsum_inv;
211 m_yav = m_ysum * wsum_inv;
216 m_xxavp = m_xxsum * wsum_inv;
217 m_xyavp = m_xysum * wsum_inv;
218 m_yyavp = m_yysum * wsum_inv;
219 m_xrravp = m_xrrsum * wsum_inv;
220 m_yrravp = m_yrrsum * wsum_inv;
221 m_rrrravp = m_rrrrsum * wsum_inv;
244 o <<
" nc=" << a.m_nc <<
" chisq=" << a.m_chisq <<
" " << (
Lpar&)a;
248double Lpav::solve_lambda(
void ) {
249 if ( m_rscale <= 0 )
return -1;
250 double xrrxrr = m_xrravp * m_xrravp;
251 double yrryrr = m_yrravp * m_yrravp;
252 double rrrrm1 = m_rrrravp - 1;
253 double xxyy = m_xxavp * m_yyavp;
255 double c0 = rrrrm1 * xxyy - xrrxrr * m_yyavp - yrryrr * m_xxavp;
256 double c1 = -rrrrm1 + xrrxrr + yrryrr - 4 * xxyy;
257 double c2 = 4 + rrrrm1 - 4 * xxyy;
267 double chiscl = m_wsum_temp * m_rscale * m_rscale;
268 double dlamax = 0.001 / chiscl;
271 double dlambda = dlamax;
272 while ( itry < ntry && std::fabs( dlambda ) >= dlamax )
274 double cpoly = c0 + lambda * ( c1 + lambda * ( c2 + lambda * lambda * c4 ) );
275 double dcpoly = c1 + lambda * ( c2d + lambda * lambda * c4d );
276 dlambda = -cpoly / dcpoly;
280 lambda = lambda < 0 ? 0 : lambda;
284double Lpav::solve_lambda3(
void ) {
285 if ( m_rscale <= 0 )
return -1;
286 double xrrxrr = m_xrravp * m_xrravp;
287 double yrryrr = m_yrravp * m_yrravp;
288 double rrrrm1 = m_rrrravp - 1;
289 double xxyy = m_xxavp * m_yyavp;
291 double a = m_rrrravp;
292 double b = xrrxrr + yrryrr - m_rrrravp * ( m_xxavp + m_yyavp );
293 double c = m_rrrravp * m_xxavp * m_yyavp - m_yyavp * xrrxrr - m_xxavp * yrryrr +
294 2 * m_xyavp * m_xrravp * m_yrravp - m_rrrravp * m_xyavp * m_xyavp;
295 if ( c >= 0 && b <= 0 ) {
return ( -b - std::sqrt( b * b - 4 * a * c ) ) / 2 / a; }
296 else if ( c >= 0 && b > 0 )
298 std::cerr <<
" returning " << -1 << std::endl;
301 else if ( c < 0 ) {
return ( -b + std::sqrt( b * b - 4 * a * c ) ) / 2 / a; }
306 double lambda = solve_lambda();
309 if ( lambda < 0 )
return -1;
310 double h11 = m_xxavp - lambda;
311 double h22 = m_yyavp - lambda;
312 if ( h11 == 0.0 )
return -1;
313 double h14 = m_xrravp;
314 double h24 = m_yrravp;
315 double h34 = 1 + 2 * lambda;
316 double rootsq = ( h14 * h14 / h11 / h11 ) + 4 * h34;
317 if ( std::fabs( h22 ) > std::fabs( h24 ) )
319 if ( h22 == 0.0 )
return -1;
320 double ratio = h24 / h22;
321 rootsq += ratio * ratio;
322 m_kappa = 1 / std::sqrt( rootsq );
323 m_beta = -ratio * m_kappa;
327 if ( h24 == 0.0 )
return -1;
328 double ratio = h22 / h24;
329 rootsq = 1 + ratio * ratio * rootsq;
330 m_beta = 1 / std::sqrt( rootsq );
331 m_beta = h24 > 0 ? -m_beta : m_beta;
332 m_kappa = -ratio * m_beta;
334 m_alpha = -( h14 / h11 ) * m_kappa;
335 m_gamma = -h34 * m_kappa;
351 rotate( m_cosrot, -m_sinrot );
355 move( -m_xav, -m_yav );
356 if ( m_yrravp < 0 )
neg();
357 if ( lambda >= 0 ) m_chisq = lambda * m_wsum_temp * m_rscale * m_rscale;
362 double lambda = solve_lambda3();
365 if ( lambda < 0 )
return -1;
366 double h11 = m_xxavp - lambda;
367 double h22 = m_yyavp - lambda;
368 double h14 = m_xrravp;
369 double h24 = m_yrravp;
371 double h12 = m_xyavp;
372 double det = h11 * h22 - h12 * h12;
375 double r1 = ( h14 * h22 - h24 * h12 ) / ( det );
376 double r2 = ( h24 * h11 - h14 * h12 ) / ( det );
377 double kinvsq = r1 * r1 + r2 * r2;
378 m_kappa = std::sqrt( 1 / kinvsq );
379 if ( h11 != 0 ) m_alpha = -m_kappa * r1;
381 if ( h22 != 0 ) m_beta = -m_kappa * r2;
387 if ( h11 != 0 && h22 != 0 )
389 m_beta = 1 / std::sqrt( 1 + h12 * h12 / h11 / h11 );
390 m_alpha = std::sqrt( 1 - m_beta * m_beta );
403 if ( ( m_alpha * m_xav + m_beta * m_yav ) * ( m_beta * m_xav - m_alpha * m_yav ) < 0 )
neg();
407 if ( lambda >= 0 ) m_chisq = lambda * m_wsum_temp * m_rscale * m_rscale;
412 if ( m_nc <= 3 )
return -1;
419 if (
q > 0 ) m_chisq =
q * m_wsum_temp * m_rscale * m_rscale;
425 if (
q > 0 ) m_chisq =
q * m_wsum_temp * m_rscale * m_rscale;
431 if ( m_nc <= 3 )
return -1;
438 if (
q > 0 ) m_chisq =
q * m_wsum_temp * m_rscale * m_rscale;
444 if (
q > 0 ) m_chisq =
q * m_wsum_temp * m_rscale * m_rscale;
450#ifdef BELLE_OPTIMIZED_RETURN
455 HepSymMatrix vret( 4 );
457 vret( 1, 1 ) = m_xxsum;
458 vret( 2, 1 ) = m_xysum;
459 vret( 2, 2 ) = m_yysum;
460 vret( 3, 1 ) = m_xsum;
461 vret( 3, 2 ) = m_ysum;
462 vret( 3, 3 ) = m_wsum;
463 vret( 4, 1 ) = m_xrrsum;
464 vret( 4, 2 ) = m_yrrsum;
465 vret( 4, 3 ) = m_xxsum + m_yysum;
466 vret( 4, 4 ) = m_rrrrsum;
474 std::cerr <<
"Lpav::cov:could not invert nc=" << m_nc << vret;
484#ifdef BELLE_OPTIMIZED_RETURN
489 HepSymMatrix vret( 3 );
495 vret =
cov( 1 ).similarity( dldc() );
498 { THROW( Lpav::cov_c1, Singular_c ); }
507 std::cerr <<
"Lpav::cov_c:could not invert " << vret;
509 THROW( Lpav::cov_c2, Singular_c );
518 if ( m_chisq < 0 )
return -1;
519 if ( xy( r, x, y ) != 0 )
return -1;
520 phi = std::atan2( y, x );
535 double l =
cov().similarity(
v );
538 double ls = std::sqrt( l );
551 if ( m_nc <= 3 )
return -1;
556 v( 4 ) = x * x + y * y;
562 l =
cov().similarity(
v );
570void Lpav::add(
double xi,
double yi,
double w,
double a,
double b ) {
571 register double wi = err_dis_inv( xi, yi,
w, a, b );
575void Lpav::add_point(
register double xi,
register double yi,
register double wi ) {
579 m_xxsum += wi * xi * xi;
580 m_yysum += wi * yi * yi;
581 m_xysum += wi * xi * yi;
582 register double rri = ( xi * xi + yi * yi );
583 register double wrri = wi * rri;
584 m_xrrsum += wrri * xi;
585 m_yrrsum += wrri * yi;
586 m_rrrrsum += wrri * rri;
591 register double wi =
w * a;
595 m_xxsum += wi * xi * xi;
596 m_yysum += wi * yi * yi;
597 m_xysum += wi * xi * yi;
598 register double rri = ( xi * xi + yi * yi );
599 register double wrri = wi * rri;
600 m_xrrsum += wrri * xi;
601 m_yrrsum += wrri * yi;
602 m_rrrrsum += wrri * rri;
606void Lpav::sub(
double xi,
double yi,
double w,
double a,
double b ) {
607 register double wi = err_dis_inv( xi, yi,
w, a, b );
611 m_xxsum -= wi * xi * xi;
612 m_yysum -= wi * yi * yi;
613 m_xysum -= wi * xi * yi;
614 register double rri = ( xi * xi + yi * yi );
615 register double wrri = wi * rri;
616 m_xrrsum -= wrri * xi;
617 m_yrrsum -= wrri * yi;
618 m_rrrrsum -= wrri * rri;
623 m_wsum += la1.m_wsum;
624 m_xsum += la1.m_xsum;
625 m_ysum += la1.m_ysum;
626 m_xxsum += la1.m_xxsum;
627 m_yysum += la1.m_yysum;
628 m_xysum += la1.m_xysum;
629 m_xrrsum += la1.m_xrrsum;
630 m_yrrsum += la1.m_yrrsum;
631 m_rrrrsum += la1.m_rrrrsum;
637#ifdef BELLE_OPTIMIZED_RETURN
644 la.m_wsum = la1.m_wsum + la2.m_wsum;
645 la.m_xsum = la1.m_xsum + la2.m_xsum;
646 la.m_ysum = la1.m_ysum + la2.m_ysum;
647 la.m_xxsum = la1.m_xxsum + la2.m_xxsum;
648 la.m_yysum = la1.m_yysum + la2.m_yysum;
649 la.m_xysum = la1.m_xysum + la2.m_xysum;
650 la.m_xrrsum = la1.m_xrrsum + la2.m_xrrsum;
651 la.m_yrrsum = la1.m_yrrsum + la2.m_yrrsum;
652 la.m_rrrrsum = la1.m_rrrrsum + la2.m_rrrrsum;
653 la.m_nc = la1.m_nc + la2.m_nc;
658 if ( m_nc <= 3 )
return 0;
659 if ( m_chisq < 0 )
return 0;
661 int nci = (int)m_nc - 3;
662 double p = (double)
prob_( &c, &nci );
667 if ( m_nc <= 3 )
return -1;
668 else return m_chisq / ( m_nc - 3 );
673 if ( sim < 0 )
return -1;
674 double d = d0( x, y );
675 double delta = std::sqrt(
d ) *
w / ( 1 + sim *
w );
TFile f("ana_bhabha660a_dqa_mcPat_zy_old.root")
float prob_(float *, int *)
****INTEGER imax DOUBLE PRECISION m_pi *DOUBLE PRECISION m_amfin DOUBLE PRECISION m_Chfin DOUBLE PRECISION m_Xenph DOUBLE PRECISION m_sinw2 DOUBLE PRECISION m_GFermi DOUBLE PRECISION m_MfinMin DOUBLE PRECISION m_ta2 INTEGER m_out INTEGER m_KeyFSR INTEGER m_KeyQCD *COMMON c_Semalib $ !copy of input $ !CMS energy $ !beam mass $ !final mass $ !beam charge $ !final charge $ !smallest final mass $ !Z mass $ !Z width $ !EW mixing angle $ !Gmu Fermi $ alphaQED at q
**********Class see also m_nmax DOUBLE PRECISION m_amel DOUBLE PRECISION m_x2 DOUBLE PRECISION m_alfinv DOUBLE PRECISION m_Xenph INTEGER m_KeyWtm INTEGER m_idyfs DOUBLE PRECISION m_zini DOUBLE PRECISION m_q2 DOUBLE PRECISION m_Wt_KF DOUBLE PRECISION m_WtCut INTEGER m_KFfin *COMMON c_KarLud $ !Input CMS energy[GeV] $ !CMS energy after beam spread beam strahlung[GeV] $ !Beam energy spread[GeV] $ !z boost due to beam spread $ !electron beam mass *ff pair spectrum $ !minimum v
std::ostream & operator<<(std::ostream &o, const Lpav &a)
Lpav operator+(const Lpav &la1, const Lpav &la2)
float prob_(float *, int *)
double d(double x, double y) const
double phi(double r, int dir=0) const
double calculate_lpar3(void)
void add_point(double x, double y, double w=1)
HepSymMatrix cov_c(int=0) const
void calculate_average3(void)
void calculate_average(void)
double calculate_lpar(void)
void add_point_frac(double x, double y, double w, double f)
double delta_chisq(double x, double y, double w=1) const
int extrapolate(double, double &, double &) const
const Lpav & operator+=(const Lpav &)
double similarity(double, double) const
HepSymMatrix cov(int=0) const