graph.go

package gdi

import (
	"fmt"
	"strings"
	"sync"
)

//
//var g *graph
//
//func init() {
//	g = &graph{
//		nodes: make(map[string]*node),
//	}
//}

type node struct {
	name   string
	fields []*nodeItem
	edges  []*edge
	//fieldMap map[string]struct{}
	//fields map[string]string
	//edges  map[string]string
}

type edge struct {
	from string
	to   string
}

type nodeItem struct {
	fieldName string
	fieldType string
}

type graph struct {
	lock  sync.Mutex
	nodes map[string]*node
}

func (n *node) addFiled(f *nodeItem) {
	//if n.fields ==nil {
	//	n.fields =make(map[string]string)
	//}
	//n.fields[f.fieldType]=f.fieldName

	n.fields = append(n.fields, f)
}
func (n *node) addEdge(e *edge) {
	//if n.edges==nil {
	//	n.edges=make(map[string]string)
	//}
	//n.edges[e.from]=e.to
	n.edges = append(n.edges, e)
}

func (gdi *GDIPool) addNode(n *node) {
	gdi.g.lock.Lock()
	defer  gdi.g.lock.Unlock()
	if _, ok := gdi.g.nodes[n.name]; !ok {
		gdi.g.nodes[n.name] = n
	}
}

func (gdi *GDIPool) Graph() string {
	var gs []string
	/*
		"node0" [
		label = "<f0> 0x10ba8| <f1>"
		shape = "record"
		];
	*/
	nodeTpl := `
   "%v" [
     label = <%v>
     shape = "none"
 ]
`
	gTpl := `
digraph { 

rankdir=LR;
  %v
}
`
	for _, n := range gdi.g.nodes {
		var fields []string
		fields = append(fields, fmt.Sprintf(`<table BORDER="1" CELLBORDER="1" CELLSPACING="0"><tr><td PORT="f100"><font POINT-SIZE="18"><b>struct %v</b></font></td></tr>`, n.name))
		for _, field := range n.fields {
			ns := strings.Split(field.fieldName, "#")
			fields = append(fields, fmt.Sprintf(`<tr><td PORT="%v">%v %v</td></tr>`, ns[0], ns[1], field.fieldType))
		}
		gs = append(gs, fmt.Sprintf(nodeTpl, n.name, strings.Join(fields, "")+"</table>"))
		var edges []string
		for _, e := range n.edges {
			edges = append(edges, fmt.Sprintf(`%v->"%v":f100;`, e.from, e.to))
		}
		gs = append(gs, strings.Join(edges, "\n"))

	}
	return fmt.Sprintf(gTpl, strings.Join(gs, "\n"))
}