# Wrapping Other Objects
# ======================

# You can `wrap()` several other object types using pyvista including:
# `numpy` arrays
# `trimesh.Trimesh` meshes
# VTK objects
# This allows for the "best of both worlds" programming special to
# Python due to its modularity.  If there's some limitation of pyvista
# (or trimesh), then you can adapt your scripts to use the best features
# of more than one module.
import numpy as np
import pyvista as pv
import trimesh
import vtk

# Wrap Point Arrays
# -----------------

# Wrap a point cloud composed of random points from numpy.
rng = np.random.default_rng(seed=0)
points = rng.random((30, 3))
cloud = pv.wrap(points)
pv.plot(
    cloud,
    scalars=points[:, 2],
    render_points_as_spheres=True,
    point_size=50,
    opacity=points[:, 0],
    cpos='xz',
)

# Wrap Trimesh Objects
# --------------------

# Create a simple `trimesh.Trimesh` and wrap it.
points = [[0, 0, 0], [0, 0, 1], [0, 1, 0]]
faces = [[0, 1, 2]]
tmesh = trimesh.Trimesh(points, faces=faces, process=False)
mesh = pv.wrap(tmesh)
print(mesh)

# We can also convert to a `trimesh.Trimesh` mesh with `to_trimesh()`.
# First create `PolyData` with data.
mesh = pv.Sphere()
mesh.point_data['point_ids'] = np.arange(mesh.n_points)
mesh.cell_data['cell_ids'] = np.arange(mesh.n_cells)
mesh.field_data['misc'] = np.array([1, 2, 3])
mesh.user_dict['name'] = 'sphere'
mesh.user_dict['number'] = 42

# Now convert it.
tmesh = pv.to_trimesh(mesh)
print(tmesh)

# Point data is accessible via `vertex_attributes`.
print(tmesh.vertex_attributes.keys())

# Cell data is accessible via `face_attributes`.
print(tmesh.face_attributes.keys())

# Field data is accessible via `metadata`. Both field data keys and
# `user_dict` keys are stored.
print(tmesh.metadata)

# Use `from_trimesh()` to convert it back to a `PolyData`
# mesh. This is the same as using `wrap()`.
pvmesh = pv.from_trimesh(tmesh)

# The data contained in the `trimesh` object is recovered as point data,
print(pvmesh.point_data.keys())

# cell data,
print(pvmesh.cell_data.keys())

# and field data, including field data stored in the
# `user_dict`:
print(pvmesh.field_data.keys())
print(pvmesh.user_dict)

# Wrap VTK Meshes
# ---------------

# Wrap an instance of vtkPolyData
points = vtk.vtkPoints()
p = [1.0, 2.0, 3.0]
vertices = vtk.vtkCellArray()
pid = points.InsertNextPoint(p)
vertices.InsertNextCell(1)
vertices.InsertCellPoint(pid)
point = vtk.vtkPolyData()
point.SetPoints(points)
point.SetVerts(vertices)
mesh = pv.wrap(point)
print(mesh)

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