Initial merge and minor refactor of simpegPF.

SimPEG/Examples/Mesh_Basic_Types.py

@@ -1,4 +1,5 @@
-from SimPEG import *
+from SimPEG import Mesh, Utils, np
+
 
 def run(plotIt=True):
     """
@@ -8,15 +9,15 @@ def run(plotIt=True):
         Here we show SimPEG used to create three different types of meshes.
 
     """
-    sz = [16,16]
+    sz = [16, 16]
     tM = Mesh.TensorMesh(sz)
     qM = Mesh.TreeMesh(sz)
     qM.refine(lambda cell: 4 if np.sqrt(((np.r_[cell.center]-0.5)**2).sum()) < 0.4 else 3)
-    rM = Mesh.CurvilinearMesh(Utils.meshutils.exampleLrmGrid(sz,'rotate'))
+    rM = Mesh.CurvilinearMesh(Utils.meshutils.exampleLrmGrid(sz, 'rotate'))
 
     if plotIt:
         import matplotlib.pyplot as plt
-        fig, axes = plt.subplots(1,3,figsize=(14,5))
+        fig, axes = plt.subplots(1, 3, figsize=(14, 5))
         opts = {}
         tM.plotGrid(ax=axes[0], **opts)
         axes[0].set_title('TensorMesh')

SimPEG/Examples/PF_Magnetics_Analytics.py

@@ -0,0 +1,65 @@
+from SimPEG import Mesh, np, PF
+
+
+def run(plotIt=True):
+    """
+        PF: Magnetics: Analytics
+        ========================
+
+        Comparing the magnetics field in Vancouver to Seoul
+
+    """
+
+    xr = np.linspace(-300, 300, 41)
+    yr = np.linspace(-300, 300, 41)
+    X, Y = np.meshgrid(xr, yr)
+    Z = np.ones((np.size(xr), np.size(yr)))*150
+
+    # Bz component in Korea
+    inckr = -8. + 3./60
+    deckr = 54. + 9./60
+    btotkr = 50898.6
+    Bokr = PF.MagAnalytics.IDTtoxyz(inckr, deckr, btotkr)
+
+    bx, by, bz = PF.MagAnalytics.MagSphereAnaFunA(
+        X, Y, Z, 100., 0., 0., 0., 0.01, Bokr, 'secondary'
+    )
+    Bzkr = np.reshape(bz, (np.size(xr), np.size(yr)), order='F')
+
+    # Bz component in Canada
+    incca = 16. + 49./60
+    decca = 70. + 19./60
+    btotca = 54692.1
+    Boca = PF.MagAnalytics.IDTtoxyz(incca, decca, btotca)
+
+    bx, by, bz = PF.MagAnalytics.MagSphereAnaFunA(
+        X, Y, Z, 100., 0., 0., 0., 0.01, Boca, 'secondary'
+    )
+    Bzca = np.reshape(bz, (np.size(xr), np.size(yr)), order='F')
+
+    if plotIt:
+        import matplotlib.pyplot as plt
+        from mpl_toolkits.axes_grid1 import make_axes_locatable
+        fig = plt.figure(figsize=(14, 5))
+
+        ax1 = plt.subplot(121)
+        dat1 = plt.imshow(Bzkr, extent=[min(xr), max(xr), min(yr), max(yr)])
+        divider = make_axes_locatable(ax1)
+        cax1 = divider.append_axes("right", size="5%", pad=0.05)
+        ax1.set_xlabel('East-West (m)')
+        ax1.set_ylabel('South-North (m)')
+        plt.colorbar(dat1, cax=cax1)
+        ax1.set_title('$B_z$ field at Seoul, South Korea')
+
+        ax2 = plt.subplot(122)
+        dat2 = plt.imshow(Bzca, extent=[min(xr), max(xr), min(yr), max(yr)])
+        divider = make_axes_locatable(ax2)
+        cax2 = divider.append_axes("right", size="5%", pad=0.05)
+        ax2.set_xlabel('East-West (m)')
+        ax2.set_ylabel('South-North (m)')
+        plt.colorbar(dat2, cax=cax2)
+        ax2.set_title('$B_z$ field at Vancouver, Canada')
+        plt.show()
+
+if __name__ == '__main__':
+    run()

SimPEG/Examples/__init__.py

@@ -20,9 +20,10 @@ import Mesh_QuadTree_HangingNodes
 import Mesh_Tensor_Creation
 import MT_1D_ForwardAndInversion
 import MT_3D_Foward
+import PF_Magnetics_Analytics
 import Utils_surface2ind_topo
 
-__examples__ = ["DC_Analytic_Dipole", "DC_Forward_PseudoSection", "EM_FDEM_1D_Inversion", "EM_FDEM_Analytic_MagDipoleWholespace", "EM_Schenkel_Morrison_Casing", "EM_TDEM_1D_Inversion", "FLOW_Richards_1D_Celia1990", "Inversion_IRLS", "Inversion_Linear", "Mesh_Basic_ForwardDC", "Mesh_Basic_PlotImage", "Mesh_Basic_Types", "Mesh_Operators_CahnHilliard", "Mesh_QuadTree_Creation", "Mesh_QuadTree_FaceDiv", "Mesh_QuadTree_HangingNodes", "Mesh_Tensor_Creation", "MT_1D_ForwardAndInversion", "MT_3D_Foward", "Utils_surface2ind_topo"]
+__examples__ = ["DC_Analytic_Dipole", "DC_Forward_PseudoSection", "EM_FDEM_1D_Inversion", "EM_FDEM_Analytic_MagDipoleWholespace", "EM_Schenkel_Morrison_Casing", "EM_TDEM_1D_Inversion", "FLOW_Richards_1D_Celia1990", "Inversion_IRLS", "Inversion_Linear", "Mesh_Basic_ForwardDC", "Mesh_Basic_PlotImage", "Mesh_Basic_Types", "Mesh_Operators_CahnHilliard", "Mesh_QuadTree_Creation", "Mesh_QuadTree_FaceDiv", "Mesh_QuadTree_HangingNodes", "Mesh_Tensor_Creation", "MT_1D_ForwardAndInversion", "MT_3D_Foward", "PF_Magnetics_Analytics", "Utils_surface2ind_topo"]
 
 ##### AUTOIMPORTS #####