helpers.py

from PyQt5.QtGui import QColor, QBrush
from PyQt5.QtCore import *
from qgis.gui import *
from qgis.utils import *
from qgis.core import *

from .globals import AOI_COORDINATES
import datetime

class CaptureCoordinates(QgsMapToolEmitPoint):
    canvasDoubleClicked = QtCore.pyqtSignal(object, object)

    def __init__(self, canvas, textEdit, layer=None, source_crs=None, destination_crs=None):
        QgsMapToolEmitPoint.__init__(self, canvas)
        self.canvas = canvas
        self.textEdit = textEdit
        self.layer = layer
        self.setCursor(Qt.CrossCursor)
        self.source_crs = source_crs
        self.destination_crs = destination_crs
        self.rubberBand = QgsRubberBand(self.canvas, QgsWkbTypes.geometryType(QgsWkbTypes.Polygon))
        self.rubberBand.setColor(QColor("transparent"))
        self.rubberBand.setWidth(2)
        # self.rubberBand.setIcon(QgsRubberBand.ICON_FULL_DIAMOND)
        # self.rubberBand.setIconSize(75)
        # self.rubberBand.setBrushStyle(QBrush(Qt.FDiagPattern))
        self.rubberBand.setStrokeColor(Qt.red)
        self.reset()

    def reset(self):
        self.rubberBand.reset(QgsWkbTypes.geometryType(QgsWkbTypes.Polygon))


    def transformCRS(self, source_crs, destination_crs):
        return QgsCoordinateTransform(QgsCoordinateReferenceSystem(source_crs),
                    QgsCoordinateReferenceSystem(destination_crs), 
                    QgsProject.instance())


    def canvasPressEvent(self, event):
        if event.button() == Qt.LeftButton:
            transform = self.transformCRS(self.source_crs, self.destination_crs)
            point = self.toLayerCoordinates(self.layer, event.pos())
            self.rubberBand.addPoint(point,True)
            self.rubberBand.show()
            tr_point = transform.transform(point)
            AOI_COORDINATES.append([round(tr_point.x(), 7), round(tr_point.y(), 7)])
            self.textEdit.appendPlainText("["+str(datetime.datetime.now().strftime ("%H:%M:%S")) + "] " +"Добавлена точка с координатами: "+ str(round(tr_point.x(), 3)) + ", " + str(round(tr_point.y(), 3)))
        elif event.button() == Qt.RightButton:
            self.reset()
            AOI_COORDINATES.clear()


    def canvasDoubleClickEvent(self, event):
        if not self.rubberBand.asGeometry().isGeosValid():
            self.textEdit.appendPlainText("["+str(datetime.datetime.now().strftime ("%H:%M:%S")) + "] " +" ВНИМАНИЕ: Не валидный полигон! Дальнейший поиск будет осуществляться без учета выбранной территории")
        
        self.canvas.unsetMapTool(self)


    def addCoordinates(self, x, y):
        point = QgsPointXY(x,y)
        AOI_COORDINATES.append([round(x,7), round(y,7)])
        transform = self.transformCRS(self.destination_crs, self.source_crs)
        tr_point = transform.transform(point)
        self.rubberBand.addPoint(tr_point, True)
        self.textEdit.appendPlainText("["+str(datetime.datetime.now().strftime ("%H:%M:%S")) + "] " +"Добавлена точка с координатами: "+ str(point.x()) + "," + str(point.y()))
        # self.textEdit.appendPlainText("["+str(datetime.datetime.now().strftime ("%H:%M:%S")) + "] " +"Координаты полигона: "+ str(AOI_COORDINATES))
        self.rubberBand.show()

    
    def cancelCoordinates(self):
        self.reset()
        AOI_COORDINATES.clear()