def write_image(cls, filename, scene):
try:
scene = scene.scene()
scenerect = scene.sceneRect() # preserve scene bounding rectangle
viewrect = scene.views()[0].sceneRect()
scene.setSceneRect(viewrect)
backgroundbrush = scene.backgroundBrush(
) # preserve scene background brush
scene.setBackgroundBrush(QtCore.Qt.white)
exporter = cls._get_exporter()
cls._export(exporter(scene), filename)
scene.setBackgroundBrush(
backgroundbrush) # reset scene background brush
scene.setSceneRect(scenerect) # reset scene bounding rectangle
except Exception:
if isinstance(scene, (QGraphicsScene, QGraphicsView)):
rect = scene.sceneRect()
elif isinstance(scene, QWidget):
rect = scene.rect()
rect = rect.adjusted(-15, -15, 15, 15)
buffer = cls._get_buffer(rect.size(), filename)
painter = QtGui.QPainter()
painter.begin(buffer)
painter.setRenderHint(QtGui.QPainter.Antialiasing)
target = cls._get_target(scene, painter, buffer, rect)
try:
scene.render(painter, target, rect)
except TypeError:
scene.render(
painter) # QWidget.render() takes different params
painter.end()
cls._save_buffer(buffer, filename)
def createAttributePixmap(char, background=Qt.black, color=Qt.white):
"""
Create a QIcon with a given character. The icon is 13 pixels high and wide.
:param char: The character that is printed in the icon
:type char: str
:param background: the background color (default: black)
:type background: QColor
:param color: the character color (default: white)
:type color: QColor
:rtype: QIcon
"""
icon = QtGui.QIcon()
for size in (13, 16, 18, 20, 22, 24, 28, 32, 64):
pixmap = QtGui.QPixmap(size, size)
pixmap.fill(Qt.transparent)
painter = QtGui.QPainter()
painter.begin(pixmap)
painter.setRenderHints(painter.Antialiasing | painter.TextAntialiasing
| painter.SmoothPixmapTransform)
painter.setPen(background)
painter.setBrush(background)
margin = 1 + size // 16
text_margin = size // 20
rect = QtCore.QRectF(margin, margin, size - 2 * margin,
size - 2 * margin)
painter.drawRoundedRect(rect, 30.0, 30.0, Qt.RelativeSize)
painter.setPen(color)
font = painter.font() # type: QtGui.QFont
font.setPixelSize(size - 2 * margin - 2 * text_margin)
painter.setFont(font)
painter.drawText(rect, Qt.AlignCenter, char)
painter.end()
icon.addPixmap(pixmap)
return icon
def write_image(cls, filename, scene):
# export via svg to temp file then print that
# NOTE: can't use NamedTemporaryFile with delete = True
# (see https://bugs.python.org/issue14243)
fd, tmpname = tempfile.mkstemp(suffix=".svg")
os.close(fd)
try:
SvgFormat.write_image(tmpname, scene)
with open(tmpname, "rb") as f:
svgcontents = f.read()
finally:
os.unlink(tmpname)
svgrend = QtSvg.QSvgRenderer(QtCore.QByteArray(svgcontents))
vbox = svgrend.viewBox()
if not vbox.isValid():
size = svgrend.defaultSize()
else:
size = vbox.size()
writer = QtGui.QPdfWriter(filename)
writer.setPageSizeMM(QtCore.QSizeF(size) * 0.282)
painter = QtGui.QPainter(writer)
svgrend.render(painter)
painter.end()
del svgrend
del painter
def render_as_png(self, filename):
png = self.__transparent_png()
painter = QtGui.QPainter(png)
painter.setRenderHint(QtGui.QPainter.Antialiasing, 1)
self.scene.render(painter, QRectF(0, 0, 50, 50),
QRectF(-25, -25, 50, 50))
painter.end()
png.save(filename)
