fix: fix parsing of unsigned electrical broadband data

sciencecorp · Oct 16, 2024 · 731492e · 731492e
1 parent 3ebca1a
commit 731492e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/synapse/utils/ndtp_types.py b/synapse/utils/ndtp_types.py
@@ -53,12 +53,14 @@ def pack(self, seq_number: int):
 
     @staticmethod
     def from_ndtp_message(msg: NDTPMessage):
+        dtype = np.int16 if msg.payload.is_signed else np.uint16
         return ElectricalBroadbandData(
             t0=msg.header.timestamp,
             bit_width=msg.payload.bit_width,
+            is_signed=msg.payload.is_signed,
             sample_rate=msg.payload.sample_rate,
             samples=[
-                (ch.channel_id, np.array(ch.channel_data, dtype=np.int16))
+                (ch.channel_id, np.array(ch.channel_data, dtype=dtype))
                 for ch in msg.payload.channels
             ],
         )