some properties

Author: gpiantoni

.travis.yml

@@ -20,6 +20,7 @@ before_install:
  # create display with large resolution
  - "/sbin/start-stop-daemon --start --quiet --pidfile /tmp/custom_xvfb_99.pid --make-pidfile --background --exec /usr/bin/Xvfb -- :99 -ac -screen 0 1920x1080x16"
  - python setup_wonambi.py --get_files
+ - git clone [email protected]:bids-standard/bids-examples.git tests/data/bids-example
 
 install: 
  # install requirements

setup_wonambi.py

@@ -67,7 +67,6 @@
     }
 
 
-
 def _files_in_downloads():
 
     print('Files stored in cached ' + str(DOWNLOADS_PATH.resolve()) + ':\n\t' +

wonambi/bids/structure.py

@@ -1,5 +1,8 @@
-from wonambi.bids.utils import _match
 from pathlib import Path
+from json import load
+
+from .utils import _match, read_tsv
+
 
 BIDS_ENTITIES = (
     'sub',
@@ -35,12 +38,68 @@ def __init__(self, filename):
         self.format = _match(self._filename, f'_([a-zA-Z0-9-]+){self.extension}')
 
     @property
-    def filename(self):
+    def _stem(self):
         parts = []
         for k, v in self.values.items():
             if v is None:
                 continue
             parts.append(f'{k}-{v}')
 
-        stem = '_'.join(parts)
-        return self.path / f'{stem}_{self.format}{self.extension}'
+        return '_'.join(parts)
+
+    @property
+    def filename(self):
+        return self.path / f'{self._stem}_{self.format}{self.extension}'
+
+
+class BIDSMain(BIDSName):
+    def __init__(self, filename):
+        super().__init__(filename)
+
+    @property
+    def _file_header(self):
+        return self.path / f'{self._stem}_{self.format}.json'
+
+    @property
+    def header(self):
+        with self._file_header.open() as f:
+            return load(f)
+
+    @property
+    def _file_events(self):
+        """
+        TODO
+        ----
+        There might be a json file
+        """
+        return self.path / f'{self._stem}_events.tsv'
+
+    @property
+    def events(self):
+        """
+        TODO
+        ----
+        how to handle n/a
+        """
+        return read_tsv(self._file_events)
+
+
+class BIDSEEG(BIDSMain):
+    def __init__(self, filename):
+        super().__init__(filename)
+
+    @property
+    def _file_channels(self):
+        return self.path / f'{self._stem}_channels.tsv'
+
+    @property
+    def channels(self):
+        return read_tsv(self._file_channels)
+
+    @property
+    def _file_electrodes(self):
+        return self.path / f'{self._stem}_electrodes.tsv'
+
+    @property
+    def electrodes(self):
+        return read_tsv(self._file_electrodes)

wonambi/bids/utils.py

@@ -1,6 +1,18 @@
+from numpy import genfromtxt
 from re import search
 
 
+def read_tsv(filename):
+    tsv = genfromtxt(
+        fname=filename,
+        delimiter='\t',
+        names=True,
+        dtype=None,  # forces it to read strings
+        deletechars='',
+        encoding='utf-8')
+    return tsv
+
+
 def _match(filename, pattern):
     m = search(pattern, filename.name)
     if m is None: