geoscience/SectionMapTool.py at master · rolandhill/geoscience · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# -*- coding: utf-8 -*-
"""
Created on Thu May  2 17:56:42 2019

@author: HillR
"""
from qgis.gui import QgsMapToolEmitPoint, QgsRubberBand, QgsMessageBar
from qgis.core import Qgis, QgsWkbTypes
from qgis.utils import iface

from qgis.PyQt import QtCore
from qgis.PyQt.QtGui import QColorConstants

import numpy as np

from .Utils import *
from .sectionMapCanvas_dialog import SectionMapCanvasDialog


class SectionMapTool(QgsMapToolEmitPoint):
    def __init__(self, canvas, parent):
        self.canvas = canvas
        self.sectionManagerDlg = parent
        self.drillManager = self.sectionManagerDlg.drillManager
        QgsMapToolEmitPoint.__init__(self, self.canvas)
        self.rubberBand = QgsRubberBand(self.canvas)
        self.rubberBand.setColor(QColorConstants.Red)
        self.rubberBand.setWidth(1)

        self.reset()

    def reset(self):
        self.startPoint = self.endPoint = None
        self.isEmittingPoint = False
        self.rubberBand.reset(QgsWkbTypes.LineGeometry)

    def canvasPressEvent(self, e):
        self.startPoint = self.toMapCoordinates(e.pos())
        self.endPoint = self.startPoint
        self.isEmittingPoint = True
        self.showLine(self.startPoint, self.endPoint)

    def canvasReleaseEvent(self, e):
        self.endPoint = self.toMapCoordinates(e.pos())
        self.isEmittingPoint = False

        # Get the layers to incude in the section from a dlg
        dlg = SectionMapCanvasDialog()
        dlg.leName.setText(self.suggestName())
        result = dlg.exec()
        if result:
            self.drillManager.sectionName = dlg.leName.text()
#            self.drillManager.sectionWidth = float(dlg.leSectionWidth.text())
            # Save the name of each checked attribute field in a list
            self.drillManager.sectionLayers = getCheckedLayers(dlg.listLayers)
            self.drillManager.elevationLayers = getCheckedLayers(dlg.listElevation)

        dlg.close()

        if result:
            s = self.drillManager.sectionManager.createSection(self.drillManager.sectionName, \
               self.startPoint.x(), self.startPoint.y(), self.endPoint.x(), self.endPoint.y(), \
               self.drillManager.sectionWidth, self.drillManager.sectionLayers, self.drillManager.elevationLayers)

            self.sectionManagerDlg.fillSectionList()

            self.sectionManagerDlg.sectionManager.showSection(s)

        self.rubberBand.reset(QgsWkbTypes.LineGeometry)
        iface.mapCanvas().setMapTool(self.oldMapTool)

    def canvasMoveEvent(self, e):
        if not self.isEmittingPoint:
            return

        self.endPoint = self.toMapCoordinates(e.pos())
        self.showLine(self.startPoint, self.endPoint)

    def suggestName(self):
        dir = 'N'
        if abs(self.endPoint.y() - self.startPoint.y()) > abs(self.endPoint.x() - self.startPoint.x()):
            dir = 'E'

        name = "00"
        if dir == 'N':
            name = "{:d}".format(int((self.endPoint.y() + self.startPoint.y())/2.0)) + dir
        else:
            name = "{:d}".format(int((self.endPoint.x() + self.startPoint.x())/2.0)) + dir

        return name

    def showLine(self, startPoint, endPoint):
        if startPoint == endPoint:
            return
        start = np.array([startPoint.x(), startPoint.y(), 0.0])
        end = np.array([endPoint.x(), endPoint.y(), 0.0])
        v = end - start
        u = v / np.linalg.norm(v)
        n = np.cross(np.array([0.0, 0.0, 1.0]), u)
        width = self.drillManager.sectionWidth
        nw = n * (width*0.5)
        qnw = QgsVector(nw[0], nw[1])
        p0 = endPoint + qnw
        p1 = startPoint + qnw
        p2 = startPoint - qnw
        p3 = endPoint - qnw
        self.rubberBand.reset(QgsWkbTypes.LineGeometry)
        self.rubberBand.addPoint(startPoint, False)
        self.rubberBand.addPoint(endPoint, False)  # true to update canvas
        self.rubberBand.addPoint(p0, False)
        self.rubberBand.addPoint(p1, False)
        self.rubberBand.addPoint(p2, False)
        self.rubberBand.addPoint(p3, False)
        self.rubberBand.addPoint(endPoint, True)  # true to update canvas
        self.rubberBand.show()

#    def rectangle(self):
#        if self.startPoint is None or self.endPoint is None:
#            return None
#        elif self.startPoint.x() == self.endPoint.x() or self.startPoint.y() == self.endPoint.y():
#            return None
#
#        return QgsRectangle(self.startPoint, self.endPoint)