Merge pull request #3958 from RobertoDF/patch-3

keep probe_planar_contour when aggregating channels

Author: samuelgarcia

src/spikeinterface/core/channelsaggregationrecording.py

@@ -105,6 +105,20 @@ def __init__(self, recording_list_or_dict=None, renamed_channel_ids=None, record
                 "Locations are not unique! " "Cannot aggregate recordings!"
             )
 
+        planar_contour_keys = [
+            key for recording in recording_list for key in recording.get_annotation_keys() if "planar_contour" in key
+        ]
+        if len(planar_contour_keys) > 0:
+            if all(
+                k == planar_contour_keys[0] for k in planar_contour_keys
+            ):  # we add the 'planar_contour' annotations only if there is a unique one in the recording_list
+                planar_contour_key = planar_contour_keys[0]
+                collect_planar_contours = []
+                for rec in recording_list:
+                    collect_planar_contours.append(rec.get_annotation(planar_contour_key))
+                if all(np.array_equal(arr, collect_planar_contours[0]) for arr in collect_planar_contours):
+                    self.set_annotation(planar_contour_key, collect_planar_contours[0])
+
         # finally add segments, we need a channel mapping
         ch_id = 0
         channel_map = {}