def test_ui_radio_button_default(interactive=False):
filename = "test_ui_radio_button"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
radio_button_test = ui.RadioButton(
labels=["option 1", "option 2\nOption 2", "option 3", "option 4"],
position=(10, 10), checked_labels=[])
old_positions = []
for option in radio_button_test.options.values():
old_positions.append(option.position)
old_positions = np.asarray(old_positions)
radio_button_test.position = (100, 100)
new_positions = []
for option in radio_button_test.options.values():
new_positions.append(option.position)
new_positions = np.asarray(new_positions)
npt.assert_allclose(new_positions - old_positions,
90 * np.ones((4, 2)))
selected_option = []
def _on_change(radio_button):
selected_option.append(radio_button.checked_labels)
# Set up a callback when selection changes
radio_button_test.on_change = _on_change
event_counter = EventCounter()
event_counter.monitor(radio_button_test)
# Create a show manager and record/play events.
show_manager = window.ShowManager(size=(600, 600),
title="FURY Checkbox")
show_manager.scene.add(radio_button_test)
# Recorded events:
# 1. Click on button of option 1.
# 2. Click on button of option 2.
# 3. Click on button of option 2.
# 4. Click on text of option 2.
# 5. Click on button of option 1.
# 6. Click on text of option 3.
# 7. Click on button of option 4.
# 8. Click on text of option 4.
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
# Check if the right options were selected.
expected = [['option 1'], ['option 2\nOption 2'], ['option 2\nOption 2'],
['option 2\nOption 2'], ['option 1'], ['option 3'],
['option 4'], ['option 4']]
npt.assert_equal(len(selected_option), len(expected))
assert_arrays_equal(selected_option, expected)
del show_manager
if interactive:
radio_button_test = ui.RadioButton(
labels=["option 1", "option 2\nOption 2", "option 3", "option 4"],
position=(100, 100), checked_labels=[])
showm = window.ShowManager(size=(600, 600))
showm.scene.add(radio_button_test)
showm.start()
def test_ui_listbox_2d(interactive=False):
filename = "test_ui_listbox_2d"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
# Values that will be displayed by the listbox.
values = list(range(1, 42 + 1))
values.append("A Very Very Long Item To Test Text Overflow of List Box 2D")
if interactive:
listbox = ui.ListBox2D(values=values,
size=(500, 500),
multiselection=True,
reverse_scrolling=False,
background_opacity=0.3)
listbox.center = (300, 300)
listbox.panel.opacity = 0.2
show_manager = window.ShowManager(size=(600, 600),
title="FURY ListBox")
show_manager.initialize()
show_manager.scene.add(listbox)
show_manager.start()
# Recorded events:
# 1. Click on 1
# 2. Ctrl + click on 2,
# 3. Ctrl + click on 2.
# 4. Use scroll bar to scroll to the bottom.
# 5. Click on "A Very Very Long Item...".
# 6. Use scroll bar to scroll to the top.
# 7. Click on 1
# 8. Use mouse wheel to scroll down.
# 9. Shift + click on "A Very Very Long Item...".
# 10. Use mouse wheel to scroll back up.
listbox = ui.ListBox2D(values=values,
size=(500, 500),
multiselection=True,
reverse_scrolling=False)
listbox.center = (300, 300)
# We will collect the sequence of values that have been selected.
selected_values = []
def _on_change():
selected_values.append(list(listbox.selected))
# Set up a callback when selection changes.
listbox.on_change = _on_change
# Assign the counter callback to every possible event.
event_counter = EventCounter()
event_counter.monitor(listbox)
show_manager = window.ShowManager(size=(600, 600),
title="FURY ListBox")
show_manager.scene.add(listbox)
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
# Check if the right values were selected.
expected = [[1], [1, 2], [1], ["A Very Very Long Item To \
Test Text Overflow of List Box 2D"], [1], values]
npt.assert_equal(len(selected_values), len(expected))
assert_arrays_equal(selected_values, expected)
# Test without multiselection enabled.
listbox.multiselection = False
del selected_values[:] # Clear the list.
show_manager.play_events_from_file(recording_filename)
# Check if the right values were selected.
expected = [[1], [2], [2], ["A Very Very Long Item To \
Test Text Overflow of List Box 2D"], [1], ["A Very Very Long Item To Test \
Text Overflow of List Box 2D"]]
npt.assert_equal(len(selected_values), len(expected))
assert_arrays_equal(selected_values, expected)
def test_ui_button_panel(recording=False):
filename = "test_ui_button_panel"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
# Rectangle
rectangle_test = ui.Rectangle2D(size=(10, 10))
another_rectangle_test = ui.Rectangle2D(size=(1, 1))
# Button
fetch_viz_icons()
icon_files = []
icon_files.append(('stop', read_viz_icons(fname='stop2.png')))
icon_files.append(('play', read_viz_icons(fname='play3.png')))
button_test = ui.Button2D(icon_fnames=icon_files)
button_test.center = (20, 20)
def make_invisible(i_ren, _obj, button):
# i_ren: CustomInteractorStyle
# obj: vtkActor picked
# button: Button2D
button.set_visibility(False)
i_ren.force_render()
i_ren.event.abort()
def modify_button_callback(i_ren, _obj, button):
# i_ren: CustomInteractorStyle
# obj: vtkActor picked
# button: Button2D
button.next_icon()
i_ren.force_render()
button_test.on_right_mouse_button_pressed = make_invisible
button_test.on_left_mouse_button_pressed = modify_button_callback
button_test.scale((2, 2))
button_color = button_test.color
button_test.color = button_color
# TextBlock
text_block_test = ui.TextBlock2D()
text_block_test.message = 'TextBlock'
text_block_test.color = (0, 0, 0)
# Panel
panel = ui.Panel2D(size=(300, 150),
position=(290, 15),
color=(1, 1, 1),
align="right")
panel.add_element(rectangle_test, (290, 135))
panel.add_element(button_test, (0.1, 0.1))
panel.add_element(text_block_test, (0.7, 0.7))
npt.assert_raises(ValueError, panel.add_element, another_rectangle_test,
(10., 0.5))
npt.assert_raises(ValueError, panel.add_element, another_rectangle_test,
(-0.5, 0.5))
# Assign the counter callback to every possible event.
event_counter = EventCounter()
event_counter.monitor(button_test)
event_counter.monitor(panel.background)
current_size = (600, 600)
show_manager = window.ShowManager(size=current_size, title="FURY Button")
show_manager.scene.add(panel)
if recording:
show_manager.record_events_to_file(recording_filename)
print(list(event_counter.events_counts.items()))
event_counter.save(expected_events_counts_filename)
else:
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
def test_ui_checkbox_initial_state(interactive=False):
filename = "test_ui_checkbox"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
checkbox_test = ui.Checkbox(labels=["option 1", "option 2\nOption 2",
"option 3", "option 4"],
position=(100, 100),
checked_labels=["option 1", "option 4"])
# Collect the sequence of options that have been checked in this list.
selected_options = []
def _on_change(checkbox):
selected_options.append(list(checkbox.checked_labels))
# Set up a callback when selection changes
checkbox_test.on_change = _on_change
event_counter = EventCounter()
event_counter.monitor(checkbox_test)
# Create a show manager and record/play events.
show_manager = window.ShowManager(size=(600, 600),
title="FURY Checkbox")
show_manager.scene.add(checkbox_test)
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
# Recorded events:
# 1. Click on button of option 1.
# 2. Click on button of option 2.
# 3. Click on button of option 1.
# 4. Click on text of option 3.
# 5. Click on text of option 1.
# 6. Click on button of option 4.
# 7. Click on text of option 1.
# 8. Click on text of option 2.
# 9. Click on text of option 4.
# 10. Click on button of option 3.
# Check if the right options were selected.
expected = [['option 4'], ['option 4', 'option 2\nOption 2'],
['option 4', 'option 2\nOption 2', 'option 1'],
['option 4', 'option 2\nOption 2', 'option 1', 'option 3'],
['option 4', 'option 2\nOption 2', 'option 3'],
['option 2\nOption 2', 'option 3'],
['option 2\nOption 2', 'option 3', 'option 1'],
['option 3', 'option 1'], ['option 3', 'option 1', 'option 4'],
['option 1', 'option 4']]
npt.assert_equal(len(selected_options), len(expected))
assert_arrays_equal(selected_options, expected)
del show_manager
if interactive:
checkbox_test = ui.Checkbox(labels=["option 1", "option 2\nOption 2",
"option 3", "option 4"],
position=(100, 100), checked_labels=[])
showm = window.ShowManager(size=(600, 600))
showm.scene.add(checkbox_test)
showm.start()
def test_grid_ui(interactive=False):
vol1 = np.zeros((100, 100, 100))
vol1[25:75, 25:75, 25:75] = 100
colors = distinguishable_colormap(nb_colors=3)
contour_actor1 = actor.contour_from_roi(vol1, np.eye(4),
colors[0], 1.)
vol2 = np.zeros((100, 100, 100))
vol2[25:75, 25:75, 25:75] = 100
contour_actor2 = actor.contour_from_roi(vol2, np.eye(4),
colors[1], 1.)
vol3 = np.zeros((100, 100, 100))
vol3[25:75, 25:75, 25:75] = 100
contour_actor3 = actor.contour_from_roi(vol3, np.eye(4),
colors[2], 1.)
scene = window.Scene()
actors = []
texts = []
actors.append(contour_actor1)
text_actor1 = actor.text_3d('cube 1', justification='center')
texts.append(text_actor1)
actors.append(contour_actor2)
text_actor2 = actor.text_3d('cube 2', justification='center')
texts.append(text_actor2)
actors.append(contour_actor3)
text_actor3 = actor.text_3d('cube 3', justification='center')
texts.append(text_actor3)
actors.append(shallow_copy(contour_actor1))
text_actor1 = actor.text_3d('cube 4', justification='center')
texts.append(text_actor1)
actors.append(shallow_copy(contour_actor2))
text_actor2 = actor.text_3d('cube 5', justification='center')
texts.append(text_actor2)
actors.append(shallow_copy(contour_actor3))
text_actor3 = actor.text_3d('cube 6', justification='center')
texts.append(text_actor3)
actors.append(shallow_copy(contour_actor1))
text_actor1 = actor.text_3d('cube 7', justification='center')
texts.append(text_actor1)
actors.append(shallow_copy(contour_actor2))
text_actor2 = actor.text_3d('cube 8', justification='center')
texts.append(text_actor2)
actors.append(shallow_copy(contour_actor3))
text_actor3 = actor.text_3d('cube 9', justification='center')
texts.append(text_actor3)
counter = itertools.count()
show_m = window.ShowManager(scene)
show_m.initialize()
def timer_callback(_obj, _event):
cnt = next(counter)
show_m.scene.zoom(1)
show_m.render()
if cnt == 10:
show_m.exit()
show_m.destroy_timers()
# show the grid with the captions
grid_ui = ui.GridUI(actors=actors, captions=texts,
caption_offset=(0, -50, 0),
cell_padding=(60, 60), dim=(3, 3),
rotation_axis=(1, 0, 0))
scene.add(grid_ui)
show_m.add_timer_callback(True, 200, timer_callback)
show_m.start()
arr = window.snapshot(scene)
report = window.analyze_snapshot(arr)
npt.assert_equal(report.objects > 9, True)
# this needs to happen automatically when start() ends.
for act in actors:
act.RemoveAllObservers()
filename = "test_grid_ui"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
current_size = (900, 600)
scene = window.Scene()
show_manager = window.ShowManager(scene,
size=current_size,
title="FURY GridUI")
show_manager.initialize()
grid_ui2 = ui.GridUI(actors=actors, captions=texts,
caption_offset=(0, -50, 0),
cell_padding=(60, 60), dim=(3, 3),
rotation_axis=None)
scene.add(grid_ui2)
event_counter = EventCounter()
event_counter.monitor(grid_ui2)
if interactive:
show_manager.start()
recording = False
if recording:
# Record the following events
# 1. Left click on top left box (will rotate the box)
show_manager.record_events_to_file(recording_filename)
# print(list(event_counter.events_counts.items()))
event_counter.save(expected_events_counts_filename)
else:
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
def test_ui_tab_ui(interactive=False):
filename = "test_ui_tab_ui"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
tab_ui = ui.TabUI(position=(50, 50), size=(300, 300), nb_tabs=3,
draggable=True)
tab_ui.tabs[0].title = "Tab 1"
tab_ui.tabs[1].title = "Tab 2"
tab_ui.tabs[2].title = "Tab 3"
tab_ui.add_element(0, ui.Checkbox(["Option 1", "Option 2"]), (0.5, 0.5))
tab_ui.add_element(1, ui.LineSlider2D(), (0.0, 0.5))
tab_ui.add_element(2, ui.TextBlock2D(), (0.5, 0.5))
with npt.assert_raises(IndexError):
tab_ui.add_element(3, ui.TextBlock2D(), (0.5, 0.5, 0.5))
with npt.assert_raises(IndexError):
tab_ui.remove_element(3, ui.TextBlock2D())
with npt.assert_raises(IndexError):
tab_ui.update_element(3, ui.TextBlock2D(), (0.5, 0.5, 0.5))
npt.assert_equal("Tab 1", tab_ui.tabs[0].title)
npt.assert_equal("Tab 2", tab_ui.tabs[1].title)
npt.assert_equal("Tab 3", tab_ui.tabs[2].title)
npt.assert_equal(3, tab_ui.nb_tabs)
collapses = itertools.count()
changes = itertools.count()
def collapse(tab_ui):
if tab_ui.collapsed:
next(collapses)
def tab_change(tab_ui):
next(changes)
tab_ui.on_change = tab_change
tab_ui.on_collapse = collapse
event_counter = EventCounter()
event_counter.monitor(tab_ui)
current_size = (800, 800)
show_manager = window.ShowManager(
size=current_size, title="Tab UI Test")
show_manager.scene.add(tab_ui)
if interactive:
show_manager.record_events_to_file(recording_filename)
print(list(event_counter.events_counts.items()))
event_counter.save(expected_events_counts_filename)
else:
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
npt.assert_equal(0, tab_ui.active_tab_idx)
npt.assert_equal(11, next(changes))
npt.assert_equal(5, next(collapses))
def test_ui_combobox_2d(interactive=False):
filename = "test_ui_combobox_2d"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
values = ["An Item" + str(i) for i in range(0, 5)]
new_values = ["An Item5", "An Item6"]
combobox = ui.ComboBox2D(
items=values, position=(400, 400), size=(300, 200))
# Assign the counter callback to every possible event.
event_counter = EventCounter()
event_counter.monitor(combobox)
current_size = (800, 800)
show_manager = window.ShowManager(
size=current_size, title="ComboBox UI Example")
show_manager.scene.add(combobox)
values.extend(new_values)
combobox.append_item(*new_values)
npt.assert_equal(values, combobox.items)
values.append("An Item7")
combobox.append_item("An Item7")
npt.assert_equal(values, combobox.items)
values.append("An Item8")
values.append("An Item9")
combobox.append_item("An Item8", "An Item9")
npt.assert_equal(values, combobox.items)
complex_list = [[0], (1, [[2, 3], 4], 5)]
combobox.append_item(*complex_list)
values.extend([str(i) for i in range(6)])
npt.assert_equal(values, combobox.items)
invalid_item = {"Hello": 1, "World": 2}
npt.assert_raises(TypeError, combobox.append_item, invalid_item)
npt.assert_equal(values, combobox.items)
npt.assert_equal((60, 60), combobox.drop_button_size)
npt.assert_equal([300, 140], combobox.drop_menu_size)
npt.assert_equal([300, 200], combobox.size)
ui.ComboBox2D(items=values, draggable=False)
if interactive:
show_manager.record_events_to_file(recording_filename)
print(list(event_counter.events_counts.items()))
event_counter.save(expected_events_counts_filename)
else:
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
npt.assert_equal("An Item1", combobox.selected_text)
npt.assert_equal(1, combobox.selected_text_index)
combobox.resize((450, 300))
npt.assert_equal((360, 90), combobox.text_block_size)
npt.assert_equal((90, 90), combobox.drop_button_size)
npt.assert_equal((450, 210), combobox.drop_menu_size)
def test_ui_file_menu_2d(interactive=False):
filename = "test_ui_file_menu_2d"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
tmpdir = InTemporaryDirectory()
test_dir = os.path.join(tmpdir.name, "testdir")
os.mkdir(test_dir)
os.chdir(test_dir)
os.mkdir(os.path.join(test_dir, "tempdir"))
for i in range(10):
open(os.path.join(test_dir, "tempdir", f"test{i}.txt"),
'wt').close()
open("testfile.txt", 'wt').close()
filemenu = ui.FileMenu2D(size=(500, 500), extensions=["txt"],
directory_path=os.getcwd())
# We will collect the sequence of files that have been selected.
selected_files = []
def _on_change():
selected_files.append(list(filemenu.listbox.selected))
# Set up a callback when selection changes.
filemenu.listbox.on_change = _on_change
# Assign the counter callback to every possible event.
event_counter = EventCounter()
event_counter.monitor(filemenu)
# Create a show manager and record/play events.
show_manager = window.ShowManager(size=(600, 600),
title="FURY FileMenu")
show_manager.scene.add(filemenu)
# Recorded events:
# 1. Click on 'testfile.txt'
# 2. Click on 'tempdir/'
# 3. Click on 'test0.txt'.
# 4. Shift + Click on 'test6.txt'.
# 5. Click on '../'.
# 2. Click on 'testfile.txt'.
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
# Check if the right files were selected.
expected = [["testfile.txt"], ["tempdir"], ["test0.txt"],
["test0.txt", "test1.txt", "test2.txt", "test3.txt",
"test4.txt", "test5.txt", "test6.txt"],
["../"], ["testfile.txt"]]
npt.assert_equal(len(selected_files), len(expected))
assert_arrays_equal(selected_files, expected)
if interactive:
filemenu = ui.FileMenu2D(size=(500, 500),
directory_path=os.getcwd())
show_manager = window.ShowManager(size=(600, 600),
title="FURY FileMenu")
show_manager.scene.add(filemenu)
show_manager.start()
shutil.rmtree(os.path.join(tmpdir.name, "testdir"))
def test_ui_checkbox_default(recording=False):
filename = "test_ui_checkbox_initial_state"
recording_filename = pjoin(DATA_DIR, filename + ".log.gz")
expected_events_counts_filename = pjoin(DATA_DIR, filename + ".json")
checkbox_test = ui.Checkbox(labels=["option 1", "option 2\nOption 2",
"option 3", "option 4"],
position=(10, 10), checked_labels=[])
old_positions = []
for option in checkbox_test.options.values():
old_positions.append(option.position)
old_positions = np.asarray(old_positions)
checkbox_test.position = (100, 100)
new_positions = []
for option in checkbox_test.options.values():
new_positions.append(option.position)
new_positions = np.asarray(new_positions)
npt.assert_allclose(new_positions - old_positions,
90.0 * np.ones((4, 2)))
# Collect the sequence of options that have been checked in this list.
selected_options = []
def _on_change(checkbox):
selected_options.append(list(checkbox.checked_labels))
# Set up a callback when selection changes
checkbox_test.on_change = _on_change
event_counter = EventCounter()
event_counter.monitor(checkbox_test)
# Create a show manager and record/play events.
show_manager = window.ShowManager(size=(600, 600),
title="FURY Checkbox")
show_manager.scene.add(checkbox_test)
if recording:
show_manager.record_events_to_file(recording_filename)
print(list(event_counter.events_counts.items()))
event_counter.save(expected_events_counts_filename)
else:
# Recorded events:
# 1. Click on button of option 1.
# 2. Click on button of option 2.
# 3. Click on button of option 1.
# 4. Click on text of option 3.
# 5. Click on text of option 1.
# 6. Click on button of option 4.
# 7. Click on text of option 1.
# 8. Click on text of option 2.
# 9. Click on text of option 4.
# 10. Click on button of option 3.
show_manager.play_events_from_file(recording_filename)
expected = EventCounter.load(expected_events_counts_filename)
event_counter.check_counts(expected)
# Check if the right options were selected.
expected = [['option 1'], ['option 1', 'option 2\nOption 2'],
['option 2\nOption 2'], ['option 2\nOption 2', 'option 3'],
['option 2\nOption 2', 'option 3', 'option 1'],
['option 2\nOption 2', 'option 3', 'option 1', 'option 4'],
['option 2\nOption 2', 'option 3', 'option 4'],
['option 3', 'option 4'], ['option 3'], []]
npt.assert_equal(len(selected_options), len(expected))
assert_arrays_equal(selected_options, expected)