wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-06-29 03:04:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

update points to nodes in meshUtils

Browse Source
This commit is contained in:
rowanc1
2014-04-15 08:56:07 -07:00
parent dbd1334e0b
commit 47a148361b
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Utils/meshutils.py
+7 -4
View File
@@ -65,19 +65,22 @@ def points2nodes(mesh, pts):
pts = np.atleast_2d(pts)
assert mesh._meshType == 'TENSOR'
assert mesh._meshType in ['TENSOR', 'CYL']
assert pts.shape[1] == mesh.dim
nodeInds = np.empty(pts.shape[0], dtype=int)
for i, pt in enumerate(pts):
nodeInds[i] = ((np.tile(pt, (mesh.nN,1)) - mesh.gridN)**2).sum(axis=1).argmin()
nodeInds[i] = ((np.tile(pt, (mesh.gridN.shape[0],1)) - mesh.gridN)**2).sum(axis=1).argmin()
return nodeInds
if __name__ == '__main__':
from SimPEG import mesh
from SimPEG import Mesh
import matplotlib.pyplot as plt
M = mesh.TensorMesh(meshTensors(((10,10),(40,10),(10,10)), ((10,10),(20,10),(0,0))))
M = Mesh.TensorMesh(meshTensors(((10,10),(40,10),(10,10)), ((10,10),(20,10),(0,0))))
M.plotGrid()
plt.gca().axis('tight')
plt.show()