actually enable BDF writing

Author: drammock

mne_bids/write.py

@@ -1280,23 +1280,7 @@ def _write_raw_brainvision(raw, bids_fname, events, overwrite):
     )
 
 
-def _write_raw_bdf(raw, bids_fname, overwrite):
-    """Store data as BDF.
-
-    Parameters
-    ----------
-    raw : mne.io.Raw
-        Raw data to save.
-    bids_fname : str
-        The output filename.
-    overwrite : bool
-        Whether to overwrite an existing file or not.
-    """
-    assert str(bids_fname).endswith(".bdf")
-    raw.export(bids_fname, overwrite=overwrite)
-
-
-def _write_raw_edf(raw, bids_fname, overwrite):
+def _write_raw_edf_bdf(raw, bids_fname, overwrite):
     """Store data as EDF.
 
     Parameters
@@ -1308,7 +1292,7 @@ def _write_raw_edf(raw, bids_fname, overwrite):
     overwrite : bool
         Whether to overwrite an existing file or not.
     """
-    assert str(bids_fname).endswith(".edf")
+    assert bids_fname.suffix in (".edf", ".bdf")
     raw.export(bids_fname, overwrite=overwrite)
 
 
@@ -1657,14 +1641,14 @@ def write_raw_bids(
             ``source`` column of ``scans.tsv``. By default, this information
             is not stored.
 
-    format : 'auto' | 'BrainVision' | 'EDF' | 'FIF' | 'EEGLAB'
+    format : 'auto' | 'BrainVision' | 'BDF' | 'EDF' | 'FIF' | 'EEGLAB'
         Controls the file format of the data after BIDS conversion. If
         ``'auto'``, MNE-BIDS will attempt to convert the input data to BIDS
         without a change of the original file format. A conversion to a
         different file format will then only take place if the original file
         format lacks some necessary features.
         Conversion may be forced to BrainVision, EDF, or EEGLAB for (i)EEG,
-        and to FIF for MEG data.
+        to BDF or EDF for EMG, and to FIF for MEG data.
     symlink : bool
         Instead of copying the source files, only create symbolic links to
         preserve storage space. This is only allowed when not anonymizing the
@@ -1905,6 +1889,8 @@ def write_raw_bids(
     else:
         if format == "BrainVision":
             ext = ".vhdr"
+        elif format == "BDF":
+            ext = ".bdf"
         elif format == "EDF":
             ext = ".edf"
         elif format == "EEGLAB":
@@ -1914,7 +1900,7 @@ def write_raw_bids(
         else:
             msg = (
                 'For preloaded data, you must set the "format" parameter '
-                "to one of: BrainVision, EDF, EEGLAB, or FIF"
+                "to one of: BrainVision, BDF, EDF, EEGLAB, or FIF"
             )
             if format != "auto":  # the default was changed
                 msg += f', but got: "{format}"'
@@ -2108,6 +2094,12 @@ def write_raw_bids(
                 warn("Converting data files to BrainVision format for anonymization")
                 convert = True
                 bids_path.update(extension=".vhdr")
+        elif bids_path.datatype == "emg":
+            if ext not in [".edf", ".bdf", ".EDF", ".BDF"]:
+                warn("Converting data files to BDF format for anonymization")
+                convert = True
+                bids_path.update(extension=".bdf")
+
     # Read in Raw object and extract metadata from Raw object if needed
     orient = ORIENTATION.get(ext, "n/a")
     unit = EXT_TO_UNIT_MAP.get(ext, "n/a")
@@ -2263,9 +2255,12 @@ def write_raw_bids(
         if format == "BrainVision" and bids_path.datatype in ["ieeg", "eeg"]:
             convert = True
             bids_path.update(extension=".vhdr")
-        elif format == "EDF" and bids_path.datatype in ["ieeg", "eeg"]:
+        elif format == "EDF" and bids_path.datatype in ["ieeg", "eeg", "emg"]:
             convert = True
             bids_path.update(extension=".edf")
+        elif format == "BDF" and bids_path.datatype in ["emg"]:
+            convert = True
+            bids_path.update(extension=".bdf")
         elif format == "EEGLAB" and bids_path.datatype in ["ieeg", "eeg"]:
             convert = True
             bids_path.update(extension=".set")
@@ -2326,20 +2321,20 @@ def write_raw_bids(
                     else bids_path.fpath
                 ),
             )
+        elif bids_path.datatype in ["emg"] and format == "BDF":
+            bids_path.update(extension=".bdf")
+            _write_raw_edf_bdf(raw, bids_path.fpath, overwrite=overwrite)
         elif bids_path.datatype in ["eeg", "emg", "ieeg"] and format == "EDF":
             warn("Converting data files to EDF format")
             bids_path.update(extension=".edf")
-            _write_raw_edf(raw, bids_path.fpath, overwrite=overwrite)
-        elif bids_path.datatype in ["emg"] and format == "BDF":
-            bids_path.update(extension=".bdf")
-            _write_raw_bdf(raw, bids_path.fpath, overwrite=overwrite)
+            _write_raw_edf_bdf(raw, bids_path.fpath, overwrite=overwrite)
         elif bids_path.datatype in ["eeg", "ieeg"] and format == "EEGLAB":
             warn("Converting data files to EEGLAB format")
             _write_raw_eeglab(raw, bids_path.fpath, overwrite=overwrite)
         elif bids_path.datatype in ["emg"]:
             bids_path.update(extension=".bdf")
             warn("Converting data files to BDF format")
-            _write_raw_edf(raw, bids_path.fpath, overwrite=overwrite)
+            _write_raw_edf_bdf(raw, bids_path.fpath, overwrite=overwrite)
         else:
             warn("Converting data files to BrainVision format")
             bids_path.update(suffix=bids_path.datatype, extension=".vhdr")