def _generate_indicators(self, **kwargs): # pylint: disable=unused-argument
"""
Paint arrow indicators of selected instructions and labels.
"""
scene = self.scene()
for item in self._map_indicator_items:
scene.removeItem(item)
self._map_indicator_items.clear()
for addr in list(self.disasm_view.infodock.selected_insns) + list(
self.disasm_view.infodock.selected_labels):
pos = self._get_pos_from_addr(addr)
if pos is None:
continue
pos -= 1 # this is the top-left x coordinate of our arrow body (the rectangle)
pen = QPen(Qt.yellow)
brush = QBrush(Qt.yellow)
item = QGraphicsRectItem(QRectF(pos, 0, 2, 10), parent=self)
item.setPen(pen)
item.setBrush(brush)
item.setZValue(self.ZVALUE_INDICATOR)
self._map_indicator_items.append(item)
triangle = QPolygonF()
triangle.append(QPointF(pos - 1, 10))
triangle.append(QPointF(pos + 3, 10))
triangle.append(QPointF(pos + 1, 12))
triangle.append(QPointF(pos - 1, 10))
item = QGraphicsPolygonItem(triangle, parent=self)
item.setPen(pen)
item.setBrush(brush)
item.setZValue(self.ZVALUE_INDICATOR)
self._map_indicator_items.append(item)
def _generate_hover_region(self):
"""
Paint the memory region indicator.
"""
if self._map_hover_region_item:
self.scene().removeItem(self._map_hover_region_item)
self._map_hover_region_item = None
mr = self._map_hover_region
if mr is None:
return
pw = 1.0
hpw = pw / 2
pen = QPen(Qt.red)
pen.setWidth(pw)
r = self._get_region_rect(mr)
r = r.marginsRemoved(QMarginsF(hpw, hpw, hpw, hpw))
item = QGraphicsRectItem(r, parent=self)
item.setPen(pen)
item.setZValue(self.ZVALUE_HOVER)
self._map_hover_region_item = item
def _create_line_indicator(self, addr, item_map, color=Qt.yellow, show_frontier=False, z=None, z_frontier=None):
"""
Generate a cursor at a given address.
"""
pos_x = self._get_pos_from_addr(addr)
if pos_x is None:
return
pen = QPen(color)
brush = QBrush(color)
height = self.height
tri_width = 7
tri_height = 4
pos_x = int(pos_x - tri_width / 2) # Center drawing
center = pos_x + int(tri_width / 2)
pos_y = 0
frontier_width = int(0.15 * max(self.width, self.height))
if show_frontier:
# Draw frontier gradients
r = QRectF(center - frontier_width, pos_y, frontier_width, height)
bg = QLinearGradient(r.topLeft(), r.topRight())
color = Qt.red
top_color = QColor(color)
top_color.setAlpha(0)
bg.setColorAt(0, top_color)
bottom_color = QColor(color)
bottom_color.setAlpha(180)
bg.setColorAt(1, bottom_color)
i = QGraphicsRectItem(r, parent=self)
i.setPen(Qt.NoPen)
i.setBrush(bg)
if z_frontier is not None:
i.setZValue(z_frontier)
item_map.append(i)
r = QRectF(center, pos_y, frontier_width, height)
bg = QLinearGradient(r.topLeft(), r.topRight())
color = Qt.blue
top_color = QColor(color)
bg.setColorAt(0, top_color)
bottom_color = QColor(color)
bottom_color.setAlpha(0)
bg.setColorAt(1, bottom_color)
i = QGraphicsRectItem(r, parent=self)
i.setPen(Qt.NoPen)
i.setBrush(bg)
if z_frontier is not None:
i.setZValue(z_frontier)
item_map.append(i)
# Draw line
i = QGraphicsLineItem(center, 0, center, height, parent=self)
i.setPen(pen)
if z is not None:
i.setZValue(z)
item_map.append(i)
# Draw top and bottom triangles
t = QPolygonF()
t.append(QPointF(pos_x, pos_y))
t.append(QPointF(pos_x + tri_width - 1, pos_y))
t.append(QPointF(center, pos_y + tri_height - 1))
t.append(QPointF(pos_x, pos_y))
pos_y += height - 1
b = QPolygonF()
b.append(QPointF(pos_x, pos_y))
b.append(QPointF(center, pos_y - tri_height + 1))
b.append(QPointF(pos_x + tri_width - 1, pos_y))
b.append(QPointF(pos_x, pos_y))
for i in [QGraphicsPolygonItem(t, parent=self),
QGraphicsPolygonItem(b, parent=self)]:
i.setPen(pen)
i.setBrush(brush)
if z is not None:
i.setZValue(z)
item_map.append(i)
class QScene2dViewer(AbstractGraphicViewer):
drawaxis = False
axis_center = None
axis_x = None
axis_y = None
G = None
def __init__(self, G):
super().__init__()
self.G = G
# self.setMinimunSize(400, 400)
self.scale(1, -1)
# AXIS
self.axis_center = QGraphicsRectItem(-100, -100, 200, 200)
self.axis_center.setPen(QPen(QColor("black")))
self.axis_center.setBrush(QBrush(QColor("black")))
self.axis_center.setZValue(5000)
self.axis_x = QGraphicsRectItem(0, 0, 1000, 30)
self.axis_x.setPen(QPen(QColor("red")))
self.axis_x.setBrush(QBrush(QColor("red")))
self.axis_x.setZValue(5000)
self.axis_y = QGraphicsRectItem(0, 0, 30, 1000)
self.axis_y.setPen(QPen(QColor("blue")))
self.axis_y.setBrush(QBrush(QColor("blue")))
self.axis_y.setZValue(5000)
self.set_draw_axis(True)
self.draw_axis()
def create_graph(self):
pass
def set_draw_axis(self, visible):
self.drawaxis = visible
def draw_axis(self):
if self.drawaxis:
self.scene.addItem(self.axis_center)
self.scene.addItem(self.axis_x)
self.scene.addItem(self.axis_y)
else:
self.scene.removeItem(self.axis_center)
self.scene.removeItem(self.axis_x)
self.scene.removeItem(self.axis_y)
# SLOTS
def add_or_assign_node_slot(self, node_id, node_type, name=""):
pass
def add_or_assign_node_slot(self, node_id, node):
pass
def add_or_assign_edge_slot(self, from_node, to_node, node_type):
pass
def del_edge_slot(self, from_node, to_node, edge_tag):
pass
def del_node_slot(self, node_id):
pass
def node_change_slot(self, value, node_id, node_type, parent=None):
pass
def category_change_slot(self, value, parent=None):
pass
