3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-15 02:37:49 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Handle Trimesh returning scene of objects

Browse source 
These objects are added as a group node, if they are plural.

Contributes to issue CURA-6739.
This commit is contained in:
Ghostkeeper 2019-08-29 14:56:49 +02:00
parent cead90c5ba
commit 1f534db508
No known key found for this signature in database
GPG key ID: 86BEF881AE2CF276
1 changed files with 42 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

56
plugins/TrimeshReader/TrimeshReader.py
View file

@ -3,6 +3,7 @@
# The _toMeshData function is taken from the AMFReader class which was built by fieldOfView. # The _toMeshData function is taken from the AMFReader class which was built by fieldOfView.
from typing import List, Union, TYPE_CHECKING
import numpy # To create the mesh data. import numpy # To create the mesh data.
import os.path # To create the mesh name for the resulting mesh. import os.path # To create the mesh name for the resulting mesh.
import trimesh # To load the files into a Trimesh. import trimesh # To load the files into a Trimesh.
@ -10,12 +11,17 @@ import trimesh # To load the files into a Trimesh.
from UM.Mesh.MeshData import MeshData, calculateNormalsFromIndexedVertices # To construct meshes from the Trimesh data. from UM.Mesh.MeshData import MeshData, calculateNormalsFromIndexedVertices # To construct meshes from the Trimesh data.
from UM.Mesh.MeshReader import MeshReader # The plug-in type we're extending. from UM.Mesh.MeshReader import MeshReader # The plug-in type we're extending.
from UM.MimeTypeDatabase import MimeTypeDatabase, MimeType # To add file types that we can open. from UM.MimeTypeDatabase import MimeTypeDatabase, MimeType # To add file types that we can open.
from UM.Scene.GroupDecorator import GroupDecorator # Added to the parent node if we load multiple nodes at once.
from cura.CuraApplication import CuraApplication from cura.CuraApplication import CuraApplication
from cura.Scene.BuildPlateDecorator import BuildPlateDecorator # Added to the resulting scene node. from cura.Scene.BuildPlateDecorator import BuildPlateDecorator # Added to the resulting scene node.
from cura.Scene.ConvexHullDecorator import ConvexHullDecorator # Added to group nodes if we load multiple nodes at once.
from cura.Scene.CuraSceneNode import CuraSceneNode # To create a node in the scene after reading the file. from cura.Scene.CuraSceneNode import CuraSceneNode # To create a node in the scene after reading the file.
from cura.Scene.SliceableObjectDecorator import SliceableObjectDecorator # Added to the resulting scene node. from cura.Scene.SliceableObjectDecorator import SliceableObjectDecorator # Added to the resulting scene node.
if TYPE_CHECKING:
from UM.Scene.SceneNode import SceneNode
## Class that leverages Trimesh to import files. ## Class that leverages Trimesh to import files.
class TrimeshReader(MeshReader): class TrimeshReader(MeshReader):
def __init__(self) -> None: def __init__(self) -> None:
@ -34,21 +40,43 @@ class TrimeshReader(MeshReader):
# \param file_name The file path. This is assumed to be one of the file # \param file_name The file path. This is assumed to be one of the file
# types that Trimesh can read. It will not be checked again. # types that Trimesh can read. It will not be checked again.
# \return A scene node that contains the file's contents. # \return A scene node that contains the file's contents.
def _read(self, file_name: str) -> CuraSceneNode: def _read(self, file_name: str) -> Union["SceneNode", List["SceneNode"]]:
mesh = trimesh.load(file_name) mesh_or_scene = trimesh.load(file_name)
mesh.merge_vertices() meshes = []
mesh.remove_unreferenced_vertices() if isinstance(mesh_or_scene, trimesh.Trimesh):
mesh.fix_normals() meshes = [mesh_or_scene]
mesh_data = self._toMeshData(mesh) elif isinstance(mesh_or_scene, trimesh.Scene):
meshes = [mesh for mesh in mesh_or_scene.geometry.values()]
file_base_name = os.path.basename(file_name) active_build_plate = CuraApplication.getInstance().getMultiBuildPlateModel().activeBuildPlate
new_node = CuraSceneNode() nodes = []
new_node.setMeshData(mesh_data) for mesh in meshes:
new_node.setSelectable(True) if not isinstance(mesh, trimesh.Trimesh): # Trimesh can also receive point clouds, 2D paths, 3D paths or metadata. Skip those.
new_node.setName(file_base_name) continue
new_node.addDecorator(BuildPlateDecorator(CuraApplication.getInstance().getMultiBuildPlateModel().activeBuildPlate)) mesh.merge_vertices()
new_node.addDecorator(SliceableObjectDecorator()) mesh.remove_unreferenced_vertices()
return new_node mesh.fix_normals()
mesh_data = self._toMeshData(mesh)
file_base_name = os.path.basename(file_name)
new_node = CuraSceneNode()
new_node.setMeshData(mesh_data)
new_node.setSelectable(True)
new_node.setName(file_base_name if len(meshes) == 1 else "{file_base_name} {counter}".format(file_base_name = file_base_name, counter = str(len(nodes) + 1)))
new_node.addDecorator(BuildPlateDecorator(active_build_plate))
new_node.addDecorator(SliceableObjectDecorator())
nodes.append(new_node)
if len(nodes) == 1:
return nodes[0]
# Add all nodes to a group so they stay together.
group_node = CuraSceneNode()
group_node.addDecorator(GroupDecorator())
group_node.addDecorator(ConvexHullDecorator())
group_node.addDecorator(BuildPlateDecorator(active_build_plate))
for node in nodes:
node.setParent(group_node)
return group_node
## Converts a Trimesh to Uranium's MeshData. ## Converts a Trimesh to Uranium's MeshData.
# \param tri_node A Trimesh containing the contents of a file that was # \param tri_node A Trimesh containing the contents of a file that was