summaryrefslogtreecommitdiff
path: root/examples/triage/entry.h
blob: 013e679389b5a51f75942a5f8e6ff251a696c286 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#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, FilterOptions options);

	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);
	void copySelection();
	bool canCopySelection() const;

	virtual void setFilter(const std::string& filterText, FilterOptions options) override;
	virtual void scrollToFirstItem() override;
	virtual void scrollToCurrentItem() override;
	virtual void ensureSelection() override;
	virtual void activateSelection() 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);
};