Skip to content

Commit 0e1c924

Browse files
Chiu PeterChiu Peter
authored andcommitted
add retain_atom_idx
1 parent d16ac38 commit 0e1c924

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

crystal_toolkit/renderables/site.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def get_site_scene(
137137
phiEnd=phiEnd,
138138
clickable=True,
139139
tooltip=name,
140-
_meta=[site_idx],
140+
_meta=[site_idx] if retain_atom_idx else None,
141141
)
142142
atoms.append(sphere)
143143

@@ -210,7 +210,9 @@ def get_site_scene(
210210
radius=bond_radius / 2,
211211
clickable=True,
212212
tooltip=name_cyl,
213-
_meta=[site_idx, connected_site.index],
213+
_meta=[site_idx, connected_site.index]
214+
if retain_atom_idx
215+
else None,
214216
)
215217
)
216218
trans_vector = trans_vector + 0.25 * max_radius
@@ -222,7 +224,9 @@ def get_site_scene(
222224
radius=bond_radius,
223225
clickable=True,
224226
tooltip=name_cyl,
225-
_meta=[site_idx, connected_site.index],
227+
_meta=[site_idx, connected_site.index]
228+
if retain_atom_idx
229+
else None,
226230
)
227231
bonds.append(cylinder)
228232

@@ -233,7 +237,7 @@ def get_site_scene(
233237
radius=bond_radius,
234238
clickable=True,
235239
tooltip=name_cyl,
236-
_meta=[site_idx, connected_site.index],
240+
_meta=[site_idx, connected_site.index] if retain_atom_idx else None,
237241
)
238242
bonds.append(cylinder)
239243
all_positions.append(connected_position.tolist())
@@ -257,7 +261,7 @@ def get_site_scene(
257261
positionPairs=[[position, bond_midpoint.tolist()]],
258262
color=color,
259263
radius=bond_radius,
260-
_meta=[site_idx, connected_site.index],
264+
_meta=[site_idx, connected_site.index] if retain_atom_idx else None,
261265
)
262266
bonds.append(cylinder)
263267
all_positions.append(connected_position.tolist())

0 commit comments

Comments
 (0)