U f{j-@sbdZddlmZmZmZed dddddddddddd ddZedd d ZGd d d eZdS)z'Convenience functions for opening GUIs.)verbose get_configwarnFTN) orient_to_surfacescale_by_distance mark_inside interactionscaleadvanced_rendering head_inside fullscreenshowblockrc CsT|df|df| df|df|||d}|D]@\}}t|trP|d|dk}n|dk }|r,td|dq,t}|dkr|d d d k}| dkr|d d d k} |dkr|d d d k}|dkr|d d}|dkr|dd d k}|dkr|dd}|dkr|dd}|dkrHd|kr6|d}nd|krH|d}| dkrb|ddd k} | dkr||ddd k} |dkr|dd}|dkr|ddd k}|dkr|dd}|dkr|ddd k}t|}t|}t |}t |}t|}dd l m }dd!l m }|r*d}}||||| || | || ||f|||||d"S)#aCoregister an MRI with a subject's head shape. The GUI can be launched through the command line interface: .. code-block:: bash $ mne coreg or using a python interpreter as shown in :ref:`tut-source-alignment`. Parameters ---------- tabbed : bool Combine the data source panel and the coregistration panel into a single panel with tabs. split : bool Split the main panels with a movable splitter (good for QT4 but unnecessary for wx backend). width : int | None Specify the width for window (in logical pixels). Default is None, which uses ``MNE_COREG_WINDOW_WIDTH`` config value (which defaults to 800). inst : None | str Path to an instance file containing the digitizer data. Compatible for Raw, Epochs, and Evoked files. subject : None | str Name of the mri subject. %(subjects_dir)s guess_mri_subject : bool When selecting a new head shape file, guess the subject's name based on the filename and change the MRI subject accordingly (default True). height : int | None Specify a height for window (in logical pixels). Default is None, which uses ``MNE_COREG_WINDOW_WIDTH`` config value (which defaults to 400). head_opacity : float | None The opacity of the head surface in the range [0., 1.]. Default is None, which uses ``MNE_COREG_HEAD_OPACITY`` config value (which defaults to 1.). head_high_res : bool | None Use a high resolution head surface. Default is None, which uses ``MNE_COREG_HEAD_HIGH_RES`` config value (which defaults to True). trans : str | None The transform file to use. scrollable : bool Make the coregistration panel vertically scrollable (default True). orient_to_surface : bool | None If True (default), orient EEG electrode and head shape points to the head surface. .. versionadded:: 0.16 scale_by_distance : bool | None If True (default), scale the digitization points by their distance from the scalp surface. .. versionadded:: 0.16 mark_inside : bool | None If True (default), mark points inside the head surface in a different color. .. versionadded:: 0.16 %(interaction_scene_none)s Defaults to ``'terrain'``. .. versionadded:: 0.16 .. versionchanged:: 1.0 Default interaction mode if ``None`` and no config setting found changed from ``'trackball'`` to ``'terrain'``. scale : float | None The scaling for the scene. .. versionadded:: 0.16 advanced_rendering : bool Use advanced OpenGL rendering techniques (default True). For some renderers (such as MESA software) this can cause rendering bugs. .. versionadded:: 0.18 head_inside : bool If True (default), add opaque inner scalp head surface to help occlude points behind the head. .. versionadded:: 0.23 %(fullscreen)s Default is None, which uses ``MNE_COREG_FULLSCREEN`` config value (which defaults to False). .. versionadded:: 1.1 show : bool Show the GUI if True. block : bool Whether to halt program execution until the figure is closed. %(verbose)s Returns ------- frame : instance of CoregistrationUI The coregistration frame. Notes ----- Many parameters (e.g., ``head_opacity``) take None as a parameter, which means that the default will be read from the MNE-Python configuration file (which gets saved when exiting). Step by step instructions for the coregistrations are shown below: .. youtube:: uK4n5g6DBcg FT)tabbedsplit scrollabler guess_mri_subjectr r NzThe parameter zD is not supported with the pyvistaqt 3d backend. It will be ignored.ZMNE_COREG_GUESS_MRI_SUBJECTtrueZMNE_COREG_HEAD_HIGH_RESZMNE_COREG_ADVANCED_RENDERINGZMNE_COREG_HEAD_OPACITYg?ZMNE_COREG_HEAD_INSIDEZMNE_COREG_WINDOW_WIDTHi ZMNE_COREG_WINDOW_HEIGHTiXZ SUBJECTS_DIRZMNE_COREG_SUBJECTS_DIRZMNE_COREG_ORIENT_TO_SURFACEZMNE_COREG_SCALE_BY_DISTANCEZMNE_COREG_INTERACTIONZterrainZMNE_COREG_MARK_INSIDEZMNE_COREG_SCENE_SCALEg{Gz?ZMNE_COREG_FULLSCREENr)MNE_3D_BACKEND_TESTINGCoregistrationUI)Z info_filesubject subjects_dirZhead_resolution head_opacityZ orient_glyphsrrtranssizer rrr r)items isinstancetuplerrgetlowerfloatboolintZviz.backends.rendererr_coregr)rrwidthinstrrrheightrZ head_high_resrrrrrrr r r r r rrZunsupported_paramskeyvalZto_raiseconfigrrr.O/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/gui/__init__.pycoregistration sx                       r0c  Csfddlm} ddlm} ddlm} | } | dkr>| dg} | ||||||||d } |rb| | | S) a4Locate intracranial electrode contacts. Parameters ---------- %(info_not_none)s %(trans_not_none)s aligned_ct : path-like | nibabel.spatialimages.SpatialImage The CT image that has been aligned to the Freesurfer T1. Path-like inputs and nibabel image objects are supported. %(subject)s %(subjects_dir)s groups : dict | None A dictionary with channels as keys and their group index as values. If None, the groups will be inferred by the channel names. Channel names must have a format like ``LAMY 7`` where a string prefix like ``LAMY`` precedes a numeric index like ``7``. If the channels are formatted improperly, group plotting will work incorrectly. Group assignments can be adjusted in the GUI. show : bool Show the GUI if True. block : bool Whether to halt program execution until the figure is closed. %(verbose)s Returns ------- gui : instance of IntracranialElectrodeLocator The graphical user interface (GUI) window. r) _qt_app_execrIntracranialElectrodeLocatorr) QApplicationNzIntracranial Electrode Locator)rrgroupsr r)Zviz.backends._utilsr1 _ieeg_locater3Zqtpy.QtWidgetsr4instance)inforZ aligned_ctrrr5r rrr1r3r4Zappguir.r.r/ locate_ieegs&    r:c@s eZdZdZddZddZdS) _GUIScraperzScrape GUI outputs.cCsdS)Nz r.)selfr.r.r/__repr__sz_GUIScraper.__repr__c Cs2ddlm}ddlm}ddlm}ddlm}|dD]}t |||frZqtpyr?valuesr getattrr@nexthasattrrCZ_windowrBplotterZ screenshotZQPixmapZQPainterZ drawPixmapZmapTorectZtopLeftsaveclose Exception)r<rZ block_varsZ gallery_confr3rr>r?r9Z img_fnameZwindowZpixmaprIZ sub_pixmapr.r.r/__call__sH             z_GUIScraper.__call__N)__name__ __module__ __qualname____doc__r=rNr.r.r.r/r;sr;) FTNNNNNNNNNT)NNNTFN) rRutilsrrrr0r:objectr;r.r.r.r/sBA .