def saveResult(self, folder=None, size=(16, 10), **kwargs): """Save the results in the specified folder. Saved items are: Inverted profile Velocity vector Coverage vector Standardized coverage vector Mesh (bms and vtk with results) """ # TODO: How to extract the chi2 etc. from each iteration??? subfolder = '/' + self.__class__.__name__ path = getSavePath(folder, subfolder) if self.verbose: print('Saving refraction data to: {}'.format(path)) np.savetxt(path + '/velocity.vector', self.velocity) np.savetxt(path + '/velocity-cov.vector', self.rayCoverage()) np.savetxt(path + '/velocity-scov.vector', self.standardizedCoverage()) self.mesh.addExportData('Velocity', self.velocity) self.mesh.addExportData('Coverage', self.rayCoverage()) self.mesh.addExportData('S_Coverage', self.standardizedCoverage()) self.mesh.exportVTK(path + 'velocity') self.mesh.save(path + 'velocity-mesh') self.pd.save(path + 'velocity-pd') fig, ax = plt.subplots() self.showResult(ax=ax, cov=self.standardizedCoverage(), **kwargs) fig.set_size_inches(size) fig.savefig(path + '/velocity.pdf', bbox_inches='tight') return path, fig, ax
def saveResult(self, folder=None, size=(16, 10), **kwargs): """ Saves the results in the specified folder. Saved items are: Inverted profile Velocity vector Coverage vector Standardized coverage vector Mesh (bms and vtk with results) """ # TODO: How to extract the chi2 etc. from each iteration??? subfolder = '/' + self.__class__.__name__ path = getSavePath(folder, subfolder) print('Saving refraction data to: {}'.format(path)) np.savetxt(path + '/velocity.vector', self.velocity) np.savetxt(path + '/velocity-cov.vector', self.rayCoverage()) np.savetxt(path + '/velocity-scov.vector', self.standardizedCoverage()) self.mesh.addExportData('Velocity', self.velocity) self.mesh.addExportData('Coverage', self.rayCoverage()) self.mesh.addExportData('S_Coverage', self.standardizedCoverage()) self.mesh.exportVTK(path + 'velocity') self.mesh.save(path + 'velocity-mesh') self.pd.save(path + 'velocity-pd') fig, ax = plt.subplots() self.showResult(ax=ax, cov=self.standardizedCoverage(), **kwargs) fig.set_size_inches(size) fig.savefig(path + '/velocity.pdf') return path, fig, ax