Skip to content

Reference for pyronn/ct_reconstruction/helpers/phantoms/primitives_2d.py


pyronn.ct_reconstruction.helpers.phantoms.primitives_2d.circle 

circle(shape, pos, radius, value=1.0)

Creates a simple circle primitive.

Args: shape: Shape (in [Y, X]) pos: Center (in [Y, X]) from upper left corner radius: Radius value: Value

Returns:

Type Description

np.array filled with circle
Source code in pyronn/ct_reconstruction/helpers/phantoms/primitives_2d.py 
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
def circle(shape, pos, radius, value=1.0):
    """
        Creates a simple circle primitive.
    Args:
        shape:      Shape (in [Y, X])
        pos:        Center (in [Y, X]) from upper left corner
        radius:     Radius
        value:      Value

    Returns:
        np.array filled with circle
    """
    # create meshgrid of coords
    xx, yy = np.mgrid[:shape[0], :shape[1]].astype(dtype=np.float32)

    # calc squared distance to pos
    circle = (xx - pos[1]) ** 2 + (yy - pos[0]) ** 2

    return (circle <= radius ** 2) *  (np.float32)(value)




pyronn.ct_reconstruction.helpers.phantoms.primitives_2d.ellipse 

ellipse(shape, pos, half_axes, value=1.0, phi=0.0)

Creates a simple ellipse primitive.

Args: shape: Shape (in [Y, X]) pos: Center (in [Y, X]) from upper left corner half_axes: Half axes of the ellipse (in [b, a]) value: Value phi: Rotation Angle in radians

Returns:

Type Description

np.array filled with ellipse
Source code in pyronn/ct_reconstruction/helpers/phantoms/primitives_2d.py 
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
def ellipse(shape, pos, half_axes, value=1.0, phi=0.0):
    """
        Creates a simple ellipse primitive.
    Args:
        shape:          Shape (in [Y, X])
        pos:            Center (in [Y, X]) from upper left corner
        half_axes:      Half axes of the ellipse (in [b, a])
        value:          Value
        phi:            Rotation Angle in radians

    Returns:
        np.array filled with ellipse
    """
    # create meshgrid of coords
    xx, yy = np.mgrid[:shape[0], :shape[1]].astype(dtype=np.float32)

    # move to pos
    xc = (xx - pos[1])
    yc = (yy - pos[0])

    # rotate
    xx = xc * np.cos(phi) + yc * np.sin(phi)
    yy = yc * np.cos(phi) - xc * np.sin(phi)

    a = half_axes[1]
    b = half_axes[0]

    # calc squared distance to pos
    ellipse_points = (xx ** 2) / (a ** 2) + (yy ** 2) / (b ** 2)

    return (ellipse_points <= 1) * value




pyronn.ct_reconstruction.helpers.phantoms.primitives_2d.rect 

rect(shape, pos, size, value=1.0)

Creates a simple rect primitive.

Args: shape: Shape (in [Y, X]) pos: Pos (upper left corner) (in [Y, X]) from upper left corner size: Size (in [Y, X]) value: Value

Returns:

Type Description

np.array filled with rectangle
Source code in pyronn/ct_reconstruction/helpers/phantoms/primitives_2d.py 
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
def rect(shape, pos, size, value=1.0):
    """
        Creates a simple rect primitive.
    Args:
        shape:      Shape (in [Y, X])
        pos:        Pos (upper left corner) (in [Y, X]) from upper left corner
        size:       Size  (in [Y, X])
        value:      Value

    Returns:
        np.array filled with rectangle
    """
    # create array and populate it with value
    rectangle = np.zeros(shape, dtype=np.float32)
    rectangle[pos[0]:pos[0] + size[0], pos[1]:pos[1] + size[1]] = value

    return rectangle