Tetris Domino Le

Experiments/exp7/SweepParameters.py

@@ -10,11 +10,30 @@
 """
 
 import os
+import sys
 from multiprocessing import Pool
 from joblib import dump
 from tetris_ballistic.tetris_ballistic import Tetris_Ballistic, load_density_from_config
 
 
+class DualLogger:
+    def __init__(self, filepath, mode='a'):
+        self.terminal = sys.stdout
+        self.log = open(filepath, mode)
+
+    def write(self, message):
+        self.terminal.write(message)
+        self.log.write(message)
+
+    def flush(self):  # This flush method is needed for python 3 compatibility.
+        # This flushes the stream to the file, but not the terminal
+        self.terminal.flush()
+        self.log.flush()
+
+    def close(self):
+        self.log.close()
+
+
 def simulate(params, total_iterations):
     # Extract parameters
     w, seed, config_name, density, current_iteration = params
@@ -23,6 +42,9 @@ def simulate(params, total_iterations):
     joblib_filename = f'{basename}_w={w}_seed={seed}.joblib'
     config_filename = f'{basename}_w={w}_seed={seed}.yaml'
     fig_filename = f'{basename}_w={w}_seed={seed}.png'
+    log_file_path = f'{basename}_w={w}_seed={seed}.log'
+
+    sys.stdout = DualLogger(log_file_path, mode='a')
 
     # Check if this simulation has already been completed
     if os.path.exists(joblib_filename):
@@ -66,6 +88,9 @@ def simulate(params, total_iterations):
     progress_message = f"Progress: {progress:.2f}% Completed simulation: {joblib_filename}"
     log_progress(progress_message)
 
+    sys.stdout.close()  # Assuming sys.stdout was set to an instance of DualLogger
+    sys.stdout = sys.__stdout__
+
 
 def log_progress(progress_message):
     with open("simulation_progress.log", "a") as log_file:

Experiments/exp7/cleanup.sh

@@ -2,3 +2,4 @@
 rm *.joblib
 rm *.yaml
 rm *.png
+rm *.log

Experiments/exp8/SweepParameters.py

@@ -10,11 +10,30 @@
 """
 
 import os
+import sys
 from multiprocessing import Pool
 from joblib import dump
 from tetris_ballistic.tetris_ballistic import Tetris_Ballistic, load_density_from_config
 
 
+class DualLogger:
+    def __init__(self, filepath, mode='a'):
+        self.terminal = sys.stdout
+        self.log = open(filepath, mode)
+
+    def write(self, message):
+        self.terminal.write(message)
+        self.log.write(message)
+
+    def flush(self):  # This flush method is needed for python 3 compatibility.
+        # This flushes the stream to the file, but not the terminal
+        self.terminal.flush()
+        self.log.flush()
+
+    def close(self):
+        self.log.close()
+
+
 def simulate(params, total_iterations):
     # Extract parameters
     w, seed, config_name, density, current_iteration = params
@@ -23,6 +42,9 @@ def simulate(params, total_iterations):
     joblib_filename = f'{basename}_w={w}_seed={seed}.joblib'
     config_filename = f'{basename}_w={w}_seed={seed}.yaml'
     fig_filename = f'{basename}_w={w}_seed={seed}.png'
+    log_file_path = f'{basename}_w={w}_seed={seed}.log'
+
+    sys.stdout = DualLogger(log_file_path, mode='a')
 
     # Check if this simulation has already been completed
     if os.path.exists(joblib_filename):
@@ -66,6 +88,9 @@ def simulate(params, total_iterations):
     progress_message = f"Progress: {progress:.2f}% Completed simulation: {joblib_filename}"
     log_progress(progress_message)
 
+    sys.stdout.close()  # Assuming sys.stdout was set to an instance of DualLogger
+    sys.stdout = sys.__stdout__
+
 
 def log_progress(progress_message):
     with open("simulation_progress.log", "a") as log_file:

Experiments/exp8/cleanup.sh

@@ -2,3 +2,4 @@
 rm *.joblib
 rm *.yaml
 rm *.png
+rm *.log