pyedb 0.55.0__py3-none-any.whl → 0.57.0__py3-none-any.whl

pyedb/grpc/database/net/differential_pair.py

@@ -20,15 +20,18 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pyedb.grpc.database.net.net import Net
 import re
 
 from ansys.edb.core.net.differential_pair import (
     DifferentialPair as GrpcDifferentialPair,
 )
 
-from pyedb.grpc.database.net.net import Net
-
 
 class DifferentialPairs:
     def __init__(self, pedb):
@@ -131,9 +134,13 @@ class DifferentialPair(GrpcDifferentialPair):
     @property
     def positive_net(self) -> Net:
         """Positive Net."""
+        from pyedb.grpc.database.net.net import Net
+
         return Net(self._pedb, super().positive_net)
 
     @property
     def negative_net(self) -> Net:
         """Negative Net."""
+        from pyedb.grpc.database.net.net import Net
+
         return Net(self._pedb, super().negative_net)

pyedb/grpc/database/net/extended_net.py

@@ -20,9 +20,13 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-from ansys.edb.core.net.extended_net import ExtendedNet as GrpcExtendedNet
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
-from pyedb.grpc.database.net.net import Net
+if TYPE_CHECKING:
+    from pyedb.grpc.database.net.net import Net
+from ansys.edb.core.net.extended_net import ExtendedNet as GrpcExtendedNet
 
 
 class ExtendedNets:
@@ -204,12 +208,21 @@ class ExtendedNets:
                 val_value = cmp.rlc_values
                 if refdes in exception_list:
                     pass
-                elif cmp.type == "inductor" and val_value[1] < inductor_below:
-                    pass
-                elif cmp.type == "resistor" and val_value[0] < resistor_below:
-                    pass
-                elif cmp.type == "capacitor" and val_value[2] > capacitor_above:
-                    pass
+                elif cmp.type == "inductor":
+                    if val_value[1] is None:
+                        continue
+                    elif not val_value[1] < inductor_below:
+                        continue
+                elif cmp.type == "resistor":
+                    if val_value[0] is None:
+                        continue
+                    elif not val_value[0] < resistor_below:
+                        continue
+                elif cmp.type == "capacitor":
+                    if val_value[2] is None:
+                        continue
+                    elif not val_value[2] > capacitor_above:
+                        continue
                 else:
                     continue
                 for net in comp_dict[refdes]:
@@ -263,7 +276,7 @@ class ExtendedNet(GrpcExtendedNet):
         self._pedb = pedb
 
     @property
-    def nets(self):
+    def nets(self) -> dict[str, Net]:
         """Nets dictionary.
 
         Returns
@@ -271,6 +284,8 @@ class ExtendedNet(GrpcExtendedNet):
         Dict[str, :class:`Net <pyedb.grpc.database.net.net.Net>`]
             Dict[net name, Net object].
         """
+        from pyedb.grpc.database.net.net import Net
+
         return {net.name: Net(self._pedb, net) for net in super().nets}
 
     @property

pyedb/grpc/database/net/net.py

@@ -20,14 +20,17 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-from typing import Union
+from __future__ import annotations
 
+from typing import TYPE_CHECKING, Union
+
+if TYPE_CHECKING:
+    from pyedb.grpc.database.primitive.circle import Circle
+    from pyedb.grpc.database.primitive.padstack_instance import PadstackInstance
 from ansys.edb.core.net.net import Net as GrpcNet
 from ansys.edb.core.primitive.primitive import PrimitiveType as GrpcPrimitiveType
 
 from pyedb.grpc.database.primitive.bondwire import Bondwire
-from pyedb.grpc.database.primitive.circle import Circle
-from pyedb.grpc.database.primitive.padstack_instance import PadstackInstance
 from pyedb.grpc.database.primitive.path import Path
 from pyedb.grpc.database.primitive.polygon import Polygon
 from pyedb.grpc.database.primitive.rectangle import Rectangle
