U f{jJ-@sdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z da dadZed d d Zdad d ZddZddZed.ddZddZddZeddZed/ddZeddZe d0dd Zd1d"d#Zd2ddd%d&d'Zd(d)Zd*d+Z d,d-Z!dS)3zCore visualization operations.)contextmanagerN)VALID_3D_BACKENDS)loggerverbose get_config _check_optionfill_doc_validate_typeFz._qtz ._notebook) pyvistaqtnotebookcCs$tjt|ddatd|dS)Nzmne.viz.backends)namepackagezUsing %s 3d backend. ) importlib import_module_backend_name_mapbackendrinfo backend_namerX/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/viz/backends/renderer.py_reload_backends rcOsttj||S)N)_get_3d_backendrZ _Renderer)argskwargsrrr _get_renderer$srcCs,t|td|dkrdn|}td|t|S)NrZpyvistar )r strr rrrrr_check_3d_backend_name)s  rcCs$t}t|}t|kr t||a|S)u Set the 3D backend for MNE. The backend will be set as specified and operations will use that backend. Parameters ---------- backend_name : str The 3d backend to select. See Notes for the capabilities of each backend (``'pyvistaqt'`` and ``'notebook'``). .. versionchanged:: 0.24 The ``'pyvista'`` backend was renamed ``'pyvistaqt'``. %(verbose)s Returns ------- old_backend_name : str | None The old backend that was in use. Notes ----- To use PyVista, set ``backend_name`` to ``pyvistaqt`` but the value ``pyvista`` is still supported for backward compatibility. This table shows the capabilities of each backend ("✓" for full support, and "-" for partial support): .. table:: :widths: auto +--------------------------------------+-----------+----------+ | **3D function:** | pyvistaqt | notebook | +======================================+===========+==========+ | :func:`plot_vector_source_estimates` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`plot_source_estimates` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`plot_alignment` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`plot_sparse_source_estimates` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`plot_evoked_field` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`snapshot_brain_montage` | ✓ | ✓ | +--------------------------------------+-----------+----------+ | :func:`link_brains` | ✓ | | +--------------------------------------+-----------+----------+ +--------------------------------------+-----------+----------+ | **Feature:** | +--------------------------------------+-----------+----------+ | Large data | ✓ | ✓ | +--------------------------------------+-----------+----------+ | Opacity/transparency | ✓ | ✓ | +--------------------------------------+-----------+----------+ | Support geometric glyph | ✓ | ✓ | +--------------------------------------+-----------+----------+ | Smooth shading | ✓ | ✓ | +--------------------------------------+-----------+----------+ | Subplotting | ✓ | ✓ | +--------------------------------------+-----------+----------+ | Inline plot in Jupyter Notebook | | ✓ | +--------------------------------------+-----------+----------+ | Inline plot in JupyterLab | | ✓ | +--------------------------------------+-----------+----------+ | Inline plot in Google Colab | | | +--------------------------------------+-----------+----------+ | Toolbar | ✓ | ✓ | +--------------------------------------+-----------+----------+ )MNE_3D_BACKENDrr)rrZold_backend_namerrrset_3d_backend0s Ir!c CsDz t}Wn4tk r>}zd}tt|W5d}~XYnX|S)aiReturn the 3D backend currently used. Returns ------- backend_used : str | None The 3d backend currently in use. If no backend is found, returns ``None``. .. versionchanged:: 0.24 The ``'pyvista'`` backend has been renamed ``'pyvistaqt'``, so ``'pyvista'`` is no longer returned by this function. N)r RuntimeErrorrrr)rexcrrrget_3d_backends   r$c Cstdkrtdddatdkrt}tD]H}z t|Wn.tk rd}zt|||<W5d}~XYq&X|aqq&tdddd| Dddnt tattt tatS) z'Load and return the current 3d backend.Nr )keydefaultz$Could not load any valid 3D backend  css |]\}}|d|VqdS)z: Nr).0r%valrrr sz"_get_3d_backend..)z) install pyvistaqt, using pip or conda:z'pip install pyvistaqt'z('conda install -c conda-forge pyvistaqt'z4 or install ipywidgets, if using a notebook backendz'pip install ipywidgets'z)'conda install -c conda-forge ipywidgets') r rdictrr ImportErrorrr"joinitemsr)errorsrr#rrrrs,   rccsDt|}z dVW5|dk r>z t|Wntk r<YnXXdS)a0Create a 3d visualization context using the designated backend. See :func:`mne.viz.set_3d_backend` for more details on the available 3d backends and their capabilities. Parameters ---------- backend_name : {'pyvistaqt', 'notebook'} The 3d backend to use in the context. N)r! Exception)rZ old_backendrrruse_3d_backends   r1c csFt6t|"t| dVW5QRXW5QRXW5QRXdS)zCreate a testing viz context. Parameters ---------- backend_name : str The 3d backend to use in the context. interactive : bool If True, ensure interactive elements are accessible. N)_actors_invisibler1rZ_testing_context)r interactiverrr_use_test_3d_backends   r4ccst}daz dVW5|aXdS)NT)MNE_3D_BACKEND_TESTING)Z orig_testingrrrr2s  r2Tc Cstj|||||||ddS)a&Configure the view of the given scene. Parameters ---------- figure : object The scene which is modified. %(azimuth)s %(elevation)s %(focalpoint)s %(distance)s %(roll)s reset_camera : bool If True, reset the camera properties beforehand. figureZazimuthZ elevationZ focalpointZdistanceZrollZ reset_cameraN)rZ _set_3d_viewr6rrr set_3d_viewsr8(cCstj|||ddS)zConfigure the title of the given scene. Parameters ---------- figure : object The scene which is modified. title : str The title of the scene. size : int The size of the title. r7titlesizeN)rZ _set_3d_titler:rrr set_3d_titles r=rrr)sceneshowcCs&t|||||d}|r|S|SdS)aHReturn an empty figure based on the current 3d backend. .. warning:: Proceed with caution when the renderer object is returned (with ``scene=False``) because the _Renderer API is not necessarily stable enough for production, it's still actively in development. Parameters ---------- size : tuple The dimensions of the 3d figure (width, height). bgcolor : tuple The color of the background. smooth_shading : bool If True, smooth shading is enabled. Defaults to True. handle : int | None The figure identifier. scene : bool If True (default), the returned object is the Figure3D. If False, an advanced, undocumented Renderer object is returned (the API is not stable or documented, so this is not recommended). show : bool If True, show the renderer immediately. .. versionadded:: 1.0 Returns ------- figure : instance of Figure3D or ``Renderer`` The requested empty figure or renderer, depending on ``scene``. )Zfigr<bgcolorsmooth_shadingr@N)rr?)r<rArBhandler?r@rendererrrrcreate_3d_figures!rEcCst|dS)zzClose the given scene. Parameters ---------- figure : object The scene which needs to be closed. N)rZ_close_3d_figure)r7rrrclose_3d_figureAsrFcCs tdS)z/Close all the scenes of the current 3d backend.N)rZ _close_allrrrrclose_all_3d_figuresLsrGcCsddlm}|S)zReturn the proper Brain class based on the current 3d backend. Returns ------- brain : object The Brain class corresponding to the current 3d backend. rBrain)Z viz._brainrIrHrrrget_brain_classQs rJ)N)F)NNNNNT)r9)r>TN)"__doc__ contextlibrrZ_utilsrutilsrrrr r r r r5ZMNE_3D_BACKEND_INTERACTIVEr+rrrrrr!r$rr1r4r2r8r=rErFrGrJrrrrsT    P#      .