def rowCount(self, parent=QtCore.QModelIndex()):
if self.file is None:
return 0
if self.file.layer() is None:
return 0
return len(self.file.layers()) + 1
def restoreSelectedLayers(self):
proj = QgsProject.instance()
cutDefStr = proj.readEntry("CutPlugin", "GeologyCut", "[]")[0]
pointsStr = proj.readEntry("CutPlugin", "PointToDraw", '[]')[0]
self.dockwidget.blockAllSignals(True)
# try:
self.dockwidget.mXyAspectRatio.setValue(float(proj.readEntry("CutPlugin", "XyAspectRatio", '1.0')[0]))
self.dockwidget.plotComboBox.setCurrentIndex(int(proj.readEntry("CutPlugin", "plotIndex", '0')[0]))
self.dockwidget.comboBox.setCurrentIndex(int(proj.readEntry("CutPlugin", "comboBox", '0')[0]))
self.dockwidget.mWellDistance.setValue(float(proj.readEntry("CutPlugin", "distanceToWell", '50')[0]))
self.dockwidget.wellTopDepthEdit.setValue(float(proj.readEntry("CutPlugin", "wellTopDepth", '-9999')[0]))
self.dockwidget.wellBottomDepthEdit.setValue(
float(proj.readEntry("CutPlugin", "wellBottomDepth", '-9999')[0]))
self.dockwidget.currentTemplateId = int(proj.readEntry("CutPlugin", "currentTemplateId", "-1")[0])
self.dockwidget.isDefaultTrackWidth = proj.readEntry("CutPlugin", "isDefaultTrackWidth", 'True')[0] == 'True'
self.dockwidget.trackWidth = int(proj.readEntry("CutPlugin", "trackWidth", '10')[0])
self.dockwidget.setShowModel(proj.readEntry("CutPlugin", "isShowModel", 'True')[0] == 'True')
propsStr = proj.readEntry("CutPlugin", "colorRamp1", "{'color1': '202,0,32,255', "
"'color2': '64,64,64,255', "
"'discrete': '0', "
"'rampType': 'gradient', "
"'stops': '0.25;244,165,130,255:0.5;255,255,255,255:0.75;186,186,186,255'}")[0]
self.dockwidget.mColorRamp.setColorRamp(QgsGradientColorRamp.create(ast.literal_eval(propsStr)))
try:
model_no = int(proj.readEntry("CutPlugin", "model_no", '0')[0])
for i in range(self.dockwidget.modelListWidget.count()):
item = self.dockwidget.modelListWidget.item(i)
if int(item.data(Qt.UserRole)) == model_no:
self.dockwidget.modelListWidget.setCurrentItem(item)
break
propKey = proj.readEntry("CutPlugin", "property_key", "")[0]
items = self.dockwidget.propertyListWidget.findItems(propKey, Qt.MatchExactly)
if len(items):
self.dockwidget.propertyListWidget.setCurrentItem(items[0])
except:
pass
cutDefs = ast.literal_eval(cutDefStr)
self.dockwidget.tableViewTool.blockSignals(True)
reg = QgsProject.instance()
for row in cutDefs:
layer = reg.mapLayer(row['id'])
if layer:
self.dockwidget.addLayer(layer)
mdl = self.dockwidget.mdl
r = mdl.rowCount() - 1
mdl.setData(mdl.index(r, COL_VISIBLE, QtCore.QModelIndex()), bool(row['selected']),
QtCore.Qt.CheckStateRole)
mdl.setData(mdl.index(r, COL_COLOR, QtCore.QModelIndex()), QtGui.QColor(row['color']),
QtCore.Qt.BackgroundRole)
symbolXml = str(row['backColor'])
symbol = self.stringToSymbol(symbolXml)
if not symbol:
symbol = QgsFillSymbol.createSimple({'color': '#0000FF',
'style': 'solid',
'style_border': 'solid',
'color_border': 'black',
'width_border': '0.3'})
icon = QgsSymbolLayerUtils.symbolPreviewIcon(symbol, QtCore.QSize(50, 50))
mdl.setData(mdl.index(r, COL_BACKGROUND, QtCore.QModelIndex()), symbol, QtCore.Qt.UserRole)
mdl.setData(mdl.index(r, COL_BACKGROUND, QtCore.QModelIndex()), icon, QtCore.Qt.DecorationRole)
mdl.setData(mdl.index(r, COL_NAME, QtCore.QModelIndex()), icon, QtCore.Qt.DecorationRole)
self.pointstoDraw = ast.literal_eval(pointsStr)
if len(self.pointstoDraw):
self.isInitialised = True
PlottingTool().addAllLayers(self.dockwidget)
# self.updateProfil(self.pointstoDraw, False, True)
# self.updateWells()
self.toolrenderer.setPointsToDraw(self.pointstoDraw)
# except Exception as e:
# QgsMessageLog.logMessage(u"Cut restore error: {0}".format(str(e)), tag="CutPlugin")
self.dockwidget.tableViewTool.blockSignals(False)
self.isInitialised = True
self.dockwidget.blockAllSignals(False)
def rowCount(self, parent=QtCore.QModelIndex()):
return len(self._fields)
def columnCount(self, parent=QtCore.QModelIndex()):
if parent.isValid():
return 0
return self._dataframe.columns.size
def rowCount(self, parent=QtCore.QModelIndex()):
if parent.isValid():
return 0
return len(self._dataframe.index)
def columnCount(self, parent=QtCore.QModelIndex()):
if self.section is None:
return 0
return len(self.columns)
def rowCount(self, parent=QtCore.QModelIndex()):
if self.section is None:
return 0
return len(self.section.distances)
