Skip to content

Black formatting #1881

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
zm711 merged 1 commit into from
Jul 21, 2026
+49 −43
Merged

Black formatting #1881

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
92 changes: 49 additions & 43 deletions neo/rawio/blackrockrawio.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -650,8 +650,7 @@ def _get_signal_t_start(self, block_index, seg_index, stream_index):
return self._sigs_t_starts[nsx_nb][seg_index]

@staticmethod
def _create_mmap_view(fid, dtype, offset, num_samples, num_channels=None,
packet_size=None, item_offset=0):
def _create_mmap_view(fid, dtype, offset, num_samples, num_channels=None, packet_size=None, item_offset=0):
"""
Create an np.ndarray view over a raw mmap buffer from an open file.

Expand Down Expand Up @@ -1271,15 +1270,17 @@ def _parse_nsx_data_v22_v30(self, spec, nsx_nb):
data_offset = current_offset + header.dtype.itemsize
timestamp = header["timestamp"]

parsed_data_headers.append({
"timestamp": timestamp,
"memmap_kwargs": {
"filename": filename,
"dtype": "int16",
"offset": data_offset,
"num_samples": num_samples,
},
})
parsed_data_headers.append(
{
"timestamp": timestamp,
"memmap_kwargs": {
"filename": filename,
"dtype": "int16",
"offset": data_offset,
"num_samples": num_samples,
},
}
)

# Jump to next block
data_size_bytes = num_samples * channels * 2 # int16 = 2 bytes
Expand Down Expand Up @@ -1432,13 +1433,15 @@ def _segment_nsx_data(self, parsed_data_headers, nsx_nb):
# Case 1: Multiple blocks (Standard format) - each block is a segment
if len(parsed_data_headers) > 1:
for block_info in parsed_data_headers:
segments.append({
"timestamp": block_info["timestamp"],
"nb_data_points": block_info["memmap_kwargs"]["num_samples"],
"header": 1, # Standard format has headers
"offset_to_data_block": None, # Not needed
"memmap_kwargs": block_info["memmap_kwargs"],
})
segments.append(
{
"timestamp": block_info["timestamp"],
"nb_data_points": block_info["memmap_kwargs"]["num_samples"],
"header": 1, # Standard format has headers
"offset_to_data_block": None, # Not needed
"memmap_kwargs": block_info["memmap_kwargs"],
}
)

# Case 2: Single block - check if PTP (has ptp_timestamps_memmap_kwargs) or simple
elif len(parsed_data_headers) == 1:
Expand Down Expand Up @@ -1510,32 +1513,36 @@ def _segment_nsx_data(self, parsed_data_headers, nsx_nb):
seg_samples_offset = base_samples_offset + int(start) * packet_size
seg_ts_offset = base_ts_offset + int(start) * packet_size

segments.append({
"timestamp": None, # PTP timestamps read on demand
"nb_data_points": seg_num_samples,
"header": None, # PTP has no headers
"offset_to_data_block": None,
"memmap_kwargs": {
**samples_kw,
"offset": seg_samples_offset,
"num_samples": seg_num_samples,
},
"timestamps_memmap_kwargs": {
**ts_kw,
"offset": seg_ts_offset,
"num_samples": seg_num_samples,
},
})
segments.append(
{
"timestamp": None, # PTP timestamps read on demand
"nb_data_points": seg_num_samples,
"header": None, # PTP has no headers
"offset_to_data_block": None,
"memmap_kwargs": {
**samples_kw,
"offset": seg_samples_offset,
"num_samples": seg_num_samples,
},
"timestamps_memmap_kwargs": {
**ts_kw,
"offset": seg_ts_offset,
"num_samples": seg_num_samples,
},
}
)

# V2.1 or single block standard format: no segmentation needed
else:
segments.append({
"timestamp": block_info["timestamp"],
"nb_data_points": block_info["memmap_kwargs"]["num_samples"],
"header": None,
"offset_to_data_block": None,
"memmap_kwargs": block_info["memmap_kwargs"],
})
segments.append(
{
"timestamp": block_info["timestamp"],
"nb_data_points": block_info["memmap_kwargs"]["num_samples"],
"header": None,
"offset_to_data_block": None,
"memmap_kwargs": block_info["memmap_kwargs"],
}
)

return segments

Expand Down Expand Up @@ -1820,8 +1827,7 @@ def _match_nsx_and_nev_segment_ids(self, nsx_nb):
# Nonempty segments are those containing at least 2 samples
# These have to be able to be mapped to nev
nonempty_nsx_segment_indices = [
seg_index for seg_index, seg in enumerate(self._nsx_data_header[nsx_nb])
if seg["nb_data_points"] > 1
seg_index for seg_index, seg in enumerate(self._nsx_data_header[nsx_nb]) if seg["nb_data_points"] > 1
]
nonempty_nsx_segments = [
self._nsx_data_header[nsx_nb][seg_index] for seg_index in nonempty_nsx_segment_indices
Expand Down
Loading