src/math/order-stats.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2008, 2009, 2011 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17 #include <config.h>
  18
  19 #include "math/order-stats.h"
  20
  21 #include <string.h>
  22
  23 #include "data/casereader.h"
  24 #include "data/val-type.h"
  25 #include "data/variable.h"
  26 #include "libpspp/assertion.h"
  27
  28 #include "gl/xalloc.h"
  29
  30 #if 0
  31
  32 #include <stdio.h>
  33
  34 static void
  35 order_stats_dump_k1 (const struct order_stats *os)
  36 {
  37   struct k *k = &os->k[0];
  38   printf ("K1: tc %g; c %g cc %g ccp %g\n",
  39           k->tc, k->c, k->cc, k->cc_p1);
  40
  41 }
  42
  43 static void
  44 order_stats_dump_k2 (const struct order_stats *os)
  45 {
  46   struct k *k = &os->k[1];
  47   printf ("K2: tc %g; c %g cc %g ccp %g\n",
  48           k->tc, k->c, k->cc, k->cc_p1);
  49 }
  50
  51
  52 void
  53 order_stats_dump (const struct order_stats *os)
  54 {
  55   order_stats_dump_k1 (os);
  56   order_stats_dump_k2 (os);
  57 }
  58
  59 #endif
  60
  61 static void
  62 update_k_lower (struct k *kk,
  63                 double y_i, double c_i, double cc_i)
  64 {
  65   if ( cc_i <= kk->tc )
  66     {
  67       kk->cc = cc_i;
  68       kk->c = c_i;
  69       kk->y = y_i;
  70     }
  71 }
  72
  73
  74 static void
  75 update_k_upper (struct k *kk,
  76                 double y_i, double c_i, double cc_i)
  77 {
  78   if ( cc_i > kk->tc && kk->c_p1 == 0)
  79     {
  80       kk->cc_p1 = cc_i;
  81       kk->c_p1 = c_i;
  82       kk->y_p1 = y_i;
  83     }
  84 }
  85
  86
  87 static void
  88 update_k_values (const struct ccase *cx, double y_i, double c_i, double cc_i,
  89                  struct order_stats **os, size_t n_os)
  90 {
  91   int j;
  92
  93   for (j = 0 ; j < n_os ; ++j)
  94     {
  95       int k;
  96       struct order_stats *tos = os[j];
  97       struct statistic  *stat = &tos->parent;
  98       for (k = 0 ; k < tos->n_k; ++k)
  99         {
 100           struct k *myk = &tos->k[k];
 101           update_k_lower (myk, y_i, c_i, cc_i);
 102           update_k_upper (myk, y_i, c_i, cc_i);
 103         }
 104
 105       if ( stat->accumulate )
 106         stat->accumulate (stat, cx, c_i, cc_i, y_i);
 107
 108       tos->cc = cc_i;
 109     }
 110 }
 111
 112
 113 void
 114 order_stats_accumulate (struct order_stats **os, size_t nos,
 115                         struct casereader *reader,
 116                         const struct variable *wv,
 117                         const struct variable *var,
 118                         enum mv_class exclude)
 119 {
 120   struct ccase *cx;
 121   struct ccase *prev_cx = NULL;
 122   double prev_value = -DBL_MAX;
 123
 124   double cc_i = 0;
 125   double c_i = 0;
 126
 127   for (; (cx = casereader_read (reader)) != NULL; case_unref (cx))
 128     {
 129       const double weight = wv ? case_data (cx, wv)->f : 1.0;
 130       const double this_value = case_data (cx, var)->f;
 131
 132       /* The casereader MUST be sorted */
 133       assert (this_value >= prev_value);
 134
 135       if ( var_is_value_missing (var, case_data (cx, var), exclude))
 136         continue;
 137
 138       if ( prev_value == -DBL_MAX || prev_value == this_value)
 139         c_i += weight;
 140
 141       if ( prev_value > -DBL_MAX && this_value > prev_value)
 142         {
 143           update_k_values (prev_cx, prev_value, c_i, cc_i, os, nos);
 144           c_i = weight;
 145         }
 146
 147       case_unref (prev_cx);
 148       cc_i += weight;
 149       prev_value = this_value;
 150       prev_cx = case_ref (cx);
 151     }
 152
 153   update_k_values (prev_cx, prev_value, c_i, cc_i, os, nos);
 154   case_unref (prev_cx);
 155
 156   casereader_destroy (reader);
 157 }
 158
 159
 160