initial commit

mne/datasets/brainstorm/__init__.py

@@ -0,0 +1,7 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+"""Brainstorm datasets."""
+
+from . import bst_raw, bst_resting, bst_auditory, bst_phantom_ctf, bst_phantom_elekta

mne/datasets/brainstorm/bst_auditory.py

@@ -0,0 +1,69 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+from ...utils import verbose
+from ..utils import (
+    _data_path_doc_accept,
+    _download_mne_dataset,
+    _get_version,
+    _version_doc,
+)
+
+_description = """
+URL: http://neuroimage.usc.edu/brainstorm/DatasetAuditory
+    - One subject, two acquisition runs of 6 minutes each
+    - Subject stimulated binaurally with intra-aural earphones
+      (air tubes+transducers)
+    - Each run contains:
+        - 200 regular beeps (440Hz)
+        - 40 easy deviant beeps (554.4Hz, 4 semitones higher)
+    - Random inter-stimulus interval: between 0.7s and 1.7s seconds, uniformly
+      distributed
+    - The subject presses a button when detecting a deviant with the right
+      index finger
+    - Auditory stimuli generated with the Matlab Psychophysics toolbox
+"""
+
+
+@verbose
+def data_path(
+    path=None,
+    force_update=False,
+    update_path=True,
+    download=True,
+    accept=False,
+    *,
+    verbose=None,
+):  # noqa: D103
+    return _download_mne_dataset(
+        name="bst_auditory",
+        processor="nested_untar",
+        path=path,
+        force_update=force_update,
+        update_path=update_path,
+        download=download,
+        accept=accept,
+    )
+
+
+_data_path_doc = _data_path_doc_accept.format(
+    name="brainstorm", conf="MNE_DATASETS_BRAINSTORM_DATA_PATH"
+)
+_data_path_doc = _data_path_doc.replace(
+    "brainstorm dataset", "brainstorm (bst_auditory) dataset"
+)
+data_path.__doc__ = _data_path_doc
+
+
+def get_version():  # noqa: D103
+    return _get_version("bst_auditory")
+
+
+get_version.__doc__ = _version_doc.format(name="brainstorm")
+
+
+def description():
+    """Get description of brainstorm (bst_auditory) dataset."""
+    for desc in _description.splitlines():
+        print(desc)

mne/datasets/brainstorm/bst_phantom_ctf.py

@@ -0,0 +1,58 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+from ...utils import verbose
+from ..utils import (
+    _data_path_doc_accept,
+    _download_mne_dataset,
+    _get_version,
+    _version_doc,
+)
+
+_description = """
+URL: http://neuroimage.usc.edu/brainstorm/Tutorials/PhantomCtf
+"""
+
+
+@verbose
+def data_path(
+    path=None,
+    force_update=False,
+    update_path=True,
+    download=True,
+    accept=False,
+    *,
+    verbose=None,
+):  # noqa: D103
+    return _download_mne_dataset(
+        name="bst_phantom_ctf",
+        processor="nested_untar",
+        path=path,
+        force_update=force_update,
+        update_path=update_path,
+        download=download,
+        accept=accept,
+    )
+
+
+_data_path_doc = _data_path_doc_accept.format(
+    name="brainstorm", conf="MNE_DATASETS_BRAINSTORM_DATA_PATH"
+)
+_data_path_doc = _data_path_doc.replace(
+    "brainstorm dataset", "brainstorm (bst_phantom_ctf) dataset"
+)
+data_path.__doc__ = _data_path_doc
+
+
+def get_version():  # noqa: D103
+    return _get_version("bst_phantom_ctf")
+
+
+get_version.__doc__ = _version_doc.format(name="brainstorm")
+
+
+def description():
+    """Get description of brainstorm (bst_phantom_ctf) dataset."""
+    for desc in _description.splitlines():
+        print(desc)

mne/datasets/brainstorm/bst_phantom_elekta.py

@@ -0,0 +1,58 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+from ...utils import verbose
+from ..utils import (
+    _data_path_doc_accept,
+    _download_mne_dataset,
+    _get_version,
+    _version_doc,
+)
+
+_description = """
+URL: http://neuroimage.usc.edu/brainstorm/Tutorials/PhantomElekta
+"""
+
+
+@verbose
+def data_path(
+    path=None,
+    force_update=False,
+    update_path=True,
+    download=True,
+    accept=False,
+    *,
+    verbose=None,
+):  # noqa: D103
+    return _download_mne_dataset(
+        name="bst_phantom_elekta",
+        processor="nested_untar",
+        path=path,
+        force_update=force_update,
+        update_path=update_path,
+        download=download,
+        accept=accept,
+    )
+
+
+_data_path_doc = _data_path_doc_accept.format(
+    name="brainstorm", conf="MNE_DATASETS_BRAINSTORM_DATA_PATH"
+)
+_data_path_doc = _data_path_doc.replace(
+    "brainstorm dataset", "brainstorm (bst_phantom_elekta) dataset"
+)
+data_path.__doc__ = _data_path_doc
+
+
+def get_version():  # noqa: D103
+    return _get_version("bst_phantom_elekta")
+
+
+get_version.__doc__ = _version_doc.format(name="brainstorm")
+
+
+def description():
+    """Get description of brainstorm (bst_phantom_elekta) dataset."""
+    for desc in _description.splitlines():
+        print(desc)

