Save and Load of CArrays.

Author: henrique-borba

config.m4

@@ -129,6 +129,7 @@ PHP_NEW_EXTENSION(carray,
       kernel/common/compare.c \
       kernel/exp_logs.c \
       kernel/random.c \
+      kernel/storage.c \
       kernel/range.c \
       kernel/conversion_utils.c \
 	  kernel/buffer.c ,

kernel/storage.c

@@ -0,0 +1,86 @@
+#include "carray.h"
+#include "storage.h"
+#include "buffer.h"
+
+/**
+ * Save CArrray to binary file
+ *
+ * @param filename
+ * @param target
+ * @return 1 - Success or 0 error
+ */
+int
+CArrayStorage_SaveBin(char * filename, CArray *target)
+{
+    char * fullname = emalloc(sizeof(char) * (strlen(filename) + 4));
+
+    strcpy(fullname, filename);
+    strcat(fullname, ".carray");
+
+    FILE* fp = fopen(fullname, "wb");
+
+    if (fp == NULL || fp == 0) {
+        return 0;
+    }
+
+    assert(fp != 0);
+
+    fwrite(&(target->ndim), sizeof(int), 1, fp);
+    fwrite(target->dimensions, sizeof(int) * CArray_NDIM(target), 1, fp);
+    fwrite(target->strides, sizeof(int), CArray_NDIM(target), fp);
+    fwrite(&(target->flags), sizeof(int), 1, fp);
+    fwrite(&(target->descriptor->type_num), sizeof(int), 1, fp);
+    fwrite(target->data, CArray_DESCR(target)->elsize, CArray_MultiplyList(CArray_DIMS(target), CArray_NDIM(target)), fp);
+
+    fclose(fp);
+    efree(fullname);
+
+    return 1;
+}
+
+int
+CArrayStorage_LoadBin(char *filename, MemoryPointer *out)
+{
+    CArrayDescriptor *descr;
+    CArray *array;
+    char * fullname = emalloc(sizeof(char) * (strlen(filename) + 4));
+    int type_num;
+
+    strcpy(fullname, filename);
+    strcat(fullname, ".carray");
+
+    FILE* fp = fopen(fullname, "r");
+
+    if (fp == NULL || fp == 0) {
+        return 0;
+    }
+
+    assert(fp != 0);
+
+    array = emalloc(sizeof(CArray));
+    fread(&(array->ndim), sizeof(int), 1, fp);
+
+    array->dimensions = emalloc(sizeof(int) * array->ndim);
+    fread(array->dimensions, sizeof(int) * array->ndim, 1, fp);
+
+    array->strides = emalloc(sizeof(int) * array->ndim);
+    fread(array->strides, sizeof(int), array->ndim, fp);
+    fread(&(array->flags), sizeof(int), 1, fp);
+    fread(&type_num, sizeof(int), 1, fp);
+
+    descr = CArray_DescrFromType(type_num);
+    array->descriptor = descr;
+
+    array->data = emalloc(array->descriptor->elsize * CArray_MultiplyList(CArray_DIMS(array), CArray_NDIM(array)));
+    fread(array->data, array->descriptor->elsize, CArray_MultiplyList(CArray_DIMS(array), CArray_NDIM(array)), fp);
+
+    array->descriptor->numElements =  CArray_MultiplyList(CArray_DIMS(array), CArray_NDIM(array));
+    array->refcount = 0;
+
+
+    array->flags |= CARRAY_ARRAY_OWNDATA;
+
+    add_to_buffer(out, array, sizeof(CArray));
+    efree(fullname);
+    return 1;
+}

kernel/storage.h

@@ -0,0 +1,8 @@
+#ifndef CARRAY_STORAGE_H
+#define CARRAY_STORAGE_H
+
+#include "carray.h"
+
+int CArrayStorage_SaveBin(char * filename, CArray *target);
+int CArrayStorage_LoadBin(char * filename, MemoryPointer * out);
+#endif //CARRAY_STORAGE_H

phpsci.c

@@ -56,6 +56,7 @@
 #include "kernel/search.h"
 #include "kernel/exp_logs.h"
 #include "kernel/clip.h"
+#include "kernel/storage.h"
 #include "kernel/round.h"
 
 typedef struct _zend_carray_cdata {
@@ -2566,6 +2567,45 @@ PHP_METHOD(CArray, __invoke)
     throw_notimplemented_exception();
 }
 
+/**
+ * STORAGE
+ */
+PHP_METHOD(CArray, save)
+{
+    MemoryPointer ptr;
+    zval * obj = getThis();
+    char * filename;
+    size_t filename_len;
+    ZEND_PARSE_PARAMETERS_START(1, 1)
+        Z_PARAM_STRING(filename, filename_len)
+    ZEND_PARSE_PARAMETERS_END();
+
+    ZVAL_TO_MEMORYPOINTER(obj, &ptr, NULL);
+
+    CArray *target = CArray_FromMemoryPointer(&ptr);
+    if(!CArrayStorage_SaveBin(filename, target)){
+        throw_memory_exception("An error occurred");
+        return;
+    }
+}
+PHP_METHOD(CArray, load)
+{
+    MemoryPointer rtn;
+    char * filename;
+    size_t filename_len;
+    ZEND_PARSE_PARAMETERS_START(1, 1)
+            Z_PARAM_STRING(filename, filename_len)
+    ZEND_PARSE_PARAMETERS_END();
+
+    if(!CArrayStorage_LoadBin(filename, &rtn)){
+        throw_memory_exception("An error occurred");
+        return;
+    }
+
+    RETURN_MEMORYPOINTER(return_value, &rtn);
+}
+
+
 /**
  * CLASS METHODS
  */
@@ -2704,6 +2744,10 @@ static zend_function_entry carray_class_methods[] =
         PHP_ME(CArray, cosh, NULL, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
         PHP_ME(CArray, tanh, NULL, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
 
+        // STORAGAE
+        PHP_ME(CArray, save, NULL, ZEND_ACC_PUBLIC)
+        PHP_ME(CArray, load, NULL, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
+
         // CARRAY ITERATOR
         PHP_ME(CArray, offsetUnset, arginfo_array_offsetGet, ZEND_ACC_PUBLIC)
         PHP_ME(CArray, offsetSet, arginfo_offsetSet, ZEND_ACC_PUBLIC)