reworked simulation or evolution in run_main python script

Author: adampdp

helper_funcs.py

@@ -0,0 +1,4 @@
+dir_name = None
+def rename_file(file_name):
+    if dir_name is None: return file_name
+    return dir_name + '/' + file_name

load_data.py

@@ -6,6 +6,7 @@
 import numpy as np
 from matplotlib import pyplot as plt
 import random
+import helper_funcs as hf
 
 N_muscles = 24;               # Number of muscles alongside the body
 N_units = 10;                 # Number of neural units in VNC
@@ -20,7 +21,7 @@
 
 ###  Worm neuron/muscle activation
 
-act_data = np.loadtxt('act.dat').T
+act_data = np.loadtxt(hf.rename_file('act.dat')).T
 
 offset=1
 
@@ -64,7 +65,7 @@
 
 ###  Worm body curvature
 
-curv_data = np.loadtxt('curv.dat')
+curv_data = np.loadtxt(hf.rename_file('curv.dat'))
 curv_data_less_time = curv_data.T[1:,:]
 
 axs[3, 1].set_title('Body curvature', fontsize=title_font_size)
@@ -73,7 +74,7 @@
 ###  Body position
 
 
-body_data = np.loadtxt('body.dat').T
+body_data = np.loadtxt(hf.rename_file('body.dat')).T
 
 tmax = 1520
 if tmax >= body_data.shape[1]:
@@ -103,7 +104,7 @@
 axs[3, 0].set_aspect('equal')
 
 
-plt.savefig("ExampleActivity.png", bbox_inches="tight", dpi=300)
+plt.savefig(hf.rename_file("ExampleActivity.png"), bbox_inches="tight", dpi=300)
 
 plt.show()
 

main.cpp

@@ -270,16 +270,18 @@ int main (int argc, const char* argv[])
 
     if (argc==2) randomseed += atoi(argv[1]);
 
