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)
