Mercurial > ldforge2 / file revision

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

class Canvas : public PartRenderer
{
	Q_OBJECT
public:
	struct MouseClickInfo;
	struct MouseMoveInfo;
	using OverpaintCallback = std::function<void(Canvas*, QPainter*)>;
	Canvas(
		Model* model,
		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 drawWorldPolygon(QPainter* painter, const std::vector<glm::vec3>& points);
public slots:
	void handleSelectionChange(const QSet<ldraw::id_t>& selectedIds, const QSet<ldraw::id_t>& deselectedIds);
protected:
	void mouseMoveEvent(QMouseEvent* event) override;
	void mousePressEvent(QMouseEvent* event) override;
	void mouseReleaseEvent(QMouseEvent* event) override;
	void initializeGL() override;
	void paintGL() override;
signals:
	void newStatusText(const QString& newStatusText);
	void selectionChanged(const QSet<ldraw::id_t>& newSelection);
	void mouseClick(const MouseClickInfo& info);
	void mouseMove(const MouseMoveInfo& info);
private:
	void updateGridMatrix();
	glm::vec3 cameraVector() const;
	bool isGridPerpendicularToScreen(float threshold) const;
	std::optional<GridProgram> gridProgram;
	std::optional<AxesProgram> axesProgram;
	std::optional<glm::vec3> worldPosition;
	glm::mat4 gridMatrix;
	geom::Plane gridPlane;
	int totalMouseMove = 0;
	QSet<ldraw::id_t> selection;
	OverpaintCallback overpaintCallback = nullptr;
};

struct Canvas::MouseClickInfo
{
	std::optional<glm::vec3> worldPosition;
	Canvas* invoker;
};

struct Canvas::MouseMoveInfo
{
	std::optional<glm::vec3> worldPosition;
	Canvas* invoker;
};