47 lines
1.2 KiB
Ruby
47 lines
1.2 KiB
Ruby
module Pruning
|
|
module Processing
|
|
# Contains methods that store the tree and prune everything
|
|
# except the branches with requested indicator
|
|
class Pruner
|
|
def initialize(tree)
|
|
@tree = tree
|
|
end
|
|
|
|
def prune_tree(indicator_ids)
|
|
pruned_tree = @tree
|
|
prune_subtree(pruned_tree, indicator_ids)
|
|
pruned_tree
|
|
end
|
|
|
|
private
|
|
|
|
# Depth First Search implementation
|
|
# With pruning on the same sweep
|
|
def prune_subtree(nodes, indicator_ids)
|
|
nodes_to_examine = nodes.dup
|
|
nodes_to_examine.each do |node|
|
|
if indicator_node?(node)
|
|
indicator_not_wanted = !indicator_ids.include?(node['id'])
|
|
nodes.delete(node) if indicator_not_wanted
|
|
else
|
|
no_wanted_children = prune_subtree(children(node), indicator_ids)
|
|
nodes.delete(node) if no_wanted_children
|
|
end
|
|
end
|
|
nodes.empty?
|
|
end
|
|
|
|
def children(node)
|
|
node.fetch('sub_themes', false) ||
|
|
node.fetch('categories', false) ||
|
|
node.fetch('indicators', false) ||
|
|
[]
|
|
end
|
|
|
|
def indicator_node?(node)
|
|
children(node).empty?
|
|
end
|
|
end
|
|
end
|
|
end
|