Skip to content

Commit 5ff61c7

Browse files
Add docs to get_auto_lims
1 parent 3728499 commit 5ff61c7

File tree

1 file changed

+36
-1
lines changed

1 file changed

+36
-1
lines changed

src/probeinterface/utils.py

Lines changed: 36 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,42 @@ def generate_unique_ids(min: int, max: int, n: int, trials: int = 20) -> np.arra
139139
return ids
140140

141141

142-
def get_auto_lims(probe, margin=40):
142+
def get_auto_lims(probe: Probe, margin: float = 40.0) -> tuple[float, float, float]:
143+
"""
144+
Automatically compute a probe boundaries given its contour and an optional
145+
margin. Both planar and tridimensional probes are accepted.
146+
147+
Parameters
148+
----------
149+
probe : Probe
150+
The probe whose limits are to be computed.
151+
margin : float, default: 40
152+
An isotropic margin added to the exact probe boundaries.
153+
154+
Returns
155+
-------
156+
lims : a tuple containing xlims, ylims, and zlims. If the provided probe
157+
is planar, then zlims is None.
158+
159+
"""
160+
161+
"""
162+
Compute the boundaries of a given probe, considering its contour and an optional margin.
163+
The function is designed to handle both planar and tridimensional probes.
164+
165+
Parameters
166+
----------
167+
probe : Probe
168+
The probe for which the limits are to be computed.
169+
margin : float, default: 40
170+
An isotropic margin that is added to the exact probe boundaries.
171+
172+
Returns
173+
-------
174+
lims : a tuple containing the limits in the x, y, and z directions
175+
(xlims, ylims, zlims). If the provided probe is planar, then
176+
zlims is None.
177+
"""
143178
positions = probe.contact_positions
144179
planar_contour = probe.probe_planar_contour
145180

0 commit comments

Comments
 (0)