[MLIO] add main MLIO

Author: sprenger

neo/io/__init__.py

@@ -37,6 +37,7 @@
 * :attr:`KwikIO`
 * :attr:`MaxwellIO`
 * :attr:`MicromedIO`
+* :attr:`MonkeyLogicIO`
 * :attr:`NeoMatlabIO`
 * :attr:`NestIO`
 * :attr:`NeuralynxIO`
@@ -158,6 +159,10 @@
 
     .. autoattribute:: extensions
 
+.. autoclass:: neo.io.MonkeyLogicIO
+
+    .. autoattribute:: extensions
+
 .. autoclass:: neo.io.NeoMatlabIO
 
     .. autoattribute:: extensions
@@ -295,6 +300,7 @@
 from neo.io.mearecio import MEArecIO
 from neo.io.maxwellio import MaxwellIO
 from neo.io.micromedio import MicromedIO
+from neo.io.monkeylogicio import MonkeyLogicIO
 from neo.io.neomatlabio import NeoMatlabIO
 from neo.io.nestio import NestIO
 from neo.io.neuralynxio import NeuralynxIO
@@ -345,6 +351,7 @@
     MEArecIO,
     MaxwellIO,
     MicromedIO,
+    MonkeyLogicIO,
     NixIO,  # place NixIO before other IOs that use HDF5 to make it the default for .h5 files
     NeoMatlabIO,
     NestIO,

neo/io/monkeylogicio.py

@@ -0,0 +1,21 @@
+from neo.io.basefromrawio import BaseFromRaw
+from neo.rawio.monkeylogicrawio import MonkeyLogicRawIO
+
+
+class MonkeyLogicIO(MonkeyLogicRawIO, BaseFromRaw):
+
+    name = 'MonkeyLogicIO'
+
+    _prefered_signal_group_mode = 'group-by-same-units'
+    _prefered_units_group_mode = 'all-in-one'
+
+    def __init__(self, filename):
+        MonkeyLogicRawIO.__init__(self, filename)
+        BaseFromRaw.__init__(self, filename)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        self.header = None
+        self.file.close()

neo/test/iotest/test_monkeylogicio.py

@@ -0,0 +1,32 @@
+"""
+Tests of neo.io.monkeylogicio
+"""
+
+import unittest
+
+from neo.io import MonkeyLogicIO
+from neo.test.iotest.common_io_test import BaseTestIO
+
+# class TestMonkeyLogicIO(BaseTestIO, unittest.TestCase):
+#     entities_to_download = [
+#         'monkeylogic'
+#     ]
+#     entities_to_test = [
+#         'monkeylogic/mearec_test_10s.h5'
+#     ]
+#     ioclass = MonkeyLogicIO
+
+
+class TestMonkeyLogicIO(unittest.TestCase):
+    # TODO: Adjust this once ML files are on GIN
+
+    def test_read(self):
+        filename = '/home/sprengerj/projects/monkey_logic/210909_TSCM_5cj_5cl_Riesling.bhv2'
+        filename = '/home/sprengerj/projects/monkey_logic/sabrina/210810__learndms_userloop.bhv2'
+        # filename = '/home/sprengerj/projects/monkey_logic/sabrina/210916__learndms_userloop.bhv2'
+        # filename = '/home/sprengerj/projects/monkey_logic/sabrina/210917__learndms_userloop.bhv2'
+        io = MonkeyLogicIO(filename)
+        io.read_block()
+
+if __name__ == "__main__":
+    unittest.main()