src/data/casewriter-translator.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2007 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 <data/casewriter.h>
  20 #include <data/casewriter-provider.h>
  21
  22 #include <stdlib.h>
  23
  24 #include <libpspp/taint.h>
  25
  26 #include "xalloc.h"
  27
  28 struct casewriter_translator
  29   {
  30     struct casewriter *subwriter;
  31
  32     void (*translate) (const struct ccase *input, struct ccase *output,
  33                        void *aux);
  34     bool (*destroy) (void *aux);
  35     void *aux;
  36   };
  37
  38 static struct casewriter_class casewriter_translator_class;
  39
  40 struct casewriter *
  41 casewriter_create_translator (struct casewriter *subwriter,
  42                               size_t translated_value_cnt,
  43                               void (*translate) (const struct ccase *input,
  44                                                  struct ccase *output,
  45                                                  void *aux),
  46                               bool (*destroy) (void *aux),
  47                               void *aux)
  48 {
  49   struct casewriter_translator *ct = xmalloc (sizeof *ct);
  50   struct casewriter *writer;
  51   ct->subwriter = casewriter_rename (subwriter);
  52   ct->translate = translate;
  53   ct->destroy = destroy;
  54   ct->aux = aux;
  55   writer = casewriter_create (translated_value_cnt,
  56                               &casewriter_translator_class, ct);
  57   taint_propagate (casewriter_get_taint (ct->subwriter),
  58                    casewriter_get_taint (writer));
  59   return writer;
  60 }
  61
  62 static void
  63 casewriter_translator_write (struct casewriter *writer UNUSED,
  64                              void *ct_, struct ccase *c)
  65 {
  66   struct casewriter_translator *ct = ct_;
  67   struct ccase tmp;
  68
  69   ct->translate (c, &tmp, ct->aux);
  70   casewriter_write (ct->subwriter, &tmp);
  71 }
  72
  73 static void
  74 casewriter_translator_destroy (struct casewriter *writer UNUSED, void *ct_)
  75 {
  76   struct casewriter_translator *ct = ct_;
  77   casewriter_destroy (ct->subwriter);
  78   ct->destroy (ct->aux);
  79   free (ct);
  80 }
  81
  82 static struct casereader *
  83 casewriter_translator_convert_to_reader (struct casewriter *writer UNUSED,
  84                                          void *ct_)
  85 {
  86   struct casewriter_translator *ct = ct_;
  87   struct casereader *reader = casewriter_make_reader (ct->subwriter);
  88   free (ct);
  89   ct->destroy (ct->aux);
  90   return reader;
  91 }
  92
  93 static struct casewriter_class casewriter_translator_class =
  94   {
  95     casewriter_translator_write,
  96     casewriter_translator_destroy,
  97     casewriter_translator_convert_to_reader,
  98   };