From 32e7413aad5359453bc2cb73f0dd7bcc88353681 Mon Sep 17 00:00:00 2001 From: Pierre-Antoine Rouby Date: Thu, 18 Jun 2026 15:45:11 +0200 Subject: [PATCH] Solver: Mage: read_bin: Put velocity into np table. --- src/Solver/Mage.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py index a876f4a7..5d34b17f 100644 --- a/src/Solver/Mage.py +++ b/src/Solver/Mage.py @@ -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...") - for r in reachs: - for t in ts_list: - for i, p in enumerate(r.profiles): + velocity_arr = np.zeros((len(ts), len(iprofiles))) + for ti, t in enumerate(ts_list): + j = 0 + for r in reachs: + 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")