src/ui/canvas.h

#pragma once
#include <QPainter>
#include <QBrush>
#include <QPen>
#include "gl/partrenderer.h"
#include "gl/gridprogram.h"
#include "gl/axesprogram.h"
#include "gl/vertexprogram.h"

enum EditingMode
{
	SelectMode,
	DrawMode
};

struct DrawState
{
	std::vector<glm::vec3> polygon;
	std::vector<glm::vec3> previewPolygon;
	glm::vec3 previewPoint;
	bool isconcave = false;
};

class Canvas : public PartRenderer
{
	Q_OBJECT
public:
	using OverpaintCallback = std::function<void(Canvas*, QPainter*)>;
	Canvas(
		Model* model,
		Document* document,
		DocumentManager* documents,
		const ldraw::ColorTable& colorTable,
		QWidget* parent = nullptr);
	void clearSelection();
	void addToSelection(ldraw::id_t id);
	void setOverpaintCallback(OverpaintCallback fn);
	void drawWorldPoint(QPainter* painter, const glm::vec3& worldPoint) const;
	void drawWorldPolyline(QPainter* painter, const std::vector<glm::vec3>& points);
	void drawWorldPolygon(QPainter* painter, const std::vector<glm::vec3>& points);
	Winding worldPolygonWinding(const std::vector<glm::vec3>& points) const;
	const std::optional<glm::vec3>& getWorldPosition() const;
	const QSet<ldraw::id_t> selectedObjects() const;
	const glm::mat4& getGridMatrix() const;
	void closeShape();
	glm::vec3 cameraVector() const;
	DrawState drawState;
	EditingMode mode = SelectMode;
public Q_SLOTS:
	void handleSelectionChange(const QSet<ldraw::id_t>& selectedIds, const QSet<ldraw::id_t>& deselectedIds);
	void rebuildVertices(Document *document);
	void setGridMatrix(const glm::mat4 &newMatrix);
protected:
	void mouseMoveEvent(QMouseEvent* event) override;
	void mousePressEvent(QMouseEvent* event) override;
	void mouseReleaseEvent(QMouseEvent* event) override;
	void initializeGL() override;
	void paintGL() override;
Q_SIGNALS:
	void newStatusText(const QString& newStatusText);
	void selectionChanged(const QSet<ldraw::id_t>& newSelection);
	void mouseClick(Canvas* canvas, QMouseEvent* event);
	void mouseMove(Canvas* canvas, QMouseEvent* event);
private:
	bool isGridPerpendicularToScreen(float threshold) const;
	QVector<QPointF> convertWorldPointsToScreenPoints(const std::vector<glm::vec3>& worldPoints) const;
	Q_SLOT void updateCanvasRenderPreferences();
	void renderAxesLabels(QPainter& painter);
	std::optional<GridProgram> gridProgram;
	std::optional<AxesProgram> axesProgram;
	std::optional<VertexProgram> vertexProgram;
	std::optional<glm::vec3> worldPosition;
	glm::mat4 gridMatrix;
	geom::Plane gridPlane;
	int totalMouseMove = 0;
	bool isDark = true;
	QSet<ldraw::id_t> selection;
	OverpaintCallback overpaintCallback = nullptr;
	Document* document;
};

void adjustGridToView(Canvas* canvas);