def _make_repr_widget(self):
button_refresh = Button(description='Refresh',
tooltip='Get representation info')
button_update = Button(
description='Update',
tooltip='Update representation by updating rinfo box')
button_remove = Button(description='Remove',
tooltip='Remove current representation')
button_hide = Button(description='Hide',
tooltip='Hide/Show current representation')
button_center_selection = Button(description='Center',
tooltip='center selected atoms')
button_center_selection._ngl_name = 'button_center_selection'
bbox = HBox([
button_refresh, button_center_selection, button_hide, button_remove
])
repr_name_text = Text(value='', description='')
repr_name_text._ngl_name = 'repr_name_text'
repr_selection = Text(value='', description='')
repr_selection._ngl_name = 'repr_selection'
repr_selection.width = repr_name_text.width = DEFAULT_TEXT_WIDTH
repr_info_box = VBox([repr_name_text, repr_selection])
repr_info_box._ngl_name = 'repr_info_box'
component_slider = IntSlider(value=0, description='component')
component_slider._ngl_name = 'component_slider'
component_slider.visible = False
cvalue = ' '
component_dropdown = Dropdown(value=cvalue,
options=[
cvalue,
],
description='component')
component_dropdown._ngl_name = 'component_dropdown'
repr_slider = IntSlider(value=0,
description='representation',
width=DEFAULT_SLIDER_WIDTH)
repr_slider._ngl_name = 'repr_slider'
repr_slider.visible = True
repr_text_info = Textarea(value='',
description='representation parameters')
repr_text_info.visible = False
checkbox_repr_text = Checkbox(value=False,
description='show parameters')
checkbox_repr_text.visible = False
repr_text_box = VBox([checkbox_repr_text, repr_text_info])
repr_text_box._ngl_name = 'repr_text_box'
checkbox_reprlist = Checkbox(value=False, description='reprlist')
checkbox_reprlist._ngl_name = 'checkbox_reprlist'
reprlist_choices = self._make_repr_name_choices(
component_slider, repr_slider)
reprlist_choices._ngl_name = 'reprlist_choices'
def on_update_checkbox_reprlist(change):
reprlist_choices.visible = change['new']
checkbox_reprlist.observe(on_update_checkbox_reprlist, names='value')
def on_click_refresh(button):
self._view._request_repr_parameters(
component=component_slider.value, repr_index=repr_slider.value)
self._view._remote_call('requestReprInfo', target='Widget')
button_refresh.on_click(on_click_refresh)
def on_click_update(button):
parameters = json.loads(
repr_text_info.value.replace("False",
"false").replace("True", "true"))
self._view.update_representation(component=component_slider.value,
repr_index=repr_slider.value,
**parameters)
self._view._set_selection(repr_selection.value,
component=component_slider.value,
repr_index=repr_slider.value)
button_update.on_click(on_click_update)
def on_click_remove(button_remove):
self._view._remove_representation(component=component_slider.value,
repr_index=repr_slider.value)
self._view._request_repr_parameters(
component=component_slider.value, repr_index=repr_slider.value)
button_remove.on_click(on_click_remove)
def on_click_hide(button_hide):
component = component_slider.value
repr_index = repr_slider.value
if button_hide.description == 'Hide':
hide = True
button_hide.description = 'Show'
elif button_hide.description == 'Show':
hide = False
button_hide.description = 'Hide'
else:
raise ValueError("must be Hide or Show")
self._view._remote_call('setVisibilityForRepr',
target='Widget',
args=[component, repr_index, not hide])
button_hide.on_click(on_click_hide)
def on_click_center(center_selection):
self._view.center_view(selection=repr_selection.value,
component=component_slider.value)
button_center_selection.on_click(on_click_center)
def on_change_repr_name(change):
name = change['new'].strip()
old = change['old'].strip()
should_update = (self._real_time_update and old and name
and name in REPR_NAMES
and name != change['old'].strip())
if should_update:
component = component_slider.value
repr_index = repr_slider.value
self._view._remote_call(
'setRepresentation',
target='Widget',
args=[change['new'], {}, component, repr_index])
self._view._request_repr_parameters(component, repr_index)
def update_slider_info(change):
self._view._request_repr_parameters(
component=component_slider.value, repr_index=repr_slider.value)
component_dropdown.options = tuple(self._view._ngl_component_names)
def on_change_selection(change):
if self._real_time_update:
component = component_slider.value
repr_index = repr_slider.value
self._view._set_selection(change['new'],
component=component,
repr_index=repr_index)
def on_change_component_dropdown(change):
choice = change['new']
if choice:
component_slider.value = self._view._ngl_component_names.index(
choice)
component_dropdown.observe(on_change_component_dropdown, names='value')
def on_change_checkbox_repr_text(change):
repr_text_info.visible = change['new']
repr_slider.observe(update_slider_info, names='value')
component_slider.observe(update_slider_info, names='value')
repr_name_text.observe(on_change_repr_name, names='value')
repr_selection.observe(on_change_selection, names='value')
checkbox_repr_text.observe(on_change_checkbox_repr_text, names='value')
# HC
repr_parameters_box = self._make_repr_parameter_slider()
repr_parameters_box._ngl_name = 'repr_parameters_box'
# NOTE: if you update below list, make sure to update _make_repr_parameter_slider
# or refactor
# try to "refresh"
vbox = VBox([
component_dropdown, bbox, repr_info_box, component_slider,
repr_slider, reprlist_choices, repr_text_box, repr_parameters_box
])
self._view._request_repr_parameters(component=component_slider.value,
repr_index=repr_slider.value)
return vbox
def _make_widget_repr(self):
self.widget_repr_name = Text(value='', description='representation')
self.widget_repr_name._ngl_name = 'repr_name_text'
repr_selection = Text(value=' ', description='selection')
repr_selection._ngl_name = 'repr_selection'
repr_selection.width = self.widget_repr_name.width = default.DEFAULT_TEXT_WIDTH
max_n_components = max(self._view.n_components - 1, 0)
self.widget_component_slider = IntSlider(
value=0, max=max_n_components, min=0, description='component')
self.widget_component_slider._ngl_name = 'component_slider'
cvalue = ' '
self.widget_component_dropdown = Dropdown(
value=cvalue, options=[
cvalue,
], description='component')
self.widget_component_dropdown._ngl_name = 'component_dropdown'
self.widget_repr_slider = IntSlider(
value=0,
description='representation',
width=default.DEFAULT_SLIDER_WIDTH)
self.widget_repr_slider._ngl_name = 'repr_slider'
self.widget_repr_slider.visible = True
self.widget_component_slider.layout.width = default.DEFAULT_SLIDER_WIDTH
self.widget_repr_slider.layout.width = default.DEFAULT_SLIDER_WIDTH
self.widget_component_dropdown.layout.width = self.widget_component_dropdown.max_width = default.DEFAULT_TEXT_WIDTH
# turn off for now
self.widget_component_dropdown.layout.display = 'none'
self.widget_component_dropdown.description = ''
# self.widget_accordion_repr_parameters = Accordion()
self.widget_accordion_repr_parameters = Tab()
self.widget_repr_parameters = self._make_widget_repr_parameters(
self.widget_component_slider, self.widget_repr_slider,
self.widget_repr_name)
self.widget_accordion_repr_parameters.children = [
self.widget_repr_parameters, Box()
]
self.widget_accordion_repr_parameters.set_title(0, 'Parameters')
self.widget_accordion_repr_parameters.set_title(1, 'Hide')
self.widget_accordion_repr_parameters.selected_index = 1
checkbox_reprlist = Checkbox(value=False, description='reprlist')
checkbox_reprlist._ngl_name = 'checkbox_reprlist'
self.widget_repr_choices = self._make_repr_name_choices(
self.widget_component_slider, self.widget_repr_slider)
self.widget_repr_choices._ngl_name = 'reprlist_choices'
self.widget_repr_add = self._make_add_widget_repr(
self.widget_component_slider)
def on_update_checkbox_reprlist(change):
self.widget_repr_choices.visible = change['new']
checkbox_reprlist.observe(on_update_checkbox_reprlist, names='value')
def on_repr_name_text_value_changed(change):
name = change['new'].strip()
old = change['old'].strip()
should_update = (self._real_time_update and old and name and
name in REPRESENTATION_NAMES and
name != change['old'].strip())
if should_update:
component = self.widget_component_slider.value
repr_index = self.widget_repr_slider.value
self._view._remote_call(
'setRepresentation',
target='Widget',
args=[change['new'], {}, component, repr_index])
self._view._request_repr_parameters(component, repr_index)
def on_component_or_repr_slider_value_changed(change):
self._view._request_repr_parameters(
component=self.widget_component_slider.value,
repr_index=self.widget_repr_slider.value)
self.widget_component_dropdown.options = tuple(
self._view._ngl_component_names)
if self.widget_accordion_repr_parameters.selected_index >= 0:
self.widget_repr_parameters.name = self.widget_repr_name.value
self.widget_repr_parameters.repr_index = self.widget_repr_slider.value
self.widget_repr_parameters.component_index = self.widget_component_slider.value
def on_repr_selection_value_changed(change):
if self._real_time_update:
component = self.widget_component_slider.value
repr_index = self.widget_repr_slider.value
self._view._set_selection(
change['new'], component=component, repr_index=repr_index)
def on_change_component_dropdown(change):
choice = change['new']
if choice and choice.strip(): # bool(" ") is True
self.widget_component_slider.value = self._view._ngl_component_names.index(
choice)
self.widget_component_dropdown.observe(
on_change_component_dropdown, names='value')
self.widget_repr_slider.observe(
on_component_or_repr_slider_value_changed, names='value')
self.widget_component_slider.observe(
on_component_or_repr_slider_value_changed, names='value')
self.widget_repr_name.observe(
on_repr_name_text_value_changed, names='value')
repr_selection.observe(on_repr_selection_value_changed, names='value')
self.widget_repr_control_buttons = self._make_button_repr_control(
self.widget_component_slider, self.widget_repr_slider,
repr_selection)
blank_box = Box([Label("")])
all_kids = [
self.widget_repr_control_buttons, blank_box, self.widget_repr_add,
self.widget_component_dropdown, self.widget_repr_name,
repr_selection, self.widget_component_slider,
self.widget_repr_slider, self.widget_repr_choices,
self.widget_accordion_repr_parameters
]
vbox = VBox(all_kids)
self._view._request_repr_parameters(
component=self.widget_component_slider.value,
repr_index=self.widget_repr_slider.value)
self.widget_repr = _relayout_master(vbox, width='100%')
self._refresh(self.widget_component_slider, self.widget_repr_slider)
setattr(self.widget_repr, "_saved_widgets", [])
for _box in self.widget_repr.children:
if hasattr(_box, 'children'):
for kid in _box.children:
self.widget_repr._saved_widgets.append(kid)
return self.widget_repr
def _make_widget_repr(self):
self.widget_repr_name = Text(value='', description='representation')
self.widget_repr_name._ngl_name = 'repr_name_text'
repr_selection = Text(value=' ', description='selection')
repr_selection._ngl_name = 'repr_selection'
repr_selection.width = self.widget_repr_name.width = default.DEFAULT_TEXT_WIDTH
max_n_components = max(self._view.n_components-1, 0)
self.widget_component_slider = IntSlider(value=0, max=max_n_components, min=0, description='component')
self.widget_component_slider._ngl_name = 'component_slider'
cvalue = ' '
self.widget_component_dropdown = Dropdown(value=cvalue, options=[cvalue,],
description='component')
self.widget_component_dropdown._ngl_name = 'component_dropdown'
self.widget_repr_slider = IntSlider(value=0, description='representation', width=default.DEFAULT_SLIDER_WIDTH)
self.widget_repr_slider._ngl_name = 'repr_slider'
self.widget_repr_slider.visible = True
self.widget_component_slider.layout.width = default.DEFAULT_SLIDER_WIDTH
self.widget_repr_slider.layout.width = default.DEFAULT_SLIDER_WIDTH
self.widget_component_dropdown.layout.width = self.widget_component_dropdown.max_width = default.DEFAULT_TEXT_WIDTH
# turn off for now
self.widget_component_dropdown.layout.display = 'none'
self.widget_component_dropdown.description = ''
# self.widget_accordion_repr_parameters = Accordion()
self.widget_accordion_repr_parameters = Tab()
self.widget_repr_parameters = self._make_widget_repr_parameters(self.widget_component_slider,
self.widget_repr_slider,
self.widget_repr_name)
self.widget_accordion_repr_parameters.children = [self.widget_repr_parameters, Box()]
self.widget_accordion_repr_parameters.set_title(0, 'Parameters')
self.widget_accordion_repr_parameters.set_title(1, 'Hide')
self.widget_accordion_repr_parameters.selected_index = 1
checkbox_reprlist = Checkbox(value=False, description='reprlist')
checkbox_reprlist._ngl_name = 'checkbox_reprlist'
self.widget_repr_choices = self._make_repr_name_choices(self.widget_component_slider,
self.widget_repr_slider)
self.widget_repr_choices._ngl_name = 'reprlist_choices'
self.widget_repr_add = self._make_add_widget_repr(self.widget_component_slider)
def on_update_checkbox_reprlist(change):
self.widget_repr_choices.visible= change['new']
checkbox_reprlist.observe(on_update_checkbox_reprlist, names='value')
def on_repr_name_text_value_changed(change):
name = change['new'].strip()
old = change['old'].strip()
should_update = (self._real_time_update
and old and name
and name in REPRESENTATION_NAMES
and name != change['old'].strip())
if should_update:
component=self.widget_component_slider.value
repr_index=self.widget_repr_slider.value
self._view._remote_call('setRepresentation',
target='Widget',
args=[change['new'], {}, component, repr_index])
self._view._request_repr_parameters(component, repr_index)
def on_component_or_repr_slider_value_changed(change):
self._view._request_repr_parameters(component=self.widget_component_slider.value,
repr_index=self.widget_repr_slider.value)
self.widget_component_dropdown.options = tuple(self._view._ngl_component_names)
if self.widget_accordion_repr_parameters.selected_index >= 0:
self.widget_repr_parameters.name = self.widget_repr_name.value
self.widget_repr_parameters.repr_index = self.widget_repr_slider.value
self.widget_repr_parameters.component_index = self.widget_component_slider.value
def on_repr_selection_value_changed(change):
if self._real_time_update:
component = self.widget_component_slider.value
repr_index = self.widget_repr_slider.value
self._view._set_selection(change['new'],
component=component,
repr_index=repr_index)
def on_change_component_dropdown(change):
choice = change['new']
if choice:
self.widget_component_slider.value = self._view._ngl_component_names.index(choice)
self.widget_component_dropdown.observe(on_change_component_dropdown, names='value')
self.widget_repr_slider.observe(on_component_or_repr_slider_value_changed, names='value')
self.widget_component_slider.observe(on_component_or_repr_slider_value_changed, names='value')
self.widget_repr_name.observe(on_repr_name_text_value_changed, names='value')
repr_selection.observe(on_repr_selection_value_changed, names='value')
self.widget_repr_control_buttons = self._make_button_repr_control(self.widget_component_slider,
self.widget_repr_slider, repr_selection)
blank_box = Box([Label("")])
all_kids = [self.widget_repr_control_buttons,
blank_box,
self.widget_repr_add,
self.widget_component_dropdown,
self.widget_repr_name,
repr_selection,
self.widget_component_slider,
self.widget_repr_slider,
self.widget_repr_choices,
self.widget_accordion_repr_parameters
]
vbox = VBox(all_kids)
self._view._request_repr_parameters(component=self.widget_component_slider.value,
repr_index=self.widget_repr_slider.value)
self.widget_repr = _relayout_master(vbox, width='100%')
self._refresh(self.widget_component_slider, self.widget_repr_slider)
setattr(self.widget_repr, "_saved_widgets", [])
for _box in self.widget_repr.children:
if hasattr(_box, 'children'):
for kid in _box.children:
self.widget_repr._saved_widgets.append(kid)
return self.widget_repr
