qelectrotech-source-mirror/sources/dvevent/dveventaddshape.cpp

/*
	Copyright 2006-2014 The QElectroTech Team
	This file is part of QElectroTech.

	QElectroTech 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 2 of the License, or
	(at your option) any later version.

	QElectroTech 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 QElectroTech.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "dveventaddshape.h"
#include "diagramview.h"
#include "qetshapeitem.h"
#include "diagram.h"
#include "diagramcommands.h"
#include <QMouseEvent>

/**
 * @brief DVEventAddShape::DVEventAddShape
 * Default constructor
 * @param dv, the diagram view where operate this event
 * @param shape_type, the shape type to draw
 */
DVEventAddShape::DVEventAddShape(DiagramView *dv, QetShapeItem::ShapeType shape_type) :
	DVEventInterface(dv),
	m_shape_type (shape_type),
	m_shape_item (nullptr),
	m_help_horiz (nullptr),
	m_help_verti (nullptr)
{}

/**
 * @brief DVEventAddShape::~DVEventAddShape
 */
DVEventAddShape::~DVEventAddShape() {
	if (m_running || m_abort) {
		m_diagram -> removeItem(m_shape_item);
		delete m_shape_item;
	}
	delete m_help_horiz;
	delete m_help_verti;
	m_dv -> setContextMenuPolicy(Qt::DefaultContextMenu);
}

/**
 * @brief DVEventAddShape::mousePressEvent
 * Action when mouse is pressed
 * @param event : event of mouse press
 * @return : true if this event is managed, otherwise false
 */
bool DVEventAddShape::mousePressEvent(QMouseEvent *event) {

	if (!m_dv->isInteractive() && m_diagram->isReadOnly()) return false;

	QPointF pos = m_dv->mapToScene(event->pos());

	//@m_running false => shape isn't created yet, we create a new shape
	if (m_running == false && event -> button() == Qt::LeftButton) {
		m_shape_item = new QetShapeItem(pos, pos, m_shape_type);
		m_dv -> setContextMenuPolicy (Qt::NoContextMenu);
		m_diagram -> addItem (m_shape_item);
		m_running = true;
		return true;
	}

	//At this point m_shape_item must be created
	if (!m_shape_item) return false;

	// Next left click finish all shape item except the polyline
	if (m_shape_type != QetShapeItem::Polyline && event->button() == Qt::LeftButton) {
		m_shape_item -> setP2 (pos);
		m_diagram -> undoStack().push (new AddItemCommand<QetShapeItem *> (m_shape_item, m_diagram));
		m_dv -> setContextMenuPolicy(Qt::DefaultContextMenu);
		m_running = false;
		return true;
	}

	// Next left click create new segment for polyline
	if (m_shape_type == QetShapeItem::Polyline && event -> button() == Qt::LeftButton) {
		m_shape_item -> setNextPoint (Diagram::snapToGrid(pos)); //< this point is ok for pos
		m_shape_item -> setNextPoint (Diagram::snapToGrid(pos)); //< Add new point for next segment. the pos of this point
																 //< can be changed by calling QetShapItem::setP2()
		return true;
	}

	// If shape item is polyline and click is right button, the shape item is finish
	// m_running is set to false at the release of right button.
	if (m_shape_type == QetShapeItem::Polyline && event -> button() == Qt::RightButton) {
		m_shape_item -> setP2 (pos);
		m_diagram -> undoStack().push (new AddItemCommand<QetShapeItem *> (m_shape_item, m_diagram));
		return true;
	}

	return false;
}

/**
 * @brief DVEventAddShape::mouseMoveEvent
 * Action when mouse move
 * @param event : event of mouse move
 * @return : true if this event is managed, otherwise false
 */
bool DVEventAddShape::mouseMoveEvent(QMouseEvent *event) {
	updateHelpCross(event->pos());
	if (!m_running) return false;
	if (m_shape_item && event -> buttons() == Qt::NoButton) {
		m_shape_item -> setP2 (m_dv -> mapToScene (event -> pos()));
		return true;
	}
	return false;
}

/**
 * @brief DVEventAddShape::mouseReleaseEvent
 * Action when mouse button is released
 * @param event : event of mouse release
 * @return : true if this event is managed, otherwise false
 */
bool DVEventAddShape::mouseReleaseEvent(QMouseEvent *event) {
	//When the shape is polyline, we set default context menu to diagram view
	//only when the right button is released
	if (m_shape_type == QetShapeItem::Polyline && event -> button() == Qt::RightButton ) {
		m_dv -> setContextMenuPolicy(Qt::DefaultContextMenu);
		m_running = false;
		return true;
	}
	return false;
}

/**
 * @brief DVEventAddShape::updateHelpCross
 * Create and update the position of the cross to help user for draw new shape
 * @param event
 */
void DVEventAddShape::updateHelpCross(const QPoint &p) {
	//If line isn't created yet, we create it.
	if (!m_help_horiz || !m_help_verti) {
		QPen pen;
		pen.setColor(Qt::yellow);
		//Add +5 for each line, because the topleft of diagram isn't draw at position (0:0) but (5:5)
		if (!m_help_horiz) {
			m_help_horiz = new QGraphicsLineItem(m_diagram -> border_and_titleblock.rowsHeaderWidth() + 5, 0,
												 m_diagram -> border_and_titleblock.diagramWidth() + 5, 0,
												 0, m_diagram);
			m_help_horiz->setPen(pen);
		}
		if (!m_help_verti) {
			m_help_verti = new QGraphicsLineItem(0, m_diagram -> border_and_titleblock.columnsHeaderHeight() + 5,
												 0, m_diagram -> border_and_titleblock.diagramHeight() + 5,
												 0, m_diagram);
			m_help_verti->setPen(pen);
		}
	}

	//Update the position of the cross
	QPointF point = Diagram::snapToGrid(m_dv->mapToScene(p));

	m_help_horiz->setY(point.y());
	m_help_verti->setX(point.x());
}