Skip to content

cellects.gui.numba_precompilation

cellects.gui.numba_precompilation

This module precompile all njit decorated functions used by Cellects.

warming_up_numba_functions(loading)

This function calls all njit decorated functions used by Cellects.

It also update a loading object to display a progress bar during launching.

Source code in src/cellects/gui/numba_precompilation.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
def warming_up_numba_functions(loading):
    """
    This function calls all njit decorated functions used by Cellects.

    It also update a loading object to display a progress bar during launching.
    """
    integer = np.uint8(1)
    vect = np.zeros((1), dtype=np.float64)
    img = np.zeros((1, 1, 3), dtype=np.uint8)
    b_img = np.zeros((1, 1), dtype=np.uint8)
    b_vid = np.zeros((1, 1, 1), dtype=np.uint8)
    c_space_dict = Dict()
    c_space_dict['bgr'] = List([0, 0, 0])
    all_c_spaces = Dict()
    all_c_spaces['bgr'] = img
    subtract_background = img[:, :, 0]
    mo = Dict()
    mo['m00'] = 1.0
    mo["m10"] = 2.
    mo["m01"] = 3.
    mo["m20"] = 4.
    mo["m02"] = 1.
    mo["m11"] = 5.

    loading.add_progress()
    _ = combine_color_spaces(c_space_dict, all_c_spaces, subtract_background)

    loading.add_progress()
    _ = otsu_thresholding(img)

    loading.add_progress()
    _ = sum_of_abs_differences(vect, vect)

    loading.add_progress()
    _ = bracket_to_uint8_image_contrast(img)

    loading.add_progress()
    _, _ = get_power_dists(img, 5.0, 5.0, 2)

    loading.add_progress()
    _, _ = get_var(mo, img[:, :, 0], vect, vect)

    loading.add_progress()
    _, _ = get_skewness_kurtosis(1.5, 2.0, 0.5, 0.75, 3)

    loading.add_progress()
    _, _, _, _, _ = get_inertia_axes(mo)

    loading.add_progress()
    _ = get_newly_explored_area(b_vid)

    loading.add_progress()
    _, _ = _get_counts_jit(integer, vect, vect)

    loading.add_progress()
    get_line_points((0, 0), (1, 0))

    loading.add_progress()
    _ = reduce_image_size_for_speed(b_img)

    loading.add_progress()
    indexed_shapes = np.array([1, 0, 1], dtype=np.int32)

    loading.add_progress()
    _ = keep_largest_shape(indexed_shapes)

    loading.add_progress()
    _ = nonzero_to_set(b_img)

    loading.add_progress()
    _ = greater_along_first_axis(b_img, vect)

    loading.add_progress()
    _ = less_along_first_axis(b_img, vect)