parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Solver: Mage: read_bin: Put velocity into np table.

opt-result
Pierre-Antoine 2026-06-18 15:45:11 +02:00
parent d87db5fd48
commit 32e7413aad
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/Solver/Mage.py
View File

@ -1175,7 +1175,6 @@ class Mage8(Mage):
for k in tmp_table:
table[k] = np.array(tmp_table[k])
results.set("table", table)
results.set("timestamps", ts)
ts_list = sorted(ts)
logger.info(f"compute tab...")
@ -1185,14 +1184,22 @@ class Mage8(Mage):
logger.info(f"compute velocity...")
velocity_arr = np.zeros((len(ts), len(iprofiles)))
for ti, t in enumerate(ts_list):
j = 0
for r in reachs:
for t in ts_list:
for i, p in enumerate(r.profiles):
for pi, p in enumerate(r.profiles):
v = p.geometry.speed(
p.get_ts_key(t, "Q"),
p.get_ts_key(t, "Z")
)
r.set(i, t, "V", v)
r.set(pi, t, "V", v)
velocity_arr[ti, j] = v
j += 1
table["V"] = velocity_arr
results.set("table", table)
logger.info(f"read_bin: ... end with {len(ts)} timestamp read")
results.bufferize("Z")