def f():
doc = QTextDocument()
tables = lookup(line.text())
doc.setHtml(
template.render(request={"script_root": ""},
tables=tables,
url_for=lambda *_, **__: None))
doc.setTextWidth(doc.size().width())
pixmap = QPixmap(doc.size().width(), doc.size().height())
pixmap.fill(QColor(0, 0, 0, 0))
painter = QPainter(pixmap)
doc.drawContents(painter)
painter.end()
lab.setPixmap(pixmap)
lab.setFixedSize(pixmap.size())
def sizeHint(self, option, index):
options = QStyleOptionViewItem(option)
self.initStyleOption(options, index)
doc = QTextDocument()
doc.setHtml(options.text)
doc.setTextWidth(options.rect.width())
return QSize(doc.idealWidth(), doc.size().height())
class QBlockCode(QCachedGraphicsItem):
"""
Top-level code widget for a selection of text. Will construct an AST using
QBlockCodeObj, mirroring the structure associated with the target object.
This text is then rendered using a QTextDocument, with appropriate styles
applied to it. Interaction events will be propagated to corresponding
objects.
"""
GRAPH_ADDR_SPACING = 20
addr: int
_addr_str: str
obj: QBlockCodeObj
_config: ConfigurationManager
disasm_view: 'QDisassemblyBaseControl'
workspace: 'Workspace'
infodock: InfoDock
parent: Any
def __init__(self,
addr: int,
obj: QBlockCodeObj,
config: ConfigurationManager,
disasm_view: 'QDisassemblyBaseControl',
workspace: 'Workspace',
infodock: InfoDock,
parent: Any = None):
super().__init__(parent=parent)
self.addr = addr
self._addr_str = "%08x" % self.addr
self._addr_item: QGraphicsSimpleTextItem = None
self.obj = obj
self._width = 0
self._height = 0
self._config = config
self.parent = parent
self.workspace = workspace
self.infodock = infodock
self._disasm_view = disasm_view
self._qtextdoc = QTextDocument()
self._qtextdoc.setDefaultFont(self._config.disasm_font)
self._qtextdoc.setDocumentMargin(0)
self._addr_item = QGraphicsSimpleTextItem(self._addr_str, self)
self._addr_item.setBrush(Conf.disasm_view_node_address_color)
self._addr_item.setFont(Conf.disasm_font)
self.update_document()
self.setToolTip("Address: " + self._addr_str)
self.refresh()
def refresh(self):
self._addr_item.setVisible(self._disasm_view.show_address)
self._layout_items_and_update_size()
def update_document(self):
self._qtextdoc.clear()
cur = QTextCursor(self._qtextdoc)
self.obj.render_to_doc(cur)
def paint(self, painter, option, widget): #pylint: disable=unused-argument
self.update_document()
painter.setRenderHints(QPainter.Antialiasing
| QPainter.SmoothPixmapTransform
| QPainter.HighQualityAntialiasing)
painter.setFont(self._config.disasm_font)
if self.infodock.is_instruction_selected(
self.addr) or self.obj.should_highlight_line:
highlight_color = Conf.disasm_view_node_instruction_selected_background_color
painter.setBrush(highlight_color)
painter.setPen(highlight_color)
painter.drawRect(0, 0, self.width, self.height)
x = 0
if self._disasm_view.show_address:
x += self._addr_item.boundingRect().width(
) + self.GRAPH_ADDR_SPACING
painter.translate(QPointF(x, 0))
self._qtextdoc.drawContents(painter)
#
# Event handlers
#
def get_obj_for_mouse_event(self, event: QMouseEvent) -> QBlockCodeObj:
p = event.pos()
if self._disasm_view.show_address:
offset = self._addr_item.boundingRect().width(
) + self.GRAPH_ADDR_SPACING
p.setX(p.x() - offset)
if p.x() >= 0:
hitpos = self._qtextdoc.documentLayout().hitTest(
p, Qt.HitTestAccuracy.ExactHit)
if hitpos >= 0:
return self.obj.get_hit_obj(hitpos)
return None
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.infodock.select_instruction(self.addr)
obj = self.get_obj_for_mouse_event(event)
if obj is not None:
obj.mousePressEvent(event)
def mouseDoubleClickEvent(self, event):
obj = self.get_obj_for_mouse_event(event)
if obj is not None:
obj.mouseDoubleClickEvent(event)
#
# Private methods
#
def _layout_items_and_update_size(self):
self.update_document()
x, y = 0, 0
if self._disasm_view.show_address:
self._addr_item.setPos(x, y)
x += self._addr_item.boundingRect().width(
) + self.GRAPH_ADDR_SPACING
x += self._qtextdoc.size().width()
y += self._qtextdoc.size().height()
self._width = x
self._height = y
self.recalculate_size()
def _boundingRect(self):
return QRectF(0, 0, self._width, self._height)
