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)