Merge pull request #6 from buttleofx/ScreenPicker

Screen picker

Author: MaximeEngel

example/screenPicker/main.py

@@ -0,0 +1,24 @@
+import sys
+from PyQt5 import QtCore, QtWidgets, QtQuick
+#Fix execution shader bug
+from OpenGL import GL
+
+
+import os
+
+currentFilePath = os.path.dirname(os.path.abspath(__file__))
+quickmambaPath = os.path.join(currentFilePath, '../QuickMamba')
+sys.path.append(quickmambaPath)
+
+import quickmamba
+
+if __name__ == '__main__':
+    quickmamba.qmlRegister()
+    app = QtWidgets.QApplication(sys.argv)
+    view = QtQuick.QQuickView()
+    view.setResizeMode(QtQuick.QQuickView.SizeRootObjectToView)
+    # Create a declarative view
+    view.setSource(QtCore.QUrl("source.qml"))
+
+    view.show()
+    app.exec_()

example/screenPicker/source.qml

@@ -0,0 +1,32 @@
+import QtQuick 2.0
+import QtQuick.Controls 1.1
+import "../../qml/QuickMamba/." // Qt-BUG import qmldir
+import QtQuick.Layouts 1.1
+
+Rectangle
+{
+    id: root
+    color: "black"
+    width: 300
+    height: 250
+
+    RowLayout {
+        anchors.centerIn: parent
+        spacing: 20
+
+        ScreenPicker {
+            id:screenPicker
+
+            onGrabbedColor: colorPreview.color = color
+        }
+
+        Rectangle {
+            id:colorPreview
+            color:"#00b2a1"
+
+            radius: 3
+            Layout.minimumWidth: 80
+            Layout.minimumHeight: 80
+        }
+    }
+}

qml/QuickMamba/ScreenPicker.qml

@@ -0,0 +1,43 @@
+import QtQuick 2.0
+import QuickMamba 1.0
+
+Rectangle
+{
+    id:root
+    property string pickerImg: "assets/img/screenpicker/screenPicker.png"
+    property string pickerImgHover: "assets/img/screenpicker/screenPickerHover.png"
+    // Default style
+    color: "#212121"
+    border.width: 2
+    border.color: "#333"
+    radius: 3
+    height: pickerImg.height * 2
+    width: pickerImg.width * 2
+
+    signal accepted
+    signal grabbedColor(var color)
+
+    Image {
+        id: pickerImg
+        source: root.pickerImg
+        anchors.centerIn: parent
+    }
+
+    MouseArea {
+        anchors.fill: root
+        hoverEnabled: true
+
+        onEntered: pickerImg.source = root.pickerImgHover
+        onExited: pickerImg.source = root.pickerImg
+
+        onPressed: screenPicker.grabbing = true
+        cursorShape: Qt.OpenHandCursor
+    }
+
+    ColorPicker {
+        id: screenPicker
+
+        onAccepted: root.accepted()
+        onCurrentColorChanged: root.grabbedColor(currentColor)
+    }
+}

qml/QuickMamba/assets/img/screenpicker/screenPicker.png

@@ -5,3 +5,4 @@ WheelArea 1.0 WheelArea.qml
 ExternDropArea 1.0 ExternDropArea.qml
 QuickEditableNumberInput 1.0 QuickEditableNumberInput.qml
 BlinkCursor 1.0 BlinkCursor.qml
+ScreenPicker 1.0 ScreenPicker.qml

qml/QuickMamba/assets/img/screenpicker/screenPickerHover.png

@@ -1,12 +1,13 @@
 from .wheelArea import WheelArea
 from .externDropArea import ExternDropArea
 from .colorExtended import ColorExtended
+from .colorPicker import ColorPicker
 
 from PyQt5 import QtQml
 
 def qmlRegister():
     QtQml.qmlRegisterType(WheelArea, "QuickMamba", 1, 0, "WheelAreaImpl")
     QtQml.qmlRegisterType(ExternDropArea, "QuickMamba", 1, 0, "ExternDropAreaImpl")
     QtQml.qmlRegisterType(ColorExtended, "QuickMamba", 1, 0, "ColorExtended")
-
+    QtQml.qmlRegisterType(ColorPicker, "QuickMamba", 1, 0, "ColorPicker")
 

qml/QuickMamba/qmldir

@@ -0,0 +1,78 @@
+from PyQt5 import QtCore
+from PyQt5 import QtWidgets
+from PyQt5 import QtQuick
+from PyQt5.QtGui import QCursor, QColor, QGuiApplication
+
+class ColorPickingEventFilter(QtCore.QObject):
+    def __init__(self, screenpicker):
+        QtCore.QObject.__init__(self, screenpicker)
+        self._screenpicker = screenpicker
+
+    def eventFilter(self, obj, event):
+        if(event.type() == QtCore.QEvent.MouseMove):
+            self._screenpicker.updateCurrentColor()
+        elif(event.type() == QtCore.QEvent.MouseButtonRelease):
+            self._screenpicker.setGrabbing(False)
+            self._screenpicker.accepted.emit()
+
+        return False
+
+
+class ColorPicker(QtQuick.QQuickItem):
+    """
+        Define the common methods and fields for screenPicker.
+    """
+
+    def __init__(self, parent = None):
+        QtQuick.QQuickItem.__init__(self, parent)
+        self._currentColor = QColor("#FFFFFF")
+        self._grabbing = False
+        self._desktop = QtWidgets.QDesktopWidget()
+        self._colorPickingEventFilter = ColorPickingEventFilter(self)
+
+    # ######################################## Methods private to this class ####################################### #
+
+    # ## Getters ## #
+
+    def getCurrentColor(self):
+        return self._currentColor
+
+    def setCurrentColor(self, currentColor):
+        self._currentColor = currentColor
+        self.currentColorChanged.emit()
+
+    def isGrabbing(self):
+        return self._grabbing
+
+    def setGrabbing(self, grabbing):
+        if(self._grabbing != grabbing):
+            self._grabbing = grabbing
+            if(self._grabbing):
+                self.installEventFilter(self._colorPickingEventFilter)
+                QGuiApplication.setOverrideCursor(QCursor(QtCore.Qt.CrossCursor))
+                self.grabMouse()
+            else:
+                self.ungrabMouse()
+                self.removeEventFilter(self._colorPickingEventFilter)
+                QGuiApplication.restoreOverrideCursor()
+            self.grabbingChanged.emit()
+
+    # ## Others ## #
+
+    def updateCurrentColor(self):
+        cursorPos = QCursor.pos()
+        # Catch the pixel pointed by the mouse on a pixmap
+        pixmap = QGuiApplication.screens()[self._desktop.screenNumber()].grabWindow(self._desktop.winId(), cursorPos.x(), cursorPos.y(), 1, 1)
+        qImage = pixmap.toImage()
+        qColor = QColor(qImage.pixel(0, 0))
+        self.setCurrentColor(qColor)
+
+# ############################################# Data exposed to QML ############################################## #
+
+    accepted = QtCore.pyqtSignal()
+
+    currentColorChanged = QtCore.pyqtSignal()
+    currentColor = QtCore.pyqtProperty(QColor, getCurrentColor, setCurrentColor, notify=currentColorChanged)
+
+    grabbingChanged = QtCore.pyqtSignal()
+    grabbing = QtCore.pyqtProperty(bool, isGrabbing, setGrabbing, notify=grabbingChanged)