from compas.datastructures import Mesh
from compas.geometry import add_vectors
from compas.geometry import distance_point_point
from compas.geometry import subtract_vectors
from compas_blender.geometry._geometry import BaseBlenderGeometry
__all__ = [
'BlenderMesh',
]
[docs]class BlenderMesh(BaseBlenderGeometry):
[docs] def __init__(self, obj):
super().__init__(obj)
[docs] def get_vertex_coordinates(self, vertex):
return add_vectors(self.location, self.geometry.vertices[vertex].co)
[docs] def get_vertices_coordinates(self):
xyzs = [vertex.co for vertex in self.geometry.vertices]
return {vertex: add_vectors(self.location, xyz) for vertex, xyz in enumerate(xyzs)}
[docs] def set_vertices_coordinates(self, xyzs):
for vertex, xyz in xyzs.items():
self.geometry.vertices[vertex].co = subtract_vectors(xyz, self.location)
[docs] def get_vertices_colors(self, vertices=None):
colors = {}
col = self.geometry.vertex_colors.active
if col:
if not vertices:
vertices = range(len(self.geometry.vertices))
for face in self.geometry.polygons:
for i in face.loop_indices:
j = self.geometry.loops[i].vertex_index
if (j in vertices) and (not colors.get(j, None)):
colors[j] = list(col.data[i].color)[:3]
return colors
[docs] def set_vertices_colors(self, colors):
if self.geometry.vertex_colors:
col = self.geometry.vertex_colors.active
else:
col = self.geometry.vertex_colors.new()
for face in self.geometry.polygons:
for i in face.loop_indices:
j = self.geometry.loops[i].vertex_index
if j in colors:
col.data[i].color = list(colors[j]) + [1]
[docs] def unset_vertices_colors(self):
vertex_colors = self.geometry.vertex_colors
while vertex_colors:
vertex_colors.remove(vertex_colors[0])
[docs] def get_vertex_index(self):
raise NotImplementedError
[docs] def get_vertices_indices(self):
raise NotImplementedError
[docs] def get_vertex_face_indices(self):
raise NotImplementedError
[docs] def get_vertices_face_indices(self):
raise NotImplementedError
[docs] def get_edge_vertex_indices(self, edge):
return list(self.geometry.edges[edge].vertices)
[docs] def get_edges_vertex_indices(self, edges=None):
if not edges:
edges = range(len(self.geometry.edges))
return {edge: self.get_edge_vertex_indices(edge=edge) for edge in edges}
[docs] def get_edge_index(self):
raise NotImplementedError
[docs] def get_edges_indices(self):
raise NotImplementedError
[docs] def edge_length(self, edge):
u, v = self.geometry.edges[edge].vertices
sp, ep = [list(self.geometry.vertices[i].co) for i in [u, v]]
return distance_point_point(sp, ep)
[docs] def edges_lengths(self, edges=None):
if not edges:
edges = range(len(self.geometry.edges))
return {edge: self.edge_length(edge=edge) for edge in edges}
[docs] def get_face_vertex_indices(self, face):
return list(self.geometry.polygons[face].vertices)
[docs] def get_faces_vertex_indices(self, faces=None):
if not faces:
faces = range(len(self.geometry.polygons))
return {face: self.get_face_vertex_indices(face=face) for face in faces}
[docs] def get_face_vertex_coordinates(self, face):
raise NotImplementedError
[docs] def get_faces_vertex_coordinates(self, faces=None):
raise NotImplementedError
[docs] def get_face_index(self):
raise NotImplementedError
[docs] def get_faces_indices(self):
raise NotImplementedError
[docs] def face_normal(self, face):
return list(self.geometry.polygons[face].normal)
[docs] def faces_normals(self, faces=None):
if not faces:
faces = range(len(self.geometry.polygons))
return {face: self.face_normal(face=face) for face in faces}
[docs] def face_area(self, face):
return self.geometry.polygons[face].area
[docs] def faces_areas(self, faces=None):
if not faces:
faces = range(len(self.geometry.polygons))
return {face: self.face_area(face=face) for face in faces}
[docs] def bevel(self, width=0.2, segments=1, only_vertices=False):
self.object.modifiers.new('bevel', type='BEVEL')
self.object.modifiers['bevel'].width = width
self.object.modifiers['bevel'].segments = segments
self.object.modifiers['bevel'].use_only_vertices = only_vertices
self.refresh()
[docs] def subdivide(self, levels=1, type='SIMPLE'):
self.object.modifiers.new('subdivision', type='SUBSURF')
self.object.modifiers['subdivision'].levels = levels
self.object.modifiers['subdivision'].subdivision_type = type # or 'CATMULL_CLARK'
self.refresh()
[docs] def triangulate(self):
self.object.modifiers.new('triangulate', type='TRIANGULATE')
self.refresh()
[docs] def get_vertices_and_faces(self):
vertices = self.get_vertices_coordinates()
faces = self.get_faces_vertex_indices()
return vertices, faces
[docs] def get_border(self):
raise NotImplementedError
[docs] def closest_point(self, point, maxdist=None):
raise NotImplementedError
[docs] def closest_points(self, points, maxdist=None):
raise NotImplementedError
[docs] def to_datastructure(self):
vertices = self.get_vertices_coordinates().values()
faces = self.get_faces_vertex_indices().values()
return Mesh.from_vertices_and_faces(vertices=vertices, faces=faces)
# ==============================================================================
# Main
# ==============================================================================
if __name__ == '__main__':
from compas_blender.utilities import get_object_by_name
object = get_object_by_name(name='Cube')
mesh = BlenderMesh(object)
print(mesh.geometry)
print(mesh.object)
print(mesh.otype)
print(mesh.name)
print(mesh.location)
print(mesh.get_vertex_coordinates(vertex=0))
print(mesh.get_vertices_coordinates())
# mesh.set_vertices_coordinates({0: [0, 0, 0], 1: [1, 1, 0]})
mesh.set_vertices_colors(colors={0: [1, 0, 0], 1: [0, 0, 1], 2: [0, 1, 0]})
print(mesh.get_vertices_colors())
mesh.unset_vertices_colors()
print(mesh.get_edge_vertex_indices(edge=4))
print(mesh.get_edges_vertex_indices())
print(mesh.edge_length(edge=2))
print(mesh.edges_lengths())
print(mesh.get_face_vertex_indices(face=4))
print(mesh.get_faces_vertex_indices())
print(mesh.face_normal(face=2))
print(mesh.faces_normals())
print(mesh.face_area(face=2))
print(mesh.faces_areas())
print(mesh.get_vertices_and_faces())
print(mesh.to_datastructure())
mesh.bevel(width=0.2, segments=1, only_vertices=False)
mesh.subdivide(levels=2, type='CATMULL_CLARK')
mesh.triangulate()