Load event data only on demand and when not already cached

Author: JuliaSprenger

neo/rawio/plexon2rawio/plexon2rawio.py

@@ -176,7 +176,6 @@ def _parse_header(self):
             # event channels are characterized by (name, id, type), with type in ['event', 'epoch']
             channel_name = echannel_info.m_Name.decode()
             event_channels.append((channel_name, echannel_info.m_Channel, 'event'))
-            self._event_channel_cache[channel_name] = self.pl2reader.pl2_get_digital_channel_data_by_name(channel_name)
 
         event_channels = np.array(event_channels, dtype=_event_channel_dtype)
 
@@ -452,6 +451,10 @@ def _event_count(self, block_index, seg_index, event_channel_index):
         channel_header = self.header['event_channels'][event_channel_index]
         channel_name = channel_header['name']
 
+        # loading event channel data on demand when not already cached
+        if channel_name not in self._event_channel_cache:
+            self._event_channel_cache[channel_name] = self.pl2reader.pl2_get_digital_channel_data_by_name(channel_name)
+
         event_times, values = self._event_channel_cache[channel_name]
 
         return len(event_times)
@@ -465,6 +468,10 @@ def _get_event_timestamps(self, block_index, seg_index, event_channel_index, t_s
         channel_header = self.header['event_channels'][event_channel_index]
         channel_name = channel_header['name']
 
+        # loading event channel data on demand when not already cached
+        if channel_name not in self._event_channel_cache:
+            self._event_channel_cache[channel_name] = self.pl2reader.pl2_get_digital_channel_data_by_name(channel_name)
+
         event_times, labels = self._event_channel_cache[channel_name]
         labels = np.asarray(labels, dtype='U')