def dunlop(palint_object, ax,
component='m', norm_factor=1,
plt_idx=0,
out='show', folder=None, name='output.pdf',
plt_opt={},
**options):
components = {'x': 1, 'y': 2, 'z': 3, 'm': 4}
idx = components[component]
colors = helper.get_colors()
marker = helper.get_marker()
ax.plot(palint_object.th[:, 0], palint_object.th[:, idx] / norm_factor, '.-',
marker=marker[plt_idx],
color=colors[0], **plt_opt)
ax.plot(palint_object.sum[:, 0], palint_object.sum[:, idx] / norm_factor, '.-',
marker=marker[plt_idx],
color=colors[1], **plt_opt)
ax.plot(palint_object.ptrm[:, 0], palint_object.ptrm[:, idx] / norm_factor, '.-',
marker=marker[plt_idx],
color=colors[2], **plt_opt)
# todo write dunlop_add_tr()
# ax.plot(palint_object.tr[:, 0], palint_object.tr[:, idx] / norm_factor,
# color='k', marker=marker[plt_idx], alpha=0.8, ls='')
dunlop_std(palint_object=palint_object, ax=ax, component=component, norm_factor=norm_factor, plt_idx=plt_idx,
plt_opt=plt_opt)
return ax
def dunlop_std(palint_object, ax,
component='m', norm_factor=1,
plt_idx=0,
out='show', folder=None, name='output.pdf',
plt_opt={}):
components = {'x': 1, 'y': 2, 'z': 3, 'm': 4}
idx = components[component]
colors = helper.get_colors()
if palint_object.th_stdev is not None:
try:
ax.fill_between(palint_object.th[:, 0],
(palint_object.th[:, idx] - palint_object.th_stdev[:, idx]) / norm_factor,
(palint_object.th[:, idx] + palint_object.th_stdev[:, idx]) / norm_factor,
color=colors[0], alpha=0.2)
ax.fill_between(palint_object.sum[:, 0],
(palint_object.sum[:, idx] - palint_object.sum_stdev[:, idx]) / norm_factor,
(palint_object.sum[:, idx] + palint_object.sum_stdev[:, idx]) / norm_factor,
color=colors[1], alpha=0.2)
ax.fill_between(palint_object.ptrm[:, 0],
(palint_object.ptrm[:, idx] - palint_object.ptrm_stdev[:, idx]) / norm_factor,
(palint_object.ptrm[:, idx] + palint_object.ptrm_stdev[:, idx]) / norm_factor,
color=colors[2], alpha=0.2)
except AttributeError:
return
return ax
def add_ck_check(palint_object, ax, component='m', norm=None, norm_factor=[1, 1], plt_idx=0, plt_opt=None, **options):
if not plt_opt: plt_opt = {}
colors = helper.get_colors()
idx = palint_object.components[component]
check_data = palint_object._get_ck_data()
for i in check_data:
ck_i = i[0][idx] / norm_factor[0]
th_i = i[1][idx] / norm_factor[0]
ptrm_j = i[2][idx] / norm_factor[1]
th_j = i[3][idx] / norm_factor[1]
hline = lines.Line2D([ck_i, ptrm_j], [th_j, th_j], color=colors[plt_idx], linewidth=1.2)
vline = lines.Line2D([ck_i, ck_i], [th_j, th_i], color=colors[plt_idx], linewidth=1.2)
ax.add_line(hline)
ax.add_line(vline)
ax.plot(ck_i, th_i, '^', markeredgecolor=colors[plt_idx], markerfacecolor='w')
def add_ac_check(palint_object, ax, component='m', norm=None,
norm_factor=[1, 1],
plt_idx=0, plt_opt={},
**options):
colors = helper.get_colors()
idx = palint_object.components[component]
check_data = palint_object._get_ac_data()
for i in check_data:
ac_i = i[0][idx] / norm_factor[0]
th_i = i[1][idx] / norm_factor[1]
ptrm_j = i[2][idx] / norm_factor[0]
th_j = i[3][idx] / norm_factor[1]
vline = lines.Line2D([ptrm_j, ptrm_j], [th_j, th_i], color=colors[plt_idx], ls='--', linewidth=1.2)
hline = lines.Line2D([ac_i, ptrm_j], [th_i, th_i], color=colors[plt_idx], ls='--', linewidth=1.2)
ax.add_line(hline)
ax.add_line(vline)
ax.plot(ac_i, th_i, 's', markeredgecolor=colors[plt_idx], markerfacecolor='w', linewidth=2)
