pyedb 0.42.0__py3-none-any.whl → 0.43.0__py3-none-any.whl

pyedb/configuration/cfg_operations.py

@@ -24,9 +24,38 @@ from pyedb.configuration.cfg_common import CfgBase
 
 
 class CfgCutout(CfgBase):
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def get_parameters_from_edb(self):
+            if "pyedb_cutout" in self._pedb.stackup.all_layers:
+                polygons = self._pedb.layout.find_primitive(layer_name="pyedb_cutout")
+                if polygons:
+                    poly = polygons[0]
+                    self.parent.custom_extent = poly.polygon_data.points
+                    net_names = []
+                    for name, obj in self._pedb.nets.nets.items():
+                        if obj.primitives:
+                            if obj.primitives[0].layer.name == "pyedb_cutout":
+                                continue
+                            else:
+                                net_names.append(name)
+                    self.parent.reference_list = []
+                    self.parent.signal_list = net_names
+                return self.parent.export_properties()
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, **kwargs):
         self._pedb = pedb
-
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.signal_list = kwargs.get("signal_list")
         self.reference_list = kwargs.get("reference_list")
         self.extent_type = kwargs.get("extent_type")
@@ -52,23 +81,7 @@ class CfgCutout(CfgBase):
         self.keep_lines_as_path = kwargs.get("keep_lines_as_path")
 
     def get_data_from_db(self):
-        if "pyedb_cutout" in self._pedb.stackup.all_layers:
-            polygons = self._pedb.layout.find_primitive(layer_name="pyedb_cutout")
-            if polygons:
-                poly = polygons[0]
-                self.custom_extent = poly.polygon_data.points
-
-                net_names = []
-                for name, obj in self._pedb.nets.nets.items():
-                    if obj.primitives:
-                        if obj.primitives[0].layer.name == "pyedb_cutout":
-                            continue
-                        else:
-                            net_names.append(name)
-
-                self.reference_list = []
-                self.signal_list = net_names
-            return self.export_properties()
+        return self.api.get_parameters_from_edb()
 
     def export_properties(self):
         return {
@@ -79,24 +92,43 @@ class CfgCutout(CfgBase):
 
 
 class CfgOperations(CfgBase):
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def apply_on_edb(self):
+            if self.parent.op_cutout:
+                polygon_points = self._pedb.cutout(**self.parent.op_cutout.get_attributes())
+                if "pyedb_cutout" not in self._pedb.stackup.all_layers:
+                    self._pedb.stackup.add_document_layer(name="pyedb_cutout")
+                    self._pedb.modeler.create_polygon(
+                        polygon_points, layer_name="pyedb_cutout", net_name="pyedb_cutout"
+                    )
+
+        def get_parameter_from_edb(self):
+            self.parent.op_cutout = CfgCutout(self._pedb)
+            data_from_db = self.parent.op_cutout.get_data_from_db()
+            if data_from_db:
+                return {"cutout": data_from_db}
+            else:
+                return {}
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, data):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.op_cutout = CfgCutout(pedb, **data["cutout"]) if "cutout" in data else None
 
     def apply(self):
         """Imports operation information from JSON."""
-        if self.op_cutout:
-            polygon_points = self._pedb.cutout(**self.op_cutout.get_attributes())
-            if "pyedb_cutout" not in self._pedb.stackup.all_layers:
-                self._pedb.stackup.add_document_layer(name="pyedb_cutout")
-                self._pedb.modeler.create_polygon(polygon_points, layer_name="pyedb_cutout", net_name="pyedb_cutout")
-
-            # create a polygon on pyedb layer
+        self.api.apply_on_edb()
 
     def get_data_from_db(self):
-        self.op_cutout = CfgCutout(self._pedb)
-        data_from_db = self.op_cutout.get_data_from_db()
-        if data_from_db:
-            return {"cutout": data_from_db}
-        else:
-            return {}
+        return self.api.get_parameter_from_edb()

