src/sample.c

   1 /* PSPP - computes sample statistics.
   2    Copyright (C) 1997-9, 2000 Free Software Foundation, Inc.
   3    Written by Ben Pfaff <blp@gnu.org>.
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public License as
   7    published by the Free Software Foundation; either version 2 of the
   8    License, or (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful, but
  11    WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  18    02111-1307, USA. */
  19
  20 #include <config.h>
  21 #include <limits.h>
  22 #include <stdio.h>
  23 #include <math.h>
  24 #include "alloc.h"
  25 #include "command.h"
  26 #include "error.h"
  27 #include "lexer.h"
  28 #include "random.h"
  29 #include "str.h"
  30 #include "var.h"
  31
  32 #include "debug-print.h"
  33
  34 /* The two different types of samples. */
  35 enum
  36   {
  37     TYPE_A_FROM_B,              /* 5 FROM 10 */
  38     TYPE_FRACTION               /* 0.5 */
  39   };
  40
  41 /* SAMPLE transformation. */
  42 struct sample_trns
  43   {
  44     struct trns_header h;
  45     int type;                   /* One of TYPE_*. */
  46     int n, N;                   /* TYPE_A_FROM_B: n from N. */
  47     int m, t;                   /* TYPE_A_FROM_B: # picked so far; # so far. */
  48     unsigned frac;              /* TYPE_FRACTION: a fraction of UINT_MAX. */
  49   };
  50
  51 static trns_proc_func sample_trns_proc;
  52
  53 int
  54 cmd_sample (void)
  55 {
  56   struct sample_trns *trns;
  57
  58   int type;
  59   int a, b;
  60   unsigned frac;
  61
  62   if (!lex_force_num ())
  63     return CMD_FAILURE;
  64   if (!lex_integer_p ())
  65     {
  66       type = TYPE_FRACTION;
  67       if (tokval <= 0 || tokval >= 1)
  68         {
  69           msg (SE, _("The sampling factor must be between 0 and 1 "
  70                      "exclusive."));
  71           return CMD_FAILURE;
  72         }
  73
  74       frac = tokval * UINT_MAX;
  75       a = b = 0;
  76     }
  77   else
  78     {
  79       type = TYPE_A_FROM_B;
  80       a = lex_integer ();
  81       lex_get ();
  82       if (!lex_force_match_id ("FROM"))
  83         return CMD_FAILURE;
  84       if (!lex_force_int ())
  85         return CMD_FAILURE;
  86       b = lex_integer ();
  87       if (a >= b)
  88         {
  89           msg (SE, _("Cannot sample %d observations from a population of "
  90                      "%d."),
  91                a, b);
  92           return CMD_FAILURE;
  93         }
  94
  95       frac = 0;
  96     }
  97   lex_get ();
  98
  99   trns = xmalloc (sizeof *trns);
 100   trns->h.proc = sample_trns_proc;
 101   trns->h.free = NULL;
 102   trns->type = type;
 103   trns->n = a;
 104   trns->N = b;
 105   trns->m = trns->t = 0;
 106   trns->frac = frac;
 107   add_transformation ((struct trns_header *) trns);
 108
 109   return lex_end_of_command ();
 110 }
 111
 112 /* Executes a SAMPLE transformation. */
 113 static int
 114 sample_trns_proc (struct trns_header * trns, struct ccase *c UNUSED,
 115                   int case_num UNUSED)
 116 {
 117   struct sample_trns *t = (struct sample_trns *) trns;
 118   double U;
 119
 120   if (t->type == TYPE_FRACTION)
 121     {
 122       if (rng_get_unsigned (pspp_rng ()) <= t->frac)
 123         return -1;
 124       else
 125         return -2;
 126     }
 127
 128   if (t->m >= t->n)
 129     return -2;
 130
 131   U = rng_get_double (pspp_rng ());
 132   if ((t->N - t->t) * U >= t->n - t->m)
 133     {
 134       t->t++;
 135       return -2;
 136     }
 137   else
 138     {
 139       t->m++;
 140       t->t++;
 141       return -1;
 142     }
 143 }