Fixed Select operator

CMakeLists.txt

@@ -7,6 +7,11 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
 set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wall -Wextra -Werror")
 
+if (CMAKE_BUILD_TYPE MATCHES "Debug")
+    add_compile_options(-fsanitize=address,undefined)
+    add_link_options(-fsanitize=address,undefined)
+endif()
+
 add_subdirectory(tests)
 
 # Option to build benchmarks

src/file_source.c

@@ -11,6 +11,7 @@
 #include <unistd.h>
 
 #include "generator.h"
+#include "utils.h"
 
 
 data_t *get_next_file(const source_t *source)
@@ -23,8 +24,8 @@ data_t *get_next_file(const source_t *source)
     data_t *data = malloc(sizeof(data_t));
     assert(data);
     data->data = fs->source.buffer.data + (fs->index * fs->source.buffer.width);
-    data->size = fs->inc;
-    data->width = data->width;
+    data->size = min(fs->inc, fs->source.buffer.size - fs->index);
+    data->width = source->buffer.width;
 
     fs->index += fs->inc;
 
@@ -56,7 +57,7 @@ source_t *create_file_source(const char *filename, uint8_t width, uint32_t inc)
         return NULL;
     }
     fs->source.buffer.data = triples;
-    fs->source.buffer.size = sb.st_size / sizeof(triple_t);
+    fs->source.buffer.size = sb.st_size / (sizeof(triple_t) * width);
     fs->source.buffer.width = width;
     fs->source.get_next = get_next_file;
     fs->index = 0;

src/query.c

@@ -84,9 +84,13 @@ void select_query(const data_t *in, data_t *out, const parameter_t param)
     out->width = param.select.size;
     uint32_t out_idx = 0;
 
-    for (uint32_t i = 0; i < in->size * in->width; ++i) {
-        if (select_check(in, i, param.select))
-            out->data[out_idx++] = in->data[i];
+    for (uint32_t i = 0; i < in->size; ++i) {
+        for (uint32_t j = 0; j < in->width; ++j) {
+            if (select_check(in, i * in->width + j, param.select)) {
+                out->data[out_idx++] = in->data[i * in->width + j];
+                break;
+            }
+        }
     }
 }