pyedb/configuration/cfg_package_definition.py

@@ -64,61 +64,82 @@ class CfgHeatSink(CfgBase):
 
 
 class CfgPackageDefinitions:
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def set_parameter_to_edb(self):
+            for pkg in self.parent.packages:
+                comp_def_from_db = self._pedb.definitions.component[pkg.component_definition]
+                if pkg.name in self._pedb.definitions.package:
+                    self._pedb.definitions.package[pkg.name].delete()
+
+                if pkg.extent_bounding_box:
+                    package_def = PackageDef(self._pedb, name=pkg.name, extent_bounding_box=pkg.extent_bounding_box)
+                else:
+                    package_def = PackageDef(self._pedb, name=pkg.name, component_part_name=pkg.component_definition)
+                pkg.set_attributes(package_def)
+
+                if pkg.heatsink:
+                    attrs = pkg.heatsink.get_attributes()
+                    for attr, value in attrs.items():
+                        package_def.set_heatsink(**attrs)
+
+                comp_list = dict()
+                if pkg.apply_to_all:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes not in pkg.components
+                        }
+                    )
+                else:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes in pkg.components
+                        }
+                    )
+                for _, i in comp_list.items():
+                    i.package_def = pkg.name
+
+        def get_parameter_from_edb(self):
+            package_definitions = []
+            for pkg_name, pkg_obj in self._pedb.definitions.package.items():
+                pkg = {}
+                pkg_attrs = [i for i in dir(pkg_obj) if not i.startswith("_")]
+                pkg_attrs = {i for i in pkg_attrs if i in CfgPackage().__dict__}
+                for pkg_attr_name in pkg_attrs:
+                    pkg[pkg_attr_name] = getattr(pkg_obj, pkg_attr_name)
+                hs_obj = pkg_obj.heatsink
+                if hs_obj:
+                    hs = {}
+                    hs_attrs = [i for i in dir(hs_obj) if not i.startswith("_")]
+                    hs_attrs = [i for i in hs_attrs if i in CfgHeatSink().__dict__]
+                    for hs_attr_name in hs_attrs:
+                        hs[hs_attr_name] = getattr(hs_obj, hs_attr_name)
+                    pkg["heatsink"] = hs
+                package_definitions.append(pkg)
+
+            return package_definitions
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, data):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.packages = [CfgPackage(**package) for package in data]
 
     def apply(self):
-        for pkg in self.packages:
-            comp_def_from_db = self._pedb.definitions.component[pkg.component_definition]
-            if pkg.name in self._pedb.definitions.package:
-                self._pedb.definitions.package[pkg.name].delete()
-
-            if pkg.extent_bounding_box:
-                package_def = PackageDef(self._pedb, name=pkg.name, extent_bounding_box=pkg.extent_bounding_box)
-            else:
-                package_def = PackageDef(self._pedb, name=pkg.name, component_part_name=pkg.component_definition)
-            pkg.set_attributes(package_def)
-
-            if pkg.heatsink:
-                attrs = pkg.heatsink.get_attributes()
-                for attr, value in attrs.items():
-                    package_def.set_heatsink(**attrs)
-
-            comp_list = dict()
-            if pkg.apply_to_all:
-                comp_list.update(
-                    {
-                        refdes: comp
-                        for refdes, comp in comp_def_from_db.components.items()
-                        if refdes not in pkg.components
-                    }
-                )
-            else:
-                comp_list.update(
-                    {refdes: comp for refdes, comp in comp_def_from_db.components.items() if refdes in pkg.components}
-                )
-            for _, i in comp_list.items():
-                i.package_def = pkg.name
+        self.api.set_parameter_to_edb()
 
     def get_data_from_db(self):
