def load_data_local(self):
source_path = QAanalyzer.run_info.source_path
data_sources = self.info.data_sources
files = self.info.files
if 'stat' in data_sources:
filepath = os.path.join(source_path, files.stat)
hr = HDFreader(filepath)
if not hr._success:
self.warn(
' hdf file seems to be corrupted, skipping contents:\n '
+ filepath)
#end if
hdf = hr.obj
self.data = QAHDFdata()
self.data.transfer_from(hdf)
#end if
remove = []
for name, value in self.iteritems():
if isinstance(value, HDFAnalyzer):
value.load_data_local(self.data)
value.info.data_loaded = True
if value.info.should_remove:
remove.append(name)
#end if
#end if
#end for
for name in remove:
del self[name]
def load_data_local(self,data=None):
if data==None:
self.error('attempted load without data')
#end if
exclude = self.info.exclude
self.data = QAHDFdata()
for var in data.keys():
if not var in exclude and not str(var)[0]=='_':
self.data[var] = data[var]
del data[var]
#end if
#end for
corrvars = ['LocalEnergy','ElecElec','MPC','KEcorr']
if set(corrvars)<set(self.data.keys()):
Ed,Ved,Vmd,Kcd = self.data.tuple(*corrvars)
E,E2 = Ed.value,Ed.value_squared
Ve,Ve2 = Ved.value,Ved.value_squared
Vm,Vm2 = Vmd.value,Vmd.value_squared
Kc,Kc2 = Kcd.value,Kcd.value_squared
self.data.LocalEnergy_mpc_kc = obj(
value = E-Ve+Vm+Kc,
value_squared = E2+Ve2+Vm2+Kc2 + 2*(E*(-Ve+Vm+Kc)-Ve*(Vm+Kc)+Vm*Kc)
)
