wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-09-10 12:37:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

SubMap as well as SubModel

Browse Source
This commit is contained in:
Rowan Cockett
2016-01-20 12:49:05 -07:00
parent f734888cb5
commit d7a0c29d3a
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Problem.py
+13 -1
View File
@@ -311,11 +311,13 @@ class GlobalProblem(BaseProblem):
assert type(ind) in [int,long] and ind >= 0 and ind < self.nGroups, 'ind must be an index into the group list' assert type(ind) in [int,long] and ind >= 0 and ind < self.nGroups, 'ind must be an index into the group list'
subMesh = self.meshes[ind] subMesh = self.meshes[ind]
subMap = Maps.IdentityMap(subMesh) # this is probably a mesh2mesh mapping?
subMap = self.getSubMap(subMesh, ind)
if self.PropMap is None: if self.PropMap is None:
prob = self.SubProblem(subMesh, mapping=subMap * self.mapping, **self.probKwargs) prob = self.SubProblem(subMesh, mapping=subMap * self.mapping, **self.probKwargs)
else: else:
# This will not work with a fancier propmap...
prob = self.SubProblem(subMesh, mapping=subMap * self._propMapMapping, **self.probKwargs) prob = self.SubProblem(subMesh, mapping=subMap * self._propMapMapping, **self.probKwargs)
survey = self.survey.__class__(srcList=self.survey.srcList[self.groups[ind]], **self.surveyKwargs) survey = self.survey.__class__(srcList=self.survey.srcList[self.groups[ind]], **self.surveyKwargs)
@@ -323,6 +325,16 @@ class GlobalProblem(BaseProblem):
return prob return prob
def getSubMap(self, subMesh, ind):
"""The sub"""
mesh2mesh = Maps.IdentityMap(subMesh) # this is probably a mesh2mesh mapping?
if self.PropMap is None:
subMap = mesh2mesh * self.mapping
else:
subMap = mesh2mesh * self._propMapMapping
return subMap
if __name__ == '__main__': if __name__ == '__main__':