Turbo Colorscale, EM : IONQ QPU Integration

em/excitation.py

@@ -502,7 +502,7 @@ def update_initial_state_preview():
         plotter.add_mesh(
             mesh,
             scalars='scalars',
-            cmap="Blues",
+            cmap=EXCITATION_SURFACE_COLORSCALE,
             show_scalar_bar=False,
             show_edges=show_grid_edges,
             edge_color='grey',

em/globals.py

@@ -30,16 +30,10 @@ __all__ = [
 ]
 
 # --- Constants ---
-GRID_SIZES = ["16", "32", "64", "128", "256", "512"]
+GRID_SIZES = [8, 16, 32, 64, 128, 256, 512]
 DEFAULT_AXIS_TICKS = (0.0, 0.25, 0.5, 0.75, 1.0)
 
-EXCITATION_SURFACE_COLORSCALE = [
-    [0.0, "#001219"],
-    [0.25, "#005F73"],
-    [0.5, "#94D2BD"],
-    [0.75, "#EE9B00"],
-    [1.0, "#CA6702"],
-]
+EXCITATION_SURFACE_COLORSCALE = "Turbo"
 
 # --- PyVista and Simulation Data ---
 plotter = pv.Plotter()

em/simulation.py

@@ -9,8 +9,16 @@ import numpy as np
 
 from .state import state, ctrl, _apply_workflow_highlights, is_statevector_estimator_selected, is_ibm_qpu_selected
 from .globals import (
-    plotter, simulation_data, current_mesh, snapshot_times,
-    stop_simulation, qpu_ts_cache, sim_ts_cache, set_stop_simulation, reset_globals
+    plotter,
+    simulation_data,
+    current_mesh,
+    snapshot_times,
+    stop_simulation,
+    qpu_ts_cache,
+    sim_ts_cache,
+    set_stop_simulation,
+    reset_globals,
+    EXCITATION_SURFACE_COLORSCALE,
 )
 
 # Import backend functions
@@ -126,10 +134,9 @@ def setup_surface_plot_data(sim_data, nx):
     g.X_grids['Hy'], g.Y_grids['Hy'] = np.meshgrid(x_m1, y)
     
     # Compute z_scale for visualization
-    # finite_vals = [abs(float(v)) for pair in g.surface_clims.values() for v in pair if np.isfinite(v)]
-    # max_abs = max(finite_vals) if finite_vals else 1e-9
-    # g.z_scale = (nx / 2) / max(max_abs, 1e-9)
-    g.z_scale = 1.0
+    finite_vals = [abs(float(v)) for pair in g.surface_clims.values() for v in pair if np.isfinite(v)]
+    max_abs = max(finite_vals) if finite_vals else 1e-9
+    g.z_scale = (nx / 2) / max(max_abs, 1e-9)
     
     g.simulation_data = sim_data
 
@@ -209,7 +216,7 @@ def redraw_surface_plot():
         mesh,
         scalars='scalars',
         # clim=g.surface_clims[field],
-        cmap="Blues",
+        cmap=EXCITATION_SURFACE_COLORSCALE,
         show_scalar_bar=False,
         show_edges=True,
         edge_color='grey',
@@ -490,7 +497,7 @@ def run_simulation_only():
                 impulse_pos=impulse_pos,
                 shots=10000,
                 pm_optimization_level=2,
-                simulation="True",
+                simulation="False",
                 optimization="True",
                 platform="IBM",
                 progress_callback=_ibm_progress_callback,

em/state.py

@@ -6,6 +6,8 @@ These allow using state.update(), @state.change(), and ctrl.xxx at module
 load time, then applying them when set_server() is called.
 """
 
+from .globals import GRID_SIZES
+
 __all__ = [
     "state", "ctrl", "set_server", "init_state",
     "enable_point_picking_on_plotter",
@@ -234,6 +236,7 @@ def _init_state_defaults():
         "stop_button_disabled": True,
         "export_format": "vtk",
         "nx_slider_index": None,
+    "grid_size_labels": [int(val) for val in GRID_SIZES],
         "show_export_status": False,
         "export_status_message": "",
         "logo_src": None,

em/ui.py

@@ -8,7 +8,7 @@ from trame.widgets import html as trame_html, vuetify3, plotly as plotly_widgets
 from pyvista.trame.ui import plotter_ui
 
 from .state import state, ctrl, get_server
-from .globals import plotter
+from .globals import plotter, GRID_SIZES
 from .geometry import build_geometry_placeholder as _build_geometry_placeholder
 from .excitation import build_excitation_placeholder as _build_excitation_placeholder
 from .simulation import run_simulation_only, reset_to_defaults, stop_simulation_handler
@@ -346,14 +346,17 @@ def _build_meshing_card():
                         v_model=("nx_slider_index", None),
                         label="No. of points per direction:",
                         min=0,
-                        max=5,
+                        max=len(GRID_SIZES) - 1,
                         step=1,
                         show_ticks="always",
                         thumb_label="always",
                         density="compact",
                         color="primary",
                     ):
-                        vuetify3.Template(v_slot_thumb_label="{ modelValue }", children=["{{ modelValue === null ? 'Select' : [8, 16, 32, 64, 128, 256, 512][modelValue] }}"])
+                        vuetify3.Template(
+                            v_slot_thumb_label="{ modelValue }",
+                            children=["{{ modelValue === null ? 'Select' : grid_size_labels[modelValue] }}"],
+                        )
                 # Hover content: enlarged Plotly graph
                 with vuetify3.VSheet(classes="pa-2", elevation=6, rounded=True, style="width: 644px;"):
                     qubit_fig_widget = plotly_widgets.Figure(

qlbm/qlbm_sample_app.py

@@ -12,6 +12,8 @@ import plotly.graph_objects as go
 
 dim=3
 
+QLBM_PLOT_COLORSCALE = "Turbo"
+
 
 def bin_to_gray(bin_s):
   XOR=lambda x,y: (x or y) and not (x and y) 
@@ -869,7 +871,7 @@ def _create_slider_figure(output_list, T_list, X):
       opacity=0.4,
       surface_count=10,
       caps=dict(x_show=False, y_show=False, z_show=False),
-      colorscale='Blues',
+      colorscale=QLBM_PLOT_COLORSCALE,
       colorbar=dict(title="Density"),
       visible=visible,
       name=f"T={T}"
@@ -1043,7 +1045,7 @@ def show_initial_distribution(
       opacity=0.4,
       surface_count=10,
       caps=dict(x_show=False, y_show=False, z_show=False),
-      colorscale='Blues',
+      colorscale=QLBM_PLOT_COLORSCALE,
     ))
     
     fig.update_layout(

qlbm/visualize_counts.py

@@ -234,7 +234,7 @@ def plot_density_isosurface_slider(outputs, T_list=None):
             opacity=0.4,
             surface_count=10,
             caps=dict(x_show=False, y_show=False, z_show=False),
-            colorscale='Blues',
+            colorscale='Turbo',
             colorbar=dict(title="Density"),
             visible=visible,
             name=f"T={T_list[i]}"

qlbm_embedded.py

@@ -100,6 +100,8 @@ _PROBLEM_GEOMETRY_MAP = {
     "Laminar flow & heat transfer for a heated body in water.": "Rectangular domain with a heated box (3D)",
 }
 
+_QLBM_PREVIEW_COLORSCALE = "Turbo"
+
 
 def set_server(server):
     """Set the server for embedded mode."""
@@ -488,7 +490,7 @@ def get_initial_distribution_figure(distribution_type, N, show_edges=False):
         isomin=isomin,
         isomax=isomax,
         surface_count=surface_count,
-        colorscale='Blues',
+        colorscale=_QLBM_PREVIEW_COLORSCALE,
         opacity=0.35,
         caps=dict(x_show=False, y_show=False, z_show=False)
     )]
@@ -717,7 +719,8 @@ def _create_box_figure(lx, ly, lz, title):
         j=[3, 4, 1, 2, 5, 6, 5, 2, 0, 1, 6, 3],
         k=[0, 7, 2, 3, 6, 7, 1, 1, 5, 5, 7, 6],
         opacity=0.2,
-        color="rgba(255,218,185,0.25)",
+        intensity=list(range(len(x))),
+        colorscale=_QLBM_PREVIEW_COLORSCALE,
         flatshading=True,
         name=title,
         showscale=False