def print_figure(self, fo, dpi=150, facecolor='w', edgecolor='w', orientation='portrait', format=None, **kwargs): if _debug: print '%s.%s()' % (self.__class__.__name__, _fn_name()) # settings for printing self.figure.dpi.set(dpi) self.figure.set_facecolor(facecolor) self.figure.set_edgecolor(edgecolor) if format is None and isinstance (fo, basestring): # get format from filename extension format = os.path.splitext(fo)[1][1:] if format is not None: format = format.lower() if format == 'png': self._save_png (fo) elif format in ('pdf', 'ps', 'svg'): self._save (fo, format, orientation, **kwargs) elif format == 'eps': # backend_ps for eps from backend_ps import FigureCanvasPS as FigureCanvas fc = FigureCanvas(self.figure) fc.print_figure (filename, dpi, facecolor, edgecolor, orientation, **kwargs) else: warnings.warn('Format "%s" is not supported.\n' 'Supported formats: ' '%s.' % (format, ', '.join(IMAGE_FORMAT)))
def print_figure(self, fo, dpi=150, facecolor='w', edgecolor='w', orientation='portrait', format=None, **kwargs): if _debug: print '%s.%s()' % (self.__class__.__name__, _fn_name()) # settings for printing self.figure.dpi.set(dpi) self.figure.set_facecolor(facecolor) self.figure.set_edgecolor(edgecolor) if format is None and isinstance(fo, basestring): # get format from filename extension format = os.path.splitext(fo)[1][1:] if format is not None: format = format.lower() if format == 'png': self._save_png(fo) elif format in ('pdf', 'ps', 'svg'): self._save(fo, format, orientation, **kwargs) elif format == 'eps': # backend_ps for eps from backend_ps import FigureCanvasPS as FigureCanvas fc = FigureCanvas(self.figure) fc.print_figure(filename, dpi, facecolor, edgecolor, orientation, **kwargs) else: warnings.warn('Format "%s" is not supported.\n' 'Supported formats: ' '%s.' % (format, ', '.join(IMAGE_FORMAT)))
_save_png(figure, filename) elif ext in ('png', 'ps'): # native formats try: fileObject = file(filename, 'wb') except IOError, exc: verbose.report_error("%s: %s" % (exc.filename, exc.strerror)) else: if ext == 'png': _save_png(figure, fileObject) else: _save_ps(figure, fileObject, orientation) elif ext in ('eps', 'svg'): # backend_svg/ps if ext == 'eps': from backend_ps import FigureCanvasPS as FigureCanvas else: from backend_svg import FigureCanvasSVG as FigureCanvas fc = FigureCanvas(figure) fc.print_figure(filename, dpi, facecolor, edgecolor, orientation) else: verbose.report_error( 'Format "%s" is not supported.\nSupported formats: %s.' % (ext, ', '.join(IMAGE_FORMAT))) # restore the new params figure.dpi.set(origDPI) figure.set_facecolor(origfacecolor) figure.set_edgecolor(origedgecolor) def _save_png(figure, fileObject): width, height = figure.get_width_height()
_save_png (figure, filename) elif ext in ('png', 'ps'): # native formats try: fileObject = file(filename,'wb') except IOError, exc: verbose.report_error("%s: %s" % (exc.filename, exc.strerror)) else: if ext == 'png': _save_png (figure, fileObject) else: _save_ps (figure, fileObject, orientation) elif ext in ('eps', 'svg'): # backend_svg/ps if ext == 'eps': from backend_ps import FigureCanvasPS as FigureCanvas else: from backend_svg import FigureCanvasSVG as FigureCanvas fc = FigureCanvas(figure) fc.print_figure(filename, dpi, facecolor, edgecolor, orientation) else: verbose.report_error('Format "%s" is not supported.\nSupported formats: %s.' % (ext, ', '.join(IMAGE_FORMAT))) # restore the new params figure.dpi.set(origDPI) figure.set_facecolor(origfacecolor) figure.set_edgecolor(origedgecolor) def _save_png (figure, fileObject): width, height = figure.get_width_height() width, height = int(width), int(height)