cust_stream.H

/* cust_stream.H */
/* $Id: cust_stream.H,v 1.2 2002/09/03 15:30:37 ibadm Exp $ */
#ifndef __cust_stream_H
#define __cust_stream_H

/***********************************************************************
 * Setup new directory for stream I/O.
 */
template<class T>
void
CubeStorage<T>::stream_dir (const char* dir)
{
  if(NULL != szStreamDir)
    {
      delete[] szStreamDir;
      szStreamDir = NULL;
    }
  szStreamDir = new char[strlen(dir) + 1];
  if(NULL == szStreamDir)
    Throw(ERROR__ALLOC);
  strcpy(szStreamDir, dir);

  if(0 != mkdir(szStreamDir, 0777))
    {
      if(EEXIST != errno)
    Throw(ERROR__IO);
    }
}


/***********************************************************************
 * Get name of directory for stream I/O.
 */
template<class T>
const char*
CubeStorage<T>::stream_dir () const
{
  return szStreamDir;
}


/***********************************************************************
 * Put actual contents to stream with given item index
 */
template<class T>
void
CubeStorage<T>::stream_put (int ii)
{
  if(NULL == szStreamDir)
    Throw(ERROR__NULL);

  int   fildes;

  char  *szStreamFile = new char[strlen(szStreamDir) + 20 + 5];
  sprintf(szStreamFile, "%s/%d.css", szStreamDir, ii);

  fildes = open(szStreamFile, O_CREAT|O_RDWR, 0666);
  if(-1 == fildes)
    {
      delete[] szStreamFile;
      Throw(ERROR__OPEN_FILE);
    }

  Try{
    int mem_size = items_number(nDim, pDim) * sizeof(T);

    stream_put_header(fildes);

    if(mem_size != write(fildes, pStorage, mem_size))
      Throw(ERROR__WRITE);
  }Catch{
    close(fildes);
    delete[] szStreamFile;
    Throw(exCode);
  }

  close(fildes);
  delete[] szStreamFile;
}

/***********************************************************************
 * Replace actual contents from stream at given item index
 */
template<class T>
void
CubeStorage<T>::stream_get (int ii)
{
  if(NULL == szStreamDir)
    Throw(ERROR__NULL);

  int   fildes;

  char  *szStreamFile = new char[strlen(szStreamDir) + 20 + 5];
  sprintf(szStreamFile, "%s/%d.css", szStreamDir, ii);

  fildes = open(szStreamFile, O_RDONLY);
  if(-1 == fildes)
    {
      delete[] szStreamFile;
      Throw(ERROR__OPEN_FILE);
    }

  Try{
    int mem_size = items_number(nDim, pDim) * sizeof(T);

    stream_get_header(fildes);

    if(mem_size != read(fildes, pStorage, mem_size))
      Throw(ERROR__READ);
  }Catch{
    close(fildes);
    delete[] szStreamFile;
    Throw(exCode);
  }

  close(fildes);
  delete[] szStreamFile;
}


/***********************************************************************
 * Put header to stream
 */
template<class T>
void
CubeStorage<T>::stream_put_header (int fildes)
{
  cust_stream_header_t  hdr;

  memcpy(hdr.signature, CSS_SIGNATURE, strlen(CSS_SIGNATURE));
  hdr.magic = 0x12345678;
  memset(hdr.item_order, 0, 4);
  switch(eItemOrder)
    {
    case AlaC:
      hdr.item_order[0] = 'c';
      break;
    case AlaFortran:
      hdr.item_order[0] = 'f';
      break;
    }
  hdr.dim_num = nDim;
  hdr.item_size = sizeof(T);

  if(sizeof(hdr) != write(fildes, &hdr, sizeof(hdr)))
    Throw(ERROR__WRITE);
}


/***********************************************************************
 * Get header from stream (simply skip it)
 */
template<class T>
void
CubeStorage<T>::stream_get_header (int fildes)
{
  cust_stream_header_t  hdr;
  Int4  magic = 0x12345678;

  if(sizeof(hdr) != read(fildes, &hdr, sizeof(hdr)))
    Throw(ERROR__READ);

  if(memcmp(hdr.signature, CSS_SIGNATURE, strlen(CSS_SIGNATURE)))
    /* bad file format or unknown version */
    Throw(ERROR__BAD_PARAMS);

  if(hdr.magic != 0x12345678)
    /* another byte order */
    Throw(ERROR__BAD_PARAMS);

  switch(eItemOrder)
    {
    case AlaC:
      if(hdr.item_order[0] != 'c')
    /* another item order */
    Throw(ERROR__BAD_PARAMS);
      break;
    case AlaFortran:
      if(hdr.item_order[0] != 'f')
    /* another item order */
    Throw(ERROR__BAD_PARAMS);
      break;
    }

  if(hdr.dim_num != nDim)
    /* another number of dimensions */
    Throw(ERROR__BAD_PARAMS);

  if(hdr.item_size != sizeof(T))
    /* another size of item */
    Throw(ERROR__BAD_PARAMS);
}


#endif /* cust_stream.H */