From 1aa8c3983d0290dbb3f581895269edeb8439e457 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 1 Jun 2023 10:54:43 -0700 Subject: [PATCH] wip --- aider/repomap.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/aider/repomap.py b/aider/repomap.py index 50c95b50b..0b438114c 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -333,8 +333,8 @@ def call_map(): for ident in idents: definers = defines[ident] num_defs = len(definers) - if num_defs > 3: - continue + # if num_defs > 3: + # continue for referencer, num_refs in Counter(references[ident]).items(): for definer in definers: @@ -353,6 +353,10 @@ def call_map(): dangling=personalization, ) + for node in G.nodes: + # todo: sum up the weight of every inbound edge to node + pass + N = 20 top_10_nodes = sorted(ranked, key=ranked.get, reverse=True)[:N] nodes_to_remove = [node for node in G.nodes if node not in top_10_nodes] @@ -390,7 +394,7 @@ def call_map(): dump(refs, defs, data) color = get_random_color() - weight = weight * 2 + weight = weight dot.edge(refs, defs, penwidth=str(weight), color=color, fontcolor=color, label=label) top_rank = sorted([(rank, node) for (node, rank) in ranked.items()], reverse=True)