func (sg *SubGraph) Dotty(labels *digraph.Labels, highlightVertices, highlightEdges map[int]bool) string { V := make([]string, 0, len(sg.V)) E := make([]string, 0, len(sg.E)) for vidx, v := range sg.V { highlight := "" if highlightVertices[vidx] { highlight = " color=red" } V = append(V, fmt.Sprintf( "n%v [label=\"%v\"%v];", vidx, labels.Label(v.Color), highlight, )) } for eidx, e := range sg.E { highlight := "" if highlightEdges[eidx] { highlight = " color=red" } E = append(E, fmt.Sprintf( "n%v->n%v [label=\"%v\"%v]", e.Src, e.Targ, labels.Label(e.Color), highlight, )) } return fmt.Sprintf("digraph{\n%v\n%v\n}", strings.Join(V, "\n"), strings.Join(E, "\n")) }
func (emb *Embedding) Dotty(labels *digraph.Labels, attrs map[int]map[string]interface{}) string { V := make([]string, 0, len(emb.SG.V)) E := make([]string, 0, len(emb.SG.E)) // TODO: Replace this with strconv.Quote safeStr := func(i interface{}) string { s := fmt.Sprint(i) s = strings.Replace(s, "\n", "\\n", -1) s = strings.Replace(s, "\"", "\\\"", -1) return s } renderAttrs := func(color, id int) string { a := attrs[id] label := labels.Label(color) strs := make([]string, 0, len(a)+1) strs = append(strs, fmt.Sprintf(`idx="%v"`, id)) if line, has := a["start_line"]; has { strs = append(strs, fmt.Sprintf(`label="%v\n[line: %v]"`, safeStr(label), safeStr(line))) } else { strs = append(strs, fmt.Sprintf(`label="%v"`, safeStr(label))) } for name, value := range a { if name == "label" || name == "id" { continue } strs = append(strs, fmt.Sprintf("%v=\"%v\"", name, safeStr(value))) } return strings.Join(strs, ",") } for idx, id := range emb.Ids { V = append(V, fmt.Sprintf( "%v [%v];", id, renderAttrs(emb.SG.V[idx].Color, id), )) } for idx := range emb.SG.E { e := &emb.SG.E[idx] E = append(E, fmt.Sprintf( "%v -> %v [label=\"%v\"];", emb.Ids[e.Src], emb.Ids[e.Targ], safeStr(labels.Label(e.Color)), )) } return fmt.Sprintf( `digraph { %v %v } `, strings.Join(V, "\n "), strings.Join(E, "\n ")) }
func (sg *SubGraph) Pretty(labels *digraph.Labels) string { V := make([]string, 0, len(sg.V)) E := make([]string, 0, len(sg.E)) for _, v := range sg.V { V = append(V, fmt.Sprintf( "(%v)", labels.Label(v.Color), )) } for _, e := range sg.E { E = append(E, fmt.Sprintf( "[%v->%v:%v]", e.Src, e.Targ, labels.Label(e.Color), )) } return fmt.Sprintf("{%v:%v}%v%v", len(sg.E), len(sg.V), strings.Join(V, ""), strings.Join(E, "")) }