SimplePETSc

import os
from pathlib import Path
from subprocess import run

import matplotlib.pyplot as plt
import numpy as np
import ogstools as ot

prj_name = "square_1e1_neumann"
data_dir = os.environ.get("OGS_DATA_DIR", "../../../Data")
prj_file = f"{data_dir}/EllipticPETSc/{prj_name}.prj"

out_dir = Path(os.environ.get("OGS_TESTRUNNER_OUT_DIR", "_out"))
if not out_dir.exists():
    out_dir.mkdir(parents=True)

print(f"mpirun --bind-to none -np 2 ogs {prj_file} > out.txt")
run(f"mpirun --bind-to none -np 2 ogs {prj_file} > out.txt", shell=True, check=True)

mesh_series = ot.MeshSeries(f"{prj_name}.pvd")
points_coords = np.array([[0.3, 0.5, 0.0], [0.24, 0.21, 0.0]])
points_labels = ["pt0", "pt1"]

fig, ax = plt.subplots(nrows=1, ncols=1)

mesh_series.plot_probe(
    points_coords,
    ot.variables.pressure,
    labels=["{label} linear interpolated" for label in points_labels],
    time_unit="a",
    interp_method="linear",
    ax=fig.axes[0],
    colors=["b", "r"],
    linestyles=["-", "-"],
)