Allow driving to be specified by file

1 files changed, 23 insertions, 0 deletions

diff --git a/src/driving.c b/src/driving.c
index c329848..7920d45 100644
--- a/src/driving.c
+++ b/src/driving.c
@@ -1,5 +1,6 @@
 #include "driving.h"
 #include "navier-stokes.h"
+#include "io.h"
 #include <math.h>
 
 // test driving function
@@ -35,3 +36,25 @@ int g_zero(
 
   return 0;
 }
+
+// From file
+// txt input
+int driving_file_txt (
+  _Complex double* g,
+  int K1,
+  int K2,
+  FILE* file
+){
+  read_vec(g, K1, K2, file);
+  return 0;
+}
+// binary input
+int driving_file_bin (
+  _Complex double* g,
+  int K1,
+  int K2,
+  FILE* file
+){
+  read_vec_bin(g, K1, K2, file);
+  return 0;
+}