mne/datasets/brainstorm/bst_raw.py

@@ -0,0 +1,88 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+from functools import partial
+
+from ...utils import get_config, verbose
+from ..utils import (
+    _data_path_doc_accept,
+    _download_mne_dataset,
+    _get_version,
+    _version_doc,
+    has_dataset,
+)
+
+has_brainstorm_data = partial(has_dataset, name="bst_raw")
+
+_description = """
+URL: http://neuroimage.usc.edu/brainstorm/DatasetMedianNerveCtf
+    - One subject, one acquisition run of 6 minutes
+    - Subject stimulated using Digitimer Constant Current Stimulator
+      (model DS7A)
+    - The run contains 200 electric stimulations randomly distributed between
+      left and right:
+        - 102 stimulations of the left hand
+        - 98 stimulations of the right hand
+    - Inter-stimulus interval: jittered between [1500, 2000]ms
+    - Stimuli generated using PsychToolBox on Windows PC (TTL pulse generated
+      with the parallel port connected to the Digitimer via the rear panel BNC)
+"""
+
+
+@verbose
+def data_path(
+    path=None,
+    force_update=False,
+    update_path=True,
+    download=True,
+    accept=False,
+    *,
+    verbose=None,
+):  # noqa: D103
+    return _download_mne_dataset(
+        name="bst_raw",
+        processor="nested_untar",
+        path=path,
+        force_update=force_update,
+        update_path=update_path,
+        download=download,
+        accept=accept,
+    )
+
+
+_data_path_doc = _data_path_doc_accept.format(
+    name="brainstorm", conf="MNE_DATASETS_BRAINSTORM_DATA_PATH"
+)
+_data_path_doc = _data_path_doc.replace(
+    "brainstorm dataset", "brainstorm (bst_raw) dataset"
+)
+data_path.__doc__ = _data_path_doc
+
+
+def get_version():  # noqa: D103
+    return _get_version("bst_raw")
+
+
+get_version.__doc__ = _version_doc.format(name="brainstorm")
+
+
+def description():  # noqa: D103
+    """Get description of brainstorm (bst_raw) dataset."""
+    for desc in _description.splitlines():
+        print(desc)
+
+
+def _skip_bstraw_data():
+    skip_testing = get_config("MNE_SKIP_TESTING_DATASET_TESTS", "false") == "true"
+    skip = skip_testing or not has_brainstorm_data()
+    return skip
+
+
+def requires_bstraw_data(func):
+    """Skip testing data test."""
+    import pytest
+
+    return pytest.mark.skipif(
+        _skip_bstraw_data(), reason="Requires brainstorm dataset"
+    )(func)

mne/datasets/brainstorm/bst_resting.py

@@ -0,0 +1,61 @@
+# Authors: The MNE-Python contributors.
+# License: BSD-3-Clause
+# Copyright the MNE-Python contributors.
+
+from ...utils import verbose
+from ..utils import (
+    _data_path_doc_accept,
+    _download_mne_dataset,
+    _get_version,
+    _version_doc,
+)
+
+_description = """
+URL: http://neuroimage.usc.edu/brainstorm/DatasetResting
+    - One subject
+    - Two runs of 10 min of resting state recordings
+    - Eyes open
+"""
+
+
+@verbose
+def data_path(
+    path=None,
+    force_update=False,
+    update_path=True,
+    download=True,
+    accept=False,
+    *,
+    verbose=None,
+):  # noqa: D103
+    return _download_mne_dataset(
+        name="bst_resting",
+        processor="nested_untar",
+        path=path,
+        force_update=force_update,
+        update_path=update_path,
+        download=download,
+        accept=accept,
+    )
+
+
+_data_path_doc = _data_path_doc_accept.format(
+    name="brainstorm", conf="MNE_DATASETS_BRAINSTORM_DATA_PATH"
+)
+_data_path_doc = _data_path_doc.replace(
+    "brainstorm dataset", "brainstorm (bst_resting) dataset"
+)
+data_path.__doc__ = _data_path_doc
+
+
+def get_version():  # noqa: D103
+    return _get_version("bst_resting")
+
+
+get_version.__doc__ = _version_doc.format(name="brainstorm")
+
+
+def description():
+    """Get description of brainstorm (bst_resting) dataset."""
+    for desc in _description.splitlines():
+        print(desc)