// Copyright (C) 2008  Lukas Lalinsky
//
// 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 2 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, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

#include <cmath>
#include <QMouseEvent>
#include <QScrollBar>
#include <QDebug>
#include "diagramview.h"
#include "diagramdocument.h"

using namespace std;

DiagramView::DiagramView(QWidget *parent)
	: QGraphicsView(parent), m_handScrolling(false)
{
	setAlignment(Qt::AlignLeft | Qt::AlignTop);
	setRenderHint(QPainter::Antialiasing);
	setDragMode(QGraphicsView::RubberBandDrag);
	setRubberBandSelectionMode(Qt::ContainsItemBoundingRect);
}

DiagramDocument *
DiagramView::document() const
{
	return static_cast<DiagramDocument *>(scene());
}

void
DiagramView::setScene(QGraphicsScene *scene)
{
	QGraphicsView::setScene(scene);
	updateSceneRect2(scene->sceneRect());
	connect(scene, SIGNAL(sceneRectChanged(const QRectF &)), SLOT(updateSceneRect2(const QRectF &)));
}

void
DiagramView::updateSceneRect2(const QRectF &rect)
{
	QRectF sceneRect = rect.united(QRectF(QPointF(0, 0), QPointF(100, 100)));
	setSceneRect(sceneRect);
}

void
DiagramView::mousePressEvent(QMouseEvent *event)
{
	if (event->button() == Qt::MidButton) {
		m_handScrolling = true;
        m_handScrollingOrigin = QPoint(event->pos());
		m_handScrollingValueX = horizontalScrollBar()->value();
        m_handScrollingValueY = verticalScrollBar()->value();
		m_savedCursor = viewport()->cursor();
		viewport()->setCursor(Qt::ClosedHandCursor);
		event->accept();
		return;
	}
	QGraphicsView::mousePressEvent(event);
}

void
DiagramView::mouseMoveEvent(QMouseEvent *event)
{
	if (m_handScrolling) {
		QPoint delta = event->pos() - m_handScrollingOrigin;
		horizontalScrollBar()->setValue(m_handScrollingValueX - delta.x());
		verticalScrollBar()->setValue(m_handScrollingValueY - delta.y());
		event->accept();
		return;
	}
	QGraphicsView::mouseMoveEvent(event);
}

void
DiagramView::mouseReleaseEvent(QMouseEvent *event)
{
	if (m_handScrolling) {
		m_handScrolling = false;
		viewport()->setCursor(m_savedCursor);
		event->accept();
		return;
	}
	QGraphicsView::mouseReleaseEvent(event);
}

void
DiagramView::drawBackground(QPainter *painter, const QRectF &rect)
{
	DiagramDocument *doc = document();
	if (doc && doc->isGridVisible()) {
		drawGrid(painter, rect);
	}
}

void
DiagramView::drawGrid(QPainter *painter, const QRectF &rect)
{
	const int pointBufferSize = 5000;
	static QPoint pointBuffer[pointBufferSize];
	DiagramDocument *doc = document();
	int gridSize = doc->gridSize();
	int x0 = gridSize * floor(rect.left() / gridSize);
	int y0 = gridSize * floor(rect.top() / gridSize);
	int x1 = gridSize * ceil(rect.right() / gridSize);
	int y1 = gridSize * ceil(rect.bottom() / gridSize);
	painter->save();
	QPen pen;
	pen.setColor(doc->gridColor());
	pen.setWidth(0);
	painter->setPen(pen);
	painter->setRenderHint(QPainter::Antialiasing, false);
	int pointsUsed = 0;
	for (int x = x0; x < x1; x += gridSize) {
		for (int y = y0; y < y1; y += gridSize) {
			pointBuffer[pointsUsed].setX(x);
			pointBuffer[pointsUsed].setY(y);
			pointsUsed++;
			if (pointsUsed == pointBufferSize) {
				painter->drawPoints(pointBuffer, pointsUsed);
				pointsUsed = 0;
			}
		}
	}
	if (pointsUsed > 0) {
		painter->drawPoints(pointBuffer, pointsUsed);
		pointsUsed = 0;
	}
	painter->restore();
}