@@ -76,6 +79,7 @@ class Net(GrpcNet):
             - :class:`Rectangle <pyedb.grpc.database.primitive.rectangle.Rectangle>`
             - :class:`Bondwire <pyedb.grpc.database.primitive.bondwire.Bondwire>`
         """
+
         primitives = super().primitives
         if not len(self.__primitives) == len(primitives):
             for primitive in primitives:
@@ -100,6 +104,8 @@ class Net(GrpcNet):
         list of :class:`PadstackInstance <pyedb.grpc.database.primitive.padstack_instance.PadstackInstance>`
             Padstack instances associated with the net.
         """
+        from pyedb.grpc.database.primitive.padstack_instance import PadstackInstance
+
         return [PadstackInstance(self._pedb, i) for i in super().padstack_instances]
 
     @property
@@ -118,8 +124,11 @@ class Net(GrpcNet):
             if component:
                 try:
                     components[component.name] = component
-                except:
-                    pass
+                except Exception as e:
+                    self._pedb.logger.error(
+                        f"A(n) {type(e).__name__} error occurred while attempting to access "
+                        f"'components' property for object {self} - Empty dict is returned: {str(e)}"
+                    )
         return components
 
     def find_dc_short(self, fix=False) -> list[list[str, str]]:

pyedb/grpc/database/net/net_class.py

@@ -20,9 +20,14 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from ansys.edb.core.net.net_class import NetClass as GrpcNetClass
 
-from pyedb.grpc.database.net.net import Net
+if TYPE_CHECKING:
+    from pyedb.grpc.database.net.net import Net
 
 
 class NetClass(GrpcNetClass):
@@ -49,6 +54,8 @@ class NetClass(GrpcNetClass):
         List[:class:`Net <pyedb.grpc.database.net.net.Net>`].
             List of Net object.
         """
+        from pyedb.grpc.database.net.net import Net
+
         return [Net(self._pedb, i) for i in super().nets]
 
     def add_net(self, net):
@@ -59,21 +66,30 @@ class NetClass(GrpcNetClass):
         bool
         """
         if isinstance(net, str):
+            from pyedb.grpc.database.net.net import Net
+
             net = Net.find_by_name(self._pedb.active_layout, name=net)
         if isinstance(net, Net) and not net.is_null:
             self.add_net(net)
             return True
         return False
 
-    def contains_net(self, net):
-        """Determine if a net exists in the net class.
+    def contains_net(self, net) -> bool:
+        """
+        Determine if a net exists in the net class.
+
+        Parameters
+        ----------
+        net : str or Net
+            The net to check. This can be a string representing the net name or a `Net` object.
 
-        returns
+        Returns
         -------
-        List[:class:`Net <pyedb.grpc.database.net.net.Net>`].
-            List of Net object.
+        bool
+            True if the net exists in the net class, False otherwise.
 
         """
+
         if isinstance(net, str):
             net = Net.find_by_name(self._pedb.active_layout, name=net)
         return super().contains_net(net)
@@ -85,9 +101,10 @@ class NetClass(GrpcNetClass):
         -------
         bool
         """
+
         if isinstance(net, str):
             net = Net.find_by_name(self._pedb.active_layout, name=net)
         if isinstance(net, Net) and not net.is_null:
-            self.remove(net)
+            self.remove_net(net)
             return True
         return False

pyedb/grpc/database/nets.py

@@ -95,34 +95,21 @@ class Nets(CommonNets):
     >>> print("Eligible power nets:", [net.name for net in eligible_pwr])
 
     >>> # Generate extended nets (deprecated)
-    >>> nets.generate_extended_nets(
-    >>> resistor_below=5,
-    >>>inductor_below=0.5,
-    >>> capacitor_above=0.1
-    >>> )
+    >>> nets.generate_extended_nets(resistor_below=5, inductor_below=0.5, capacitor_above=0.1)
 
     >>> # Classify nets
-    >>> nets.classify_nets(
-    >>> power_nets=["VDD_CPU", "VDD_MEM"],
-    >>> signal_nets=["PCIe_TX", "ETH_RX"]
-    >>> )
+    >>> nets.classify_nets(power_nets=["VDD_CPU", "VDD_MEM"], signal_nets=["PCIe_TX", "ETH_RX"])
 
     >>> # Check power/ground status
     >>> is_power = nets.is_power_gound_net(["VDD_CPU", "PCIe_TX"])
     >>> print("Is power net:", is_power)
 
     >>> # Get DC-connected nets
