handles new probe naming in latest Open Ephys format

Author: Thinh Nguyen

element_array_ephys/readers/openephys.py

@@ -140,6 +140,21 @@ def load_probe_data(self):
         return probes
 
 
+# For more details on supported probes,
+# see: https://open-ephys.github.io/gui-docs/User-Manual/Plugins/Neuropixels-PXI.html
+_probe_model_name_mapper = {
+    "Neuropix-PXI": "neuropixels 1.0 - 3B",
+    "Neuropix-3a": "neuropixels 1.0 - 3A",
+    "Neuropixels 1.0": "neuropixels 1.0 - 3B",
+    "Neuropixels Ultra": "neuropixels UHD",
+    "Neuropixels Ultra (Switchable)": "neuropixels UHD",
+    "Neuropixels 21": "neuropixels 2.0 - SS",
+    "Neuropixels 24": "neuropixels 2.0 - MS",
+    "Neuropixels 2.0 - Single Shank": "neuropixels 2.0 - SS",
+    "Neuropixels 2.0 - Four Shank": "neuropixels 2.0 - MS"
+}
+
+
 class Probe:
 
     def __init__(self, processor, probe_index=0):
@@ -152,19 +167,13 @@ def __init__(self, processor, probe_index=0):
         if processor['@pluginName'] == 'Neuropix-3a' or 'NP_PROBE' not in processor['EDITOR']:
             self.probe_info = processor['EDITOR']['PROBE'] if isinstance(processor['EDITOR']['PROBE'], dict) else processor['EDITOR']['PROBE'][probe_index]
             self.probe_SN = self.probe_info['@probe_serial_number']
-            self.probe_model = {
-                "Neuropix-PXI": "neuropixels 1.0 - 3B",
-                "Neuropix-3a": "neuropixels 1.0 - 3A"}[processor['@pluginName']]
+            self.probe_model = _probe_model_name_mapper[processor['@pluginName']]
             self._channels_connected = {int(re.search(r'\d+$', k).group()): int(v)
                                         for k, v in self.probe_info.pop('CHANNELSTATUS').items()}
         else:
             self.probe_info = processor['EDITOR']['NP_PROBE'][probe_index]
             self.probe_SN = self.probe_info['@probe_serial_number']
-            self.probe_model = {
-                "Neuropixels 1.0": "neuropixels 1.0 - 3B",
-                "Neuropixels Ultra": "neuropixels UHD",
-                "Neuropixels 21": "neuropixels 2.0 - SS",
-                "Neuropixels 24": "neuropixels 2.0 - MS"}[self.probe_info['@probe_name']]
+            self.probe_model = _probe_model_name_mapper[self.probe_info['@probe_name']]
             self._channels_connected = {int(re.search(r'\d+$', k).group()): 1
                                         for k in self.probe_info.pop('CHANNELS')}