package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-07 08:01:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

graph/encoding: provide attribute handling out of the box

Browse Source
This commit is contained in:
Dan Kortschak
2021-05-27 07:18:04 +09:30
parent c8613b2ab9
commit 97f387b38f
4 changed files with 175 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
graph/encoding/dot/decode_test.go
View File

@@ -532,7 +532,13 @@ type dotDirectedGraph struct {
// newDotDirectedGraph returns a new directed capable of creating user-defined
// nodes and edges.
func newDotDirectedGraph() *dotDirectedGraph {
return &dotDirectedGraph{DirectedGraph: simple.NewDirectedGraph()}
return &dotDirectedGraph{
DirectedGraph: simple.NewDirectedGraph(),
graph: &encoding.Attributes{},
node: &encoding.Attributes{},
edge: &encoding.Attributes{},
}
}
// NewNode returns a new node with a unique node ID for the graph.
@@ -552,7 +558,7 @@ func (g *dotDirectedGraph) DOTAttributers() (graph, node, edge encoding.Attribut
// DOTAttributeSetters implements the dot.AttributeSetters interface.
func (g *dotDirectedGraph) DOTAttributeSetters() (graph, node, edge encoding.AttributeSetter) {
return &g.graph, &g.node, &g.edge
return g.graph, g.node, g.edge
}
// SetDOTID sets the DOT ID of the graph.
@@ -579,7 +585,13 @@ type dotUndirectedGraph struct {
// newDotUndirectedGraph returns a new undirected capable of creating user-
// defined nodes and edges.
func newDotUndirectedGraph() *dotUndirectedGraph {
return &dotUndirectedGraph{UndirectedGraph: simple.NewUndirectedGraph()}
return &dotUndirectedGraph{
UndirectedGraph: simple.NewUndirectedGraph(),
graph: &encoding.Attributes{},
node: &encoding.Attributes{},
edge: &encoding.Attributes{},
}
}
// NewNode adds a new node with a unique node ID to the graph.
@@ -599,7 +611,7 @@ func (g *dotUndirectedGraph) DOTAttributers() (graph, node, edge encoding.Attrib
// DOTUnmarshalerAttrs implements the dot.UnmarshalerAttrs interface.
func (g *dotUndirectedGraph) DOTAttributeSetters() (graph, node, edge encoding.AttributeSetter) {
return &g.graph, &g.node, &g.edge
return g.graph, g.node, g.edge
}
// SetDOTID sets the DOT ID of the graph.
@@ -705,15 +717,9 @@ func (e *dotEdge) ToPort() (port, compass string) {
return e.ToPortLabels.Port, e.ToPortLabels.Compass
}
// attributes is a helper for global attributes.
type attributes []encoding.Attribute
func (a attributes) Attributes() []encoding.Attribute {
return []encoding.Attribute(a)
}
func (a *attributes) SetAttribute(attr encoding.Attribute) error {
*a = append(*a, attr)
return nil
type attributes interface {
encoding.Attributer
encoding.AttributeSetter
}
const undirectedSelfLoopGraph = `graph {