-    >>> dc_connected = nets.get_dcconnected_net_list(
-    >>> ground_nets=["GND"],
-    >>> res_value=0.002
-    >>> )
+    >>> dc_connected = nets.get_dcconnected_net_list(ground_nets=["GND"], res_value=0.002)
         print("DC-connected nets:", dc_connected)
 
     >>> # Get power tree
-    >>> comp_list, columns, net_group = nets.get_powertree(
-    >>> power_net_name="VDD_CPU",
-    >>> ground_nets=["GND"]
-    >>> )
+    >>> comp_list, columns, net_group = nets.get_powertree(power_net_name="VDD_CPU", ground_nets=["GND"])
     >>> print("Power tree components:", comp_list)
 
     >>> # Find net by name
@@ -142,10 +129,7 @@ class Nets(CommonNets):
     >>> print("Net in component:", in_component)
 
     >>> # Find and fix disjoint nets (deprecated)
-    >>> fixed_nets = nets.find_and_fix_disjoint_nets(
-    >>> net_list=["PCIe_TX"],
-    >>> clean_disjoints_less_than=1e-6
-    >>> )
+    >>> fixed_nets = nets.find_and_fix_disjoint_nets(net_list=["PCIe_TX"], clean_disjoints_less_than=1e-6)
     >>> print("Fixed nets:", fixed_nets)
 
     >>> # Merge net polygons
@@ -422,11 +406,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> edb_nets.generate_extended_nets(
-        ...     resistor_below=5,
-        ...     inductor_below=0.5,
-        ...     capacitor_above=0.1
-        ... )
+        >>> edb_nets.generate_extended_nets(resistor_below=5, inductor_below=0.5, capacitor_above=0.1)
         """
         warnings.warn("Use new method :func:`edb.extended_nets.generate_extended_nets` instead.", DeprecationWarning)
         self._pedb.extended_nets.generate_extended_nets(
@@ -487,10 +467,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> edb_nets.classify_nets(
-        ...     power_nets=["VDD_CPU", "VDD_MEM"],
-        ...     signal_nets=["PCIe_TX", "ETH_RX"]
-        ... )
+        >>> edb_nets.classify_nets(power_nets=["VDD_CPU", "VDD_MEM"], signal_nets=["PCIe_TX", "ETH_RX"])
         """
         if isinstance(power_nets, str):
             power_nets = []
@@ -551,10 +528,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> dc_connected = edb_nets.get_dcconnected_net_list(
-        ...     ground_nets=["GND"],
-        ...     res_value=0.002
-        ... )
+        >>> dc_connected = edb_nets.get_dcconnected_net_list(ground_nets=["GND"], res_value=0.002)
         >>> for net_group in dc_connected:
         ...     print("Connected nets:", net_group)
         """
@@ -611,10 +585,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> comp_list, columns, net_group = edb_nets.get_powertree(
-        ...     power_net_name="VDD_CPU",
-        ...     ground_nets=["GND"]
-        ... )
+        >>> comp_list, columns, net_group = edb_nets.get_powertree(power_net_name="VDD_CPU", ground_nets=["GND"])
         >>> print("Power tree components:", comp_list)
         """
         flag_in_ng = False
@@ -700,7 +671,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> deleted_nets = database.nets.delete(["Net1","Net2"])
+        >>> deleted_nets = database.nets.delete(["Net1", "Net2"])
         """
         if isinstance(netlist, str):
             netlist = [netlist]
@@ -858,10 +829,7 @@ class Nets(CommonNets):
 
         Examples
         --------
-        >>> fixed_nets = edb_nets.find_and_fix_disjoint_nets(
-        ...     net_list=["PCIe_TX"],
-        ...     clean_disjoints_less_than=1e-6
-        ... )
+        >>> fixed_nets = edb_nets.find_and_fix_disjoint_nets(net_list=["PCIe_TX"], clean_disjoints_less_than=1e-6)
         >>> print("Fixed nets:", fixed_nets)
         """
 

