jgrewe/nixview-python
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[file_handler] additional functions for count checking

Browse Source
This commit is contained in:
Jan Grewe 2021-01-16 18:12:02 +01:00
parent 7d21a9d072
commit 084aec12de
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
nixview/file_handler.py
View File

@ -111,15 +111,26 @@ class FileHandler(metaclass=Singleton):
def filename(self): def filename(self):
return self._filename return self._filename
def valid_count(self, shape, offset, count):
valid_count = np.empty(len(shape), dtype=int)
for i, (o, c) in enumerate(zip(offset, count)):
if o + c > shape[i]:
valid_count[i] = shape[i] - o
return valid_count
def count_is_valid(self, shape, offset, count):
res = True
for s, o, c in zip(shape, offset, count):
res = res and o + c <= s
return res
def request_data(self, entity_descriptor, offset=None, count=None): def request_data(self, entity_descriptor, offset=None, count=None):
entity = self._entity_buffer.get(entity_descriptor.id) entity = self._entity_buffer.get(entity_descriptor.id)
if entity is None: if entity is None:
print("need to do something else") raise ValueError("Entity is invalid: %s" % entity_descriptor)
for i, (o, c) in enumerate(zip(offset, count)): if not self.count_is_valid(entity.shape, offset, count):
if o + c > entity.shape[i]: count = self.valid_count(entity.shape, offset, count)
count[i] = entity.shape[i] - o
seg = tuple([slice(o, o + c) for o, c in zip(offset, count)]) seg = tuple([slice(o, o + c) for o, c in zip(offset, count)])
print(seg)
return entity[seg] return entity[seg]
def request_section_descriptor(self, id): def request_section_descriptor(self, id):