first stab at how to structure sources and fields for primary secondary formulations

simpegEM/Utils/EMUtils.py

@@ -1,10 +1,46 @@
 import numpy as np
 from scipy.constants import mu_0, epsilon_0
 
+# useful params
 def omega(freq):
     """Angular frequency, omega"""
     return 2.*np.pi*freq
 
 def k(freq, sigma, mu=mu_0, eps=epsilon_0):
-	w = omega(freq)
-	return np.sqrt(mu * eps * w**2 - 1j * w* mu * sigma)
+    w = omega(freq)
+    return np.sqrt(mu * eps * w**2 - 1j * w* mu * sigma)
+
+# Constitutive relations
+def e_from_j(prob,j):
+    eqLocs = prob._eqLocs
+    if eqLocs is 'FE':
+        MSigmaI = prob.MeSigmaI
+    elif eqLocs is 'EF':
+        MSigmaI = prob.MfSigmai
+    return MSigmaI*j
+
+def j_from_e(prob,e):
+    eqLocs = prob._eqLocs
+    if eqLocs is 'FE':
+        MSigma = prob.MeSigma
+    elif eqLocs is 'EF':
+        MSigma = prob.MfSigma
+    return MSigma*e
+
+def b_from_h(prob,h):
+    eqLocs = prob._eqLocs
+    if eqLocs is 'FE':
+        MMu = prob.MfMu
+    elif eqLocs is 'EF':
+        MMu = prob.MeMu
+    return MMu*h
+
+def h_from_b(prob,b):
+    eqLocs = prob._eqLocs
+    if eqLocs is 'FE':
+        MMuI = prob.MfMui
+    elif eqLocs is 'EF':
+        MMuI = prob.MeMuI
+    return MMuI*b 
+
+