dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 00:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

graph

Browse source
This commit is contained in:
Paul Gauthier 2023-06-02 11:25:10 -07:00
parent 903d318a8a
commit 6e51675072
1 changed files with 27 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

27
aider/repomap.py
View file

@ -370,10 +370,37 @@ def call_map():
ident = data["ident"]
ranked_definitions[(dst, ident)] += data["rank"]
dot = graphviz.Digraph(graph_attr={"ratio": ".5"})
clusters = dict()
for fname in set(show_fnames):
clusters[fname] = graphviz.Digraph(f"cluster_{fname}")
clusters[fname].attr(label=fname, style="filled")
clusters[fname].node(f"invis_{fname}", style="invis", width="0", label="")
for fname, ident in set(ranked_definitions.keys()):
clusters[fname].node(str((fname, ident)), label=ident)
for src, dst, data in G.edges(data=True):
frm = f"invis_{src}"
ident = data["ident"]
to = str((dst, ident))
dot.edge(
frm,
to,
# penwidth=str(weight), color=color, fontcolor=color, label=label,
)
for cluster in clusters.values():
dump(cluster)
dot.subgraph(cluster)
ranked_definitions = sorted(ranked_definitions.items(), reverse=True, key=lambda x: x[1])
for (fname, ident), rank in ranked_definitions:
print(f"{rank:.03f} {fname} {ident}")
dot.render("tmp", format="pdf", view=True)
return
#############