enh: support reading 'tables' from DCOR datasets (close #221)

CHANGELOG

@@ -1,3 +1,5 @@
+0.50.2
+ - enh: support reading 'tables' from DCOR datasets (#221)
 0.50.1
  - enh: support reading 'logs' from DCOR datasets
  - enh: hierarchy parents inherit their parent's logs and tables

dclab/rtdc_dataset/fmt_dcor/__init__.py

@@ -5,6 +5,8 @@
 import re
 import time
 
+import numpy as np
+
 from ...util import hashobj
 
 from ..config import Configuration
@@ -141,6 +143,36 @@ def _logs(self):
         return self.api.get(query="logs")
 
 
+class DCORTables:
+    def __init__(self, api):
+        self.api = api
+
+    def __getitem__(self, key):
+        return self._tables[key]
+
+    def __len__(self):
+        return len(self._tables)
+
+    def keys(self):
+        return self._tables.keys()
+
+    @property
+    @functools.lru_cache()
+    def _tables(self):
+        table_data = self.api.get(query="tables")
+        # assemble the tables
+        tables = {}
+        for key in table_data:
+            columns, data = table_data[key]
+            ds_dt = np.dtype({'names': columns,
+                              'formats': [float] * len(columns)})
+            tab_data = np.asarray(data)
+            rec_arr = np.rec.array(tab_data, dtype=ds_dt)
+            tables[key] = rec_arr
+
+        return tables
+
+
 class RTDC_DCOR(RTDCBase):
     def __init__(self, url, host="dcor.mpl.mpg.de", api_key="",
                  use_ssl=None, cert_path=None, *args, **kwargs):
@@ -199,6 +231,9 @@ def __init__(self, url, host="dcor.mpl.mpg.de", api_key="",
         # Lazy logs
         self.logs = DCORLogs(self.api)
 
+        # Lazy tables
+        self.tables = DCORTables(self.api)
+
         # Get size
         self._size = int(self.api.get(query="size"))