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) )