github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 02:36:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

misc: several improvements to the graph module

Browse source
This commit is contained in:
Simone Margaritelli 2021-04-08 18:41:30 +02:00
commit 71634058a7
6 changed files with 468 additions and 263 deletions
Download patch file Download diff file Expand all files Collapse all files

21
modules/graph/edge.go
View file

@ -16,25 +16,26 @@ const (
)
type EdgeEvent struct {
Left *Node
Edge *Edge
Left *Node
Edge *Edge
Right *Node
}
type Edge struct {
Type EdgeType `json:"type"`
CreatedAt time.Time `json:"created_at"`
Position session.GPS `json:"position"`
Type EdgeType `json:"type"`
CreatedAt time.Time `json:"created_at"`
Position *session.GPS `json:"position,omitempty"`
}
func (e Edge) Dot(left, right *Node) string {
edgeLen := 2.0
func (e Edge) Dot(left, right *Node, width float64) string {
edgeLen := 1.0
if e.Type == Is {
edgeLen = 1.0
edgeLen = 0.3
}
return fmt.Sprintf("\"%s\" -> \"%s\" [label=\"%s\", len=%.2f];",
return fmt.Sprintf("\"%s\" -> \"%s\" [label=\"%s\", len=%.2f, penwidth=%.2f];",
left.String(),
right.String(),
e.Type,
edgeLen)
edgeLen,
width)
}