-
Notifications
You must be signed in to change notification settings - Fork 18
/
visibility_provider.py
124 lines (91 loc) · 4.17 KB
/
visibility_provider.py
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
124
# -*- coding: utf-8 -*-
"""
/***************************************************************************
Visibility analysis for QGIS (**to be installed in ViewshedAnalysis folder**)
-------------------
begin : 2018-03-18
copyright : (C) 2018 by Z. Cuckovic
homepage : https://zoran-cuckovic.github.io/QGIS-visibility-analysis/
***************************************************************************/
/***************************************************************************
* *
* 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__ = 'AUthor'
__date__ = '2018-03-18'
__copyright__ = '(C) 2018 by AUthor'
# This will get replaced with a git SHA1 when you do a git archive
__revision__ = '$Format:%H$'
from qgis.core import QgsProcessingProvider
from processing.core.ProcessingConfig import Setting, ProcessingConfig
from PyQt5.QtGui import QIcon
from os import path
from .algorithms.viewshed_points import ViewshedPoints
from .algorithms.viewshed_raster import ViewshedRaster
from .algorithms.viewshed_intervisibility import Intervisibility
from .algorithms.viewshed_index import VisibilityIndex
class VisibilityProvider(QgsProcessingProvider):
def __init__(self):
super().__init__()
def load(self):
ProcessingConfig.settingIcons[self.name()] = self.icon()
# Activate provider by default
ProcessingConfig.addSetting(
Setting(self.name(), 'VISIBILITY_ANALYSIS_ACTIVATED',
'Activate', True))
ProcessingConfig.addSetting(
Setting(self.name(), 'MEMORY_BUFFER_SIZE',
'Memory buffer size (mega-pixels)', 100))
ProcessingConfig.readSettings()
self.refreshAlgorithms()
return True
def unload(self):
"""
Unloads the provider. Any tear-down steps required by the provider
should be implemented here.
"""
ProcessingConfig.removeSetting('VISIBILITY_ANALYSIS_ACTIVATED')
def loadAlgorithms(self):
"""
Loads all algorithms belonging to this provider.
"""
for alg in [ViewshedPoints(), ViewshedRaster(),
Intervisibility() , VisibilityIndex()]: self.addAlgorithm( alg )
def isActive(self):
"""Return True if the provider is activated and ready to run algorithms"""
return ProcessingConfig.getSetting('VISIBILITY_ANALYSIS_ACTIVATED')
def setActive(self, active):
ProcessingConfig.setSettingValue('VISIBILITY_ANALYSIS_ACTIVATED', active)
def id(self):
"""
Returns the unique provider id, used for identifying the provider. This
string should be a unique, short, character only string, eg "qgis" or
"gdal". This string should not be localised.
"""
return 'visibility'
def name(self):
"""
Returns the provider name, which is used to describe the provider
within the GUI.
This string should be short (e.g. "Lastools") and localised.
"""
return 'Visibility analysis'
def longName(self):
"""
Returns the a longer version of the provider name, which can include
extra details such as version numbers. E.g. "Lastools LIDAR tools
(version 2.2.1)". This string should be localised. The default
implementation returns the same string as name().
"""
return self.name()
def icon(self):
"""
We return the default icon.
QgsProcessingProvider.icon(self)
"""
return QIcon(path.dirname(__file__) + '/icon.png')