|
#!/usr/bin/env python3
|
|
|
|
import subprocess
|
|
import time
|
|
import re
|
|
import sys
|
|
import platform
|
|
from datetime import datetime
|
|
|
|
|
|
# Intervalle entre deux mesures, en secondes
|
|
INTERVAL = 1
|
|
# Fichier de sortie
|
|
LOG_FILE = "memory.log"
|
|
# Executable Herezh
|
|
exe_hz = "HZppfast_Vn-1"
|
|
|
|
|
|
|
|
COMMAND = [exe_hz, "-f", "test.info"]
|
|
|
|
|
|
def get_memory_kb(pid):
|
|
"""Retourne la memoire RSS du processus en Ko."""
|
|
|
|
system = platform.system()
|
|
|
|
# -------------------------
|
|
# Linux
|
|
# -------------------------
|
|
if system == "Linux":
|
|
try:
|
|
with open(f"/proc/{pid}/status", "r") as f:
|
|
for line in f:
|
|
if line.startswith("VmRSS:"):
|
|
match = re.search(r"(\d+)", line)
|
|
if match:
|
|
return int(match.group(1))
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
# -------------------------
|
|
# macOS / Darwin
|
|
# -------------------------
|
|
elif system == "Darwin":
|
|
try:
|
|
result = subprocess.run(
|
|
["ps", "-o", "rss=", "-p", str(pid)],
|
|
capture_output=True,
|
|
text=True,
|
|
check=True
|
|
)
|
|
|
|
output = result.stdout.strip()
|
|
|
|
if output:
|
|
return int(output)
|
|
|
|
except (subprocess.SubprocessError, ValueError):
|
|
pass
|
|
|
|
return None
|
|
|
|
|
|
def main():
|
|
print(f"Lancement de : {' '.join(COMMAND)}")
|
|
|
|
process = subprocess.Popen(COMMAND)
|
|
|
|
print(f"PID : {process.pid}")
|
|
|
|
start_time = time.perf_counter()
|
|
|
|
with open(LOG_FILE, "w") as log:
|
|
log.write("#temps(s) memoire(kb)\n")
|
|
|
|
try:
|
|
while process.poll() is None:
|
|
elapsed = time.perf_counter() - start_time
|
|
memory_kb = get_memory_kb(process.pid)
|
|
|
|
if memory_kb is not None:
|
|
print(f"{elapsed}s : {memory_kb} Ko")
|
|
log.write(f"{elapsed:.3f} {memory_kb}\n")
|
|
log.flush()
|
|
|
|
time.sleep(INTERVAL)
|
|
|
|
except KeyboardInterrupt:
|
|
print("\nArret demande.")
|
|
|
|
finally:
|
|
# Attend que le processus se termine s il est encore actif
|
|
if process.poll() is None:
|
|
print("Arret du processus...")
|
|
process.terminate()
|
|
|
|
process.wait()
|
|
|
|
|
|
print(f"\n\nFichier cree : {LOG_FILE}\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
