From da8a85fea6dcb1375510790ab3f37c017942ef30 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 25 May 2023 06:39:40 -0700 Subject: [PATCH] wip --- aider/repomap.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/aider/repomap.py b/aider/repomap.py index a60142818..860a2d39b 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -210,14 +210,16 @@ if __name__ == "__main__": edges = defaultdict(int) for ident in idents: + defs = defines[ident] + if len(defs) > 1: + dump(ident, len(defs), defs) + continue + for refs in references[ident]: - defs = defines[ident] - if len(defs) != 1: - continue - defs = list(defs)[0] - if refs == defs: - continue - edges[(refs, defs)] += 1 + for defs in defines[ident]: + if refs == defs: + continue + edges[(refs, defs)] += 1 max_w = max(edges.values()) @@ -228,7 +230,7 @@ if __name__ == "__main__": b = random.randint(0, 255) color = f"#{r:02x}{g:02x}{b:02x}80" weight = weight * 10 / max_w - weight = max(weight, 1) + # weight = max(weight, 1) dot.edge(refs, defs, penwidth=str(weight), color=color) # print(f"{refs} -{weight}-> {defs}")