X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=code%2Fnetwork-generator%2Fmodel%2Fpython%2Ftower.py;h=7def5e6d4c550887db555a86ee39f40ee7fc33b1;hb=527f95ae29a40bfc8535d8b2177d9ca976c48b81;hp=869b3f5d6eb034b5e3f104dfb83be9fa9b7df557;hpb=809c7bfdf19790349202dd42496758d0fe0f4782;p=oam.git

diff --git a/code/network-generator/model/python/tower.py b/code/network-generator/model/python/tower.py
index 869b3f5..7def5e6 100644
--- a/code/network-generator/model/python/tower.py
+++ b/code/network-generator/model/python/tower.py
@@ -15,50 +15,80 @@
 #!/usr/bin/python
 
 """
-A Class representing a Tower to mount O-RAN RUx
+A Class representing a Tower to mount O-RAN RUs
+It can be interpreted as 'resource pool' for physical network
+functions.
 """
-from typing import Any, Dict
-
-import model.python.hexagon as Hexagon
-from model.python.hexagon import Hex, Layout, Point
-from model.python.geo_location import GeoLocation
-
 from model.python.o_ran_object import IORanObject
+from model.python.o_ran_ru import ORanRu
 from model.python.o_ran_node import ORanNode
 import xml.etree.ElementTree as ET
 
 
-# Define the "ITower" interface
+# Define the "IORanDu" interface
 class ITower(IORanObject):
-    def __init__(self, layout: Layout, hex: Hex = None, **kwargs):
+    def __init__(self, o_ran_ru_count: int, **kwargs):
         super().__init__(**kwargs)
+        self._o_ran_ru_count = o_ran_ru_count
 
 
-# Define an abstract O-RAN Node class
-class Tower(ORanNode, ITower):
+# Implement a concrete O-RAN Node class
+class Tower(ORanNode):
     def __init__(self, tower_data: ITower = None, **kwargs):
         super().__init__(tower_data, **kwargs)
+        self._o_ran_ru_count = (
+            tower_data["oRanRuCount"]
+            if tower_data and "oRanRuCount" in tower_data
+            else 3
+        )
+        self._o_ran_rus: list[ORanRu] = self._create_o_ran_rus()
+
+    def _create_o_ran_rus(self) -> list[ORanRu]:
+        result: list[ORanRu] = []
+        for index in range(self._o_ran_ru_count):
+            s: str = "00" + str(index)
+            name: str = "-".join(
+                [self.name.replace("Tower", "RU"), s[len(s) - 2 : len(s)]]
+            )
+            cell_count: int = self.parent.parent.parent.parent.parent.configuration()[
+                "pattern"
+            ]["o-ran-ru"]["nr-cell-du-count"]
+            cell_angle : int = self.parent.parent.parent.parent.parent.configuration()[
+                "pattern"
+            ]["nr-cell-du"]["cell-angle"]
+            ru_angle: int = cell_count * cell_angle
+            ru_azimuth: int = index * ru_angle
+            result.append(
+                ORanRu(
+                    {
+                        "name": name,
+                        "geoLocation": self.geoLocation,
+                        "position": self.position,
+                        "layout": self.layout,
+                        "spiralRadiusProfile": self.spiralRadiusProfile,
+                        "parent": self,
+                        "cellCount": cell_count,
+                        "ruAngle": ru_angle,
+                        "ruAzimuth": ru_azimuth,
+                    }
+                )
+            )
+        return result
+
+    @property
+    def o_ran_rus(self) -> list[ORanRu]:
+        return self._o_ran_rus
 
-    def toKml(self):
-        placemark: ET.Element = ET.Element("Placemark")
-        name: ET.Element = ET.SubElement(placemark, "name")
+    def toKml(self) -> ET.Element:
+        tower: ET.Element = ET.Element("Folder")
+        open: ET.Element = ET.SubElement(tower, "open")
+        open.text = "1"
+        name: ET.Element = ET.SubElement(tower, "name")
         name.text = self.name
-        style: ET.Element = ET.SubElement(placemark, "styleUrl")
-        style.text = "#" + self.__class__.__name__
-        polygon: ET.Element = ET.SubElement(placemark, "Polygon")
-        outer_boundary: ET.Element = ET.SubElement(polygon, "outerBoundaryIs")
-        linear_ring: ET.Element = ET.SubElement(outer_boundary, "LinearRing")
-        coordinates: ET.Element = ET.SubElement(linear_ring, "coordinates")
-        points: list[Point] = Hexagon.polygon_corners(self.layout, self.position)
-        points.append(points[0])
+        for o_ran_ru in self.o_ran_rus:
+            tower.append(o_ran_ru.toKml())
+        return tower
 
-        method = GeoLocation(self.geoLocation).point_to_geo_location
-        geo_locations: list[GeoLocation] = map(method, points)
-        text:list[str] = []
-        for geo_location in list(geo_locations):
-            text.append(f"{geo_location.longitude},{geo_location.latitude},{geo_location.aboveMeanSeaLevel}")
-        coordinates.text = " ".join(text)
-        return placemark
 
-    def toSvg(self):
+    def toSvg(self) -> None:
         return None