FloodRiskSwatPlus/FloodRiskSwatPlus.py at main · icra/FloodRiskSwatPlus · 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
# -*- coding: utf-8 -*-

"""
/***************************************************************************
 FloodRiskSwatPlus
                                 A QGIS plugin
 FloodRiskSwatPlus
 Generated by Plugin Builder: http://g-sherman.github.io/Qgis-Plugin-Builder/
                              -------------------
        begin                : 2025-01-30
        copyright            : (C) 2025 by ICRA - Oliu Llorente
        email                : ollorente@icra.cat
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
"""

__author__ = 'ICRA - Oliu Llorente'
__date__ = '2025-01-30'
__copyright__ = '(C) 2025 by ICRA - Oliu Llorente'

# This will get replaced with a git SHA1 when you do a git archive

__revision__ = '$Format:%H$'

import os
import sys
import inspect

from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from qgis.core import *
from qgis.core import QgsProcessingAlgorithm, QgsApplication
from .FloodRiskSwatPlus_provider import FloodRiskSwatPlusProvider

from processing.gui.AlgorithmDialog import AlgorithmDialog
from qgis.core import QgsApplication, QgsProcessingFeedback, QgsProcessingParameters

from .allProcess.DamageUnits import DamageUnitsLayer as damageUnits
from .allProcess.FloodRisk import FloodRiskMitigation as FloodRisk


cmd_folder = os.path.split(inspect.getfile(inspect.currentframe()))[0]

if cmd_folder not in sys.path:
    sys.path.insert(0, cmd_folder)


class FloodRiskSwatPlusPlugin(object):

    def __init__(self, iface):

        self.provider = None
        self.iface = iface
        self.provider = FloodRiskSwatPlusProvider()

    def initProcessing(self):
        """Init Processing provider for QGIS >= 3.8."""
        self.provider = FloodRiskSwatPlusProvider()
        QgsApplication.processingRegistry().addProvider(self.provider)

    def DamageUnits(self):
        servei = damageUnits()
        servei.initAlgorithm()
        servei.setProvider(self.provider)
        dlg = AlgorithmDialog(servei, parent=self.iface.mainWindow())
        dlg.show()
        # do NOT remove!!!! if you do then sip forgets the python subclass of AlgorithmDialog and you get a broken
        # dialog
        dlg.exec_()

    def FloodRiskMitigation(self):
        servei = FloodRisk()
        servei.initAlgorithm()
        servei.setProvider(self.provider)
        dlg = AlgorithmDialog(servei, parent=self.iface.mainWindow())
        dlg.show()
        # do NOT remove!!!! if you do then sip forgets the python subclass of AlgorithmDialog and you get a broken
        # dialog
        dlg.exec_()

    def initGui(self):
        self.menu = QMenu("FloodRiskSwat+", self.iface.mainWindow().menuBar())
        self.action_damageUnits = QAction("Damage units", self.iface.mainWindow())
        self.mitigation = QAction("Risk Mitigation", self.iface.mainWindow())

        actions = self.iface.mainWindow().menuBar().actions()
        lastAction = actions[-1]
        self.iface.mainWindow().menuBar().insertMenu(lastAction, self.menu)
        self.menu.addAction(self.action_damageUnits)
        self.menu.addAction(self.mitigation)

        self.action_damageUnits.triggered.connect(self.DamageUnits)
        self.mitigation.triggered.connect(self.FloodRiskMitigation)
        self.initProcessing()

    def unload(self):
        QgsApplication.processingRegistry().removeProvider(self.provider)