Change isActor

This commit is contained in:
Júlio César Batista 2022-12-02 10:26:07 -08:00
parent f918991550
commit 7fcf62ca67
No known key found for this signature in database
GPG key ID: 10C4B861BF314878

View file

@ -14,6 +14,7 @@ import (
) )
type sequenceDiagram struct { type sequenceDiagram struct {
root *d2graph.Object
messages []*d2graph.Edge messages []*d2graph.Edge
lifelines []*d2graph.Edge lifelines []*d2graph.Edge
actors []*d2graph.Object actors []*d2graph.Object
@ -48,6 +49,7 @@ func newSequenceDiagram(actors []*d2graph.Object, messages []*d2graph.Edge) *seq
} }
for rank, actor := range actors { for rank, actor := range actors {
sd.root = actor.Parent
sd.objectRank[actor] = rank sd.objectRank[actor] = rank
sd.maxActorHeight = math.Max(sd.maxActorHeight, actor.Height) sd.maxActorHeight = math.Max(sd.maxActorHeight, actor.Height)
@ -273,13 +275,7 @@ func (sd *sequenceDiagram) getMessageY(rank int) float64 {
} }
func (sd *sequenceDiagram) isActor(obj *d2graph.Object) bool { func (sd *sequenceDiagram) isActor(obj *d2graph.Object) bool {
// TODO: map to avoid looping around every time? return obj.Parent == sd.root
for _, actor := range sd.actors {
if actor == obj {
return true
}
}
return false
} }
func (sd *sequenceDiagram) getWidth() float64 { func (sd *sequenceDiagram) getWidth() float64 {