pyedb 0.42.0__tar.gz → 0.44.0__tar.gz

{pyedb-0.42.0 → pyedb-0.44.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pyedb
-Version: 0.42.0
+Version: 0.44.0
 Summary: Higher-Level Pythonic Ansys Electronics Data Base
 Author-email: "ANSYS, Inc." <pyansys.core@ansys.com>
 Maintainer-email: PyEDB developers <simon.vandenbrouck@ansys.com>

{pyedb-0.42.0 → pyedb-0.44.0}/src/pyedb/__init__.py

@@ -44,7 +44,7 @@ deprecation_warning()
 #
 
 pyedb_path = os.path.dirname(__file__)
-__version__ = "0.42.0"
+__version__ = "0.44.0"
 version = __version__
 
 #

pyedb-0.44.0/src/pyedb/configuration/cfg_boundaries.py

@@ -0,0 +1,187 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from pyedb.configuration.cfg_common import CfgBase
+
+
+class CfgBoundaries(CfgBase):
+    class Grpc:
+        @property
+        def pyedb_obj(self):
+            return self.parent.pyedb_obj
+
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent.pedb
+            self.open_region = self.parent.boundary_data.get("open_region", None)
+            self.open_region_type = self.parent.boundary_data.get("map_open_region_type", None)
+            self.pml_visible = self.parent.boundary_data.get("pml_visible", None)
+            self.pml_operation_frequency = self.parent.boundary_data.get("pml_operation_frequency", None)
+            self.pml_radiation_factor = self.parent.boundary_data.get("pml_radiation_factor", None)
+            self.dielectric_extent_type = self.parent.boundary_data.get("dielectric_extent_type", None)
+            self.horizontal_padding = self.parent.boundary_data.get("horizontal_padding", None)
+            self.honor_primitives_on_dielectric_layers = self.parent.boundary_data.get(
+                "honor_primitives_on_dielectric_layers", False
+            )
+            self.air_box_extent_type = self.parent.boundary_data.get("air_box_extent_type", None)
+            self.air_box_base_polygon = self.parent.boundary_data.get("air_box_base_polygon", None)
+            self.air_box_truncate_model_ground_layers = self.parent.boundary_data.get(
+                "air_box_truncate_model_ground_layers", None
+            )
+            self.air_box_horizontal_padding = self.parent.boundary_data.get("air_box_horizontal_padding", None)
+            self.air_box_positive_vertical_padding = self.parent.boundary_data.get(
+                "air_box_positive_vertical_padding", None
+            )
+            self.air_box_negative_vertical_padding = self.parent.boundary_data.get(
+                "air_box_negative_vertical_padding", None
+            )
+
+        def set_parameters_to_edb(self):
+            if self.open_region is not None:
+                self._pedb.hfss.hfss_extent_info.use_open_region = self.open_region
+            if self.open_region_type:
+                self._pedb.hfss.hfss_extent_info.open_region_type = self.open_region_type.lower()
+            if self.pml_visible is not None:
+                self._pedb.hfss.hfss_extent_info.is_pml_visible = self.pml_visible
+            if self.pml_operation_frequency:
+                self._pedb.hfss.hfss_extent_info.operating_freq = self.pml_operation_frequency
+            if self.pml_radiation_factor:
+                self._pedb.hfss.hfss_extent_info.pml_radiation_factor = self.pml_radiation_factor
+            if self.dielectric_extent_type:
+                self._pedb.hfss.hfss_extent_info.extent_type = self.dielectric_extent_type.lower()
+            if self.horizontal_padding:
+                self._pedb.hfss.hfss_extent_info.dielectric_extent_size = float(self.horizontal_padding)
+            if self.honor_primitives_on_dielectric_layers is not None:
+                self._pedb.hfss.hfss_extent_info.honor_user_dielectric = self.honor_primitives_on_dielectric_layers
+            if self.air_box_extent_type:
+                self._pedb.hfss.hfss_extent_info.extent_type = self.air_box_extent_type.lower()
+            if self.air_box_truncate_model_ground_layers is not None:
+                self._pedb.hfss.hfss_extent_info.truncate_air_box_at_ground = self.air_box_truncate_model_ground_layers
+            if self.air_box_horizontal_padding:
+                self._pedb.hfss.hfss_extent_info.air_box_horizontal_extent = float(self.air_box_horizontal_padding)
+            if self.air_box_positive_vertical_padding:
+                self._pedb.hfss.hfss_extent_info.parentair_box_positive_vertical_extent = float(
+                    self.air_box_positive_vertical_padding
+                )
+            if self.air_box_negative_vertical_padding:
+                self._pedb.hfss.hfss_extent_info.air_box_negative_vertical_extent = float(
+                    self.air_box_negative_vertical_padding
+                )
+
+        def get_parameters_from_edb(self):
+            self.parent.open_region = self._pedb.hfss.hfss_extent_info.use_open_region
+            self.parent.open_region_type = self._pedb.hfss.hfss_extent_info.open_region_type
+            self.parent.pml_visible = self._pedb.hfss.hfss_extent_info.is_pml_visible
+            self.parent.pml_operation_frequency = self._pedb.hfss.hfss_extent_info.operating_freq
+            self.parent.pml_radiation_factor = self._pedb.hfss.hfss_extent_info.pml_radiation_factor
+            self.parent.dielectric_extent_type = self._pedb.hfss.hfss_extent_info.extent_type
+            self.parent.horizontal_padding = self._pedb.hfss.hfss_extent_info.dielectric_extent_size
+            self.parent.honor_primitives_on_dielectric_layers = self._pedb.hfss.hfss_extent_info.honor_user_dielectric
+            self.parent.air_box_extent_type = self._pedb.hfss.hfss_extent_info.extent_type
+            self.parent.air_box_truncate_model_ground_layers = (
+                self._pedb.hfss.hfss_extent_info.truncate_air_box_at_ground
+            )
+            self.parent.air_box_horizontal_padding = self._pedb.hfss.hfss_extent_info.air_box_horizontal_extent
+            self.parent.air_box_positive_vertical_padding = (
+                self._pedb.hfss.hfss_extent_info.air_box_positive_vertical_extent
+            )
+            self.parent.air_box_negative_vertical_padding = (
+                self._pedb.hfss.hfss_extent_info.air_box_negative_vertical_extent
+            )
+            return self.parent.get_attributes(exclude="boundary_data")
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
+        def get_parameters_from_edb(self):
+            self.parent.open_region = self._pedb.hfss.hfss_extent_info.use_open_region
+            self.parent.open_region_type = self._pedb.hfss.hfss_extent_info.open_region_type
+            self.parent.pml_visible = self._pedb.hfss.hfss_extent_info.is_pml_visible
+            self.parent.pml_operation_frequency = self._pedb.hfss.hfss_extent_info.operating_freq.tostring
+            self.parent.pml_radiation_factor = self._pedb.hfss.hfss_extent_info.radiation_level.tostring
+            self.parent.dielectric_extent_type = self._pedb.hfss.hfss_extent_info.extent_type
+            self.parent.horizontal_padding = self._pedb.hfss.hfss_extent_info.dielectric_extent_size
+            self.parent.honor_primitives_on_dielectric_layers = self._pedb.hfss.hfss_extent_info.honor_user_dielectric
+            self.parent.air_box_extent_type = self._pedb.hfss.hfss_extent_info.extent_type
+            self.parent.air_box_truncate_model_ground_layers = (
+                self._pedb.hfss.hfss_extent_info.truncate_air_box_at_ground
+            )
+            self.parent.air_box_horizontal_padding = self._pedb.hfss.hfss_extent_info.air_box_horizontal_extent
+            self.parent.air_box_positive_vertical_padding = (
+                self._pedb.hfss.hfss_extent_info.air_box_positive_vertical_extent
+            )
+            self.parent.air_box_negative_vertical_padding = (
+                self._pedb.hfss.hfss_extent_info.air_box_negative_vertical_extent
+            )
+            return self.parent.get_attributes(exclude="boundary_data")
+
+        def set_parameters_to_edb(self):
+            """Imports boundary information from JSON."""
+            if self.open_region is not None:
+                self._pedb.hfss.hfss_extent_info.use_open_region = self.open_region
+            if self.open_region_type:
+                self._pedb.hfss.hfss_extent_info.open_region_type = self.open_region_type.lower()
+            if self.pml_visible is not None:
+                self._pedb.hfss.hfss_extent_info.is_pml_visible = self.pml_visible
+            if self.pml_operation_frequency:
+                self._pedb.hfss.hfss_extent_info.operating_freq = self.pml_operation_frequency
+            if self.pml_radiation_factor:
+                if self._pedb.grpc:
+                    self._pedb.hfss.hfss_extent_info.pml_radiation_factor = self.pml_radiation_factor
+                else:
+                    self._pedb.hfss.hfss_extent_info.radiation_level = self.pml_radiation_factor
+            if self.dielectric_extent_type:
+                self._pedb.hfss.hfss_extent_info.extent_type = self.dielectric_extent_type.lower()
+            if self.horizontal_padding:
+                self._pedb.hfss.hfss_extent_info.dielectric_extent_size = float(self.horizontal_padding)
+            if self.honor_primitives_on_dielectric_layers is not None:
+                self._pedb.hfss.hfss_extent_info.honor_user_dielectric = self.honor_primitives_on_dielectric_layers
+            if self.air_box_extent_type:
+                self._pedb.hfss.hfss_extent_info.extent_type = self.air_box_extent_type.lower()
+            if self.air_box_truncate_model_ground_layers is not None:
+                self._pedb.hfss.hfss_extent_info.truncate_air_box_at_ground = self.air_box_truncate_model_ground_layers
+            if self.air_box_horizontal_padding:
+                self._pedb.hfss.hfss_extent_info.air_box_horizontal_extent = float(self.air_box_horizontal_padding)
+            if self.air_box_positive_vertical_padding:
+                self._pedb.hfss.hfss_extent_info.air_box_positive_vertical_extent = float(
+                    self.air_box_positive_vertical_padding
+                )
+            if self.air_box_negative_vertical_padding:
+                self._pedb.hfss.hfss_extent_info.air_box_negative_vertical_extent = float(
+                    self.air_box_negative_vertical_padding
+                )
+
+    def __init__(self, pedb, boundary_data):
+        self.pedb = pedb
+        self.boundary_data = boundary_data
+        if self.pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
+
+    def apply(self):
+        """Imports boundary information from JSON."""
+        self.api.set_parameters_to_edb()
+
+    def get_data_from_db(self):
+        return self.api.get_parameters_from_edb()

{pyedb-0.42.0 → pyedb-0.44.0}/src/pyedb/configuration/cfg_components.py

@@ -262,7 +262,7 @@ class CfgComponent(CfgBase):
         def set_parameters_to_edb(self):
             if self.parent.type:
                 self.pyedb_obj.type = self.parent.type
-            if self.parent.enabled:
+            if self.parent.enabled is not None:
                 self.pyedb_obj.enabled = self.parent.enabled
 
             self._set_model_properties_to_edb()

{pyedb-0.42.0 → pyedb-0.44.0}/src/pyedb/configuration/cfg_general.py

@@ -24,26 +24,44 @@
 class CfgGeneral:
     """Manage configuration general settings."""
 
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self.pedb = parent.pedb
+
+        def set_parameters_to_edb(self):
+            self.pedb.active_cell.anti_pads_always_on = self.parent.anti_pads_always_on
+            self.pedb.active_cell.suppress_pads = self.parent.suppress_pads
+
+        def get_parameters_from_edb(self):
+            anti_pads_always_on = self.pedb.design_options.antipads_always_on
+            suppress_pads = self.pedb.design_options.suppress_pads
+            data = {"anti_pads_always_on": anti_pads_always_on, "suppress_pads": suppress_pads}
+            return data
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
+        def set_parameters_to_edb(self):
+            if self.parent.anti_pads_always_on is not None:
+                self.pedb.design_options.anti_pads_always_on = self.parent.anti_pads_always_on
+            if self.parent.suppress_pads is not None:
+                self.pedb.design_options.suppress_pads = self.parent.suppress_pads
+
     def __init__(self, pedb, data):
-        self._pedb = pedb
+        self.pedb = pedb
+        if self.pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.spice_model_library = data.get("spice_model_library", "")
         self.s_parameter_library = data.get("s_parameter_library", "")
-        self.anti_pads_always_on = data.get("anti_pads_always_on", False)
-        self.suppress_pads = data.get("suppress_pads", True)
+        self.anti_pads_always_on = data.get("anti_pads_always_on", None)
+        self.suppress_pads = data.get("suppress_pads", None)
 
     def apply(self):
-        if self._pedb.grpc:
-            self._pedb.active_cell.anti_pads_always_on = self.anti_pads_always_on
-            self._pedb.active_cell.suppress_pads = self.suppress_pads
-        else:
-            self._pedb.design_options.antipads_always_on = self.anti_pads_always_on
-            self._pedb.design_options.suppress_pads = self.suppress_pads
+        self.api.set_parameters_to_edb()
 
     def get_data_from_db(self):
-        self.anti_pads_always_on = self._pedb.design_options.antipads_always_on
-        self.suppress_pads = self._pedb.design_options.suppress_pads
-
-        data = {}
-        data["anti_pads_always_on"] = self.anti_pads_always_on
-        data["suppress_pads"] = self.suppress_pads
-        return data
+        return self.api.get_parameters_from_edb()

pyedb-0.44.0/src/pyedb/configuration/cfg_modeler.py

@@ -0,0 +1,235 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from pyedb.configuration.cfg_components import CfgComponent
+from pyedb.configuration.cfg_padstacks import CfgPadstackDefinition, CfgPadstackInstance
+from pyedb.dotnet.database.edb_data.padstacks_data import EDBPadstack
+
+
+class CfgTrace:
+    def __init__(self, **kwargs):
+        self.name = kwargs.get("name", "")
+        self.layer = kwargs["layer"]
+        self.path = kwargs["path"]
+        self.width = kwargs["width"]
+        self.net_name = kwargs.get("net_name", "")
+        self.start_cap_style = kwargs.get("start_cap_style", "round")
+        self.end_cap_style = kwargs.get("end_cap_style", "round")
+        self.corner_style = kwargs.get("corner_style", "sharp")
+
+
+class CfgPlane:
+    def __init__(self, **kwargs):
+        self.name = kwargs.get("name", "")
+        self.layer = kwargs["layer"]
+        self.net_name = kwargs.get("net_name", "")
+        self.type = kwargs.get("type", "rectangle")
+
+        # rectangle
+        self.lower_left_point = kwargs.get("lower_left_point", [])
+        self.upper_right_point = kwargs.get("upper_right_point", [])
+        self.corner_radius = kwargs.get("corner_radius", 0)
+        self.rotation = kwargs.get("rotation", 0)
+        self.voids = kwargs.get("voids", [])
+
+        # polygon
+        self.points = kwargs.get("points", [])
+
+
+class CfgModeler:
+    """Manage configuration general settings."""
+
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def set_parameter_to_edb(self):
+            from ansys.edb.core.definition.padstack_def import (
+                PadstackDef as GrpcPadstackDef,
+            )
+            from ansys.edb.core.definition.padstack_def_data import (
+                PadstackDefData as GrpcPadstackDefData,
+            )
+
+            from pyedb.grpc.database.definition.padstack_def import PadstackDef
+
+            if self.parent.traces:
+                for t in self.parent.traces:
+                    obj = self._pedb.modeler.create_trace(
+                        path_list=t.path,
+                        layer_name=t.layer,
+                        net_name=t.net_name,
+                        width=t.width,
+                        start_cap_style=t.start_cap_style,
+                        end_cap_style=t.end_cap_style,
+                        corner_style=t.corner_style,
+                    )
+                    obj.aedt_name = t.name
+
+            if self.parent.padstack_defs:
+                for p in self.parent.padstack_defs:
+                    pdata = GrpcPadstackDefData.create()
+                    pdef = GrpcPadstackDef.create(self._pedb.active_db, p.name)
+                    pdef.data = pdata
+                    pdef = PadstackDef(self._pedb, pdef)
+                    p.pyedb_obj = pdef
+                    p.api.set_parameters_to_edb()
+
+            if self.parent.padstack_instances:
+                for p in self.parent.padstack_instances:
+                    p_inst = self._pedb.padstacks.place(
+                        via_name=p.name,
+                        net_name=p.net_name,
+                        position=p.position,
+                        definition_name=p.definition,
+                    )
+                    p.pyedb_obj = p_inst
+                    p.api.set_parameters_to_edb()
+
+            if self.parent.planes:
+                for p in self.parent.planes:
+                    if p.type == "rectangle":
+                        obj = self._pedb.modeler.create_rectangle(
+                            layer_name=p.layer,
+                            net_name=p.net_name,
+                            lower_left_point=p.lower_left_point,
+                            upper_right_point=p.upper_right_point,
+                            corner_radius=p.corner_radius,
+                            rotation=p.rotation,
+                        )
+                        obj.aedt_name = p.name
+                    elif p.type == "polygon":
+                        obj = self._pedb.modeler.create_polygon(
+                            points=p.points, layer_name=p.layer, net_name=p.net_name
+                        )
+                        obj.aedt_name = p.name
+
+                    for v in p.voids:
+                        for i in self._pedb.layout.primitives:
+                            if i.aedt_name == v:
+                                self._pedb.modeler.add_void(obj, i)
+
+            if self.parent.components:
+                pedb_p_inst = self._pedb.padstacks.instances_by_name
+                for c in self.parent.components:
+                    obj = self._pedb.components.create(
+                        [pedb_p_inst[i] for i in c.pins],
+                        component_name=c.reference_designator,
+                        placement_layer=c.placement_layer,
+                        component_part_name=c.definition,
+                    )
+                    c.pyedb_obj = obj
+                    c.api.set_parameters_to_edb()
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
+        def set_parameter_to_edb(self):
+            if self.parent.traces:
+                for t in self.parent.traces:
+                    obj = self._pedb.modeler.create_trace(
+                        path_list=t.path,
+                        layer_name=t.layer,
+                        net_name=t.net_name,
+                        width=t.width,
+                        start_cap_style=t.start_cap_style,
+                        end_cap_style=t.end_cap_style,
+                        corner_style=t.corner_style,
+                    )
+                    obj.aedt_name = t.name
+
+            if self.parent.padstack_defs:
+                for p in self.parent.padstack_defs:
+                    pdata = self._pedb._edb.Definition.PadstackDefData.Create()
+                    pdef = self._pedb._edb.Definition.PadstackDef.Create(self._pedb.active_db, p.name)
+                    pdef.SetData(pdata)
+                    pdef = EDBPadstack(pdef, self._pedb.padstacks)
+                    p.pyedb_obj = pdef
+                    p.api.set_parameters_to_edb()
+
+            if self.parent.padstack_instances:
+                for p in self.parent.padstack_instances:
+                    p_inst = self._pedb.padstacks.place(
+                        via_name=p.name,
+                        net_name=p.net_name,
+                        position=p.position,
+                        definition_name=p.definition,
+                        rotation=p.rotation if p.rotation is not None else 0,
+                    )
+                    p.pyedb_obj = p_inst
+                    p.api.set_parameters_to_edb()
+
+            if self.parent.planes:
+                for p in self.parent.planes:
+                    if p.type == "rectangle":
+                        obj = self._pedb.modeler.create_rectangle(
+                            layer_name=p.layer,
+                            net_name=p.net_name,
+                            lower_left_point=p.lower_left_point,
+                            upper_right_point=p.upper_right_point,
+                            corner_radius=p.corner_radius,
+                            rotation=p.rotation,
+                        )
+                        obj.aedt_name = p.name
+                    elif p.type == "polygon":
+                        obj = self._pedb.modeler.create_polygon(
+                            main_shape=p.points, layer_name=p.layer, net_name=p.net_name
+                        )
+                        obj.aedt_name = p.name
+
+                    for v in p.voids:
+                        for i in self._pedb.layout.primitives:
+                            if i.aedt_name == v:
+                                self._pedb.modeler.add_void(obj, i)
+
+            if self.parent.components:
+                pedb_p_inst = self._pedb.padstacks.instances_by_name
+                for c in self.parent.components:
+                    obj = self._pedb.components.create(
+                        [pedb_p_inst[i] for i in c.pins],
+                        component_name=c.reference_designator,
+                        placement_layer=c.placement_layer,
+                        component_part_name=c.definition,
+                    )
+                    c.pyedb_obj = obj
+                    c.api.set_parameters_to_edb()
+
+    def __init__(self, pedb, data):
+        self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
+        self.traces = [CfgTrace(**i) for i in data.get("traces", [])]
+        self.padstack_defs = [
+            CfgPadstackDefinition(self._pedb, None, **i) for i in data.get("padstack_definitions", [])
+        ]
+        self.padstack_instances = [
+            CfgPadstackInstance(self._pedb, None, **i) for i in data.get("padstack_instances", [])
+        ]
+        self.planes = [CfgPlane(**i) for i in data.get("planes", [])]
+        self.components = [CfgComponent(self._pedb, None, **i) for i in data.get("components", [])]
+
+    def apply(self):
+        self.api.set_parameter_to_edb()

{pyedb-0.42.0 → pyedb-0.44.0}/src/pyedb/configuration/cfg_nets.py

@@ -24,8 +24,39 @@
 class CfgNets:
     """Manage configuration net class."""
 
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def set_parameter_to_edb(self):
+            for signal_net in self.parent.signal_nets:
+                if signal_net in self._pedb.nets:
+                    self._pedb.nets.nets[signal_net].is_power_ground = False
+            for power_net in self.parent.power_nets:
+                if power_net in self._pedb.nets:
+                    self._pedb.nets.nets[power_net].is_power_ground = True
+
+        def get_parameter_from_edb(self):
+            """Get net information."""
+            for net in self._pedb.nets.signal:
+                self.parent.signal_nets.append(net)
+            for net in self._pedb.nets.power:
+                self.parent.power_nets.append(net)
+            data = {"signal_nets": self.parent.signal_nets, "power_ground_nets": self.parent.power_nets}
+            return data
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            self.parent = parent
+            super().__init__(parent)
+
     def __init__(self, pdata, signal_nets=None, power_nets=None):
         self._pedb = pdata._pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.signal_nets = []
         self.power_nets = []
         if signal_nets:
@@ -35,23 +66,8 @@ class CfgNets:
 
     def apply(self):
         """Apply net on layout."""
-        for signal_net in self.signal_nets:
-            if signal_net in self._pedb.nets:
-                self._pedb.nets.nets[signal_net].is_power_ground = False
-        for power_net in self.power_nets:
-            if power_net in self._pedb.nets:
-                self._pedb.nets.nets[power_net].is_power_ground = True
-
-    def _load_data_from_db(self):
-        self.signal_nets = []
-        self.power_nets = []
-        for net in self._pedb.nets.signal:
-            self.signal_nets.append(net)
-        for net in self._pedb.nets.power:
-            self.power_nets.append(net)
+        self.api.set_parameter_to_edb()
 
     def get_data_from_db(self):
         """Get net information."""
-        self._load_data_from_db()
-        data = {"signal_nets": self.signal_nets, "power_ground_nets": self.power_nets}
-        return data
+        return self.api.get_parameter_from_edb()