wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-01 08:19:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Clean up init code, put inside run, remove mpl from TL import.

Browse Source
This commit is contained in:
Rowan Cockett
2016-01-20 12:52:51 -07:00
parent 98f209d1f0
commit 91e37c5fa7
1 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Examples/EM_FDEM_Analytic_MagDipoleWholespace.py
+14 -18
View File
@@ -1,33 +1,29 @@
from SimPEG import *
import SimPEG.EM as EM
import matplotlib.pyplot as plt
from matplotlib.colors import LogNorm
x = np.arange(-100.5,100.5,step = 1.) #(avoid putting measurement points where source is located)
y = np.r_[0]
z = x
sig = 1.
freq = 1.
XYZ = Utils.ndgrid(x,y,z)
def run(XYZ=XYZ,sig=sig,freq=freq,orientation='Z', plotIt=True):
def run(XYZ=None, sig=1.0, freq=1.0, orientation='Z', plotIt=True):
"""
EM: Magnetic Dipole in a Whole-Space
====================================
Here we plot the magnetic flux density from a harmonic dipole in a wholespace.
Here we plot the magnetic flux density from a harmonic dipole in a wholespace.
"""
Bx, By, Bz = EM.Analytics.FDEM.MagneticDipoleWholeSpace(XYZ, np.r_[0.,0.,0.], sig, freq,orientation=orientation)
if XYZ is None:
x = np.arange(-100.5,100.5,step = 1.) #(avoid putting measurement points where source is located)
y = np.r_[0]
z = x
XYZ = Utils.ndgrid(x,y,z)
Bx, By, Bz = EM.Analytics.FDEM.MagneticDipoleWholeSpace(XYZ, np.r_[0.,0.,0.], sig, freq, orientation=orientation)
absB = np.sqrt(Bx*Bx.conj()+By*By.conj()+Bz*Bz.conj()).real
if plotIt:
if plotIt:
import matplotlib.pyplot as plt
from matplotlib.colors import LogNorm
fig, ax = plt.subplots(1,1,figsize=(6,5))
bxplt = Bx.reshape(x.size,z.size)
bzplt = Bz.reshape(x.size,z.size)
@@ -44,4 +40,4 @@ def run(XYZ=XYZ,sig=sig,freq=freq,orientation='Z', plotIt=True):
return fig, ax
if __name__ == '__main__':
run()
run()