+    bool do_evol = 1;
 
     if (argc>2){
 
     const bool is_even = ((argc-1) % 2) == 0; //todo: check even
 
     bool seed_flag = 1;
-
+    
     for (int arg = 1; arg<argc; arg+=2)
     { 
-    //if (strcmp(argv[arg],"--doevol")==0) {do_evol = atoi(argv[arg+1]);}
+    if (strcmp(argv[arg],"--doevol")==0) do_evol = atoi(argv[arg+1]);
+    if (strcmp(argv[arg],"--folder")==0) dir_name = argv[arg+1];
     if (seed_flag){ 
     if (strcmp(argv[arg],"-R")==0) randomseed = atoi(argv[arg+1]);
     if (strcmp(argv[arg],"-r")==0) randomseed += atoi(argv[arg+1]);
@@ -291,28 +293,29 @@ int main (int argc, const char* argv[])
 
     }
 
-    bool do_evol;
+   /*  bool do_evol;
     cout << "Do you want to perform an evolutionary search (E) or run a simulation (S) ";
     string ans;
     while(true){
     getline(cin,ans);
     if (ans == "E" || ans == "e") {do_evol = 1;break;} 
     if (ans == "S" || ans == "s") {do_evol = 0;break;}
     cout << "Try again ";
-    };
+    }; */
 
     InitializeBodyConstants();
 
     if (do_evol){
 
+/* 
     std::cout << "Directory name to save genotype data, leave blank for current directory: ";
     getline(cin,dir_name);
     //TODO: check for directory name safety, length
     while (dir_name != "" && mkdir(dir_name.c_str(), 0777) != 0){
         //cerr << "Error :  " << strerror(errno) << endl;
         std::cout << "Folder exists. Try again or leave blank for current directory: ";
         getline(cin,dir_name);
-    }
+    } */
 
 
 
@@ -363,22 +366,22 @@ int main (int argc, const char* argv[])
         evolfile.close();
     #endif
 
-    std::cout << "Finished, now rerunning simulation with the best fit...\n";
+    std::cout << "Finished, now saving the best fit...\n";
 
     }
 
     else
     {
 
-    std::cout << "Directory name for saved genotype data, leave blank for current directory: ";
+    /* std::cout << "Directory name for saved genotype data, leave blank for current directory: ";
     getline(cin,dir_name);
     //TODO: check for directory name safety, length
     struct stat st;
     while (dir_name != "" && stat(dir_name.c_str(), &st) != 0){
         //cerr << "Error :  " << strerror(errno) << endl;
         std::cout << "Folder does not exist, try again or leave blank for current directory: ";
         getline(cin,dir_name);
-    }
+    } */
 
     RandomState rs;
     long seed = static_cast<long>(time(NULL));
@@ -389,7 +392,7 @@ int main (int argc, const char* argv[])
     Best >> best;
     save_traces(best, rs);
 
-    std::cout << "Finished final run\n" << endl;
+    std::cout << "Finished run, saving data\n" << endl;
 
     }
 

run_main.py

@@ -1,37 +1,81 @@
 import subprocess 
 import argparse
+import os
+import sys
+import helper_funcs as hf
+
+def make_directory(directory_name):
+    try:
+        os.mkdir(directory_name)
+        print(f"Directory '{directory_name}' created successfully.")
+        return True
+    except FileExistsError:
+        print(f"Directory '{directory_name}' already exists.")
+        return False
+    except PermissionError:
+        print(f"Permission denied: Unable to create '{directory_name}'.")
+        sys.exit(1)
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        sys.exit(1)
+
 
 def run_cpp_program(): 
-   
+    
     parser=argparse.ArgumentParser(description="argument parser")
 
     parser.add_argument("-R", "--Rand_seed", type=int, nargs='?', default=42)
     parser.add_argument("-r", "--rand_seed", type=int, nargs='?', default=42)
     parser.add_argument('-p', "--pop_size", type=int, nargs='?', default=96)
     parser.add_argument('-d', "--duration", type=int, nargs='?', default=24)
-
+    
+    while True:
+        do_evol_str = input("Do you want to perform an evolutionary search (E) or run a simulation (S)? ")
+        if do_evol_str == "E":
+           do_evol = 1
+           while True:
+               folder_name = input("Please enter the name of a folder to store data: ")
+               if make_directory(folder_name): break
+           break
+        if do_evol_str == "S":
+           do_evol = 0
+           while True:
+               folder_name = str(input("Please enter the name of a folder to read data: "))
+               if os.path.isdir(folder_name): break
+               print("Folder does not exist.")
+           break
+       
+      
     args=parser.parse_args()
     if args.Rand_seed is not None:
-        cmd = ['./main', '-R', str(args.Rand_seed), '-p', str(args.pop_size), '-d', str(args.duration)]
+        cmd = ['./main', '-R', str(args.Rand_seed)]
     else:    
-        cmd = ['./main', '-r', str(args.rand_seed), '-p', str(args.pop_size), '-d', str(args.duration)]
-    
+        cmd = ['./main', '-r', str(args.rand_seed)]
+        
+    cmd += ['-p', str(args.pop_size), 
+            '-d', str(args.duration),
+            '--doevol', str(do_evol),
+            '--folder', folder_name]
+
     # Run the C++  
-    #result = subprocess.run(cmd, capture_output=True, text=True)
+    result = subprocess.run(cmd, capture_output=True, text=True)
     #result = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
     #output, errors = result.communicate()
 
-    p = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-    output = p.communicate(input='some data'.encode())[0]
-
-    print(output)
+    #p = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    
+    #print(output)
 
-    """ if result.stdout: 
+    if result.stdout: 
        print(result.stdout)
 
     if result.stderr: 
         print("Error:") 
-        print(result.stderr)  """
+        print(result.stderr)
+
+    if not do_evol:
+       hf.dir_name = folder_name
+       import load_data    
 
 if __name__ == "__main__": 
     run_cpp_program()