pyedb/grpc/database/padstacks.py

@@ -23,6 +23,7 @@
 """
 This module contains the `EdbPadstacks` class.
 """
+
 from collections import defaultdict
 import math
 from typing import Any, Dict, List, Optional, Tuple, Union
@@ -30,20 +31,12 @@ import warnings
 
 from ansys.edb.core.definition.padstack_def_data import (
     PadGeometryType as GrpcPadGeometryType,
-)
-from ansys.edb.core.definition.padstack_def_data import (
     PadstackDefData as GrpcPadstackDefData,
-)
-from ansys.edb.core.definition.padstack_def_data import (
     PadstackHoleRange as GrpcPadstackHoleRange,
-)
-from ansys.edb.core.definition.padstack_def_data import (
+    PadType as GrpcPadType,
     SolderballPlacement as GrpcSolderballPlacement,
-)
-from ansys.edb.core.definition.padstack_def_data import (
     SolderballShape as GrpcSolderballShape,
 )
-from ansys.edb.core.definition.padstack_def_data import PadType as GrpcPadType
 from ansys.edb.core.geometry.point_data import PointData as GrpcPointData
 from ansys.edb.core.geometry.polygon_data import PolygonData as GrpcPolygonData
 import numpy as np
@@ -334,7 +327,7 @@ class Padstacks(object):
         >>> groups = edb_padstacks._layout.pin_groups  # New way
         """
         warnings.warn(
-            "`pingroups` is deprecated and is now located here " "`pyedb.grpc.core.layout.pin_groups` instead.",
+            "`pingroups` is deprecated and is now located here `pyedb.grpc.core.layout.pin_groups` instead.",
             DeprecationWarning,
         )
         return self._layout.pin_groups
@@ -380,10 +373,7 @@ class Padstacks(object):
         Examples
         --------
         >>> via_name = edb_padstacks.create_circular_padstack(
-        ...     padstackname="VIA1",
-        ...     holediam="200um",
-        ...     paddiam="400um",
-        ...     antipaddiam="600um"
+        ...     padstackname="VIA1", holediam="200um", paddiam="400um", antipaddiam="600um"
         ... )
         """
 
