module RGL::GraphVisitor::DistanceMapSupport

def distance_to_root(v)
  @distance_map[v]
end

def handle_tree_edge(u, v)
  super
  @distance_map[v] = @distance_map[u] + 1
end