Mercurial > ldforge / file revision

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 - 2018 Teemu Piippo
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <QPainter>
#include <QMouseEvent>
#include "drawMode.h"
#include "../linetypes/modelobject.h"
#include "../glrenderer.h"
#include "../linetypes/edgeline.h"
#include "../linetypes/quadrilateral.h"
#include "../linetypes/triangle.h"

DrawMode::DrawMode (Canvas* canvas) :
    Super (canvas) {}

EditModeType DrawMode::type() const
{
	return EditModeType::Draw;
}

void DrawMode::render (QPainter& painter) const
{
	QVector<Vertex> poly;

	for (Vertex const& vert : m_drawedVerts)
		poly << vert;

	// Draw the cursor vertex as the last one in the list.
	if (countof(poly) < 4)
		poly << getCursorVertex();

	renderPolygon (painter, poly, true, true);
}

bool DrawMode::preAddVertex (Vertex const& pos)
{
	// If we picked an already-existing vertex, stop drawing
	for (Vertex& vert : m_drawedVerts)
	{
		if (vert == pos)
		{
			if (countof(m_drawedVerts) >= 2)
				endDraw();

			return true;
		}
	}

	return false;
}

void DrawMode::endDraw()
{
	// Clean the selection and create the object
	QList<Vertex>& verts = m_drawedVerts;
	Model model {m_documents};

	switch (countof(verts))
	{
	    case 2:
			// 2 verts - make a line
		    model.emplace<LDEdgeLine>(verts[0], verts[1]);
		    break;

		case 3:
		case 4:
		{
		    LDObject* obj = (countof(verts) == 3) ?
			            static_cast<LDObject*>(model.emplace<LDTriangle>()) :
			            static_cast<LDObject*>(model.emplace<LDQuadrilateral>());

			for (int i = 0; i < countof(verts); ++i)
				obj->setVertex (i, verts[i]);

			break;
		}
	}

	finishDraw (model);
}