2017-09-25 12:55:41 +00:00
|
|
|
#include <QTreeWidget>
|
2017-10-02 16:18:40 +00:00
|
|
|
#include "RelocsWidget.h"
|
|
|
|
#include "ui_RelocsWidget.h"
|
2017-10-01 19:09:42 +00:00
|
|
|
#include "MainWindow.h"
|
|
|
|
#include "utils/Helpers.h"
|
2017-04-13 15:14:02 +00:00
|
|
|
|
2018-03-25 10:57:55 +00:00
|
|
|
RelocsModel::RelocsModel(QList<RelocDescription> *relocs, QObject *parent) :
|
|
|
|
QAbstractTableModel(parent),
|
|
|
|
relocs(relocs)
|
|
|
|
{}
|
|
|
|
|
|
|
|
int RelocsModel::rowCount(const QModelIndex &parent) const
|
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
return parent.isValid() ? 0 : relocs->count();
|
2018-03-25 10:57:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
int RelocsModel::columnCount(const QModelIndex&) const
|
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
return RelocsModel::ColumnCount;
|
2018-03-25 10:57:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
QVariant RelocsModel::data(const QModelIndex &index, int role) const
|
|
|
|
{
|
|
|
|
const RelocDescription &reloc = relocs->at(index.row());
|
|
|
|
switch (role)
|
|
|
|
{
|
|
|
|
case Qt::DisplayRole:
|
|
|
|
switch (index.column())
|
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::VAddrColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return RAddressString(reloc.vaddr);
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::TypeColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return reloc.type;
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::NameColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return reloc.name;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
2018-06-26 16:17:03 +00:00
|
|
|
break;
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::RelocDescriptionRole:
|
|
|
|
return QVariant::fromValue(reloc);
|
|
|
|
case RelocsModel::AddressRole:
|
|
|
|
return reloc.vaddr;
|
2018-03-25 10:57:55 +00:00
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant RelocsModel::headerData(int section, Qt::Orientation, int role) const
|
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
if (role == Qt::DisplayRole)
|
2018-03-25 10:57:55 +00:00
|
|
|
switch (section)
|
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::VAddrColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return tr("Address");
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::TypeColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return tr("Type");
|
2018-04-24 18:40:40 +00:00
|
|
|
case RelocsModel::NameColumn:
|
2018-03-25 10:57:55 +00:00
|
|
|
return tr("Name");
|
|
|
|
}
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
|
|
|
|
void RelocsModel::beginReload()
|
|
|
|
{
|
|
|
|
beginResetModel();
|
|
|
|
}
|
|
|
|
|
|
|
|
void RelocsModel::endReload()
|
|
|
|
{
|
|
|
|
endResetModel();
|
|
|
|
}
|
|
|
|
|
2018-04-24 18:40:40 +00:00
|
|
|
RelocsProxyModel::RelocsProxyModel(RelocsModel *sourceModel, QObject *parent)
|
|
|
|
: QSortFilterProxyModel(parent)
|
|
|
|
{
|
|
|
|
setSourceModel(sourceModel);
|
|
|
|
setFilterCaseSensitivity(Qt::CaseInsensitive);
|
|
|
|
setSortCaseSensitivity(Qt::CaseInsensitive);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool RelocsProxyModel::filterAcceptsRow(int row, const QModelIndex &parent) const
|
|
|
|
{
|
|
|
|
QModelIndex index = sourceModel()->index(row, 0, parent);
|
|
|
|
auto reloc = index.data(RelocsModel::RelocDescriptionRole).value<RelocDescription>();
|
|
|
|
|
|
|
|
return reloc.name.contains(filterRegExp());
|
|
|
|
}
|
|
|
|
|
|
|
|
bool RelocsProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
|
|
|
|
{
|
|
|
|
if (!left.isValid() || !right.isValid())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (left.parent().isValid() || right.parent().isValid())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
auto leftReloc = left.data(RelocsModel::RelocDescriptionRole).value<RelocDescription>();
|
|
|
|
auto rightReloc = right.data(RelocsModel::RelocDescriptionRole).value<RelocDescription>();
|
|
|
|
|
|
|
|
switch (left.column()) {
|
|
|
|
case RelocsModel::VAddrColumn:
|
|
|
|
return leftReloc.vaddr < rightReloc.vaddr;
|
|
|
|
case RelocsModel::TypeColumn:
|
|
|
|
return leftReloc.type < rightReloc.type;
|
|
|
|
case RelocsModel::NameColumn:
|
|
|
|
return leftReloc.name < rightReloc.name;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
2018-03-25 10:57:55 +00:00
|
|
|
|
2018-03-16 21:46:57 +00:00
|
|
|
RelocsWidget::RelocsWidget(MainWindow *main, QAction *action) :
|
|
|
|
CutterDockWidget(main, action),
|
2018-03-25 10:57:55 +00:00
|
|
|
ui(new Ui::RelocsWidget),
|
2018-04-24 18:40:40 +00:00
|
|
|
relocsModel(new RelocsModel(&relocs, this)),
|
|
|
|
relocsProxyModel(new RelocsProxyModel(relocsModel, this))
|
2017-03-29 10:18:37 +00:00
|
|
|
{
|
|
|
|
ui->setupUi(this);
|
|
|
|
|
2018-04-24 18:40:40 +00:00
|
|
|
ui->relocsTreeView->setModel(relocsProxyModel);
|
|
|
|
ui->relocsTreeView->sortByColumn(RelocsModel::NameColumn, Qt::AscendingOrder);
|
|
|
|
|
|
|
|
// Ctrl-F to show/hide the filter entry
|
|
|
|
QShortcut *searchShortcut = new QShortcut(QKeySequence::Find, this);
|
|
|
|
connect(searchShortcut, &QShortcut::activated, ui->quickFilterView, &QuickFilterView::showFilter);
|
|
|
|
searchShortcut->setContext(Qt::WidgetWithChildrenShortcut);
|
|
|
|
|
|
|
|
// Esc to clear the filter entry
|
|
|
|
QShortcut *clearShortcut = new QShortcut(QKeySequence(Qt::Key_Escape), this);
|
|
|
|
connect(clearShortcut, &QShortcut::activated, ui->quickFilterView, &QuickFilterView::clearFilter);
|
|
|
|
clearShortcut->setContext(Qt::WidgetWithChildrenShortcut);
|
|
|
|
|
|
|
|
connect(ui->quickFilterView, SIGNAL(filterTextChanged(const QString &)),
|
|
|
|
relocsProxyModel, SLOT(setFilterWildcard(const QString &)));
|
|
|
|
connect(ui->quickFilterView, SIGNAL(filterClosed()), ui->relocsTreeView, SLOT(setFocus()));
|
2018-03-25 10:57:55 +00:00
|
|
|
|
2017-04-13 15:14:02 +00:00
|
|
|
setScrollMode();
|
|
|
|
|
2018-03-25 10:57:55 +00:00
|
|
|
connect(Core(), SIGNAL(refreshAll()), this, SLOT(refreshRelocs()));
|
2017-04-13 15:14:02 +00:00
|
|
|
}
|
|
|
|
|
2017-11-19 12:56:10 +00:00
|
|
|
RelocsWidget::~RelocsWidget() {}
|
2017-04-13 15:14:02 +00:00
|
|
|
|
2018-04-12 06:33:03 +00:00
|
|
|
void RelocsWidget::on_relocsTreeView_doubleClicked(const QModelIndex &index)
|
2017-03-29 10:18:37 +00:00
|
|
|
{
|
2018-04-23 07:54:06 +00:00
|
|
|
if (!index.isValid())
|
|
|
|
return;
|
|
|
|
|
2018-03-25 10:57:55 +00:00
|
|
|
Core()->seek(index.data(RelocsModel::AddressRole).toLongLong());
|
2017-04-13 15:14:02 +00:00
|
|
|
}
|
|
|
|
|
2018-03-25 10:57:55 +00:00
|
|
|
void RelocsWidget::refreshRelocs()
|
2017-04-13 15:14:02 +00:00
|
|
|
{
|
2018-04-24 18:40:40 +00:00
|
|
|
relocsModel->beginReload();
|
2018-03-25 10:57:55 +00:00
|
|
|
relocs = Core()->getAllRelocs();
|
2018-04-24 18:40:40 +00:00
|
|
|
relocsModel->endReload();
|
2018-04-12 06:33:03 +00:00
|
|
|
qhelpers::adjustColumns(ui->relocsTreeView, 3, 0);
|
2017-04-13 15:14:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void RelocsWidget::setScrollMode()
|
|
|
|
{
|
2018-04-12 06:33:03 +00:00
|
|
|
qhelpers::setVerticalScrollMode(ui->relocsTreeView);
|
2017-03-29 10:18:37 +00:00
|
|
|
}
|