X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Ftrimmed-mean.c;h=dff256c0d3319cd4bd0892adbcf250219a0e7505;hb=affc364e2d29fad9f18acba43ca81ce7ec0589aa;hp=a687dd9b5fcffcd711ed983d0566edbb54847213;hpb=fb38178d67da062cf6bbbce5ad560aeca8e259c3;p=pspp diff --git a/src/math/trimmed-mean.c b/src/math/trimmed-mean.c index a687dd9b5f..dff256c0d3 100644 --- a/src/math/trimmed-mean.c +++ b/src/math/trimmed-mean.c @@ -33,10 +33,10 @@ acc (struct statistic *s, const struct ccase *cx UNUSED, double c, double cc, do struct trimmed_mean *tm = UP_CAST (s, struct trimmed_mean, parent.parent); struct order_stats *os = &tm->parent; - if ( cc > os->k[0].tc && cc <= os->k[1].tc) + if (cc > os->k[0].tc && cc <= os->k[1].tc) tm->sum += c * y; - if ( tm->cyk1p1 == SYSMIS && cc > os->k[0].tc) + if (tm->cyk1p1 == SYSMIS && cc > os->k[0].tc) tm->cyk1p1 = c * y; } @@ -44,38 +44,34 @@ static void destroy (struct statistic *s) { struct trimmed_mean *tm = UP_CAST (s, struct trimmed_mean, parent.parent); - struct order_stats *os = &tm->parent; - free (os->k); free (tm); } struct trimmed_mean * trimmed_mean_create (double W, double tail) { - struct trimmed_mean *tm = xzalloc (sizeof (*tm)); - struct order_stats *os = &tm->parent; - struct statistic *stat = &os->parent; - - os->n_k = 2; - os->k = xcalloc (2, sizeof (*os->k)); - assert (tail >= 0); assert (tail <= 1); - os->k[0].tc = tail * W; - os->k[1].tc = W * (1 - tail); - - stat->accumulate = acc; - stat->destroy = destroy; - - tm->cyk1p1 = SYSMIS; - tm->w = W; - tm->tail = tail; - + struct trimmed_mean *tm = xmalloc (sizeof *tm); + *tm = (struct trimmed_mean) { + .parent = { + .parent = { + .destroy = destroy, + }, + .accumulate = acc, + .k = tm->k, + .n_k = 2, + }, + .k[0] = { .tc = tail * W }, + .k[1] = { .tc = W * (1 - tail) }, + .cyk1p1 = SYSMIS, + .w = W, + .tail = tail, + }; return tm; } - double trimmed_mean_calculate (const struct trimmed_mean *tm) { @@ -88,6 +84,6 @@ trimmed_mean_calculate (const struct trimmed_mean *tm) (tm->w - os->k[1].cc - os->k[0].tc) * os->k[1].y_p1 + tm->sum - ) +) / ((1.0 - tm->tail * 2) * tm->w); }