test #133

Author: svandenb-dev

src/pyedb/grpc/database/stackup.py

@@ -47,6 +47,7 @@
 )
 from ansys.edb.core.layer.layer_collection import LayerCollection as GrpcLayerCollection
 from ansys.edb.core.layer.layer_collection import LayerTypeSet as GrpcLayerTypeSet
+from ansys.edb.core.layer.stackup_layer import StackupLayer as GrpcStackupLayer
 from ansys.edb.core.layout.mcad_model import McadModel as GrpcMcadModel
 from ansys.edb.core.utility.transform3d import Transform3D as GrpcTransform3D
 from ansys.edb.core.utility.value import Value as GrpcValue
@@ -80,6 +81,7 @@
 class LayerCollection(GrpcLayerCollection):
     def __init__(self, pedb, edb_object):
         super().__init__(edb_object.msg)
+        self._layer_collection = edb_object
         self._pedb = pedb
 
     def update_layout(self):
@@ -106,12 +108,17 @@ def add_layer_top(self, name, layer_type="signal", **kwargs):
         -------
 
         """
-        added_layer = self.add_layer_top(name)
-        added_layer.type = GrpcLayerType.SIGNAL_LAYER
+        thickness = GrpcValue(0.0)
+        if "thickness" in kwargs:
+            thickness = GrpcValue(kwargs["thickness"])
+        elevation = GrpcValue(0.0)
+        _layer_type = GrpcLayerType.SIGNAL_LAYER
         if layer_type.lower() == "dielectric":
-            added_layer.type = GrpcLayerType.DIELECTRIC_LAYER
-        added_layer.name = name
-        return added_layer
+            _layer_type = GrpcLayerType.DIELECTRIC_LAYER
+        layer = GrpcStackupLayer.create(
+            name=name, layer_type=_layer_type, thickness=thickness, material="copper", elevation=elevation
+        )
+        return self._layer_collection.add_layer_top(layer)
 
     def add_layer_bottom(self, name, layer_type="signal", **kwargs):
         """Add a layer on bottom of the stackup.
@@ -128,12 +135,17 @@ def add_layer_bottom(self, name, layer_type="signal", **kwargs):
         -------
 
         """
-        added_layer = self.add_layer_bottom(name)
-        added_layer.type = GrpcLayerType.SIGNAL_LAYER
+        thickness = GrpcValue(0.0)
+        if "thickness" in kwargs:
+            thickness = GrpcValue(kwargs["thickness"])
+        elevation = GrpcValue(0.0)
+        _layer_type = GrpcLayerType.SIGNAL_LAYER
         if layer_type.lower() == "dielectric":
-            added_layer.type = GrpcLayerType.DIELECTRIC_LAYER
-        added_layer.name = name
-        return added_layer
+            _layer_type = GrpcLayerType.DIELECTRIC_LAYER
+        layer = GrpcStackupLayer.create(
+            name=name, layer_type=_layer_type, thickness=thickness, material="copper", elevation=elevation
+        )
+        return self._layer_collection.add_layer_bottom(layer)
 
     def add_layer_below(self, name, base_layer_name, layer_type="signal", **kwargs):
         """Add a layer below a layer.
@@ -152,12 +164,17 @@ def add_layer_below(self, name, base_layer_name, layer_type="signal", **kwargs):
         -------
 
         """
-        added_layer = self.add_layer_below(name, base_layer_name)
-        added_layer.type = GrpcLayerType.SIGNAL_LAYER
+        thickness = GrpcValue(0.0)
+        if "thickness" in kwargs:
+            thickness = GrpcValue(kwargs["thickness"])
+        elevation = GrpcValue(0.0)
+        _layer_type = GrpcLayerType.SIGNAL_LAYER
         if layer_type.lower() == "dielectric":
-            added_layer.type = GrpcLayerType.DIELECTRIC_LAYER
-        added_layer.name = name
-        return added_layer
+            _layer_type = GrpcLayerType.DIELECTRIC_LAYER
+        layer = GrpcStackupLayer.create(
+            name=name, layer_type=_layer_type, thickness=thickness, material="copper", elevation=elevation
+        )
+        return self._layer_collection.add_layer_below(layer, base_layer_name)
 
     def add_layer_above(self, name, base_layer_name, layer_type="signal", **kwargs):
         """Add a layer above a layer.
@@ -176,12 +193,17 @@ def add_layer_above(self, name, base_layer_name, layer_type="signal", **kwargs):
         -------
 
         """
-        added_layer = self.add_layer_above(name, base_layer_name)
-        added_layer.type = GrpcLayerType.SIGNAL_LAYER
+        thickness = GrpcValue(0.0)
+        if "thickness" in kwargs:
+            thickness = GrpcValue(kwargs["thickness"])
+        elevation = GrpcValue(0.0)
+        _layer_type = GrpcLayerType.SIGNAL_LAYER
         if layer_type.lower() == "dielectric":
-            added_layer.type = GrpcLayerType.DIELECTRIC_LAYER
-        added_layer.name = name
-        return added_layer
+            _layer_type = GrpcLayerType.DIELECTRIC_LAYER
+        layer = GrpcStackupLayer.create(
+            name=name, layer_type=_layer_type, thickness=thickness, material="copper", elevation=elevation
+        )
+        return self._layer_collection.add_layer_above(layer, base_layer_name)
 
     def add_document_layer(self, name, layer_type="user", **kwargs):
         """Add a document layer.

tests/grpc/system/test_edb_stackup.py

@@ -1117,3 +1117,4 @@ def test_19(self, edb_examples):
         base_layer = edbapp.stackup.layers["1_Top"]
         l_id = edbapp.stackup.layers_by_id.index([base_layer.id, base_layer.name])
         assert edbapp.stackup.layers_by_id[l_id - 1][1] == "add_layer_above"
+        edbapp.close()