wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-08-16 11:28:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

generalize the derivs for innerProducts

Browse Source
This commit is contained in:
rowanc1
2014-07-02 16:52:46 -07:00
parent e77f431370
commit 9d048f2884
1 changed files with 21 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Mesh/InnerProducts.py
+21 -25
View File
@@ -109,10 +109,9 @@ class InnerProducts(object):
return [V*proj(*locs[node][d-1]) for node in nodes] return [V*proj(*locs[node][d-1]) for node in nodes]
def getFaceInnerProductDeriv(self, tensorType, P=None, doFast=True): def getFaceInnerProductDeriv(self, tensorType, doFast=True):
""" """
:param TensorType tensorType: type of the tensor: TensorType(mesh, sigma) :param TensorType tensorType: type of the tensor: TensorType(mesh, sigma)
:param list P: list of projection matrices
:param bool doFast: do a faster implementation if available. :param bool doFast: do a faster implementation if available.
:rtype: function :rtype: function
:return: dMdmu(u), the derivative of the inner product matrix (u) :return: dMdmu(u), the derivative of the inner product matrix (u)
@@ -123,27 +122,22 @@ class InnerProducts(object):
:rtype: scipy.csr_matrix :rtype: scipy.csr_matrix
:return: dMdmu, the derivative of the inner product matrix for a certain u :return: dMdmu, the derivative of the inner product matrix for a certain u
""" """
assert isinstance(tensorType, TensorType), 'tensorType must be an instance of TensorType.' return self._getInnerProductDeriv(tensorType, 'F', doFast=doFast)
fast = None
if hasattr(self, '_fastInnerProductDeriv') and doFast:
fast = self._fastInnerProductDeriv('F', tensorType)
if fast is not None:
return fast
if P is None:
P = self._getInnerProductProjectionMatrices('F', tensorType=tensorType)
def innerProductDeriv(v):
return self._getInnerProductDeriv(tensorType, P, self.nF, v)
return innerProductDeriv
def getEdgeInnerProductDeriv(self, tensorType, P=None, doFast=True): def getEdgeInnerProductDeriv(self, tensorType, doFast=True):
""" """
:param TensorType tensorType: type of the tensor: TensorType(mesh, sigma) :param TensorType tensorType: type of the tensor: TensorType(mesh, sigma)
:param list P: list of projection matrices :param bool doFast: do a faster implementation if available.
:rtype: scipy.csr_matrix
:return: dMdm, the derivative of the inner product matrix (nE, nC*nA)
"""
return self._getInnerProductDeriv(tensorType, 'E', doFast=doFast)
def _getInnerProductDeriv(self, tensorType, projType, doFast=True):
"""
:param TensorType tensorType: type of the tensor: TensorType(mesh, sigma)
:param str projType: 'F' for faces 'E' for edges
:param bool doFast: do a faster implementation if available. :param bool doFast: do a faster implementation if available.
:rtype: scipy.csr_matrix :rtype: scipy.csr_matrix
:return: dMdm, the derivative of the inner product matrix (nE, nC*nA) :return: dMdm, the derivative of the inner product matrix (nE, nC*nA)
@@ -152,26 +146,28 @@ class InnerProducts(object):
fast = None fast = None
if hasattr(self, '_fastInnerProductDeriv') and doFast: if hasattr(self, '_fastInnerProductDeriv') and doFast:
fast = self._fastInnerProductDeriv('E', tensorType) fast = self._fastInnerProductDeriv(projType, tensorType)
if fast is not None: if fast is not None:
return fast return fast
if P is None: P = self._getInnerProductProjectionMatrices(projType, tensorType=tensorType)
P = self._getInnerProductProjectionMatrices('E', tensorType=tensorType)
def innerProductDeriv(v): def innerProductDeriv(v):
return self._getInnerProductDeriv(tensorType, P, self.nE, v) return self._getInnerProductDerivFunction(tensorType, P, projType, v)
return innerProductDeriv return innerProductDeriv
def _getInnerProductDeriv(self, tensorType, P, n, v): def _getInnerProductDerivFunction(self, tensorType, P, projType, v):
""" """
:param numpy.array prop: material property (tensor properties are possible) at each cell center (nC, (1, 3, or 6)) :param numpy.array prop: material property (tensor properties are possible) at each cell center (nC, (1, 3, or 6))
:param numpy.array v: vector to multiply (required in the general implementation) :param numpy.array v: vector to multiply (required in the general implementation)
:param list P: list of projection matrices :param list P: list of projection matrices
:param int n: nF or nE :param str projType: 'F' for faces 'E' for edges
:rtype: scipy.csr_matrix :rtype: scipy.csr_matrix
:return: dMdm, the derivative of the inner product matrix (n, nC*nA) :return: dMdm, the derivative of the inner product matrix (n, nC*nA)
""" """
assert projType in ['F', 'E'], "projType must be 'F' for faces or 'E' for edges"
n = getattr(self,'n'+projType)
if tensorType == -1: if tensorType == -1:
return None return None