mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-29 00:39:55 +08:00
Rowan Cockett
101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
from scipy.sparse import linalg
|
|
from scipy import sparse
|
|
from sputils import *
|
|
from utils import *
|
|
from numpy import *
|
|
|
|
#----- Cell Centers from Nodal locations -----
|
|
def getCellCenterFromNodal(X,Y,Z):
|
|
|
|
XC = 1.0/8.0 * (X[0:-1,0:-1,0:-1] + X[1:,0:-1,0:-1] + X[0:-1,1:,0:-1] + X[1:,1:,0:-1] +
|
|
X[0:-1,0:-1,1:] + X[1:,0:-1,1:] + X[0:-1,1:,1:] + X[1:,1:,1:])
|
|
|
|
YC = 1.0/8.0 * (Y[0:-1,0:-1,0:-1] + Y[1:,0:-1,0:-1] + Y[0:-1,1:,0:-1] + Y[1:,1:,0:-1] +
|
|
Y[0:-1,0:-1,1:] + Y[1:,0:-1,1:] + Y[0:-1,1:,1:] + Y[1:,1:,1:])
|
|
|
|
ZC = 1.0/8.0 * (Z[0:-1,0:-1,0:-1] + Z[1:,0:-1,0:-1] + Z[0:-1,1:,0:-1] + Z[1:,1:,0:-1] +
|
|
Z[0:-1,0:-1,1:] + Z[1:,0:-1,1:] + Z[0:-1,1:,1:] + Z[1:,1:,1:])
|
|
|
|
return (XC,YC,ZC)
|
|
|
|
#----- Edges from Nodal locations -----
|
|
|
|
def getEdgesFromNodal(X,Y,Z):
|
|
#
|
|
# node(i,j,k+1) ------ edge2(i,j,k+1) ----- node(i,j+1,k+1)
|
|
# / /
|
|
# / / |
|
|
# edge3(i,j,k) face1(i,j,k) edge3(i,j+1,k)
|
|
# / / |
|
|
# / / |
|
|
# node(i,j,k) ------ edge2(i,j,k) ----- node(i,j+1,k)
|
|
# | | |
|
|
# | | node(i+1,j+1,k+1)
|
|
# | | /
|
|
# edge1(i,j,k) face3(i,j,k) edge1(i,j+1.k)
|
|
# | | /
|
|
# | | /
|
|
# | |/
|
|
# node(i+1,j,k) ------ edge2(i+1,j,k) ----- node(i+1,j+1,k)
|
|
|
|
XE1 = (X[1:,:,:]+X[0:-1,:,:])/2.0
|
|
YE1 = (Y[1:,:,:]+Y[0:-1,:,:])/2.0
|
|
ZE1 = (Z[1:,:,:]+Z[0:-1,:,:])/2.0
|
|
|
|
XE2 = (X[:,1:,:]+X[:,0:-1,:])/2.0
|
|
YE2 = (Y[:,1:,:]+Y[:,0:-1,:])/2.0
|
|
ZE2 = (Z[:,1:,:]+Z[:,0:-1,:])/2.0
|
|
|
|
XE3 = (X[:,:,1:]+X[:,:,0:-1])/2.0
|
|
YE3 = (Y[:,:,1:]+Y[:,:,0:-1])/2.0
|
|
ZE3 = (Z[:,:,1:]+Z[:,:,0:-1])/2.0
|
|
|
|
return (XE1,YE1,ZE1,XE2,YE2,ZE2,XE3,YE3,ZE3)
|
|
|
|
#-- Get faces from nodal --
|
|
|
|
def getFacesFromNodal(X,Y,Z):
|
|
|
|
XF1 = 1.0/4.0*(X[:,0:-1,0:-1]+X[:,1:,0:-1]+X[:,0:-1,1:]+X[:,1:,1:])
|
|
YF1 = 1.0/4.0*(Y[:,0:-1,0:-1]+Y[:,1:,0:-1]+Y[:,0:-1,1:]+Y[:,1:,1:])
|
|
ZF1 = 1.0/4.0*(Z[:,0:-1,0:-1]+Z[:,1:,0:-1]+Z[:,0:-1,1:]+Z[:,1:,1:])
|
|
|
|
XF2 = 1.0/4.0*(X[0:-1,:,0:-1]+X[1:,:,0:-1]+X[0:-1,:,1:]+X[1:,:,1:])
|
|
YF2 = 1.0/4.0*(Y[0:-1,:,0:-1]+Y[1:,:,0:-1]+Y[0:-1,:,1:]+Y[1:,:,1:])
|
|
ZF2 = 1.0/4.0*(Z[0:-1,:,0:-1]+Z[1:,:,0:-1]+Z[0:-1,:,1:]+Z[1:,:,1:])
|
|
|
|
XF3 = 1.0/4.0*(X[0:-1,0:-1,:]+X[1:,0:-1,:]+X[0:-1,1:,:]+X[1:,1:,:])
|
|
YF3 = 1.0/4.0*(Y[0:-1,0:-1,:]+Y[1:,0:-1,:]+Y[0:-1,1:,:]+Y[1:,1:,:])
|
|
ZF3 = 1.0/4.0*(Z[0:-1,0:-1,:]+Z[1:,0:-1,:]+Z[0:-1,1:,:]+Z[1:,1:,:])
|
|
|
|
return (XF1,YF1,ZF1,XF2,YF2,ZF2,XF3,YF3,ZF3)
|
|
|
|
#-- Project Edge vector field
|
|
|
|
def projectEdgeVectorField(EV1,EV2,EV3,X,Y,Z):
|
|
|
|
t1x,t1y,t1z,t2x,t2y,t2z,t3x,t3y,t3z,nrm1,nrm2,nrm3 = getEdgeTangent(X,Y,Z)
|
|
|
|
|
|
E1 = EV1[:,0]*mkvc(t1x) + EV1[:,1]*mkvc(t1y) + EV1[:,2]*mkvc(t1z)
|
|
E2 = EV2[:,0]*mkvc(t2x) + EV2[:,1]*mkvc(t2y) + EV2[:,2]*mkvc(t2z)
|
|
E3 = EV3[:,0]*mkvc(t3x) + EV3[:,1]*mkvc(t3y) + EV3[:,2]*mkvc(t3z)
|
|
|
|
return hstack((hstack((mkvc(E1),mkvc(E2))),mkvc(E3)))
|
|
|
|
|
|
#-- Prolect Face vector field
|
|
|
|
def projectFaceVectorField(FV1,FV2,FV3,X,Y,Z):
|
|
|
|
n1x,n1y,n1z,n2x,n2y,n2z,n3x,n3y,n3z,ar1,ar2,ar3 = getFaceNormals(X,Y,Z)
|
|
|
|
F1 = FV1[:,0]*mkvc(n1x) + FV1[:,1]*mkvc(n1y) + FV1[:,2]*mkvc(n1z)
|
|
F2 = FV2[:,0]*mkvc(n2x) + FV2[:,1]*mkvc(n2y) + FV2[:,2]*mkvc(n2z)
|
|
F3 = FV3[:,0]*mkvc(n3x) + FV3[:,1]*mkvc(n3y) + FV3[:,2]*mkvc(n3z)
|
|
|
|
return hstack((hstack((mkvc(F1),mkvc(F2))),mkvc(F3)))
|
|
|
|
|
|
|
|
|