Source code for yt.utilities.png_writer

Writing PNGs

# Copyright (c) 2013, yt Development Team.
# Distributed under the terms of the Modified BSD License.
# The full license is in the file COPYING.txt, distributed with this software.

import matplotlib._png as _png
from yt.extern.six import PY2

if PY2:
    from cStringIO import StringIO
    from io import BytesIO as StringIO

[docs]def call_png_write_png(buffer, width, height, filename, dpi): _png.write_png(buffer, filename, dpi)
[docs]def write_png(buffer, filename, dpi=100): width = buffer.shape[1] height = buffer.shape[0] call_png_write_png(buffer, width, height, filename, dpi)
[docs]def write_png_to_string(buffer, dpi=100, gray=0): width = buffer.shape[1] height = buffer.shape[0] fileobj = StringIO() call_png_write_png(buffer, width, height, fileobj, dpi) png_str = fileobj.getvalue() fileobj.close() return png_str