luciano/d2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

set dagre ranksep according to max edge label size

Browse source
This commit is contained in:
Gavin Nishizawa 2022-12-20 16:05:35 -08:00
parent 2cffb51217
commit 707f5b9dd7
No known key found for this signature in database
GPG key ID: AE3B177777CE55CD
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
d2layouts/d2dagrelayout/layout.go
View file

@ -64,22 +64,35 @@ func Layout(ctx context.Context, g *d2graph.Graph) (err error) {
} }
rootAttrs := dagreGraphAttrs{ rootAttrs := dagreGraphAttrs{
ranksep: 100,
edgesep: 40, edgesep: 40,
nodesep: 60, nodesep: 60,
} }
isHorizontal := false
switch g.Root.Attributes.Direction.Value { switch g.Root.Attributes.Direction.Value {
case "down": case "down":
rootAttrs.rankdir = "TB" rootAttrs.rankdir = "TB"
case "right": case "right":
rootAttrs.rankdir = "LR" rootAttrs.rankdir = "LR"
isHorizontal = true
case "left": case "left":
rootAttrs.rankdir = "RL" rootAttrs.rankdir = "RL"
isHorizontal = true
case "up": case "up":
rootAttrs.rankdir = "BT" rootAttrs.rankdir = "BT"
default: default:
rootAttrs.rankdir = "TB" rootAttrs.rankdir = "TB"
} }
maxLabelSize := 0
for _, edge := range g.Edges {
size := edge.LabelDimensions.Width
if !isHorizontal {
size = edge.LabelDimensions.Height
}
maxLabelSize = go2.Max(maxLabelSize, size)
}
rootAttrs.ranksep = go2.Max(100, maxLabelSize+40)
configJS := setGraphAttrs(rootAttrs) configJS := setGraphAttrs(rootAttrs)
if _, err := vm.RunString(configJS); err != nil { if _, err := vm.RunString(configJS); err != nil {
return err return err