From ad3580b8427f56b7659056f2ac8c78bb5732add0 Mon Sep 17 00:00:00 2001
From: Clemens Brunner <clemens.brunner@gmail.com>
Date: Tue, 17 Oct 2017 16:55:29 +0200
Subject: [PATCH] Add read_raw wrapper function

---
 mne/io/__init__.py |  2 ++
 mne/io/raw.py      | 44 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+)
 create mode 100644 mne/io/raw.py

diff --git a/mne/io/__init__.py b/mne/io/__init__.py
index 2b4dab3fc98..99df7257594 100644
--- a/mne/io/__init__.py
+++ b/mne/io/__init__.py
@@ -48,3 +48,5 @@
 from .base import concatenate_raws
 from .reference import (set_eeg_reference, set_bipolar_reference,
                         add_reference_channels)
+
+from .raw import read_raw
diff --git a/mne/io/raw.py b/mne/io/raw.py
new file mode 100644
index 00000000000..18c3ae0bb17
--- /dev/null
+++ b/mne/io/raw.py
@@ -0,0 +1,44 @@
+# Authors: Clemens Brunner <clemens.brunner@gmail.com>
+#
+# License: BSD (3-clause)
+
+from os.path import splitext
+
+from .artemis123 import read_raw_artemis123
+from .brainvision import read_raw_brainvision
+from .bti import read_raw_bti  # TODO: which extension?
+from .cnt import read_raw_cnt
+from .ctf import read_raw_ctf
+from .edf import read_raw_edf
+from .egi import read_raw_egi
+from .kit import read_raw_kit
+from .fiff import read_raw_fif
+from .nicolet import read_raw_nicolet
+from .eeglab import read_raw_eeglab
+
+
+def read_raw(input_fname, **kwargs):
+    _, ext = splitext(input_fname)
+
+    if ext == '.bin':
+        return read_raw_artemis123(input_fname, **kwargs)
+    elif ext == '.vhdr':
+        return read_raw_brainvision(input_fname, **kwargs)
+    elif ext == '.cnt':
+        return read_raw_cnt(input_fname, **kwargs)
+    elif ext == '.ds':
+        return read_raw_ctf(input_fname, **kwargs)
+    elif ext in ['.edf', '.bdf', '.gdf']:
+        return read_raw_edf(input_fname, **kwargs)
+    elif ext == '.mff':
+        return read_raw_egi(input_fname, **kwargs)
+    elif ext == '.sqd':
+        return read_raw_kit(input_fname, **kwargs)
+    elif ext == '.fif':
+        return read_raw_fif(input_fname, **kwargs)
+    elif ext == '.data':
+        return read_raw_nicolet(input_fname, **kwargs)
+    elif ext == '.set':
+        return read_raw_eeglab(input_fname, **kwargs)
+    else:
+        raise ValueError('File type {} not supported.'.format(ext))