mmserv

commit f354fc7a4af7e5f71ce51e6cd00b6059594ca97e
parent d838f0bd87ace7c216a57728eab59694b9fb996a
Author: Egor Achkasov <eaachkasov@edu.hse.ru>
Date:   Fri, 15 Nov 2024 00:06:36 +0100

Return load_data after merge

Diffstat:
M
main.c
 | 
33
++++++++++++++++++++++++++++-----

1 file changed, 28 insertions(+), 5 deletions(-)
diff --git a/main.c b/main.c
@@ -1,15 +1,38 @@
 #include "include/mmserv.h"
 
-#include "printf.h"
+#include <stdio.h>
 
-extern data_t x_raw[NUM_TX_ANT][NUM_SC][2];              /* Transmitted signal */
-extern data_t H_raw[NUM_RX_ANT][NUM_TX_ANT][NUM_SC][2];  /* Channel */
-extern data_t R_raw[NUM_TX_ANT][NUM_TX_ANT][NUM_SC][2];  /* Noise covariance matrix */
-extern data_t y_raw[NUM_RX_ANT][NUM_SC][2];              /* Received signal */
+void load_data(
+  IN char* filepath,
+  IN size_t size,
+  OUT data_t* buff)
+{
+  FILE* f = fopen(filepath, "r");
+  if (!f) {
+    fprintf(stderr, "Error: could not open file %s\n", filepath);
+    exit(8);
+  }
+  if ((fread(buff, sizeof(data_t), size, f)) != size) {
+    fprintf(stderr, "Error: could not read file %s\n", filepath);
+    exit(8);
+  }
+  fclose(f);
+}
 
 int main() {
   uint32_t i, j, k;
 
+  /* Load the data */
+  data_t x_raw[NUM_TX_ANT][NUM_SC][2];              /* Transmitted signal */
+  data_t H_raw[NUM_RX_ANT][NUM_TX_ANT][NUM_SC][2];  /* Channel */
+  data_t R_raw[NUM_TX_ANT][NUM_TX_ANT][NUM_SC][2];  /* Noise covariance matrix */
+  data_t y_raw[NUM_RX_ANT][NUM_SC][2];              /* Received signal */
+
+  load_data("data/x.bin", NUM_TX_ANT * NUM_SC * 2, x_raw);
+  load_data("data/H.bin", NUM_RX_ANT * NUM_TX_ANT * NUM_SC * 2, H_raw);
+  load_data("data/R.bin", NUM_TX_ANT * NUM_TX_ANT * NUM_SC * 2, R_raw);
+  load_data("data/y.bin", NUM_RX_ANT * NUM_SC * 2, y_raw);
+
   /* Cast the data into complex data structures */
   complex x[NUM_TX_ANT][NUM_SC];
   complex H[NUM_RX_ANT][NUM_TX_ANT][NUM_SC];