'''
Created on Mar 12, 2024
@author: sima9999
'''
import sys
import logging
import ipywidgets
import ipympl
from IPython.core.display_functions import display
import numpy as np
import scipy.spatial
import scipy.stats
from pyOMA.core.Helpers import get_method_dict
import os
from pathlib import Path
# import ipywidgets
# import ipympl.backend_nbagg # the ''%matplotlib widget' backend
# Monkeypatch webagg to support blitting until https://github.com/matplotlib/matplotlib/pull/27160 is merged upstream
# copied and modified from https://github.com/raphaelquast/EOmaps/blob/66d32e2f5219059ab32a02457c535652d3e3f881/eomaps/_maps_base.py#L147
# #Creating a logger
# logger = logging.getLogger(__name__)
# logger.setLevel(level=logging.INFO)
#
# # log exceptions to logger instead of stderr
# def showtraceback(self, *args, **kwargs):
#
# logger.critical("Unhandled exception", exc_info=sys.exc_info())
#
# from IPython import get_ipython
# ipython = get_ipython()
# ipython.set_custom_exc((Exception,), showtraceback)
[docs]
class SnappingCursor:
"""
A cross-hair cursor that snaps to the data point of a line, which is
closest to the cursor.
.. TODO::
waiting for https://github.com/matplotlib/matplotlib/pull/27160 to be approved
then blitting should be tested and enabled
"""
[docs]
def __init__(self, ax, f_data, order_data):
self.ax = ax
self.horizontal_line = ax.axhline(color='k', lw=0.8, ls='--') # , animated=True)
self.vertical_line = ax.axvline(color='k', lw=0.8, ls='--') # , animated=True)
self.data_shape = f_data.shape
self.n_points = np.prod(self.data_shape)
self.data = np.ma.empty((self.n_points, 2))
self.data[:, 0] = f_data.reshape((self.n_points,))
self.data[:, 1] = order_data.reshape((self.n_points,))
# copy mask from frequency-data (x) to order-data (y)
if isinstance(f_data, np.ma.MaskedArray):
mask = f_data.mask.reshape((self.n_points,))
else:
mask = np.ma.nomask
self.data.mask[:, 0] = mask
self.data.mask[:, 1] = mask
self._last_index = None
self.callbacks = {'show_current_info':lambda *args, **kwargs: None,
'mode_selected':lambda *args, **kwargs: None,
'mode_deselected':lambda *args, **kwargs: None, }
self.update_pix_data()
def add_callback(self, name, func):
assert name in ['show_current_info', 'mode_selected', 'mode_deselected']
self.callbacks[name] = func
def set_mask(self, mask, name=None): # name just for backwards compatibility
n_mask = mask.reshape((self.n_points,))
self.data.mask[:, 0] = n_mask
self.data.mask[:, 1] = n_mask
self.update_pix_data()
def set_cross_hair_visible(self, visible):
need_redraw = self.horizontal_line.get_visible() != visible
self.horizontal_line.set_visible(visible)
self.vertical_line.set_visible(visible)
return need_redraw
def on_mouse_move(self, event):
if not event.inaxes:
self._last_index = None
need_redraw = self.set_cross_hair_visible(False)
if need_redraw:
self.ax.figure.canvas.draw()
else:
self.set_cross_hair_visible(True)
x, y = event.xdata, event.ydata
x_, y_ = self.ax.transData.transform(
np.vstack([x, y]).T).T
index = self.findIndexNearestXY(x_, y_)
if index == self._last_index:
return # still on the same data point. Nothing to do.
self._last_index = index
x = self.data[index, 0:1]
y = self.data[index, 1:2]
# update the line positions
self.horizontal_line.set_ydata(y)
self.vertical_line.set_xdata(x)
# self.ax.figure.canvas.restore_region(self.bg)
# self.ax.draw_artist(self.horizontal_line)
# self.ax.draw_artist(self.vertical_line)
# self.ax.figure.canvas.blit(self.ax.figure.bbox)
# self.ax.figure.canvas.flush_events()
self.ax.figure.canvas.draw()
if np.ma.is_masked(x):
self.data_index = None
else:
self.data_index = np.unravel_index(index, self.data_shape)
self.callbacks['show_current_info'](self.data_index)
def on_button(self, event=None):
if self.data_index is not None:
self.callbacks['mode_selected'](self.data_index)
[docs]
def findIndexNearestXY(self, x_point, y_point):
'''
Finds the nearest neighbour
'''
# distance = np.square(self.pix_data[:, 1] - y_point) + np.square(self.pix_data[:, 0] - x_point)
# index = np.argmin(distance)
d, index = self.tree.query(np.hstack([x_point, y_point]), 1)
return index
def update_pix_data(self, event=None):
data = self.data
self.pix_data = self.ax.transData.transform(self.data)
self.pix_data._mask = self.data._mask
# the slow thing is the redraw of all poles
# the following only speeds up tree lookup
#
# xmin, xmax = self.ax.get_xlim()
# ymin, ymax = self.ax.get_ylim()
#
# xmask = np.logical_and(self.data[:,0]>xmin, self.data[:,0]<xmax)
# ymask = np.logical_and(self.data[:,1]>ymin, self.data[:,1]<ymax)
#
# datamask = ~np.logical_and(xmask, ymask)
# data_mask = np.hstack([datamask[:,np.newaxis],datamask[:,np.newaxis]])
#
# self.pix_data._mask = np.logical_or(self.data._mask, data_mask)
self.tree = scipy.spatial.KDTree(self.pix_data)
# fig = self.ax.get_figure()
# self.bg = fig.canvas.copy_from_bbox(fig.bbox)
# self.ax.draw_artist(self.horizontal_line)
# self.ax.draw_artist(self.vertical_line)
# fig.canvas.blit(fig.bbox)
[docs]
def StabilGUIWeb(stabil_plot):
stabil_calc = stabil_plot.stabil_calc
stabil_plot.update_stabilization() # initialize the plot
# print('Which backend are we using? ', plt.get_backend())
df_max = stabil_calc.df_max * 100
dd_max = stabil_calc.dd_max * 100
dmac_max = stabil_calc.dmac_max * 100
d_range = stabil_calc.d_range
mpc_min = stabil_calc.mpc_min
mpd_max = stabil_calc.mpd_max
fig = stabil_plot.fig
ax = stabil_plot.ax
canvas = ipympl.backend_nbagg.Canvas(fig)
manager = ipympl.backend_nbagg.FigureManager(canvas, 0)
canvas.header_visible = False
canvas.toolbar_position = 'left'
canvas.footer_visible = False
canvas.resizable = False
snap_cursor = SnappingCursor(ax, stabil_calc.masked_frequencies, stabil_calc.order_dummy)
# setup logger for output in UI
logger = logging.getLogger('core.StabilDiagram')
handler = OutputWidgetHandler()
handler.out.layout.width = '1360px'
handler.setFormatter(logging.Formatter(logging.BASIC_FORMAT))
loggers = [logging.getLogger(name) for name in logging.root.manager.loggerDict]
for logger in loggers:
if 'pyOMA.' in logger.name:
logger.addHandler(handler)
# logger.setLevel(logging.INFO)
# Soft criteria
widgets = []
lb = ipywidgets.Label("Soft criteria:")
widgets.append(lb)
if stabil_calc.capabilities['f']:
sl_df = ipywidgets.FloatLogSlider(value=stabil_calc.df_max, base=10, min=-4, max=0, step=0.1, description="Frequency [%]")
sl_df.observe(lambda change: stabil_plot.update_stabilization(df_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_df)
if stabil_calc.capabilities['d']:
sl_dd = ipywidgets.FloatLogSlider(value=stabil_calc.dd_max, base=10, min=-4, max=0, step=0.1, description="Damping [%]")
sl_dd.observe(lambda change: stabil_plot.update_stabilization(dd_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_dd)
if stabil_calc.capabilities['msh']:
sl_dmac = ipywidgets.FloatLogSlider(value=stabil_calc.dmac_max, base=10, min=-4, max=0, step=0.1, description="MAC [%]")
sl_dmac.observe(lambda change: stabil_plot.update_stabilization(dmac_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_dmac)
# ..TODO:: add Eigenvalue distance selector
softbox = ipywidgets.VBox(widgets,
layout=ipywidgets.Layout(width='350px', border='solid 1px'))
# Hard criteria
widgets = []
lb = ipywidgets.Label("Hard criteria:")
widgets.append(lb)
if stabil_calc.capabilities['std']:
# sl_stdf = ipywidgets.FloatLogSlider(value=stabil_calc.stdf_max, base=10, min=-2, max=4, step=0.1, description='CI F. [Hz]')
sl_stdf = ipywidgets.FloatSlider(value=stabil_calc.stdf_max, min=0, max=43, step=0.1, description='CI F. [Hz]')
sl_stdf.observe(lambda change: stabil_plot.update_stabilization(stdf_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_stdf)
# sl_stdd = ipywidgets.FloatLogSlider(value=stabil_calc.stdd_max, base=10, min=-2, max=4, step=0.1, description='CI D. [%]')
sl_stdd = ipywidgets.FloatSlider(value=stabil_calc.stdd_max, min=0, max=100, step=0.1, description='CI D. [%]')
sl_stdd.observe(lambda change: stabil_plot.update_stabilization(stdd_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_stdd)
if stabil_calc.capabilities['d']:
sl_d_range = ipywidgets.FloatRangeSlider(value=stabil_calc.d_range, min=0, max=20, step=0.1, description='Damping range [%]')
sl_d_range.observe(lambda change: stabil_plot.update_stabilization(d_range=change['new']),
names='value', type='change')
widgets.append(sl_d_range)
if stabil_calc.capabilities['msh']:
sl_mpc = ipywidgets.FloatSlider(value=stabil_calc.mpc_min, min=0, max=1, step=0.01, description='MPC_min')
sl_mpc.observe(lambda change: stabil_plot.update_stabilization(mpc_min=float(change['new'])),
names='value', type='change')
widgets.append(sl_mpc)
sl_mpd = ipywidgets.FloatSlider(value=stabil_calc.mpd_max, min=0, max=180, step=1, description='MPD_max [°]')
sl_mpd.observe(lambda change: stabil_plot.update_stabilization(mpd_max=float(change['new'])),
names='value', type='change')
widgets.append(sl_mpd)
if stabil_calc.capabilities['mtn']:
sl_mtn = ipywidgets.FloatSlider(value=stabil_calc.mtn_min, min=0, max=100, step=1, description='MTN_max []')
widgets.append(sl_mtn)
# ..TODO:: implement
if stabil_calc.capabilities['MC']:
sl_mc = ipywidgets.FloatSlider(value=stabil_calc.MC_min, min=0, max=1, step=0.01, description='MC_min []')
sl_mc.observe(lambda change: stabil_plot.update_stabilization(MC_min=float(change['new'])),
names='value', type='change')
widgets.append(sl_mc)
hardbox = ipywidgets.VBox(widgets,
layout=ipywidgets.Layout(width='350px', border='solid 1px'))
# View settings
lb = ipywidgets.Label('View')
cb_stb = ipywidgets.Checkbox(value=stabil_plot.stable_plot['plot_stable'].get_visible(), description='Stable poles', indent=False, layout=ipywidgets.Layout(width='100px'))
cb_all = ipywidgets.Checkbox(value=stabil_plot.stable_plot['plot_pre'].get_visible(), description='All poles', indent=False, layout=ipywidgets.Layout(width='100px'))
cb_psd = ipywidgets.Checkbox(value=stabil_plot.psd_plot[0][0].get_visible() if stabil_plot.psd_plot else False, description='Show PSD', indent=False, layout=ipywidgets.Layout(width='100px'))
rbs = ipywidgets.RadioButtons(options=['Stable', 'All', 'Off'], value='Off', description='Cursor', layout=ipywidgets.Layout(width='100px'))
viewbox = ipywidgets.VBox([lb, cb_stb, cb_all, cb_psd, rbs],
layout=ipywidgets.Layout(width='200px', border='solid 1px'))
# Selected modes
frequencies = [f'{f:1.3f}' for f in stabil_calc.get_frequencies()]
dd = ipywidgets.Dropdown(options=frequencies , value=frequencies[-1] if frequencies else None, description='Selected mode:', style={'description_width': '100px'}, layout=ipywidgets.Layout(width='200px'))
select_mode_values = ipywidgets.HTMLMath(value='')
selectbox = ipywidgets.VBox([dd, select_mode_values],
layout=ipywidgets.Layout(width='230px', border='solid 1px'))
# Current mode
lb = ipywidgets.Label('Current mode:')
current_mode_values = ipywidgets.HTMLMath(value='')
currentbox = ipywidgets.VBox([lb, current_mode_values],
layout=ipywidgets.Layout(width='230px', border='solid 1px'))
# build final layout
hbox = ipywidgets.HBox([softbox, hardbox, viewbox, selectbox, currentbox],
layout=ipywidgets.Layout(justify_content='space-around'))
vbox = ipywidgets.VBox([fig.canvas, hbox, handler.out],
layout=ipywidgets.Layout(align_items='center', padding="0px 0px 0px 100px", overflow="scroll"))
global cid
cid = None
def toggle_cursor_snap(change):
global cid
if change['new'] == 'Stable':
snap_cursor.set_mask(stabil_calc.get_stabilization_mask('mask_stable'))
elif change['new'] == 'All':
snap_cursor.set_mask(stabil_calc.get_stabilization_mask('mask_pre'))
if change['new'] == 'Off':
snap_cursor.horizontal_line.set_visible(False)
snap_cursor.vertical_line.set_visible(False)
canvas.mpl_disconnect(cid)
else:
cid = canvas.mpl_connect('motion_notify_event', snap_cursor.on_mouse_move)
snap_cursor.horizontal_line.set_visible(True)
snap_cursor.vertical_line.set_visible(True)
# assign as callback to: stabil_calc.add_callback('add_mode', mode_selector_change)
def mode_selector_change(index):
# update Dropdown widget with new frequencies and set it to the current
frequencies = [f'{f:1.3f}' for f in stabil_calc.get_frequencies()]
if index in stabil_calc.select_modes:
current = f'{stabil_calc.masked_frequencies[index[0], index[1]]:1.3f}'
else:
current = frequencies[0]
dd.options = frequencies
dd.value = current
def update_value_view(widget, frequency=None, mode_index=None,):
if frequency is not None:
selected_indices = stabil_calc.select_modes
frequencies = np.array([stabil_calc.masked_frequencies[index[0], index[1]]
for index in selected_indices])
f_delta = abs(frequencies - frequency)
index = np.argmin(f_delta)
mode_index = selected_indices[index]
n, f, stdf, d, stdd, mpc, mp, mpd, dmp, dmpd, mtn, MC, ex_1, ex_2 = stabil_calc.get_modal_values(mode_index)
if stabil_calc.capabilities['std']:
num_blocks = stabil_calc.modal_data.num_blocks
stdf = scipy.stats.t.ppf(
0.95, num_blocks) * stdf / np.sqrt(num_blocks)
stdd = scipy.stats.t.ppf(
0.95, num_blocks) * stdd / np.sqrt(num_blocks)
s = '<table>\n'
s += ''.join([f'<tr>\n<td> Frequency [Hz]:</td>\n <td> {f:1.3f} </td>\n</tr>\n' if not np.isnan(f) else '',
f'<tr>\n<td> CI Frequency [Hz]:</td>\n <td> {stdf:1.3e} </td>\n</tr>\n' if not np.isnan(stdf) else '',
f'<tr>\n<td> Model order:</td>\n <td> {n:1.0f} </td>\n</tr>\n' if not np.isnan(n) else '',
f'<tr>\n<td> Damping [%]:</td>\n <td> {d:1.3f} </td>\n</tr>\n' if not np.isnan(d) else '',
f'<tr>\n<td> CI Damping [%]:</td>\n <td> {stdd:1.3e} </td>\n</tr>\n' if not np.isnan(stdd) else '',
f'<tr>\n<td> MPC [-]:</td>\n <td> {mpc:1.5f} </td>\n</tr>\n' if not np.isnan(mpc) else '',
f'<tr>\n<td> MP [\u00b0]:</td>\n <td> {mp:1.3f} </td>\n</tr>\n' if not np.isnan(mp) else '',
f'<tr>\n<td> MPD [-]:</td>\n <td> {mpd:1.5f} </td>\n</tr>\n' if not np.isnan(mpd) else '',
f'<tr>\n<td> dMP [\u00b0]:</td>\n <td> {dmp:1.3f} </td>\n</tr>\n' if not np.isnan(dmp) else '',
f'<tr>\n<td> MTN [%]:</td>\n <td> {mtn:1.5f} </td>\n</tr>\n' if not np.isnan(mtn) else '',
f'<tr>\n<td> MC [%]:</td>\n <td> {MC:1.5f} </td>\n</tr>\n' if not np.isnan(MC) else '',
f'<tr>\n<td> Ext [-]:</td>\n <td> {ex_1:1.5f} </td>\n</tr>\n' if not np.isnan(ex_1) else '',
f'<tr>\n<td> Ext [-]:</td>\n <td> {ex_2:1.5f} </td>\n</tr>\n' if not np.isnan(ex_2) else ''
])
s += '</table>'
widget.value = s
cb_stb.observe(handler=lambda change: stabil_plot.toggle_stable(bool(change['new'])), names='value', type='change')
cb_all.observe(handler=lambda change: stabil_plot.toggle_all(bool(change['new'])), names='value', type='change')
cb_psd.observe(handler=lambda change: stabil_plot.plot_sv_psd(bool(change['new'])), names='value', type='change')
stabil_calc.add_callback('add_mode', mode_selector_change)
stabil_calc.add_callback('remove_mode', mode_selector_change)
dd.observe(handler=lambda change: update_value_view(select_mode_values, frequency=float(change['new'])) , names='value', type='change')
snap_cursor.add_callback('show_current_info', lambda mode_index: update_value_view(current_mode_values, mode_index=mode_index))
snap_cursor.add_callback('mode_selected', stabil_plot.toggle_mode)
canvas.mpl_connect('button_press_event', snap_cursor.on_button)
canvas.mpl_connect('resize_event', snap_cursor.update_pix_data)
ax.callbacks.connect('xlim_changed', snap_cursor.update_pix_data)
ax.callbacks.connect('ylim_changed', snap_cursor.update_pix_data)
rbs.observe(handler=toggle_cursor_snap, names='value', type='change')
rbs.value = 'Stable'
if frequencies:
update_value_view(select_mode_values, frequency=float(frequencies[0]))
dpi = fig.get_dpi()
height = fig.get_figheight()
fig.set_size_inches((1360 / dpi, height))
fig.canvas.draw()
snap_cursor.update_pix_data(None)
return vbox, snap_cursor
[docs]
def PlotMSHWeb(msp):
# setup Figure for display with ipympl
fig = msp.fig
ax = msp.subplot
canvas = ipympl.backend_nbagg.Canvas(fig)
# uncommenting following line breaks display on some windows systems
# msp.canvas = canvas
manager = ipympl.backend_nbagg.FigureManager(canvas, 0)
canvas.header_visible = False
canvas.toolbar_position = 'right'
canvas.footer_visible = False
canvas.resizable = False
# reset view
msp.reset_view()
# setup logger for output in UI
logger = logging.getLogger('core.PlotMSH')
handler = OutputWidgetHandler()
handler.setFormatter(logging.Formatter(logging.BASIC_FORMAT))
logger.addHandler(handler)
logger.setLevel(logging.INFO)
# build "Options" box
lb = ipywidgets.Label(value='Options:')
cb1 = ipywidgets.Checkbox(value=msp.show_axis, description='Show Axis Arrows',
indent=False, layout=ipywidgets.Layout(width='150px', height='30px'))
cb2 = ipywidgets.Checkbox(value=msp.show_nodes, description='Show Nodes',
indent=False, layout=ipywidgets.Layout(width='100px', height='30px'))
cb3 = ipywidgets.Checkbox(value=msp.show_lines, description='Show Lines',
indent=False, layout=ipywidgets.Layout(width='100px', height='30px'))
cb4 = ipywidgets.Checkbox(value=msp.show_cn_lines, description='Show Connecting Lines',
indent=False, layout=ipywidgets.Layout(width='170px', height='30px'))
cb5 = ipywidgets.Checkbox(value=msp.show_nd_lines, description='Show Non-displaced Lines',
indent=False, layout=ipywidgets.Layout(width='180px', height='30px'))
cb6 = ipywidgets.Checkbox(value=msp.show_parent_childs, description='Show Parent-Child Assignm.',
indent=False, layout=ipywidgets.Layout(width='200px', height='30px'))
cb7 = ipywidgets.Checkbox(value=msp.show_chan_dofs, description='Show Channel-DOF Assignm.',
indent=False, layout=ipywidgets.Layout(width='190px', height='30px'))
optbox = ipywidgets.VBox([lb, cb1, cb2, cb3, cb4, cb5, cb6, cb7],
layout=ipywidgets.Layout(border='solid 1px'))
# build "View" box
lb = ipywidgets.Label(value='View:')
fse = ipywidgets.FloatSlider(value=msp.subplot.elev, min=-180, max=180, step=1,
description='Elevation', continuous_update=True)
fsa = ipywidgets.FloatSlider(value=msp.subplot.azim, min=-180, max=180, step=1,
description='Azimuth', continuous_update=True)
fsr = ipywidgets.FloatSlider(value=msp.subplot.roll, min=-180, max=180, step=1,
description='Roll', continuous_update=True)
view_buttons = []
btn = ipywidgets.Button(description='X',
layout=ipywidgets.Layout(width='30px', height='30px'))
view_buttons.append(btn)
btn = ipywidgets.Button(description='Y',
layout=ipywidgets.Layout(width='30px', height='30px'))
view_buttons.append(btn)
btn = ipywidgets.Button(description='Z',
layout=ipywidgets.Layout(width='30px', height='30px'))
view_buttons.append(btn)
btn = ipywidgets.Button(description='ISO',
layout=ipywidgets.Layout(width='40px', height='30px'))
view_buttons.append(btn)
hbox = ipywidgets.HBox(view_buttons)
res_btn = ipywidgets.Button(description='Reset')
viewbox = ipywidgets.VBox([lb, hbox, fse, fsa, fsr, res_btn],
layout=ipywidgets.Layout(border='solid 1px'))
# build "Mode" box
lb = ipywidgets.Label(value='Mode:')
frequencies = [f'{f:1.3f}' for f in msp.get_frequencies()]
if msp.mode_index is not None:
current = f'{msp.modal_frequencies[msp.mode_index[0], msp.mode_index[1]]:1.3f}'
dd = ipywidgets.Dropdown(options=frequencies , value=current)
else:
dd = ipywidgets.Dropdown(options=frequencies)
ft = ipywidgets.FloatText(value=msp.amplitude, description='Amplitude')
cb = ipywidgets.Checkbox(value=msp.real , description='Real Modeshape',)
buttons = []
btn = ipywidgets.Button(icon='play')
btn.on_click(lambda change: msp.animate())
buttons.append(btn)
btn = ipywidgets.Button(icon='stop')
btn.on_click(lambda change: msp.stop_ani())
buttons.append(btn)
hbox = ipywidgets.HBox(buttons)
reload_btn = ipywidgets.Button(description='Reload Mode Selection', layout={'width':'90%'})
modebox = ipywidgets.VBox([lb, dd, ft, cb, hbox, reload_btn],
layout=ipywidgets.Layout(border='solid 1px'))
# build "Info" box
lb = ipywidgets.Label(value='Info:')
html = ipywidgets.HTMLMath(value='')
infobox = ipywidgets.VBox([lb, html], layout=ipywidgets.Layout(border='solid 1px'))
# build final layout
hbox = ipywidgets.HBox([optbox, viewbox, modebox, infobox],
layout=ipywidgets.Layout(justify_content='space-around'))
vbox = ipywidgets.VBox([fig.canvas, ipywidgets.Label(value='Left click to rotate, middle click to pan, right click to zoom.'), hbox, handler.out],
layout=ipywidgets.Layout(align_items='center'))
# define callbacks and other logic
def observe_opt_btns(b):
if b:
cb1.observe(lambda d: msp.refresh_axis(d['new']) if d['name'] == 'value' else None)
cb2.observe(lambda d: msp.refresh_nodes(d['new']) if d['name'] == 'value' else None)
cb3.observe(lambda d: msp.refresh_lines(d['new']) if d['name'] == 'value' else None)
cb4.observe(lambda d: msp.refresh_cn_lines(d['new']) if d['name'] == 'value' else None)
cb5.observe(lambda d: msp.refresh_nd_lines(d['new']) if d['name'] == 'value' else None)
cb6.observe(lambda d: msp.refresh_parent_childs(d['new']) if d['name'] == 'value' else None)
cb7.observe(lambda d: msp.refresh_chan_dofs(d['new']) if d['name'] == 'value' else None)
else:
for cb in [cb1, cb2, cb3, cb4, cb5, cb6, cb7]:
cb.unobserve_all()
def observe_sliders(b):
if b:
fse.observe(handler=change_viewport, names='value', type='change')
fsa.observe(handler=change_viewport, names='value', type='change')
fsr.observe(handler=change_viewport, names='value', type='change')
else:
for fs in [fse, fsa, fsr]:
fs.unobserve_all()
def change_viewport(change):
if isinstance(change, ipywidgets.Button):
observe_sliders(False)
sender = change.description
if sender == 'X':
fsa.value, fse.value = 0.0, 0.0
elif sender == 'Y':
fsa.value, fse.value = -90.0, 0.0
elif sender == 'Z':
fsa.value, fse.value = 0.0, 90.0
elif sender == 'ISO':
fsa.value, fse.value = -60.0, 30.0
msp.change_viewport(sender)
observe_sliders(True)
else:
msp.change_viewport((fse.value, fsa.value, fsr.value))
def reset_view(self):
msp.stop_ani()
observe_opt_btns(False)
observe_sliders(False)
cb1.value = True
cb2.value = True
cb3.value = True
cb4.value = True
cb5.value = True
cb6.value = False
cb7.value = False
fse.value = 30
fsa.value = -60
fsr.value = 0
msp.reset_view()
observe_opt_btns(True)
observe_sliders(True)
def mode_change(current):
mode, order, frequency, damping, MPC, MP, MPD = msp.change_mode(float(current))
if msp.stabil_calc is not None:
n, f, stdf, d, stdd, mpc, mp, mpd, dmp, dmpd, mtn, MC, ex_1, ex_2 = msp.stabil_calc.get_modal_values((order, mode))
# print(n, f, stdf, d, stdd, mpc, mp, mpd, dmp, dmpd, mtn, MC, ex_1, ex_2)
if msp.stabil_calc.capabilities['std']:
num_blocks = msp.tabil_calc.modal_data.num_blocks
stdf = scipy.stats.t.ppf(
0.95, num_blocks) * stdf / np.sqrt(num_blocks)
stdd = scipy.stats.t.ppf(
0.95, num_blocks) * stdd / np.sqrt(num_blocks)
text = '<table>\n'
text += ''.join([f'<tr>\n<td> Frequency [Hz]:</td>\n <td> {f:1.3f} </td>\n</tr>\n' if not np.isnan(f) else '',
f'<tr>\n<td> CI Frequency [Hz]:</td>\n <td> {stdf:1.3e} </td>\n</tr>\n' if not np.isnan(stdf) else '',
f'<tr>\n<td> Model order:</td>\n <td> {n:1.0f} </td>\n</tr>\n' if not np.isnan(n) else '',
f'<tr>\n<td> Damping [%]:</td>\n <td> {d:1.3f} </td>\n</tr>\n' if not np.isnan(d) else '',
f'<tr>\n<td> CI Damping [%]:</td>\n <td> {stdd:1.3e} </td>\n</tr>\n' if not np.isnan(stdd) else '',
f'<tr>\n<td> MPC [-]:</td>\n <td> {mpc:1.5f} </td>\n</tr>\n' if not np.isnan(mpc) else '',
f'<tr>\n<td> MP [\u00b0]:</td>\n <td> {mp:1.3f} </td>\n</tr>\n' if not np.isnan(mp) else '',
f'<tr>\n<td> MPD [-]:</td>\n <td> {mpd:1.5f} </td>\n</tr>\n' if not np.isnan(mpd) else '',
f'<tr>\n<td> dMP [\u00b0]:</td>\n <td> {dmp:1.3f} </td>\n</tr>\n' if not np.isnan(dmp) else '',
f'<tr>\n<td> MTN [%]:</td>\n <td> {mtn:1.5f} </td>\n</tr>\n' if not np.isnan(mtn) else '',
f'<tr>\n<td> MC [%]:</td>\n <td> {MC:1.5f} </td>\n</tr>\n' if not np.isnan(MC) else '',
f'<tr>\n<td> Ext [-]:</td>\n <td> {ex_1:1.5f} </td>\n</tr>\n' if not np.isnan(ex_1) else '',
f'<tr>\n<td> Ext [-]:</td>\n <td> {ex_2:1.5f} </td>\n</tr>\n' if not np.isnan(ex_2) else ''
])
text += '</table>'
else:
text = f'''
<table>
<tr>
<td> Frequency [Hz]:</td>
<td> {frequency:1.3f} </td>
</tr>
<tr>
<td> Damping [%]:</td>
<td> {damping:1.3f} </td>
</tr>
'''
if order is not None:
text += f'''
<tr>
<td> Model order:</td>
<td> {order} </td>
</tr>
'''
if mode is not None:
text += f'''
<tr>
<td> Mode number:</td>
<td> {mode} </td>
</tr>
'''
if MPC is not None:
text += f'''
<tr>
<td> MPC [-]:</td>
<td> {MPC:1.3f} </td>
</tr>
'''
if MP is not None:
text += f'''
<tr>
<td> MP [\u00b0]:</td>
<td> {MP:1.3f} </td>
</tr>
'''
if MPD is not None:
text += f'''
<tr>
<td> MPD [-]:</td>
<td> {MPD:1.3f} </td>
</tr>
'''
text += f'''
</table>
'''
dd.value = f'{msp.modal_frequencies[msp.mode_index[0], msp.mode_index[1]]:1.3f}'
html.value = text
def reload_modes(btn):
# update Dropdown widget with new frequencies and set it to the current
current = dd.value
frequencies = [f'{f:1.3f}' for f in msp.get_frequencies()]
dd.options = frequencies
if current in frequencies:
dd.value = current
# connect widgets and callbacks
observe_opt_btns(True)
for button in view_buttons:
button.on_click(change_viewport)
res_btn.on_click(reset_view)
dd.observe(handler=lambda change: mode_change(float(change['new'])) , names='value', type='change')
ft.observe(handler=lambda change: msp.change_amplitude(float(change['new'])) , names='value', type='change')
cb.observe(handler=lambda change: msp.change_part(bool(change['new'])) , names='value', type='change')
reload_btn.on_click(reload_modes)
if msp.mode_index is not None:
mode_change(current)
return vbox
[docs]
def ConfigGUIWeb(config_dict):
def read_and_display(widget, file):
if os.path.exists(file):
with open(file, 'r') as f:
widget.value = f.read()
else:
widget.value = 'File does not exist'
def save_contents(widget, file):
with open(file, 'w') as f:
contents = widget.value
f.write(contents)
project_dir = config_dict.get('project_dir', '')
setup_dir = config_dict.get('setup_dir', '')
result_dir = config_dict.get('result_dir', '')
meas_file = config_dict.get('meas_file', '')
method = config_dict.get('method', '')
nodes_file = config_dict.get('nodes_file', '')
lines_file = config_dict.get('lines_file', '')
parent_child_file = config_dict.get('parent_child_file', '')
setup_info_file = config_dict.get('setup_info_file', '')
chan_dofs_file = config_dict.get('chan_dofs_file', '')
oma_conf_file = config_dict.get('oma_conf_file', '')
skip_existing = config_dict.get('skip_existing', '')
save_results = config_dict.get('save_results', '')
method_dict = get_method_dict()
if method in method_dict.values():
method_name = [name for name, method_ in method_dict.items() if method == method_][0]
else:
method_name = list(method_dict.keys())[0]
tab_contents = ['General', 'Geometry', 'Setup Info', 'Channel-DOF-Assignments', 'OMA Config']
layout = {'width':'200px'}
# General
project_dir_widg = ipywidgets.Text(value=str(project_dir), description='Project Directory', layout={'width':'800px'}, style={'description_width': '200px'})
project_dir_widg.observe(handler=lambda change: config_dict.update({'project_dir':Path(change['new'])}), names='value', type='change')
setup_dir_widg = ipywidgets.Text(value=str(setup_dir), description='Setup Directory', layout={'width':'800px'}, style={'description_width': '200px'})
setup_dir_widg.observe(handler=lambda change: config_dict.update({'setup_dir':Path(change['new'])}), names='value', type='change')
result_dir_widg = ipywidgets.Text(value=str(result_dir), description='Result Directory', layout={'width':'800px'}, style={'description_width': '200px'})
result_dir_widg.observe(handler=lambda change: config_dict.update({'result_dir':Path(change['new'])}), names='value', type='change')
meas_file_widg = ipywidgets.Text(value=str(meas_file), description='Measurement File', layout={'width':'800px'}, style={'description_width': '200px'})
meas_file_widg.observe(handler=lambda change: config_dict.update({'meas_file':Path(change['new'])}), names='value', type='change')
method_widg = ipywidgets.Dropdown(options=list(method_dict.keys()), value=method_name, layout={'width':'800px'})
method_widg.observe(handler=lambda change: config_dict.update({'method':method_dict[change['new']]}), names='value', type='change')
config_dict.update({'method':method_dict[method_name]})
skip_existing_widg = ipywidgets.Checkbox(value=skip_existing, description='Skip existing results')
skip_existing_widg.observe(handler=lambda change: config_dict.update({'skip_existing':bool(change['new'])}), names='value', type='change')
save_results_widg = ipywidgets.Checkbox(value=save_results, description='Save new results')
save_results_widg.observe(handler=lambda change: config_dict.update({'save_results':bool(change['new'])}), names='value', type='change')
general_box = ipywidgets.VBox([project_dir_widg, setup_dir_widg, result_dir_widg, meas_file_widg, method_widg, ipywidgets.HBox([skip_existing_widg, save_results_widg])])
# Geometry
geometry_contents = ['Nodes', 'Lines', 'Parent-Child-Assignments']
nodes_file_widg = ipywidgets.Text(value=str(nodes_file), description='Nodes File', layout={'width':'800px'}, style={'description_width': '200px'})
nodes_file_widg.observe(handler=lambda change: config_dict.update({'nodes_file':Path(change['new'])}), names='value', type='change')
nodes_text = ipywidgets.Textarea(placeholder='Load nodes file', layout={'width':'800px', 'height':'400px'})
nodes_save_btn = ipywidgets.Button(description='Save')
nodes_load_btn = ipywidgets.Button(description='Load')
nodes_save_btn.on_click(lambda b: save_contents(nodes_text, nodes_file_widg.value))
nodes_load_btn.on_click(lambda b: read_and_display(nodes_text, nodes_file_widg.value))
nodes_load_btn.click()
btn_box = ipywidgets.HBox([nodes_load_btn, nodes_save_btn])
nodes_box = ipywidgets.VBox([nodes_file_widg, nodes_text, btn_box])
lines_file_widg = ipywidgets.Text(value=str(lines_file), description='Lines File', layout={'width':'800px'}, style={'description_width': '200px'})
lines_file_widg.observe(handler=lambda change: config_dict.update({'lines_file':Path(change['new'])}), names='value', type='change')
lines_text = ipywidgets.Textarea(placeholder='Load lines file', layout={'width':'800px', 'height':'400px'})
lines_save_btn = ipywidgets.Button(description='Save')
lines_load_btn = ipywidgets.Button(description='Load')
lines_save_btn.on_click(lambda b: save_contents(lines_text, lines_file_widg.value))
lines_load_btn.on_click(lambda b: read_and_display(lines_text, lines_file_widg.value))
lines_load_btn.click()
btn_box = ipywidgets.HBox([lines_load_btn, lines_save_btn])
lines_box = ipywidgets.VBox([lines_file_widg, lines_text, btn_box])
parent_child_file_widg = ipywidgets.Text(value=str(parent_child_file), description='Parent-Child Assignments File', layout={'width':'800px'}, style={'description_width': '200px'})
parent_child_file_widg.observe(handler=lambda change: config_dict.update({'parent_child_file':Path(change['new'])}), names='value', type='change')
parent_child_text = ipywidgets.Textarea(placeholder='Load parent child assignments file', layout={'width':'800px', 'height':'400px'})
parent_child_save_btn = ipywidgets.Button(description='Save')
parent_child_load_btn = ipywidgets.Button(description='Load')
parent_child_save_btn.on_click(lambda b: save_contents(parent_child_text, parent_child_file_widg.value))
parent_child_load_btn.on_click(lambda b: read_and_display(parent_child_text, parent_child_file_widg.value))
parent_child_load_btn.click()
btn_box = ipywidgets.HBox([parent_child_load_btn, parent_child_save_btn])
parent_child_box = ipywidgets.VBox([parent_child_file_widg, parent_child_text, btn_box])
geometry_tab = ipywidgets.Tab()
geometry_tab.children = [nodes_box, lines_box, parent_child_box]
geometry_tab.titles = geometry_contents
# Setup Info
setup_info_file_widg = ipywidgets.Text(value=str(setup_info_file), description='Setup Info File', layout={'width':'800px'}, style={'description_width': '200px'})
setup_info_file_widg.observe(handler=lambda change: config_dict.update({'setup_info_file':Path(change['new'])}), names='value', type='change')
setup_info_text = ipywidgets.Textarea(placeholder='Load setup info file', layout={'width':'800px', 'height':'400px'})
setup_info_save_btn = ipywidgets.Button(description='Save')
setup_info_load_btn = ipywidgets.Button(description='Load')
setup_info_save_btn.on_click(lambda b: save_contents(setup_info_text, setup_info_file_widg.value))
setup_info_load_btn.on_click(lambda b: read_and_display(setup_info_text, setup_info_file_widg.value))
setup_info_load_btn.click()
btn_box = ipywidgets.HBox([setup_info_load_btn, setup_info_save_btn])
setup_info_box = ipywidgets.VBox([setup_info_file_widg, setup_info_text, btn_box])
# Chan-DOFs
channel_dof_file_widg = ipywidgets.Text(value=str(chan_dofs_file), description='Channel-DOF-Assignments File', layout={'width':'800px'}, style={'description_width': '200px'})
channel_dof_file_widg.observe(handler=lambda change: config_dict.update({'chan_dofs_file':Path(change['new'])}), names='value', type='change')
channel_dof_text = ipywidgets.Textarea(placeholder='Load channel-DOF-assignments file', layout={'width':'800px', 'height':'400px'})
channel_dof_save_btn = ipywidgets.Button(description='Save')
channel_dof_load_btn = ipywidgets.Button(description='Load')
channel_dof_save_btn.on_click(lambda b: save_contents(channel_dof_text, channel_dof_file_widg.value))
channel_dof_load_btn.on_click(lambda b: read_and_display(channel_dof_text, channel_dof_file_widg.value))
channel_dof_load_btn.click()
btn_box = ipywidgets.HBox([channel_dof_load_btn, channel_dof_save_btn])
channel_dof_box = ipywidgets.VBox([channel_dof_file_widg, channel_dof_text, btn_box])
# Chan-DOFs
oma_conf_file_widg = ipywidgets.Text(value=str(oma_conf_file), description='Modal Analysis Config File', layout={'width':'800px'}, style={'description_width': '200px'})
oma_conf_file_widg.observe(handler=lambda change: config_dict.update({'oma_conf_file':Path(change['new'])}), names='value', type='change')
oma_conf_text = ipywidgets.Textarea(placeholder='Load modal analysis config file', layout={'width':'800px', 'height':'400px'})
oma_conf_save_btn = ipywidgets.Button(description='Save')
oma_conf_load_btn = ipywidgets.Button(description='Load')
oma_conf_save_btn.on_click(lambda b: save_contents(oma_conf_text, oma_conf_file_widg.value))
oma_conf_load_btn.on_click(lambda b: read_and_display(oma_conf_text, oma_conf_file_widg.value))
oma_conf_load_btn.click()
btn_box = ipywidgets.HBox([oma_conf_load_btn, oma_conf_save_btn])
oma_conf_box = ipywidgets.VBox([oma_conf_file_widg, oma_conf_text, btn_box])
children = [general_box, geometry_tab, setup_info_box, channel_dof_box, oma_conf_box]
tab = ipywidgets.Tab()
tab.children = children
tab.titles = tab_contents
return tab
