libs/basekit/source/UArray_stream.c

   1 /*
   2         copyright: Steve Dekorte, 2006. All rights reserved.
   3         license: See _BSDLicense.txt.
   4 */
   5
   6 #include "UArray.h"
   7 #include <string.h>
   8 #include <stdarg.h>
   9 #include <stdio.h>
  10 #include <stddef.h>
  11
  12 // read ------------------------------------------------------
  13
  14 size_t UArray_fread_(UArray *self, FILE *fp)
  15 {
  16         size_t itemsRead = fread(self->data, self->itemSize, self->size, fp);
  17         UArray_setSize_(self, itemsRead);
  18         return itemsRead;
  19 }
  20
  21 long UArray_readFromCStream_(UArray *self, FILE *fp)
  22 {
  23         long totalItemsRead = 0;
  24         long itemsPerBuffer =  4096 / self->itemSize;
  25         UArray *buffer = UArray_new();
  26         UArray_setItemType_(buffer, self->itemType);
  27         UArray_setSize_(buffer, itemsPerBuffer);
  28
  29         if (!fp) { perror("UArray_readFromCStream_"); return -1; }
  30
  31         while(!feof(fp) && !ferror(fp))
  32         {
  33                 size_t itemsRead;
  34                 UArray_setSize_(buffer, itemsPerBuffer);
  35                 itemsRead = UArray_fread_(buffer, fp);
  36
  37                 totalItemsRead += itemsRead;
  38                 UArray_append_(self, buffer);
  39                 if (itemsRead != itemsPerBuffer) break;
  40         }
  41
  42         if (ferror(fp)) { perror("UArray_readFromCStream_"); return -1; }
  43
  44         UArray_free(buffer);
  45         return totalItemsRead;
  46 }
  47
  48 long UArray_readNumberOfItems_fromCStream_(UArray *self, size_t size, FILE *stream)
  49 {
  50         size_t itemsRead;
  51         UArray *buffer = UArray_new();
  52         UArray_setItemType_(buffer, self->itemType);
  53         UArray_setSize_(buffer, size);
  54
  55         itemsRead = UArray_fread_(buffer, stream);
  56         UArray_append_(self, buffer);
  57
  58         UArray_free(buffer);
  59         return itemsRead;
  60 }
  61
  62 long UArray_readFromFilePath_(UArray *self, const UArray *path)
  63 {
  64         FILE *stream;
  65         long itemsRead;
  66         UArray *sysPath = (UArray_itemSize(path) == 1) ? (UArray *)path : UArray_asUTF8(path);
  67         const char *p = UArray_asCString(sysPath);
  68
  69         //printf("UArray_readFromFilePath_(\"%s\")\n", p);
  70
  71         stream = fopen(p, "r");
  72         if (!stream) return -1;
  73         itemsRead = UArray_readFromCStream_(self, stream);
  74         fclose(stream);
  75
  76         if(sysPath != path) UArray_free(sysPath);
  77         return itemsRead;
  78 }
  79
  80
  81 #define CHUNK_SIZE 4096
  82
  83 int UArray_readLineFromCStream_(UArray *self, FILE *stream)
  84 {
  85         int readSomething = 0;
  86
  87         if(self->itemSize == 1)
  88         {
  89                 char *s = (char *)io_calloc(1, CHUNK_SIZE);
  90
  91                 while (fgets(s, CHUNK_SIZE, stream) != NULL)
  92                 {
  93                         char *eol1 = strchr(s, '\n');
  94                         char *eol2 = strchr(s, '\r');
  95
  96                         readSomething = 1;
  97
  98                         if (eol1) { *eol1 = 0; } // remove the \n return character
  99                         if (eol2) { *eol2 = 0; } // remove the \r return character
 100
 101                         if (*s)
 102                         {
 103                                 UArray_appendCString_(self, s);
 104                         }
 105
 106                         if (eol1 || eol2)
 107                         {
 108                                 break;
 109                         }
 110                 }
 111
 112                 io_free(s);
 113         }
 114
 115         return readSomething;
 116 }
 117
 118 // write ------------------------------------------------------
 119
 120 size_t UArray_fwrite_(const UArray *self, size_t size, FILE *fp)
 121 {
 122         return fwrite(self->data, 1, self->itemSize * size, fp);
 123 }
 124
 125 long UArray_writeToCStream_(const UArray *self, FILE *stream)
 126 {
 127         size_t totalItemsRead = UArray_fwrite_(self, self->size, stream);
 128         if (ferror(stream)) { perror("UArray_readFromCStream_"); return -1; }
 129         return totalItemsRead;
 130 }
 131
 132 long UArray_writeToFilePath_(const UArray *self, const UArray *path)
 133 {
 134         UArray *sysPath = (UArray_itemSize(path) == 1) ? (UArray *)path : UArray_asUTF8(path);
 135         FILE *fp = fopen(UArray_asCString(sysPath), "w");
 136         long itemsWritten = -1;
 137
 138         if (fp)
 139         {
 140                 itemsWritten = UArray_writeToCStream_(self, fp);
 141                 fclose(fp);
 142         }
 143
 144         return itemsWritten;
 145 }
 146