mirror of
https://github.com/Findus23/rebound-collisions.git
synced 2024-09-19 15:53:48 +02:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import os
|
|
import socket
|
|
import subprocess
|
|
from pathlib import Path
|
|
from sys import argv
|
|
|
|
from setproctitle import setproctitle
|
|
|
|
|
|
def filename_from_argv(argument: str = None) -> Path:
|
|
if len(argv) < 2 and not argument:
|
|
raise ValueError("specify filename")
|
|
if argument:
|
|
fn = argument
|
|
else:
|
|
fn = argv[1]
|
|
fn = fn.replace(".bin", "").replace(".meta.json", "")
|
|
if fn.endswith("."):
|
|
fn = fn[:-1]
|
|
return Path(fn.replace(".bin", "").replace(".meta.json", ""))
|
|
|
|
|
|
def git_hash() -> str:
|
|
output = subprocess.run(["git", "rev-parse", "--verify", "HEAD"], capture_output=True)
|
|
return output.stdout.decode()
|
|
|
|
|
|
def check_heartbeat_needs_recompile() -> None:
|
|
library = Path("heartbeat/heartbeat.so")
|
|
code = library.with_suffix(".c")
|
|
if code.stat().st_mtime > library.stat().st_mtime:
|
|
raise RuntimeError("heartbeat.so is out of date. Please recompile it from source.")
|
|
|
|
|
|
def process_friendlyness(fn: Path) -> None:
|
|
if socket.gethostname() == "standpc":
|
|
# only handle other computers specially
|
|
return
|
|
set_process_title(fn, 0, 0)
|
|
os.nice(5)
|
|
|
|
|
|
def set_process_title(fn: Path, progress: float, bodies: int) -> None:
|
|
setproctitle(f"watersim [{fn.stem}] {progress * 100:.2f}% finished, {bodies} bodies")
|
|
|
|
|
|
def is_ci() -> bool:
|
|
return "CI" in os.environ
|