def __init__(self, iface, settings):
self.iface = iface
self.s = settings
QDialog.__init__(self, iface.mainWindow())
# Set up the user interface from Designer.
self.ui = Ui_VoGISRaumplanungPlot()
self.ui.setupUi(self)
try:
self.json_settings = JsonSettings(self.iface, self.s.read(self.s.key_file_settings))
except:
msg = u'Einstellungsdatei konnte nicht geladen werden!\nBitte die Datei auf korrektes JSON-Format überprüfen.\n\n\nFehlerprotokoll:\n{0}'.format(traceback.format_exc())
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.dkm_coverage_layer = None
self.gstke = {}
self.curr_gem_name = None
#hold kgs and gnrs if there's already a selection in the active document
self.autoselect_kgs = None
self.autoselect_gnrs = None
self.__add_themen()
#add Gemeinde names
self.gem_src = VRPGemeinden(self.iface, self.s, self.json_settings)
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
gem_names = self.gem_src.get_names()
#if VRP_DEBUG is True: QgsMessageLog.logMessage('\n'.join(gem_names), DLG_CAPTION)
for gem in gem_names:
item = QListWidgetItem(gem)
item.setData(Qt.UserRole, gem)
self.ui.LST_GEMEINDEN.addItem(item)
finally:
QApplication.restoreOverrideCursor()
#insert layouts
for name, layout in self.json_settings.layouts().iteritems():
self.ui.CB_Layout.addItem(name, layout)
#speed up items insertion
self.ui.LST_GSTKE.setLayoutMode( QListWidget.Batched )
self.ui.LST_GSTKE.setBatchSize( 100 )
self.ui.LST_GSTKE.setUniformItemSizes(True)
#check if the active layer is a DKM layer and if there are any selected features
if not self.iface.activeLayer() is None:
act_lyr = self.iface.activeLayer()
if isinstance(act_lyr, QgsVectorLayer):
sel_feats = act_lyr.selectedFeatures()
if len(sel_feats) > 0:
fld_gemname = self.json_settings.fld_pgem_name()
fld_kg = self.json_settings.fld_kg()
fld_gnr = self.json_settings.fld_gnr()
dkm_flds = [fld_gemname, fld_kg, fld_gnr]
if all( dkm_fld in [fld.name() for fld in act_lyr.pendingFields()] for dkm_fld in dkm_flds):
#self.iface.messageBar().pushMessage(u'alles da!', QgsMessageBar.INFO)
self.__apply_selected_features(sel_feats, fld_gemname, fld_kg, fld_gnr)
def __init__(self, iface, settings):
self.iface = iface
self.s = settings
QDialog.__init__(self, iface.mainWindow())
# Set up the user interface from Designer.
self.ui = Ui_VoGISRaumplanungPlot()
self.ui.setupUi(self)
try:
self.json_settings = JsonSettings(
self.iface, self.s.read(self.s.key_file_settings))
except:
msg = u'Einstellungsdatei konnte nicht geladen werden!\nBitte die Datei auf korrektes JSON-Format überprüfen.\n\n\nFehlerprotokoll:\n{0}'.format(
traceback.format_exc())
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.dkm_coverage_layer = None
self.gstke = {}
self.curr_gem_name = None
#hold kgs and gnrs if there's already a selection in the active document
self.autoselect_kgs = None
self.autoselect_gnrs = None
self.__add_themen()
#add Gemeinde names
self.gem_src = VRPGemeinden(self.iface, self.s, self.json_settings)
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
gem_names = self.gem_src.get_names()
#if VRP_DEBUG is True: QgsMessageLog.logMessage('\n'.join(gem_names), DLG_CAPTION)
for gem in gem_names:
item = QListWidgetItem(gem)
item.setData(Qt.UserRole, gem)
self.ui.LST_GEMEINDEN.addItem(item)
finally:
QApplication.restoreOverrideCursor()
#insert layouts
for name, layout in self.json_settings.layouts().iteritems():
self.ui.CB_Layout.addItem(name, layout)
#speed up items insertion
self.ui.LST_GSTKE.setLayoutMode(QListWidget.Batched)
self.ui.LST_GSTKE.setBatchSize(100)
self.ui.LST_GSTKE.setUniformItemSizes(True)
#check if the active layer is a DKM layer and if there are any selected features
if not self.iface.activeLayer() is None:
act_lyr = self.iface.activeLayer()
if isinstance(act_lyr, QgsVectorLayer):
sel_feats = act_lyr.selectedFeatures()
if len(sel_feats) > 0:
fld_gemname = self.json_settings.fld_pgem_name()
fld_kg = self.json_settings.fld_kg()
fld_gnr = self.json_settings.fld_gnr()
dkm_flds = [fld_gemname, fld_kg, fld_gnr]
if all(dkm_fld in
[fld.name() for fld in act_lyr.pendingFields()]
for dkm_fld in dkm_flds):
#self.iface.messageBar().pushMessage(u'alles da!', QgsMessageBar.INFO)
self.__apply_selected_features(sel_feats, fld_gemname,
fld_kg, fld_gnr)
class VoGISRaumplanungPlotDialog(QDialog):
def __init__(self, iface, settings):
self.iface = iface
self.s = settings
QDialog.__init__(self, iface.mainWindow())
# Set up the user interface from Designer.
self.ui = Ui_VoGISRaumplanungPlot()
self.ui.setupUi(self)
try:
self.json_settings = JsonSettings(self.iface, self.s.read(self.s.key_file_settings))
except:
msg = u'Einstellungsdatei konnte nicht geladen werden!\nBitte die Datei auf korrektes JSON-Format überprüfen.\n\n\nFehlerprotokoll:\n{0}'.format(traceback.format_exc())
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.dkm_coverage_layer = None
self.gstke = {}
self.curr_gem_name = None
#hold kgs and gnrs if there's already a selection in the active document
self.autoselect_kgs = None
self.autoselect_gnrs = None
self.__add_themen()
#add Gemeinde names
self.gem_src = VRPGemeinden(self.iface, self.s, self.json_settings)
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
gem_names = self.gem_src.get_names()
#if VRP_DEBUG is True: QgsMessageLog.logMessage('\n'.join(gem_names), DLG_CAPTION)
for gem in gem_names:
item = QListWidgetItem(gem)
item.setData(Qt.UserRole, gem)
self.ui.LST_GEMEINDEN.addItem(item)
finally:
QApplication.restoreOverrideCursor()
#insert layouts
for name, layout in self.json_settings.layouts().iteritems():
self.ui.CB_Layout.addItem(name, layout)
#speed up items insertion
self.ui.LST_GSTKE.setLayoutMode( QListWidget.Batched )
self.ui.LST_GSTKE.setBatchSize( 100 )
self.ui.LST_GSTKE.setUniformItemSizes(True)
#check if the active layer is a DKM layer and if there are any selected features
if not self.iface.activeLayer() is None:
act_lyr = self.iface.activeLayer()
if isinstance(act_lyr, QgsVectorLayer):
sel_feats = act_lyr.selectedFeatures()
if len(sel_feats) > 0:
fld_gemname = self.json_settings.fld_pgem_name()
fld_kg = self.json_settings.fld_kg()
fld_gnr = self.json_settings.fld_gnr()
dkm_flds = [fld_gemname, fld_kg, fld_gnr]
if all( dkm_fld in [fld.name() for fld in act_lyr.pendingFields()] for dkm_fld in dkm_flds):
#self.iface.messageBar().pushMessage(u'alles da!', QgsMessageBar.INFO)
self.__apply_selected_features(sel_feats, fld_gemname, fld_kg, fld_gnr)
def __apply_selected_features(self, feats, fldgem, fldkg, fldgnr):
msg = u''
gem_name = feats[0][fldgem]
kgs = []
gnrs = []
for feat in feats:
msg += u'{0}\t{1}\t{2}\n'.format(feat[fldgem], feat[fldkg], feat[fldgnr])
kgs.append(feat[fldkg])
gnrs.append(feat[fldgnr])
msg += u'\nIst ' if len(feats) < 2 else u'\nSind '
msg += u'bereits selektiert!\nSelektion übernehmen?'
answer = QMessageBox.question(
self.iface.mainWindow(),
DLG_CAPTION,
msg,
QMessageBox.Yes | QMessageBox.No
)
if QMessageBox.No == answer:
return
self.connect(self, SIGNAL('GNRS_LOADED'), self.__gnrs_loaded)
for idx in range(0, self.ui.LST_GEMEINDEN.count()):
item = self.ui.LST_GEMEINDEN.item(idx)
if gem_name == item.data(Qt.UserRole):
#item.setSelected(True)
self.autoselect_kgs = kgs
self.autoselect_gnrs = gnrs
self.ui.LST_GEMEINDEN.setCurrentRow(idx)
break
def accept(self):
if len(self.ui.LST_GEMEINDEN.selectedItems()) < 1:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'Keine Gemeinde gewählt!')
return
gstk_filter, gnrs = self.__get_checked_gstke()
if gstk_filter is None:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'Keine Grundstücke gewählt!')
return
QgsMapLayerRegistry.instance().removeAllMapLayers()
QgsMapLayerRegistry.instance().clearAllLayerCaches()
themen = OrderedDict()
for i in xrange(0, self.ui.TREE_THEMEN.topLevelItemCount()):
node_thema = self.ui.TREE_THEMEN.topLevelItem(i)
#check, if any childnodes are checked
sub_themen = []
for j in xrange(0, node_thema.childCount()):
node_subthema = node_thema.child(j)
sub_thema = node_subthema.data(0, Qt.UserRole)
if node_subthema.checkState(0) == Qt.Checked:
sub_themen.append(sub_thema)
#thema without subthemen
thema = node_thema.data(0, Qt.UserRole)
if (
node_thema.checkState(0) == Qt.Checked
and not thema.quellen is None
and len(sub_themen) < 1
):
themen[thema] = None
elif len(sub_themen) > 0:
themen[thema] = sub_themen
#check, if all sources are valid
errors = []
for thema, subthemen in themen.iteritems():
if not thema.quellen is None and len(thema.quellen) > 0:
for quelle in thema.quellen:
quelle.pfad = quelle.pfad.replace('{gem_name}', self.curr_gem_name)
if path.isfile(quelle.pfad) is False: errors.append(quelle.pfad)
if not quelle.qml is None:
quelle.qml = quelle.qml.replace('{gem_name}', self.curr_gem_name)
if path.isfile(quelle.qml) is False: errors.append(quelle.qml)
if not subthemen is None:
for subthema in subthemen:
if not subthema.quellen is None and len(subthema.quellen) > 0:
for quelle in subthema.quellen:
quelle.pfad = quelle.pfad.replace('{gem_name}', self.curr_gem_name)
if path.isfile(quelle.pfad) is False: errors.append(quelle.pfad)
if not quelle.qml is None:
quelle.qml = quelle.qml.replace('{gem_name}', self.curr_gem_name)
if path.isfile(quelle.qml) is False: errors.append(quelle.qml)
if len(errors) > 0:
msg = u'Diese Datenquellen sind ungültig:\n\n ' + u'\n'.join(errors)
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.__add_dkm_layers()
if self.dkm_coverage_layer is None:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'DKM konnte nicht geladen werden!')
return
file_dlg = QFileDialog(self.iface.mainWindow())
pdf_out = file_dlg.getSaveFileName(
self.iface.mainWindow(),
"PDF speichern unter ...",
self.s.read(self.s.key_file_pdf),
'PDF Datei (*.pdf)'
)
if pdf_out is None or pdf_out == '':
return
self.s.store(self.s.key_file_pdf, pdf_out)
layout = self.ui.CB_Layout.itemData(self.ui.CB_Layout.currentIndex())
if VRP_DEBUG is True:
QgsMessageLog.logMessage('selected Layout: {0}'.format(layout.name), DLG_CAPTION)
QgsMessageLog.logMessage('selected Layout: {0}'.format(layout.pfad), DLG_CAPTION)
ortho = None
if self.ui.CHK_Ortho.checkState() == Qt.Checked:
ortho = self.json_settings.luftbild()
composer = VRPPrintComposer(
self.iface,
self.curr_gem_name,
self.dkm_coverage_layer,
self.json_settings,
gnrs,
gstk_filter,
ortho,
themen,
layout.pfad,
pdf_out
)
#result = composer.export_atlas()
result = composer.export_all_features()
if not result is None:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'PDF konnte nicht exportiert werden:\n{0}'.format(result))
return
#open pdf
if sys.platform.startswith('darwin'):
subprocess.call(('open', pdf_out))
elif os.name == 'nt':
os.startfile(pdf_out)
elif os.name == 'posix':
subprocess.call(('xdg-open', pdf_out))
QDialog.accept(self)
def reject(self):
QDialog.reject(self)
def __add_thema_layer(self, node):
layers = []
thema = node.data(0, Qt.UserRole)
if thema.quellen is None:
return layers
for quelle in thema.quellen:
pfad = quelle.pfad.replace('{gem_name}', self.curr_gem_name)
qml = None
if not quelle.qml is None:
qml = quelle.qml.replace('{gem_name}', self.curr_gem_name)
if VRP_DEBUG is True:
QgsMessageLog.logMessage('ADD LAYER: {0}'.format(pfad), DLG_CAPTION)
QgsMessageLog.logMessage('LAYER QML: {0}'.format(qml), DLG_CAPTION)
if os.path.isfile(pfad) is False:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'Thema [{0}] nicht vorhanden:\n{1}'.format(thema.name, pfad))
else:
layer = None
if pfad.endswith('.shp') is True:
layer = QgsVectorLayer(pfad, quelle.name, "ogr")
else:
fileinfo = QFileInfo(pfad)
basename = fileinfo.baseName()
layer = QgsRasterLayer(pfad, basename)
if not layer.isValid():
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, u'Raster [{0}] konnte nicht geladen werden:\n{1}'.format(thema.name, pfad))
continue
if not qml is None and not layer is None:
layer.loadNamedStyle(qml)
#QgsMapLayerRegistry.instance().addMapLayer(layer)
if not layer is None:
layers.append(deepcopy(layer))
#if len(layers) > 1:
# leg = self.iface.legendInterface()
# idx = leg.addGroup(thema.name)
# for lyr in layers:
# leg.moveLayer(lyr, idx)
return layers
def __add_dkm_layers(self):
dkmgem = self.json_settings.dkm_gemeinde(self.curr_gem_name)
if VRP_DEBUG is True:
QgsMessageLog.logMessage('__add_dkm_layers: {0}'.format(dkmgem.keys()), DLG_CAPTION)
QgsMessageLog.logMessage('__add_dkm_layers: {0}'.format(dkmgem['shpgnr']), DLG_CAPTION)
QgsMessageLog.logMessage('__add_dkm_layers: {0}'.format(dkmgem['qmlgnr']), DLG_CAPTION)
QgsMessageLog.logMessage('__add_dkm_layers: {0}'.format(dkmgem['shpgstk']), DLG_CAPTION)
QgsMessageLog.logMessage('__add_dkm_layers: {0}'.format(dkmgem['qmlgstk']), DLG_CAPTION)
lyr = QgsVectorLayer(dkmgem['shpgstk'], dkmgem['lyrnamegstk'], 'ogr')
lyr.loadNamedStyle(dkmgem['qmlgstk'])
QgsMapLayerRegistry.instance().addMapLayer(lyr)
self.dkm_coverage_layer = lyr
lyr = QgsVectorLayer(dkmgem['shpgnr'], dkmgem['lyrnamegnr'], 'ogr')
lyr.loadNamedStyle(dkmgem['qmlgnr'])
QgsMapLayerRegistry.instance().addMapLayer(lyr)
def lst_gem_currentItem_changed(self, new_item, old_item):
if VRP_DEBUG is True: QgsMessageLog.logMessage('current item changed: {0} {1}'.format(old_item, new_item), DLG_CAPTION)
self.lst_gem_clicked(new_item)
def lst_gem_clicked(self, item):
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
self.ui.LST_GSTKE.clear()
self.curr_gem_name = item.data(Qt.UserRole)
if VRP_DEBUG is True: QgsMessageLog.logMessage('GEM SELECTION: {0}'.format(self.curr_gem_name), DLG_CAPTION)
self.gstke = self.gem_src.get_gst(self.curr_gem_name)
self.__add_gstke(self.ui.LE_GST_FILTER.text())
finally:
QApplication.restoreOverrideCursor()
self.emit(SIGNAL('GNRS_LOADED'))
def __gnrs_loaded(self):
if VRP_DEBUG is True: QgsMessageLog.logMessage('GNRS_LOADED fired', DLG_CAPTION)
self.disconnect(self, SIGNAL('GNRS_LOADED'), self.__gnrs_loaded)
if self.autoselect_kgs is None or self.autoselect_gnrs is None:
return
for i in range(0, self.ui.LST_GSTKE.count()):
item = self.ui.LST_GSTKE.item(i)
gstk_props = item.data(Qt.UserRole)
for idx, kg in enumerate(self.autoselect_kgs):
if kg == gstk_props['kg'] and self.autoselect_gnrs[idx] == gstk_props['gnr']:
item.setCheckState(Qt.Checked)
self.autoselect_kgs = None
self.autoselect_gnrs = None
def gst_text_changed(self, msg):
self.__add_gstke(msg)
def lst_themen_item_changed(self, item, idx):
if VRP_DEBUG is True: QgsMessageLog.logMessage('treeitem changed: {0} {1}'.format(item, idx), DLG_CAPTION)
checked = item.checkState(0)
if checked == Qt.Unchecked and item.text(0) == 'DKM':
item.setCheckState(0, Qt.Checked)
return
for i in xrange(0, item.childCount()):
item.child(i).setCheckState(0, checked)
def __get_checked_gstke(self):
checked = []
fldkg = self.json_settings.fld_kg()
fldgnr = self.json_settings.fld_gnr()
for i in range(0, self.ui.LST_GSTKE.count()):
item = self.ui.LST_GSTKE.item(i)
if item.checkState() == Qt.Checked:
gstk_props = item.data(Qt.UserRole)
if VRP_DEBUG is True: QgsMessageLog.logMessage('gstk_props: {0}'.format(gstk_props), DLG_CAPTION)
checked.append([gstk_props['kg'], gstk_props['gnr']])
if len(checked) < 1:
return None, None
gstk_filter = ''
gnrs = []
for i in range(0, len(checked)):
if i > 0:
gstk_filter += ' OR '
kg_nr = checked[i][0]
gnr = checked[i][1]
gnrs.append(gnr)
gstk_filter += u'("{0}" LIKE \'{1}\' AND "{2}" LIKE \'{3}\')'.format(fldkg, kg_nr, fldgnr, gnr)
if VRP_DEBUG is True: QgsMessageLog.logMessage('gstk_filter: {0}'.format(gstk_filter), DLG_CAPTION)
return gstk_filter, gnrs
def __add_gstke(self, filter):
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
self.ui.LST_GSTKE.clear()
#show them sorted
for gst, props in sorted(self.gstke.iteritems()):
item = None
if gst == 'EXCEPTION':
item = QListWidgetItem(props)
else:
if filter in gst:
item = QListWidgetItem(gst)
item.setData(Qt.UserRole, props)
if props != 'FEHLER':
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
#important! won't work otherwise
item.setCheckState(Qt.Unchecked)
if not item is None:
self.ui.LST_GSTKE.addItem(item)
finally:
QApplication.restoreOverrideCursor()
def __add_themen(self):
self.ui.TREE_THEMEN.headerItem().setText(0, '')
#TODO: resize column to contents
#all of these don't work??
self.ui.TREE_THEMEN.header().resizeSection(0, 250)
self.ui.TREE_THEMEN.header().setResizeMode(QHeaderView.ResizeToContents);
self.ui.TREE_THEMEN.header().setStretchLastSection(True);
themen = self.json_settings.themen()
for thema_name, thema in themen.iteritems():
tree_thema = QTreeWidgetItem(self.ui.TREE_THEMEN)
tree_thema.setText(0, thema_name)
tree_thema.setData(0, Qt.UserRole, thema)
tree_thema.setFlags(tree_thema.flags() | Qt.ItemIsUserCheckable)
if thema_name == 'DKM':
tree_thema.setCheckState(0, Qt.Checked)
else:
tree_thema.setCheckState(0, Qt.Unchecked)
if not thema.subthemen is None:
for subthema in thema.subthemen:
if VRP_DEBUG is True: QgsMessageLog.logMessage(u'Subthema: {0}'.format(subthema.name), DLG_CAPTION)
tree_subthema = QTreeWidgetItem()
tree_subthema.setText(0, subthema.name)
tree_subthema.setData(0, Qt.UserRole, subthema)
tree_subthema.setFlags(tree_subthema.flags() | Qt.ItemIsUserCheckable)
tree_subthema.setCheckState(0, Qt.Unchecked)
tree_thema.addChild(tree_subthema)
class VoGISRaumplanungPlotDialog(QDialog):
def __init__(self, iface, settings):
self.iface = iface
self.s = settings
QDialog.__init__(self, iface.mainWindow())
# Set up the user interface from Designer.
self.ui = Ui_VoGISRaumplanungPlot()
self.ui.setupUi(self)
try:
self.json_settings = JsonSettings(
self.iface, self.s.read(self.s.key_file_settings))
except:
msg = u'Einstellungsdatei konnte nicht geladen werden!\nBitte die Datei auf korrektes JSON-Format überprüfen.\n\n\nFehlerprotokoll:\n{0}'.format(
traceback.format_exc())
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.dkm_coverage_layer = None
self.gstke = {}
self.curr_gem_name = None
#hold kgs and gnrs if there's already a selection in the active document
self.autoselect_kgs = None
self.autoselect_gnrs = None
self.__add_themen()
#add Gemeinde names
self.gem_src = VRPGemeinden(self.iface, self.s, self.json_settings)
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
gem_names = self.gem_src.get_names()
#if VRP_DEBUG is True: QgsMessageLog.logMessage('\n'.join(gem_names), DLG_CAPTION)
for gem in gem_names:
item = QListWidgetItem(gem)
item.setData(Qt.UserRole, gem)
self.ui.LST_GEMEINDEN.addItem(item)
finally:
QApplication.restoreOverrideCursor()
#insert layouts
for name, layout in self.json_settings.layouts().iteritems():
self.ui.CB_Layout.addItem(name, layout)
#speed up items insertion
self.ui.LST_GSTKE.setLayoutMode(QListWidget.Batched)
self.ui.LST_GSTKE.setBatchSize(100)
self.ui.LST_GSTKE.setUniformItemSizes(True)
#check if the active layer is a DKM layer and if there are any selected features
if not self.iface.activeLayer() is None:
act_lyr = self.iface.activeLayer()
if isinstance(act_lyr, QgsVectorLayer):
sel_feats = act_lyr.selectedFeatures()
if len(sel_feats) > 0:
fld_gemname = self.json_settings.fld_pgem_name()
fld_kg = self.json_settings.fld_kg()
fld_gnr = self.json_settings.fld_gnr()
dkm_flds = [fld_gemname, fld_kg, fld_gnr]
if all(dkm_fld in
[fld.name() for fld in act_lyr.pendingFields()]
for dkm_fld in dkm_flds):
#self.iface.messageBar().pushMessage(u'alles da!', QgsMessageBar.INFO)
self.__apply_selected_features(sel_feats, fld_gemname,
fld_kg, fld_gnr)
def __apply_selected_features(self, feats, fldgem, fldkg, fldgnr):
msg = u''
gem_name = feats[0][fldgem]
kgs = []
gnrs = []
for feat in feats:
msg += u'{0}\t{1}\t{2}\n'.format(feat[fldgem], feat[fldkg],
feat[fldgnr])
kgs.append(feat[fldkg])
gnrs.append(feat[fldgnr])
msg += u'\nIst ' if len(feats) < 2 else u'\nSind '
msg += u'bereits selektiert!\nSelektion übernehmen?'
answer = QMessageBox.question(self.iface.mainWindow(), DLG_CAPTION,
msg, QMessageBox.Yes | QMessageBox.No)
if QMessageBox.No == answer:
return
self.connect(self, SIGNAL('GNRS_LOADED'), self.__gnrs_loaded)
for idx in range(0, self.ui.LST_GEMEINDEN.count()):
item = self.ui.LST_GEMEINDEN.item(idx)
if gem_name == item.data(Qt.UserRole):
#item.setSelected(True)
self.autoselect_kgs = kgs
self.autoselect_gnrs = gnrs
self.ui.LST_GEMEINDEN.setCurrentRow(idx)
break
def accept(self):
if len(self.ui.LST_GEMEINDEN.selectedItems()) < 1:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION,
u'Keine Gemeinde gewählt!')
return
gstk_filter, gnrs = self.__get_checked_gstke()
if gstk_filter is None:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION,
u'Keine Grundstücke gewählt!')
return
QgsMapLayerRegistry.instance().removeAllMapLayers()
QgsMapLayerRegistry.instance().clearAllLayerCaches()
themen = OrderedDict()
for i in xrange(0, self.ui.TREE_THEMEN.topLevelItemCount()):
node_thema = self.ui.TREE_THEMEN.topLevelItem(i)
#check, if any childnodes are checked
sub_themen = []
for j in xrange(0, node_thema.childCount()):
node_subthema = node_thema.child(j)
sub_thema = node_subthema.data(0, Qt.UserRole)
if node_subthema.checkState(0) == Qt.Checked:
sub_themen.append(sub_thema)
#thema without subthemen
thema = node_thema.data(0, Qt.UserRole)
if (node_thema.checkState(0) == Qt.Checked
and not thema.quellen is None and len(sub_themen) < 1):
themen[thema] = None
elif len(sub_themen) > 0:
themen[thema] = sub_themen
#check, if all sources are valid
errors = []
for thema, subthemen in themen.iteritems():
if not thema.quellen is None and len(thema.quellen) > 0:
for quelle in thema.quellen:
quelle.pfad = quelle.pfad.replace('{gem_name}',
self.curr_gem_name)
if path.isfile(quelle.pfad) is False:
errors.append(quelle.pfad)
if not quelle.qml is None:
quelle.qml = quelle.qml.replace(
'{gem_name}', self.curr_gem_name)
if path.isfile(quelle.qml) is False:
errors.append(quelle.qml)
if not subthemen is None:
for subthema in subthemen:
if not subthema.quellen is None and len(
subthema.quellen) > 0:
for quelle in subthema.quellen:
quelle.pfad = quelle.pfad.replace(
'{gem_name}', self.curr_gem_name)
if path.isfile(quelle.pfad) is False:
errors.append(quelle.pfad)
if not quelle.qml is None:
quelle.qml = quelle.qml.replace(
'{gem_name}', self.curr_gem_name)
if path.isfile(quelle.qml) is False:
errors.append(quelle.qml)
if len(errors) > 0:
msg = u'Diese Datenquellen sind ungültig:\n\n ' + u'\n'.join(
errors)
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION, msg)
return
self.__add_dkm_layers()
if self.dkm_coverage_layer is None:
QMessageBox.warning(self.iface.mainWindow(), DLG_CAPTION,
u'DKM konnte nicht geladen werden!')
return
file_dlg = QFileDialog(self.iface.mainWindow())
pdf_out = file_dlg.getSaveFileName(self.iface.mainWindow(),
"PDF speichern unter ...",
self.s.read(self.s.key_file_pdf),
'PDF Datei (*.pdf)')
if pdf_out is None or pdf_out == '':
return
self.s.store(self.s.key_file_pdf, pdf_out)
layout = self.ui.CB_Layout.itemData(self.ui.CB_Layout.currentIndex())
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'selected Layout: {0}'.format(layout.name), DLG_CAPTION)
QgsMessageLog.logMessage(
'selected Layout: {0}'.format(layout.pfad), DLG_CAPTION)
ortho = None
if self.ui.CHK_Ortho.checkState() == Qt.Checked:
ortho = self.json_settings.luftbild()
composer = VRPPrintComposer(self.iface, self.curr_gem_name,
self.dkm_coverage_layer,
self.json_settings, gnrs, gstk_filter,
ortho, themen, layout.pfad, pdf_out)
#result = composer.export_atlas()
result = composer.export_all_features()
if not result is None:
QMessageBox.warning(
self.iface.mainWindow(), DLG_CAPTION,
u'PDF konnte nicht exportiert werden:\n{0}'.format(result))
return
#open pdf
if sys.platform.startswith('darwin'):
subprocess.call(('open', pdf_out))
elif os.name == 'nt':
os.startfile(pdf_out)
elif os.name == 'posix':
subprocess.call(('xdg-open', pdf_out))
QDialog.accept(self)
def reject(self):
QDialog.reject(self)
def __add_thema_layer(self, node):
layers = []
thema = node.data(0, Qt.UserRole)
if thema.quellen is None:
return layers
for quelle in thema.quellen:
pfad = quelle.pfad.replace('{gem_name}', self.curr_gem_name)
qml = None
if not quelle.qml is None:
qml = quelle.qml.replace('{gem_name}', self.curr_gem_name)
if VRP_DEBUG is True:
QgsMessageLog.logMessage('ADD LAYER: {0}'.format(pfad),
DLG_CAPTION)
QgsMessageLog.logMessage('LAYER QML: {0}'.format(qml),
DLG_CAPTION)
if os.path.isfile(pfad) is False:
QMessageBox.warning(
self.iface.mainWindow(), DLG_CAPTION,
u'Thema [{0}] nicht vorhanden:\n{1}'.format(
thema.name, pfad))
else:
layer = None
if pfad.endswith('.shp') is True:
layer = QgsVectorLayer(pfad, quelle.name, "ogr")
else:
fileinfo = QFileInfo(pfad)
basename = fileinfo.baseName()
layer = QgsRasterLayer(pfad, basename)
if not layer.isValid():
QMessageBox.warning(
self.iface.mainWindow(), DLG_CAPTION,
u'Raster [{0}] konnte nicht geladen werden:\n{1}'.
format(thema.name, pfad))
continue
if not qml is None and not layer is None:
layer.loadNamedStyle(qml)
#QgsMapLayerRegistry.instance().addMapLayer(layer)
if not layer is None:
layers.append(deepcopy(layer))
#if len(layers) > 1:
# leg = self.iface.legendInterface()
# idx = leg.addGroup(thema.name)
# for lyr in layers:
# leg.moveLayer(lyr, idx)
return layers
def __add_dkm_layers(self):
dkmgem = self.json_settings.dkm_gemeinde(self.curr_gem_name)
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'__add_dkm_layers: {0}'.format(dkmgem.keys()), DLG_CAPTION)
QgsMessageLog.logMessage(
'__add_dkm_layers: {0}'.format(dkmgem['shpgnr']), DLG_CAPTION)
QgsMessageLog.logMessage(
'__add_dkm_layers: {0}'.format(dkmgem['qmlgnr']), DLG_CAPTION)
QgsMessageLog.logMessage(
'__add_dkm_layers: {0}'.format(dkmgem['shpgstk']), DLG_CAPTION)
QgsMessageLog.logMessage(
'__add_dkm_layers: {0}'.format(dkmgem['qmlgstk']), DLG_CAPTION)
lyr = QgsVectorLayer(dkmgem['shpgstk'], dkmgem['lyrnamegstk'], 'ogr')
lyr.loadNamedStyle(dkmgem['qmlgstk'])
QgsMapLayerRegistry.instance().addMapLayer(lyr)
self.dkm_coverage_layer = lyr
lyr = QgsVectorLayer(dkmgem['shpgnr'], dkmgem['lyrnamegnr'], 'ogr')
lyr.loadNamedStyle(dkmgem['qmlgnr'])
QgsMapLayerRegistry.instance().addMapLayer(lyr)
def lst_gem_currentItem_changed(self, new_item, old_item):
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'current item changed: {0} {1}'.format(old_item, new_item),
DLG_CAPTION)
self.lst_gem_clicked(new_item)
def lst_gem_clicked(self, item):
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
self.ui.LST_GSTKE.clear()
self.curr_gem_name = item.data(Qt.UserRole)
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'GEM SELECTION: {0}'.format(self.curr_gem_name),
DLG_CAPTION)
self.gstke = self.gem_src.get_gst(self.curr_gem_name)
self.__add_gstke(self.ui.LE_GST_FILTER.text())
finally:
QApplication.restoreOverrideCursor()
self.emit(SIGNAL('GNRS_LOADED'))
def __gnrs_loaded(self):
if VRP_DEBUG is True:
QgsMessageLog.logMessage('GNRS_LOADED fired', DLG_CAPTION)
self.disconnect(self, SIGNAL('GNRS_LOADED'), self.__gnrs_loaded)
if self.autoselect_kgs is None or self.autoselect_gnrs is None:
return
for i in range(0, self.ui.LST_GSTKE.count()):
item = self.ui.LST_GSTKE.item(i)
gstk_props = item.data(Qt.UserRole)
for idx, kg in enumerate(self.autoselect_kgs):
if kg == gstk_props['kg'] and self.autoselect_gnrs[
idx] == gstk_props['gnr']:
item.setCheckState(Qt.Checked)
self.autoselect_kgs = None
self.autoselect_gnrs = None
def gst_text_changed(self, msg):
self.__add_gstke(msg)
def lst_themen_item_changed(self, item, idx):
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'treeitem changed: {0} {1}'.format(item, idx), DLG_CAPTION)
checked = item.checkState(0)
if checked == Qt.Unchecked and item.text(0) == 'DKM':
item.setCheckState(0, Qt.Checked)
return
for i in xrange(0, item.childCount()):
item.child(i).setCheckState(0, checked)
def __get_checked_gstke(self):
checked = []
fldkg = self.json_settings.fld_kg()
fldgnr = self.json_settings.fld_gnr()
for i in range(0, self.ui.LST_GSTKE.count()):
item = self.ui.LST_GSTKE.item(i)
if item.checkState() == Qt.Checked:
gstk_props = item.data(Qt.UserRole)
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
'gstk_props: {0}'.format(gstk_props), DLG_CAPTION)
checked.append([gstk_props['kg'], gstk_props['gnr']])
if len(checked) < 1:
return None, None
gstk_filter = ''
gnrs = []
for i in range(0, len(checked)):
if i > 0:
gstk_filter += ' OR '
kg_nr = checked[i][0]
gnr = checked[i][1]
gnrs.append(gnr)
gstk_filter += u'("{0}" LIKE \'{1}\' AND "{2}" LIKE \'{3}\')'.format(
fldkg, kg_nr, fldgnr, gnr)
if VRP_DEBUG is True:
QgsMessageLog.logMessage('gstk_filter: {0}'.format(gstk_filter),
DLG_CAPTION)
return gstk_filter, gnrs
def __add_gstke(self, filter):
QApplication.setOverrideCursor(Qt.WaitCursor)
try:
self.ui.LST_GSTKE.clear()
#show them sorted
for gst, props in sorted(self.gstke.iteritems()):
item = None
if gst == 'EXCEPTION':
item = QListWidgetItem(props)
else:
if filter in gst:
item = QListWidgetItem(gst)
item.setData(Qt.UserRole, props)
if props != 'FEHLER':
item.setFlags(item.flags()
| Qt.ItemIsUserCheckable)
#important! won't work otherwise
item.setCheckState(Qt.Unchecked)
if not item is None:
self.ui.LST_GSTKE.addItem(item)
finally:
QApplication.restoreOverrideCursor()
def __add_themen(self):
self.ui.TREE_THEMEN.headerItem().setText(0, '')
#TODO: resize column to contents
#all of these don't work??
self.ui.TREE_THEMEN.header().resizeSection(0, 250)
self.ui.TREE_THEMEN.header().setResizeMode(
QHeaderView.ResizeToContents)
self.ui.TREE_THEMEN.header().setStretchLastSection(True)
themen = self.json_settings.themen()
for thema_name, thema in themen.iteritems():
tree_thema = QTreeWidgetItem(self.ui.TREE_THEMEN)
tree_thema.setText(0, thema_name)
tree_thema.setData(0, Qt.UserRole, thema)
tree_thema.setFlags(tree_thema.flags() | Qt.ItemIsUserCheckable)
if thema_name == 'DKM':
tree_thema.setCheckState(0, Qt.Checked)
else:
tree_thema.setCheckState(0, Qt.Unchecked)
if not thema.subthemen is None:
for subthema in thema.subthemen:
if VRP_DEBUG is True:
QgsMessageLog.logMessage(
u'Subthema: {0}'.format(subthema.name),
DLG_CAPTION)
tree_subthema = QTreeWidgetItem()
tree_subthema.setText(0, subthema.name)
tree_subthema.setData(0, Qt.UserRole, subthema)
tree_subthema.setFlags(tree_subthema.flags()
| Qt.ItemIsUserCheckable)
tree_subthema.setCheckState(0, Qt.Unchecked)
tree_thema.addChild(tree_subthema)
