def run(self):
area_type_dict = {} # (key: area_type, value: nsite index list)
d = QgsDistanceArea() # 计算距离
allNSiteFeatures = getFeaturesList(self.nsiteLayer)
# 把所有的nsite按区域类型分类
for nsiteFeature in allNSiteFeatures:
area_type = nsiteFeature[u"区域类型"]
if area_type not in [u"普通市区", u"密集市区", u"农村", u"郊区乡镇"]:
area_type = u"普通市区"
if not area_type_dict.has_key(area_type):
area_type_dict[area_type] = [nsiteFeature]
else:
t = area_type_dict[area_type]
t.append(nsiteFeature)
area_type_dict[area_type] = t
del t
total = len(allNSiteFeatures) # 写进度条
progess = Progess(self.parent, total, u"优化中...")
progess.show()
delete_list = [] # 要删除的 nsiteFeaures.id() list
for (area_type, nsiteFeatures_list) in area_type_dict.items():
if area_type == u"密集市区":
mergeDis = 200
elif area_type == u"郊区乡镇":
mergeDis = 500
elif area_type == u"农村":
mergeDis = 800
else:
# 默认普通市区
mergeDis = 300
for nsiteFeatures1 in nsiteFeatures_list:
progess.count()
lon1 = nsiteFeatures1[u"经度"]
lat1 = nsiteFeatures1[u"纬度"]
point1 = QgsPoint(float(lon1), float(lat1))
rectangle = createARectangleByCenterPoint(point1, 2000)
if nsiteFeatures1.id() in delete_list:
progess.count()
continue
else:
request = QgsFeatureRequest()
request.setFilterRect(rectangle)
for nsiteFeature2 in self.nsiteLayer.getFeatures(request):
if nsiteFeatures1.id() == nsiteFeature2.id():
continue
if nsiteFeature2.id() in delete_list:
continue
lon2 = nsiteFeatures1[u"经度"]
lat2 = nsiteFeatures1[u"纬度"]
point2 = QgsPoint(float(lon2), float(lat2))
distance = d.convertMeasurement(
d.measureLine(point1, point2), 2, 0, False)[0]
if distance <= mergeDis:
delete_list.append(nsiteFeature2.id())
break
progess.kill()
# 删除features
delFeatures(self.nsiteLayer, delete_list)
return True
def changed_geom(layerId, geoms):
fids = geoms.keys()
freq = QgsFeatureRequest()
freq.setFilterFids(fids)
feats = list(layer_from_name(shpName).getFeatures(freq))
fks_to_change = get_fk_set(shpName, shpKeyName, skipFirst=False, fids=fids)
global shpChange
shpChange = {k: v for (k, v) in zip(fks_to_change, feats)}
def get_fk_set(layerName, fkName, skipFirst=True, fids=None):
layer = layer_from_name(layerName)
freq = QgsFeatureRequest()
if fids is not None:
freq.setFilterFids(fids)
feats = [f for f in layer.getFeatures(freq)]
fkSet = []
first = True
for f in feats:
if skipFirst and first:
first = False
continue
fk = f.attribute(fkName)
fkSet.append(fk)
return fkSet
def get_fk_set(layerName, fkName, skipFirst=1, fids=None, useProvider=False):
"""
skipFirst: number of initial lines to skip (header lines in excel)
"""
layer = layer_from_name(layerName)
freq = QgsFeatureRequest()
if fids is not None:
freq.setFilterFids(fids)
if not useProvider:
feats = [f for f in layer.getFeatures(freq)]
else:
feats = [f for f in layer.dataProvider().getFeatures(freq)]
fkSet = []
for f in feats[skipFirst:]:
QgsMessageLog.logMessage('FK {}'.format(f.attribute(fkName)), logTag,
QgsMessageLog.CRITICAL)
fk = f.attribute(fkName)
if fk: # Skip NULL ids that may be reported from excel files
fkSet.append(fk)
return fkSet
def get_fk_set(layerName, fkName, skipFirst=1, fids=None, useProvider=False):
"""
skipFirst: number of initial lines to skip (header lines in excel)
"""
layer = layer_from_name(layerName)
freq = QgsFeatureRequest()
if fids is not None:
freq.setFilterFids(fids)
if not useProvider:
feats = [f for f in layer.getFeatures(freq)]
else:
feats = [f for f in layer.dataProvider().getFeatures(freq)]
fkSet = []
for f in feats[skipFirst:]:
QgsMessageLog.logMessage(
'FK {}'.format(f.attribute(fkName)), logTag, QgsMessageLog.CRITICAL)
fk = f.attribute(fkName)
if fk: # Skip NULL ids that may be reported from excel files
fkSet.append(fk)
return fkSet
def query_layer_for_fids(layerName, fids):
layer = layer_from_name(layerName)
freq = QgsFeatureRequest()
freq.setFilterFids(fids)
return list(layer.getFeatures(freq))
def query_layer_for_fids(layerName, fids):
layer = layer_from_name(layerName)
freq = QgsFeatureRequest()
freq.setFilterFids(fids)
return list(layer.getFeatures(freq))