EGI: restore _read_mff helper (mffpy-backed read path)

Author: PragnyaKhandelwal

mne/io/egi/egimff.py

@@ -83,6 +83,13 @@ def _parse_egi_datetime(time_str):
     return datetime.datetime.strptime(time_str, "%Y-%m-%dT%H:%M:%S.%f%z")
 
 
+def _get_mff_reader(input_fname):
+    mffpy = _import_mffpy()
+    mff_reader = mffpy.Reader(input_fname)
+    mff_reader.set_unit("EEG", "V")  # XXX: set PNS unit
+    return mff_reader
+
+
 def _read_events(input_fname, egi_info):
     """Read EGI event tracks from an MFF directory."""
     from mffpy.xml_files import XML, EventTrack
@@ -157,13 +164,6 @@ def _read_events(input_fname, egi_info):
     return egi_events, egi_info, mff_events
 
 
-def _get_mff_reader(input_fname):
-    mffpy = _import_mffpy()
-    mff_reader = mffpy.Reader(input_fname)
-    mff_reader.set_unit("EEG", "V")  # XXX: set PNS unit
-    return mff_reader
-
-
 def _get_montage(mff_reader):
     mffpy = _import_mffpy()
     xml_files = mff_reader.directory.files_by_type[".xml"]
@@ -318,6 +318,25 @@ def _get_annotations(mff_reader, mne_info):
     return event_annots + gap_annots
 
 
+def _read_mff(input_fname):
+    """Read EGI MFF file using the mffpy-backed helpers.
+
+    Returns
+    -------
+    eeg : array-like
+        Raw EEG data as returned by `_get_eeg_data`.
+    info : dict
+        MNE `info` structure built by `_get_info`.
+    annotations : instance of `mne.Annotations`
+        Annotations built from events and gaps via `_get_annotations`.
+    """
+    mff_reader = _get_mff_reader(input_fname)
+    eeg = _get_eeg_data(mff_reader)
+    info = _get_info(mff_reader)
+    annotations = _get_annotations(mff_reader, info)
+    return eeg, info, annotations
+
+
 def _read_mff_header(filepath):
     """Read mff header."""
     _soft_import("defusedxml", "reading EGI MFF data")