src/test/utils.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <math.h>
   4 #include <string.h>
   5
   6 // Check if array == refArray
   7 void compareArray(const char* ID, const void* array, const void* refArray, int size,
   8     int isinteger)
   9 {
  10     float EPS = 1e-5; //precision
  11     printf("Checking %s\n",ID);
  12     float maxError = 0.0;
  13     for (int i=0; i<size; i++)
  14     {
  15         float error = isinteger
  16             ? fabs(((int*)array)[i] - ((int*)refArray)[i])
  17             : fabs(((float*)array)[i] - ((float*)refArray)[i]);
  18         if (error >= maxError)
  19             maxError = error;
  20     }
  21     if (maxError >= EPS)
  22         printf("    Inaccuracy: max(abs(error)) = %g >= %g\n",maxError,EPS);
  23     else
  24         printf("    OK\n");
  25 }
  26
  27 void compareArray_real(const char* ID, const void* array, const void* refArray, int size)
  28 {
  29     return compareArray(ID, array, refArray, size, 0);
  30 }
  31
  32 void compareArray_int(const char* ID, const void* array, const void* refArray, int size)
  33 {
  34     return compareArray(ID, array, refArray, size, 1);
  35 }
  36
  37 // Read array by columns (as in MATLAB) and return by-rows encoding
  38 void* readArray(const char* fileName, int isinteger)
  39 {
  40     // need to prepend '../data/' (not really nice code...)
  41     char* fullFileName = (char*)calloc(5+strlen(fileName)+1, sizeof(char));
  42     strcat(fullFileName, "data/");
  43     strcat(fullFileName, fileName);
  44
  45     // first pass to know how many elements to allocate
  46     char* command = (char*)calloc(12+strlen(fullFileName)+8+1, sizeof(char));
  47     strcat(command, "grep -o ' ' ");
  48     strcat(command, fullFileName);
  49     strcat(command, " | wc -l");
  50     FILE *countSpaces = popen(command, "r");
  51     char* buffer = (char*)calloc(32, sizeof(char));
  52     fgets(buffer, sizeof(buffer), countSpaces);
  53     int n = atoi(buffer) + 1;
  54     free(buffer);
  55     pclose(countSpaces);
  56
  57     // open file for reading
  58     FILE* file = fopen(fullFileName, "r");
  59     free(fullFileName);
  60
  61     int d = 1;
  62     size_t elementSize = isinteger
  63         ? sizeof(int)
  64         : sizeof(float);
  65
  66     // read all values, and convert them to by-rows matrices format
  67     void* array = malloc(n*elementSize);
  68     char curChar = ' ';
  69     char bufferNum[64];
  70     for (int u=0; u<n; u++)
  71     {
  72         // read number (as a string)
  73         int bufferIndex = 0;
  74         while (!feof(file) && curChar!=' ')
  75         {
  76             curChar = fgetc(file);
  77             bufferNum[bufferIndex++] = curChar;
  78         }
  79         bufferNum[bufferIndex] = 0;
  80         // transform string into float, and store it at appropriate location
  81         if (isinteger)
  82             ((int*)array)[u] = atoi(bufferNum);
  83         else
  84             ((float*)array)[u] = atof(bufferNum);
  85         // position to next non-separator character
  86         curChar = fgetc(file);
  87     }
  88     fclose(file);
  89
  90     return array;
  91 }
  92
  93 int* readArray_int(const char* fileName)
  94 {
  95     return (int*)readArray(fileName, 1);
  96 }
  97
  98 float* readArray_real(const char* fileName)
  99 {
 100     return (float*)readArray(fileName, 0);
 101 }
 102
 103 int read_int(const char* fileName)
 104 {
 105     return readArray_int(fileName)[0];
 106 }
 107
 108 float read_real(const char* fileName)
 109 {
 110     return readArray_real(fileName)[0];
 111 }