@@ -1299,8 +1289,7 @@ class Padstacks(object):
             List of padstack instances associated with the specified net.
         """
         warnings.warn(
-            "`get_padstack_instance_by_net_name` is deprecated, use `get_instances` with `net_name` "
-            "parameter instead.",
+            "`get_padstack_instance_by_net_name` is deprecated, use `get_instances` with `net_name` parameter instead.",
             DeprecationWarning,
         )
         return self.get_instances(net_name=net)
@@ -1853,3 +1842,90 @@ class Padstacks(object):
             clusters[int(label)].append(padstack_ids[i])
 
         return dict(clusters)
+
+    def reduce_via_by_density(
+        self, padstacks: List[int], cell_size_x: float = 1e-3, cell_size_y: float = 1e-3, delete: bool = False
+    ) -> tuple[List[int], List[List[List[float]]]]:
+        """
+        Reduce the number of vias by density. Keep only one via which is closest to the center of the cell. The cells
+        are automatically populated based on the input vias.
+
+        Parameters
+        ----------
+        padstacks: List[int]
+            List of padstack ids to be reduced.
+
+        cell_size_x : float
+            Width of each grid cell (default is 1e-3).
+
+        cell_size_y : float
+            Height of each grid cell (default is 1e-3).
+
+        delete: bool
+            If True, delete vias that are not kept (default is False).
+
+        Returns
+        -------
+        List[int]
+            IDs of vias kept after reduction.
+
+        List[List[float]]
+            coordinates for grid lines (for plotting).
+
+        """
+        to_keep = set()
+
+        all_instances = self.instances
+        positions = np.array([all_instances[_id].position for _id in padstacks])
+
+        x_coords, y_coords = positions[:, 0], positions[:, 1]
+        x_min, x_max = np.min(x_coords), np.max(x_coords)
+        y_min, y_max = np.min(y_coords), np.max(y_coords)
+
+        padstacks_array = np.array(padstacks)
+        cell_map = {}  # {(cell_x, cell_y): [(id1, [x1, y1]), (id2, [x2, y2), ...]}
+        grid = []
+
+        for idx, pos in enumerate(positions):
+            i = int((pos[0] - x_min) // cell_size_x)
+            j = int((pos[1] - y_min) // cell_size_y)
+            cell_key = (i, j)
+            cell_map.setdefault(cell_key, []).append((padstacks_array[idx], pos))
+
+        for (i, j), items in cell_map.items():
+            # cell center
+            cell_x_min = x_min + i * cell_size_x
+            cell_y_min = y_min + j * cell_size_y
+            cell_x_mid = cell_x_min + 0.5 * cell_size_x
+            cell_y_mid = cell_y_min + 0.5 * cell_size_y
+
+            grid.append(
+                [
+                    [
+                        cell_x_min,
+                        cell_x_min + cell_size_x,
+                        cell_x_min + cell_size_x,
+                        cell_x_min,
+                        cell_x_min,
+                    ],
+                    [
+                        cell_y_min,
+                        cell_y_min,
+                        cell_y_min + cell_size_y,
+                        cell_y_min + cell_size_y,
+                        cell_y_min,
+                    ],
+                ]
+            )
+
+            # Find closest via to cell center
+            distances = [np.linalg.norm(pos - [cell_x_mid, cell_y_mid]) for _, pos in items]
+            closest_idx = np.argmin(distances)
+            to_keep.add(items[closest_idx][0])
+
+        if delete:
+            to_delete = set(padstacks) - to_keep
+            for _id in to_delete:
+                all_instances[_id].delete()
+
+        return list(to_keep), grid

pyedb/grpc/database/primitive/bondwire.py

@@ -21,10 +21,10 @@
 # SOFTWARE.
 
 from ansys.edb.core.primitive.bondwire import (
+    Bondwire as GrpcBondWire,
     BondwireCrossSectionType as GrpcBondwireCrossSectionType,
+    BondwireType as GrpcBondWireType,
 )
-from ansys.edb.core.primitive.bondwire import Bondwire as GrpcBondWire
-from ansys.edb.core.primitive.bondwire import BondwireType as GrpcBondWireType
 
 from pyedb.grpc.database.utility.value import Value
 
@@ -36,6 +36,7 @@ class Bondwire(GrpcBondWire):
         super().__init__(edb_object.msg)
         self._pedb = _pedb
         self._edb_object = edb_object
+        # TODO add create and delete methods to keep cache in sync
 
     @property
     def material(self):
@@ -53,25 +54,6 @@ class Bondwire(GrpcBondWire):
     def material(self, value):
         self.set_material(value)
 
-    # def __create(self, **kwargs):
-    #     return Bondwire.create(
-    #         self._pedb.layout,
-    #         kwargs.get("net"),
-    #         self._bondwire_type[kwargs.get("bondwire_type")],
-    #         kwargs.get("definition_name"),
-    #         kwargs.get("placement_layer"),
-    #         kwargs.get("width"),
-    #         kwargs.get("material"),
-    #         kwargs.get("start_context"),
-    #         kwargs.get("start_layer_name"),
-    #         kwargs.get("start_x"),
-    #         kwargs.get("start_y"),
-    #         kwargs.get("end_context"),
-    #         kwargs.get("end_layer_name"),
-    #         kwargs.get("end_x"),
-    #         kwargs.get("end_y"),
-    #     )
-
     @property
     def type(self):
         """str: Bondwire-type of a bondwire object. Supported values for setter: `"apd"`, `"jedec4"`, `"jedec5"`,
@@ -120,28 +102,6 @@ class Bondwire(GrpcBondWire):
     def cross_section_height(self, cross_section_height):
         super(Bondwire, self.__class__).cross_section_height.__set__(self, Value(cross_section_height))
 
