Module sverchok.utils.modules.polygon_utils
Expand source code
# This file is part of project Sverchok. It's copyrighted by the contributors
# recorded in the version control history of the file, available from
# its original location https://github.com/nortikin/sverchok/commit/master
#
# SPDX-License-Identifier: GPL3
# License-Filename: LICENSE
import numpy as np
from mathutils import Vector
from mathutils.geometry import area_tri as area
from mathutils.geometry import tessellate_polygon as tessellate
from sverchok.data_structure import has_element
from sverchok.utils.math import np_normalize_vectors
from sverchok.utils.sv_bmesh_utils import bmesh_from_pydata
from sverchok.utils.modules.matrix_utils import vectors_center_axis_to_matrix
from sverchok.utils.modules.vertex_utils import vertex_shell_factor, adjacent_edg_pol, adjacent_edg_pol_idx
from sverchok.nodes.analyzer.mesh_filter import Faces
from .edge_utils import adjacent_faces_idx
def areas_from_polygons(verts, polygons, sum_faces=False):
'''
returns pols area as [float, float,...]
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
sum_faces if True it will return the sum of the areas as [float]
'''
areas = []
concat_area = areas.append
for polygon in polygons:
num = len(polygon)
if num == 3:
concat_area(area(verts[polygon[0]], verts[polygon[1]], verts[polygon[2]]))
elif num == 4:
area_1 = area(verts[polygon[0]], verts[polygon[1]], verts[polygon[2]])
area_2 = area(verts[polygon[0]], verts[polygon[2]], verts[polygon[3]])
concat_area(area_1 + area_2)
elif num > 4:
ngon_area = 0.0
subcoords = [Vector(verts[idx]) for idx in polygon]
for tri in tessellate([subcoords]):
ngon_area += area(*[verts[polygon[i]] for i in tri])
concat_area(ngon_area)
else:
concat_area(0)
if sum_faces:
areas = [sum(areas)]
return areas
def pols_perimeters(verts, polygons, sum_perimeters=False, output_numpy=False):
'''
returns pols perimeter as [float, float,...]
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
sum_perimeters if True it will return the sum of the perimenters as [float]
'''
vals = np_process_polygons(verts, polygons, func=np_faces_perimeters, dims=1, output_numpy=True)
if sum_perimeters:
if output_numpy:
return [np.sum(vals)]
return [np.sum(vals).tolist()]
if output_numpy:
return vals
return vals.tolist()
def pols_vertices(vertices, faces):
'''
Explodes geometry.
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
returns [[vertex, vertex, vertex,..], [vertex, vertex, vertex,..], ...] and [[polygon], [polygon],...]
'''
verts = [[vertices[c] for c in p] for p in faces]
pols = [[list(range(len(p)))] for p in faces]
vals = [verts, pols]
return vals
def pols_to_edges(obj, unique_edges=False):
'''
Generate edges from pols
obj: list as [[polygon, polygon,..], [polygon,...],...], being each polygon [int, int, ...].
unique_edges: if False will return all polygon edges if True will remove the doubled edges
'''
out = []
for faces in obj:
out_edges = []
seen = set()
for face in faces:
for edge in zip(face, list(face[1:]) + list([face[0]])):
if unique_edges and tuple(sorted(edge)) in seen:
continue
if unique_edges:
seen.add(tuple(sorted(edge)))
out_edges.append(edge)
out.append(out_edges)
return out
def pols_sides(faces, sum_sides=False):
'''
returns the number of sides of each polygon as [int, int, ... ]
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
sum_sides: if True will return the sum of sides of the whole object returning [int]
'''
vals = [len(p) for p in faces]
if sum_sides:
vals = [sum(vals)]
return vals
def pols_adjacent(pols):
"""
returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
"""
return [[pols[i] for i in p_indexes] for p_indexes in pols_adjacent_idx(pols)]
def pols_adjacent_idx(pols):
"""
returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
"""
edges = {tuple(sorted(e)): None for pol_edges in pols_edges(pols) for e in pol_edges}
edges = {e: adj_indexes for e, adj_indexes in zip(edges.keys(), adjacent_faces_idx(edges.keys(), pols))}
adj_pols = []
for ei, pol_edges in enumerate(pols_edges(pols)):
adj_pols.append(list({fi for e in pol_edges for fi in edges[tuple(sorted(e))]} - {ei}))
return adj_pols
def pols_adjacent_num(pols):
"""
returns the number polygons that share a vertex with each polygon [int, int,..]]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
"""
return [len(p) for p in pols_adjacent_idx(pols)]
def pols_neighbor(verts, pols):
'''
returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
v_adj = adjacent_edg_pol(verts, pols)
vals = []
for pol in pols:
pol_adj = []
for v_id in pol:
for related_pol in v_adj[v_id]:
if not related_pol in pol_adj:
pol_adj.append(related_pol)
pol_adj.remove(pol)
vals.append(pol_adj)
return vals
def pols_neighbor_idx(verts, pols):
'''
returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
v_adj = adjacent_edg_pol_idx(verts, pols)
vals = []
for idx, pol in enumerate(pols):
pol_adj = []
for v_id in pol:
for related_pol in v_adj[v_id]:
if not related_pol in pol_adj:
pol_adj.append(related_pol)
pol_adj.remove(idx)
vals.append(pol_adj)
return vals
def pols_neighbor_num(verts, pols):
'''
returns the number of polygons that share one edges with each polygon [int, int,...]
pols: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
return [len(p) for p in pols_neighbor(verts, pols)]
def pols_normals(vertices, faces, output_numpy):
'''
Returns Faces normals as [vector, vector,...]
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
vals = np_process_polygons(vertices, faces, func=np_faces_normals, output_numpy=output_numpy)
return vals
def np_faces_normals(v_pols):
pol_sides = v_pols.shape[1]
if pol_sides > 3:
f_normals = np.zeros((len(v_pols), 3), dtype=np.float64)
for i in range(pol_sides - 2):
f_normals += np.cross(v_pols[::, (1+i)%pol_sides] - v_pols[::, 0], v_pols[::, (2+i)%pol_sides] - v_pols[::, 0])
else:
f_normals = np.cross(v_pols[::, 1] - v_pols[::, 0], v_pols[::, 2] - v_pols[::, 0])
np_normalize_vectors(f_normals)
return f_normals
def np_faces_absolute_normals(v_pols):
return np_center_median(v_pols) + np_faces_normals(v_pols)
def pols_absolute_normals(vertices, faces, output_numpy):
'''
Returns Faces center + faces normals as [vector, vector,...]
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
return np_process_polygons(vertices, faces, func=np_faces_absolute_normals, output_numpy=output_numpy)
def pols_shell_factor(vertices, faces):
'''
Average of vertex shell_factor
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
edges: list as [edge, edge,..], being each edge [int, int].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
returns vals as [float, float,...]
'''
v_shell = vertex_shell_factor(vertices, [], faces)
vals = []
for face in faces:
p_shell = 0
for v_id in face:
p_shell += v_shell[v_id]
vals.append(p_shell/len(face))
return vals
def np_faces_perimeters(v_pols):
return np.sum(edges_lengths(v_pols), axis=1)
def vertex_weight_factor(v_pols):
ed_lengths = edges_lengths(v_pols)
perimeters = np.sum(ed_lengths, axis=1)
edges_prop = ed_lengths/perimeters[:, np.newaxis]
return (edges_prop + np.roll(edges_prop, 1, axis=1))/2
def np_center_median(v_pols):
return np.sum(v_pols, axis=1) / v_pols.shape[1]
def np_center_bbox(v_pols):
return (np.amin(v_pols, axis=1) + np.amax(v_pols, axis=1))/2
def np_center_weighted(v_pols):
v_factor = vertex_weight_factor(v_pols)
return np.sum(v_pols * v_factor[:, :, np.newaxis], axis=1)
def edges_lengths(v_pols):
return np.linalg.norm(v_pols-np.roll(v_pols, -1, axis=1), axis=2)
def np_tangent_longest_edge(v_pols):
edges_dir = v_pols-np.roll(v_pols, 1, axis=1)
ed_length = np.linalg.norm(edges_dir, axis=2)
ed_idx = np.argmax(ed_length, axis=1)
return np_normalize_vectors(edges_dir[np.arange(len(v_pols)), ed_idx, :])
def np_tangent_center_orig(v_pols):
return np_normalize_vectors(np_center_median(v_pols) - v_pols[:, 0, :])
def np_process_polygons(verts, faces, func=None, dims=3, output_numpy=False):
if not func:
return
if not (has_element(verts) and has_element(faces)):
return
if isinstance(verts, np.ndarray):
np_verts = verts
else:
np_verts = np.array(verts)
if isinstance(faces, np.ndarray):
np_faces = faces
else:
np_faces = np.array(faces)
if np_faces.dtype == object:
np_len = np.vectorize(len)
lens = np_len(np_faces)
pol_types = np.unique(lens)
if dims == 1:
vals = np.zeros(np_faces.shape[0], dtype=float)
else:
vals = np.zeros((np_faces.shape[0], dims), dtype=float)
for p in pol_types:
mask = lens == p
np_faces_g = np.array(np_faces[mask].tolist())
v_pols = np_verts[np_faces_g]
if dims == 1:
vals[mask] = func(v_pols)
else:
vals[mask, :] = func(v_pols)
else:
v_pols = np_verts[np_faces] #num pols, num sides
vals = func(v_pols)
if output_numpy:
return vals
return vals.tolist()
def pols_center(vertices, faces, origin, output_numpy):
'''
Cemter of faces
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
edges: list as [edge, edge,..], being each edge [int, int].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
origin: String that can be any key of pols_origin_modes_dict
returns vals as [float, float,...]
'''
if origin == 'Median Center':
centers_func = np_center_median
elif origin == 'Bounds Center':
centers_func = np_center_bbox
else:
centers_func = np_center_weighted
vals = np_process_polygons(vertices, faces, func=centers_func, output_numpy=output_numpy)
return vals
def pols_center_bounds(bm_faces):
return [tuple(bm_face.calc_center_bounds()) for bm_face in bm_faces]
def pols_center_median(bm_faces):
return [tuple(bm_face.calc_center_median()) for bm_face in bm_faces]
def pols_center_median_weighted(bm_faces):
return [tuple(bm_face.calc_center_median_weighted()) for bm_face in bm_faces]
def pols_first_vert(bm_faces):
return [tuple(bm_face.verts[0].co) for bm_face in bm_faces]
def pols_last_vert(bm_faces):
return [tuple(bm_face.verts[-1].co) for bm_face in bm_faces]
def pols_perimeter(vertices, faces):
'''
returns pols perimeter.
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
bm = bmesh_from_pydata(vertices, [], faces, normal_update=True)
vals = [bm_face.calc_perimeter() for bm_face in bm.faces]
bm.free()
return vals
def pols_tangent(vertices, faces, direction):
'''
returns pols tangents.
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
direction: String that can be any key of tangent_modes_dict
'''
if direction == 'Edge':
return np_process_polygons(vertices, faces,
func=np_tangent_longest_edge,
output_numpy=False)
if direction == 'Center - Origin':
return np_process_polygons(vertices, faces,
func=np_tangent_center_orig,
output_numpy=False)
else:
bm = bmesh_from_pydata(vertices, [], faces, normal_update=True)
vals = tangent_modes_dict[direction][1](bm.faces)
bm.free()
return vals
def pols_tangent_edge(bm_faces):
return [tuple(bm_face.calc_tangent_edge()) for bm_face in bm_faces]
def pols_tangent_edge_diagonal(bm_faces):
return [tuple(bm_face.calc_tangent_edge_diagonal()) for bm_face in bm_faces]
def pols_tangent_edge_pair(bm_faces):
return [tuple(bm_face.calc_tangent_edge_pair()) for bm_face in bm_faces]
def pols_tangent_center_origin(bm_faces):
return [tuple((Vector(bm_face.verts[0].co)-Vector(bm_face.calc_center_median())).normalized()) for bm_face in bm_faces]
def pols_tangent_vert_diagonal(bm_faces):
return [tuple(bm_face.calc_tangent_vert_diagonal()) for bm_face in bm_faces]
def pols_is_boundary(vertices, faces):
'''
bridge to mesh filter node Faces class
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
bm = bmesh_from_pydata(vertices, [], faces, normal_update=True)
interior, boundary, mask = Faces.process(bm, [], [])
bm.free()
return mask, interior, boundary
def pols_edges(faces):
'''
extracts the edges from polygons order [0, 1, 2] -> [[0, 1], [1, 2], [2, 0]]
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
vals = [[(c, cn) for c, cn in zip(face, face[1:] + [face[0]])] for face in faces]
return vals
def pols_inverted(faces):
'''
inverts the polygon order [0, 1, 2] -> [2, 1, 0]
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
'''
vals = [list(reversed(f)) for f in faces]
return vals
def pols_matrix(vertices, faces, origin, direc):
'''
Matrix aligned with faces
vertices: list as [vertex, vertex, ...], being each vertex [float, float, float].
faces: list as [polygon, polygon,..], being each polygon [int, int, ...].
orientation: contains origin and direction
origin: String that can be any key of pols_origin_modes_dict
direction: String that can be any key of tangent_modes_dict
outputs each polygon matrix [matrix, matrix, matrix]
'''
bm = bmesh_from_pydata(vertices, [], faces, normal_update=True)
normals = [Vector(face.normal) for face in bm.faces]
centers = pols_origin_modes_dict[origin][1](bm.faces)
tangents = tangent_modes_dict[direc][1](bm.faces)
vals = vectors_center_axis_to_matrix(centers, normals, tangents)
bm.free()
return vals
tangent_modes_dict = {
'Edge': (1, pols_tangent_edge, 'Face tangent based on longest edge'),
'Edge Diagonal': (2, pols_tangent_edge_diagonal, 'Face tangent based on the edge farthest from any vertex'),
'Edge Pair': (3, pols_tangent_edge_pair, 'Face tangent based on the two longest disconnected edges'),
'Vert Diagonal': (4, pols_tangent_vert_diagonal, 'Face tangent based on the two most distant vertices'),
'Center - Origin': (5, pols_tangent_center_origin, 'Face tangent based on the mean center and first corner'),
}
pols_origin_modes_dict = {
'Bounds Center': (30, pols_center_bounds, 'Center of bounding box of faces'),
'Median Center': (31, pols_center_median, 'Mean of vertices of each face'),
'Median Weighted Center': (32, pols_center_median_weighted, 'Mean of vertices of each face weighted by edges length'),
'First Vertex': (33, pols_first_vert, 'First Vertex of Face'),
'Last Vertex': (34, pols_last_vert, 'Last Vertex of Face'),
}
# Name: (index, input_sockets, func_options, output_options, function, output_sockets, output_sockets_names, description)
faces_modes_dict = {
'Geometry': (0, 'vp', '', 'u', pols_vertices, 'vs', 'Vertices, Faces', "Geometry of each face. (explode)"),
'Center': (10, 'vp', 'ca', '', pols_center, 'v', 'Center', 'Center faces'),
'Normal': (20, 'vp', 'a', '', pols_normals, 'v', 'Normal', 'Normal of faces'),
'Normal Absolute': (21, 'vp', 'a', '', pols_absolute_normals, 'v', 'Normal_Abs', 'Median Center + Normal'),
'Tangent': (30, 'vp', 't', '', pols_tangent, 'v', 'Tangent', 'Face tangent.'),
'Matrix': (40, 'vp', 'qt', 'u', pols_matrix, 'm', 'Matrix', 'Matrix of face. Z axis on normal. X to first corner'),
'Area': (50, 'vp', 's', '', areas_from_polygons, 's', 'Area', "Area of faces"),
'Perimeter': (51, 'vp', 'sa', '', pols_perimeters, 's', 'Perimeter', 'Perimeter of faces'),
'Sides Number': (52, 'p', 's', '', pols_sides, 's', 'Sides', "Number of sides of faces"),
'Adjacent Faces Num': (53, 'p', '', '', pols_adjacent_num, 's', 'Number', "Number of Faces that share a edge with face"),
'Neighbor Faces Num': (54, 'vp', '', '', pols_neighbor_num, 's', 'Number', "Number of Faces that share a vertex with face"),
'Sharpness': (55, 'vp', '', '', pols_shell_factor, 's', 'Sharpness', 'Average of curvature of mesh in faces vertices'),
'Inverse': (60, 'p', '', '', pols_inverted, 's', 'Faces', 'Reversed Polygons (Flipped)'),
'Edges': (61, 'p', '', 'u', pols_edges, 's', 'Edges', 'Face Edges'),
'Adjacent Faces': (62, 'p', '', 'u', pols_adjacent, 's', 'Faces', 'Faces that share a edge with face'),
'Neighbor Faces': (63, 'vp', '', 'u', pols_neighbor, 's', 'Faces', 'Faces that share a vertex with face'),
'Adjacent Faces Idx': (64, 'p', '', 'u', pols_adjacent_idx, 's', 'Faces Idx', 'Index of faces that share a edge with face'),
'Neighbor Faces Idx': (65, 'vp', '', 'u', pols_neighbor_idx, 's', 'Faces Idx', 'Index of faces that share a vertex with face'),
'Is Boundary': (70, 'vp', '', '', pols_is_boundary, 'sss', 'Mask, Boundary, Interior', 'Is the face boundary'),
}Functions
def areas_from_polygons(verts, polygons, sum_faces=False)-
returns pols area as [float, float,…] vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. sum_faces if True it will return the sum of the areas as [float]
Expand source code
def areas_from_polygons(verts, polygons, sum_faces=False): ''' returns pols area as [float, float,...] vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. sum_faces if True it will return the sum of the areas as [float] ''' areas = [] concat_area = areas.append for polygon in polygons: num = len(polygon) if num == 3: concat_area(area(verts[polygon[0]], verts[polygon[1]], verts[polygon[2]])) elif num == 4: area_1 = area(verts[polygon[0]], verts[polygon[1]], verts[polygon[2]]) area_2 = area(verts[polygon[0]], verts[polygon[2]], verts[polygon[3]]) concat_area(area_1 + area_2) elif num > 4: ngon_area = 0.0 subcoords = [Vector(verts[idx]) for idx in polygon] for tri in tessellate([subcoords]): ngon_area += area(*[verts[polygon[i]] for i in tri]) concat_area(ngon_area) else: concat_area(0) if sum_faces: areas = [sum(areas)] return areas def edges_lengths(v_pols)-
Expand source code
def edges_lengths(v_pols): return np.linalg.norm(v_pols-np.roll(v_pols, -1, axis=1), axis=2) def np_center_bbox(v_pols)-
Expand source code
def np_center_bbox(v_pols): return (np.amin(v_pols, axis=1) + np.amax(v_pols, axis=1))/2 def np_center_median(v_pols)-
Expand source code
def np_center_median(v_pols): return np.sum(v_pols, axis=1) / v_pols.shape[1] def np_center_weighted(v_pols)-
Expand source code
def np_center_weighted(v_pols): v_factor = vertex_weight_factor(v_pols) return np.sum(v_pols * v_factor[:, :, np.newaxis], axis=1) def np_faces_absolute_normals(v_pols)-
Expand source code
def np_faces_absolute_normals(v_pols): return np_center_median(v_pols) + np_faces_normals(v_pols) def np_faces_normals(v_pols)-
Expand source code
def np_faces_normals(v_pols): pol_sides = v_pols.shape[1] if pol_sides > 3: f_normals = np.zeros((len(v_pols), 3), dtype=np.float64) for i in range(pol_sides - 2): f_normals += np.cross(v_pols[::, (1+i)%pol_sides] - v_pols[::, 0], v_pols[::, (2+i)%pol_sides] - v_pols[::, 0]) else: f_normals = np.cross(v_pols[::, 1] - v_pols[::, 0], v_pols[::, 2] - v_pols[::, 0]) np_normalize_vectors(f_normals) return f_normals def np_faces_perimeters(v_pols)-
Expand source code
def np_faces_perimeters(v_pols): return np.sum(edges_lengths(v_pols), axis=1) def np_process_polygons(verts, faces, func=None, dims=3, output_numpy=False)-
Expand source code
def np_process_polygons(verts, faces, func=None, dims=3, output_numpy=False): if not func: return if not (has_element(verts) and has_element(faces)): return if isinstance(verts, np.ndarray): np_verts = verts else: np_verts = np.array(verts) if isinstance(faces, np.ndarray): np_faces = faces else: np_faces = np.array(faces) if np_faces.dtype == object: np_len = np.vectorize(len) lens = np_len(np_faces) pol_types = np.unique(lens) if dims == 1: vals = np.zeros(np_faces.shape[0], dtype=float) else: vals = np.zeros((np_faces.shape[0], dims), dtype=float) for p in pol_types: mask = lens == p np_faces_g = np.array(np_faces[mask].tolist()) v_pols = np_verts[np_faces_g] if dims == 1: vals[mask] = func(v_pols) else: vals[mask, :] = func(v_pols) else: v_pols = np_verts[np_faces] #num pols, num sides vals = func(v_pols) if output_numpy: return vals return vals.tolist() def np_tangent_center_orig(v_pols)-
Expand source code
def np_tangent_center_orig(v_pols): return np_normalize_vectors(np_center_median(v_pols) - v_pols[:, 0, :]) def np_tangent_longest_edge(v_pols)-
Expand source code
def np_tangent_longest_edge(v_pols): edges_dir = v_pols-np.roll(v_pols, 1, axis=1) ed_length = np.linalg.norm(edges_dir, axis=2) ed_idx = np.argmax(ed_length, axis=1) return np_normalize_vectors(edges_dir[np.arange(len(v_pols)), ed_idx, :]) def pols_absolute_normals(vertices, faces, output_numpy)-
Returns Faces center + faces normals as [vector, vector,…] vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_absolute_normals(vertices, faces, output_numpy): ''' Returns Faces center + faces normals as [vector, vector,...] vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' return np_process_polygons(vertices, faces, func=np_faces_absolute_normals, output_numpy=output_numpy) def pols_adjacent(pols)-
returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_adjacent(pols): """ returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. """ return [[pols[i] for i in p_indexes] for p_indexes in pols_adjacent_idx(pols)] def pols_adjacent_idx(pols)-
returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_adjacent_idx(pols): """ returns the polygons that share a vertex with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. """ edges = {tuple(sorted(e)): None for pol_edges in pols_edges(pols) for e in pol_edges} edges = {e: adj_indexes for e, adj_indexes in zip(edges.keys(), adjacent_faces_idx(edges.keys(), pols))} adj_pols = [] for ei, pol_edges in enumerate(pols_edges(pols)): adj_pols.append(list({fi for e in pol_edges for fi in edges[tuple(sorted(e))]} - {ei})) return adj_pols def pols_adjacent_num(pols)-
returns the number polygons that share a vertex with each polygon [int, int,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_adjacent_num(pols): """ returns the number polygons that share a vertex with each polygon [int, int,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. """ return [len(p) for p in pols_adjacent_idx(pols)] def pols_center(vertices, faces, origin, output_numpy)-
Cemter of faces vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. edges: list as [edge, edge,..], being each edge [int, int]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. origin: String that can be any key of pols_origin_modes_dict returns vals as [float, float,…]
Expand source code
def pols_center(vertices, faces, origin, output_numpy): ''' Cemter of faces vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. edges: list as [edge, edge,..], being each edge [int, int]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. origin: String that can be any key of pols_origin_modes_dict returns vals as [float, float,...] ''' if origin == 'Median Center': centers_func = np_center_median elif origin == 'Bounds Center': centers_func = np_center_bbox else: centers_func = np_center_weighted vals = np_process_polygons(vertices, faces, func=centers_func, output_numpy=output_numpy) return vals def pols_center_bounds(bm_faces)-
Expand source code
def pols_center_bounds(bm_faces): return [tuple(bm_face.calc_center_bounds()) for bm_face in bm_faces] def pols_center_median(bm_faces)-
Expand source code
def pols_center_median(bm_faces): return [tuple(bm_face.calc_center_median()) for bm_face in bm_faces] def pols_center_median_weighted(bm_faces)-
Expand source code
def pols_center_median_weighted(bm_faces): return [tuple(bm_face.calc_center_median_weighted()) for bm_face in bm_faces] def pols_edges(faces)-
extracts the edges from polygons order [0, 1, 2] -> [[0, 1], [1, 2], [2, 0]] faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_edges(faces): ''' extracts the edges from polygons order [0, 1, 2] -> [[0, 1], [1, 2], [2, 0]] faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' vals = [[(c, cn) for c, cn in zip(face, face[1:] + [face[0]])] for face in faces] return vals def pols_first_vert(bm_faces)-
Expand source code
def pols_first_vert(bm_faces): return [tuple(bm_face.verts[0].co) for bm_face in bm_faces] def pols_inverted(faces)-
inverts the polygon order [0, 1, 2] -> [2, 1, 0] faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_inverted(faces): ''' inverts the polygon order [0, 1, 2] -> [2, 1, 0] faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' vals = [list(reversed(f)) for f in faces] return vals def pols_is_boundary(vertices, faces)-
bridge to mesh filter node Faces class vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_is_boundary(vertices, faces): ''' bridge to mesh filter node Faces class vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' bm = bmesh_from_pydata(vertices, [], faces, normal_update=True) interior, boundary, mask = Faces.process(bm, [], []) bm.free() return mask, interior, boundary def pols_last_vert(bm_faces)-
Expand source code
def pols_last_vert(bm_faces): return [tuple(bm_face.verts[-1].co) for bm_face in bm_faces] def pols_matrix(vertices, faces, origin, direc)-
Matrix aligned with faces vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. orientation: contains origin and direction origin: String that can be any key of pols_origin_modes_dict direction: String that can be any key of tangent_modes_dict outputs each polygon matrix [matrix, matrix, matrix]
Expand source code
def pols_matrix(vertices, faces, origin, direc): ''' Matrix aligned with faces vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. orientation: contains origin and direction origin: String that can be any key of pols_origin_modes_dict direction: String that can be any key of tangent_modes_dict outputs each polygon matrix [matrix, matrix, matrix] ''' bm = bmesh_from_pydata(vertices, [], faces, normal_update=True) normals = [Vector(face.normal) for face in bm.faces] centers = pols_origin_modes_dict[origin][1](bm.faces) tangents = tangent_modes_dict[direc][1](bm.faces) vals = vectors_center_axis_to_matrix(centers, normals, tangents) bm.free() return vals def pols_neighbor(verts, pols)-
returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_neighbor(verts, pols): ''' returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' v_adj = adjacent_edg_pol(verts, pols) vals = [] for pol in pols: pol_adj = [] for v_id in pol: for related_pol in v_adj[v_id]: if not related_pol in pol_adj: pol_adj.append(related_pol) pol_adj.remove(pol) vals.append(pol_adj) return vals def pols_neighbor_idx(verts, pols)-
returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_neighbor_idx(verts, pols): ''' returns the polygons that share one edges with each polygon [[pol, pol,..], [pol,..]] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' v_adj = adjacent_edg_pol_idx(verts, pols) vals = [] for idx, pol in enumerate(pols): pol_adj = [] for v_id in pol: for related_pol in v_adj[v_id]: if not related_pol in pol_adj: pol_adj.append(related_pol) pol_adj.remove(idx) vals.append(pol_adj) return vals def pols_neighbor_num(verts, pols)-
returns the number of polygons that share one edges with each polygon [int, int,…] pols: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_neighbor_num(verts, pols): ''' returns the number of polygons that share one edges with each polygon [int, int,...] pols: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' return [len(p) for p in pols_neighbor(verts, pols)] def pols_normals(vertices, faces, output_numpy)-
Returns Faces normals as [vector, vector,…] vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_normals(vertices, faces, output_numpy): ''' Returns Faces normals as [vector, vector,...] vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' vals = np_process_polygons(vertices, faces, func=np_faces_normals, output_numpy=output_numpy) return vals def pols_perimeter(vertices, faces)-
returns pols perimeter. vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …].
Expand source code
def pols_perimeter(vertices, faces): ''' returns pols perimeter. vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. ''' bm = bmesh_from_pydata(vertices, [], faces, normal_update=True) vals = [bm_face.calc_perimeter() for bm_face in bm.faces] bm.free() return vals def pols_perimeters(verts, polygons, sum_perimeters=False, output_numpy=False)-
returns pols perimeter as [float, float,…] vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. sum_perimeters if True it will return the sum of the perimenters as [float]
Expand source code
def pols_perimeters(verts, polygons, sum_perimeters=False, output_numpy=False): ''' returns pols perimeter as [float, float,...] vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. sum_perimeters if True it will return the sum of the perimenters as [float] ''' vals = np_process_polygons(verts, polygons, func=np_faces_perimeters, dims=1, output_numpy=True) if sum_perimeters: if output_numpy: return [np.sum(vals)] return [np.sum(vals).tolist()] if output_numpy: return vals return vals.tolist() def pols_shell_factor(vertices, faces)-
Average of vertex shell_factor vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. edges: list as [edge, edge,..], being each edge [int, int]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. returns vals as [float, float,…]
Expand source code
def pols_shell_factor(vertices, faces): ''' Average of vertex shell_factor vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. edges: list as [edge, edge,..], being each edge [int, int]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. returns vals as [float, float,...] ''' v_shell = vertex_shell_factor(vertices, [], faces) vals = [] for face in faces: p_shell = 0 for v_id in face: p_shell += v_shell[v_id] vals.append(p_shell/len(face)) return vals def pols_sides(faces, sum_sides=False)-
returns the number of sides of each polygon as [int, int, … ] faces: list as [polygon, polygon,..], being each polygon [int, int, …]. sum_sides: if True will return the sum of sides of the whole object returning [int]
Expand source code
def pols_sides(faces, sum_sides=False): ''' returns the number of sides of each polygon as [int, int, ... ] faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. sum_sides: if True will return the sum of sides of the whole object returning [int] ''' vals = [len(p) for p in faces] if sum_sides: vals = [sum(vals)] return vals def pols_tangent(vertices, faces, direction)-
returns pols tangents. vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. direction: String that can be any key of tangent_modes_dict
Expand source code
def pols_tangent(vertices, faces, direction): ''' returns pols tangents. vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. direction: String that can be any key of tangent_modes_dict ''' if direction == 'Edge': return np_process_polygons(vertices, faces, func=np_tangent_longest_edge, output_numpy=False) if direction == 'Center - Origin': return np_process_polygons(vertices, faces, func=np_tangent_center_orig, output_numpy=False) else: bm = bmesh_from_pydata(vertices, [], faces, normal_update=True) vals = tangent_modes_dict[direction][1](bm.faces) bm.free() return vals def pols_tangent_center_origin(bm_faces)-
Expand source code
def pols_tangent_center_origin(bm_faces): return [tuple((Vector(bm_face.verts[0].co)-Vector(bm_face.calc_center_median())).normalized()) for bm_face in bm_faces] def pols_tangent_edge(bm_faces)-
Expand source code
def pols_tangent_edge(bm_faces): return [tuple(bm_face.calc_tangent_edge()) for bm_face in bm_faces] def pols_tangent_edge_diagonal(bm_faces)-
Expand source code
def pols_tangent_edge_diagonal(bm_faces): return [tuple(bm_face.calc_tangent_edge_diagonal()) for bm_face in bm_faces] def pols_tangent_edge_pair(bm_faces)-
Expand source code
def pols_tangent_edge_pair(bm_faces): return [tuple(bm_face.calc_tangent_edge_pair()) for bm_face in bm_faces] def pols_tangent_vert_diagonal(bm_faces)-
Expand source code
def pols_tangent_vert_diagonal(bm_faces): return [tuple(bm_face.calc_tangent_vert_diagonal()) for bm_face in bm_faces] def pols_to_edges(obj, unique_edges=False)-
Generate edges from pols obj: list as [[polygon, polygon,..], [polygon,…],…], being each polygon [int, int, …]. unique_edges: if False will return all polygon edges if True will remove the doubled edges
Expand source code
def pols_to_edges(obj, unique_edges=False): ''' Generate edges from pols obj: list as [[polygon, polygon,..], [polygon,...],...], being each polygon [int, int, ...]. unique_edges: if False will return all polygon edges if True will remove the doubled edges ''' out = [] for faces in obj: out_edges = [] seen = set() for face in faces: for edge in zip(face, list(face[1:]) + list([face[0]])): if unique_edges and tuple(sorted(edge)) in seen: continue if unique_edges: seen.add(tuple(sorted(edge))) out_edges.append(edge) out.append(out_edges) return out def pols_vertices(vertices, faces)-
Explodes geometry. vertices: list as [vertex, vertex, …], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, …]. returns [[vertex, vertex, vertex,..], [vertex, vertex, vertex,..], …] and [[polygon], [polygon],…]
Expand source code
def pols_vertices(vertices, faces): ''' Explodes geometry. vertices: list as [vertex, vertex, ...], being each vertex [float, float, float]. faces: list as [polygon, polygon,..], being each polygon [int, int, ...]. returns [[vertex, vertex, vertex,..], [vertex, vertex, vertex,..], ...] and [[polygon], [polygon],...] ''' verts = [[vertices[c] for c in p] for p in faces] pols = [[list(range(len(p)))] for p in faces] vals = [verts, pols] return vals def vertex_weight_factor(v_pols)-
Expand source code
def vertex_weight_factor(v_pols): ed_lengths = edges_lengths(v_pols) perimeters = np.sum(ed_lengths, axis=1) edges_prop = ed_lengths/perimeters[:, np.newaxis] return (edges_prop + np.roll(edges_prop, 1, axis=1))/2