167 lines
4.5 KiB
C++
167 lines
4.5 KiB
C++
|
|
|
|
#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>
|
|
#include <QPaintEvent>
|
|
#include <QPainter>
|
|
#include <QCoreApplication>
|
|
#include <QCursor>
|
|
#include <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));
|
|
break;
|
|
|
|
case QEvent::MouseMove:
|
|
mouseMoveEvent(widget, static_cast<QMouseEvent *>(e));
|
|
break;
|
|
|
|
case QEvent::MouseButtonRelease:
|
|
mouseReleaseEvent(widget, static_cast<QMouseEvent *>(e));
|
|
};
|
|
}
|
|
|
|
//------------------------------------------------------------------
|
|
|
|
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();
|
|
|
|
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");
|