#pragma once

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

#include "uitypes.h"

class BINARYNINJAUIAPI PossibleValueSetDialog : public QDialog
{
	Q_OBJECT

	BinaryViewRef m_view;
	uint64_t m_addr;

	QComboBox* m_combo;
	QLineEdit* m_input;
	QPushButton* m_acceptButton;
	QLabel* m_formatLabel;

	std::map<BNRegisterValueType, QString> m_typeText;
	std::map<BNRegisterValueType, QString> m_formatText;
	std::map<QString, BNRegisterValueType> m_regValueTypes;
	BNRegisterValueType m_curRegValueType;

	BinaryNinja::PossibleValueSet m_valueSet;

  public:
	PossibleValueSetDialog(
	    QWidget* parent, BinaryViewRef view, uint64_t addr, BinaryNinja::PossibleValueSet existingValue);

	BinaryNinja::PossibleValueSet getPossibleValueSet() const { return m_valueSet; }
	void validate(const QString& input);
};