mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 14:49:19 +08:00
Rowan Cockett
37 lines
956 B
Python
37 lines
956 B
Python
from scipy.sparse import linalg
|
|
from utils import *
|
|
from sputils import *
|
|
|
|
|
|
def inv3X3BlockDiagonal(a11,a12,a13,a21,a22,a23,a31,a32,a33):
|
|
|
|
a11 = mkvc(a11)
|
|
a12 = mkvc(a12)
|
|
a13 = mkvc(a13)
|
|
a21 = mkvc(a21)
|
|
a22 = mkvc(a22)
|
|
a23 = mkvc(a23)
|
|
a31 = mkvc(a31)
|
|
a32 = mkvc(a32)
|
|
a33 = mkvc(a33)
|
|
|
|
detA = a31*a12*a23 - a31*a13*a22 - a21*a12*a33 + a21*a13*a32 + a11*a22*a33 - a11*a23*a32
|
|
|
|
b11 = (a22*a33 - a23*a32)/detA
|
|
b12 = -(a12*a33 - a13*a32)/detA
|
|
b13 = (a12*a23 - a13*a22)/detA
|
|
|
|
b21 = (a31*a23 - a21*a33)/detA
|
|
b22 = -(a31*a13 - a11*a33)/detA
|
|
b23 = (a21*a13 - a11*a23)/detA
|
|
|
|
b31 = -(a31*a22 - a21*a32)/detA
|
|
b32 = (a31*a12 - a11*a32)/detA
|
|
b33 = -(a21*a12 - a11*a22)/detA
|
|
|
|
B = appendBottom3(
|
|
appendRight3(sdiag(b11), sdiag(b12), sdiag(b13)),
|
|
appendRight3(sdiag(b21), sdiag(b22), sdiag(b23)),
|
|
appendRight3(sdiag(b31), sdiag(b32), sdiag(b33)))
|
|
|
|
return B |