add bapctools limits

RagnarGrootKoerkamp · Jan 25, 2025 · 39aea9a · 39aea9a
1 parent bdbdb21
commit 39aea9a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/bin/problem.py b/bin/problem.py
@@ -139,6 +139,9 @@ def _read_settings(self):
             'validation_time': 60,  # in seconds
             'validation_memory': 2048,  # in MiB
             # 'validation_output': 8,  # in MiB
+            # BAPCtools extension:
+            'generator_time': config.DEFAULT_TIMEOUT,  # in seconds
+            'visualizer_time': config.DEFAULT_TIMEOUT,  # in seconds
         }
 
         yaml_path = self.path / 'problem.yaml'
@@ -218,6 +221,8 @@ def _read_settings(self):
             config.args.timeout or self.limits.time_safety_margin * self.settings.timelimit + 1
         )
         self.limits.validation_time = config.args.timeout or self.limits.validation_time
+        self.limits.generator_time = config.args.timeout or self.limits.generator_time
+        self.limits.visualizer_time = config.args.timeout or self.limits.visualizer_time
         self.limits.memory = config.args.memory or self.limits.memory
         self.limits.validation_memory = config.args.memory or self.limits.validation_memory
 

diff --git a/bin/program.py b/bin/program.py
@@ -509,7 +509,7 @@ def run(self, bar, cwd, name, args=[]):
             else:
                 f.unlink()
 
-        timeout = config.args.timeout or config.DEFAULT_TIMEOUT
+        timeout = self.problem.limits.generator_time
 
         with stdout_path.open('w') as stdout_file:
             result = exec_command(
@@ -552,7 +552,7 @@ def run(self, cwd, args=[]):
         assert self.run_command is not None
         return exec_command(
             self.run_command + args,
-            timeout=config.args.timeout or config.DEFAULT_TIMEOUT,
+            timeout=self.problem.limits.visualizer_time,
             cwd=cwd,
             memory=None,
         )