mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-29 03:54:39 +08:00
seogi_macbook
39 lines
986 B
Python
39 lines
986 B
Python
import SimPEG
|
|
from SimPEG.EM.Base import BaseEMSurvey
|
|
from SimPEG import sp, Survey
|
|
from SimPEG.Utils import Zero, Identity
|
|
from RxDC import BaseRx
|
|
from SrcDC import BaseSrc
|
|
|
|
class Survey(BaseEMSurvey):
|
|
rxPair = BaseRx
|
|
srcPair = BaseSrc
|
|
|
|
def __init__(self, srcList, **kwargs):
|
|
self.srcList = srcList
|
|
BaseEMSurvey.__init__(self, srcList, **kwargs)
|
|
|
|
class Survey_ky(BaseEMSurvey):
|
|
rxPair = BaseRx
|
|
srcPair = BaseSrc
|
|
|
|
def __init__(self, srcList, **kwargs):
|
|
self.srcList = srcList
|
|
BaseEMSurvey.__init__(self, srcList, **kwargs)
|
|
|
|
def eval(self, f):
|
|
"""
|
|
Project fields to receiver locations
|
|
:param Fields u: fields object
|
|
:rtype: numpy.ndarray
|
|
:return: data
|
|
"""
|
|
data = SimPEG.Survey.Data(self)
|
|
kys = self.prob.kys
|
|
for src in self.srcList:
|
|
for rx in src.rxList:
|
|
data[src, rx] = rx.eval(kys, src, self.mesh, f)
|
|
return data
|
|
|
|
|