2017-10-02 16:18:40 +00:00
|
|
|
#include "ExportsWidget.h"
|
|
|
|
#include "ui_ExportsWidget.h"
|
2017-10-01 19:09:42 +00:00
|
|
|
#include "MainWindow.h"
|
|
|
|
#include "utils/Helpers.h"
|
2017-05-19 07:45:26 +00:00
|
|
|
|
|
|
|
ExportsModel::ExportsModel(QList<ExportDescription> *exports, QObject *parent)
|
2017-06-03 12:27:23 +00:00
|
|
|
: QAbstractListModel(parent),
|
|
|
|
exports(exports)
|
2017-05-19 07:45:26 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
int ExportsModel::rowCount(const QModelIndex &) const
|
|
|
|
{
|
|
|
|
return exports->count();
|
|
|
|
}
|
|
|
|
|
|
|
|
int ExportsModel::columnCount(const QModelIndex &) const
|
|
|
|
{
|
2018-04-23 07:53:35 +00:00
|
|
|
return ExportsModel::ColumnCount;
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
QVariant ExportsModel::data(const QModelIndex &index, int role) const
|
|
|
|
{
|
2017-06-03 12:27:23 +00:00
|
|
|
if (index.row() >= exports->count())
|
2017-05-19 07:45:26 +00:00
|
|
|
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()) {
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::OffsetColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return RAddressString(exp.vaddr);
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::SizeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return RSizeString(exp.size);
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::TypeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return exp.type;
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::NameColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return exp.name;
|
2017-05-19 07:45:26 +00:00
|
|
|
default:
|
|
|
|
return QVariant();
|
2017-06-03 12:27:23 +00:00
|
|
|
}
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::ExportDescriptionRole:
|
2017-06-03 12:27:23 +00:00
|
|
|
return QVariant::fromValue(exp);
|
|
|
|
default:
|
|
|
|
return QVariant();
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
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) {
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::OffsetColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return tr("Address");
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::SizeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return tr("Size");
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::TypeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return tr("Type");
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::NameColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return tr("Name");
|
2017-05-19 07:45:26 +00:00
|
|
|
default:
|
|
|
|
return QVariant();
|
2017-06-03 12:27:23 +00:00
|
|
|
}
|
|
|
|
default:
|
|
|
|
return QVariant();
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExportsModel::beginReloadExports()
|
|
|
|
{
|
|
|
|
beginResetModel();
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExportsModel::endReloadExports()
|
|
|
|
{
|
|
|
|
endResetModel();
|
|
|
|
}
|
|
|
|
|
2018-03-21 20:32:32 +00:00
|
|
|
ExportsSortFilterProxyModel::ExportsSortFilterProxyModel(ExportsModel *source_model,
|
|
|
|
QObject *parent)
|
2017-06-03 12:27:23 +00:00
|
|
|
: QSortFilterProxyModel(parent)
|
2017-05-19 07:45:26 +00:00
|
|
|
{
|
|
|
|
setSourceModel(source_model);
|
2018-04-23 07:53:35 +00:00
|
|
|
setFilterCaseSensitivity(Qt::CaseInsensitive);
|
|
|
|
setSortCaseSensitivity(Qt::CaseInsensitive);
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool ExportsSortFilterProxyModel::filterAcceptsRow(int row, const QModelIndex &parent) const
|
|
|
|
{
|
|
|
|
QModelIndex index = sourceModel()->index(row, 0, parent);
|
|
|
|
ExportDescription exp = index.data(ExportsModel::ExportDescriptionRole).value<ExportDescription>();
|
|
|
|
return exp.name.contains(filterRegExp());
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ExportsSortFilterProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
|
|
|
|
{
|
2018-03-21 20:32:32 +00:00
|
|
|
ExportDescription left_exp = left.data(
|
|
|
|
ExportsModel::ExportDescriptionRole).value<ExportDescription>();
|
|
|
|
ExportDescription right_exp = right.data(
|
|
|
|
ExportsModel::ExportDescriptionRole).value<ExportDescription>();
|
2017-05-19 07:45:26 +00:00
|
|
|
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (left.column()) {
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::SizeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
if (left_exp.size != right_exp.size)
|
|
|
|
return left_exp.size < right_exp.size;
|
|
|
|
// fallthrough
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::OffsetColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
if (left_exp.vaddr != right_exp.vaddr)
|
|
|
|
return left_exp.vaddr < right_exp.vaddr;
|
|
|
|
// fallthrough
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::NameColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
return left_exp.name < right_exp.name;
|
2018-04-23 07:53:35 +00:00
|
|
|
case ExportsModel::TypeColumn:
|
2017-06-03 12:27:23 +00:00
|
|
|
if (left_exp.type != right_exp.type)
|
|
|
|
return left_exp.type < right_exp.type;
|
|
|
|
default:
|
|
|
|
break;
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// fallback
|
|
|
|
return left_exp.vaddr < right_exp.vaddr;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
2018-03-16 21:46:57 +00:00
|
|
|
ExportsWidget::ExportsWidget(MainWindow *main, QAction *action) :
|
|
|
|
CutterDockWidget(main, action),
|
|
|
|
ui(new Ui::ExportsWidget)
|
2017-05-19 07:45:26 +00:00
|
|
|
{
|
|
|
|
ui->setupUi(this);
|
|
|
|
|
2018-04-23 07:53:35 +00:00
|
|
|
exportsModel = new ExportsModel(&exports, this);
|
|
|
|
exportsProxyModel = new ExportsSortFilterProxyModel(exportsModel, this);
|
|
|
|
ui->exportsTreeView->setModel(exportsProxyModel);
|
|
|
|
ui->exportsTreeView->sortByColumn(ExportsModel::OffsetColumn, 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 &)),
|
|
|
|
exportsProxyModel, SLOT(setFilterWildcard(const QString &)));
|
|
|
|
connect(ui->quickFilterView, SIGNAL(filterClosed()), ui->exportsTreeView, SLOT(setFocus()));
|
2017-05-19 07:45:26 +00:00
|
|
|
|
|
|
|
setScrollMode();
|
|
|
|
|
2017-11-19 12:56:10 +00:00
|
|
|
connect(Core(), SIGNAL(refreshAll()), this, SLOT(refreshExports()));
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
|
2017-11-19 12:56:10 +00:00
|
|
|
ExportsWidget::~ExportsWidget() {}
|
2017-05-19 07:45:26 +00:00
|
|
|
|
|
|
|
void ExportsWidget::refreshExports()
|
|
|
|
{
|
2018-04-23 07:53:35 +00:00
|
|
|
exportsModel->beginReloadExports();
|
2018-04-12 06:33:30 +00:00
|
|
|
exports = Core()->getAllExports();
|
2018-04-23 07:53:35 +00:00
|
|
|
exportsModel->endReloadExports();
|
2017-05-19 07:45:26 +00:00
|
|
|
|
2018-04-01 08:25:31 +00:00
|
|
|
qhelpers::adjustColumns(ui->exportsTreeView, 3, 0);
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void ExportsWidget::setScrollMode()
|
|
|
|
{
|
|
|
|
qhelpers::setVerticalScrollMode(ui->exportsTreeView);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExportsWidget::on_exportsTreeView_doubleClicked(const QModelIndex &index)
|
|
|
|
{
|
|
|
|
ExportDescription exp = index.data(ExportsModel::ExportDescriptionRole).value<ExportDescription>();
|
2018-04-12 06:33:30 +00:00
|
|
|
Core()->seek(exp.vaddr);
|
2017-05-19 07:45:26 +00:00
|
|
|
}
|