# Create a PointSet
# =================

# A `pyvista.PointSet` is a concrete class representing a set of points
# that specifies the interface for datasets that explicitly use "point" arrays to
# represent geometry. This class is useful for improving the performance of
# filters on point clouds.
# This example shows the performance improvement when clipping using the
# `pyvista.DataObjectFilters.clip()` filter on a `pyvista.PointSet`.
import time

from pyvista import examples

lidar = examples.download_lidar()

tstart = time.time()
clipped = lidar.clip(origin=(0, 0, 1.76e3), normal=(0, 0, 1))
t_elapsed = time.time() - tstart
print(f'Time to clip with a PolyData {t_elapsed:.2f} seconds.')

# Plot the clipped polydata
clipped.plot(show_scalar_bar=False)

# Show the performance improvement when using a PointSet.

# NOTE:
#     For VTK 9.4.X, there is no performance improvement due to a VTK bug which
#     was introduced into this version. See https://gitlab.kitware.com/vtk/vtk/-/issues/19649.

# pset = lidar.cast_to_pointset()

lidar_pset = lidar.cast_to_pointset()
tstart = time.time()
clipped_pset = lidar_pset.clip(origin=(0, 0, 1.76e3), normal=(0, 0, 1))
t_elapsed = time.time() - tstart
print(f'Time to clip with a PointSet {t_elapsed:.2f} seconds.')

# Plot the same dataset.

# NOTE:
#     PyVista must still create an intermediate PolyData to be able to plot, so
#     there is no performance improvement when using a `pyvista.PointSet`

clipped_pset.plot(show_scalar_bar=False)

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