mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-29 17:21:30 +08:00
rowanc1
96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
import numpy as np
|
|
from matutils import mkvc
|
|
import warnings
|
|
|
|
def _checkAccuracy(A, b, X, accuracyTol):
|
|
nrm = np.linalg.norm(mkvc(A*X - b), np.inf)
|
|
nrm_b = np.linalg.norm(mkvc(b), np.inf)
|
|
if nrm_b > 0:
|
|
nrm /= nrm_b
|
|
if nrm > accuracyTol:
|
|
msg = '### SolverWarning ###: Accuracy on solve is above tolerance: %e > %e' % (nrm, accuracyTol)
|
|
print msg
|
|
warnings.warn(msg, RuntimeWarning)
|
|
|
|
|
|
def DSolverWrap(fun, factorize=True, checkAccuracy=True, accuracyTol=1e-6):
|
|
|
|
def __init__(self, A, **kwargs):
|
|
self.A = A.tocsc()
|
|
self.kwargs = kwargs
|
|
if factorize:
|
|
self.solver = fun(self.A, **kwargs)
|
|
|
|
def solve(self, b):
|
|
if len(b.shape) == 1 or b.shape[1] == 1:
|
|
b = b.flatten()
|
|
# Just one RHS
|
|
if factorize:
|
|
X = self.solver.solve(b, **self.kwargs)
|
|
else:
|
|
X = fun(self.A, b, **self.kwargs)
|
|
else: # Multiple RHSs
|
|
X = np.empty_like(b)
|
|
for i in range(b.shape[1]):
|
|
if factorize:
|
|
X[:,i] = self.solver.solve(b[:,i])
|
|
else:
|
|
X[:,i] = fun(self.A, b[:,i], **self.kwargs)
|
|
|
|
if checkAccuracy:
|
|
_checkAccuracy(self.A, b, X, accuracyTol)
|
|
return X
|
|
|
|
def clean(self):
|
|
if hasattr(self.solver, 'clean'):
|
|
return self.solver.clean()
|
|
|
|
|
|
def __mul__(self, val):
|
|
if type(val) is np.ndarray:
|
|
return self.solve(val)
|
|
raise TypeError('Can only multiply by a numpy array.')
|
|
|
|
return type(fun.__name__, (object,), {"__init__": __init__, "solve": solve, "clean": clean, "__mul__": __mul__})
|
|
|
|
|
|
|
|
def ISolverWrap(fun, checkAccuracy=True, accuracyTol=1e-5):
|
|
|
|
def __init__(self, A, **kwargs):
|
|
self.A = A
|
|
self.kwargs = kwargs
|
|
|
|
def solve(self, b):
|
|
if len(b.shape) == 1 or b.shape[1] == 1:
|
|
b = b.flatten()
|
|
# Just one RHS
|
|
out = fun(self.A, b, **self.kwargs)
|
|
if type(out) is tuple and len(out) == 2:
|
|
# We are dealing with scipy output with an info!
|
|
X = out[0]
|
|
self.info = out[1]
|
|
else:
|
|
X = out
|
|
else: # Multiple RHSs
|
|
X = np.empty_like(b)
|
|
for i in range(b.shape[1]):
|
|
out = fun(self.A, b[:,i], **self.kwargs)
|
|
if type(out) is tuple and len(out) == 2:
|
|
# We are dealing with scipy output with an info!
|
|
X[:,i] = out[0]
|
|
self.info = out[1]
|
|
else:
|
|
X[:,i] = out
|
|
|
|
if checkAccuracy:
|
|
_checkAccuracy(self.A, b, X, accuracyTol)
|
|
return X
|
|
|
|
def __mul__(self, val):
|
|
if type(val) is np.ndarray:
|
|
return self.solve(val)
|
|
raise TypeError('Can only multiply by a numpy array.')
|
|
|
|
return type(fun.__name__, (object,), {"__init__": __init__, "solve": solve, "__mul__": __mul__})
|