# Examples from pyvista.examples.downloads.download_caffeine
# ==========================================================

import pyvista as pv
from pyvista import examples
filename = examples.download_caffeine(load=False)
reader = pv.get_reader(filename)
poly = reader.read()

# Add atoms and bonds to the plotter.
pl = pv.Plotter()
atoms = pl.add_mesh(poly.glyph(geom=pv.Sphere(radius=0.1)), color='red')
bonds = pl.add_mesh(poly.tube(radius=0.1), color='gray')
pl.show(cpos='xy')

# SEE ALSO:
#     Caffeine Dataset https://docs.pyvista.org/api/examples/dataset_gallery.html#caffeine-dataset
#     See this dataset in the Dataset Gallery for more info.

# ----------------------------------------------------------------------
# Generated by sphinx-examples-as-code https://github.com/pyvista/sphinx-examples-as-code