From cea3ab272549da196eea801150ce743adefdc8a1 Mon Sep 17 00:00:00 2001
From: Padraig Gleeson <p.gleeson@gmail.com>
Date: Mon, 7 Oct 2024 18:27:27 +0100
Subject: [PATCH] Initial replay script

---
 replay.py | 129 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 129 insertions(+)
 create mode 100644 replay.py

diff --git a/replay.py b/replay.py
new file mode 100644
index 0000000..90da5df
--- /dev/null
+++ b/replay.py
@@ -0,0 +1,129 @@
+import pyvista as pv
+
+import sys
+
+import time
+
+points = []
+
+types = []
+
+file = "pressure_buffer.txt"
+
+if len(sys.argv) == 2:
+    file = sys.argv[1]
+
+colours = {1.1: "lightblue", 2.1: "green", 2.2: "turquoise", 3: "#eeeeee"}
+colours = {1.1: "blue", 2.2: "turquoise"}
+
+line_count = 0
+pcount = 0
+
+all_points = []
+all_point_types = []
+
+time_count = 0
+
+logStep = None
+
+include_boundary = False
+
+for line in open(file):
+    ws = line.split()
+    # print(ws)
+    if line_count == 6:
+        numOfElasticP = int(ws[0])
+    if line_count == 7:
+        numOfLiquidP = int(ws[0])
+    if line_count == 8:
+        numOfBoundaryP = int(ws[0])
+    if line_count == 9:
+        timeStep = float(ws[0])
+    if line_count == 10:
+        logStep = int(ws[0])
+
+    if len(ws) == 4:
+        type = float(ws[3])
+
+        if not (type == 3 and not include_boundary):
+            points.append([float(ws[0]), float(ws[1]), float(ws[2])])
+            types.append(type)
+
+    if logStep is not None:
+        pcount += 1
+
+        if pcount == numOfBoundaryP + numOfElasticP + numOfLiquidP:
+            print(
+                "End of one batch of %i added, %i total points at line %i, time: %i"
+                % (len(points), pcount, line_count, time_count)
+            )
+            all_points.append(points)
+            all_point_types.append(types)
+
+            points = []
+            types = []
+            pcount = 0
+            numOfBoundaryP = 0
+
+            time_count += 1
+
+    line_count += 1
+
+# all_points_np = np.array(all_points)
+
+print(
+    "Loaded positions with %i elastic, %i liquid and %i boundary points (%i total), %i lines"
+    % (
+        numOfElasticP,
+        numOfLiquidP,
+        numOfBoundaryP,
+        numOfElasticP + numOfLiquidP + numOfBoundaryP,
+        line_count,
+    )
+)
+
+print("Num of time points found: %i" % len(all_points))
+
+pl = pv.Plotter()
+pl.set_background("lightgrey")
+
+last_mesh = None
+
+
+def create_mesh(step):
+    step_count = step
+    value = step_count
+    global last_mesh
+    index = int(value)
+
+    print("Changing to time point: %s (%s) " % (index, value))
+    curr_points = all_points[index]
+    curr_types = all_point_types[index]
+    if last_mesh is None:
+        last_mesh = pv.PolyData(curr_points)
+        last_mesh["types"] = curr_types
+        print(last_mesh)
+
+        pl.add_mesh(
+            last_mesh,
+            render_points_as_spheres=True,
+            cmap=[c for c in colours.values()],
+            point_size=3,
+        )
+    else:
+        last_mesh.points = curr_points
+
+    pl.render()
+
+    time.sleep(0.1)
+
+    return
+
+
+create_mesh(0)
+
+max_time = len(all_points) - 1
+pl.add_slider_widget(create_mesh, rng=[0, max_time], value=max_time, title="Time point")
+pl.add_timer_event(max_steps=len(all_points), duration=200, callback=create_mesh)
+
+pl.show()