refactor and generalize where things are being saved

This commit is contained in:
rowanc1
2013-12-04 17:55:09 -08:00
parent 191d0e242c
commit e25afacd1b
3 changed files with 33 additions and 19 deletions
+6
View File
@@ -349,6 +349,12 @@ class BaseInversion(object):
return dmisfit
def save(self, group):
group.attrs['phi_d'] = self.phi_d
group.attrs['phi_m'] = self.phi_m
group.setArray('m', self.m)
group.setArray('dpred', self.dpred)
class Inversion(Cooling, Remember, BaseInversion):
maxIter = 10
+11
View File
@@ -431,6 +431,17 @@ class Minimize(object):
if self.debug: self.printDone()
def save(self, group):
group.setArray('searchDirection', self.searchDirection)
if getattr(self,'parent',None) is None:
group.setArray('x', self.xc)
else: # Assume inversion is the parent
group.setArray('m', self.xc)
group.setArray('dpred', self.parent.dpred)
class Remember(object):
"""
This mixin remembers all the things you tend to forget.