solder ball height bug fix (#3776)

* fix

* fix

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix

---------

Co-authored-by: ring630 <@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

_unittest/test_00_EDB.py

@@ -953,7 +953,11 @@ def test_075_flatten_planes(self):
         assert self.edbapp.modeler.unite_polygons_on_layer("1_Top")
 
     def test_076_create_solder_ball_on_component(self):
-        assert self.edbapp.components.set_solder_ball("U1")
+        assert self.edbapp.components.set_solder_ball("U1", shape="Spheroid")
+        assert self.edbapp.components.set_solder_ball("U6", sball_height=None)
+        assert self.edbapp.components.set_solder_ball(
+            "U6", sball_height="100um", auto_reference_size=False, chip_orientation="chip_up"
+        )
 
     def test_077_add_void(self):
         plane_shape = self.edbapp.modeler.Shape("rectangle", pointA=["-5mm", "-5mm"], pointB=["5mm", "5mm"])

pyaedt/edb_core/components.py

@@ -1775,11 +1775,15 @@ def disable_rlc_component(self, component_name):
     def set_solder_ball(
         self,
         component="",
-        sball_diam="100um",
-        sball_height="150um",
+        sball_diam=None,
+        sball_height=None,
         shape="Cylinder",
         sball_mid_diam=None,
         chip_orientation="chip_down",
+        auto_reference_size=True,
+        reference_size_x=0,
+        reference_size_y=0,
+        reference_height=0,
     ):
         """Set cylindrical solder balls on a given component.
 
@@ -1799,7 +1803,14 @@ def set_solder_ball(
         chip_orientation : str, optional
             Give the chip orientation, ``"chip_down"`` or ``"chip_up"``. Default is ``"chip_down"``. Only applicable on
             IC model.
-
+        auto_reference_size : bool, optional
+            Whether to automatically set reference size.
+        reference_size_x : int, str, float, optional
+            X size of the reference. Applicable when auto_reference_size is False.
+        reference_size_y : int, str, float, optional
+            Y size of the reference. Applicable when auto_reference_size is False.
+        reference_height : int, str, float, optional
+            Height of the reference. Applicable when auto_reference_size is False.
         Returns
         -------
         bool
@@ -1816,52 +1827,55 @@ def set_solder_ball(
         if not isinstance(component, self._pedb.edb_api.cell.hierarchy.component):
             edb_cmp = self.get_component_by_name(component)
             cmp = self.instances[component]
-        else:
+        else:  # pragma: no cover
             edb_cmp = component
             cmp = self.instances[edb_cmp.GetName()]
-        if edb_cmp:
-            cmp_type = edb_cmp.GetComponentType()
-            if not sball_diam:
-                pin1 = list(cmp.pins.values())[0].pin
-                pin_layers = pin1.GetPadstackDef().GetData().GetLayerNames()
-                pad_params = self._padstack.get_pad_parameters(pin=pin1, layername=pin_layers[0], pad_type=0)
-                _sb_diam = min([self._get_edb_value(val).ToDouble() for val in pad_params[1]])
-                sball_diam = _sb_diam
-            sball_height = round(self._edb.utility.Value(sball_diam).ToDouble(), 9) / 2
-            if not sball_mid_diam:
-                sball_mid_diam = sball_diam
-
-            if shape == "Cylinder":
-                sball_shape = self._edb.definition.SolderballShape.Cylinder
-            else:
-                sball_shape = self._edb.definition.SolderballShape.Spheroid
-
-            cmp_property = edb_cmp.GetComponentProperty().Clone()
-            if cmp_type == self._edb.definition.ComponentType.IC:
-                ic_die_prop = cmp_property.GetDieProperty().Clone()
-                ic_die_prop.SetType(self._edb.definition.DieType.FlipChip)
-                if chip_orientation.lower() == "chip_down":
-                    ic_die_prop.SetOrientation(self._edb.definition.DieOrientation.ChipDown)
-                if chip_orientation.lower() == "chip_up":
-                    ic_die_prop.SetOrientation(self._edb.definition.DieOrientation.ChipUp)
-                else:
-                    ic_die_prop.SetOrientation(self._edb.definition.DieOrientation.ChipDown)
-                cmp_property.SetDieProperty(ic_die_prop)
 
-            solder_ball_prop = cmp_property.GetSolderBallProperty().Clone()
-            solder_ball_prop.SetDiameter(self._get_edb_value(sball_diam), self._get_edb_value(sball_mid_diam))
-            solder_ball_prop.SetHeight(self._get_edb_value(sball_height))
+        cmp_type = edb_cmp.GetComponentType()
+        if not sball_diam:
+            pin1 = list(cmp.pins.values())[0].pin
+            pin_layers = pin1.GetPadstackDef().GetData().GetLayerNames()
+            pad_params = self._padstack.get_pad_parameters(pin=pin1, layername=pin_layers[0], pad_type=0)
+            _sb_diam = min([self._get_edb_value(val).ToDouble() for val in pad_params[1]])
+            sball_diam = _sb_diam
+        if sball_height:
+            sball_height = round(self._edb.utility.Value(sball_height).ToDouble(), 9)
+        else:
+            sball_height = round(self._edb.utility.Value(sball_diam).ToDouble(), 9) / 2
 
-            solder_ball_prop.SetShape(sball_shape)
-            cmp_property.SetSolderBallProperty(solder_ball_prop)
+        if not sball_mid_diam:
+            sball_mid_diam = sball_diam
 
-            port_prop = cmp_property.GetPortProperty().Clone()
-            port_prop.SetReferenceSizeAuto(True)
-            cmp_property.SetPortProperty(port_prop)
-            edb_cmp.SetComponentProperty(cmp_property)
-            return True
+        if shape == "Cylinder":
+            sball_shape = self._edb.definition.SolderballShape.Cylinder
         else:
-            return False
+            sball_shape = self._edb.definition.SolderballShape.Spheroid
+
+        cmp_property = edb_cmp.GetComponentProperty().Clone()
+        if cmp_type == self._edb.definition.ComponentType.IC:
+            ic_die_prop = cmp_property.GetDieProperty().Clone()
+            ic_die_prop.SetType(self._edb.definition.DieType.FlipChip)
+            if chip_orientation.lower() == "chip_up":
+                ic_die_prop.SetOrientation(self._edb.definition.DieOrientation.ChipUp)
+            else:
+                ic_die_prop.SetOrientation(self._edb.definition.DieOrientation.ChipDown)
+            cmp_property.SetDieProperty(ic_die_prop)
+
+        solder_ball_prop = cmp_property.GetSolderBallProperty().Clone()
+        solder_ball_prop.SetDiameter(self._get_edb_value(sball_diam), self._get_edb_value(sball_mid_diam))
+        solder_ball_prop.SetHeight(self._get_edb_value(sball_height))
+
+        solder_ball_prop.SetShape(sball_shape)
+        cmp_property.SetSolderBallProperty(solder_ball_prop)
+
+        port_prop = cmp_property.GetPortProperty().Clone()
+        port_prop.SetReferenceSizeAuto(auto_reference_size)
+        if not auto_reference_size:
+            port_prop.SetReferenceSize(self._pedb.edb_value(reference_size_x), self._pedb.edb_value(reference_size_y))
+            port_prop.SetReferenceHeight(self._pedb.edb_value(reference_height))
+        cmp_property.SetPortProperty(port_prop)
+        edb_cmp.SetComponentProperty(cmp_property)
+        return True
 
     @pyaedt_function_handler()
     def set_component_rlc(