X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Fbox-whisker.c;h=506251682c7bf17106fba7525c2f6dc823ecac1c;hb=922dfe227e0a157f895c025b8f8590e2bfc59f23;hp=2e4545906f9d558059a0950d9ad5a551adeaf7f3;hpb=81579d9e9f994fb2908f50af41c3eb033d216e58;p=pspp diff --git a/src/math/box-whisker.c b/src/math/box-whisker.c index 2e4545906f..506251682c 100644 --- a/src/math/box-whisker.c +++ b/src/math/box-whisker.c @@ -22,6 +22,7 @@ #include #include "data/case.h" +#include "data/data-out.h" #include "data/val-type.h" #include "data/variable.h" #include "libpspp/assertion.h" @@ -39,7 +40,7 @@ destroy (struct statistic *s) struct order_stats *os = &bw->parent; struct ll *ll; - for (ll = ll_head (&bw->outliers); ll != ll_null (&bw->outliers); ) + for (ll = ll_head (&bw->outliers); ll != ll_null (&bw->outliers);) { struct outlier *e = ll_data (ll, struct outlier, ll); @@ -60,36 +61,51 @@ acc (struct statistic *s, const struct ccase *cx, { struct box_whisker *bw = UP_CAST (s, struct box_whisker, parent.parent); bool extreme; - struct outlier *o; - if ( y < bw->hinges[2] + bw->step) - bw->whiskers[1] = y; + if (y > bw->hinges[2] + bw->step) /* Upper outlier */ + { + extreme = (y > bw->hinges[2] + 2 * bw->step) ; + } + + else if (y < bw->hinges[0] - bw->step) /* Lower outlier */ + { + extreme = (y < bw->hinges[0] - 2 * bw->step) ; + } - if (bw->whiskers[0] == SYSMIS || bw->hinges[0] - bw->step > y) - bw->whiskers[0] = y; + else /* Not an outlier */ + { + if (bw->whiskers[0] == SYSMIS) + bw->whiskers[0] = y; - if ( y > bw->hinges[2] + bw->step) - extreme = (y > bw->hinges[2] + 2 * bw->step) ; + if (y > bw->whiskers[1]) + bw->whiskers[1] = y; - else if (y < bw->hinges[0] - bw->step) - extreme = (y < bw->hinges[0] - 2 * bw->step) ; + return; + } - else - return; + /* y is an outlier */ - o = xzalloc (sizeof *o) ; + struct outlier *o = XZALLOC (struct outlier); o->value = y; o->extreme = extreme; ds_init_empty (&o->label); if (bw->id_var) - var_append_value_name (bw->id_var, - case_data (cx, bw->id_var), - &o->label); + { + char *s = data_out (case_data_idx (cx, bw->id_idx), + var_get_encoding (bw->id_var), + var_get_print_format (bw->id_var), + settings_get_fmt_settings ()); + + ds_put_cstr (&o->label, s); + free (s); + } else - ds_put_format (&o->label, - "%ld", - (casenumber) case_data_idx (cx, bw->casenumber_idx)->f); + { + ds_put_format (&o->label, + "%ld", + (casenumber) case_num_idx (cx, bw->id_idx)); + } ll_push_head (&bw->outliers, &o->ll); } @@ -115,11 +131,20 @@ box_whisker_outliers (const struct box_whisker *bw) return &bw->outliers; } +/* + Create a box_whisker struct, suitable for generating a boxplot. + + TH are the tukey hinges of the dataset. + + id_idx is the index into the casereader which will be used to label + outliers. + id_var is the variable from which that label came, or NULL +*/ struct box_whisker * box_whisker_create (const struct tukey_hinges *th, - const struct variable *id_var, size_t casenumber_idx) + size_t id_idx, const struct variable *id_var) { - struct box_whisker *w = xzalloc (sizeof (*w)); + struct box_whisker *w = XZALLOC (struct box_whisker); struct order_stats *os = &w->parent; struct statistic *stat = &os->parent; @@ -130,7 +155,7 @@ box_whisker_create (const struct tukey_hinges *th, tukey_hinges_calculate (th, w->hinges); - w->casenumber_idx = casenumber_idx; + w->id_idx = id_idx; w->id_var = id_var; w->step = (w->hinges[2] - w->hinges[0]) * 1.5;