changed regular expression and added an evtype

Author: JulesLebert

neo/rawio/tdtrawio.py

@@ -197,7 +197,8 @@ def _parse_header(self):
                 for seg_index, segment_name in enumerate(segment_names):
                     # get data index
                     tsq = self._tsq[seg_index]
-                    mask = (tsq['evtype'] == EVTYPE_STREAM) & \
+                    mask = ((tsq['evtype'] == EVTYPE_STREAM) | \
+                           (tsq['evtype'] == int(33041))) & \
                            (tsq['evname'] == info['StoreName']) & \
                            (tsq['channel'] == chan_id)
                     data_index = tsq[mask].copy()
@@ -248,16 +249,23 @@ def _parse_header(self):
                         sev_filename = (path / sev_stem).with_suffix('.sev')
                     else:
                         # for single block datasets the exact name of sev files in not known
-                        sev_regex = f".*_ch{chan_id}.sev"
+                        store = info['StoreName'].decode('ascii')
+                        # sev_regex = f".*_ch{chan_id}.sev"
+                        sev_regex = f'*_{store}_Ch{chan_id}.sev'
                         sev_filename = list(self.dirname.parent.glob(str(sev_regex)))
 
                         # in case non or multiple sev files are found for current stream + channel
                         if len(sev_filename) != 1:
                             missing_sev_channels.append(chan_id)
                             sev_filename = None
+                        else:
+                            sev_filename = sev_filename[0]
 
                     if (sev_filename is not None) and sev_filename.exists():
                         data = np.memmap(sev_filename, mode='r', offset=0, dtype='uint8')
+                        # sev_info = tdt.read_sev(sev_filename)
+                        # data = sev_info[store].data
+                        # sampling_rate = sev_info[store].fs
                     else:
                         data = self._tev_datas[seg_index]
                     assert data is not None, 'no TEV nor SEV'
@@ -593,4 +601,4 @@ def is_tdtblock(blockpath):
     if file_ext >= tdt_ext:  # if containing all the necessary files
         return True
     else:
-        return False
+        return False