def filterOnFocus(window, focused): loclist = getattr(window, 'build_loclist', None) if not loclist: return model = loclist.model() if not getattr(model, 'isFilterOnFocus', False): orig = model model = QSortFilterProxyModel() model.isFilterOnFocus = True loclist.setModel(model) model.setSourceModel(orig) model.setFilterRole(AbsolutePathRole) model.setFilterRegExp(QRegExp.escape(focused.path or ''))
def filterOnFocus(window, focused):
loclist = getattr(window, 'build_loclist', None)
if not loclist:
return
model = loclist.model()
if not getattr(model, 'isFilterOnFocus', False):
orig = model
model = QSortFilterProxyModel()
model.isFilterOnFocus = True
loclist.setModel(model)
model.setSourceModel(orig)
model.setFilterRole(AbsolutePathRole)
model.setFilterRegExp(QRegExp.escape(focused.path or ''))
class IncidentView(QtWidgets.QWidget):
def __init__(self, parent=None):
super(IncidentView, self).__init__(parent)
self.view = QtWidgets.QTreeView()
self.view.setAllColumnsShowFocus(True)
self.view.setUniformRowHeights(True)
box = QtWidgets.QVBoxLayout()
box.addWidget(self.view)
self.load_trace = QtWidgets.QPushButton('&Trace')
self.load_trace.setToolTip('Load into the Trace Window')
self.load_trace.setEnabled(False)
for activation_signal in [
self.view.activated,
self.view.entered,
self.view.pressed]:
activation_signal.connect(lambda _: self.update_controls_state())
self.load_trace.clicked.connect(self.load_current_trace)
self.view.doubleClicked.connect(self.jump_to_index)
hbox = QtWidgets.QHBoxLayout()
self.filter = QtWidgets.QLineEdit()
self.filter.textChanged.connect(self.filter_model)
filter_label = QtWidgets.QLabel('&Search')
filter_label.setBuddy(self.filter)
hbox.addWidget(filter_label)
hbox.addWidget(self.filter)
hbox.addWidget(self.load_trace)
box.addLayout(hbox)
self.setLayout(box)
self.model = None
self.proxy = None
def display(self, incidents, locations):
self.model = IncidentModel(incidents, locations, self)
self.proxy = QSortFilterProxyModel(self)
self.proxy.setSourceModel(self.model)
self.proxy.setFilterRole(self.model.filter_role)
self.proxy.setFilterRegExp(QRegExp(self.filter.text()))
self.view.setModel(self.proxy)
def filter_model(self, txt):
if self.proxy:
self.proxy.setFilterRegExp(QRegExp(txt))
def update_controls_state(self):
curr = self.view.currentIndex()
self.load_trace.setEnabled(curr.isValid() and
curr.parent().isValid())
def load_current_trace(self):
idx = self.proxy.mapToSource(self.view.currentIndex())
if not idx.isValid() or index_level(idx) not in (1, 2):
raise ValueError('load_current_trace: invalid index')
if index_level(idx) == 2:
idx = idx.parent()
incident = self.model.incidents[idx.parent().row()]
location = incident.locations[idx.row()]
backtrace = self.model.locations[location]
for p in reversed(backtrace):
self.load_trace_point(p)
def jump_to_index(self, idx):
idx = self.proxy.mapToSource(idx)
if index_level(idx) != 2:
# don't mess with parents, they are used to create children
return
grandpa = idx.parent().parent()
incident = self.model.incidents[grandpa.row()]
location = incident.locations[idx.parent().row()]
trace = self.model.locations[location]
point = trace[idx.row()]
self.show_trace_point(point)
def load_trace_point(self, p):
add_insn_to_trace_view(p.addr)
def show_trace_point(self, p):
idaapi.jumpto(p.addr)
class FontViewWidget(QWidget):
def __init__(self, statusBar, config, *args, **kwargs):
super(FontViewWidget, self).__init__(*args, **kwargs)
self.statusBar = statusBar
layout = QVBoxLayout(self)
self.retButton = QToolButton(self)
self.retButton.setIconSize(QSize(60, 60))
self.retButton.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
self.retButton.setMinimumSize(200, 100)
self.retButton.setMaximumSize(200, 100)
layout.addWidget(self.retButton)
# 过滤输入框
layout.addWidget(
QLineEdit(self, textChanged=self.doFilter,
placeholderText='过滤...'))
# Material Font
self.listView = QListView(self)
self.listView.setMouseTracking(True)
self.listView.setViewMode(QListView.IconMode)
self.listView.setMovement(QListView.Static)
self.listView.setFlow(QListView.LeftToRight)
self.listView.setWrapping(True)
self.listView.setEditTriggers(QListView.NoEditTriggers)
self.listView.setResizeMode(QListView.Adjust)
self.listView.doubleClicked.connect(self.onDoubleClicked)
self.listView.entered.connect(self.onEntered)
self.dmodel = QStandardItemModel(self.listView)
self.fmodel = QSortFilterProxyModel(self.listView)
self.fmodel.setSourceModel(self.dmodel)
self.fmodel.setFilterRole(Qt.ToolTipRole)
self.listView.setModel(self.fmodel)
layout.addWidget(self.listView)
# 字体加载器
loader = config[0]
# 添加Item
fontMap = json.loads(open(config[1],
'rb').read().decode(encoding='utf_8',
errors='ignore'),
encoding='utf_8')
for name, _ in fontMap.items():
item = QStandardItem(loader.icon(name), '')
item.setData(name, Qt.ToolTipRole)
item.setData(name, Qt.StatusTipRole)
item.setTextAlignment(Qt.AlignCenter)
item.setFlags(item.flags())
self.dmodel.appendRow(item)
def doFilter(self, _):
self.fmodel.setFilterRegExp(self.sender().text())
def onEntered(self, index):
index = self.fmodel.mapToSource(index)
text = index.data(Qt.ToolTipRole)
if text:
self.retButton.setText(text)
self.retButton.setIcon(self.dmodel.itemFromIndex(index).icon())
def onDoubleClicked(self, index):
index = self.fmodel.mapToSource(index)
text = index.data(Qt.ToolTipRole)
if text:
QApplication.clipboard().setText(text)
self.statusBar.showMessage('已复制: %s' % text)
class QtComparePanel(QWidget):
filterChanged = pyqtSignal(str)
def __init__(self, parent=None):
super(QtComparePanel, self).__init__(parent)
self.visibility_flags = []
self.visibility_buttons = []
self.labels_projects = []
self.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.setMinimumWidth(400)
self.setMinimumHeight(100)
self.data_table = QTableView()
self.data_table.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.data_table.setSelectionMode(QAbstractItemView.MultiSelection)
self.data_table.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.data_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.model = None
self.data = None
self.combodelegate1 = ComboBoxItemDelegate(self.data_table)
self.combodelegate2 = ComboBoxItemDelegate(self.data_table)
lblFilter = QLabel("Filter: ")
self.comboboxFilter = QComboBox()
self.comboboxFilter.setMinimumWidth(80)
self.comboboxFilter.addItem("All")
self.comboboxFilter.addItem("Same")
self.comboboxFilter.addItem("Born")
self.comboboxFilter.addItem("Dead")
self.comboboxFilter.addItem("Grow")
self.comboboxFilter.addItem("Shrink")
filter_layout = QHBoxLayout()
filter_layout.addWidget(lblFilter)
filter_layout.addWidget(self.comboboxFilter)
filter_layout.addStretch()
layout = QVBoxLayout()
layout.addLayout(filter_layout)
layout.addWidget(self.data_table)
self.setLayout(layout)
self.correspondences = None
self.data = None
self.comboboxFilter.currentTextChanged.connect(self.changeFilter)
def setTable(self, project, img1idx, img2idx):
self.correspondences = project.getImagePairCorrespondences(
img1idx, img2idx)
self.data = self.correspondences.data
self.model = TableModel(self.data)
self.sortfilter = QSortFilterProxyModel(self)
self.sortfilter.setSourceModel(self.model)
self.data_table.setModel(self.sortfilter)
self.data_table.setVisible(False)
self.data_table.verticalHeader().hide()
self.data_table.resizeColumnsToContents()
self.data_table.setVisible(True)
self.data_table.setItemDelegateForColumn(5, self.combodelegate1)
self.data_table.setItemDelegateForColumn(6, self.combodelegate2)
self.data_table.setEditTriggers(QAbstractItemView.DoubleClicked)
self.data_table.update()
self.data_table.setStyleSheet(
"QHeaderView::section { background-color: rgb(40,40,40) }")
def updateData(self, corr):
if corr is None:
return
self.correspondences = corr
self.sortfilter.beginResetModel()
self.model.beginResetModel()
self.model._data = corr.data
self.sortfilter.endResetModel()
self.model.endResetModel()
self.data_table.update()
def selectRows(self, rows):
self.data_table.clearSelection()
indexes = [self.model.index(r, 0) for r in rows]
mode = QItemSelectionModel.Select | QItemSelectionModel.Rows
[
self.data_table.selectionModel().select(index, mode)
for index in indexes
]
if len(rows) > 0:
value = self.data_table.horizontalScrollBar().value()
column = self.data_table.columnAt(value)
self.data_table.scrollTo(self.data_table.model().index(
rows[0], column))
@pyqtSlot(QModelIndex)
def getData(self, index):
pass
#column = index.column()
#row = index.row()
#self.data_table.model().index(row, column).data()
@pyqtSlot(str)
def changeFilter(self, txt):
if self.data is None:
return
if txt == 'All':
self.sortfilter.setFilterRegExp(QRegExp())
else:
self.sortfilter.setFilterKeyColumn(5)
self.sortfilter.setFilterRegExp(txt.lower())
self.sortfilter.setFilterRole(Qt.DisplayRole)
self.filterChanged.emit(txt.lower())
