mirror of
https://github.com/shlomif/PySolFC.git
synced 2025-04-05 00:02:29 -04:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim:fenc=utf-8
|
|
#
|
|
# Distributed under terms of the CC-BY-SA licence, see:
|
|
# https://stackoverflow.com/questions/22583035
|
|
# Thanks!
|
|
|
|
import cairo
|
|
|
|
from gi import require_version
|
|
require_version('Rsvg', '2.0')
|
|
from gi.repository import Rsvg # noqa: E402
|
|
|
|
import pysnooper # noqa: E402
|
|
|
|
|
|
from pysollib.mfxutil import Image # noqa: E402
|
|
|
|
|
|
class SVGManager:
|
|
"""docstring for SVGManager"""
|
|
def __init__(self, filename):
|
|
self.filename = filename
|
|
self.svg = Rsvg.Handle().new_from_file(filename)
|
|
|
|
# Taken from https://stackoverflow.com/questions/44471795
|
|
# Under MIT License - thanks.
|
|
def pixbuf2image(self, pxb):
|
|
""" Convert GdkPixbuf.Pixbuf to PIL image """
|
|
data = pxb.get_pixels()
|
|
w = pxb.get_width()
|
|
h = pxb.get_height()
|
|
stride = pxb.get_rowstride()
|
|
mode = "RGB"
|
|
if pxb.get_has_alpha():
|
|
mode = "RGBA"
|
|
img = Image.frombytes(mode, (w, h), data, "raw", mode, stride)
|
|
return img
|
|
|
|
@pysnooper.snoop()
|
|
def render_fragment(self, id_, width, height):
|
|
id__ = '#' + id_
|
|
"""docstring for render_"""
|
|
dims = self.svg.get_dimensions_sub(id__)[1]
|
|
width_, height_ = dims.width, dims.height
|
|
pix = self.svg.get_pixbuf_sub(id__)
|
|
if False:
|
|
surface = cairo.ImageSurface(cairo.FORMAT_ARGB32,
|
|
int(width_), int(height_))
|
|
context = cairo.Context(surface)
|
|
# context.scale(width_, height_)
|
|
assert self.svg.render_cairo_sub(context, id__)
|
|
buf = bytes(surface.get_data())
|
|
image = Image.frombuffer('RGBA', (width_, height_), buf,
|
|
'raw', 'BGRA', 0, 1)
|
|
image = self.pixbuf2image(pix)
|
|
return image.resize((width, height))
|