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

Fix legend when res_flag is True

Browse Source
This commit is contained in:
usov_i 2022-05-05 19:04:14 +02:00
parent 585d4b3f54
commit 2810cf3d2b
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyzebra/app/panel_ccl_prepare.py
View File

@ -623,10 +623,15 @@ def create():
# Legend items for propagation vector (color) # Legend items for propagation vector (color)
if prop_legend_flag: if prop_legend_flag:
labels, inds = np.unique(scan_source.data["c"], return_index=True) if res_flag:
source, render = ellipse_source, ellipse
else:
source, render = scan_source, mline
labels, inds = np.unique(source.data["c"], return_index=True)
for label, ind in zip(labels, inds): for label, ind in zip(labels, inds):
label = f"k={k[cmap.index(label)]}" label = f"k={k[cmap.index(label)]}"
legend_items.append(LegendItem(label=label, renderers=[mline], index=ind)) legend_items.append(LegendItem(label=label, renderers=[render], index=ind))
plot.legend.items = legend_items plot.legend.items = legend_items
@ -656,7 +661,7 @@ def create():
) )
ellipse_source = ColumnDataSource(dict(x=[], y=[], w=[], h=[], c=[])) ellipse_source = ColumnDataSource(dict(x=[], y=[], w=[], h=[], c=[]))
plot.add_glyph( ellipse = plot.add_glyph(
ellipse_source, Ellipse(x="x", y="y", width="w", height="h", fill_color="c", line_color="c") ellipse_source, Ellipse(x="x", y="y", width="w", height="h", fill_color="c", line_color="c")
) )