From 2016b6dad103b9bf3e7fb848da5c739d5092b5ae Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Wed, 30 Dec 2020 16:52:10 +0100
Subject: [PATCH] [fileHandler] request functions for tree building

---
 file_handler.py | 108 +++++++++++++++++++++++++++++++++++++++++++++++-
 nixview.py      |   1 -
 2 files changed, 106 insertions(+), 3 deletions(-)
 mode change 100644 => 100755 nixview.py

diff --git a/file_handler.py b/file_handler.py
index b251ed2..0e6ac68 100644
--- a/file_handler.py
+++ b/file_handler.py
@@ -1,7 +1,19 @@
 import os
 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):
     _instances = {}
     def __call__(cls, *args, **kwargs):
@@ -13,6 +25,7 @@ class FileHandler(metaclass=Singleton):
     
     def __init__(self) -> None:
         super().__init__()
+        self._filename = None
         self._nix_file = None
         self._file_requests = []
     
@@ -28,6 +41,7 @@ class FileHandler(metaclass=Singleton):
             return False, "File %s could not be found!" % filename
         try:
             self._nix_file = nix.File.open(filename, nix.FileMode.ReadOnly)
+            self._filename = filename
             return True, "Successfully opened file %s." % filename.split(os.sep)[-1]
         except RuntimeError as 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._file_requests = []
     
+    def file_descriptor(self):
+        
+        return ItemDescriptor()
+    
+   
     @property
     def is_valid(self):
-        return self._nix_file is not None and self._nix_file.is_open()
\ No newline at end of file
+        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
\ No newline at end of file
diff --git a/nixview.py b/nixview.py
old mode 100644
new mode 100755
index 2fe1d2b..6ecde7e
--- a/nixview.py
+++ b/nixview.py
@@ -1,5 +1,4 @@
 #!/usr/bin/python3
-
 import sys
 from PyQt5.QtWidgets import QApplication
 from main_window import NixView