[fileHandler] request functions for tree building

This commit is contained in:
Jan Grewe 2020-12-30 16:52:10 +01:00
parent b569231b47
commit 2016b6dad1
2 changed files with 106 additions and 3 deletions

View File

@ -1,6 +1,18 @@
import os import os
import nixio as nix import nixio as nix
from nixio import file
class ItemDescriptor():
def __init__(self) -> None:
super().__init__()
self.name = None
self.type = None
self.id = None
self.definition = None
self.value = None
self.entity_type = None
class Singleton(type): class Singleton(type):
_instances = {} _instances = {}
@ -13,6 +25,7 @@ class FileHandler(metaclass=Singleton):
def __init__(self) -> None: def __init__(self) -> None:
super().__init__() super().__init__()
self._filename = None
self._nix_file = None self._nix_file = None
self._file_requests = [] self._file_requests = []
@ -28,6 +41,7 @@ class FileHandler(metaclass=Singleton):
return False, "File %s could not be found!" % filename return False, "File %s could not be found!" % filename
try: try:
self._nix_file = nix.File.open(filename, nix.FileMode.ReadOnly) self._nix_file = nix.File.open(filename, nix.FileMode.ReadOnly)
self._filename = filename
return True, "Successfully opened file %s." % filename.split(os.sep)[-1] return True, "Successfully opened file %s." % filename.split(os.sep)[-1]
except RuntimeError as e: except RuntimeError as e:
return False, "Failed to open file %s! \n Error message is: %s" % (filename, e) return False, "Failed to open file %s! \n Error message is: %s" % (filename, e)
@ -41,6 +55,96 @@ class FileHandler(metaclass=Singleton):
self._nix_file = None self._nix_file = None
self._file_requests = [] self._file_requests = []
def file_descriptor(self):
return ItemDescriptor()
@property @property
def is_valid(self): def is_valid(self):
return self._nix_file is not None and self._nix_file.is_open() return self._nix_file is not None and self._nix_file.is_open()
@property
def filename(self):
return self._filename
def request_metadata(self, root_id=None, depth=1):
"""[summary]
Args:
root_id ([type], optional): [description]. Defaults to None.
depth (int, optional): [description]. Defaults to 1.
"""
def get_subsections(section):
sub_sections = []
for s in section.sections:
sub_sections.append({"name": s.name, "id": s.id, "item_type":s.type, "description": s.definition})
return sub_sections
def get_properties(section):
props = []
for p in section.props:
props.append({"name": p.name, "id":p.id, "unit": p.unit})
return props
sections = []
properties = []
if root_id is None:
sections = get_subsections(self._nix_file)
else:
found_section = self._nix_file.find_sections(lambda s: s.id == root_id)
for fs in found_section:
sections.extend(get_subsections(fs))
properties.extend(get_properties(fs))
return sections, properties
def _entity_info(self, entities):
infos = []
for e in entities:
infos.append({"name": e.name, "id": e.id, "item_type": e.type, "description": e.definition})
return infos
def request_blocks(self):
return self._entity_info(self._nix_file.blocks)
def request_data_arrays(self, block_id):
return self._entity_info(self._nix_file.blocks[block_id].data_arrays)
def request_tags(self, block_id):
tags = self._entity_info(self._nix_file.blocks[block_id].tags)
tags.extend(self._entity_info(self._nix_file.blocks[block_id].multi_tags))
return tags
def request_references(self, block_id, tag_id, is_mtag):
b = self._nix_file.blocks[block_id]
t = None
if is_mtag:
t = b.multi_tags[tag_id]
else:
t = b.tags[tag_id]
return self._entity_info(t.references)
def request_features(self, block_id, tag_id, is_mtag):
b = self._nix_file.blocks[block_id]
t = None
if is_mtag:
t = b.multi_tags[tag_id]
else:
t = b.tags[tag_id]
feats = []
for f in t.features:
itd = ItemDescriptor()
itd.name = f.data.name
itd.id = f.id
itd.type = f.type
itd.definition = f.data.definition
itd.entity_type = "nix.Feature"
feats.append(itd)
return feats
def request_dimensions(self, block_id, array_id):
dimensions = []
for i, d in enumerate(self._nix_file.blocks[block_id].data_arrays[array_id].dimensions):
dim_name = "%i. dim: %s" % (i+1, d.label if hasattr(d, "label") else "")
dimensions.append({"name": dim_name, "item_type": d.dimension_type, "id":None, "description": None})
return dimensions

1
nixview.py Normal file → Executable file
View File

@ -1,5 +1,4 @@
#!/usr/bin/python3 #!/usr/bin/python3
import sys import sys
from PyQt5.QtWidgets import QApplication from PyQt5.QtWidgets import QApplication
from main_window import NixView from main_window import NixView