def test_two_components(self):
components = np.zeros((10, 10, 20), dtype=np.uint8)
components[4:6, 4:6, 4:6] = 1
components[4:6, 4:6, 14:16] = 2
sprawl_area = np.zeros(components.shape, dtype=np.uint8)
sprawl_area[2:8, 2:8, 2:18] = True
sprawl_area[components > 0] = False
fdt = np.zeros(components.shape, dtype=np.float64)
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
res = mso.optimum_erosion_calculate(fdt, components, sprawl_area)
assert np.all(res == components)
fdt[:] = 2
fdt[:, :, 10] = 1
components2 = np.zeros(components.shape, dtype=np.uint8)
components2[2:8, 2:8, 2:10] = 1
components2[2:8, 2:8, 11:18] = 2
res = mso.optimum_erosion_calculate(fdt, components, sprawl_area)
assert np.all(res == components2)
fdt[5, 5, 10] = 2
res = mso.optimum_erosion_calculate(fdt, components, sprawl_area)
assert np.all(res == components)
def test_fdt(self):
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
components = np.zeros((10, 10, 10), dtype=np.uint8)
components[3:7, 3:7, 3:7] = 1
mso.set_neighbourhood(neigh, dist)
with pytest.raises(RuntimeError):
mso.calculate_FDT()
mso.set_components(components)
with pytest.raises(RuntimeError):
mso.calculate_FDT()
def test_bridges(self):
components = np.zeros((10, 20, 30), dtype=np.uint8)
components[4:6, 4:6, 4:6] = 1
components[4:6, 4:6, 24:26] = 2
components[4:6, 14:16, 14:16] = 3
sprawl_area = np.zeros(components.shape, dtype=np.uint8)
sprawl_area[2:9, 2:10, 2:28] = True
sprawl_area[2:9, 10:18, 11:20] = True
sprawl_area[components > 0] = False
fdt = np.ones(components.shape, dtype=np.float64) * 2
fdt[:, 10, :] = 1
fdt[:, :, 10] = 1
fdt[:, :, 20] = 1
comp = np.copy(components)
comp[2:9, 2:10, 2:10] = 1
comp[2:9, 2:10, 21:28] = 2
comp[2:9, 11:18, 11:20] = 3
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso = PyMSO()
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
res = mso.optimum_erosion_calculate(fdt, components, sprawl_area)
assert np.all(res == comp)
fdt2 = np.copy(fdt)
fdt2[5, 5, 10] = 2
comp2 = np.copy(comp)
comp2[5, 5, 10] = 1
comp2[2:9, 2:10, 11:20] = 1
res = mso.optimum_erosion_calculate(fdt2, components, sprawl_area)
assert np.all(res == comp2)
fdt2 = np.copy(fdt)
fdt2[5, 5, 20] = 2
comp2 = np.copy(comp)
comp2[5, 5, 20] = 2
comp2[2:9, 2:10, 11:20] = 2
res = mso.optimum_erosion_calculate(fdt2, components, sprawl_area)
assert np.all(res == comp2)
fdt2 = np.copy(fdt)
fdt2[5, 10, 15] = 2
comp2 = np.copy(comp)
comp2[5, 10, 15] = 3
comp2[2:9, 2:10, 11:20] = 3
res = mso.optimum_erosion_calculate(fdt2, components, sprawl_area)
assert np.all(res == comp2)
def test_fdt_simple(self):
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
components = np.zeros((3, 3, 3), dtype=np.uint8)
components[1, 1, 1] = 1
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
mso.set_mu_array(np.ones(components.shape))
arr = np.zeros(components.shape)
arr[(0, 0, 0, 0, 2, 2, 2, 2), (0, 0, 2, 2, 0, 0, 2, 2),
(0, 2, 0, 2, 0, 2, 0, 2)] = np.sqrt(3)
arr[(0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2),
(0, 1, 1, 2, 0, 0, 2, 2, 0, 1, 1, 2),
(1, 0, 2, 1, 0, 2, 0, 2, 1, 0, 2, 1), ] = np.sqrt(2)
arr[(0, 1, 1, 1, 1, 2), (1, 0, 1, 1, 2, 1), (1, 1, 0, 2, 1, 1)] = 1
res = mso.calculate_FDT()
assert np.all(res == arr)
def test_two_components_base(self):
components = np.zeros((10, 10, 20), dtype=np.uint8)
components[4:6, 4:6, 4:6] = 1
components[4:6, 4:6, 14:16] = 2
sprawl_area = np.zeros(components.shape, dtype=np.uint8)
sprawl_area[2:8, 2:8, 2:18] = True
sprawl_area[components > 0] = False
fdt = np.zeros(components.shape, dtype=np.float64)
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
mso.set_mu_array(np.ones(components.shape))
res = mso.constrained_dilation(fdt, components, sprawl_area)
assert np.all(res == components)
fdt[:] = 20
# fdt[:, :, 10] = 1
components2 = np.zeros(components.shape, dtype=np.uint8)
components2[2:8, 2:8, 2:10] = 1
components2[2:8, 2:8, 10:18] = 2
res = mso.constrained_dilation(fdt, components, sprawl_area)
assert np.all(res == components2)
def test_chain_component(self):
for i in range(2, 10):
components = np.zeros((10, 10, i * 10), dtype=np.uint8)
for j in range(i):
components[4:6, 4:6, (10 * j + 4):(10 * j + 6)] = j + 1
fdt = np.ones(components.shape, dtype=np.float64) * i * 10
sprawl_area = np.zeros(components.shape, dtype=np.uint8)
sprawl_area[2:8, 2:8, 2:(10 * i) - 2] = True
sprawl_area[components > 0] = False
components2 = np.zeros(components.shape, dtype=np.uint8)
for j in range(i):
components2[2:8, 2:8, (j * 10):(j + 1) * 10] = j + 1
components2[:, :, :2] = 0
components2[:, :, -2:] = 0
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1),
NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
mso.set_mu_array(np.ones(components.shape))
res = mso.constrained_dilation(fdt, components, sprawl_area)
assert np.all(res == components2)
def test_chain_component_base(self):
for i in range(2, 10):
components = np.zeros((4, 5, i * 5), dtype=np.uint8)
for j in range(i):
components[2, 2, 5 * j + 2] = j + 1
fdt = np.ones(components.shape, dtype=np.float64) * 2
for j in range(i - 1):
fdt[:, :, j * 5 + 4] = 1
sprawl_area = np.ones(components.shape, dtype=np.uint8)
sprawl_area[components > 0] = False
components2 = np.zeros(components.shape, dtype=np.uint8)
for j in range(i):
components2[:, :, (j * 5):(j * 5) + 4] = j + 1
components2[:, :, -1] = i
neigh, dist = calculate_distances_array((1, 1, 1),
NeighType.vertex)
mso = PyMSO()
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
res = mso.optimum_erosion_calculate(fdt, components, sprawl_area)
assert np.all(res == components2)
def _test_background_simple(self):
components = np.ones((20, 20, 20), dtype=np.uint8)
components[1:-1, 1:-1, 1:-1] = 0
components[9:11, 9:11, 9:11] = 2
mu_array = np.zeros(components.shape)
mu_array[1:-1, 1:-1, 1:-1] = 0.7
mu_array[3:-3, 3:-3, 3:-3] = 0.6
mu_array[5:-5, 5:-5, 5:-5] = 0.4
mu_array[6:-6, 6:-6, 6:-6] = 0.6
mu_array[8:-8, 8:-8, 8:-8] = 0.7
mu_array[components > 0] = 1.0
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_use_background(True)
mso.set_components(components, 3)
mso.set_mu_array(mu_array)
mso.set_components_num(3)
mso.run_MSO(10)
mso.get_result_catted()
def test_two_components_bridge(self):
components = np.zeros((10, 10, 20), dtype=np.uint8)
components[:] = 1
components[2:8, 2:8, 2:18] = 0
components[4:6, 4:6, 4:6] = 2
components[4:6, 4:6, 14:16] = 3
mu_arr = np.zeros(components.shape, dtype=np.float64)
mu_arr[components == 0] = 0.5
mu_arr[components > 1] = 1
mu_arr[2:8, 2:8, (9, 10)] = 0.1
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_components(components, 3)
mso.set_mu_array(mu_arr)
mso.set_components_num(3)
mso.run_MSO(10)
res = mso.get_result_catted()
assert mso.steps_done() == 2
arr = np.copy(components)
arr[arr == 1] = 0
arr[2:8, 2:8, 2:9] = 2
arr[2:8, 2:8, 11:18] = 3
assert np.all(res == arr)
def test_fdt_base(self):
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
components = np.zeros((10, 10, 10), dtype=np.uint8)
components[3:7, 3:7, 3:7] = 1
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
mso.set_mu_array(np.ones(components.shape))
res = mso.calculate_FDT()
arr = calculate_euclidean((components == 0).astype(np.uint8),
components, neigh, dist)
assert np.all(res == arr)
mso.set_mu_array(np.ones(components.shape) * 0.5)
res = mso.calculate_FDT()
assert np.all(res == arr * 0.5)
mu = np.ones(components.shape) * 0.5
mu[components > 0] = 1
mso.set_mu_array(mu)
arr *= 0.5
arr[arr > 0] += 0.25
arr[3:7, (0, 1, 2, 0, 1, 2, 9, 8, 7, 9, 8, 7),
(0, 1, 2, 9, 8, 7, 0, 1, 2, 9, 8, 7)] += np.sqrt(2) / 4 - 0.25
for i in range(3):
lb = i
ub = 9 - i
arr[lb, lb + 1:ub, (lb, ub)] += np.sqrt(2) / 4 - 0.25
arr[lb, (lb, ub), lb + 1:ub] += np.sqrt(2) / 4 - 0.25
arr[ub, lb + 1:ub, (lb, ub)] += np.sqrt(2) / 4 - 0.25
arr[ub, (lb, ub), lb + 1:ub] += np.sqrt(2) / 4 - 0.25
for el in itertools.product([lb, ub], repeat=3):
arr[el] += np.sqrt(3) / 4 - 0.25
for z, (y, x) in itertools.product([2, 7],
itertools.product([0, 9],
repeat=2)):
arr[z, y, x] += np.sqrt(2) / 4 - 0.25
for z, (y, x) in itertools.product([2, 7],
itertools.product([1, 8],
repeat=2)):
arr[z, y, x] += np.sqrt(2) / 4 - 0.25
for z, (y, x) in itertools.product([1, 8],
itertools.product([0, 9],
repeat=2)):
arr[z, y, x] += np.sqrt(2) / 4 - 0.25
res2 = mso.calculate_FDT()
assert np.allclose(res2, arr)
def test_high_fdt_two_components(self):
components = np.zeros((10, 10, 20), dtype=np.uint8)
components[4:6, 4:6, 4:6] = 1
components[4:6, 4:6, 14:16] = 2
sprawl_area = np.zeros(components.shape, dtype=np.uint8)
sprawl_area[2:8, 2:8, 2:18] = True
sprawl_area[components > 0] = False
fdt = np.ones(components.shape, dtype=np.float64) * 20
mso = PyMSO()
neigh, dist = calculate_distances_array((1, 1, 1), NeighType.vertex)
mso.set_neighbourhood(neigh, dist)
mso.set_components(components)
mso.set_mu_array(np.ones(components.shape))
components2 = np.zeros(components.shape, dtype=np.uint8)
components2[2:8, 2:8, 2:10] = 1
components2[2:8, 2:8, 10:18] = 2
fdt2 = np.copy(fdt)
fdt2[:, :, 7] = 25
components3 = np.copy(components2)
components3[2:8, 2:8, 8:18] = 2
components3[2:8, 2:8, 7] = 0
res = mso.constrained_dilation(fdt2, components, sprawl_area)
assert np.all(res == components3)
fdt2 = np.copy(fdt)
fdt2[:, :, 12] = 25
components3 = np.copy(components2)
components3[2:8, 2:8, 2:12] = 1
components3[2:8, 2:8, 12] = 0
res = mso.constrained_dilation(fdt2, components, sprawl_area)
assert np.all(res == components3)
fdt2 = np.copy(fdt)
fdt2[:, :, 12] = 25
fdt2[:, :, 7] = 25
components3 = np.copy(components2)
components3[2:8, 2:8, 7:13] = 0
res = mso.constrained_dilation(fdt2, components, sprawl_area)
assert np.all(res == components3)
def test_mso_construct():
data = PyMSO()
assert hasattr(data, "set_image")