#pragma once

#include <QtGui/QPainter>

#include "binaryninjaapi.h"

#include "uicontext.h"

#include "action.h"

/*!

\defgroup render Render

\ingroup uiapi

*/

/*!

\ingroup render

*/

enum HexEditorHighlightMode

{

NoHighlight,

ColumnHighlight,

ByteValueHighlight

};

/*!

\ingroup render

*/

enum HexEditorColorMode

{

NoColorization,

AsciiColorization,

ModificationColorization

};

/*!

\ingroup render

*/

enum HexEditorHighlightContrast

{

NormalContrastHighlight,

MediumContrastHighlight,

HighContrastHighlight

};

/*!

\ingroup render

*/

struct BINARYNINJAUIAPI HexEditorHighlightState

{

HexEditorHighlightMode mode;

HexEditorColorMode color;

HexEditorHighlightContrast contrast;

void restoreSettings();

void saveSettings();

};

/*!

A piece of text together with the character cell it starts at.

\ingroup render

*/

struct BINARYNINJAUIAPI CellAlignedText

{

QString text;

size_t cell;

};

/*!

\ingroup render

*/

class BINARYNINJAUIAPI FontParameters

{

QWidget* m_owner;

QFont m_font, m_emojiFont;

int m_baseline, m_charWidth, m_charHeight, m_charOffset;

float m_fontScale;

bool m_customFont;

public:

FontParameters(QWidget* parent, float fontScale = 1.0f);

void update();

const QFont& getFont() const { return m_font; }

QFont& getFont() { return m_font; }

const QFont& getEmojiFont() const { return m_emojiFont; }

QFont& getEmojiFont() { return m_emojiFont; }

void setFont(const QFont& font);

void setEmojiFont(const QFont& emojiFont);

int getBaseline() const { return m_baseline; }

int getWidth() const { return m_charWidth; }

int getHeight() const { return m_charHeight; }

int getOffset() const { return m_charOffset; }

};

/*!

\ingroup render

*/

class BINARYNINJAUIAPI RenderContext

{

QWidget* m_owner;

FontParameters m_fontParams;

bool m_drawIndents;

std::vector<QColor> m_braceColors;

QIcon m_collapsedIcon;

QIcon m_expandedIcon;

public:

RenderContext(QWidget* parent, float fontScale = 1.0f);

void update();

FontParameters& getFontParamters() { return m_fontParams; }

const FontParameters& getFontParameters() const { return m_fontParams; }

int getFontWidth() const { return m_fontParams.getWidth(); }

int getFontHeight() const { return m_fontParams.getHeight(); }

void init(QPainter& p);

QColor getColorForHexDumpByte(

const HexEditorHighlightState& state, BNModificationStatus modification, uint8_t byte);

QColor getHighlightColor(BNHighlightColor color);

HighlightTokenState getTokenForDisassemblyLinePosition(

int64_t col, const std::vector<BinaryNinja::InstructionTextToken>& tokens);

HighlightTokenState getTokenForDisassemblyTokenIndex(

size_t tokenIndex, const std::vector<BinaryNinja::InstructionTextToken>& tokens);

HighlightTokenState getHighlightTokenForTextToken(const BinaryNinja::InstructionTextToken& token);

void drawText(QPainter& p, int x, int y, QColor color, const QString& text) const;

/*! Draws text positioned on the character cell grid, so that wide code points occupy the cells

their token width accounted for. ASCII text is drawn in a single call as a fast path. */

void drawCellAlignedText(QPainter& p, int x, int y, QColor color, const std::string& text) const;

void drawUnderlinedText(QPainter& p, int x, int y, QColor color, const QString& text);

void drawSeparatorLine(QPainter& p, QColor top, QColor bottom, QColor line, const QRect& rect);

void drawInstructionHighlight(QPainter& p, const QRect& rect, bool focused = true);

void drawLinearDisassemblyLineBackground(

QPainter& p, BNLinearDisassemblyLineType type, const QRect& rect, const QRect& dirtyRect, int gutterWidth);

int drawDisassemblyLine(QPainter& p, int left, int top,

const std::vector<BinaryNinja::InstructionTextToken>& tokens, HighlightTokenState& highlight,

bool highlightOnly = false, bool renderCollapseIndicator = false) const;

void drawHexEditorLine(QPainter& p, int left, int top, const HexEditorHighlightState& highlight, BinaryViewRef view,

uint64_t lineStartAddr, size_t cols, size_t firstCol, size_t count, bool cursorVisible, bool cursorAscii,

size_t cursorPos, bool byteCursor);

QFont getFont() const { return m_fontParams.getFont(); }

QFont getEmojiFont() const { return m_fontParams.getEmojiFont(); }

void setFont(const QFont& font);

/*! Splits text into the pieces needed to draw it on the character cell grid.

Text is broken at grapheme cluster boundaries, since a cluster is what the user sees as a single

character and what the width accounting measured; a cluster can span many code points, as combining

marks, emoji modifiers, and ZWJ sequences all join into one. Runs of plain ASCII are kept together

so the font can still shape and apply ligatures across them.

The returned cell offsets always total the string's width as reported by

BinaryNinja::Unicode::GetDisplayWidth, so drawing stays aligned with the layout that width produced.

\param text Text to lay out

\return The pieces to draw, in order, each with the cell offset to draw it at

*/

static std::vector<CellAlignedText> layOutTextInCells(const std::string& text);

};