-    # @property
-    # def trajectory(self):
-    #     """Get trajectory parameters of a bondwire object.
-    #
-    #     Returns
-    #     -------
-    #     tuple[float, float, float, float]
-    #
-    #     Returns a tuple of the following format:
-    #     **(x1, y1, x2, y2)**
-    #     **x1** : X value of the start point.
-    #     **y1** : Y value of the start point.
-    #     **x1** : X value of the end point.
-    #     **y1** : Y value of the end point.
-    #     """
-    #     return [Value(i) for i in self.get_traj()]
-    #
-    # @trajectory.setter
-    # def trajectory(self, value):
-    #     values = [Value(i) for i in value]
-    #     self.set_traj(values[0], values[1], values[2], values[3])
-
     @property
     def width(self):
         """:class:`Value <ansys.edb.utility.Value>`: Width of a bondwire object.
@@ -156,27 +116,3 @@ class Bondwire(GrpcBondWire):
     @width.setter
     def width(self, width):
         super(Bondwire, self.__class__).width.__set__(self, Value(width))
-
-    # @property
-    # def start_elevation(self):
-    #     layer = self.get_start_elevation(self._pedb.active_cell)
-    #     return layer.name
-    #
-    # @start_elevation.setter
-    # def start_elevation(self, layer):
-    #     if not layer in self._pedb.stackup.layers:
-    #         return
-    #     layer = self._pedb.stackup.layers[layer]
-    #     self.set_start_elevation(self._pedb.active_cell, layer)
-    #
-    # @property
-    # def end_elevation(self):
-    #     layer = self.get_end_elevation(self._pedb.active_cell)
-    #     return layer.name
-    #
-    # @end_elevation.setter
-    # def end_elevation(self, layer):
-    #     if not layer in self._pedb.stackup.layers:
-    #         return
-    #     layer = self._pedb.stackup.layers[layer]
-    #     self.set_end_elevation(self._pedb.active_cell, layer)

pyedb/grpc/database/primitive/circle.py

@@ -21,18 +21,57 @@
 # SOFTWARE.
 
 
+from typing import Union
+
 from ansys.edb.core.primitive.circle import Circle as GrpcCircle
 
+from pyedb.grpc.database.layers.layer import Layer
+from pyedb.grpc.database.net.net import Net
 from pyedb.grpc.database.primitive.primitive import Primitive
 from pyedb.grpc.database.utility.value import Value
 
 
 class Circle(GrpcCircle, Primitive):
-    def __init__(self, pedb, edb_object):
-        GrpcCircle.__init__(self, edb_object.msg)
-        Primitive.__init__(self, pedb, edb_object)
+    def __init__(self, pedb, edb_object=None):
+        if edb_object:
+            GrpcCircle.__init__(self, edb_object.msg)
+            Primitive.__init__(self, pedb, edb_object)
         self._pedb = pedb
 
+    def create(
+        self,
+        layout=None,
+        layer: Union[str, Layer] = None,
+        net: Union[str, Net, None] = None,
+        center_x: float = None,
+        center_y: float = None,
+        radius: float = 0.0,
+    ):
+        if not layout:
+            layout = self._pedb.active_layout
+        if not layer:
+            raise ValueError("Layer must be provided to create a circle.")
+        if center_x is None or center_y is None:
+            raise ValueError("Center x and y values must be provided to create a circle.")
+        edb_object = GrpcCircle.create(
+            layout=layout,
+            layer=layer,
+            net=net,
+            center_x=Value(center_x),
+            center_y=Value(center_y),
+            radius=Value(radius),
+        )
+        new_circle = Circle(self._pedb, edb_object)
+        GrpcCircle.__init__(self, edb_object.msg)
+        Primitive.__init__(self, self._pedb, edb_object)
+        self._pedb.modeler._add_primitive(new_circle)
+        return new_circle
+
+    def delete(self):
+        """Delete the circle from the layout."""
+        self._pedb.modeler._remove_primitive(self)
+        super().delete()
+
     def get_parameters(self) -> tuple[float, float, float]:
         """Returns parameters.