-        package_definitions = []
-
-        for pkg_name, pkg_obj in self._pedb.definitions.package.items():
-            pkg = {}
-            pkg_attrs = [i for i in dir(pkg_obj) if not i.startswith("_")]
-            pkg_attrs = {i for i in pkg_attrs if i in CfgPackage().__dict__}
-            for pkg_attr_name in pkg_attrs:
-                pkg[pkg_attr_name] = getattr(pkg_obj, pkg_attr_name)
-
-            hs_obj = pkg_obj.heatsink
-            if hs_obj:
-                hs = {}
-                hs_attrs = [i for i in dir(hs_obj) if not i.startswith("_")]
-                hs_attrs = [i for i in hs_attrs if i in CfgHeatSink().__dict__]
-                for hs_attr_name in hs_attrs:
-                    hs[hs_attr_name] = getattr(hs_obj, hs_attr_name)
-                pkg["heatsink"] = hs
-            package_definitions.append(pkg)
-
-        return package_definitions
+        return self.api.get_parameter_from_edb()

pyedb/configuration/cfg_pin_groups.py

@@ -26,28 +26,40 @@ from pyedb.configuration.cfg_common import CfgBase
 class CfgPinGroups:
     """Manage configuration pin group class."""
 
-    def __init__(self, pedb, pingroup_data):
-        self._pedb = pedb
-        self.pin_groups = [CfgPinGroup(self._pedb, **pg) for pg in pingroup_data]
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
 
-    def apply(self):
-        for pg in self.pin_groups:
-            pg.create()
+        def set_pingroup_to_edb(self):
+            for pg in self.parent.pin_groups:
+                pg.create()
 
-    def get_data_from_db(self):
-        self.pin_groups = []
-        layout_pin_groups = self._pedb.siwave.pin_groups
-        for pg_name, pg_obj in layout_pin_groups.items():
-            if self._pedb.grpc:
-                pins = pg_obj.pins
-                refdes = pins[0].component.name
-                cfg_pg = CfgPinGroup(
-                    self._pedb,
-                    name=pg_name,
-                    reference_designator=refdes,
-                    pins=[pin.component_pin for pin in pins],
-                )
-            else:
+        def get_data_from_edb(self):
+            self.parent.pin_groups = []
+            layout_pin_groups = self._pedb.siwave.pin_groups
+            for pg_name, pg_obj in layout_pin_groups.items():
+                if self._pedb.grpc:
+                    pins = pg_obj.pins
+                    refdes = pins[0].component.name
+                    cfg_pg = CfgPinGroup(
+                        self._pedb,
+                        name=pg_name,
+                        reference_designator=refdes,
+                        pins=[pin.component_pin for pin in pins],
+                    )
+                    self.parent.pin_groups.append(cfg_pg)
+            return self.parent.export_properties()
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            self.parent = parent
+            super().__init__(parent)
+
+        def get_data_from_edb(self):
+            self.parent.pin_groups = []
+            layout_pin_groups = self._pedb.siwave.pin_groups
+            for pg_name, pg_obj in layout_pin_groups.items():
                 pins = list(pg_obj.pins.keys())
                 refdes = list(pg_obj.pins.values())[0].component.name
                 cfg_pg = CfgPinGroup(
@@ -56,8 +68,22 @@ class CfgPinGroups:
                     reference_designator=refdes,
                     pins=pins,
                 )
-            self.pin_groups.append(cfg_pg)
-        return self.export_properties()
+                self.parent.pin_groups.append(cfg_pg)
+            return self.parent.export_properties()
+
+    def __init__(self, pedb, pingroup_data):
+        self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
+        self.pin_groups = [CfgPinGroup(self._pedb, **pg) for pg in pingroup_data]
+
+    def apply(self):
+        self.api.set_pingroup_to_edb()
+
+    def get_data_from_db(self):
+        return self.api.get_data_from_edb()
 
     def export_properties(self):
         pin_groups = []
