#include "tcommon.h"

#include "toonzqt/pickrgbutils.h"

#include "toonzqt/screenboard.h"

#include "toonzqt/menubarcommand.h"

#include "tools/RGBpicker.h"

#include "tools/cursors.h"

#include "tools/cursormanager.h"

#include <qmouseevent></qmouseevent>

#include <qpaintevent></qpaintevent>

#include <qpainter></qpainter>

#include <qcoreapplication></qcoreapplication>

#include <qcursor></qcursor>

#include <qtimer></qtimer>

#include "tools/screenpicker.h"

//***********************************************************************************

//    Screen Picker implementation

//***********************************************************************************

ScreenPicker::ScreenPicker(QWidget *parent)

	: m_mousePressed(false), m_mouseGrabbed(false)

{

}

//------------------------------------------------------------------

bool ScreenPicker::acceptScreenEvents(const QRect &screenRect) const

{

	return screenRect.contains(QCursor::pos()) || //Accept mouse events

		   screenRect.intersects(m_geometry);	 //Accept paint events

}

//------------------------------------------------------------------

void ScreenPicker::event(QWidget *widget, QEvent *e)

{

	switch (e->type()) {

	case QEvent::MouseButtonPress:

		mousePressEvent(widget, static_cast<qmouseevent *="">(e));</qmouseevent>

		break;

	case QEvent::MouseMove:

		mouseMoveEvent(widget, static_cast<qmouseevent *="">(e));</qmouseevent>

		break;

	case QEvent::MouseButtonRelease:

		mouseReleaseEvent(widget, static_cast<qmouseevent *="">(e));</qmouseevent>

	};

}

//------------------------------------------------------------------

void ScreenPicker::mousePressEvent(QWidget *widget, QMouseEvent *me)

{

	m_mousePressed = true;

	m_start = widget->mapToGlobal(me->pos());

	m_geometry = QRect(m_start, QSize(1, 1));

	DVGui::ScreenBoard::instance()->update();

}

//------------------------------------------------------------------

void ScreenPicker::mouseMoveEvent(QWidget *widget, QMouseEvent *me)

{

	assert(m_mouseGrabbed);

	if (!m_mousePressed)

		return;

	QPoint pos(widget->mapToGlobal(me->pos()));

	m_geometry = QRect(QRect(m_start, QSize(1, 1)) | QRect(pos, QSize(1, 1)));

	DVGui::ScreenBoard::instance()->update();

}

//------------------------------------------------------------------

void ScreenPicker::mouseReleaseEvent(QWidget *widget, QMouseEvent *me)

{

	assert(m_mouseGrabbed);

	if (!m_mousePressed)

		return;

	m_mouseGrabbed = m_mousePressed = false;

	DVGui::ScreenBoard *screenBoard = DVGui::ScreenBoard::instance();

	QList<dvgui::screenboard::drawing *=""> &drawings = screenBoard->drawings();</dvgui::screenboard::drawing>

	drawings.removeAt(drawings.indexOf(this));

	screenBoard->releaseMouse();

	screenBoard->update();

	QPoint pos(widget->mapToGlobal(me->pos()));

	m_geometry = QRect(QRect(m_start, QSize(1, 1)) | QRect(pos, QSize(1, 1)));

	//TimerEvents execution is delayed until all other events have been processed.

	//In particular, we want to pick after the screen refreshes

	QTimer::singleShot(0, this, SLOT(pick()));

}

//------------------------------------------------------------------

void ScreenPicker::pick()

{

	//Someway, on MACOSX 10.7 (Lion) there may be screen refresh events pending at this point.

	//Process them before picking.

	QCoreApplication::processEvents();

	QColor color(pickScreenRGB(m_geometry));

	RGBPicker::setCurrentColorWithUndo(TPixel32(color.red(), color.green(), color.blue()));

	m_geometry = QRect();

}

//------------------------------------------------------------------

void ScreenPicker::paintEvent(QWidget *widget, QPaintEvent *pe)

{

	if (!m_mousePressed)

		return;

	QPainter painter(widget);

	QRect relativeGeom(

		widget->mapFromGlobal(m_geometry.topLeft()),

		widget->mapFromGlobal(m_geometry.bottomRight()));

	painter.setPen(QColor(0, 0, 255, 128));

	painter.setBrush(QColor(0, 0, 255, 64));

	painter.drawRect(relativeGeom);

}

//------------------------------------------------------------------

void ScreenPicker::startGrab()

{

	if (!m_mouseGrabbed) {

		m_mouseGrabbed = true;

		DVGui::ScreenBoard *screenBoard = DVGui::ScreenBoard::instance();

		screenBoard->drawings().push_back(this);

		screenBoard->grabMouse(getToolCursor(ToolCursor::PickerCursor));

		screenBoard->update();

	}

}

//***********************************************************************************

//    Pick Screen Command instantiation

//***********************************************************************************

class PickScreenCommandHandler : public MenuItemHandler

{

public:

	PickScreenCommandHandler(CommandId cmdId) : MenuItemHandler(cmdId) {}

	void execute()

	{

		static ScreenPicker *picker = new ScreenPicker;

		picker->startGrab();

	}

} pickScreenCHInstance("A_ToolOption_PickScreen");