def add_coupling_plots(
report: reporting.Report,
trials: typing.List[dict]
):
"""
Generates coupling report data from the analyzed coupling data and the
individual simulation trial results
:param report:
Grouped coupling data from the grouped simulation results
:param trials:
Simulation results for each trial run by the group
"""
distribution_traces = []
population_traces = []
index = 0
for trial in trials:
trial_data = reader.trial(trial['id'], paths.results())
index += 1
coupling_data = trial_data['couplings']
distribution_traces.append(dict(
x=coupling_data['distribution_profile']['x'],
y=coupling_data['distribution_profile']['y'],
type='scatter',
mode='lines',
name='{}'.format(index)
))
population_traces.append(dict(
x=coupling_data['population'],
type='box',
name='{}'.format(index),
boxpoints=False
))
report.add_plotly(
data=distribution_traces,
layout=plotting.create_layout(
title='Coupling Length Trial Distributions',
x_label='Expectation Value (au)',
y_label='Coupling Length (m)'
)
)
report.add_plotly(
data=population_traces,
layout=plotting.create_layout(
title='Coupling Length Trials',
x_label='Coupling Length (m)',
y_label='Trial Index (#)'
)
)
def plot_lengths(
report: Report,
coupling_data: dict
):
"""
:param report:
:param coupling_data:
:param times:
:param sampled:
:return:
"""
times = [e.time for e in coupling_data['lengths']]
lengths = [e.value for e in coupling_data['lengths']]
values, uncertainties = mstats.values.unzip(lengths)
plot = plotting.make_line_data(
times,
values,
uncertainties,
name='Length'
)
report.add_plotly(
data=plot['data'],
layout=plotting.create_layout(
plot['layout'],
'Coupling Lengths',
'Activity Cycle (#)',
'Length (m)'
)
)
def plot_deviations(
report: Report,
coupling_data: dict
):
"""
:param report:
:param coupling_data:
:return:
"""
times = [e.time for e in coupling_data['lengths']]
deviations = []
for dev in coupling_data['deviations']:
deviations.append(100.0 * dev)
values, uncertainties = mstats.values.unzip(deviations)
plot = plotting.make_line_data(
times,
values,
uncertainties,
name='Deviation',
color=plotting.get_color(1, as_string=True)
)
report.add_plotly(
data=plot['data'],
layout=plotting.create_layout(
plot['layout'],
'Coupling Length Median Deviations',
'Activity Cycle (#)',
'Median Deviation (%)'
)
)
def plot_strides(
report: Report,
advancement_data: dict
):
"""
:param report:
:param advancement_data:
:return:
"""
keys = (
(0, 'left_pes_strides'),
(1, 'right_pes_strides'),
(2, 'left_manus_strides'),
(4, 'right_manus_strides')
)
traces = []
layout = dict()
for entry in keys:
color_index = entry[0]
key = entry[1]
times = [e.time for e in advancement_data[key]]
values = [e.value for e in advancement_data[key]]
values, uncertainties = mstats.values.unzip(values)
name = ' '.join(key.split('_')[:2]).capitalize()
plot = plotting.make_line_data(
times,
values,
uncertainties,
name=name,
color=plotting.get_color(color_index, 0.5, True),
fill_color=plotting.get_color(color_index, 0.2, True)
)
traces.extend(plot['data'])
layout.update(plot['layout'])
report.add_plotly(
data=traces,
layout=plotting.create_layout(
layout,
'Stepping Distances',
'Activity Cycle (#)',
'Distance (m)'
)
)
def plot_advance(report: Report, coupling_data: dict):
"""
:param report:
:param coupling_data:
:param times:
:return:
"""
traces = []
sources = dict(
rear_advance={
'name': 'Rear',
'index': 0
},
forward_advance={
'name': 'Forward',
'index': 1
}
)
for key, data in sources.items():
events = coupling_data[key]
times = [e.time for e in events]
values, uncertainties = mstats.values.unzip([e.value for e in events])
plot = plotting.make_line_data(
x=times,
y=values,
y_unc=uncertainties,
name=data['name'],
color=plotting.get_color(data['index'], 0.6, as_string=True),
fill_color=plotting.get_color(data['index'], 0.2, as_string=True)
)
traces += plot['data']
report.add_plotly(
data=traces,
layout=plotting.create_layout(
title='Coupler Rate of Advancement',
x_label='Activity Cycle (#)',
y_label='Advance Rate (m/cycle)'
)
)
def plot_distribution(
report: Report,
coupling_data: dict
):
"""
:param report:
:param coupling_data:
:return:
"""
report.add_plotly(
data=[go.Scatter(
x=coupling_data['distribution_profile']['x'],
y=coupling_data['distribution_profile']['y'],
mode='lines'
)],
layout=plotting.create_layout(
title='Coupling Length Distribution',
x_label='Coupling Distance (m)',
y_label='Expectation (au)'
)
)
def add_length_plot(report: Report, separation_data: dict, times:dict):
"""
:param report:
:param separation_data:
:param times:
:return:
"""
sources = [
dict(
layout=plotting.create_layout(
title='Left & Right Separations',
x_label='Activity Cycle (#)',
y_label='Separation (m)'
),
items=[
dict(
name='Left',
data=separation_data['left_lengths'],
index=2
),
dict(
name='Right',
data=separation_data['right_lengths'],
index=4
)
]
),
dict(
layout=plotting.create_layout(
title='Forward & Rear Separations',
x_label='Activity Cycle (#)',
y_label='Separation (m)'
),
items=[
dict(
name='Forward',
data=separation_data['forward_lengths'],
index=1
),
dict(
name='Rear',
data=separation_data['rear_lengths'],
index=0
)
]
)
]
for source in sources:
traces = []
for item in source['items']:
values, uncertainties = mstats.values.unzip(item['data'])
color = plotting.get_color(item['index'], 0.7, as_string=True)
fill = plotting.get_color(item['index'], 0.2, as_string=True)
plot = plotting.make_line_data(
x=times['cycles'],
y=values,
y_unc=uncertainties,
name=item['name'],
color=color,
fill_color=fill
)
traces += plot['data']
report.add_plotly(data=traces, layout=source['layout'])