Как добавить симплексы к объекту триангуляции Делоне - PullRequest
Новая
       26

Как добавить симплексы к объекту триангуляции Делоне

0 голосов
/ 12 сентября 2018

У меня уже есть прямоугольник, триангулированный объектом scipy.spatial.Delaunay(). Мне удается растянуть и изогнуть его так, чтобы оно выглядело как кольцеобразный разрез по линии. Вот код для создания чего-то с такой же топологией: 

from scipy.spatial import Delaunay

NR = 22
NTheta = 36

Rin = 1
Rout = 3
alphaFactor = 33/64
alpha = np.pi/alphaFactor # opening angle of wedge

u=np.linspace(pi/2, pi/2 + alpha, NTheta)
v=np.linspace(Rin, Rout, NR)
u,v=np.meshgrid(u,v)
u=u.flatten()
v=v.flatten()

#evaluate the parameterization at the flattened u and v
x=v*np.cos(u)
y=v*np.sin(u)

#define 2D points, as input data for the Delaunay triangulation of U
points2D=np.vstack([u,v]).T
xy0 = np.vstack([x,y]).T
triLattice = Delaunay(points2D) #triangulate the rectangle U
triSimplices = triLattice.simplices
plt.figure()
plt.triplot(x, y, triSimplices, linewidth=0.5)

Cut annulus

Исходя из этой топологии, я теперь хочу объединить два открытых ребра и сделать замкнутое кольцо (то есть изменить топологию). Как вручную добавить новые треугольники в существующую триангуляцию?

1 Ответ

0 голосов
/ 12 сентября 2018

Решение состоит в том, чтобы объединить точки вокруг разрыва.Вот способ сделать это, отслеживая индексы соответствующих точек: 

import matplotlib.pylab as plt
from scipy.spatial import Delaunay
import numpy as np

NR = 4
NTheta = 16

Rin = 1
Rout = 3
alphaFactor = 33/64  # -- set to .5 to close the gap
alpha = np.pi/alphaFactor  # opening angle of wedge

u = np.linspace(np.pi/2, np.pi/2 + alpha, NTheta)
v = np.linspace(Rin, Rout, NR)
u_grid, v_grid = np.meshgrid(u, v)
u = u_grid.flatten()
v = v_grid.flatten()

# Get the indexes of the points on the first and last columns:
idx_grid_first = (np.arange(u_grid.shape[0]),
                  np.zeros(u_grid.shape[0], dtype=int))

idx_grid_last = (np.arange(u_grid.shape[0]),
                 (u_grid.shape[1]-1)*np.ones(u_grid.shape[0], dtype=int))

# Convert these 2D indexes to 1D indexes, on the flatten array:
idx_flat_first = np.ravel_multi_index(idx_grid_first, u_grid.shape)
idx_flat_last = np.ravel_multi_index(idx_grid_last, u_grid.shape)

# Evaluate the parameterization at the flattened u and v
x = v * np.cos(u)
y = v * np.sin(u)

# Define 2D points, as input data for the Delaunay triangulation of U
points2D = np.vstack([u, v]).T
triLattice = Delaunay(points2D) # triangulate the rectangle U
triSimplices = triLattice.simplices

# Replace the 'last' index by the corresponding 'first':
triSimplices_merged = triSimplices.copy()
for i_first, i_last in zip(idx_flat_first, idx_flat_last):
    triSimplices_merged[triSimplices == i_last] = i_first

# Graph
plt.figure(figsize=(7, 7))
plt.triplot(x, y, triSimplices, linewidth=0.5)
plt.triplot(x, y, triSimplices_merged, linewidth=0.5, color='k')
plt.axis('equal');

plt.plot(x[idx_flat_first], y[idx_flat_first], 'or', label='first')
plt.plot(x[idx_flat_last], y[idx_flat_last], 'ob', label='last')
plt.legend();

, что дает:

merged mesh

Возможно, вам придется настроить определение alphaFactor так, чтобы зазор имел правильный размер.

...