mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-13 15:04:06 +08:00
D Fournier
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
def gen_DCIPsurvey(endl, mesh, stype, a, n):
|
|
|
|
from SimPEG import np
|
|
import pylab as plt
|
|
import re
|
|
"""
|
|
Load in endpoints and survey specifications to generate Tx, Rx location
|
|
stations.
|
|
|
|
Assumes flat topo for now...
|
|
|
|
Input:
|
|
:param endl -> input endpoints [x1, y1, z1, x2, y2, z2]
|
|
:object mesh -> SimPEG mesh object
|
|
:switch stype -> "dpdp" (dipole-dipole) | "pdp" (pole-dipole)
|
|
: param a, n -> pole seperation, number of rx dipoles per tx
|
|
|
|
Output:
|
|
:param Tx, Rx -> List objects for each tx location
|
|
Lines: P1x, P1y, P1z, P2x, P2y, P2z
|
|
|
|
Created on Wed December 9th, 2015
|
|
|
|
@author: dominiquef
|
|
|
|
"""
|
|
|
|
## Evenly distribute electrodes and put on surface
|
|
# Mesure survey length and direction
|
|
dl_len = np.sqrt( np.sum((endl[1,:] - endl[0,:])**2) )
|
|
dl_x = ( endl[1,0] - endl[0,0] ) / dl_len
|
|
dl_y = ( endl[1,1] - endl[0,1] ) / dl_len
|
|
|
|
nstn = np.floor( dl_len / a )
|
|
|
|
# Compute discrete pole location along line
|
|
stn_x = endl[0,0] + np.cumsum( np.ones(nstn)*dl_x*a )
|
|
stn_y = endl[0,1] + np.cumsum( np.ones(nstn)*dl_y*a )
|
|
|
|
# Create line of P1 locations
|
|
M = np.c_[stn_x, stn_y, np.ones(nstn).T*mesh.vectorNz[-1]]
|
|
|
|
# Create line of P2 locations
|
|
N = np.c_[stn_x+a*dl_x, stn_y+a*dl_y, np.ones(nstn).T*mesh.vectorNz[-1]]
|
|
|
|
## Build list of Tx-Rx locations depending on survey type
|
|
# Dipole-dipole: Moving tx with [a] spacing -> [AB a MN1 a MN2 ... a MNn]
|
|
# Pole-dipole: Moving pole on one end -> [A a MN1 a MN2 ... MNn a B]
|
|
Tx = []
|
|
Rx = []
|
|
|
|
if re.match(stype,'pdp'):
|
|
|
|
for ii in range(0, int(nstn)-1):
|
|
|
|
indx = np.min([ii+n,nstn])
|
|
Tx.append(np.c_[M[ii,:],M[ii,:]])
|
|
Rx.append(np.c_[M[ii+1:indx,:],N[ii+1:indx,:]])
|
|
|
|
|
|
elif re.match(stype,'dpdp'):
|
|
|
|
for ii in range(0, int(nstn)-2):
|
|
|
|
indx = np.min([ii+n+1,nstn])
|
|
Tx.append(np.c_[M[ii,:],N[ii,:]])
|
|
Rx.append(np.c_[M[ii+2:indx,:],N[ii+2:indx,:]])
|
|
|
|
else:
|
|
print """stype must be either 'pdp' or 'dpdp'. """
|
|
|
|
|
|
|
|
return Tx, Rx
|