test_remove_objects.py 899 B

123456789101112131415161718192021222324252627282930313233
  1. from vedo import *
  2. from vedo.pyplot import matrix
  3. def func(event):
  4. if not event.object:
  5. return
  6. if event.object.name == "Sphere":
  7. sph = Sphere()
  8. arr = np.random.rand(sph.npoints)*np.random.rand()
  9. sph.cmap("Blues", arr)
  10. # sph.add_scalarbar(title="Elevation", c="k")
  11. sph.add_scalarbar3d(title="Elevation", c="k")
  12. plt.remove("Sphere").add(sph).render()
  13. if event.object.name == "Matrix":
  14. arr = np.eye(n, m)/2 + np.random.randn(n, m)*0.1
  15. mat = matrix(arr, scale=0.15).scale(2).y(2)
  16. plt.remove("Matrix").add(mat).render()
  17. sph = Sphere()
  18. n, m = (6, 5)
  19. mat = matrix(
  20. np.eye(n, m)/2 + np.random.randn(n, m)*0.1,
  21. scale=0.15, # size of bin labels; set it to 0 to remove labels
  22. ).scale(2).y(2)
  23. plt = Plotter()
  24. plt.add_callback("mouse left click", func)
  25. plt.show(sph, mat, 'click to change random data')