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

Added method to return index of nearest face or edge to a given point.

Browse Source
This commit is contained in:
Dave Marchant
2013-10-21 12:48:56 -07:00
parent 57271e4ecc
commit 674de34212
1 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/mesh/Cyl1DMesh.py
+27
View File
@@ -421,5 +421,32 @@ class Cyl1DMesh(object):
raise ValueError('Invalid locType')
return Q.tocsr()
def getNearest(self, loc, locType):
""" Returns the index of the closest face or edge to a given location
:param numpy.ndarray loc: Test point
:param str locType: Type of location desired (see below)
:rtype: int
:return: ind:
locType can be::
'fz' -> location of nearest z-face
'fr' -> location of nearest r-face
'et' -> location of nearest edge
"""
if locType=='et':
dr = self.gridN[:,0] - loc[0]
dz = self.gridN[:,1] - loc[1]
elif locType=='fz':
dr = self.gridFz[:,0] - loc[0]
dz = self.gridFz[:,1] - loc[1]
elif locType=='fr':
dr = self.gridFr[:,0] - loc[0]
dz = self.gridFr[:,1] - loc[1]
else:
raise ValueError('Invalid locType')
R = np.sqrt(dr**2 + dz**2)
ind = np.argmin(R)
return ind