Merge master into output branch.
[pspp-builds.git] / src / output / charts / roc-chart.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 2009 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 <output/charts/roc-chart.h>
20
21 #include <output/chart-provider.h>
22 #include <output/charts/cartesian.h>
23 #include <output/charts/plot-chart.h>
24 #include <data/casereader.h>
25 #include <language/stats/roc.h>
26
27 #include "xalloc.h"
28
29 #include "gettext.h"
30 #define _(msgid) gettext (msgid)
31
32 struct roc_var
33   {
34     char *name;
35     struct casereader *cutpoint_reader;
36   };
37
38 struct roc_chart
39   {
40     struct chart chart;
41     bool reference;
42     struct roc_var *vars;
43     size_t n_vars;
44     size_t allocated_vars;
45   };
46
47 static const struct chart_class roc_chart_class;
48
49 struct roc_chart *
50 roc_chart_create (bool reference)
51 {
52   struct roc_chart *rc = xmalloc (sizeof *rc);
53   chart_init (&rc->chart, &roc_chart_class);
54   rc->reference = reference;
55   rc->vars = NULL;
56   rc->n_vars = 0;
57   rc->allocated_vars = 0;
58   return rc;
59 }
60
61 void
62 roc_chart_add_var (struct roc_chart *rc, const char *var_name,
63                    const struct casereader *cutpoint_reader)
64 {
65   struct roc_var *rv;
66
67   if (rc->n_vars >= rc->allocated_vars)
68     rc->vars = x2nrealloc (rc->vars, &rc->allocated_vars, sizeof *rc->vars);
69
70   rv = &rc->vars[rc->n_vars++];
71   rv->name = xstrdup (var_name);
72   rv->cutpoint_reader = casereader_clone (cutpoint_reader);
73 }
74
75 struct chart *
76 roc_chart_get_chart (struct roc_chart *rc)
77 {
78   return &rc->chart;
79 }
80
81 static void
82 roc_chart_draw (const struct chart *chart, cairo_t *cr,
83                 struct chart_geometry *geom)
84 {
85   const struct roc_chart *rc = UP_CAST (chart, struct roc_chart, chart);
86   size_t i;
87
88   chart_write_title (cr, geom, _("ROC Curve"));
89   chart_write_xlabel (cr, geom, _("1 - Specificity"));
90   chart_write_ylabel (cr, geom, _("Sensitivity"));
91
92   chart_write_xscale (cr, geom, 0, 1, 5);
93   chart_write_yscale (cr, geom, 0, 1, 5);
94
95   if ( rc->reference )
96     {
97       chart_line (cr, geom, 1.0, 0,
98                   0.0, 1.0,
99                   CHART_DIM_X);
100     }
101
102   for (i = 0; i < rc->n_vars; ++i)
103     {
104       const struct roc_var *rv = &rc->vars[i];
105       struct casereader *r = casereader_clone (rv->cutpoint_reader);
106       struct ccase *cc;
107
108       chart_vector_start (cr, geom, rv->name);
109       for (; (cc = casereader_read (r)) != NULL; case_unref (cc))
110         {
111           double se = case_data_idx (cc, ROC_TP)->f;
112           double sp = case_data_idx (cc, ROC_TN)->f;
113
114           se /= case_data_idx (cc, ROC_FN)->f + case_data_idx (cc, ROC_TP)->f ;
115           sp /= case_data_idx (cc, ROC_TN)->f + case_data_idx (cc, ROC_FP)->f ;
116
117           chart_vector (cr, geom, 1 - sp, se);
118         }
119       chart_vector_end (cr, geom);
120       casereader_destroy (r);
121     }
122
123   chart_write_legend (cr, geom);
124 }
125
126 static void
127 roc_chart_destroy (struct chart *chart)
128 {
129   struct roc_chart *rc = UP_CAST (chart, struct roc_chart, chart);
130   size_t i;
131
132   for (i = 0; i < rc->n_vars; i++)
133     {
134       struct roc_var *rv = &rc->vars[i];
135       free (rv->name);
136       casereader_destroy (rv->cutpoint_reader);
137     }
138   free (rc->vars);
139   free (rc);
140 }
141
142 static const struct chart_class roc_chart_class =
143   {
144     roc_chart_draw,
145     roc_chart_destroy
146   };
147
148