diff options
Diffstat (limited to 'examples/triage/entry.h')
| -rw-r--r-- | examples/triage/entry.h | 70 |
1 files changed, 70 insertions, 0 deletions
diff --git a/examples/triage/entry.h b/examples/triage/entry.h new file mode 100644 index 00000000..3ed82db5 --- /dev/null +++ b/examples/triage/entry.h @@ -0,0 +1,70 @@ +#pragma once + +#include <QtCore/QAbstractItemModel> +#include <QtWidgets/QTreeView> +#include "filter.h" + + +class GenericEntryModel : public QAbstractItemModel +{ + BinaryViewRef m_data; + std::vector<FunctionRef> m_allEntries, m_entries; + Qt::SortOrder m_sortOrder; + int m_sortCol; + + void performSort(int col, Qt::SortOrder order); + + public: + GenericEntryModel(QWidget* parent, BinaryViewRef data); + + virtual int columnCount(const QModelIndex& parent) const override; + virtual int rowCount(const QModelIndex& parent) const override; + virtual QVariant data(const QModelIndex& index, int role) const override; + virtual QVariant headerData(int section, Qt::Orientation orientation, int role) const override; + virtual QModelIndex index(int row, int col, const QModelIndex& parent) const override; + virtual QModelIndex parent(const QModelIndex& index) const override; + virtual void sort(int col, Qt::SortOrder order) override; + void setFilter(const std::string& filterText); + + FunctionRef getEntry(const QModelIndex& index); +}; + + +class TriageView; +class EntryWidget; + +class EntryTreeView : public QTreeView, public FilterTarget +{ + BinaryViewRef m_data; + EntryWidget* m_parent; + TriageView* m_view; + UIActionHandler m_actionHandler; + GenericEntryModel* m_model; + + public: + EntryTreeView(EntryWidget* parent, TriageView* view, BinaryViewRef data); + + virtual void setFilter(const std::string& filterText) override; + virtual void scrollToFirstItem() override; + virtual void scrollToCurrentItem() override; + virtual void selectFirstItem() override; + virtual void activateFirstItem() override; + virtual void closeFilter() override; + + protected: + virtual void keyPressEvent(QKeyEvent* event) override; + + private Q_SLOTS: + void entrySelected(const QModelIndex& cur, const QModelIndex& prev); + void entryDoubleClicked(const QModelIndex& cur); +}; + + +class EntryWidget : public QWidget +{ + FilteredView* m_filter; + + public: + EntryWidget(QWidget* parent, TriageView* view, BinaryViewRef data); + void showFilter(const QString& filter); +}; |
