io_csv.c

Go to the documentation of this file.

/*
  Copyright (C) 2006 octopus team
 
  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2, or (at your option)
  any later version.
 
  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.
 
  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  02110-1301, USA.
 
*/
 
/*
The functions in this file read an array from an ascii matrix (csv) file.
Format with values "valueXYZ" as follows
 
File values.csv:
--------
value111 value112 value113
value121 value122 value123
value131 value132 value133
 
value211 value212 value213
value221 value222 value223
value231 value232 value233
 
value311 value312 value313
value321 value322 value323
value331 value332 value333
--------
 
That is, every XY-plane as a table of values and all XY-planes separated by an
empty row.
 
The given matrix is interpolated/stretched to fit the calculation
box defined in input file.
 
Calculation box shape must be "parallelepiped".
 
The delimiter can be a tab, a comma or a space.
*/
 
#include <assert.h>
#include <config.h>
#include <errno.h>
#include <fcntl.h>
#include <math.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
 
#include "string_f.h"
 
#include "io_binary.h"
 
static const int size_of[6] = {4, 8, 8, 16, 4, 8};
 
void FC_FUNC_(read_csv, READ_CSV)(unsigned long *np, void *f, int *output_type,
                                  int *ierr, STR_F_TYPE fname STR_ARG1) {
  char *filename;
  int i;
  FILE *fd;
  char *buf;
  char *c;
  const char sep[] = "\t\n ,";
  int buf_size = 65536;
 
  TO_C_STR1(fname, filename);
  fd = fopen(filename, "r");
  free(filename);
 
  *ierr = 0;
  if (fd == NULL) {
    *ierr = 2;
    return;
  }
 
  buf = (char *)malloc(buf_size * sizeof(char));
  assert(buf != NULL);
 
  if ((*output_type) == TYPE_FLOAT) {
    i = 0;
    while (fgets(buf, buf_size * sizeof(char), fd) != NULL) {
      float d;
      c = strtok(buf, sep);
      while (c != NULL) {
        assert(i / 8 < *np);
        d = strtof(c, (char **)NULL);
        c = (char *)strtok((char *)NULL, sep);
        memcpy(f + i, &d, size_of[(*output_type)]);
        i += size_of[(*output_type)];
      }
    }
  } else if ((*output_type) == TYPE_DOUBLE) {
    double d;
    i = 0;
    while (fgets(buf, buf_size * sizeof(char), fd) != NULL) {
      c = strtok(buf, sep);
      while (c != NULL) {
        assert(i / 8 < *np);
        d = strtod(c, (char **)NULL);
        memcpy(f + i, &d, size_of[(*output_type)]);
        c = (char *)strtok((char *)NULL, sep);
        i += size_of[(*output_type)];
      }
    }
  }
 
  free(buf);
  fclose(fd);
}
 
void FC_FUNC_(get_info_csv, GET_INFO_CSV)(unsigned long *dims, int *ierr,
                                          STR_F_TYPE fname STR_ARG1) {
  char *filename;
  char *buf;
  char *c;
  FILE *fd;
  int buf_size = 65536;
  const char sep[] = "\n\t ,";
 
  unsigned long curr_dims[3] = {0, 0, 0};
  unsigned long prev_dims[2] = {0, 0};
 
  TO_C_STR1(fname, filename);
  fd = fopen(filename, "r");
  free(filename);
 
  *ierr = 0;
  if (fd == NULL) {
    *ierr = 2;
    return;
  }
 
  buf = (char *)malloc(buf_size * sizeof(char));
  assert(buf != NULL);
  while (fgets(buf, buf_size * sizeof(char), fd) != NULL) {
    c = strtok(buf, sep);
 
    prev_dims[0] = curr_dims[0];
 
    curr_dims[0] = 0;
 
    while (c != NULL) {
      curr_dims[0]++;
      c = (char *)strtok((char *)NULL, sep);
    }
 
    if (prev_dims[0] > 0 && curr_dims[0] > 0)
      assert(curr_dims[0] == prev_dims[0]);
 
    if (prev_dims[0] == 0 && curr_dims[0] != 0) {
      prev_dims[1] = curr_dims[1];
      curr_dims[1] = 0;
      curr_dims[2]++;
 
      if (prev_dims[1] > 0 && curr_dims[1] > 0)
        assert(prev_dims[1] == curr_dims[1]);
    }
 
    if (curr_dims[0] > 0)
      curr_dims[1]++;
  }
 
  dims[0] = curr_dims[0];
  dims[1] = curr_dims[1];
  dims[2] = curr_dims[2];
 
  free(buf);
  fclose(fd);
}