#pragma once
#include "Common/UI/UIScreen.h"
#include "UI/MiscScreens.h"

class JitCompareScreen : public UIDialogScreenWithBackground {
public:
	JitCompareScreen();
	void CreateViews() override;

	const char *tag() const override { return "JitCompare"; }

private:
	void Flip();
	void UpdateDisasm();

	// Uses the current ListType
	void FillBlockList();

	UI::LinearLayout *comparisonView_;
	UI::LinearLayout *leftDisasm_;
	UI::LinearLayout *rightDisasm_;

	UI::LinearLayout *blockListView_;
	UI::LinearLayout *blockListContainer_;

	UI::LinearLayout *statsView_;
	UI::LinearLayout *statsContainer_;

	UI::EventReturn OnSelectBlock(UI::EventParams &e);
	UI::EventReturn OnBlockAddress(UI::EventParams &e);
	UI::EventReturn OnAddressChange(UI::EventParams &e);
	UI::EventReturn OnShowStats(UI::EventParams &e);
	UI::EventReturn OnBlockClick(UI::EventParams &e);

	// To switch, change the below things and call RecreateViews();
	enum class ViewMode {
		BLOCK_LIST,
		DISASM,
		STATS,
	};
	enum class ListType {
		ALL_BLOCKS,
		FPU_BLOCKS,
		VFPU_BLOCKS,
	};
	enum class ListSort {
		BLOCK_NUM,
		BLOCK_LENGTH_DESC,
		BLOCK_LENGTH_ASC,
		TIME_SPENT,
		EXECUTIONS,
		MAX
	};
	ViewMode viewMode_ = ViewMode::BLOCK_LIST;
	ListType listType_ = ListType::ALL_BLOCKS;
	ListSort listSort_ = ListSort::TIME_SPENT;

	int currentBlock_ = -1;  // For DISASM mode
	int64_t sumTotalNanos_ = 0;
	int64_t sumExecutions_ = 0;
	std::vector<int> blockList_;  // for BLOCK_LIST mode

	UI::TextView *blockName_;
	UI::TextEdit *blockAddr_;
	UI::TextView *blockStats_;
};

class AddressPromptScreen : public PopupScreen {
public:
	AddressPromptScreen(std::string_view title) : PopupScreen(title, "OK", "Cancel") {}

	const char *tag() const override { return "AddressPrompt"; }

	bool key(const KeyInput &key) override;

	UI::Event OnChoice;

protected:
	void CreatePopupContents(UI::ViewGroup *parent) override;
	void OnCompleted(DialogResult result) override;
	UI::EventReturn OnDigitButton(UI::EventParams &e);
	UI::EventReturn OnBackspace(UI::EventParams &e);

private:
	void AddDigit(int n);
	void BackspaceDigit();
	void UpdatePreviewDigits();

	UI::TextView *addrView_ = nullptr;
	UI::Button *buttons_[16]{};
	unsigned int addr_ = 0;
};