{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Control Scalar Array\n\nExtending our simple example to have a dropdown menu to control which\nscalar array is used to color the mesh.\n" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "collapsed": false }, "outputs": [], "source": [ "import pyvista as pv\nfrom pyvista import examples\nfrom pyvista.trame.ui import plotter_ui\nfrom trame.app import get_server\nfrom trame.ui.vuetify3 import SinglePageLayout\nfrom trame.widgets import vuetify3\n\npv.OFF_SCREEN = True\n\nserver = get_server()\nstate, ctrl = server.state, server.controller\n\nmesh = examples.download_antarctica_velocity()\n\npl = pv.Plotter()\nactor = pl.add_mesh(mesh)\npl.view_xy()\n\n\n@state.change(\"scalars\")\ndef set_scalars(scalars=mesh.active_scalars_name, **kwargs) -> None:\n actor.mapper.array_name = scalars\n actor.mapper.scalar_range = mesh.get_data_range(scalars)\n ctrl.view_update()\n\n\n@state.change(\"log_scale\")\ndef set_log_scale(log_scale=False, **kwargs) -> None: # noqa: FBT002\n actor.mapper.lookup_table.log_scale = log_scale\n ctrl.view_update()\n\n\nwith SinglePageLayout(server) as layout:\n with layout.toolbar:\n vuetify3.VSpacer()\n vuetify3.VCheckbox(\n label=\"Log Scale\",\n v_model=(\"log_scale\", False),\n hide_details=True,\n density=\"compact\",\n outlined=True,\n )\n vuetify3.VSelect(\n label=\"Scalars\",\n v_model=(\"scalars\", mesh.active_scalars_name),\n items=(\"array_list\", list(mesh.point_data.keys())),\n hide_details=True,\n density=\"compact\",\n outlined=True,\n classes=\"pt-1 ml-2\",\n style=\"max-width: 250px\",\n )\n\n with (\n layout.content,\n vuetify3.VContainer(\n fluid=True,\n classes=\"pa-0 fill-height\",\n ),\n ):\n # Use PyVista UI template for Plotters\n view = plotter_ui(pl)\n ctrl.view_update = view.update\n\n# Show UI\nawait layout.ready\nlayout" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "```{=html}\n<center>\n <a target=\"_blank\" href=\"https://colab.research.google.com/github/pyvista/pyvista-tutorial/blob/gh-pages/notebooks/tutorial/09_trame/c_trame_scalars.ipynb\">\n <img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/ width=\"150px\">\n </a>\n</center>\n```\n" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.12.11" } }, "nbformat": 4, "nbformat_minor": 0 }