gui: Fix behavior of print preview in psppire-output-window.
[pspp] / 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