X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Ftrimmed-mean.c;h=b643651a26aa4240e0340768b32af1c3ed37a320;hb=9aa6503e5f9407f221999e19901d5f437bcd2981;hp=b985125295f9754d1ffec7c6b751c02e08425516;hpb=90a008db8b2f6e5bec903e584eb68e9bb20c061e;p=pspp

diff --git a/src/math/trimmed-mean.c b/src/math/trimmed-mean.c
index b985125295..b643651a26 100644
--- a/src/math/trimmed-mean.c
+++ b/src/math/trimmed-mean.c
@@ -33,11 +33,11 @@ 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)
-      tm->sum += c * y;
+  if (cc > os->k[0].tc && cc <= os->k[1].tc)
+    tm->sum += c * y;
 
-  if ( tm->cyk1p1 == SYSMIS && cc >os->k[0].tc)
-      tm->cyk1p1 = c * y;
+  if (tm->cyk1p1 == SYSMIS && cc > os->k[0].tc)
+    tm->cyk1p1 = c * y;
 }
 
 static void
@@ -52,7 +52,7 @@ destroy (struct statistic *s)
 struct trimmed_mean *
 trimmed_mean_create (double W, double tail)
 {
-  struct trimmed_mean *tm = xzalloc (sizeof (*tm));
+  struct trimmed_mean *tm = XZALLOC (struct trimmed_mean);
   struct order_stats *os = &tm->parent;
   struct statistic *stat = &os->parent;
 
@@ -81,18 +81,13 @@ trimmed_mean_calculate (const struct trimmed_mean *tm)
 {
   const struct order_stats *os = (const struct order_stats *) tm;
 
-  assert (os->cc == tm->w);
-
   return
     (
-     (os->k[0].cc_p1 - os->k[0].tc) * os->k[0].y_p1
-     -
-     (os->k[1].cc - os->k[1].tc) * os->k[1].y_p1
+     (os->k[0].cc - os->k[0].tc) * os->k[0].y_p1
+     +
+      (tm->w - os->k[1].cc - os->k[0].tc) * os->k[1].y_p1
      +
-     tm->sum
-     -
-     tm->cyk1p1
-     )
-    /
-    ( (1.0 - 2 * tm->tail) * tm->w);
+      tm->sum
+)
+    / ((1.0 - tm->tail * 2) * tm->w);
 }