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

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2007, 2009, 2010, 2011, 2013 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 "spreadsheet-reader.h"
  20
  21 #include <libpspp/assertion.h>
  22 #include "gnumeric-reader.h"
  23 #include "ods-reader.h"
  24
  25 #include <libpspp/str.h>
  26 #include <stdio.h>
  27 #include <string.h>
  28 #include <gl/xalloc.h>
  29 #include <gl/c-xvasprintf.h>
  30 #include <stdlib.h>
  31
  32 void
  33 spreadsheet_ref (struct spreadsheet *s)
  34 {
  35   s->ref_cnt++;
  36 }
  37
  38 void
  39 spreadsheet_unref (struct spreadsheet *s)
  40 {
  41   switch (s->type)
  42     {
  43     case SPREADSHEET_ODS:
  44       ods_unref (s);
  45       break;
  46
  47     case SPREADSHEET_GNUMERIC:
  48       gnumeric_unref (s);
  49       break;
  50     default:
  51       NOT_REACHED ();
  52       break;
  53     }
  54 }
  55
  56
  57 struct casereader *
  58 spreadsheet_make_reader (struct spreadsheet *s,
  59                          const struct spreadsheet_read_options *opts)
  60 {
  61   if ( s->type == SPREADSHEET_ODS)
  62     return ods_make_reader (s, opts);
  63
  64   if ( s->type == SPREADSHEET_GNUMERIC)
  65     return gnumeric_make_reader (s, opts);
  66
  67   return NULL;
  68 }
  69
  70 const char *
  71 spreadsheet_get_sheet_name (struct spreadsheet *s, int n)
  72 {
  73   if ( s->type == SPREADSHEET_ODS)
  74     return ods_get_sheet_name (s, n);
  75
  76   if ( s->type == SPREADSHEET_GNUMERIC)
  77     return gnumeric_get_sheet_name (s, n);
  78
  79   return NULL;
  80 }
  81
  82
  83 char *
  84 spreadsheet_get_sheet_range (struct spreadsheet *s, int n)
  85 {
  86   if ( s->type == SPREADSHEET_ODS)
  87     return ods_get_sheet_range (s, n);
  88
  89   if ( s->type == SPREADSHEET_GNUMERIC)
  90     return gnumeric_get_sheet_range (s, n);
  91
  92   return NULL;
  93 }
  94
  95
  96 #define RADIX 26
  97
  98 static void
  99 reverse (char *s, int len)
 100 {
 101   int i;
 102   for (i = 0; i < len / 2; ++i)
 103     {
 104       char tmp = s[len - i - 1];
 105       s[len - i -1] = s[i];
 106       s[i] = tmp;
 107     }
 108 }
 109
 110
 111 /* Convert a string, which is an integer encoded in base26
 112    IE, A=0, B=1, ... Z=25 to the integer it represents.
 113    ... except that in this scheme, digits with an exponent
 114    greater than 1 are implicitly incremented by 1, so
 115    AA  = 0 + 1*26, AB = 1 + 1*26,
 116    ABC = 2 + 2*26 + 1*26^2 ....
 117 */
 118 int
 119 ps26_to_int (const char *str)
 120 {
 121   int i;
 122   int multiplier = 1;
 123   int result = 0;
 124   int len = strlen (str);
 125
 126   for (i = len - 1 ; i >= 0; --i)
 127     {
 128       int mantissa = (str[i] - 'A');
 129
 130       assert (mantissa >= 0);
 131       assert (mantissa < RADIX);
 132
 133       if (i != len - 1)
 134         mantissa++;
 135
 136       result += mantissa * multiplier;
 137       multiplier *= RADIX;
 138     }
 139
 140   return result;
 141 }
 142
 143 char *
 144 int_to_ps26 (int i)
 145 {
 146   char *ret = NULL;
 147
 148   int lower = 0;
 149   long long int base = RADIX;
 150   int exp = 1;
 151
 152   assert (i >= 0);
 153
 154   while (i > lower + base - 1)
 155     {
 156       lower += base;
 157       base *= RADIX;
 158       assert (base > 0);
 159       exp++;
 160     }
 161
 162   i -= lower;
 163   i += base;
 164
 165   ret = xmalloc (exp + 1);
 166
 167   exp = 0;
 168   do
 169     {
 170       ret[exp++] = (i % RADIX) + 'A';
 171       i /= RADIX;
 172     }
 173   while (i > 1);
 174
 175   ret[exp]='\0';
 176
 177   reverse (ret, exp);
 178   return ret;
 179 }
 180
 181
 182 char *
 183 create_cell_ref (int col0, int row0)
 184 {
 185   char *cs0 ;
 186   char *s ;
 187
 188   if ( col0 < 0) return NULL;
 189   if ( row0 < 0) return NULL;
 190
 191   cs0 =  int_to_ps26 (col0);
 192   s =  c_xasprintf ("%s%d", cs0, row0 + 1);
 193
 194   free (cs0);
 195
 196   return s;
 197 }
 198
 199 char *
 200 create_cell_range (int col0, int row0, int coli, int rowi)
 201 {
 202   char *s0 = create_cell_ref (col0, row0);
 203   char *si = create_cell_ref (coli, rowi);
 204
 205   char *s =  c_xasprintf ("%s:%s", s0, si);
 206
 207   free (s0);
 208   free (si);
 209
 210   return s;
 211 }
 212
 213
 214 /* Convert a cell reference in the form "A1:B2", to
 215    integers.  A1 means column zero, row zero.
 216    B1 means column 1 row 0. AA1 means column 26, row 0.
 217 */
 218 bool
 219 convert_cell_ref (const char *ref,
 220                   int *col0, int *row0,
 221                   int *coli, int *rowi)
 222 {
 223   char startcol[5];
 224   char stopcol [5];
 225
 226   int startrow;
 227   int stoprow;
 228
 229   int n = sscanf (ref, "%4[a-zA-Z]%d:%4[a-zA-Z]%d",
 230               startcol, &startrow,
 231               stopcol, &stoprow);
 232   if ( n != 4)
 233     return false;
 234
 235   str_uppercase (startcol);
 236   *col0 = ps26_to_int (startcol);
 237   str_uppercase (stopcol);
 238   *coli = ps26_to_int (stopcol);
 239   *row0 = startrow - 1;
 240   *rowi = stoprow - 1 ;
 241
 242   return true;
 243 }
 244