Skip to content

openg2g.utils

openg2g.utils

Shared utility functions.

split_integer_evenly(n, k)

Split integer n into k non-negative integers whose sum is n, differing by at most 1.

Example:

split_integer_evenly(10, 3)  # -> [4, 3, 3]
split_integer_evenly(2, 5)   # -> [1, 1, 0, 0, 0]
Source code in openg2g/utils.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
def split_integer_evenly(n: int, k: int) -> list[int]:
    """Split integer *n* into *k* non-negative integers whose sum is *n*,
    differing by at most 1.

    Example:

    ```python
    split_integer_evenly(10, 3)  # -> [4, 3, 3]
    split_integer_evenly(2, 5)   # -> [1, 1, 0, 0, 0]
    ```
    """
    q, r = divmod(int(n), int(k))
    return [q + (1 if i < r else 0) for i in range(k)]

smooth_bump(t, t_center, half_width)

Smooth bump function: 1 at center, 0 outside half_width.

Source code in openg2g/utils.py 
21
22
23
24
25
26
27
def smooth_bump(t: float, t_center: float, half_width: float) -> float:
    """Smooth bump function: 1 at center, 0 outside half_width."""
    dt = abs(t - t_center)
    if dt >= half_width:
        return 0.0
    x = dt / half_width
    return (1 - x * x) ** 2