Add option to vary dot size by category in 3D scatter

Introduces a new checklist switch in the 3D view filter sidebar to enable or disable dot size variation by categories. Updates callbacks, processing, and visualization logic to pass and respect the new 'size_vary' parameter, allowing users to control whether dot sizes reflect category grouping in scatter plots.

Author: zpeng

layouts/view3d_card_layout.py

@@ -535,6 +535,22 @@ def get_filter_sidebar():
                 target="click-hide-switch",
                 placement="top",
             ),
+            dbc.Checklist(
+                options=[
+                    {
+                        "label": "Dot size varies with categories",
+                        "value": True,
+                    }
+                ],
+                value=[],
+                id="size-vary-switch",
+                switch=True,
+            ),
+            dbc.Tooltip(
+                "When this is enabled, the dot size varies with categories",
+                target="size-vary-switch",
+                placement="top",
+            ),
             dbc.Label("Decay"),
             dcc.Slider(
                 id="decay-slider",

process_frame.py

@@ -74,6 +74,7 @@ def process_single_frame(
     colormap: str,
     visible_list: List[str],
     c_key: str,
+    size_vary: str,
     decay: int,
     session_id: str,
     file: str,
@@ -120,6 +121,7 @@ def process_single_frame(
         config,
         frame_list,
         c_key,
+        bool(size_vary),
         num_keys,
         num_values,
         frame_idx,
@@ -227,6 +229,7 @@ def process_overlay_frame(
     colormap: str,
     visible_list: List[str],
     c_key: str,
+    size_vary: str,
     session_id: str,
     file: str,
     file_list: List[str],
@@ -266,6 +269,7 @@ def process_overlay_frame(
         config,
         frame_list,
         c_key,
+        bool(size_vary),
         num_keys,
         num_values,
         frame_idx,

utils.py

@@ -150,6 +150,7 @@ def prepare_figure_kwargs(
     config: Dict[str, Any],
     frame_list: List[float],
     c_key: str,
+    size_vary: bool,
     num_keys: List[str],
     num_values: List[Tuple[float, float]],
     slider_arg: int = 0,
@@ -280,6 +281,7 @@ def prepare_figure_kwargs(
     )
 
     fig_kwargs["c_type"] = keys_dict[c_key].get("type", KEY_TYPES["NUM"])
+    fig_kwargs["size_vary"] = size_vary
     fig_kwargs["ref_name"] = "Host Vehicle"
 
     return fig_kwargs

view_callbacks/scatter_3d_view.py

@@ -133,6 +133,7 @@ def toggle_3d_config_collapse(n_clicks: int, is_open: bool) -> dict:
             "colormap": State("colormap-3d", "value"),
             "visible_list": State("visible-picker", "value"),
             "c_key": State("c-picker-3d", "value"),
+            "size_vary": State("size-vary-switch", "value"),
             "darkmode": State("darkmode-switch", "value"),
             "session_id": State("session-id", "data"),
             "file": State("current-file", "data"),
@@ -149,6 +150,7 @@ def slider_change_callback(
         colormap: str,
         visible_list: list,
         c_key: str,
+        size_vary: str,
         overlay_enable: list,
         decay: int,
         darkmode: list,
@@ -189,6 +191,7 @@ def slider_change_callback(
                 colormap,
                 visible_list,
                 c_key,
+                bool(size_vary),
                 session_id,
                 file,
                 file_list,
@@ -201,6 +204,7 @@ def slider_change_callback(
                 colormap,
                 visible_list,
                 c_key,
+                bool(size_vary),
                 decay,
                 session_id,
                 file,
@@ -335,6 +339,7 @@ def visible_table_change_callback(
             "visible_list": Input("visible-picker", "value"),
             "unused_vistable_trigger": Input("visible-table-change-trigger", "data"),
             "c_key": Input("c-picker-3d", "value"),
+            "size_vary": Input("size-vary-switch", "value"),
             "unused_left_hide_trigger": Input("left-hide-trigger", "data"),
             "unused_right_hide_trigger": Input("right-hide-trigger", "data"),
             "unused_file_loaded": Input("file-loaded-trigger", "data"),
@@ -370,6 +375,7 @@ def regenerate_figure_callback(
         ispaused: bool,
         slider_arg: int,
         c_key: str,
+        size_vary: str,
         overlay_enable: list,
         unused_left_hide_trigger: int,
         unused_right_hide_trigger: int,
@@ -455,6 +461,7 @@ def regenerate_figure_callback(
                 colormap,
                 visible_list,
                 c_key,
+                bool(size_vary),
                 session_id,
                 file,
                 file_list,
@@ -467,6 +474,7 @@ def regenerate_figure_callback(
                 colormap,
                 visible_list,
                 c_key,
+                bool(size_vary),
                 decay,
                 session_id,
                 file,
@@ -494,6 +502,7 @@ def regenerate_figure_callback(
             "num_values": State({"type": "filter-slider", "index": ALL}, "value"),
             "visible_list": State("visible-picker", "value"),
             "c_key": State("c-picker-3d", "value"),
+            "size_vary": State("size-vary-switch", "value"),
             "session_id": State("session-id", "data"),
             "file": State("current-file", "data"),
             "file_list": State("file-add", "value"),
@@ -513,6 +522,7 @@ def regenerate_figure_background_callback(
         num_values: list,
         visible_list: list,
         c_key: str,
+        size_vary: str,
         session_id: str,
         file: str,
         file_list: list,
@@ -571,6 +581,7 @@ def regenerate_figure_background_callback(
             config,
             frame_list,
             c_key,
+            bool(size_vary),
             num_keys,
             num_values,
         )
@@ -605,7 +616,6 @@ def regenerate_figure_background_callback(
                 visible_list,
             )
 
-            # fig = get_scatter3d_data(filterd_frame, **fig_kwargs)
             result = get_scatter3d_data(filterd_frame, hover=keys_dict, **fig_kwargs)
             fig = result["scatter_data"]
             hover_strings = result["hover_strings"]
@@ -703,6 +713,7 @@ def invoke_filter_trigger(
             "case": State("test-case", "value"),
             "session_id": State("session-id", "data"),
             "c_key": State("c-picker-3d", "value"),
+            "size_vary": State("size-vary-switch", "value"),
             "colormap": State("colormap-3d", "value"),
             "visible_list": State("visible-picker", "value"),
             "file": State("current-file", "data"),
@@ -723,6 +734,7 @@ def export_3d_scatter_animation(
         case: str,
         session_id: str,
         c_key: str,
+        size_vary: str,
         colormap: str,
         visible_list: list,
         file: str,
@@ -775,6 +787,7 @@ def export_3d_scatter_animation(
             config,
             frame_list,
             c_key,
+            bool(size_vary),
             num_keys,
             num_values,
         )

viz/graph_data.py

@@ -164,6 +164,8 @@ def get_scatter3d_data(
         "line_width": 0,
     }
 
+    enable_size_vary = kwargs.get("size_vary", False)
+
     def format_hover(series: pd.Series, config: Dict[str, Any]) -> pd.Series:
         """
         Format series values for hover display according to configuration.
@@ -269,8 +271,13 @@ def create_scatter(
         num_groups = len(grouped)
         for i, (name, group) in enumerate(sorted(grouped)):
             hover_text = process_hover(group)
-            size_offset = num_groups - 1 - i
-            result["scatter_data"].append(create_scatter(group, str(name), size_offset=size_offset))
+            if enable_size_vary:
+                size_offset = num_groups - 1 - i
+            else:
+                size_offset = 0
+            result["scatter_data"].append(
+                create_scatter(group, str(name), size_offset=size_offset)
+            )
             result["hover_strings"].append(hover_text.tolist())
 
     return result