Updates to documentation and DC utilities.

This commit is contained in:
Rowan Cockett
2016-06-26 16:07:35 -06:00
parent 57f8e35db7
commit 51c112a529
5 changed files with 21 additions and 12 deletions
+9 -7
View File
@@ -219,10 +219,12 @@ def gen_DCIPsurvey(endl, mesh, stype, a, b, n):
for ii in range(0, int(nstn)-1):
if stype == 'dpdp':
if stype == 'dipole-dipole':
tx = np.c_[M[ii,:],N[ii,:]]
elif stype == 'pdp':
elif stype == 'pole-dipole':
tx = np.c_[M[ii,:],M[ii,:]]
else:
raise Exception('The stype must be "dipole-dipole" or "pole-dipole"')
# Rx.append(np.c_[M[ii+1:indx,:],N[ii+1:indx,:]])
@@ -256,10 +258,10 @@ def gen_DCIPsurvey(endl, mesh, stype, a, b, n):
P2 = np.c_[stn_x+a*dl_x, np.ones(nstn).T*ztop]
rxClass = DC.Rx.Dipole_ky(P1, P2)
if stype == 'dpdp':
srcClass = DC.Src.Dipole([rxClass], M[ii,:],N[ii,:])
elif stype == 'pdp':
srcClass = DC.Src.Pole([rxClass], M[ii,:])
if stype == 'dipole-dipole':
srcClass = DC.Src.Dipole([rxClass], M[ii,:],N[ii,:])
elif stype == 'pole-dipole':
srcClass = DC.Src.Pole([rxClass], M[ii,:])
SrcList.append(srcClass)
elif stype == 'gradient':
@@ -310,7 +312,7 @@ def gen_DCIPsurvey(endl, mesh, stype, a, b, n):
srcClass = DC.Src.Dipole([rxClass], M[0,:], N[-1,:])
SrcList.append(srcClass)
else:
print """stype must be either 'pdp', 'dpdp' or 'gradient'. """
print """stype must be either 'pole-dipole', 'dipole-dipole' or 'gradient'. """
return SrcList
+1
View File
@@ -1,3 +1,4 @@
import DC
import IP
import SIP
import Utils
+8 -2
View File
@@ -1,5 +1,5 @@
from SimPEG import Mesh, Utils, np, sp
import SimPEG.DCIP as DC
from SimPEG.EM.Static import DC, Utils as StaticUtils
import time
def run(loc=None, sig=None, radi=None, param=None, surveyType='dipole-dipole', unitType='appConductivity', plotIt=True):
@@ -74,7 +74,13 @@ def run(loc=None, sig=None, radi=None, param=None, surveyType='dipole-dipole', u
# We will handle the geometry of the survey for you and create all the combination of tx-rx along line
# [Tx, Rx] = DC.gen_DCIPsurvey(locs, mesh, surveyType, param[0], param[1], param[2])
survey, Tx, Rx = DC.gen_DCIPsurvey(locs, mesh, surveyType, param[0], param[1], param[2])
srcList = StaticUtils.gen_DCIPsurvey(locs, mesh, surveyType, param[0], param[1], param[2])
print(srcList)
raise Exception('The output of the function changed. We need to create the survey and get Rx to continue.')
# Something like this!
survey, Tx, Rx
# Define some global geometry
dl_len = np.sqrt( np.sum((locs[0,:] - locs[1,:])**2) )
+2 -2
View File
@@ -40,7 +40,7 @@
Direct Current Resistivity
**************************
`SimPEG.DCIP` uses SimPEG as the framework for the forward and inverse
`SimPEG.EM.Static.DC` and `SimPEG.EM.Static.IP` uses SimPEG as the framework for the forward and inverse
direct current (DC) resistivity and induced polarization (IP) geophysical problems.
@@ -150,7 +150,7 @@ Comparing to the analytic function:
API for DC codes
================
.. automodule:: SimPEG.DCIP.BaseDC
.. automodule:: SimPEG.EM.Static.DC
:show-inheritance:
:members:
:undoc-members:
+1 -1
View File
@@ -7,7 +7,7 @@ Todo: docs for IP!
API for IP codes
================
.. automodule:: SimPEG.DCIP.BaseIP
.. automodule:: SimPEG.EM.Static.IP
:show-inheritance:
:members:
:undoc-members: