Merge pull request #400 from alejoe91/add-channel-name-to-open-ephys

Add `channel_name` contact annotation for open ephys

Author: alejoe91

src/probeinterface/neuropixels_tools.py

@@ -1221,7 +1221,6 @@ def read_openephys(
                 "probe": sliced_probe,
             }
         else:
-
             channel_names = np.array(list(channels.attrib.keys()))
             channel_ids = np.array([int(ch[2:]) for ch in channel_names])
             channel_order = np.argsort(channel_ids)
@@ -1307,6 +1306,7 @@ def read_openephys(
             np_probe_dict = {
                 "shank_ids": shank_ids,
                 "elec_ids": elec_ids,
+                "channel_names": channel_names,
                 "pt_metadata": pt_metadata,
                 "slot": slot,
                 "port": port,
@@ -1429,6 +1429,9 @@ def read_openephys(
             pt_metadata, probe_part_number, elec_ids, shank_ids=shank_ids, mux_info=mux_info
         )
 
+    if "channel_names" in np_probe_info:
+        probe.annotate_contacts(channel_name=np_probe_info["channel_names"])
+
     chans_saved = get_saved_channel_indices_from_openephys_settings(settings_file, stream_name=stream_name)
     if chans_saved is not None:
         probe = probe.get_slice(chans_saved)