Create a script to generate a Topology

- add tapi node generation for SMO

Issue-ID: OAM-249
Change-Id: I8f5c7b0b5aa71b6e1ba7b163e474d364221ae2de
Signed-off-by: demx8as6 <martin.skorupski@highstreet-technologies.com>

diff --git a/code/network-topology-instance-generator/model/python/tapi_node_smo.py b/code/network-topology-instance-generator/model/python/tapi_node_smo.py
new file mode 100644 (file)
index 0000000..87e0d07
--- /dev/null
+++ b/code/network-topology-instance-generator/model/python/tapi_node_smo.py
@@ -0,0 +1,165 @@
+# Copyright 2022 highstreet technologies GmbH
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+#!/usr/bin/python
+"""
+Module containing a class representing a SMO as TAPI Node.
+"""
+from typing import List
+from lxml import etree
+from model.python.tapi_node import TapiNode
+from model.python.tapi_node_edge_point import TapiNodeEdgePoint
+
+
+class TapiNodeSmo(TapiNode):
+    """
+    Class representing a SMO as TAPI Node
+    """
+    __width: 0
+
+    # constructor
+    def __init__(self, parent: TapiNode, config):
+        super().__init__(parent, config)
+
+        # add O2 consumer interface
+        nep_configuration = {
+            "parent": self.identifier(),
+            "nodeEdgePoint": {
+                "interface": "o2", "cep": [{"protocol": "REST", "role": "consumer"}]
+            }
+        }
+        self.add(TapiNodeEdgePoint(nep_configuration))
+
+        # add A1 consumer interface
+        nep_configuration = {
+            "parent": self.identifier(),
+            "nodeEdgePoint": {
+                "interface": "a1", "cep": [{"protocol": "REST", "role": "consumer"}]
+            }
+        }
+        self.add(TapiNodeEdgePoint(nep_configuration))
+
+        # add O1/OAM NetConf Consumer interface
+        nep_configuration = {
+            "parent": self.identifier(),
+            "nodeEdgePoint": {
+                "interface": "oam", "cep": [{"protocol": "NETCONF", "role": "consumer"}]
+            }
+        }
+        self.add(TapiNodeEdgePoint(nep_configuration))
+
+        # add O1 VES Provider interface
+        nep_configuration = {
+            "parent": self.identifier(),
+            "nodeEdgePoint": {
+                "interface": "o1", "cep": [{"protocol": "VES", "role": "provider"}]
+            }
+        }
+        self.add(TapiNodeEdgePoint(nep_configuration))
+
+        # add O1 File Transfer Consumer interface
+        nep_configuration = {
+            "parent": self.identifier(),
+            "nodeEdgePoint": {
+                "interface": "o1", "cep": [{"protocol": "FILE", "role": "consumer"}]
+            }
+        }
+        self.add(TapiNodeEdgePoint(nep_configuration))
+
+    def __smo_component(self, x: int, y: int, label: str) -> etree.Element:
+        group = etree.Element("g")
+        group.attrib["class"] = " ".join(["node", label])
+
+        width = (2 + 2) * (2.2*self.FONTSIZE)
+        height = 2 * (2*self.FONTSIZE)
+
+        rect = etree.Element("rect")
+        rect.attrib["x"] = str(int(x - width/2))
+        rect.attrib["y"] = str(int(y - height/2))
+        rect.attrib["width"] = str(int(width))
+        rect.attrib["height"] = str(int(height))
+        rect.attrib["rx"] = str(self.FONTSIZE)
+        rect.attrib["class"] = " ".join(["node", label])
+        group.append(rect)
+
+        labelElement = etree.Element('text')
+        labelElement.attrib['x'] = str(x)
+        # +4px for font-size 14px (think of chars like 'gjy')
+        labelElement.attrib['y'] = str(y + 4)
+        labelElement.attrib['class'] = " ".join(["node", label])
+        labelElement.text = label.upper()
+        group.append(labelElement)
+        return group
+
+    def svg(self, x: int, y: int) -> etree.Element:
+        """
+        Getter for a xml Element object representing the TAPI Node.
+        :return TAPI Node as svg object.
+        """
+        super().svg(x, y)
+
+        components = ["o2-controller", "non-rt-ric", "oam-controller",
+                      "ves-collector", "file-server"]
+
+        group = etree.Element("g")
+        group.attrib["class"] = "node"
+        title = etree.Element("title")
+        title.text = "\n TAPI Node\n id: " + \
+            self.identifier() + "\n name: " + self.name()
+        group.append(title)
+
+        width = (len(components)*5 +1) * (2.2*self.FONTSIZE)
+        height = 2 * (2.2*self.FONTSIZE)
+
+        rect = etree.Element("rect")
+        rect.attrib["x"] = str(int(x - width/2))
+        rect.attrib["y"] = str(int(y - height/2))
+        rect.attrib["width"] = str(int(width))
+        rect.attrib["height"] = str(int(height))
+        rect.attrib["rx"] = str(self.FONTSIZE)
+        rect.attrib["class"] = " ".join(
+            ["node", self.function_label().lower()])
+        group.append(rect)
+
+        label = etree.Element('text')
+        label.attrib['x'] = str(x)
+        # +4px for font-size 14px (think of chars like 'gjy')
+        label.attrib['y'] = str(y + 4)
+        label.attrib['class'] = " ".join(
+            ["node", self.function_label().lower()])
+        label.text = self.function_label()
+        group.append(label)
+
+        for component in components:
+            x_mapping = {
+                "o2-controller": -4*6*self.FONTSIZE,
+                "non-rt-ric": -2*6*self.FONTSIZE,
+                "oam-controller": -0*6*self.FONTSIZE,
+                "ves-collector": +2*6*self.FONTSIZE,
+                "file-server": +4*6*self.FONTSIZE
+            }
+            comp_x = x + x_mapping[component]
+            comp_y = y-0*self.FONTSIZE
+            group.append(self.__smo_component(comp_x, comp_y, component))
+
+        for nep in self.data()['owned-node-edge-point']:
+            nep_x = x + \
+                super().x_offset_by_cep_name(
+                    nep.connection_edge_points()[0].name(), 0)
+            nep_y = y + \
+                super().y_offset_by_cep_name(
+                    nep.connection_edge_points()[0].name())
+            group.append(nep.svg(nep_x, nep_y))
+
+        return group