U f{j%@sddlZddlmZddlZddlZddlmZmZmZddlm Z m Z dZ e ddd d d Z e ddde dfd d Z ddZdS)N)path) _get_path_do_path_update _mne_path)_url_to_local_pathverbosez+https://physionet.org/files/eegmmidb/1.0.0/F)r c Csddl}d}d}t|||}d}t|t||} | g} t| rF|rt| rZt| tt | szt t | |j || |dt ||||dd| D} | S) atGet path to local copy of EEGMMI dataset URL. This is a low-level function useful for getting a local copy of a remote EEGBCI dataset :footcite:`SchalkEtAl2004` which is available at PhysioNet :footcite:`GoldbergerEtAl2000`. Parameters ---------- url : str The dataset to use. path : None | str Location of where to look for the EEGBCI data storing location. If None, the environment variable or config parameter ``MNE_DATASETS_EEGBCI_PATH`` is used. If it doesn't exist, the "~/mne_data" directory is used. If the EEGBCI dataset is not found under the given path, the data will be automatically downloaded to the specified folder. force_update : bool Force update of the dataset even if a local copy exists. update_path : bool | None If True, set the MNE_DATASETS_EEGBCI_PATH in mne-python config to the given path. If None, the user is prompted. %(verbose)s Returns ------- path : list of Path Local path to the given data file. This path is contained inside a list of length one, for compatibility. Notes ----- For example, one could do: >>> from mne.datasets import eegbci >>> url = 'http://www.physionet.org/physiobank/database/eegmmidb/' >>> eegbci.data_path(url, os.getenv('HOME') + '/datasets') # doctest:+SKIP This would download the given EEGBCI data file to the 'datasets' folder, and prompt the user to save the 'datasets' path to the mne-python config, if it isn't there already. References ---------- .. footbibliography:: rNMNE_DATASETS_EEGBCI_PATHEEGBCIMNE-eegbci-data)urlrfnamecSsg|] }t|qS)r).0destrrY/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/datasets/eegbci/eegbci.py Zszdata_path..) poochrropjoinisfileosremoveisdirdirnamemakedirsretriever) r r force_update update_pathr rkeynamer destination destinationsrrr data_paths(0   r$cCsddl}t|ds|g}d}d} d} t||| }d} t| |} | dkrTtdtj|| f| } |j | |ddd d }t d td d }| |g}|D]d}d|dd|dd|dd}t| |}|rt|rt||||t|||| q|S)aGet paths to local copies of EEGBCI dataset files. This will fetch data for the EEGBCI dataset :footcite:`SchalkEtAl2004`, which is also available at PhysioNet :footcite:`GoldbergerEtAl2000`. Parameters ---------- subject : int The subject to use. Can be in the range of 1-109 (inclusive). runs : int | list of int The runs to use. See Notes for details. path : None | str Location of where to look for the EEGBCI data storing location. If None, the environment variable or config parameter ``MNE_DATASETS_EEGBCI_PATH`` is used. If it doesn't exist, the "~/mne_data" directory is used. If the EEGBCI dataset is not found under the given path, the data will be automatically downloaded to the specified folder. force_update : bool Force update of the dataset even if a local copy exists. update_path : bool | None If True, set the MNE_DATASETS_EEGBCI_PATH in mne-python config to the given path. If None, the user is prompted. base_url : str The URL root for the data. %(verbose)s Returns ------- paths : list List of local data paths of the given type. Notes ----- The run numbers correspond to: ========= =================================== run task ========= =================================== 1 Baseline, eyes open 2 Baseline, eyes closed 3, 7, 11 Motor execution: left vs right hand 4, 8, 12 Motor imagery: left vs right hand 5, 9, 13 Motor execution: hands vs feet 6, 10, 14 Motor imagery: hands vs feet ========= =================================== For example, one could do:: >>> from mne.datasets import eegbci >>> eegbci.load_data(1, [4, 10, 14], os.getenv('HOME') + '/datasets') # doctest:+SKIP This would download runs 4, 10, and 14 (hand/foot motor imagery) runs from subject 1 in the EEGBCI dataset to the 'datasets' folder, and prompt the user to save the 'datasets' path to the mne-python config, if it isn't there already. References ---------- .. footbibliography:: rN__iter__r r r z3(?:https?://.*)(files)/(eegmmidb)/(\d+\.\d+\.\d+)/?zabase_url does not match the expected EEGMI folder structure. Please notify MNE-Python developers.r)rbase_urlversionregistryZretry_if_failedZmnedatazeegbci_checksums.txtSZ03dz/SR02dz.edf)rhasattrrrecompilematch ValueErrorrrgroupscreate pkg_resourcesresource_streamZ load_registryrrrappendfetchr)subjectrunsrrrr&r rZ config_keyfolderr!patternr0 base_pathZfetcherr(Z data_pathsrunZ file_partr"rrr load_data^sB@       r>cCsnt}|jD]R}|d}|}|dr<|ddd}|drVd|dd}|||<q ||dS) zStandardize channel positions and names. Parameters ---------- raw : instance of Raw The raw data to standardize. Operates in-place. .ZNzZFPZFpr)dictZch_namesstripupperendswith startswithZrename_channels)rawrenamer!Zstd_namerrr standardizes     rJ)NFN)rrrr4r.utilsrrrrr Z EEGMI_URLr$r>rJrrrrs  K m