def createMemLayer(line, breaksList): ''' create memory layer storing all reaches :return: ''' # create layer vl = QgsVectorLayer("LineString", "sinuosity_river", "memory") pr = vl.dataProvider() # add fields pr.addAttributes([ QgsField("reach", QVariant.Int), QgsField("sinuosity", QVariant.Double), QgsField("Length", QVariant.Double) ]) vl.updateFields() # create breaks with initial and final bk = sorted(breaksList) bk.insert(0, 0) bk.append(line.length()) for breack in range(1, len(bk)): ptInt = line.interpolate(bk[breack - 1]) ptFin = line.interpolate(bk[breack]) reach = splitLine(line, ptInt, ptFin) # sinuosity calc dist = qgisdist(ptInt, ptFin) lenReach = bk[breack] - bk[breack - 1] # add a feature fet = QgsFeature() fet.setGeometry(QgsGeometry.fromPolylineXY(reach)) fet.setAttributes([breack, lenReach / dist, str(lenReach)]) pr.addFeatures([fet]) #vl.updateExtents() vl.commitChanges() return vl
def dup_layer(self, crs, names): for name in names: layer = uf.getLegendLayerByName(self.iface, "POI_Evacu8") layer_type = {'0': 'Point', '1': 'LineString', '2': 'Polygon'} mem_layer = QgsVectorLayer( layer_type[str(layer.geometryType())] + "?crs=epsg:" + str(crs), name, "memory") feats = [feat for feat in layer.getFeatures()] mem_layer_data = mem_layer.dataProvider() attr = layer.dataProvider().fields().toList() mem_layer_data.addAttributes(attr) mem_layer.updateFields() mem_layer_data.addFeatures(feats) QgsMapLayerRegistry.instance().addMapLayer(mem_layer)
def generate_and_load_geopackage_layer(self, gpkg_file, crs=None, type="Point", name="source_indicators", fields=dict(name="unit_name", type=QVariant.String)): vl = QgsVectorLayer(type, name, "memory") pr = vl.dataProvider() for field in fields: pr.addAttributes([QgsField(field[0], field[1])]) vl.updateFields() if crs is not None: vl.setCrs(crs) from mappy.mappy_utils import write_layer_to_gpkg2, add_layer_from_geopackage write_layer_to_gpkg2(vl, gpkg_file, name) # vector_file = vector_file.as_posix() + f"|layername={vl.name()}" # print(vector_file) # l = QgsVectorLayer(vector_file) # l.setName(vl.name()) l = add_layer_from_geopackage(gpkg_file, name) return l
def drawCurva(self): if not hasattr(self.c, "dados"): self.c.dados = [] self.comboCurva: QtWidgets.QComboBox layer = QgsVectorLayer( 'LineString?crs=%s' % (QgsProject.instance().crs().authid()), "Curva: " + str(self.comboCurva.currentText()), "memory") layer.setCrs( QgsCoordinateReferenceSystem(QgsProject.instance().crs())) layer.renderer().symbol().setWidth(.5) layer.renderer().symbol().setColor(QtGui.QColor("#0f16d0")) layer.triggerRepaint() fields = layerFields() layer.dataProvider().addAttributes(fields) layer.updateFields() QgsProject.instance().addMapLayer(layer, False) QgsProject.instance().layerTreeRoot().insertLayer(0, layer) self.c.layer = layer layer = self.c.layer layer.dataProvider().deleteFeatures( [f.id() for f in layer.getFeatures()]) vmax = 0 k = 0 i = -1 data = None try: if hasattr(self, "justStarted") and self.justStarted: for tipo, index, state in self.c.readData(): i += 1 if tipo == "C": data = circleArc2(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" elif tipo == "EC": data = polyTransCircle(layer, state, index, self.layer, self.next_index, self.current_index) k = 0 vmax = "120 km/h" elif tipo == "EE": data = inSpiral2(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" elif tipo == "ES": data = outSpiral2(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" elif tipo == "T": data = tangent2(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" else: continue if len(self.c.dados) - 1 < i: self.c.dados.append(None) self.c.dados[i] = tipo, data else: for tipo, index, state in self.c.readData(): i += 1 if tipo == "C": data = circleArc(layer, state, index, self.layer, self.next_index, self.current_index) k = 0 vmax = "120 km/h" elif tipo == "EC": data = polyTransCircle(layer, state, index, self.layer, self.next_index, self.current_index) k = 0 vmax = "120 km/h" elif tipo == "EE": data = inSpiral(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" elif tipo == "ES": data = outSpiral(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" elif tipo == "T": data = tangent(layer, state, index, self.layer, self.next_index) k = 0 vmax = "120 km/h" else: continue if len(self.c.dados) - 1 < i: self.c.dados.append(None) self.c.dados[i] = tipo, data except Exception as e: messageDialog(title="Erro", message="Não foi possível definir a geometria", info="Provavelmente houve a interseção de curvas") msgLog(str(traceback.format_exception(None, e, e.__traceback__))) self.curvaFailed = True refreshCanvas(self.iface, layer) #TODO compute vmax and k if self.c.lastWidget and data: self.c.lastWidget.fill(data, k=k, vmax=vmax)
def painting(self, data, fileType): currentInputCRS = self.dlg.inputCRS.text().replace(" ", "") table = data.sheets()[0] nrows = table.nrows if fileType == '4G': vl = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"4G基础信息图层", "memory") vlLabel = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"4G基础信息标签层", "memory") elif fileType == '3G': vl = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"3G基础信息图层", "memory") vlLabel = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"3G基础信息标签层", "memory") elif fileType == '2G': vl = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"2G基础信息图层", "memory") vlLabel = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"2G基础信息标签层", "memory") pr = vl.dataProvider() vlLabelPr = vlLabel.dataProvider() attrs = self.createAttributesTable(fileType) pr.addAttributes(attrs) vlLabelPr.addAttributes(attrs) vl.updateFields() vlLabel.updateFields() fet = QgsFeature() self.dlg.progressBar.setVisible(True) '''开始读取到内存''' list = [[] for row in range(nrows-1)] for i in range(nrows): if i > 0: try: if fileType == '4G': for j in range(21): list[i - 1].append(table.cell(i, j).value) if j == 20: list[i - 1].append(table.cell(i, 13).value % 3) if fileType == '3G': for j in range(73): list[i - 1].append(table.cell(i, j).value) if j == 72: list[i - 1].append(int(table.cell(i, 5).value) % 3) if fileType =='2G': for j in range(67): list[i - 1].append(table.cell(i, j).value) if j == 66: list[i - 1].append(int(table.cell(i, 2).value) % 3) except: self.showErrorDialog(u'文档读取出错', u'文档可能存在错误或格式不符合规范') self.dlg.progressBar.setVisible(False) return '''完成读取到内存''' PCI_Types = { '0': ('#F49ECC', 'Mod3=0'), '1': ('#65F0FF', 'Mod3=1'), '2': ('#00FF00', 'Mod3=2'), '3': ('#000', u'未知值'), } categories = [] for pci_type, (color, label) in PCI_Types.items(): symbol = QgsSymbolV2.defaultSymbol(vl.geometryType()) # symbol = QgsFillSymbolV2.createSimple({'color': color, 'color_border': color, 'width_border': '2'}) symbol.setColor(QColor(color)) # symbol.symbolLayer(0).setOutlineColor(QColor('#75263b')) category = QgsRendererCategoryV2(pci_type, symbol, label) categories.append(category) for i in range(len(list)): fetAttrs =[] if fileType == '4G': for j in range(22): fetAttrs.append(list[i][j]) elif fileType == '3G': for j in range(74): fetAttrs.append(list[i][j]) elif fileType == '2G': for j in range(68): fetAttrs.append(list[i][j]) fet.setAttributes(fetAttrs) if fileType == '4G': Longitude = 11 Latitude = 12 Azimuth = list[i][14] elif fileType == '3G': Longitude = 12 Latitude = 13 Azimuth = list[i][15] elif fileType == '2G': Longitude = 15 Latitude = 16 Azimuth = list[i][5] points = [QgsPoint(list[i][Longitude], list[i][Latitude])] # start point startAngle = -20 endAngle = 20 while startAngle <= endAngle: # midway points points.append(QgsPoint(list[i][Longitude] + 0.001 * math.sin(math.radians(Azimuth + startAngle)), list[i][Latitude] + 0.001 * math.cos(math.radians(Azimuth + startAngle)))) startAngle += 2 points.append(QgsPoint(list[i][Longitude], list[i][Latitude])) # end point # create the renderer and assign it to a layer expression = 'Mod3' # field name renderer = QgsCategorizedSymbolRendererV2(expression, categories) vl.setRendererV2(renderer) fet.setGeometry(QgsGeometry.fromPolygon([points])) pr.addFeatures([fet]) # fet.setGeometry(QgsGeometry.fromPoint(QgsPoint(Longitude, Latitude))) vlLabelPr.addFeatures([fet]) vlLabel.setLayerTransparency(100) vl.updateExtents() vlLabel.updateExtents() vl.commitChanges() vlLabel.commitChanges() vl.updateFields() vlLabel.updateFields() if i % 200 == 0: print 'progress:', float(pr.featureCount()) / nrows * 100 self.dlg.progressBar.setValue(float(pr.featureCount()) / nrows * 100.0) palyr = QgsPalLayerSettings() # palyr.readFromLayer(vl) palyr.enabled = True palyr.fieldName = 'CellName' if fileType =='2G': palyr.fieldName = 'CELL_NAME' palyr.placement = QgsPalLayerSettings.OverPoint # palyr.setDataDefinedProperty(QgsPalLayerSettings.Size,False,False,'','') palyr.writeToLayer(vlLabel) QgsMapLayerRegistry.instance().addMapLayer(vl) QgsMapLayerRegistry.instance().addMapLayer(vlLabel) print 'progress:', 100 self.dlg.progressBar.setValue(0) self.dlg.progressBar.setVisible(False) self.iface.messageBar().clearWidgets() qgis.utils.iface.setActiveLayer(vl) self.saveEPSGChange() self.dlg.close()
def ltefpsOKBtn(self): currentInputCRS = self.dlg.inputCRS.text().replace(" ", "") try: sourceLayer = QgsMapLayerRegistry.instance().mapLayersByName(u'4G基础信息图层')[0] except: self.showErrorDialog(u'未找到图层', u'请检查是否已加载“4G基础信息图层”') return try: inputFreq = int(self.ltefpsdlg.inputFreq.text()) inputPCI = int(self.ltefpsdlg.inputPCI.text()) except: self.showErrorDialog(u'错误的输入', u'请输入数字') return # 隐藏之前的同名图层 try: sameLayers = QgsMapLayerRegistry.instance().mapLayersByName(u'4G图层频点PCI查找') for oneLayer in sameLayers: qgis.utils.iface.legendInterface().setLayerVisible(oneLayer, False) except: pass ltefpsLayer = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"4G图层频点PCI查找", "memory") ltefpsPr = ltefpsLayer.dataProvider() ltefpsPr.addAttributes(self.createAttributesTable('4GFreqPCISearch')) ltefpsLayer.updateFields() qgis.utils.iface.legendInterface().setLayerVisible(sourceLayer, False) ltefpsFet = QgsFeature() PCI_Types = { 1: ('#FFFF00', u'频点='+str(inputFreq)+',PCI='+str(inputPCI)), 0: ('#EEEEEE', u'无关频点PCI') } categories = [] for pci_type, (color, label) in PCI_Types.items(): symbol = QgsSymbolV2.defaultSymbol(ltefpsLayer.geometryType()) symbol.setColor(QColor(color)) category = QgsRendererCategoryV2(pci_type, symbol, label) categories.append(category) # 创建属性表 fetAttrs = [] relatedAttrs = [] Longitude, Latitude, Azimuth = 11, 12, 14 for feature in sourceLayer.getFeatures(): attrs = feature.attributes() if (inputFreq == attrs[9]) & (inputPCI == int(attrs[13])): temp = [] for j in range(21): temp.append(attrs[j]) temp.append(1) relatedAttrs.append(temp) else: temp = [] for j in range(21): temp.append(attrs[j]) temp.append(0) fetAttrs.append(temp) for one in relatedAttrs: fetAttrs.append(one) for fetAttr in fetAttrs: ltefpsFet.setAttributes(fetAttr) # 开始画图 points = [QgsPoint(fetAttr[Longitude], fetAttr[Latitude])] # start point startAngle = -20 endAngle = 20 while startAngle <= endAngle: points.append( QgsPoint(fetAttr[Longitude] + 0.001 * math.sin(math.radians(fetAttr[Azimuth] + startAngle)), fetAttr[Latitude] + 0.001 * math.cos(math.radians(fetAttr[Azimuth] + startAngle)))) startAngle += 2 points.append(QgsPoint(fetAttr[Longitude], fetAttr[Latitude])) # end point ltefpsFet.setGeometry(QgsGeometry.fromPolygon([points])) ltefpsPr.addFeatures([ltefpsFet]) # 根据字段值渲染颜色 expression = 'FreqPCI' # field name renderer = QgsCategorizedSymbolRendererV2(expression, categories) ltefpsLayer.setRendererV2(renderer) ltefpsLayer.updateExtents() ltefpsLayer.commitChanges() ltefpsLayer.updateFields() # 更新图层 QgsMapLayerRegistry.instance().addMapLayer(ltefpsLayer) self.ltefpsdlg.close() self.saveEPSGChange() self.dlg.close()
def gsmfsOKBtn(self): currentInputCRS = self.dlg.inputCRS.text().replace(" ", "") inputFreq = self.gsmfsdlg.inputFreq.text() try: sourceLayer = QgsMapLayerRegistry.instance().mapLayersByName(u'2G基础信息图层')[0] except: self.showErrorDialog(u'未找到图层', u'请检查是否已加载“2G基础信息图层”') return # 隐藏之前的同名图层 try: sameLayer = QgsMapLayerRegistry.instance().mapLayersByName(u'2G图层频点查找') for oneLayer in sameLayer: qgis.utils.iface.legendInterface().setLayerVisible(oneLayer, False) except: pass gsmfsLayer = QgsVectorLayer("Polygon?crs=EPSG:"+currentInputCRS, u"2G图层频点查找", "memory") gsmfsPr = gsmfsLayer.dataProvider() gsmfsPr.addAttributes(self.createAttributesTable('2GFreqSearch')) gsmfsLayer.updateFields() # QgsMapLayerRegistry.instance().addMapLayer(gsmfsLayer) qgis.utils.iface.legendInterface().setLayerVisible(sourceLayer, False) gsmfsFet = QgsFeature() try: inputFreq.replace(" ", "") B = int(inputFreq) except: self.showErrorDialog(u'错误的输入', u'请输入数字') return A = B - 1 C = B + 1 PCI_Types = { A: ('#FF63F2', u'频点='+ str(A)), B: ('#FFFF00', u'指定频点='+ str(B)), C: ('#01FF8D', u'频点='+ str(C)), '-1': ('#eeeeee', u'无关频点') } categories = [] for pci_type, (color, label) in PCI_Types.items(): symbol = QgsSymbolV2.defaultSymbol(gsmfsLayer.geometryType()) symbol.setColor(QColor(color)) category = QgsRendererCategoryV2(pci_type, symbol, label) categories.append(category) # 创建属性表 fetAttrs = [] relatedAttrs = [] Longitude, Latitude, Azimuth = 15, 16, 5 for feature in sourceLayer.getFeatures(): attrs = feature.attributes() if (int(inputFreq) == attrs[3]): temp = [] for j in range(67): temp.append(attrs[j]) temp.append(B) relatedAttrs.append(temp) elif int(inputFreq) == int(attrs[3]) - 1: temp = [] for j in range(67): temp.append(attrs[j]) temp.append(A) relatedAttrs.append(temp) elif int(inputFreq) == int(attrs[3]) + 1: temp = [] for j in range(67): temp.append(attrs[j]) temp.append(C) relatedAttrs.append(temp) else: temp = [] for j in range(67): temp.append(attrs[j]) if j == 66: if (int(inputFreq) != attrs[3]) & (int(inputFreq) != int(attrs[3])-1) & (int(inputFreq) != int(attrs[3])+1): temp.append('-1') fetAttrs.append(temp) for one in relatedAttrs: fetAttrs.append(one) for fetAttr in fetAttrs: gsmfsFet.setAttributes(fetAttr) #开始画图 points = [QgsPoint(fetAttr[Longitude], fetAttr[Latitude])] # start point startAngle = -20 endAngle = 20 while startAngle <= endAngle: points.append(QgsPoint(fetAttr[Longitude] + 0.001 * math.sin(math.radians(fetAttr[Azimuth] + startAngle)), fetAttr[Latitude] + 0.001 * math.cos(math.radians(fetAttr[Azimuth] + startAngle)))) startAngle += 2 points.append(QgsPoint(fetAttr[Longitude], fetAttr[Latitude])) # end point gsmfsFet.setGeometry(QgsGeometry.fromPolygon([points])) gsmfsPr.addFeatures([gsmfsFet]) #根据字段值渲染颜色 expression = 'Condition' # field name renderer = QgsCategorizedSymbolRendererV2(expression, categories) gsmfsLayer.setRendererV2(renderer) gsmfsLayer.updateExtents() gsmfsLayer.commitChanges() gsmfsLayer.updateFields() QgsMapLayerRegistry.instance().addMapLayer(gsmfsLayer) self.gsmfsdlg.close() self.saveEPSGChange() self.dlg.close()