From 14ed36a61c9224c531883ed8df604a0d21e00700 Mon Sep 17 00:00:00 2001
From: ReyDoran <manetoscopio@gmail.com>
Date: Mon, 8 Jan 2024 14:37:25 +0100
Subject: [PATCH] Code modified from RAM

---
 manager/manager/manager.py | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/manager/manager/manager.py b/manager/manager/manager.py
index 20def23..5279f6d 100644
--- a/manager/manager/manager.py
+++ b/manager/manager/manager.py
@@ -4,6 +4,7 @@
 import signal
 import subprocess
 import sys
+import re
 import psutil
 import time
 if ("noetic" in str(subprocess.check_output(['bash', '-c', 'echo $ROS_DISTRO']))):
@@ -163,6 +164,29 @@ def on_prepare_visualization(self, event):
             self.gui_server.start()
         LogManager.logger.info("Visualization transition finished")
 
+    def add_frequency_control(self, code):
+        frequency_control_code_imports = """
+from datetime import datetime
+ideal_cycle = 20
+"""
+        code = frequency_control_code_imports + code
+        infinite_loop = re.search(
+            r'[^ ]while\s*\(\s*True\s*\)\s*:|[^ ]while\s*True\s*:|[^ ]while\s*1\s*:|[^ ]while\s*\(\s*1\s*\)\s*:', code)        
+        frequency_control_code_pre = """
+    start_time = datetime.now()
+            """
+        code = code[:infinite_loop.end()] + frequency_control_code_pre + code[infinite_loop.end():]
+        frequency_control_code_post = """
+    finish_time = datetime.now()
+    dt = finish_time - start_time
+    ms = (dt.days * 24 * 60 * 60 + dt.seconds) * 1000 + dt.microseconds / 1000.0
+
+    if (ms < ideal_cycle):
+        time.sleep((ideal_cycle - ms) / 1000.0)
+"""        
+        code = code + frequency_control_code_post
+        return code
+
     def on_run_application(self, event):
         application_configuration = event.kwargs.get('data', {})
         application_file = application_configuration['template']
@@ -171,13 +195,14 @@ def on_run_application(self, event):
     
         errors = self.linter.evaluate_code(code, exercise_id)
         if errors == "":
+            code = self.add_frequency_control(code)
             f = open("/workspace/code/academy.py", "w")
             f.write(code)
             f.close()
 
+            self.unpause_sim()
             self.application_process = subprocess.Popen(["python3", application_file], stdout=sys.stdout, stderr=subprocess.STDOUT,
                                 bufsize=1024, universal_newlines=True)
-            self.unpause_sim()
         else:
             print('errors')
             raise Exception(errors)