def _write_charts(z, data): if data['chart(0,0)'].dtype == numpy.dtype(numpy.uint8): ext = 'png' elif data['chart(0,0)'].dtype == numpy.dtype(numpy.float64): _write_charts_fits(z, data) return else: ext = 'png' import formats for atlasno, atlas in enumerate(formats.get_atlases(data)): for chartno, _ in enumerate(atlas): s = StringIO() filename = 'chart.%d-%d.%s' % (atlasno, chartno, ext) formats.write(filename, data, s, (atlasno, chartno)) z.writestr('chart.%d-%d.%s' % (atlasno, chartno, ext), s.getvalue())
import sys sys.path.append('..') import graphcontext import formats ctx = graphcontext.GraphContext() print "Loading spring-furball" formats.load(ctx, '../../graphs/spring-furball.mg2') print "Writing test.gml" formats.write(ctx, 'test.gml') print "Loading test.gml" ctx.graph.clear() formats.gml.load(ctx.graph, file('test.gml')) print "Writing test2.gml" formats.gml.write(ctx.graph, file('test2.gml', 'w'))
def _write_charts_fits(z, data): import formats s = StringIO() formats.write('charts.fits', data, s) z.writestr('charts.fits', s.getvalue())
def file_save(self, filename): self.resources.update(formats.wrap(self.world, self.view), True) formats.write(filename, self.resources) self.last_filename = filename