summaryrefslogtreecommitdiff
path: root/ui/createarraydialog.h
blob: 61c896247c9f0fe040887f50af1681a1b5dbc23b (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
#include "uitypes.h"

#include <QtWidgets/QDialog>
#include <QtWidgets/QSpinBox>
#include <QtWidgets/QLineEdit>
#include <QtWidgets/QComboBox>

class CreateArrayDialog : public QDialog
{
	Q_OBJECT

	BinaryViewRef m_data;

	uint64_t m_startAddress;
	size_t m_size;
	TypeRef m_elementType;
	uint64_t m_elementCount{};

	QLineEdit* m_startField;
	QComboBox* m_typeDropdown;
	QLineEdit* m_countField;

	QPushButton* m_cancelButton;
	QPushButton* m_createButton;

	void update();
	size_t guessElementCount(size_t elementWidth);

public:
	explicit CreateArrayDialog(BinaryViewRef data, BNAddressRange selection, QWidget* parent = nullptr);

	/// Set the initial start address, element type, and element count for
	/// the dialog. The element type may be null if no default is desired; a
	/// default will be chosen by the dialog.
	void setInitialState();

	/// Get the desired start address from the accepted dialog.
	[[nodiscard]] uint64_t startAddress() const;

	/// Get the desired array element type from the accepted dialog.
	///
	/// The returned value will NOT be of `Type::ArrayType(...)`, but rather
	/// the element inside.
	[[nodiscard]] TypeRef elementType() const;

	/// Get the desired element count from the accepted dialog.
	[[nodiscard]] uint64_t elementCount() const;

	void accept() override;
	void reject() override;
};