cutter/src/widgets/ExportsWidget.cpp

158 lines
4.5 KiB
C++
Raw Normal View History

#include "ExportsWidget.h"
#include "ui_ListDockWidget.h"
#include "core/MainWindow.h"
2018-10-17 07:55:53 +00:00
#include "common/Helpers.h"
#include "WidgetShortcuts.h"
2019-03-14 09:28:42 +00:00
#include <QShortcut>
ExportsModel::ExportsModel(QList<ExportDescription> *exports, QObject *parent)
: AddressableItemModel<QAbstractListModel>(parent),
2017-06-03 12:27:23 +00:00
exports(exports)
{
}
int ExportsModel::rowCount(const QModelIndex &) const
{
return exports->count();
}
int ExportsModel::columnCount(const QModelIndex &) const
{
return ExportsModel::ColumnCount;
}
QVariant ExportsModel::data(const QModelIndex &index, int role) const
{
2017-06-03 12:27:23 +00:00
if (index.row() >= exports->count())
return QVariant();
const ExportDescription &exp = exports->at(index.row());
2018-03-21 20:32:32 +00:00
switch (role) {
2017-06-03 12:27:23 +00:00
case Qt::DisplayRole:
2018-03-21 20:32:32 +00:00
switch (index.column()) {
case ExportsModel::OffsetColumn:
2017-06-03 12:27:23 +00:00
return RAddressString(exp.vaddr);
case ExportsModel::SizeColumn:
2017-06-03 12:27:23 +00:00
return RSizeString(exp.size);
case ExportsModel::TypeColumn:
2017-06-03 12:27:23 +00:00
return exp.type;
case ExportsModel::NameColumn:
2017-06-03 12:27:23 +00:00
return exp.name;
default:
return QVariant();
2017-06-03 12:27:23 +00:00
}
case ExportsModel::ExportDescriptionRole:
2017-06-03 12:27:23 +00:00
return QVariant::fromValue(exp);
default:
return QVariant();
}
}
QVariant ExportsModel::headerData(int section, Qt::Orientation, int role) const
{
2018-03-21 20:32:32 +00:00
switch (role) {
2017-06-03 12:27:23 +00:00
case Qt::DisplayRole:
2018-03-21 20:32:32 +00:00
switch (section) {
case ExportsModel::OffsetColumn:
2017-06-03 12:27:23 +00:00
return tr("Address");
case ExportsModel::SizeColumn:
2017-06-03 12:27:23 +00:00
return tr("Size");
case ExportsModel::TypeColumn:
2017-06-03 12:27:23 +00:00
return tr("Type");
case ExportsModel::NameColumn:
2017-06-03 12:27:23 +00:00
return tr("Name");
default:
return QVariant();
2017-06-03 12:27:23 +00:00
}
default:
return QVariant();
}
}
RVA ExportsModel::address(const QModelIndex &index) const
{
const ExportDescription &exp = exports->at(index.row());
return exp.vaddr;
}
QString ExportsModel::name(const QModelIndex &index) const
{
const ExportDescription &exp = exports->at(index.row());
return exp.name;
}
2018-04-24 18:40:40 +00:00
ExportsProxyModel::ExportsProxyModel(ExportsModel *source_model, QObject *parent)
: AddressableFilterProxyModel(source_model, parent)
{
setFilterCaseSensitivity(Qt::CaseInsensitive);
setSortCaseSensitivity(Qt::CaseInsensitive);
}
2018-04-24 18:40:40 +00:00
bool ExportsProxyModel::filterAcceptsRow(int row, const QModelIndex &parent) const
{
QModelIndex index = sourceModel()->index(row, 0, parent);
2018-04-24 18:40:40 +00:00
auto exp = index.data(ExportsModel::ExportDescriptionRole).value<ExportDescription>();
return exp.name.contains(filterRegExp());
}
2018-04-24 18:40:40 +00:00
bool ExportsProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
{
2018-04-24 18:40:40 +00:00
auto leftExp = left.data(ExportsModel::ExportDescriptionRole).value<ExportDescription>();
auto rightExp = right.data(ExportsModel::ExportDescriptionRole).value<ExportDescription>();
2018-03-21 20:32:32 +00:00
switch (left.column()) {
case ExportsModel::SizeColumn:
2018-04-24 18:40:40 +00:00
if (leftExp.size != rightExp.size)
return leftExp.size < rightExp.size;
2017-06-03 12:27:23 +00:00
// fallthrough
case ExportsModel::OffsetColumn:
2018-04-24 18:40:40 +00:00
if (leftExp.vaddr != rightExp.vaddr)
return leftExp.vaddr < rightExp.vaddr;
2017-06-03 12:27:23 +00:00
// fallthrough
case ExportsModel::NameColumn:
2018-04-24 18:40:40 +00:00
return leftExp.name < rightExp.name;
case ExportsModel::TypeColumn:
2018-04-24 18:40:40 +00:00
if (leftExp.type != rightExp.type)
return leftExp.type < rightExp.type;
2017-06-03 12:27:23 +00:00
default:
break;
}
// fallback
2018-04-24 18:40:40 +00:00
return leftExp.vaddr < rightExp.vaddr;
}
ExportsWidget::ExportsWidget(MainWindow *main, QAction *action) :
ListDockWidget(main, action)
{
setWindowTitle(tr("Exports"));
setObjectName("ExportsWidget");
exportsModel = new ExportsModel(&exports, this);
2018-04-24 18:40:40 +00:00
exportsProxyModel = new ExportsProxyModel(exportsModel, this);
setModels(exportsProxyModel);
ui->treeView->sortByColumn(ExportsModel::OffsetColumn, Qt::AscendingOrder);
QShortcut *toggle_shortcut = new QShortcut(widgetShortcuts["ExportsWidget"], main);
connect(toggle_shortcut, &QShortcut::activated, this, [=] (){
toggleDockWidget(true);
main->updateDockActionChecked(action);
} );
connect(Core(), SIGNAL(refreshAll()), this, SLOT(refreshExports()));
}
ExportsWidget::~ExportsWidget() {}
void ExportsWidget::refreshExports()
{
exportsModel->beginResetModel();
2018-04-12 06:33:30 +00:00
exports = Core()->getAllExports();
exportsModel->endResetModel();
qhelpers::adjustColumns(ui->treeView, 3, 0);
}