pintos-os.org Git - pspp/blob - src/data/any-reader.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2006, 2010, 2011, 2012 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/any-reader.h"
  20
  21 #include <assert.h>
  22 #include <errno.h>
  23 #include <stdbool.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26
  27 #include "data/dataset-reader.h"
  28 #include "data/file-handle-def.h"
  29 #include "data/file-name.h"
  30 #include "data/por-file-reader.h"
  31 #include "data/sys-file-reader.h"
  32 #include "libpspp/assertion.h"
  33 #include "libpspp/message.h"
  34 #include "libpspp/str.h"
  35
  36 #include "gl/xalloc.h"
  37
  38 #include "gettext.h"
  39 #define _(msgid) gettext (msgid)
  40
  41 /* Tries to detect whether FILE is a given type of file, by opening the file
  42    and passing it to DETECT, and returns a detect_result. */
  43 static enum detect_result
  44 try_detect (const char *file_name, bool (*detect) (FILE *))
  45 {
  46   FILE *file;
  47   bool is_type;
  48
  49   file = fn_open (file_name, "rb");
  50   if (file == NULL)
  51     {
  52       msg (ME, _("An error occurred while opening `%s': %s."),
  53            file_name, strerror (errno));
  54       return ANY_ERROR;
  55     }
  56
  57   is_type = detect (file);
  58
  59   fn_close (file_name, file);
  60
  61   return is_type ? ANY_YES : ANY_NO;
  62 }
  63
  64 /* Returns true if any_reader_open() would be able to open FILE as a data
  65    file, false otherwise. */
  66 enum detect_result
  67 any_reader_may_open (const char *file)
  68 {
  69   enum detect_result res = try_detect (file, sfm_detect);
  70
  71   if (res == ANY_NO)
  72     res = try_detect (file, pfm_detect);
  73
  74   return res;
  75 }
  76
  77 /* Returns a casereader for HANDLE.  On success, returns the new
  78    casereader and stores the file's dictionary into *DICT.  On
  79    failure, returns a null pointer.
  80
  81    Ordinarily the reader attempts to automatically detect the character
  82    encoding based on the file's contents.  This isn't always possible,
  83    especially for files written by old versions of SPSS or PSPP, so specifying
  84    a nonnull ENCODING overrides the choice of character encoding.  */
  85 struct casereader *
  86 any_reader_open (struct file_handle *handle, const char *encoding,
  87                  struct dictionary **dict)
  88 {
  89   switch (fh_get_referent (handle))
  90     {
  91     case FH_REF_FILE:
  92       {
  93         enum detect_result result;
  94
  95         result = try_detect (fh_get_file_name (handle), sfm_detect);
  96         if (result == ANY_ERROR)
  97           return NULL;
  98         else if (result == ANY_YES)
  99           return sfm_open_reader (handle, encoding, dict, NULL);
 100
 101         result = try_detect (fh_get_file_name (handle), pfm_detect);
 102         if (result == ANY_ERROR)
 103           return NULL;
 104         else if (result == ANY_YES)
 105           return pfm_open_reader (handle, dict, NULL);
 106
 107         msg (SE, _("`%s' is not a system or portable file."),
 108              fh_get_file_name (handle));
 109         return NULL;
 110       }
 111
 112     case FH_REF_INLINE:
 113       msg (SE, _("The inline file is not allowed here."));
 114       return NULL;
 115
 116     case FH_REF_DATASET:
 117       return dataset_reader_open (handle, dict);
 118     }
 119   NOT_REACHED ();
 120 }