mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-09 00:35:20 -05:00
377b181c4e
Also - simplify and add tests - remove large memory leak (persistant store of all non-project layers) - remove broken support for direct loading postgres/virtual layers by string (Python version was very broken and would never match a postgres/virtual layer)
34 lines
974 B
Python
34 lines
974 B
Python
##Table=group
|
|
##Input=vector
|
|
##Fields=Field Input
|
|
##Frequency=output table
|
|
|
|
from processing.tools.vector import TableWriter
|
|
from collections import defaultdict
|
|
from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
|
|
from processing.tools import dataobjects
|
|
|
|
layer = dataobjects.getLayerFromString(Input)
|
|
inputFields = layer.fields()
|
|
fieldIdxs = []
|
|
fields = Fields.split(',')
|
|
for f in fields:
|
|
idx = inputFields.indexFromName(f)
|
|
if idx == -1:
|
|
raise GeoAlgorithmExecutionException('Field not found:' + f)
|
|
fieldIdxs.append(idx)
|
|
writer = TableWriter(Frequency, None, fields + ['FREQ'])
|
|
|
|
counts = {}
|
|
feats = processing.features(layer)
|
|
nFeats = len(feats)
|
|
counts = defaultdict(int)
|
|
for i, feat in enumerate(feats):
|
|
feedback.setProgress(int(100 * i / nFeats))
|
|
attrs = feat.attributes()
|
|
clazz = tuple([attrs[i] for i in fieldIdxs])
|
|
counts[clazz] += 1
|
|
|
|
for c in counts:
|
|
writer.addRecord(list(c) + [counts[c]])
|