You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-07-30 19:23:07 +03:00
MCOL-1983 Use long double for internal calculations of regr_*** functions.
This commit is contained in:
@ -39,11 +39,11 @@ static Add_corr_ToUDAFMap addToMap;
|
|||||||
struct corr_data
|
struct corr_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumx2; // sum of (x squared)
|
long double sumx2; // sum of (x squared)
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumy2; // sum of (y squared)
|
long double sumy2; // sum of (y squared)
|
||||||
double sumxy; // sum of x * y
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -153,29 +153,29 @@ mcsv1_UDAF::ReturnCode corr::evaluate(mcsv1Context* context, static_any::any& va
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 1)
|
if (N > 1)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumx2 = data->sumx2;
|
long double sumx2 = data->sumx2;
|
||||||
double sumy2 = data->sumy2;
|
long double sumy2 = data->sumy2;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
|
|
||||||
double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
long double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
||||||
if (var_popx == 0)
|
if (var_popx == 0)
|
||||||
{
|
{
|
||||||
// When var_popx is 0, NULL is the result.
|
// When var_popx is 0, NULL is the result.
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
double var_popy = (sumy2 - (sumy * sumy / N)) / N;
|
long double var_popy = (sumy2 - (sumy * sumy / N)) / N;
|
||||||
if (var_popy == 0)
|
if (var_popy == 0)
|
||||||
{
|
{
|
||||||
// When var_popy is 0, NULL is the result
|
// When var_popy is 0, NULL is the result
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
double std_popx = sqrt(var_popx);
|
long double std_popx = sqrt(var_popx);
|
||||||
double std_popy = sqrt(var_popy);
|
long double std_popy = sqrt(var_popy);
|
||||||
double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
long double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
||||||
double corr = covar_pop / (std_popy * std_popx);
|
long double corr = covar_pop / (std_popy * std_popx);
|
||||||
valOut = corr;
|
valOut = static_cast<double>(corr);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,9 +39,9 @@ static Add_covar_pop_ToUDAFMap addToMap;
|
|||||||
struct covar_pop_data
|
struct covar_pop_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumxy; // sum of x * y
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -145,12 +145,12 @@ mcsv1_UDAF::ReturnCode covar_pop::evaluate(mcsv1Context* context, static_any::an
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 0)
|
if (N > 0)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
|
|
||||||
double covar_pop = (sumxy - ((sumx * sumy) / N)) / N ;
|
long double covar_pop = (sumxy - ((sumx * sumy) / N)) / N ;
|
||||||
valOut = covar_pop;
|
valOut = static_cast<double>(covar_pop);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,9 +39,9 @@ static Add_covar_samp_ToUDAFMap addToMap;
|
|||||||
struct covar_samp_data
|
struct covar_samp_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumxy; // sum of x * y
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -145,12 +145,12 @@ mcsv1_UDAF::ReturnCode covar_samp::evaluate(mcsv1Context* context, static_any::a
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 1)
|
if (N > 1)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
|
|
||||||
double covar_samp = (sumxy - ((sumx * sumy) / N)) / (N - 1);
|
long double covar_samp = (sumxy - ((sumx * sumy) / N)) / (N - 1);
|
||||||
valOut = covar_samp;
|
valOut = static_cast<double>(covar_samp);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
if (N == 1)
|
if (N == 1)
|
||||||
|
@ -40,7 +40,7 @@ static Add_regr_avgx_ToUDAFMap addToMap;
|
|||||||
// Use the simple data model
|
// Use the simple data model
|
||||||
struct regr_avgx_data
|
struct regr_avgx_data
|
||||||
{
|
{
|
||||||
double sum;
|
long double sum;
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -132,7 +132,7 @@ mcsv1_UDAF::ReturnCode regr_avgx::evaluate(mcsv1Context* context, static_any::an
|
|||||||
|
|
||||||
if (data->cnt > 0)
|
if (data->cnt > 0)
|
||||||
{
|
{
|
||||||
valOut = data->sum / (double)data->cnt;
|
valOut = static_cast<double>(data->sum / (long double)data->cnt);
|
||||||
}
|
}
|
||||||
|
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
|
@ -40,7 +40,7 @@ static Add_regr_avgy_ToUDAFMap addToMap;
|
|||||||
// Use the simple data model
|
// Use the simple data model
|
||||||
struct regr_avgy_data
|
struct regr_avgy_data
|
||||||
{
|
{
|
||||||
double sum;
|
long double sum;
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -130,7 +130,7 @@ mcsv1_UDAF::ReturnCode regr_avgy::evaluate(mcsv1Context* context, static_any::an
|
|||||||
|
|
||||||
if (data->cnt > 0)
|
if (data->cnt > 0)
|
||||||
{
|
{
|
||||||
valOut = data->sum / (double)data->cnt;
|
valOut = static_cast<double>(data->sum / (long double)data->cnt);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,10 +39,10 @@ static Add_regr_intercept_ToUDAFMap addToMap;
|
|||||||
struct regr_intercept_data
|
struct regr_intercept_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumx2; // sum of (x squared)
|
long double sumx2; // sum of (x squared)
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumxy; // sum of (x*y)
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -148,15 +148,15 @@ mcsv1_UDAF::ReturnCode regr_intercept::evaluate(mcsv1Context* context, static_an
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 1)
|
if (N > 1)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumx2 = data->sumx2;
|
long double sumx2 = data->sumx2;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
double numerator = sumy * sumx2 - sumx * sumxy;
|
long double numerator = sumy * sumx2 - sumx * sumxy;
|
||||||
double var_pop = (N * sumx2) - (sumx * sumx);
|
long double var_pop = (N * sumx2) - (sumx * sumx);
|
||||||
if (var_pop != 0)
|
if (var_pop != 0)
|
||||||
{
|
{
|
||||||
valOut = numerator / var_pop;
|
valOut = static_cast<double>(numerator / var_pop);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
|
@ -39,11 +39,11 @@ static Add_regr_r2_ToUDAFMap addToMap;
|
|||||||
struct regr_r2_data
|
struct regr_r2_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumx2; // sum of (x squared)
|
long double sumx2; // sum of (x squared)
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumy2; // sum of (y squared)
|
long double sumy2; // sum of (y squared)
|
||||||
double sumxy; // sum of x * y
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -153,13 +153,13 @@ mcsv1_UDAF::ReturnCode regr_r2::evaluate(mcsv1Context* context, static_any::any&
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 1)
|
if (N > 1)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumx2 = data->sumx2;
|
long double sumx2 = data->sumx2;
|
||||||
double sumy2 = data->sumy2;
|
long double sumy2 = data->sumy2;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
|
|
||||||
double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
long double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
||||||
if (var_popx == 0)
|
if (var_popx == 0)
|
||||||
{
|
{
|
||||||
// When var_popx is 0, NULL is the result.
|
// When var_popx is 0, NULL is the result.
|
||||||
@ -172,11 +172,11 @@ mcsv1_UDAF::ReturnCode regr_r2::evaluate(mcsv1Context* context, static_any::any&
|
|||||||
valOut = 1.0;
|
valOut = 1.0;
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
double std_popx = sqrt(var_popx);
|
long double std_popx = sqrt(var_popx);
|
||||||
double std_popy = sqrt(var_popy);
|
long double std_popy = sqrt(var_popy);
|
||||||
double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
long double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
||||||
double corr = covar_pop / (std_popy * std_popx);
|
long double corr = covar_pop / (std_popy * std_popx);
|
||||||
valOut = corr * corr;
|
valOut = static_cast<double>(corr * corr);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,10 +39,10 @@ static Add_regr_slope_ToUDAFMap addToMap;
|
|||||||
struct regr_slope_data
|
struct regr_slope_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumx2; // sum of (x squared)
|
long double sumx2; // sum of (x squared)
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumxy; // sum of (x*y)
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -148,16 +148,16 @@ mcsv1_UDAF::ReturnCode regr_slope::evaluate(mcsv1Context* context, static_any::a
|
|||||||
if (N > 1)
|
if (N > 1)
|
||||||
{
|
{
|
||||||
// COVAR_POP(y, x) / VAR_POP(x)
|
// COVAR_POP(y, x) / VAR_POP(x)
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumx2 = data->sumx2;
|
long double sumx2 = data->sumx2;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
double covar_pop = N * sumxy - sumx * sumy;
|
long double covar_pop = N * sumxy - sumx * sumy;
|
||||||
double var_pop = N * sumx2 - sumx * sumx;
|
long double var_pop = N * sumx2 - sumx * sumx;
|
||||||
if (var_pop != 0)
|
if (var_pop != 0)
|
||||||
{
|
{
|
||||||
double slope = covar_pop / var_pop;
|
long double slope = covar_pop / var_pop;
|
||||||
valOut = slope;
|
valOut = static_cast<double>(slope);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
|
@ -39,8 +39,8 @@ static Add_regr_sxx_ToUDAFMap addToMap;
|
|||||||
struct regr_sxx_data
|
struct regr_sxx_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumx2; // sum of (x squared)
|
long double sumx2; // sum of (x squared)
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -128,11 +128,11 @@ mcsv1_UDAF::ReturnCode regr_sxx::evaluate(mcsv1Context* context, static_any::any
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 0)
|
if (N > 0)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumx2 = data->sumx2;
|
long double sumx2 = data->sumx2;
|
||||||
|
|
||||||
double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
long double var_popx = (sumx2 - (sumx * sumx / N)) / N;
|
||||||
valOut = data->cnt * var_popx;
|
valOut = static_cast<double>(data->cnt * var_popx);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,9 +39,9 @@ static Add_regr_sxy_ToUDAFMap addToMap;
|
|||||||
struct regr_sxy_data
|
struct regr_sxy_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumx;
|
long double sumx;
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumxy; // sum of x * y
|
long double sumxy; // sum of x * y
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -88,8 +88,8 @@ mcsv1_UDAF::ReturnCode regr_sxy::nextValue(mcsv1Context* context, ColumnDatum* v
|
|||||||
static_any::any& valIn_y = valsIn[0].columnData;
|
static_any::any& valIn_y = valsIn[0].columnData;
|
||||||
static_any::any& valIn_x = valsIn[1].columnData;
|
static_any::any& valIn_x = valsIn[1].columnData;
|
||||||
struct regr_sxy_data* data = (struct regr_sxy_data*)context->getUserData()->data;
|
struct regr_sxy_data* data = (struct regr_sxy_data*)context->getUserData()->data;
|
||||||
double valx = 0.0;
|
long double valx = 0.0;
|
||||||
double valy = 0.0;
|
long double valy = 0.0;
|
||||||
|
|
||||||
valx = convertAnyTo<double>(valIn_x);
|
valx = convertAnyTo<double>(valIn_x);
|
||||||
valy = convertAnyTo<double>(valIn_y);
|
valy = convertAnyTo<double>(valIn_y);
|
||||||
@ -145,13 +145,13 @@ mcsv1_UDAF::ReturnCode regr_sxy::evaluate(mcsv1Context* context, static_any::any
|
|||||||
double N = data->cnt;
|
double N = data->cnt;
|
||||||
if (N > 0)
|
if (N > 0)
|
||||||
{
|
{
|
||||||
double sumx = data->sumx;
|
long double sumx = data->sumx;
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumxy = data->sumxy;
|
long double sumxy = data->sumxy;
|
||||||
|
|
||||||
double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
long double covar_pop = (sumxy - ((sumx * sumy) / N)) / N;
|
||||||
double regr_sxy = data->cnt * covar_pop;
|
long double regr_sxy = data->cnt * covar_pop;
|
||||||
valOut = regr_sxy;
|
valOut = static_cast<double>(regr_sxy);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -39,8 +39,8 @@ static Add_regr_syy_ToUDAFMap addToMap;
|
|||||||
struct regr_syy_data
|
struct regr_syy_data
|
||||||
{
|
{
|
||||||
uint64_t cnt;
|
uint64_t cnt;
|
||||||
double sumy;
|
long double sumy;
|
||||||
double sumy2; // sum of (y squared)
|
long double sumy2; // sum of (y squared)
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -125,14 +125,14 @@ mcsv1_UDAF::ReturnCode regr_syy::subEvaluate(mcsv1Context* context, const UserDa
|
|||||||
mcsv1_UDAF::ReturnCode regr_syy::evaluate(mcsv1Context* context, static_any::any& valOut)
|
mcsv1_UDAF::ReturnCode regr_syy::evaluate(mcsv1Context* context, static_any::any& valOut)
|
||||||
{
|
{
|
||||||
struct regr_syy_data* data = (struct regr_syy_data*)context->getUserData()->data;
|
struct regr_syy_data* data = (struct regr_syy_data*)context->getUserData()->data;
|
||||||
double N = data->cnt;
|
long double N = data->cnt;
|
||||||
if (N > 0)
|
if (N > 0)
|
||||||
{
|
{
|
||||||
double sumy = data->sumy;
|
long double sumy = data->sumy;
|
||||||
double sumy2 = data->sumy2;
|
long double sumy2 = data->sumy2;
|
||||||
|
|
||||||
double var_popy = (sumy2 - (sumy * sumy / N)) / N;
|
long double var_popy = (sumy2 - (sumy * sumy / N)) / N;
|
||||||
valOut = data->cnt * var_popy;
|
valOut = static_cast<double>(data->cnt * var_popy);
|
||||||
}
|
}
|
||||||
return mcsv1_UDAF::SUCCESS;
|
return mcsv1_UDAF::SUCCESS;
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user