/* See {array_io.h} */

#define array_io_C_COPYRIGHT "Copyright © 2008 by J. Stolfi, UNICAMP"
/* Created on 2008-07-19 by J.Stolfi, UNICAMP */
/* Last edited on 2024-11-20 07:49:41 by stolfi */

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>

#include <affirm.h>
#include <filefmt.h>
#include <nget.h>
#include <fget.h>
#include <jsmath.h>
#include <jswsize.h>

#include <array.h>
#include <array_io.h>

#define array_FILE_VERSION "2020-10-03"
  /* The previous version was "2009-08-31".  The header had a line
    "index_order = {ixor}".  Now the index order is always{ix_order_L}
    (C-like, last index in the inermost loop). */

void array_write_header
  ( FILE *wr, 
    char *type, 
    char *cmt, 
    ix_descr_t *D,
    uint32_t dix[]
  )
  {
    ix_dim_t na = D->na;

    /* Write header line: */
    filefmt_write_header(wr, type, array_FILE_VERSION);

    /* Write comment lines, if any: */
    int32_t ind = 0; /* Comment indentation. */
    if (cmt != NULL) { filefmt_write_comment(wr, cmt, ind, '#'); }

    /* Write the effective dimension {na} and the respective sizes: */
    int32_t ia;
    fprintf(wr, "axes = %d\n", na);
    fprintf(wr, "size =");
    for (ia = 0; ia < na; ia++)
      { ix_size_t szi = D->sz[ia];
        fprintf(wr, (" %" uint64_u_fmt), szi);
        dix[ia] = (szi == 0 ? 0 : digits(szi-1)); /* Number of digits of each index. */
      }
    fprintf(wr, "\n");
  }

void array_write_footer(FILE *wr, char *type)
  {
    filefmt_write_footer(wr, type);
    fflush(wr);
  }

void array_read_header
  ( FILE *rd, 
    char *type, 
    char **cmtP, 
    ix_descr_t *D
  )
  {
    /* Read header line: */
    filefmt_read_header(rd, type, array_FILE_VERSION);

    /* Read the comment lines, if any: */
    char *cmt = filefmt_read_comment(rd, '#');
    if (cmtP == NULL)
      { free(cmt); }
    else
      { (*cmtP) = cmt; }

    /* Read the effective number of axes {na}: */
    uint64_t d64 = nget_uint64(rd, "axes", 10); fget_eol(rd);
    demand(d64 <= array_MAX_AXES, "array has too many axes");
    ix_dim_t na = (ix_dim_t)d64;
    
    /* Read the size vector: */
    ix_size_t sz[na];
    nget_name_eq(rd, "size");
    int32_t i;
    for (i = 0; i < na; i++)
      { uint64_t sz64 = (i < na ? fget_uint64(rd, 10) : 1);
        demand(sz64 <= ix_MAX_SIZE, "size too large");
        sz[i] = sz64;
      }
    fget_eol(rd);

    /* Compute indexing increments and return to caller: */
    (*D) = ix_descr_from_sizes(na, sz);
  }

void array_read_footer(FILE *rd, char *type)
  {
    filefmt_read_footer(rd, type);
  }