wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-08-08 11:26:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixes #99

Browse Source
This commit is contained in:
Rowan Cockett
2015-05-29 11:17:56 -07:00
parent 59fcd3925f
commit de27c4e4ec
3 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Survey.py
+5
View File
@@ -231,6 +231,11 @@ class BaseSurvey(object):
[self._sourceOrder.setdefault(src.uid, ii) for ii, src in enumerate(self._srcList)]
def getSourceIndex(self, sources):
if type(sources) is not list:
sources = [sources]
for src in sources:
if getattr(src,'uid',None) is None:
raise KeyError('Source does not have a uid: %s'%str(src))
inds = map(lambda src: self._sourceOrder.get(src.uid, None), sources)
if None in inds:
raise KeyError('Some of the sources specified are not in this survey. %s'%str(inds))