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

Merge branch 'dcip/dev' of https://github.com/simpeg/simpeg into dcip/dev

Browse Source 
Conflicts:
	SimPEG/DCIP/DCIPUtils.py
This commit is contained in:
D Fournier
2016-03-10 14:33:06 -08:00
parent 4fefccc97d 1946e1f69e
commit 9d4e2488f3
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/DCIP/DCIPUtils.py
+13 -5
View File
@@ -169,7 +169,7 @@ def readUBC_DC2DModel(fileName):
return model
def plot_pseudoSection(DCsurvey, axs, stype):
def plot_pseudoSection(DCsurvey, axs, stype, dtype="voltage",clim=None):
"""
Read list of 2D tx-rx location and plot a speudo-section of apparent
resistivity.
@@ -230,10 +230,13 @@ def plot_pseudoSection(DCsurvey, axs, stype):
elif stype == 'dpdp':
leg = data * 2*np.pi / ( 1/MA - 1/MB - 1/NB + 1/NA )
midx = np.hstack([midx, ( Cmid + Pmid )/2 ])
midz = np.hstack([midz, -np.abs(Cmid-Pmid)/2 + z0 ])
rho = np.hstack([rho,leg])
#TODO ... let stick to list then finally convert to array.
if dtype =="voltage":
rho = np.hstack([rho,leg])
elif dtype =="appr":
rho = np.hstack([rho,data])
ax = axs
@@ -242,8 +245,12 @@ def plot_pseudoSection(DCsurvey, axs, stype):
grid_x, grid_z = np.mgrid[np.min(midx):np.max(midx), np.min(midz):np.max(midz)]
grid_rho = griddata(np.c_[midx,midz], rho.T, (grid_x, grid_z), method='linear')
if clim == None:
vmin, vmax = rho.min(), rho.max()
else:
vmin, vmax = clim[0], clim[1]
plt.imshow(grid_rho.T, extent = (np.min(midx),np.max(midx),np.min(midz),np.max(midz)), origin='lower', alpha=0.8, vmin = np.min(rho), vmax = np.max(rho))
plt.imshow(grid_rho.T, extent = (np.min(midx),np.max(midx),np.min(midz),np.max(midz)), origin='lower', alpha=0.8, vmin =vmin, vmax = vmax, clim=(vmin, vmax))
cbar = plt.colorbar(format = '%.2f',fraction=0.04,orientation="horizontal")
cmin,cmax = cbar.get_clim()
@@ -252,7 +259,7 @@ def plot_pseudoSection(DCsurvey, axs, stype):
cbar.ax.tick_params(labelsize=10)
# Plot apparent resistivity
plt.scatter(midx,midz,s=50,c=rho.T)
plt.scatter(midx,midz,s=10,c=rho.T, vmin =vmin, vmax = vmax, clim=(vmin, vmax))
ax.set_xticklabels([])
ax.set_yticklabels([])
@@ -263,6 +270,7 @@ def plot_pseudoSection(DCsurvey, axs, stype):
plt.gca().set_aspect('equal', adjustable='box')
return ax
def gen_DCIPsurvey(endl, mesh, stype, a, b, n):