U f{j*U@s6dZddlZddlmZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZddlmZddlmZmZdadd Zd d Zd Zd Zd)ddZ d*ddZ!d+ddZ"d,ddZ#d-ddZ$d.ddZ%ed/ddZ&ddZ'd d!Z(d"Z)d#d$Z*d0d%d&d'd(Z+dS)1zThe config functions.N)partial)_validate_type_check_qt_version _check_option _check_fname)fill_doc)warnloggercCs0|dk rt|std|td|dddS)ajSet the directory to be used for temporary file storage. This directory is used by joblib to store memmapped arrays, which reduces memory requirements and speeds up parallel computation. Parameters ---------- cache_dir : str or None Directory to use for temporary file storage. None disables temporary file storage. NzDirectory %s does not exist MNE_CACHE_DIRFset_env)opexistsIOError set_config) cache_dirrO/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/utils/config.py set_cache_dirs  rcCs@|dk r.t|tstd|ddkr.tdtd|dddS) aUSet the minimum size for memmaping of arrays for parallel processing. Parameters ---------- memmap_min_size : str or None Threshold on the minimum size of arrays that triggers automated memory mapping for parallel processing, e.g., '1M' for 1 megabyte. Use None to disable memmaping of large arrays. Nz%'memmap_min_size' has to be a string.)KMGzMThe size has to be given in kilo-, mega-, or gigabytes, e.g., 100K, 500M, 1G.MNE_MEMMAP_MIN_SIZEFr ) isinstancestr ValueErrorr)Zmemmap_min_sizerrrset_memmap_min_size1s   r)GZMNE_3D_OPTION_ANTIALIASZMNE_3D_OPTION_DEPTH_PEELINGZMNE_3D_OPTION_MULTI_SAMPLESZMNE_3D_OPTION_SMOOTH_SHADINGZMNE_3D_OPTION_THEMEZMNE_BROWSE_RAW_SIZEZMNE_BROWSER_BACKENDZMNE_BROWSER_OVERVIEW_MODEZMNE_BROWSER_PRECOMPUTEZMNE_BROWSER_THEMEZMNE_BROWSER_USE_OPENGLr ZMNE_COREG_ADVANCED_RENDERINGZMNE_COREG_COPY_ANNOTZMNE_COREG_FULLSCREENZMNE_COREG_GUESS_MRI_SUBJECTZMNE_COREG_HEAD_HIGH_RESZMNE_COREG_HEAD_OPACITYZMNE_COREG_HEAD_INSIDEZMNE_COREG_INTERACTIONZMNE_COREG_MARK_INSIDEZMNE_COREG_PREPARE_BEMZMNE_COREG_ORIENT_TO_SURFACEZMNE_COREG_SCALE_LABELSZMNE_COREG_SCALE_BY_DISTANCEZMNE_COREG_SCENE_SCALEZMNE_COREG_WINDOW_HEIGHTZMNE_COREG_WINDOW_WIDTHZMNE_COREG_SUBJECTS_DIRZMNE_CUDA_DEVICEZMNE_CUDA_IGNORE_PRECISIONZMNE_DATAZMNE_DATASETS_BRAINSTORM_PATHZMNE_DATASETS_EEGBCI_PATHMNE_DATASETS_EPILEPSY_ECOG_PATHZMNE_DATASETS_HF_SEF_PATHZMNE_DATASETS_MEGSIM_PATHZMNE_DATASETS_MISC_PATHZMNE_DATASETS_MTRF_PATHZMNE_DATASETS_SAMPLE_PATHZMNE_DATASETS_SOMATO_PATHZMNE_DATASETS_MULTIMODAL_PATHZMNE_DATASETS_FNIRS_MOTOR_PATHZMNE_DATASETS_OPM_PATHZ$MNE_DATASETS_SPM_FACE_DATASETS_TESTSZMNE_DATASETS_SPM_FACE_PATHZMNE_DATASETS_TESTING_PATHZ&MNE_DATASETS_VISUAL_92_CATEGORIES_PATHZMNE_DATASETS_KILOWORD_PATHZMNE_DATASETS_FIELDTRIP_CMC_PATHZMNE_DATASETS_PHANTOM_4DBTI_PATHZMNE_DATASETS_LIMO_PATHZMNE_DATASETS_REFMEG_NOISE_PATHZMNE_DATASETS_SSVEP_PATHZMNE_DATASETS_ERP_CORE_PATHrZMNE_FORCE_SERIALZMNE_KIT2FIFF_STIM_CHANNELSZ MNE_KIT2FIFF_STIM_CHANNEL_CODINGZMNE_KIT2FIFF_STIM_CHANNEL_SLOPEZ#MNE_KIT2FIFF_STIM_CHANNEL_THRESHOLDZMNE_LOGGING_LEVELrZMNE_SKIP_FTP_TESTSZMNE_SKIP_NETWORK_TESTSZMNE_SKIP_TESTING_DATASET_TESTSMNE_STIM_CHANNELZMNE_TQDMZ MNE_USE_CUDAZ MNE_USE_NUMBA SUBJECTS_DIR)r ZMNE_DATASETS_FNIRSZMNE_NIRSFc Cs`t|dL}zt|}Wn6tk rPd|}|r>t|t|t}YnXW5QRX|S)zSafely load a config file.rzOThe MNE-Python config file (%s) is not a valid JSON file and might be corrupted)openjsonloadr RuntimeErrorr dict) config_path raise_errorfidconfigmsgrrr _load_configs r-cCstt|dd}|S)aGet path to standard mne-python config file. Parameters ---------- home_dir : str | None The folder that contains the .mne config folder. If None, it is found automatically. Returns ------- config_path : str The path to the mne-python configuration file. On windows, this will be '%USERPROFILE%\.mne\mne-python.json'. On every other system, this will be ~/.mne/mne-python.json. home_dirzmne-python.json)rjoin_get_extra_data_path)r/valrrrget_config_paths r3Tc Cst|ttdfdd|dkr"tS|rB|dk rB|tjkrBtj|St|d}t|s\i}nt |}|dkr|rt | t tj}| dd|D|S|dkr||kr|rd nd}|rd |nd} |rd nd} d |} td |||| | | fn |||SdS)a'Read MNE-Python preferences from environment or config file. Parameters ---------- key : None | str The preference key to look for. The os environment is searched first, then the mne-python config file is parsed. If None, all the config parameters present in environment variables or the path are returned. If key is an empty string, a list of all valid keys (but not values) is returned. default : str | None Value to return if the key is not found. raise_error : bool If True, raise an error if the key is not found (instead of returning default). home_dir : str | None The folder that contains the .mne config folder. If None, it is found automatically. use_env : bool If True, consider env vars, if available. If False, only use MNE-Python configuration file values. .. versionadded:: 0.18 Returns ------- value : dict | str | None The preference key value. See Also -------- set_config Nkeyzstring or Noner.cSsi|]}|tj|qSr)osenviron).0r4rrr szget_config..Tzthe environment or in the z=either os.environ["%s"] = VALUE for a temporary solution, or zA You can also set the environment variable before running python.zCmne.utils.set_config("%s", VALUE, set_env=True) for a permanent onezDKey "%s" not found in %sthe mne-python config file (%s). Try %s%s.%s)rrtypeknown_config_typesr6r7r3risfiler-setunion intersectionupdateKeyErrorget) r4defaultr)r/Zuse_envr(r+Zenv_keysZloc_envZmeth_envZ extra_envZ meth_filerrr get_configsJ#     rDc stddt|tdtdfd|dk r2t|}tkr\tfddtDs\tdt|d }t |r~t |d d }nt }t d ||dkr|d|rԈtjkrtj=n||<|r|tj<t|}t|st|t|d }tj||d ddW5QRXdS)a5Set a MNE-Python preference key in the config file and environment. Parameters ---------- key : str The preference key to set. value : str | None The value to assign to the preference key. If None, the key is deleted. home_dir : str | None The folder that contains the .mne config folder. If None, it is found automatically. set_env : bool If True (default), update :data:`os.environ` in addition to updating the MNE-Python config file. See Also -------- get_config rr4 path-likeNvaluec3s|]}|VqdSN) startswith)r8kr4rr %szset_config..z&Setting non-standard config type: "%s"r.Tr)z:Attempting to create new mne-python configuration file: %swr) sort_keysindent)rrr:r;anyknown_config_wildcardsr r3rr<r-r'r infopopr6r7dirnameisdirmkdirr#r$dump)r4rFr/r r(r+ directoryr*rrJrrs6           rcCs|dkrtjd}|dkrdtjkrZtttddrNtd}qtd}nHtddd krt dkrt a t ttjt d d t }n tjd }|dkrtd t|dS)z.Get path to extra data (config, tables, etc.).NZ_MNE_FAKE_HOME_DIRntAPPDATAz.mne USERPROFILEZMNE_DONTWRITE_HOMEr5trueT) ignore_errors~zfmne-python config file path could not be determined, please report this error to mne-python developers)r6r7rBnamelowerrrUr0getenv_temp_home_dirtempfilemkdtempatexitregisterrshutilrmtreepath expanduserrr.rrrr1Bs$     r1cCsBt|dddd|dkr$td|d}|dk r>t|dd d dd }|S) aSafely use subjects_dir input to return SUBJECTS_DIR. Parameters ---------- subjects_dir : str | None If a value is provided, return subjects_dir. Otherwise, look for SUBJECTS_DIR config and return the result. raise_error : bool If True, raise a KeyError if no value for SUBJECTS_DIR can be found (instead of returning None). Returns ------- value : str | None The SUBJECTS_DIR value. )rEN subjects_dirzstr or path-like)itemtypesZ item_name type_nameNr!rLreadT)fname overwriteZ must_existneed_dirr_)rrDr)rkr)rrrget_subjects_dirfs rscs|dk r>t|ts$t|dd|g}|D]}t|ddq(|St}d}td}|dk r|dkr|||d7}td |}qP|dkr|Sd dkrd gSd dkrd gSd d lm}|dddd}t|dkrfdd|D}n |rtd|S)aDetermine the appropriate stim_channel. First, 'MNE_STIM_CHANNEL', 'MNE_STIM_CHANNEL_1', 'MNE_STIM_CHANNEL_2', etc. are read. If these are not found, it will fall back to 'STI 014' if present, then fall back to the first channel of type 'stim', if present. Parameters ---------- stim_channel : str | list of str | None The stim channel selected by the user. %(info_not_none)s Returns ------- stim_channel : str | list of str The name of the stim channel(s) to use Nrz Stim channelzEach provided stim channelrr ch_namesrzMNE_STIM_CHANNEL_%dZSTI101zSTI 014) pick_typesFT)ZmegZref_megZstimcsg|]}d|qS)rtr)r8Zch_rRrr sz%_get_stim_channel..z]No stim channels found. Consider specifying them manually using the 'stim_channel' parameter.) rlistrrDappendZio.pickrvlenr)Z stim_channelrRr)ZchannelZch_countchrvrrwr_get_stim_channels6       r}csZttttd}t|dttdrVtfdddDrVt}|S)z*Get as close to the repo root as possible.z..zsetup.pyc3s |]}tt|VqdSrG)rrUr0)r8xZup_dirrrrKsz _get_root_dir..)mneZexamplesdoc)rabspathr0rT__file__r<all)root_dirrrr _get_root_dirs  rc Csddlm}|dd}tW5QRX|d}|g}t|D]~\}}dD]p}|d|rX||d}d|krd }n2z|d dd d}Wnt k rYnX|d ||fg7}qXqLd |}|S)Nr) SilenceStdoutF)close )ZlapackZblasz %s_opt_infoz NOT AVAILABLEunknown['z%s=%sz, ) Z_testingrnpZ show_configgetvaluesplitr enumeraterH IndexErrorr0)rcapturelineslibsZliliner4librrr_get_numpy_libss&    rzRfrom pyvista import GPUInfo; gi = GPUInfo(); print(gi.version); print(gi.renderer)cCsNtjtjdtgddd}|jdd d}|j sFt |dkrJd S|S) Nz-cFT)checkcapture_output r5rru)NN) subprocessrunsys executable_gpu_cmdstdoutdecodestripreplacer returncoder{)procoutrrr _get_gpu_infos ruser) dependenciesc Cs^t|ttd|d|dkr"dnd}t}tdkrtjddd krtd }t d |}|r|d }d |d |}~~t t d|d}|d ||d|d |ttj ddd|d |tjd|d |tdz d dl}Wntk r*|dYnX||d|d |z d dl} Wntk rv|dYn X|| jtddd|dt} d} |dkr| d7} | D]} | dkr|dq|| d  |z t| } Wn tk r|d!YqX| d"krt| }d#D]6}t||r2t||}|dkr2||q~q2|d$n || j| d%kr|d&| d'n| d(krtd)d*\}}|d&|d+|d'n^| d,kr|d-| d'n>| d.kr(t!\}}|dkr|d/n|d0|d1|d'|rN|dd|d2t"#| j$|dqdS)3aPrint the system information for debugging. This function is useful for printing system information to help triage bugs. Parameters ---------- fid : file-like | None The file to write to. Will be passed to :func:`print()`. Can be None to use :data:`sys.stdout`. show_paths : bool If True, print paths for each module. dependencies : 'user' | 'developer' Show dependencies relevant for users (default) or for developers (i.e., output includes additional dependencies). .. versionadded:: 0.24 Examples -------- Running this function with no arguments prints an output that is useful when submitting bug reports:: >>> import mne >>> mne.sys_info() # doctest: +SKIP Platform: Linux-4.15.0-1067-aws-x86_64-with-glibc2.2.5 Python: 3.8.1 (default, Feb 2 2020, 08:37:37) [GCC 8.3.0] Executable: /usr/local/bin/python CPU: : 36 cores Memory: 68.7 GB mne: 0.21.dev0 numpy: 1.19.0 {blas=openblas, lapack=openblas} scipy: 1.5.1 matplotlib: 3.2.2 {backend=Qt5Agg} sklearn: 0.23.1 numba: 0.50.1 nibabel: 3.1.1 nilearn: 0.7.0 dipy: 1.1.1 cupy: Not found pandas: 1.0.5 pyvista: 0.25.3 {pyvistaqt=0.1.1, OpenGL 3.3 (Core Profile) Mesa 18.3.6 via llvmpipe (LLVM 7.0, 256 bits)} vtk: 9.0.1 qtpy: 2.0.1 {PySide6=6.2.4} pyqtgraph: 0.12.4 pooch: v1.5.1 r)r developerrDarwinNru)rzDarwin-.*?-(.*)zmacOS--r5)endfilez Platform:rzPython: z Executable:zCPU:z: zFnumber of processors unavailable (requires "multiprocessing" package) z cores zMemory:z'Unavailable (requires "psutil" package)i@z0.1fz GB )rnumpyZscipy matplotlibr5ZsklearnZnumbaZnibabelZnilearnZdipyZcupyZpandaspyvistaZ pyvistaqtZ ipyvtklinkvtkqtpyZipymplZ pyqtgraphZpoochr5Zmne_bidsZmne_nirsZ mne_featuresZmne_qt_browserZmne_connectivityZ mne_icalabel)r5ZsphinxZsphinx_galleryZnumpydocZpydata_sphinx_themeZpytestZnbclient:z Not found r)ZGetVTKVersionFullZ GetVTKVersionrrz {}rT)Z return_api=rz {backend=rz" {OpenGL could not be initialized}z {OpenGL z via u•)%rrrplatformsystemr version_infomac_verrefindallrprintljustversionrr processormultiprocessing ImportError cpu_countpsutilZvirtual_memorytotalfloatr __import__ ExceptionZ vtkVersionhasattrgetattr __version__r get_backendrrrTr)r*Z show_pathsrrZ platform_strZ macos_verZmacos_architecturerrrrZ use_mod_namesmod_namemodZ vtk_versionattrrapirendererrrrsys_infos2    $                     r)F)N)NNFNT)NT)N)NF)T)NF),__doc__re functoolsrr$r6os.pathrirrrgrrrcrrrrrrrrdocsrZ_loggingr r rbrrr;rQr-r3rDrr1rsr}rrrrrrrrrsF   K   K : $  3