@@ -67,8 +93,34 @@ class CfgPinGroups:
 
 
 class CfgPinGroup(CfgBase):
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def create(self):
+            if self.parent.pins:
+                pins = self.parent.pins if isinstance(self.parent.pins, list) else [self.parent.pins]
+                self._pedb.siwave.create_pin_group(self.parent.reference_designator, pins, self.parent.name)
+            elif self.parent.net:
+                nets = self.parent.net if isinstance(self.parent.net, list) else [self.parent.net]
+                comp = self._pedb.components.instances[self.parent.reference_designator]
+                pins = [p for p, obj in comp.pins.items() if obj.net_name in nets]
+                if not self._pedb.siwave.create_pin_group(self.parent.reference_designator, pins, self.parent.name):
+                    raise RuntimeError(f"Failed to create pin group {self.parent.name}")
+            else:
+                raise RuntimeError(f"No net and pins defined for defining pin group {self.parent.name}")
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, **kwargs):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.name = kwargs["name"]
         self.reference_designator = kwargs.get("reference_designator")
         self.pins = kwargs.get("pins")
@@ -76,17 +128,7 @@ class CfgPinGroup(CfgBase):
 
     def create(self):
         """Apply pin group on layout."""
-        if self.pins:
-            pins = self.pins if isinstance(self.pins, list) else [self.pins]
-            self._pedb.siwave.create_pin_group(self.reference_designator, pins, self.name)
-        elif self.net:
-            nets = self.net if isinstance(self.net, list) else [self.net]
-            comp = self._pedb.components.instances[self.reference_designator]
-            pins = [p for p, obj in comp.pins.items() if obj.net_name in nets]
-            if not self._pedb.siwave.create_pin_group(self.reference_designator, pins, self.name):
-                raise RuntimeError(f"Failed to create pin group {self.name}")
-        else:
-            raise RuntimeError(f"No net and pins defined for defining pin group {self.name}")
+        self.api.create()
 
     def export_properties(self):
         if self.pins:

pyedb/configuration/cfg_s_parameter_models.py

@@ -36,81 +36,106 @@ class CfgSParameterModel:
 
 
 class CfgSParameters:
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def apply(self):
+            for s_param in self.parent.s_parameters_models:
+                fpath = s_param.file_path
+                if not Path(fpath).anchor:
+                    fpath = str(Path(self.parent.path_libraries) / fpath)
+                comp_def = self._pedb.definitions.component[s_param.component_definition]
+                if s_param.pin_order:
+                    comp_def.set_properties(pin_order=s_param.pin_order)
+                comp_def.add_n_port_model(fpath, s_param.name)
+                comp_list = dict()
+                if s_param.apply_to_all:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def.components.items()
+                            if refdes not in s_param.components
+                        }
+                    )
+                else:
+                    comp_list.update(
+                        {refdes: comp for refdes, comp in comp_def.components.items() if refdes in s_param.components}
+                    )
+
+                for refdes, comp in comp_list.items():
+                    if refdes in s_param.reference_net_per_component:
+                        ref_net = s_param.reference_net_per_component[refdes]
+                    else:
+                        ref_net = s_param.reference_net
+                    comp.use_s_parameter_model(s_param.name, reference_net=ref_net)
+
+        def get_data_from_db(self, cfg_components):
+            db_comp_def = self._pedb.definitions.component
+            for name, compdef_obj in db_comp_def.items():
+                nport_models = compdef_obj.component_models
+                if not nport_models:
+                    continue
+                else:
+                    pin_order = compdef_obj.get_properties()["pin_order"]
+                    temp_comps = [i for i in cfg_components if i["definition"] == name]
+                    for model_name, model_obj in nport_models.items():
+                        temp_comp_list = []
+                        reference_net_per_component = {}
+                        for i in temp_comps:
+                            s_param_model = i.get("s_parameter_model")
+                            if s_param_model:
+                                if s_param_model["model_name"] == model_name:
+                                    temp_comp_list.append(i["reference_designator"])
+                                    reference_net_per_component[i["reference_designator"]] = s_param_model[
+                                        "reference_net"
+                                    ]
+                            else:
+                                continue
+
+                        self.parent.s_parameters_models.append(
+                            CfgSParameterModel(
+                                name=model_name,
+                                component_definition=name,
+                                file_path=model_obj.reference_file,
+                                apply_to_all=False,
+                                components=temp_comp_list,
+                                reference_net_per_component=reference_net_per_component,
+                                pin_order=pin_order,
+                            )
+                        )
+
+            data = []
+            for i in self.parent.s_parameters_models:
+                data.append(
+                    {
+                        "name": i.name,
+                        "component_definition": i.component_definition,
+                        "file_path": i.file_path,
+                        "apply_to_all": i.apply_to_all,
+                        "components": i.components,
+                        "reference_net_per_component": i.reference_net_per_component,
+                        "pin_order": i.pin_order,
+                    }
+                )
+            return data
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, data, path_lib=None):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.path_libraries = path_lib
         self.s_parameters_models = [CfgSParameterModel(**i) for i in data]
 
     def apply(self):
