cutter/src/widgets/RelocsWidget.cpp

140 lines
3.9 KiB
C++
Raw Normal View History

#include "RelocsWidget.h"
#include "ui_ListDockWidget.h"
#include "core/MainWindow.h"
2018-10-17 07:55:53 +00:00
#include "common/Helpers.h"
2019-03-14 09:28:42 +00:00
#include <QShortcut>
#include <QTreeWidget>
RelocsModel::RelocsModel(QList<RelocDescription> *relocs, QObject *parent) :
AddressableItemModel<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-09-30 20:00:53 +00:00
int RelocsModel::columnCount(const QModelIndex &) const
{
2018-04-24 18:40:40 +00:00
return RelocsModel::ColumnCount;
}
QVariant RelocsModel::data(const QModelIndex &index, int role) const
{
const RelocDescription &reloc = relocs->at(index.row());
2018-09-30 20:00:53 +00:00
switch (role) {
case Qt::DisplayRole:
2018-09-30 20:00:53 +00:00
switch (index.column()) {
2018-04-24 18:40:40 +00:00
case RelocsModel::VAddrColumn:
return RAddressString(reloc.vaddr);
2018-04-24 18:40:40 +00:00
case RelocsModel::TypeColumn:
return reloc.type;
2018-04-24 18:40:40 +00:00
case RelocsModel::NameColumn:
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;
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-09-30 20:00:53 +00:00
switch (section) {
2018-04-24 18:40:40 +00:00
case RelocsModel::VAddrColumn:
return tr("Address");
2018-04-24 18:40:40 +00:00
case RelocsModel::TypeColumn:
return tr("Type");
2018-04-24 18:40:40 +00:00
case RelocsModel::NameColumn:
return tr("Name");
}
return QVariant();
}
RVA RelocsModel::address(const QModelIndex &index) const
{
const RelocDescription &reloc = relocs->at(index.row());
return reloc.vaddr;
}
QString RelocsModel::name(const QModelIndex &index) const
{
const RelocDescription &reloc = relocs->at(index.row());
return reloc.name;
}
2018-04-24 18:40:40 +00:00
RelocsProxyModel::RelocsProxyModel(RelocsModel *sourceModel, QObject *parent)
: AddressableFilterProxyModel(sourceModel, parent)
2018-04-24 18:40:40 +00:00
{
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;
}
RelocsWidget::RelocsWidget(MainWindow *main, QAction *action) :
ListDockWidget(main, action),
2018-04-24 18:40:40 +00:00
relocsModel(new RelocsModel(&relocs, this)),
relocsProxyModel(new RelocsProxyModel(relocsModel, this))
{
setWindowTitle(tr("Relocs"));
setObjectName("RelocsWidget");
2018-04-24 18:40:40 +00:00
setModels(relocsProxyModel);
ui->treeView->sortByColumn(RelocsModel::NameColumn, Qt::AscendingOrder);
2018-04-24 18:40:40 +00:00
connect(Core(), &CutterCore::codeRebased, this, &RelocsWidget::refreshRelocs);
connect(Core(), &CutterCore::refreshAll, this, &RelocsWidget::refreshRelocs);
}
RelocsWidget::~RelocsWidget() {}
void RelocsWidget::refreshRelocs()
{
relocsModel->beginResetModel();
relocs = Core()->getAllRelocs();
relocsModel->endResetModel();
qhelpers::adjustColumns(ui->treeView, 3, 0);
}