Skip to content

Commit 5ab5c44

Browse files
authored
Fixed .csv header for SCIENTISST_V2 mode (#12)
1 parent 3f3e09c commit 5ab5c44

File tree

1 file changed

+27
-12
lines changed

1 file changed

+27
-12
lines changed

sense_src/file_writer.py

Lines changed: 27 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -42,18 +42,33 @@ def __init_file(
4242

4343
def __get_metadata(self, address, fs, channels, api_version, firmware_version, api_com_version):
4444
timestamp = datetime.now()
45-
metadata = {
46-
"API version": api_version,
47-
"Channels": channels,
48-
"Channels labels": get_channel_labels(channels, self.mv),
49-
"Device": address,
50-
"Firmware version": firmware_version,
51-
"Header": get_header(channels, self.mv, api_com_version),
52-
"Resolution (bits)": [4, 1, 1, 1, 1] + self.__get_channel_resolutions(),
53-
"Sampling rate (Hz)": fs,
54-
"Timestamp": timestamp.timestamp(),
55-
"ISO 8601": timestamp.isoformat(),
56-
}
45+
metadata = None
46+
if API_MODE_DICT[api_com_version] == API_MODE_SCIENTISST_V2:
47+
metadata = {
48+
"API version": api_version,
49+
"Channels": channels,
50+
"Channels labels": get_channel_labels(channels, self.mv),
51+
"Device": address,
52+
"Firmware version": firmware_version,
53+
"Header": get_header(channels, self.mv, api_com_version),
54+
"Resolution (bits)": [36, 1, 1, 1, 1] + self.__get_channel_resolutions(),
55+
"Sampling rate (Hz)": fs,
56+
"Timestamp": timestamp.timestamp(),
57+
"ISO 8601": timestamp.isoformat(),
58+
}
59+
else:
60+
metadata = {
61+
"API version": api_version,
62+
"Channels": channels,
63+
"Channels labels": get_channel_labels(channels, self.mv),
64+
"Device": address,
65+
"Firmware version": firmware_version,
66+
"Header": get_header(channels, self.mv, api_com_version),
67+
"Resolution (bits)": [4, 1, 1, 1, 1] + self.__get_channel_resolutions(),
68+
"Sampling rate (Hz)": fs,
69+
"Timestamp": timestamp.timestamp(),
70+
"ISO 8601": timestamp.isoformat(),
71+
}
5772
if self.mv:
5873
metadata["Channels indexes raw"] = list(
5974
map(lambda x: (x - 1) * 2 + 5, channels)

0 commit comments

Comments
 (0)