add additional meta data to channel directly

Author: danielparthier

ephys/classes/channels.py

@@ -70,6 +70,8 @@ def __init__(self, sweep_count: int, sweep_length: int, unit: str):
         self.sampling_rate: Quantity = Quantity(0, "Hz")
         self.starting_time: Quantity = Quantity(np.zeros(self.sweep_count), "s")
         self.rec_datetime: datetime.datetime | None = None
+        self.group: int | None = None
+        self.recording_type: str | None = None
         self.check_clamp()
 
     def insert_data(self, data: np.ndarray, sweep_count: int) -> None:
@@ -135,6 +137,34 @@ def channel_average(self, sweep_subset: Any = None) -> None:
         sweep_subset = _get_sweep_subset(self.data, sweep_subset)
         self.average = ChannelAverage(self, sweep_subset)
 
+    def add_group(self, group: int) -> None:
+        """
+        Assigns a group number to the channel.
+
+        Parameters:
+            group (int): The group number to be assigned to the channel.
+
+        Returns:
+            None
+        """
+        self.group = group
+
+    def add_recording_type(self, recording_type: str) -> None:
+        """
+        Assigns a recording type to the channel.
+
+        Parameters:
+            recording_type (str): The recording type to be assigned to the channel.
+
+        Returns:
+            None
+        """
+        if recording_type not in ["field", "cell"]:
+            print(
+                "Warning: Unknown recording type. Please use 'field' or 'cell'. Invalid type might cause issues later on."
+            )
+        self.recording_type = recording_type
+
 
 class ChannelInformation:
     """

ephys/classes/class_functions.py

@@ -63,6 +63,14 @@ def wcp_trace(trace, file_path: str, quick_check: bool = True) -> None:
             trace_insert.sampling_rate = trace.sampling_rate
         else:
             raise ValueError("Signal type not recognized")
+
+        trace_insert.add_group(
+            trace.channel_information.channel_grouping[channel_index]
+        )
+        trace_insert.add_recording_type(
+            trace.channel_information.recording_type[channel_index]
+        )
+
         for segment_index, segment in enumerate(data_block.segments):
             # handle channel mismatch (channels are loaded into the first channel with same label)
             # TODO: check if it can be fixed on neo side or if it is only specific file scenarios
@@ -139,6 +147,14 @@ def abf_trace(trace, file_path: str, quick_check: bool = True) -> None:
             trace_insert.sampling_rate = trace.sampling_rate
         else:
             raise ValueError("Signal type not recognized")
+
+        trace_insert.add_group(
+            trace.channel_information.channel_grouping[channel_index]
+        )
+        trace_insert.add_recording_type(
+            trace.channel_information.recording_type[channel_index]
+        )
+
         for segment_index, segment in enumerate(data_block.segments):
             trace_insert.insert_data(
                 segment.analogsignals[channel_index], segment_index