-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
3 changed files
with
105 additions
and
118 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
81 changes: 81 additions & 0 deletions
81
los_tools/gui/create_los_tool/los_digitizing_tool_with_widget.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,81 @@ | ||
| from qgis.core import Qgis | ||
| from qgis.gui import QgisInterface, QgsMapToolAdvancedDigitizing, QgsSnapIndicator | ||
| from qgis.PyQt.QtCore import Qt | ||
| from qgis.PyQt.QtGui import QKeyEvent | ||
| from qgis.PyQt.QtWidgets import QWidget | ||
|
|
||
|
|
||
| class LoSDigitizingToolWithWidget(QgsMapToolAdvancedDigitizing): | ||
|
|
||
| _widget: QWidget | ||
|
|
||
| def __init__(self, iface: QgisInterface) -> None: | ||
| super().__init__(iface.mapCanvas(), iface.cadDockWidget()) | ||
|
|
||
| self._iface = iface | ||
| self._canvas = self._iface.mapCanvas() | ||
|
|
||
| self._snapper = self._canvas.snappingUtils() | ||
| self.snap_marker = QgsSnapIndicator(self._canvas) | ||
|
|
||
| self._los_rubber_band = self.createRubberBand(Qgis.GeometryType.Line) | ||
|
|
||
| def activate(self) -> None: | ||
| super().activate() | ||
|
|
||
| self.create_widget() | ||
|
|
||
| self.messageDiscarded.emit() | ||
|
|
||
| self._canvas = self._iface.mapCanvas() | ||
| self._snapper = self._canvas.snappingUtils() | ||
|
|
||
| if self._canvas.mapSettings().destinationCrs().isGeographic(): | ||
| self.messageEmitted.emit( | ||
| "Tool only works if canvas is in projected CRS. Currently canvas is in geographic CRS.", | ||
| Qgis.MessageLevel.Critical, | ||
| ) | ||
| self.deactivate() | ||
| return | ||
|
|
||
| def deactivate(self) -> None: | ||
| self.clean() | ||
| self.delete_widget() | ||
| self._iface.mapCanvas().unsetMapTool(self) | ||
| super().deactivate() | ||
|
|
||
| def create_widget(self): | ||
| self.release_keyboard() | ||
| self._iface.addUserInputWidget(self._widget) | ||
| self._widget.setFocus(Qt.TabFocusReason) | ||
| self._widget.show() | ||
|
|
||
| def release_keyboard(self): | ||
| if self._widget: | ||
| self._widget.releaseKeyboard() | ||
|
|
||
| def delete_widget(self): | ||
| if self._widget: | ||
| self.release_keyboard() | ||
| self._widget.deleteLater() | ||
| self._widget = None | ||
|
|
||
| def clean(self) -> None: | ||
| self.snap_marker.setVisible(False) | ||
| self._los_rubber_band.hide() | ||
|
|
||
| def keyPressEvent(self, e: QKeyEvent) -> None: | ||
| if e.key() == Qt.Key_Escape: | ||
| self.deactivate() | ||
| self._iface.mapCanvas().unsetMapTool(self) | ||
| return super().keyPressEvent(e) | ||
|
|
||
| def canvas_crs_is_projected(self) -> bool: | ||
| if self._canvas.mapSettings().destinationCrs().isGeographic(): | ||
| self._iface.messageBar().pushMessage( | ||
| "LoS can be drawn only for projected CRS. Canvas is currently in geographic CRS.", | ||
| Qgis.Critical, | ||
| duration=5, | ||
| ) | ||
| return False | ||
| return True |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters