def edit_style(self):
from qgis.gui import QgsSingleSymbolRendererWidget
from qgis.core import QgsStyle
style = QgsStyle()
sw = QStackedWidget()
sw.addWidget
for i in range(3):
if self.__renderer and i == self.__render_type:
w = QgsSingleSymbolRendererWidget(self.__layer, style,
self.__renderer)
else:
w = QgsSingleSymbolRendererWidget(self.__layer, style,
self.__default_renderers[i])
sw.addWidget(w)
combo = QComboBox()
combo.addItem("Points")
combo.addItem("Line")
combo.addItem("Polygon")
combo.currentIndexChanged[int].connect(sw.setCurrentIndex)
combo.setCurrentIndex(self.__render_type)
dlg = QDialog()
vbox = QVBoxLayout()
btn = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
btn.accepted.connect(dlg.accept)
btn.rejected.connect(dlg.reject)
vbox.addWidget(combo)
vbox.addWidget(sw)
vbox.addWidget(btn)
dlg.setLayout(vbox)
dlg.resize(800, 600)
r = dlg.exec_()
if r == QDialog.Accepted:
self.__render_type = combo.currentIndex()
self.__renderer = sw.currentWidget().renderer().clone()
self.update()
self.style_updated.emit()
class StratigraphyStyleDialog(QDialog):
def __init__(self, layer, renderer, parent=None):
QDialog.__init__(self, parent)
self.__layer = layer
self.__renderer = renderer
from qgis.gui import QgsSingleSymbolRendererWidget, QgsRuleBasedRendererWidget, QgsCategorizedSymbolRendererWidget, QgsGraduatedSymbolRendererWidget
from qgis.core import QgsSingleSymbolRenderer, QgsRuleBasedRenderer, QgsCategorizedSymbolRenderer, QgsGraduatedSymbolRenderer
from qgis.core import QgsStyle
vbox = QVBoxLayout()
hbox = QHBoxLayout()
self.__combo = QComboBox()
self.__load_btn = QPushButton("Charger style")
self.__save_btn = QPushButton("Sauver style")
self.__load_btn.clicked.connect(self.on_load_style)
self.__save_btn.clicked.connect(self.on_save_style)
hbox.addWidget(self.__combo)
hbox.addWidget(self.__load_btn)
hbox.addWidget(self.__save_btn)
self.__sw = QStackedWidget()
self.__classes = [(u"Symbole unique", QgsSingleSymbolRenderer,
QgsSingleSymbolRendererWidget),
(u"Ensemble de règles", QgsRuleBasedRenderer,
QgsRuleBasedRendererWidget),
(u"Catégorisé", QgsCategorizedSymbolRenderer,
QgsCategorizedSymbolRendererWidget),
(u"Gradué", QgsGraduatedSymbolRenderer,
QgsGraduatedSymbolRendererWidget)]
self.__styles = [QgsStyle(), QgsStyle(), QgsStyle(), QgsStyle()]
for i, c in enumerate(self.__classes):
name, cls, wcls = c
w = wcls.create(self.__layer, self.__styles[i], self.__renderer)
self.__sw.addWidget(w)
self.__combo.addItem(name)
self.__combo.currentIndexChanged.connect(self.__sw.setCurrentIndex)
for i, c in enumerate(self.__classes):
_, cls, _ = c
if self.__renderer.__class__ == cls:
self.__combo.setCurrentIndex(i)
break
btn = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
btn.accepted.connect(self.accept)
btn.rejected.connect(self.reject)
vbox.addLayout(hbox)
vbox.addWidget(self.__sw)
vbox.addWidget(btn)
self.setLayout(vbox)
self.resize(800, 600)
def renderer(self):
return self.__renderer
def on_save_style(self):
fn, _ = QFileDialog.getSaveFileName(self,
"Fichier style à sauvegarder",
filter="*.xml")
if fn:
doc = QDomDocument()
elt = self.__sw.currentWidget().renderer().save(
doc, QgsReadWriteContext())
doc.appendChild(elt)
fo = open(fn, "w")
fo.write(doc.toString())
fo.close()
def on_load_style(self):
fn, _ = QFileDialog.getOpenFileName(self,
"Fichier style à charger",
filter="*.xml")
if fn:
doc = QDomDocument()
doc.setContent(open(fn, "r").read())
self.__renderer = QgsFeatureRenderer.load(doc.documentElement(),
QgsReadWriteContext())
for i, c in enumerate(self.__classes):
_, cls, wcls = c
if self.__renderer.__class__ == cls:
new_widget = wcls.create(self.__layer, self.__styles[i],
self.__renderer)
idx = i
break
# replace old widget
self.__sw.removeWidget(self.__sw.widget(idx))
self.__sw.insertWidget(idx, new_widget)
self.__sw.setCurrentIndex(idx)
self.__combo.setCurrentIndex(idx)
def accept(self):
self.__renderer = self.__sw.currentWidget().renderer().clone()
self.update()
return QDialog.accept(self)