def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByCategory, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.categories = []
self.selectedClasses = []
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0, 0, 0, 0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0, 0, 1, 1)
#Objects Connections
self.widget.connectionChanged.connect(self.listCategoriesFromDatabase)
self.widget.problemOccurred.connect(self.pushMessage)
self.pushButtonCancel.clicked.connect(self.cancel)
self.pushButtonOk.clicked.connect(self.okSelected)
self.pushButtonSelectAll.clicked.connect(self.selectAll)
self.pushButtonDeselectAll.clicked.connect(self.deselectAll)
self.pushButtonSelectOne.clicked.connect(self.selectOne)
self.pushButtonDeselectOne.clicked.connect(self.deselectOne)
self.checkBoxAll.stateChanged.connect(self.setAllGroup)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0,0,0,0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0,0,1,1)
#Objects Connections
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("connectionChanged()")), self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel, QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck, QtCore.SIGNAL(("stateChanged(int)")), self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")), self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def __init__(self, iface, codeList, parent=None):
"""Constructor."""
super(CreateInomDialog, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.iface = iface
self.map_index = UtmGrid()
self.disableAll()
self.setValidCharacters()
self.setMask()
self.codeList = codeList
self.layerFactory = LayerFactory()
def __init__(self, codeList, parent=None):
"""
Constructor
"""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0, 0, 0, 0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0, 0, 1, 1)
#Objects Connections
QtCore.QObject.connect(self.widget,
QtCore.SIGNAL(("connectionChanged()")),
self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(
("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel,
QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck,
QtCore.SIGNAL(("stateChanged(int)")),
self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")),
self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.widget.styleChanged.connect(self.populateStyleCombo)
self.codeList = codeList
self.layerFactory = LayerFactory()
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByCategory, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.categories = []
self.selectedClasses = []
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0,0,0,0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0,0,1,1)
#Objects Connections
self.widget.connectionChanged.connect(self.listCategoriesFromDatabase)
self.widget.problemOccurred.connect(self.pushMessage)
self.pushButtonCancel.clicked.connect(self.cancel)
self.pushButtonOk.clicked.connect(self.okSelected)
self.pushButtonSelectAll.clicked.connect(self.selectAll)
self.pushButtonDeselectAll.clicked.connect(self.deselectAll)
self.pushButtonSelectOne.clicked.connect(self.selectOne)
self.pushButtonDeselectOne.clicked.connect(self.deselectOne)
self.checkBoxAll.stateChanged.connect(self.setAllGroup)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def __init__(self, iface, codeList, parent=None):
"""Constructor."""
super(CreateInomDialog, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.iface = iface
self.map_index = UtmGrid()
self.disableAll()
self.setValidCharacters()
self.setMask()
self.codeList = codeList
self.layerFactory = LayerFactory()
class LoadByCategory(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByCategory, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.categories = []
self.selectedClasses = []
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0, 0, 0, 0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0, 0, 1, 1)
#Objects Connections
self.widget.connectionChanged.connect(self.listCategoriesFromDatabase)
self.widget.problemOccurred.connect(self.pushMessage)
self.pushButtonCancel.clicked.connect(self.cancel)
self.pushButtonOk.clicked.connect(self.okSelected)
self.pushButtonSelectAll.clicked.connect(self.selectAll)
self.pushButtonDeselectAll.clicked.connect(self.deselectAll)
self.pushButtonSelectOne.clicked.connect(self.selectOne)
self.pushButtonDeselectOne.clicked.connect(self.deselectOne)
self.checkBoxAll.stateChanged.connect(self.setAllGroup)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
self.categories = []
self.selectedClasses = []
self.listWidgetCategoryFrom.clear()
self.listWidgetCategoryTo.clear()
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
def listCategoriesFromDatabase(self):
self.listWidgetCategoryFrom.clear()
self.listWidgetCategoryTo.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
classes = self.widget.abstractDb.listGeomClassesFromDatabase()
for table in classes:
schema, layerName = self.widget.abstractDb.getTableSchema(table)
category = layerName.split('_')[0]
categoryName = schema + '.' + category
if layerName.split("_")[-1] == "p":
self.point.append(table)
if layerName.split("_")[-1] == "l":
self.line.append(table)
if layerName.split("_")[-1] == "a":
self.polygon.append(table)
self.insertIntoListView(categoryName)
self.listWidgetCategoryFrom.sortItems()
self.setCRS()
def insertIntoListView(self, item_name):
found = self.listWidgetCategoryFrom.findItems(item_name,
Qt.MatchExactly)
if len(found) == 0:
item = QtGui.QListWidgetItem(item_name)
self.listWidgetCategoryFrom.addItem(item)
def selectAll(self):
tam = self.listWidgetCategoryFrom.__len__()
for i in range(tam + 1, 1, -1):
item = self.listWidgetCategoryFrom.takeItem(i - 2)
self.listWidgetCategoryTo.addItem(item)
self.listWidgetCategoryTo.sortItems()
def deselectAll(self):
tam = self.listWidgetCategoryTo.__len__()
for i in range(tam + 1, 1, -1):
item = self.listWidgetCategoryTo.takeItem(i - 2)
self.listWidgetCategoryFrom.addItem(item)
self.listWidgetCategoryFrom.sortItems()
def selectOne(self):
listedItems = self.listWidgetCategoryFrom.selectedItems()
for i in listedItems:
item = self.listWidgetCategoryFrom.takeItem(
self.listWidgetCategoryFrom.row(i))
self.listWidgetCategoryTo.addItem(item)
self.listWidgetCategoryTo.sortItems()
def deselectOne(self):
listedItems = self.listWidgetCategoryTo.selectedItems()
for i in listedItems:
item = self.listWidgetCategoryTo.takeItem(
self.listWidgetCategoryTo.row(i))
self.listWidgetCategoryFrom.addItem(item)
self.listWidgetCategoryFrom.sortItems()
def setAllGroup(self):
if self.checkBoxAll.isChecked():
self.checkBoxPoint.setCheckState(2)
self.checkBoxLine.setCheckState(2)
self.checkBoxPolygon.setCheckState(2)
else:
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
def pushMessage(self, msg):
self.bar.pushMessage('', msg, level=QgsMessageBar.WARNING)
def setCRS(self):
try:
self.epsg = self.utils.findEPSG(self.db)
if self.epsg == -1:
self.bar.pushMessage(
"",
self.tr("Coordinate Reference System not set or invalid!"),
level=QgsMessageBar.WARNING)
else:
self.crs = QgsCoordinateReferenceSystem(
self.epsg, QgsCoordinateReferenceSystem.EpsgCrsId)
if self.isSpatialite:
self.spatialiteCrsEdit.setText(self.crs.description())
self.spatialiteCrsEdit.setReadOnly(True)
else:
self.postGISCrsEdit.setText(self.crs.description())
self.postGISCrsEdit.setReadOnly(True)
except:
pass
def cancel(self):
self.restoreInitialState()
self.close()
def getSelectedItems(self):
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
try:
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
if self.checkBoxOnlyWithElements.isChecked():
self.setLayersWithElements()
ponto = self.pointWithElement
linha = self.lineWithElement
area = self.polygonWithElement
else:
ponto = self.point
linha = self.line
area = self.polygon
if len(self.listWidgetCategoryTo) > 0:
categoriasSelecionadas = []
for i in range(self.listWidgetCategoryTo.__len__()):
categoriasSelecionadas.append(
self.listWidgetCategoryTo.item(i).text())
if self.checkBoxPoint.isChecked():
self.loadLayers('p', categoriasSelecionadas, ponto)
if self.checkBoxLine.isChecked():
self.loadLayers('l', categoriasSelecionadas, linha)
if self.checkBoxPolygon.isChecked():
self.loadLayers('a', categoriasSelecionadas, area)
if self.checkBoxPoint.isChecked(
) == False and self.checkBoxLine.isChecked(
) == False and self.checkBoxPolygon.isChecked() == False:
self.bar.pushMessage(
self.tr("WARNING!"),
self.tr("Please, select at least one type of layer!"),
level=QgsMessageBar.WARNING)
else:
self.restoreInitialState()
self.close()
else:
if self.widget.db and not self.widget.crs:
self.bar.pushMessage(
self.tr("CRITICAL!"),
self.
tr("Could not determine the coordinate reference system!"
),
level=QgsMessageBar.CRITICAL)
if not self.widget.db and not self.widget.crs:
self.bar.pushMessage(
self.tr("CRITICAL!"),
self.tr("Database not loaded properly!"),
level=QgsMessageBar.CRITICAL)
self.bar.pushMessage(
self.tr("CRITICAL!"),
self.
tr("Could not determine the coordinate reference system!"
),
level=QgsMessageBar.CRITICAL)
if len(self.listWidgetCategoryTo) == 0:
self.bar.pushMessage(
self.tr("WARNING!"),
self.tr("Please, select at least one category!"),
level=QgsMessageBar.WARNING)
categoriasSelecionadas = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
QApplication.restoreOverrideCursor()
except:
QApplication.restoreOverrideCursor()
def setLayersWithElements(self):
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
pontoAux = self.widget.abstractDb.countElements(self.point)
linhaAux = self.widget.abstractDb.countElements(self.line)
areaAux = self.widget.abstractDb.countElements(self.polygon)
for i in pontoAux:
if i[1] > 0:
self.pointWithElement.append(i[0])
for i in linhaAux:
if i[1] > 0:
self.lineWithElement.append(i[0])
for i in areaAux:
if i[1] > 0:
self.polygonWithElement.append(i[0])
def loadLayers(self, type, categories, table_names):
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
groupRelationshipList = qgis.utils.iface.legendInterface(
).groupLayerRelationship()
if dbName not in groupList:
self.parentTreeNode = qgis.utils.iface.legendInterface().addGroup(
dbName, -1)
else:
self.parentTreeNode = groupList.index(dbName)
if type == 'p':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Point'), True, self.parentTreeNode)
elif self.tr('Point') not in groupRelationshipList[
self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Point'), True, self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(
self.tr('Point'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
if type == 'l':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Line'), True, self.parentTreeNode)
elif self.tr('Line') not in groupRelationshipList[
self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Line'), True, self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(
self.tr('Line'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
if type == 'a':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Area'), True, self.parentTreeNode)
elif self.tr('Area') not in groupRelationshipList[
self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface().addGroup(
self.tr('Area'), True, self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(
self.tr('Area'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
def prepareLayer(self, category, table_names, idGrupo):
idSubgrupo = qgis.utils.iface.legendInterface().addGroup(
category, True, idGrupo)
table_names.sort(reverse=True)
for table_name in table_names:
schema, layerName = self.widget.abstractDb.getTableSchema(
table_name)
if (category.split('.')[1]
== layerName.split('_')[0]) and (category.split('.')[0]
== schema):
edgvLayer = self.layerFactory.makeLayer(
self.widget.abstractDb, self.codeList, table_name)
edgvLayer.load(self.widget.crs, idSubgrupo)
class LoadByClass(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""
Constructor
"""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0, 0, 0, 0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0, 0, 1, 1)
#Objects Connections
QtCore.QObject.connect(self.widget,
QtCore.SIGNAL(("connectionChanged()")),
self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(
("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel,
QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck,
QtCore.SIGNAL(("stateChanged(int)")),
self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")),
self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.widget.styleChanged.connect(self.populateStyleCombo)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
'''
Restores the dialog initial state
'''
self.selectedClasses = []
tam = self.classesListWidget.__len__()
for i in range(tam + 1, 1, -1):
item = self.classesListWidget.takeItem(i - 2)
self.selectAllCheck.setCheckState(0)
def listClassesFromDatabase(self):
'''
List all classes from database
'''
self.classes = []
self.classesListWidget.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
self.parentClassList = self.widget.abstractDb.getOrphanGeomTablesWithElements(
loading=True)
self.classes = []
try:
self.classes = self.widget.abstractDb.listGeomClassesFromDatabase()
except Exception as e:
self.bar.pushMessage(
self.tr("CRITICAL!"),
self.tr('A problem occurred! Check log for details.'),
level=QgsMessageBar.CRITICAL)
QgsMessageLog.logMessage(':'.join(e.args), 'DSG Tools Plugin',
QgsMessageLog.CRITICAL)
if self.onlyParentsCheckBox.isChecked(
) and not self.widget.isSpatialite:
self.classesListWidget.addItems(self.parentClassList)
else:
self.classesListWidget.addItems(self.classes)
self.classesListWidget.sortItems()
def on_filterEdit_textChanged(self, text):
'''
Filters shown classes
text: text used to filter classes
'''
classes = [
edgvClass for edgvClass in self.classes if text in edgvClass
]
self.classesListWidget.clear()
self.classesListWidget.addItems(classes)
self.classesListWidget.sortItems()
def cancel(self):
'''
Cancels the process
'''
self.restoreInitialState()
self.close()
def pushMessage(self, msg):
'''
Pushes a message into message bar
'''
self.bar.pushMessage("", msg, level=QgsMessageBar.CRITICAL)
def selectAll(self):
'''
Select all classes to be loaded
'''
if self.selectAllCheck.isChecked():
tam = self.classesListWidget.__len__()
for i in range(tam + 1):
item = self.classesListWidget.item(i - 1)
self.classesListWidget.setItemSelected(item, 2)
else:
tam = self.classesListWidget.__len__()
for i in range(tam + 1):
item = self.classesListWidget.item(i - 1)
self.classesListWidget.setItemSelected(item, 0)
def getSelectedItems(self):
'''
Gets the selected classes
'''
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
'''
Loads the selected layers
'''
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
self.loadLayers()
QApplication.restoreOverrideCursor()
def loadLayers(self):
'''
Actual method that load layers
'''
self.getSelectedItems()
if len(self.selectedClasses) > 0:
try:
selectedStyle = None
if self.styleDict:
if self.styleComboBox.currentText() in self.styleDict.keys(
):
selectedStyle = self.styleDict[
self.styleComboBox.currentText()]
for layer in self.selectedClasses:
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(
self.widget.abstractDb, self.codeList, layer)
if dbName in groupList:
edgvLayer.load(self.widget.crs,
groupList.index(dbName),
stylePath=selectedStyle)
else:
self.parentTreeNode = qgis.utils.iface.legendInterface(
).addGroup(self.widget.abstractDb.getDatabaseName(),
-1)
edgvLayer.load(self.widget.crs,
self.parentTreeNode,
stylePath=selectedStyle)
self.restoreInitialState()
self.close()
except:
self.bar.pushMessage(
self.tr("Error!"),
self.tr("Could not load the selected classes!"),
level=QgsMessageBar.CRITICAL)
else:
self.bar.pushMessage(self.tr("Warning!"),
self.tr("Please, select at least one class!"),
level=QgsMessageBar.WARNING)
def populateStyleCombo(self, styleDict):
self.styleComboBox.clear()
self.styleDict = styleDict
styleList = styleDict.keys()
numberOfStyles = len(styleList)
if numberOfStyles > 0:
self.styleComboBox.addItem(self.tr('Select Style'))
for i in range(numberOfStyles):
self.styleComboBox.addItem(styleList[i])
else:
self.syleComboBox.addItem(self.tr('No available styles'))
@pyqtSlot(int)
def on_onlyParentsCheckBox_stateChanged(self):
self.listClassesFromDatabase()
class LoadByClass(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0,0,0,0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0,0,1,1)
#Objects Connections
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("connectionChanged()")), self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel, QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck, QtCore.SIGNAL(("stateChanged(int)")), self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")), self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
self.selectedClasses = []
tam = self.classesListWidget.__len__()
for i in range(tam+1,1,-1):
item = self.classesListWidget.takeItem(i-2)
self.selectAllCheck.setCheckState(0)
def listClassesFromDatabase(self):
self.classes = []
self.classesListWidget.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
self.classes = self.widget.abstractDb.listGeomClassesFromDatabase()
self.classesListWidget.addItems(self.classes)
self.classesListWidget.sortItems()
def on_filterEdit_textChanged(self, text):
classes = [edgvClass for edgvClass in self.classes if text in edgvClass]
self.classesListWidget.clear()
self.classesListWidget.addItems(classes)
self.classesListWidget.sortItems()
def cancel(self):
self.restoreInitialState()
self.close()
def pushMessage(self, msg):
self.bar.pushMessage("", msg, level=QgsMessageBar.WARNING)
def selectAll(self):
if self.selectAllCheck.isChecked():
tam = self.classesListWidget.__len__()
for i in range(tam+1):
item = self.classesListWidget.item(i-1)
self.classesListWidget.setItemSelected(item,2)
else:
tam = self.classesListWidget.__len__()
for i in range(tam+1):
item = self.classesListWidget.item(i-1)
self.classesListWidget.setItemSelected(item,0)
def getSelectedItems(self):
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
self.loadLayers()
QApplication.restoreOverrideCursor()
def loadLayers(self):
self.getSelectedItems()
if len(self.selectedClasses)>0:
try:
for layer in self.selectedClasses:
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(self.widget.abstractDb, self.codeList, layer)
if dbName in groupList:
edgvLayer.load(self.widget.crs,groupList.index(dbName))
else:
self.parentTreeNode = qgis.utils.iface.legendInterface().addGroup(self.widget.abstractDb.getDatabaseName(), -1)
edgvLayer.load(self.widget.crs,self.parentTreeNode)
self.restoreInitialState()
self.close()
except:
self.bar.pushMessage(self.tr("Error!"), self.tr("Could not load the selected classes!"), level=QgsMessageBar.CRITICAL)
else:
self.bar.pushMessage(self.tr("Warning!"), self.tr("Please, select at least one class!"), level=QgsMessageBar.WARNING)
class LoadByClass(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""
Constructor
"""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0,0,0,0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0,0,1,1)
#Objects Connections
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("connectionChanged()")), self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel, QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck, QtCore.SIGNAL(("stateChanged(int)")), self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")), self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.widget.styleChanged.connect(self.populateStyleCombo)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
'''
Restores the dialog initial state
'''
self.selectedClasses = []
tam = self.classesListWidget.__len__()
for i in range(tam+1,1,-1):
item = self.classesListWidget.takeItem(i-2)
self.selectAllCheck.setCheckState(0)
def listClassesFromDatabase(self):
'''
List all classes from database
'''
self.classes = []
self.classesListWidget.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
self.parentClassList = self.widget.abstractDb.getOrphanGeomTablesWithElements(loading = True)
self.classes = []
try:
self.classes = self.widget.abstractDb.listGeomClassesFromDatabase()
except Exception as e:
self.bar.pushMessage(self.tr("CRITICAL!"), self.tr('A problem occurred! Check log for details.'), level=QgsMessageBar.CRITICAL)
QgsMessageLog.logMessage(':'.join(e.args), 'DSG Tools Plugin', QgsMessageLog.CRITICAL)
if self.onlyParentsCheckBox.isChecked() and not self.widget.isSpatialite:
self.classesListWidget.addItems(self.parentClassList)
else:
self.classesListWidget.addItems(self.classes)
self.classesListWidget.sortItems()
def on_filterEdit_textChanged(self, text):
'''
Filters shown classes
text: text used to filter classes
'''
classes = [edgvClass for edgvClass in self.classes if text in edgvClass]
self.classesListWidget.clear()
self.classesListWidget.addItems(classes)
self.classesListWidget.sortItems()
def cancel(self):
'''
Cancels the process
'''
self.restoreInitialState()
self.close()
def pushMessage(self, msg):
'''
Pushes a message into message bar
'''
self.bar.pushMessage("", msg, level=QgsMessageBar.CRITICAL)
def selectAll(self):
'''
Select all classes to be loaded
'''
if self.selectAllCheck.isChecked():
tam = self.classesListWidget.__len__()
for i in range(tam+1):
item = self.classesListWidget.item(i-1)
self.classesListWidget.setItemSelected(item,2)
else:
tam = self.classesListWidget.__len__()
for i in range(tam+1):
item = self.classesListWidget.item(i-1)
self.classesListWidget.setItemSelected(item,0)
def getSelectedItems(self):
'''
Gets the selected classes
'''
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
'''
Loads the selected layers
'''
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
self.loadLayers()
QApplication.restoreOverrideCursor()
def loadLayers(self):
'''
Actual method that load layers
'''
self.getSelectedItems()
if len(self.selectedClasses)>0:
try:
selectedStyle = None
if self.styleDict:
if self.styleComboBox.currentText() in self.styleDict.keys():
selectedStyle = self.styleDict[self.styleComboBox.currentText()]
for layer in self.selectedClasses:
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(self.widget.abstractDb, self.codeList, layer)
if dbName in groupList:
edgvLayer.load(self.widget.crs, groupList.index(dbName), stylePath = selectedStyle)
else:
self.parentTreeNode = qgis.utils.iface.legendInterface().addGroup(self.widget.abstractDb.getDatabaseName(), -1)
edgvLayer.load(self.widget.crs, self.parentTreeNode, stylePath = selectedStyle)
self.restoreInitialState()
self.close()
except:
self.bar.pushMessage(self.tr("Error!"), self.tr("Could not load the selected classes!"), level=QgsMessageBar.CRITICAL)
else:
self.bar.pushMessage(self.tr("Warning!"), self.tr("Please, select at least one class!"), level=QgsMessageBar.WARNING)
def populateStyleCombo(self, styleDict):
self.styleComboBox.clear()
self.styleDict = styleDict
styleList = styleDict.keys()
numberOfStyles = len(styleList)
if numberOfStyles > 0:
self.styleComboBox.addItem(self.tr('Select Style'))
for i in range(numberOfStyles):
self.styleComboBox.addItem(styleList[i])
else:
self.syleComboBox.addItem(self.tr('No available styles'))
@pyqtSlot(int)
def on_onlyParentsCheckBox_stateChanged(self):
self.listClassesFromDatabase()
class CreateInomDialog(QtGui.QDialog, FORM_CLASS):
def __init__(self, iface, codeList, parent=None):
"""Constructor."""
super(CreateInomDialog, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.iface = iface
self.map_index = UtmGrid()
self.disableAll()
self.setValidCharacters()
self.setMask()
self.codeList = codeList
self.layerFactory = LayerFactory()
@pyqtSlot()
def on_okButton_clicked(self):
if not self.widget.dbLoaded:
QMessageBox.warning(self, self.tr("Warning!"), self.tr('Please, select a database first.'))
return
if not self.validateMI():
QMessageBox.warning(self, self.tr("Warning!"), self.tr('Map name index not valid!'))
return
frame = self.map_index.getQgsPolygonFrame(self.inomLineEdit.text())
reprojected = self.reprojectFrame(frame)
self.insertFrameIntoLayer(reprojected)
self.done(1)
def insertFrameIntoLayer(self,reprojected):
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
layer = self.getFrameLayer(self.iface.legendInterface().layers())
if layer == None:
lyrName = self.widget.abstractDb.getFrameLayerName()
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(self.widget.abstractDb, self.codeList, lyrName)
if dbName in groupList:
layer = edgvLayer.load(self.widget.crs,groupList.index(dbName))
else:
self.parentTreeNode = qgis.utils.iface.legendInterface().addGroup(self.widget.abstractDb.getDatabaseName(), -1)
layer = edgvLayer.load(self.widget.crs,self.parentTreeNode)
if not layer:
return
layer.startEditing()
feat = QgsFeature()
feat.setFields(layer.dataProvider().fields())
feat.setGeometry(reprojected)
feat.setAttribute(2, self.inomLineEdit.text())
feat.setAttribute(3, self.scaleCombo.currentText())
layer.addFeatures([feat], makeSelected=True)
layer.commitChanges()
bbox = reprojected.boundingBox()
for feature in layer.getFeatures():
bbox.combineExtentWith(feature.geometry().boundingBox())
bbox = self.iface.mapCanvas().mapSettings().layerToMapCoordinates(layer, bbox)
self.iface.mapCanvas().setExtent(bbox)
self.iface.mapCanvas().refresh()
def getFrameLayer(self,ifaceLayers):
for lyr in ifaceLayers:
if 'aux_moldura_a' in lyr.name():
dbname = self.getDBNameFromLayer(lyr)
if dbname == self.widget.abstractDb.getDatabaseName():
return lyr
return None
def getDBNameFromLayer(self, lyr):
dbname = None
splitUri = lyr.dataProvider().dataSourceUri().split(' ')
if len(splitUri) > 0:
dbsplit = splitUri[0].split('=')
if len(dbsplit) > 1 and dbsplit[0] == 'dbname':
dbnameInString = dbsplit[1]
dbnameSplit = dbnameInString.split('\'')
if len(dbnameSplit) > 1:
dbname = dbnameSplit[1]
return dbname
@pyqtSlot()
def on_cancelButton_clicked(self):
self.done(0)
@pyqtSlot(str)
def on_miLineEdit_textChanged(self,s):
if (s!=''):
self.inomen=self.map_index.getINomenFromMI(str(s))
self.inomLineEdit.setText(self.inomen)
@pyqtSlot(str)
def on_mirLineEdit_textChanged(self,s):
if (s!=''):
self.inomen=self.map_index.getINomenFromMIR(str(s))
self.inomLineEdit.setText(self.inomen)
def reprojectFrame(self, poly):
crsSrc = QgsCoordinateReferenceSystem(self.widget.crs.geographicCRSAuthId())
coordinateTransformer = QgsCoordinateTransform(crsSrc, self.widget.crs)
polyline = poly.asMultiPolygon()[0][0]
newPolyline = []
for point in polyline:
newPolyline.append(coordinateTransformer.transform(point))
qgsPolygon = QgsGeometry.fromMultiPolygon([[newPolyline]])
return qgsPolygon
def setValidCharacters(self):
self.chars = []
chars = 'NS'
self.chars.append(chars)
chars = 'ABCDEFGHIJKLMNOPQRSTUVZ'
self.chars.append(chars)
chars = ['01','02','03','04','05','06','07','08','09','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']
self.chars.append(chars)
chars = 'VXYZ'
self.chars.append(chars)
chars = 'ABCD'
self.chars.append(chars)
chars = ['I','II','III','IV','V','VI']
self.chars.append(chars)
chars = '1234'
self.chars.append(chars)
chars = ['NO','NE','SO','SE']
self.chars.append(chars)
chars = 'ABCDEF'
self.chars.append(chars)
chars = ['I','II','III','IV']
self.chars.append(chars)
chars = '123456'
self.chars.append(chars)
chars = 'ABCD'
self.chars.append(chars)
def setMask(self):
if self.scaleCombo.currentText() == '1000k':
self.inomLineEdit.setInputMask('NN-NN')
elif self.scaleCombo.currentText() == '500k':
self.inomLineEdit.setInputMask('NN-NN-N')
elif self.scaleCombo.currentText() == '250k':
self.inomLineEdit.setInputMask('NN-NN-N-N')
elif self.scaleCombo.currentText() == '100k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn')
elif self.scaleCombo.currentText() == '50k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0')
elif self.scaleCombo.currentText() == '25k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN')
elif self.scaleCombo.currentText() == '10k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N')
elif self.scaleCombo.currentText() == '5k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn')
elif self.scaleCombo.currentText() == '2k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn-0')
elif self.scaleCombo.currentText() == '1k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn-0-N')
def validateMI(self):
mi = self.inomLineEdit.text()
split = mi.split('-')
for i in range(len(split)):
word = str(split[i])
if len(word) == 0:
return False
if i == 0:
if word[0] not in self.chars[0]:
print word
return False
if word[1] not in self.chars[1]:
print word
return False
elif i == 1:
if word not in self.chars[2]:
print word
return False
elif i == 2:
if word not in self.chars[3]:
print word
return False
elif i == 3:
if word not in self.chars[4]:
print word
return False
elif i == 4:
if word not in self.chars[5]:
print word
return False
elif i == 5:
if word not in self.chars[6]:
print word
return False
elif i == 6:
if word not in self.chars[7]:
print word
return False
elif i == 7:
if word not in self.chars[8]:
print word
return False
elif i == 8:
if word not in self.chars[9]:
print word
return False
elif i == 9:
if word not in self.chars[10]:
print word
return False
elif i == 10:
if word not in self.chars[11]:
print word
return False
return True
def disableAll(self):
self.mirLineEdit.setEnabled(False)
self.miLineEdit.setEnabled(False)
self.inomLineEdit.setEnabled(False)
@pyqtSlot(int)
def on_scaleCombo_currentIndexChanged(self):
self.setMask()
@pyqtSlot(bool)
def on_mirRadioButton_toggled(self, toggled):
if toggled:
self.mirLineEdit.setEnabled(True)
else:
self.mirLineEdit.setEnabled(False)
@pyqtSlot(bool)
def on_miRadioButton_toggled(self, toggled):
if toggled:
self.miLineEdit.setEnabled(True)
else:
self.miLineEdit.setEnabled(False)
@pyqtSlot(bool)
def on_inomRadioButton_toggled(self, toggled):
if toggled:
self.inomLineEdit.setEnabled(True)
else:
self.inomLineEdit.setEnabled(False)
class LoadByClass(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByClass, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.selectedClasses = []
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0, 0, 0, 0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum,
QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0, 0, 1, 1)
#Objects Connections
QtCore.QObject.connect(self.widget,
QtCore.SIGNAL(("connectionChanged()")),
self.listClassesFromDatabase)
QtCore.QObject.connect(self.widget, QtCore.SIGNAL(
("problemOccurred()")), self.pushMessage)
QtCore.QObject.connect(self.pushButtonCancel,
QtCore.SIGNAL(("clicked()")), self.cancel)
QtCore.QObject.connect(self.selectAllCheck,
QtCore.SIGNAL(("stateChanged(int)")),
self.selectAll)
QtCore.QObject.connect(self.pushButtonOk, QtCore.SIGNAL(("clicked()")),
self.okSelected)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
self.selectedClasses = []
tam = self.classesListWidget.__len__()
for i in range(tam + 1, 1, -1):
item = self.classesListWidget.takeItem(i - 2)
self.selectAllCheck.setCheckState(0)
def listClassesFromDatabase(self):
self.classes = []
self.classesListWidget.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
self.classes = self.widget.abstractDb.listGeomClassesFromDatabase()
self.classesListWidget.addItems(self.classes)
self.classesListWidget.sortItems()
def on_filterEdit_textChanged(self, text):
classes = [
edgvClass for edgvClass in self.classes if text in edgvClass
]
self.classesListWidget.clear()
self.classesListWidget.addItems(classes)
self.classesListWidget.sortItems()
def cancel(self):
self.restoreInitialState()
self.close()
def pushMessage(self, msg):
self.bar.pushMessage("", msg, level=QgsMessageBar.WARNING)
def selectAll(self):
if self.selectAllCheck.isChecked():
tam = self.classesListWidget.__len__()
for i in range(tam + 1):
item = self.classesListWidget.item(i - 1)
self.classesListWidget.setItemSelected(item, 2)
else:
tam = self.classesListWidget.__len__()
for i in range(tam + 1):
item = self.classesListWidget.item(i - 1)
self.classesListWidget.setItemSelected(item, 0)
def getSelectedItems(self):
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
self.loadLayers()
QApplication.restoreOverrideCursor()
def loadLayers(self):
self.getSelectedItems()
if len(self.selectedClasses) > 0:
try:
for layer in self.selectedClasses:
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(
self.widget.abstractDb, self.codeList, layer)
if dbName in groupList:
edgvLayer.load(self.widget.crs,
groupList.index(dbName))
else:
self.parentTreeNode = qgis.utils.iface.legendInterface(
).addGroup(self.widget.abstractDb.getDatabaseName(),
-1)
edgvLayer.load(self.widget.crs, self.parentTreeNode)
self.restoreInitialState()
self.close()
except:
self.bar.pushMessage(
self.tr("Error!"),
self.tr("Could not load the selected classes!"),
level=QgsMessageBar.CRITICAL)
else:
self.bar.pushMessage(self.tr("Warning!"),
self.tr("Please, select at least one class!"),
level=QgsMessageBar.WARNING)
class CreateInomDialog(QtGui.QDialog, FORM_CLASS):
def __init__(self, iface, codeList, parent=None):
"""Constructor."""
super(CreateInomDialog, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.iface = iface
self.map_index = UtmGrid()
self.disableAll()
self.setValidCharacters()
self.setMask()
self.codeList = codeList
self.layerFactory = LayerFactory()
@pyqtSlot()
def on_okButton_clicked(self):
if not self.widget.dbLoaded:
QMessageBox.warning(self, self.tr("Warning!"),
self.tr('Please, select a database first.'))
return
if not self.validateMI():
QMessageBox.warning(self, self.tr("Warning!"),
self.tr('Map name index not valid!'))
return
frame = self.map_index.getQgsPolygonFrame(self.inomLineEdit.text())
reprojected = self.reprojectFrame(frame)
self.insertFrameIntoLayer(reprojected)
self.done(1)
def insertFrameIntoLayer(self, reprojected):
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
layer = self.getFrameLayer(self.iface.legendInterface().layers())
if layer == None:
lyrName = self.widget.abstractDb.getFrameLayerName()
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
edgvLayer = self.layerFactory.makeLayer(self.widget.abstractDb,
self.codeList, lyrName)
if dbName in groupList:
layer = edgvLayer.load(self.widget.crs,
groupList.index(dbName))
else:
self.parentTreeNode = qgis.utils.iface.legendInterface(
).addGroup(self.widget.abstractDb.getDatabaseName(), -1)
layer = edgvLayer.load(self.widget.crs, self.parentTreeNode)
if not layer:
return
layer.startEditing()
feat = QgsFeature()
feat.setFields(layer.dataProvider().fields())
feat.setGeometry(reprojected)
feat.setAttribute(2, self.inomLineEdit.text())
feat.setAttribute(3, self.scaleCombo.currentText())
layer.addFeatures([feat], makeSelected=True)
layer.commitChanges()
bbox = reprojected.boundingBox()
for feature in layer.getFeatures():
bbox.combineExtentWith(feature.geometry().boundingBox())
bbox = self.iface.mapCanvas().mapSettings().layerToMapCoordinates(
layer, bbox)
self.iface.mapCanvas().setExtent(bbox)
self.iface.mapCanvas().refresh()
def getFrameLayer(self, ifaceLayers):
for lyr in ifaceLayers:
if 'aux_moldura_a' in lyr.name():
dbname = self.getDBNameFromLayer(lyr)
if dbname == self.widget.abstractDb.getDatabaseName():
return lyr
return None
def getDBNameFromLayer(self, lyr):
dbname = None
splitUri = lyr.dataProvider().dataSourceUri().split(' ')
if len(splitUri) > 0:
dbsplit = splitUri[0].split('=')
if len(dbsplit) > 1 and dbsplit[0] == 'dbname':
dbnameInString = dbsplit[1]
dbnameSplit = dbnameInString.split('\'')
if len(dbnameSplit) > 1:
dbname = dbnameSplit[1]
return dbname
@pyqtSlot()
def on_cancelButton_clicked(self):
self.done(0)
@pyqtSlot(str)
def on_miLineEdit_textChanged(self, s):
if (s != ''):
self.inomen = self.map_index.getINomenFromMI(str(s))
self.inomLineEdit.setText(self.inomen)
@pyqtSlot(str)
def on_mirLineEdit_textChanged(self, s):
if (s != ''):
self.inomen = self.map_index.getINomenFromMIR(str(s))
self.inomLineEdit.setText(self.inomen)
def reprojectFrame(self, poly):
crsSrc = QgsCoordinateReferenceSystem(
self.widget.crs.geographicCRSAuthId())
coordinateTransformer = QgsCoordinateTransform(crsSrc, self.widget.crs)
polyline = poly.asMultiPolygon()[0][0]
newPolyline = []
for point in polyline:
newPolyline.append(coordinateTransformer.transform(point))
qgsPolygon = QgsGeometry.fromMultiPolygon([[newPolyline]])
return qgsPolygon
def setValidCharacters(self):
self.chars = []
chars = 'NS'
self.chars.append(chars)
chars = 'ABCDEFGHIJKLMNOPQRSTUVZ'
self.chars.append(chars)
chars = [
'01', '02', '03', '04', '05', '06', '07', '08', '09', '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'
]
self.chars.append(chars)
chars = 'VXYZ'
self.chars.append(chars)
chars = 'ABCD'
self.chars.append(chars)
chars = ['I', 'II', 'III', 'IV', 'V', 'VI']
self.chars.append(chars)
chars = '1234'
self.chars.append(chars)
chars = ['NO', 'NE', 'SO', 'SE']
self.chars.append(chars)
chars = 'ABCDEF'
self.chars.append(chars)
chars = ['I', 'II', 'III', 'IV']
self.chars.append(chars)
chars = '123456'
self.chars.append(chars)
chars = 'ABCD'
self.chars.append(chars)
def setMask(self):
if self.scaleCombo.currentText() == '1000k':
self.inomLineEdit.setInputMask('NN-NN')
elif self.scaleCombo.currentText() == '500k':
self.inomLineEdit.setInputMask('NN-NN-N')
elif self.scaleCombo.currentText() == '250k':
self.inomLineEdit.setInputMask('NN-NN-N-N')
elif self.scaleCombo.currentText() == '100k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn')
elif self.scaleCombo.currentText() == '50k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0')
elif self.scaleCombo.currentText() == '25k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN')
elif self.scaleCombo.currentText() == '10k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N')
elif self.scaleCombo.currentText() == '5k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn')
elif self.scaleCombo.currentText() == '2k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn-0')
elif self.scaleCombo.currentText() == '1k':
self.inomLineEdit.setInputMask('NN-NN-N-N-Nnn-0-NN-N-Nnn-0-N')
def validateMI(self):
mi = self.inomLineEdit.text()
split = mi.split('-')
for i in range(len(split)):
word = str(split[i])
if len(word) == 0:
return False
if i == 0:
if word[0] not in self.chars[0]:
print word
return False
if word[1] not in self.chars[1]:
print word
return False
elif i == 1:
if word not in self.chars[2]:
print word
return False
elif i == 2:
if word not in self.chars[3]:
print word
return False
elif i == 3:
if word not in self.chars[4]:
print word
return False
elif i == 4:
if word not in self.chars[5]:
print word
return False
elif i == 5:
if word not in self.chars[6]:
print word
return False
elif i == 6:
if word not in self.chars[7]:
print word
return False
elif i == 7:
if word not in self.chars[8]:
print word
return False
elif i == 8:
if word not in self.chars[9]:
print word
return False
elif i == 9:
if word not in self.chars[10]:
print word
return False
elif i == 10:
if word not in self.chars[11]:
print word
return False
return True
def disableAll(self):
self.mirLineEdit.setEnabled(False)
self.miLineEdit.setEnabled(False)
self.inomLineEdit.setEnabled(False)
@pyqtSlot(int)
def on_scaleCombo_currentIndexChanged(self):
self.setMask()
@pyqtSlot(bool)
def on_mirRadioButton_toggled(self, toggled):
if toggled:
self.mirLineEdit.setEnabled(True)
else:
self.mirLineEdit.setEnabled(False)
@pyqtSlot(bool)
def on_miRadioButton_toggled(self, toggled):
if toggled:
self.miLineEdit.setEnabled(True)
else:
self.miLineEdit.setEnabled(False)
@pyqtSlot(bool)
def on_inomRadioButton_toggled(self, toggled):
if toggled:
self.inomLineEdit.setEnabled(True)
else:
self.inomLineEdit.setEnabled(False)
class LoadByCategory(QtGui.QDialog, FORM_CLASS):
def __init__(self, codeList, parent=None):
"""Constructor."""
super(LoadByCategory, self).__init__(parent)
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
self.categories = []
self.selectedClasses = []
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
self.bar = QgsMessageBar()
self.setLayout(QtGui.QGridLayout(self))
self.layout().setContentsMargins(0,0,0,0)
self.layout().setAlignment(QtCore.Qt.AlignTop)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Fixed)
self.bar.setSizePolicy(sizePolicy)
self.layout().addWidget(self.bar, 0,0,1,1)
#Objects Connections
self.widget.connectionChanged.connect(self.listCategoriesFromDatabase)
self.widget.problemOccurred.connect(self.pushMessage)
self.pushButtonCancel.clicked.connect(self.cancel)
self.pushButtonOk.clicked.connect(self.okSelected)
self.pushButtonSelectAll.clicked.connect(self.selectAll)
self.pushButtonDeselectAll.clicked.connect(self.deselectAll)
self.pushButtonSelectOne.clicked.connect(self.selectOne)
self.pushButtonDeselectOne.clicked.connect(self.deselectOne)
self.checkBoxAll.stateChanged.connect(self.setAllGroup)
self.widget.tabWidget.currentChanged.connect(self.restoreInitialState)
self.codeList = codeList
self.layerFactory = LayerFactory()
def restoreInitialState(self):
self.categories = []
self.selectedClasses = []
self.listWidgetCategoryFrom.clear()
self.listWidgetCategoryTo.clear()
self.point = []
self.line = []
self.polygon = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
self.parentTreeNode = None
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
self.checkBoxAll.setCheckState(0)
def listCategoriesFromDatabase(self):
self.listWidgetCategoryFrom.clear()
self.listWidgetCategoryTo.clear()
self.dbVersion = self.widget.getDBVersion()
self.qmlPath = self.widget.getQmlPath()
classes = self.widget.abstractDb.listGeomClassesFromDatabase()
for table in classes:
schema, layerName = self.widget.abstractDb.getTableSchema(table)
category = layerName.split('_')[0]
categoryName = schema+'.'+category
if layerName.split("_")[-1] == "p":
self.point.append(table)
if layerName.split("_")[-1] == "l":
self.line.append(table)
if layerName.split("_")[-1] == "a":
self.polygon.append(table)
self.insertIntoListView(categoryName)
self.listWidgetCategoryFrom.sortItems()
self.setCRS()
def insertIntoListView(self, item_name):
found = self.listWidgetCategoryFrom.findItems(item_name, Qt.MatchExactly)
if len(found) == 0:
item = QtGui.QListWidgetItem(item_name)
self.listWidgetCategoryFrom.addItem(item)
def selectAll(self):
tam = self.listWidgetCategoryFrom.__len__()
for i in range(tam+1,1,-1):
item = self.listWidgetCategoryFrom.takeItem(i-2)
self.listWidgetCategoryTo.addItem(item)
self.listWidgetCategoryTo.sortItems()
def deselectAll(self):
tam = self.listWidgetCategoryTo.__len__()
for i in range(tam+1,1,-1):
item = self.listWidgetCategoryTo.takeItem(i-2)
self.listWidgetCategoryFrom.addItem(item)
self.listWidgetCategoryFrom.sortItems()
def selectOne(self):
listedItems = self.listWidgetCategoryFrom.selectedItems()
for i in listedItems:
item = self.listWidgetCategoryFrom.takeItem(self.listWidgetCategoryFrom.row(i))
self.listWidgetCategoryTo.addItem(item)
self.listWidgetCategoryTo.sortItems()
def deselectOne(self):
listedItems = self.listWidgetCategoryTo.selectedItems()
for i in listedItems:
item = self.listWidgetCategoryTo.takeItem(self.listWidgetCategoryTo.row(i))
self.listWidgetCategoryFrom.addItem(item)
self.listWidgetCategoryFrom.sortItems()
def setAllGroup(self):
if self.checkBoxAll.isChecked():
self.checkBoxPoint.setCheckState(2)
self.checkBoxLine.setCheckState(2)
self.checkBoxPolygon.setCheckState(2)
else:
self.checkBoxPoint.setCheckState(0)
self.checkBoxLine.setCheckState(0)
self.checkBoxPolygon.setCheckState(0)
def pushMessage(self, msg):
self.bar.pushMessage('', msg, level=QgsMessageBar.WARNING)
def setCRS(self):
try:
self.epsg = self.utils.findEPSG(self.db)
if self.epsg == -1:
self.bar.pushMessage("", self.tr("Coordinate Reference System not set or invalid!"), level=QgsMessageBar.WARNING)
else:
self.crs = QgsCoordinateReferenceSystem(self.epsg, QgsCoordinateReferenceSystem.EpsgCrsId)
if self.isSpatialite:
self.spatialiteCrsEdit.setText(self.crs.description())
self.spatialiteCrsEdit.setReadOnly(True)
else:
self.postGISCrsEdit.setText(self.crs.description())
self.postGISCrsEdit.setReadOnly(True)
except:
pass
def cancel(self):
self.restoreInitialState()
self.close()
def getSelectedItems(self):
lista = self.classesListWidget.selectedItems()
self.selectedClasses = []
tam = len(lista)
for i in range(tam):
self.selectedClasses.append(lista[i].text())
self.selectedClasses.sort()
def okSelected(self):
try:
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
if self.checkBoxOnlyWithElements.isChecked():
self.setLayersWithElements()
ponto = self.pointWithElement
linha = self.lineWithElement
area = self.polygonWithElement
else:
ponto = self.point
linha = self.line
area = self.polygon
if len(self.listWidgetCategoryTo)>0:
categoriasSelecionadas = []
for i in range(self.listWidgetCategoryTo.__len__()):
categoriasSelecionadas.append(self.listWidgetCategoryTo.item(i).text())
if self.checkBoxPoint.isChecked():
self.loadLayers('p', categoriasSelecionadas, ponto)
if self.checkBoxLine.isChecked():
self.loadLayers('l', categoriasSelecionadas, linha)
if self.checkBoxPolygon.isChecked():
self.loadLayers('a', categoriasSelecionadas, area)
if self.checkBoxPoint.isChecked()== False and self.checkBoxLine.isChecked() == False and self.checkBoxPolygon.isChecked() == False:
self.bar.pushMessage(self.tr("WARNING!"), self.tr("Please, select at least one type of layer!"), level=QgsMessageBar.WARNING)
else:
self.restoreInitialState()
self.close()
else:
if self.widget.db and not self.widget.crs:
self.bar.pushMessage(self.tr("CRITICAL!"), self.tr("Could not determine the coordinate reference system!"), level=QgsMessageBar.CRITICAL)
if not self.widget.db and not self.widget.crs:
self.bar.pushMessage(self.tr("CRITICAL!"), self.tr("Database not loaded properly!"), level=QgsMessageBar.CRITICAL)
self.bar.pushMessage(self.tr("CRITICAL!"), self.tr("Could not determine the coordinate reference system!"), level=QgsMessageBar.CRITICAL)
if len(self.listWidgetCategoryTo)==0:
self.bar.pushMessage(self.tr("WARNING!"), self.tr("Please, select at least one category!"), level=QgsMessageBar.WARNING)
categoriasSelecionadas = []
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
QApplication.restoreOverrideCursor()
except:
QApplication.restoreOverrideCursor()
def setLayersWithElements(self):
self.pointWithElement = []
self.lineWithElement = []
self.polygonWithElement = []
pontoAux = self.widget.abstractDb.countElements(self.point)
linhaAux = self.widget.abstractDb.countElements(self.line)
areaAux = self.widget.abstractDb.countElements(self.polygon)
for i in pontoAux:
if i[1] > 0:
self.pointWithElement.append(i[0])
for i in linhaAux:
if i[1] > 0:
self.lineWithElement.append(i[0])
for i in areaAux:
if i[1] > 0:
self.polygonWithElement.append(i[0])
def loadLayers(self, type, categories, table_names):
dbName = self.widget.abstractDb.getDatabaseName()
groupList = qgis.utils.iface.legendInterface().groups()
groupRelationshipList = qgis.utils.iface.legendInterface().groupLayerRelationship()
if dbName not in groupList:
self.parentTreeNode = qgis.utils.iface.legendInterface (). addGroup (dbName, -1)
else:
self.parentTreeNode = groupList.index(dbName)
if type == 'p':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Point'), True,self.parentTreeNode)
elif self.tr('Point') not in groupRelationshipList[self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Point'), True,self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(self.tr('Point'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
if type == 'l':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Line'), True,self.parentTreeNode)
elif self.tr('Line') not in groupRelationshipList[self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Line'), True,self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(self.tr('Line'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
if type == 'a':
if len(groupList[self.parentTreeNode::]) == 0:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Area'), True,self.parentTreeNode)
elif self.tr('Area') not in groupRelationshipList[self.parentTreeNode][1]:
idGrupo = qgis.utils.iface.legendInterface (). addGroup (self.tr('Area'), True,self.parentTreeNode)
else:
idGrupo = groupList[self.parentTreeNode::].index(self.tr('Area'))
for category in categories:
self.prepareLayer(category, table_names, idGrupo)
def prepareLayer(self, category, table_names, idGrupo):
idSubgrupo = qgis.utils.iface.legendInterface().addGroup(category, True, idGrupo)
table_names.sort(reverse=True)
for table_name in table_names:
schema, layerName = self.widget.abstractDb.getTableSchema(table_name)
if (category.split('.')[1] == layerName.split('_')[0]) and (category.split('.')[0] == schema):
edgvLayer = self.layerFactory.makeLayer(self.widget.abstractDb, self.codeList, table_name)
edgvLayer.load(self.widget.crs, idSubgrupo)