44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import diceplayer.dice.dice_input as dice_input
|
|
from diceplayer.config import DiceConfig
|
|
from diceplayer.environment import System
|
|
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import Final
|
|
|
|
|
|
DICE_FLAG_LINE: Final[int] = -2
|
|
DICE_END_FLAG: Final[str] = "End of simulation"
|
|
|
|
|
|
class DiceWrapper:
|
|
def __init__(self, dice_config: DiceConfig, working_directory: Path):
|
|
self.dice_config = dice_config
|
|
self.working_directory = working_directory
|
|
|
|
def run(self, dice_config: dice_input.DiceInputConfig) -> None:
|
|
input_path = dice_input.write_config(dice_config, self.working_directory)
|
|
output_path = input_path.parent / (input_path.name + ".out")
|
|
|
|
with open(output_path, "w") as outfile, open(input_path, "r") as infile:
|
|
exit_status = subprocess.call(
|
|
self.dice_config.progname, stdin=infile, stdout=outfile, cwd=self.working_directory
|
|
)
|
|
|
|
if exit_status != 0:
|
|
raise RuntimeError(f"Dice simulation failed with exit status {exit_status}")
|
|
|
|
with open(output_path, "r") as outfile:
|
|
line = outfile.readlines()[DICE_FLAG_LINE]
|
|
if line.strip() == DICE_END_FLAG:
|
|
return
|
|
|
|
raise RuntimeError(f"Dice simulation failed with exit status {exit_status}")
|
|
|
|
def parse_results(self, system: System) -> dict:
|
|
results = {}
|
|
for output_file in sorted(self.working_directory.glob("phb*.xyz")):
|
|
...
|
|
|
|
return results
|