qlbm: Initial distribution edges, Progress logging

fluid3d_pyvista.py

@@ -652,7 +652,8 @@ def simulate_qlbm_3D_and_animate(num_reg_qubits: int, T: int, distribution_type:
 
         data_frames = []
         actual_timesteps = []
-        for t in time_steps:
+        total_frames = len(time_steps) if time_steps else 0
+        for idx, t in enumerate(time_steps):
             file_path = intermediate_folder_path / f"{t}_{rank}.npy"
             if file_path.exists():
                 sol_loaded = np.load(file_path)
@@ -660,7 +661,10 @@ def simulate_qlbm_3D_and_animate(num_reg_qubits: int, T: int, distribution_type:
                     data = np.reshape(sol_loaded, (downsampled_N, downsampled_N, downsampled_N))
                     data_frames.append(data)
                     actual_timesteps.append(t)
-                    print(f"Time {t}: Min={np.min(data)}, Max={np.max(data)}, Mean={np.mean(data)}")
+                    if total_frames:
+                        progress = int(((idx + 1) / total_frames) * 100)
+                        print(f"Loading frames: {progress}% complete (t = {t})")
+
                 else:
                     print(f"Warning: File {file_path} size {sol_loaded.size} != expected {downsampled_N*downsampled_N*downsampled_N}. Skipping.")
             else:

qlbm.py

@@ -118,7 +118,7 @@ state.update({
     "dist_type": None, 
     "nx_slider_index": 2,    # Default to 32
     "nx": 32,
-    "show_edges": True,
+    "show_edges": False,
     "custom_dist_params": False,
     "sine_k_x": 1.0,
     "sine_k_y": 1.0,
@@ -879,6 +879,7 @@ def reset_simulation():
     state.run_error = ""
     state.simulation_has_run = False
     state.dist_type = None
+    state.show_edges = False
     state.problems_selection = None
     state.geometry_selection = None
     state.backend_type = None
@@ -950,7 +951,16 @@ def _on_problem_selection_change(problems_selection, **_):
     finally:
         _apply_workflow_highlights(_determine_workflow_step())
 
-@state.change("dist_type", "show_edges", "sine_k_x", "sine_k_y", "sine_k_z", "gauss_cx", "gauss_cy", "gauss_cz", "gauss_sigma")
+@state.change("dist_type")
+def _on_dist_type_change(dist_type, **_):
+    # Whenever the user picks a new excitation, hide edges until they touch the slider again
+    if state.show_edges:
+        state.show_edges = False
+    update_view()
+    _apply_workflow_highlights(_determine_workflow_step())
+
+
+@state.change("show_edges", "sine_k_x", "sine_k_y", "sine_k_z", "gauss_cx", "gauss_cy", "gauss_cz", "gauss_sigma")
 def on_param_change(**kwargs):
     update_view()
     _apply_workflow_highlights(_determine_workflow_step())