U f{j~U@sddlmZddlZddlmZddlmZmZddl Z ddl Z ddl Z ddl Z ddlmZmZddlmZmZmZddlmZmZmZmZmZmZmZmZmZdd lm Z m!Z!d Z"e"#d Z$d e$d e$d <e$%dd d&e$Z$e!e"e Z"e!e$e Z$dZ'ddZ(ddZ)ddZ*d/ddZ+ddZ,ddZ-ed0ddZ.ed1d d!Z/ed2ddd"d#d$Z0d%d&Z1d'd(Z2Gd)d*d*Z3Gd+d,d,e3eZ4Gd-d.d.e3eZ5dS)3) OrderedDictN) WindowsPath PosixPath)_hcp_mmp_license_text MNE_DATASETS)read_labels_from_annotLabelwrite_labels_to_annot) get_config set_configlogger_validate_typewarnverboseget_subjects_dir_pl _safe_input)docdict _docformataWGet path to local copy of {name} dataset. Parameters ---------- path : None | str Location of where to look for the {name} dataset. If None, the environment variable or config parameter ``{conf}`` is used. If it doesn't exist, the "~/mne_data" directory is used. If the {name} 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 {name} dataset even if a local copy exists. Default is False. update_path : bool | None If True (default), set the ``{conf}`` in mne-python config to the given path. If None, the user is prompted. download : bool If False and the {name} dataset has not been downloaded yet, it will not be downloaded and the path will be returned as '' (empty string). This is mostly used for debugging purposes and can be safely ignored by most users. %(verbose)s Returns ------- path : instance of Path Path to {name} dataset directory. z %(verbose)sz %(accept)szGet version of the local {name} dataset. Returns ------- version : str | None Version of the {name} local dataset, or None if the dataset does not exist locally. c Csft|d}t|r:t|d}|}W5QRXn(td|tddd}| |d}|S)zGet the version of the dataset.z version.txtrzVersion file missing: z0.7z0.3)samplespmz0.0) opjoinexistsopenreadlinestriprdebugdictget)pathnameZ ver_fnamefidversionversionsr*Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/datasets/utils.py_dataset_versionJs     r,cCst|d||dk r|St|p dtd}|dk rXt|sPd|d}t|t|Std|tt dt dd }t|std zt |Wn t k rt d |YnXt|S) zGet a dataset path.)z path-likeNNZMNE_DATAzDownload location z as specified by MNE_DATA does not exist. Either create this directory manually and try again, or set MNE_DATA to an existing directory.z+Using default location ~/mne_data for %s...Z_MNE_FAKE_HOME_DIR~Zmne_datazCreating ~/mne_datazUser does not have write permissions at '%s', try giving the path as an argument to data_path() where user has write permissions, for ex:data_path('/home/xyz/me2/'))rr rrFileNotFoundError _mne_pathrinforosgetenv expandusermkdirOSError)r%keyr&msgr*r*r+ _get_pathYs0       r8cCs~t|}t|ddd|k}|sz|dkrdd}dtjkr`. A :class:`dict` can be used to check for user-defined datasets (see the Notes section of :func:`fetch_dataset`), and must contain keys ``dataset_name``, ``archive_name``, ``url``, ``folder_name``, ``hash``. Returns ------- has : bool True if the dataset is present. rrCrEspm_facerrFNrHF)r%rMZ check_versionZbst_rG) rPrD isinstancer#rr8 startswithr?endswith)r&rDrErJrFr%Zdpcheckr*r*r+rSs     rSTcCsddlm}m}m}m}m}m}m}m}m } m } m } m } m } m}m}m}m}m}m}m}m}m}m}m}|}||||| ||| | | | |||||jjdd|jjdd|jjdd|jjdd}|d|d|jjdd|jdddd gdd t d D]}|j|dd gdd qX|j!j"ddgdgd|d|dt#|ddd|jddd|dS)z5Download all datasets used in examples and tutorials.r)rtestingmiscrUsomato brainstormeegbci multimodalopmhf_sefmtrf fieldtrip_cmckiloword phantom_4dbtisleep_physionetlimo fnirs_motor refmeg_noisefetch_infant_templatefetch_fsaveragessveperp_core epilepsy_ecog fetch_phantomT)rNZotaniemi) subjects_dir )r@)runsr@r)subjectsZ recordingNZ6morw)rprN)subjectr@)$rrrZr[rUr\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmrnroZ data_pathZbst_rawZ bst_auditoryZ bst_restingZbst_phantom_elektaZbst_phantom_ctfZ load_datarangeZageZ fetch_datafetch_hcp_mmp_parcellation)rrrZr[rUr\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmrnroZ sample_pathZ phantom_pathZsubjr*r*r+_download_all_example_datasF h  r{c Csddl}t|dd}t|dd}tddd }td d d }d D]D}|d }t||}t|s>|j||d ||||dq>dS)aFetch the modified subdivided aparc parcellation. This will download and install the subdivided aparc parcellation :footcite:'KhanEtAl2018' files for FreeSurfer's fsaverage to the specified directory. Parameters ---------- subjects_dir : str | None The subjects directory to use. The file will be placed in ``subjects_dir + '/fsaverage/label'``. %(verbose)s References ---------- .. footbibliography:: rNTZ raise_error fsaveragelabelzhttps://osf.io/p92yb/downloadzhttps://osf.io/4kxny/downloadlhrhZ 9e4d8d6b90242b7e4b0145353436ef77Z dd6464db8e7762d969fc1d8087cd211bz.aparc_sub.annotmd5:urlZ known_hashr%fname)rOrrrr#isfileretrieve) rprrO destinationurlshasheshemirfpathr*r*r+fetch_aparc_sub_parcellation's&     r)rNrcsddl}t|dd}t|ddfddd D}td d d }td d d }tdd|Ds|sldtjkrrd}n tdt }| dkrt dt d |D]<\} } t | st| } |j|| d|| | dq|r·fddd D}tdd|Dr dStdtdddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/g} t| d0ks\tt} d D]"} tdd1| |d2d3d4dD}ttt}| D]\}d5|kstfd6dt|D}||rtt|tkstd||<fd7d|D}td8d|D}|t|}t |d9d|D}t|t!gd| d:}||_"||_#| $|q|sftqft| d;kstd| |d2|d?qdS)@aFetch the HCP-MMP parcellation. This will download and install the HCP-MMP parcellation :footcite:`GlasserEtAl2016` files for FreeSurfer's fsaverage :footcite:`Mills2016` to the specified directory. Parameters ---------- subjects_dir : str | None The subjects directory to use. The file will be placed in ``subjects_dir + '/fsaverage/label'``. combine : bool If True, also produce the combined/reduced set of 23 labels per hemisphere as ``HCPMMP1_combined.annot`` :footcite:`GlasserEtAl2016supp`. %(accept)s %(verbose)s Notes ----- Use of this parcellation is subject to terms of use on the `HCP-MMP webpage `_. References ---------- .. footbibliography:: rNTr|r}r~csg|]}td|qS)z%s.HCPMMP1.annotrr.0rrr*r+ psz.fetch_hcp_mmp_parcellation..rz.https://ndownloader.figshare.com/files/5528816z.https://ndownloader.figshare.com/files/5528819Z 46a102b59b2fb1bb4bd62d51bf02e975Z 75e96b331940227bbcb07c1c791c2463css|]}t|VqdSNrrrrr*r*r+ vsz-fetch_hcp_mmp_parcellation..z--accept-hcpmmp-licenser9z%s Agree (y/[n])? z1You must agree to the license to use this datasetrrcsg|]}td|qS)z%s.HCPMMP1_combined.annotrrrr*r+rscss|]}t|VqdSrrrr*r*r+rszCreating combined labels)zPrimary Visual Cortex (V1))ZV1)zEarly Visual Cortex)ZV2ZV3ZV4)zDorsal Stream Visual Cortex)ZV3AZV3BZV6ZV6AZV7ZIPS1)zVentral Stream Visual Cortex)ZV8ZVVCZPITZFFCZVMV1ZVMV2ZVMV3)z(MT+ Complex and Neighboring Visual Areas) ZV3CDZLO1ZLO2ZLO3ZV4tZFSTZMTMSTZPH)zSomatosensory and Motor Cortex)4Z3aZ3b12)z,Paracentral Lobular and Mid Cingulate Cortex)Z24ddZ24dvZ6mpZ6maZSCEFZ5mZ5LZ5mv)zPremotor Cortex)Z55bZ6dZ6aZFEFZ6vZ6rZPEF)zPosterior Opercular Cortex)Z43ZFOP1ZOP4ZOP1zOP2-3ZPFcm)zEarly Auditory Cortex)A1ZLBeltZMBeltZPBeltZRI)zAuditory Association Cortex)ZA4ZA5ZSTSdpZSTSdaZSTSvpZSTSvaZSTGaZTA2)z$Insular and Frontal Opercular Cortex) 52ZPIZIgZPoI1ZPoI2ZFOP2ZFOP3ZMIZAVIZAAICZPirZFOP4ZFOP5)zMedial Temporal Cortex)HZPreSZECZPeEcZPHA1ZPHA2ZPHA3)zLateral Temporal Cortex) ZPHTZTE1pZTE1mZTE1aZTE2pZTE2aZTGvZTGdZTF)z"Temporo-Parieto-Occipital Junction)ZTPOJ1ZTPOJ2ZTPOJ3ZSTVZPSL)zSuperior Parietal Cortex) ZLIPvZLIPdZVIPZAIPZMIPZ7PCZ7ALZ7AmZ7PLZ7Pm)zInferior Parietal Cortex) ZPGpZPGsZPGiZPFmZPFZPFtZPFopZIP0ZIP1ZIP2)zPosterior Cingulate Cortex)ZDVTZProSZPOS1ZPOS2ZRSCZv23abZd23abZ31pvZ31pdZ31aZ23dZ23cZPCVZ7m)z/Anterior Cingulate and Medial Prefrontal Cortex)Z33prZp24prZa24prZp24Za24Zp32prZa32prZd32Zp32Zs32Z8BMZ9mZ10vZ10rZ25)z Orbital and Polar Frontal Cortex) Z47sZ47mZa47rZ11lZ13lZa10pZp10pZ10ppZ10dZOFCZpOFC)zInferior Frontal Cortex)Z44Z45ZIFJpZIFJaZIFSpZIFSaZ47lZp47r)zDorsoLateral Prefrontal Cortex) Z8CZ8Avzi6-8zs6-8ZSFLZ8BLZ9pZ9aZ8Adzp9-46vza9-46vZ46z9-46d)???)rZHCPMMP1F)rrpsortcSs,g|]$}|jdrdn|jddqS)r_r)r&rWsplitrr~r*r*r+rs csg|]\}}|kr|qSr*r*)rliZ label_name)wantr*r+rscsg|] }|qSr*r*)rr)labelsr*r+rscSsg|]}t|jqSr*)lenZverticesrr*r*r+rscSsg|] }|jqSr*)colorrr*r*r+rs)rxr.))rleft)rrightz./%s.fsaverage164.label.giiZHCPMMP1_combined)rrpr table_name)&rOrrrr#allr<r=rrr> RuntimeErrorziprbasenamerrr0rrAssertionErrorlistr npZzerosboolitems enumerateanyarrayfloatsumdotr r&rrQr )rpcombinerNrrOfnamesrrrArrrgroupsZ labels_outZ label_namesusedr6Z these_labelswrZsiderr*)rrrr+rzNs         7    rzc Csddl}t|d}dd|D}W5QRXt|}t}|D] }tt||sD||qDt dt |t |||ft |dkr~t } t dt| d} |j|d|| t| d t d t |ft| dd} td d | D} t| t|} t | r6td d| |D]}| j||dq:W5QRXW5QRXt dt |t |fdS)NrrcSsg|] }|qSr*)r!)rr&r*r*r+rsz,_manifest_check_download..z%d file%s missing from %s in %sz"Downloading missing files remotelyztemp.ziprrzExtracting missing file%scss|]}|ds|VqdS)/N)rX)rfr*r*r+rs z+_manifest_check_download..z'Zip file did not have correct names: %s )r%z Successfully extracted %d file%s)rOr readlinesrrrrrrQrr0rrtempfileTemporaryDirectoryrzipfileZipFilesetnamelistsortedsymmetric_differencerextract)Z manifest_pathrrhash_rOr'namesZneedr&r%Z fname_pathffmembersmissingr*r*r+_manifest_check_downloadsD       &rcCs:tjdkrtnt}||f}|jjs6td|jf|S)Nntz$cannot instantiate %r on your system) r1r&MNEWindowsPath MNEPosixPath _from_parts_flavour is_supportedNotImplementedError__name__)r%klassoutr*r*r+r/s r/c@seZdZddZdS)_PathAddcCs2t|tr*tdtt|tj|StdS)Nzdata_path functions now return pathlib.Path objects which do not natively support the plus (+) operator, switch to using forward slash (/) instead. Support for plus will be removed in 1.2.)rVr?rDeprecationWarningrsepr)selfotherr*r*r+__add__s  z_PathAdd.__add__N)r __module__ __qualname__rr*r*r*r+rsrc@s eZdZdS)rNrrrr*r*r*r+r'src@s eZdZdS)rNrr*r*r*r+r+sr)F)T)NN)NT)6 collectionsrr1os.pathr%rpathlibrrr<rrnumpyrconfigrrr~r r r utilsr r rrrrrrrZ utils.docsrrZ_data_path_docrZ_data_path_doc_acceptinsertrZ _version_docr,r8rBrRrTrSr{rrzrr/rrrr*r*r*r+ sR  ,      " %) 6 &&