Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
[processing] log cases when subset is smaller than requested number of
features
  • Loading branch information
alexbruy committed Jul 4, 2018
1 parent 29207a1 commit b72f5d1
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
Expand Up @@ -125,9 +125,12 @@ def processAlgorithm(self, parameters, context, feedback):
classes[attrs[index]].append(feature)
feedback.setProgress(int(i * total))

for subset in classes.values():
for k, subset in classes.items():
selValue = value if method != 1 else int(round(value * len(subset), 0))
selran.extend(random.sample(subset, min(selValue, len(subset))))
if selValue > len(subset):
selValue = len(subset)
feedback.reportError(self.tr('Subset "{}" is smaller than requested number of features.'.format(k)))
selran.extend(random.sample(subset, selValue))

total = 100.0 / featureCount if featureCount else 1
for (i, feat) in enumerate(selran):
Expand Down
Expand Up @@ -133,12 +133,15 @@ def processAlgorithm(self, parameters, context, feedback):
feedback.setProgress(int(i * total))

selran = []
for subset in classes.values():
for k, subset in classes.items():
if feedback.isCanceled():
break

selValue = value if method != 1 else int(round(value * len(subset), 0))
selran.extend(random.sample(subset, min(selValue, len(subset))))
if selValue > len(subset):
selValue = len(subset)
feedback.reportError(self.tr('Subset "{}" is smaller than requested number of features.'.format(k)))
selran.extend(random.sample(subset, selValue))

layer.selectByIds(selran)
else:
Expand Down

0 comments on commit b72f5d1

Please sign in to comment.