More tests on more meshes, IPDeriv returns a function.

This commit is contained in:
rowanc1
2014-06-17 09:47:03 -06:00
parent d2bde11a2a
commit 7d34f596ff
4 changed files with 192 additions and 55 deletions
+9 -3
View File
@@ -114,8 +114,14 @@ class InnerProducts(object):
: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: function
:return: dMdmu(u), the derivative of the inner product matrix (u)
Given u, dMdmu returns (nF, nC*nA)
:param np.ndarray u: vector that multiplies dMdmu
:rtype: scipy.csr_matrix
:return: dMdm, the derivative of the inner product matrix (nF, nC*nA)
:return: dMdmu, the derivative of the inner product matrix for a certain u
"""
assert isinstance(tensorType, TensorType), 'tensorType must be an instance of TensorType.'
fast = None
@@ -131,7 +137,7 @@ class InnerProducts(object):
def innerProductDeriv(v):
return self._getInnerProductDeriv(tensorType, P, self.nF, v)
return DerivOperator(innerProductDeriv)
return innerProductDeriv
def getEdgeInnerProductDeriv(self, tensorType, P=None, doFast=True):
@@ -155,7 +161,7 @@ class InnerProducts(object):
P = self._getInnerProductProjectionMatrices('E', tensorType=tensorType)
def innerProductDeriv(v):
return self._getInnerProductDeriv(tensorType, P, self.nE, v)
return DerivOperator(innerProductDeriv)
return innerProductDeriv
def _getInnerProductDeriv(self, tensorType, P, n, v):
"""
+3 -3
View File
@@ -319,21 +319,21 @@ class BaseTensorMesh(BaseRectangularMesh):
# return self.dim * Av.T * V * ones
def scalarInnerProductDeriv(v):
return Utils.sdiag(v) * self.dim * Av.T * V * ones
return Utils.DerivOperator(scalarInnerProductDeriv)
return scalarInnerProductDeriv
if tensorType == 1:
Av = getattr(self, 'ave'+projType+'2CC')
V = Utils.sdiag(self.vol)
def isotropicInnerProductDeriv(v):
return Utils.sdiag(v) * self.dim * Av.T * V
return Utils.DerivOperator(isotropicInnerProductDeriv)
return isotropicInnerProductDeriv
if tensorType == 2: # anisotropic
Av = getattr(self, 'ave'+projType+'2CCV')
V = sp.kron(sp.identity(self.dim), Utils.sdiag(self.vol))
def anisotropicInnerProductDeriv(v):
return Utils.sdiag(v) * Av.T * V
return Utils.DerivOperator(anisotropicInnerProductDeriv)
return anisotropicInnerProductDeriv