"""
ZnVis: A Zincwarecode package.
License
-------
This program and the accompanying materials are made available under the terms
of the Eclipse Public License v2.0 which accompanies this distribution, and is
available at https://www.eclipse.org/legal/epl-v20.html
SPDX-License-Identifier: EPL-2.0
Copyright Contributors to the Zincwarecode Project.
Contact Information
-------------------
email: zincwarecode@gmail.com
github: https://github.com/zincware
web: https://zincwarecode.com/
Citation
--------
If you use this module please cite us with:
Summary
-------
Create a custom mesh
"""
from dataclasses import dataclass
import numpy as np
import open3d as o3d
from znvis.transformations.rotation_matrices import rotation_matrix
from .mesh import Mesh
[docs]@dataclass
class CustomMesh(Mesh):
"""
A class to produce cylinder meshes.
Attributes
----------
file : str
Path to mesh file.
"""
file: str = None
[docs] def create_mesh(
self, starting_position: np.ndarray, starting_orientation: np.ndarray = None
) -> o3d.geometry.TriangleMesh:
"""
Create a mesh object defined by the dataclass.
Parameters
----------
starting_position : np.ndarray shape=(3,)
Starting position of the mesh.
starting_orientation : np.ndarray shape=(3,) (default = None)
Starting orientation of the mesh.
Returns
-------
mesh : o3d.geometry.TriangleMesh
"""
mesh = o3d.io.read_triangle_mesh(self.file)
mesh.compute_vertex_normals()
mesh.translate(starting_position.astype(float))
if starting_orientation is not None:
matrix = rotation_matrix(np.array([0, 0, 1]), starting_orientation)
mesh.rotate(matrix)
mesh.paint_uniform_color(self.colour)
return mesh