zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add legend items for propagation vectors

Browse Source
This commit is contained in:
usov_i 2022-05-04 17:20:24 +02:00
parent 181c359ef9
commit 72fd54d268
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyzebra/app/panel_ccl_prepare.py
View File

@ -657,13 +657,20 @@ def create():
text=["h", "k"], text=["h", "k"],
) )
# Generate legend for different file entries (symbol) # Legend items for different file entries (symbol)
legend_items = [] legend_items = []
if not res_flag and file_flag: if not res_flag and file_flag:
labels, inds = np.unique(scan_source.data["l"], return_index=True) labels, inds = np.unique(scan_source.data["l"], return_index=True)
for label, ind in zip(labels, inds): for label, ind in zip(labels, inds):
legend_items.append(LegendItem(label=label, renderers=[scatter], index=ind)) legend_items.append(LegendItem(label=label, renderers=[scatter], index=ind))
# Legend items for propagation vector (color)
if prop_legend_flag:
labels, inds = np.unique(scan_source.data["c"], return_index=True)
for label, ind in zip(labels, inds):
label = f"k={k[cmap.index(label)]}"
legend_items.append(LegendItem(label=label, renderers=[mline], index=ind))
plot.legend.items = legend_items plot.legend.items = legend_items
plot_file = Button(label="Plot selected file(s)", button_type="primary", width=200) plot_file = Button(label="Plot selected file(s)", button_type="primary", width=200)
@ -697,7 +704,7 @@ def create():
) )
scan_source = ColumnDataSource(dict(xs=[], ys=[], x=[], y=[], m=[], s=[], c=[], l=[])) scan_source = ColumnDataSource(dict(xs=[], ys=[], x=[], y=[], m=[], s=[], c=[], l=[]))
plot.add_glyph(scan_source, MultiLine(xs="xs", ys="ys", line_color="c")) mline = plot.add_glyph(scan_source, MultiLine(xs="xs", ys="ys", line_color="c"))
scatter = plot.add_glyph( scatter = plot.add_glyph(
scan_source, Scatter(x="x", y="y", marker="m", size="s", fill_color="c", line_color="c") scan_source, Scatter(x="x", y="y", marker="m", size="s", fill_color="c", line_color="c")
) )