-        for s_param in self.s_parameters_models:
-            fpath = s_param.file_path
-            if not Path(fpath).anchor:
-                fpath = str(Path(self.path_libraries) / fpath)
-            comp_def = self._pedb.definitions.component[s_param.component_definition]
-            if s_param.pin_order:
-                comp_def.set_properties(pin_order=s_param.pin_order)
-            comp_def.add_n_port_model(fpath, s_param.name)
-            comp_list = dict()
-            if s_param.apply_to_all:
-                comp_list.update(
-                    {refdes: comp for refdes, comp in comp_def.components.items() if refdes not in s_param.components}
-                )
-            else:
-                comp_list.update(
-                    {refdes: comp for refdes, comp in comp_def.components.items() if refdes in s_param.components}
-                )
-
-            for refdes, comp in comp_list.items():
-                if refdes in s_param.reference_net_per_component:
-                    ref_net = s_param.reference_net_per_component[refdes]
-                else:
-                    ref_net = s_param.reference_net
-                comp.use_s_parameter_model(s_param.name, reference_net=ref_net)
+        self.api.apply()
 
     def get_data_from_db(self, cfg_components):
-        db_comp_def = self._pedb.definitions.component
-        for name, compdef_obj in db_comp_def.items():
-            nport_models = compdef_obj.component_models
-            if not nport_models:
-                continue
-            else:
-                pin_order = compdef_obj.get_properties()["pin_order"]
-                temp_comps = [i for i in cfg_components if i["definition"] == name]
-                for model_name, model_obj in nport_models.items():
-                    temp_comp_list = []
-                    reference_net_per_component = {}
-                    for i in temp_comps:
-                        s_param_model = i.get("s_parameter_model")
-                        if s_param_model:
-                            if s_param_model["model_name"] == model_name:
-                                temp_comp_list.append(i["reference_designator"])
-                                reference_net_per_component[i["reference_designator"]] = s_param_model["reference_net"]
-                        else:
-                            continue
-
-                    self.s_parameters_models.append(
-                        CfgSParameterModel(
-                            name=model_name,
-                            component_definition=name,
-                            file_path=model_obj.reference_file,
-                            apply_to_all=False,
-                            components=temp_comp_list,
-                            reference_net_per_component=reference_net_per_component,
-                            pin_order=pin_order,
-                        )
-                    )
-
-        data = []
-        for i in self.s_parameters_models:
-            data.append(
-                {
-                    "name": i.name,
-                    "component_definition": i.component_definition,
-                    "file_path": i.file_path,
-                    "apply_to_all": i.apply_to_all,
-                    "components": i.components,
-                    "reference_net_per_component": i.reference_net_per_component,
-                    "pin_order": i.pin_order,
-                }
-            )
-        return data
+        return self.api.get_data_from_db(cfg_components)