53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
|
from qgis.PyQt.QtCore import QVariant
|
||
|
|
from qgis.core import (QgsFeature, QgsField, QgsGeometry, QgsPointXY, QgsVectorLayer, QgsProject)
|
||
|
|
|
||
|
|
# Locate the edges layer
|
||
|
|
edges_layer = QgsProject.instance().mapLayersByName('EDGES')[0]
|
||
|
|
|
||
|
|
# Create a new temporary layer for the poles
|
||
|
|
poles_layer = QgsVectorLayer('Point?crs=epsg:4326', 'POLES', 'memory')
|
||
|
|
|
||
|
|
# Define the data provider
|
||
|
|
pr = poles_layer.dataProvider()
|
||
|
|
|
||
|
|
# Add a new field to the new layer
|
||
|
|
pr.addAttributes([QgsField('id', QVariant.Int)])
|
||
|
|
poles_layer.updateFields()
|
||
|
|
|
||
|
|
# Store coordinates of existing poles to avoid duplicates
|
||
|
|
existing_poles = set()
|
||
|
|
|
||
|
|
# Iterate over every feature of the EDGES layer
|
||
|
|
for feature in edges_layer.getFeatures():
|
||
|
|
# Only consider edges of type 'Aerial'
|
||
|
|
if feature['type'] == 'Aerial':
|
||
|
|
# Get the geometry of the feature
|
||
|
|
geom = feature.geometry()
|
||
|
|
|
||
|
|
# Handle both 2D and 3D geometries
|
||
|
|
if geom.isMultipart():
|
||
|
|
vertices = [v for part in geom.asMultiPolyline() for v in part]
|
||
|
|
else:
|
||
|
|
vertices = geom.asPolyline()
|
||
|
|
|
||
|
|
for i, vertex in enumerate(vertices):
|
||
|
|
# Do not create a new pole if one already exists at this location
|
||
|
|
vertex_coordinates = (vertex[0], vertex[1])
|
||
|
|
if vertex_coordinates in existing_poles:
|
||
|
|
continue
|
||
|
|
|
||
|
|
# Create a new feature in the POLES layer for each vertex
|
||
|
|
pole = QgsFeature()
|
||
|
|
pole.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(*vertex_coordinates)))
|
||
|
|
pole.setAttributes([i])
|
||
|
|
pr.addFeature(pole)
|
||
|
|
|
||
|
|
# Remember this location so we don't create a duplicate pole
|
||
|
|
existing_poles.add(vertex_coordinates)
|
||
|
|
|
||
|
|
# Update the layer's extent when new features have been added
|
||
|
|
poles_layer.updateExtents()
|
||
|
|
|
||
|
|
# Add the new layer to the project
|
||
|
|
QgsProject.instance().addMapLayer(poles_layer)
|