def picture_this_level():
# Create a QPicture drawing the contribution from this
# level only. This is all regions where the contingency is
# not empty and does not have a computed sub-contingency
# (i.e. the node does not have a child in that cell).
pic = QPicture()
painter = QPainter(pic)
ctng = node.contingencies
colors = create_image(ctng, palette, scale=scale)
x, y, w, h = node.brect
N, M = ctng.shape[:2]
# Nonzero contingency mask
any_mask = Node_mask(node)
if node.is_leaf:
skip = itertools.repeat(False)
else:
# Skip all None children they were already painted.
skip = (ch is not None for ch in node.children.flat)
painter.save()
painter.translate(x, y)
painter.scale(w / node.nbins, h / node.nbins)
indices = itertools.product(range(N), range(M))
for (i, j), skip, any_ in zip(indices, skip, any_mask.flat):
if not skip and any_:
painter.setBrush(QColor(*colors[i, j]))
if shape == Rect:
painter.drawRect(i, j, 1, 1)
elif shape == Circle:
painter.drawEllipse(i, j, 1, 1)
elif shape == RoundRect:
painter.drawRoundedRect(i, j, 1, 1, 25.0, 25.0,
Qt.RelativeSize)
painter.restore()
painter.end()
return pic
def innerShadowPixmap(color, size, pos, length=5):
"""
Args:
color (QColor): shadow color
size (QSize): size of pixmap
pos (int): shadow position int flag, use bitwise operations
1 - top
2 - right
4 - bottom
8 - left
length (int): length of cast shadow
"""
key = "InnerShadow " + \
color.name() + " " + \
str(size) + " " + \
str(pos) + " " + \
str(length)
# get cached shadow if it exists
finalShadow = QPixmapCache.find(key)
if finalShadow:
return finalShadow
shadowTemplate = shadowTemplatePixmap(color, length)
finalShadow = QPixmap(size)
finalShadow.fill(Qt.transparent)
shadowPainter = QPainter(finalShadow)
shadowPainter.setCompositionMode(QPainter.CompositionMode_Darken)
# top/bottom rect
targetRect = QRect(0, 0, size.width(), length)
# shadow on top
if pos & 1:
shadowPainter.drawPixmap(targetRect, shadowTemplate, shadowTemplate.rect())
# shadow on bottom
if pos & 4:
shadowPainter.save()
shadowPainter.translate(QPointF(0, size.height()))
shadowPainter.scale(1, -1)
shadowPainter.drawPixmap(targetRect, shadowTemplate, shadowTemplate.rect())
shadowPainter.restore()
# left/right rect
targetRect = QRect(0, 0, size.height(), shadowTemplate.rect().height())
# shadow on the right
if pos & 2:
shadowPainter.save()
shadowPainter.translate(QPointF(size.width(), 0))
shadowPainter.rotate(90)
shadowPainter.drawPixmap(targetRect, shadowTemplate, shadowTemplate.rect())
shadowPainter.restore()
# shadow on left
if pos & 8:
shadowPainter.save()
shadowPainter.translate(0, size.height())
shadowPainter.rotate(-90)
shadowPainter.drawPixmap(targetRect, shadowTemplate, shadowTemplate.rect())
shadowPainter.restore()
shadowPainter.end()
# cache shadow
QPixmapCache.insert(key, finalShadow)
return finalShadow
