func (v *VegLoader) loadVertex(labels *digraph.Labels, b *baseLoader, data []byte) (err error) { obj, err := parseJson(data) if err != nil { return err } _id, err := obj["id"].(json.Number).Int64() if err != nil { return err } label := strings.TrimSpace(obj["label"].(string)) id := int32(_id) return b.addVertex(id, labels.Color(label), label, obj) }
func (v *VegLoader) loadEdge(labels *digraph.Labels, b *baseLoader, data []byte) (err error) { obj, err := parseJson(data) if err != nil { return err } _src, err := obj["src"].(json.Number).Int64() if err != nil { return err } _targ, err := obj["targ"].(json.Number).Int64() if err != nil { return err } src := int32(_src) targ := int32(_targ) label := strings.TrimSpace(obj["label"].(string)) return b.addEdge(src, targ, labels.Color(label), label) }
func ParsePretty(str string, labels *digraph.Labels) (*SubGraph, error) { size := regexp.MustCompile(`^\{([0-9]+):([0-9]+)\}`) edge := regexp.MustCompile(`^\[([0-9]+)->([0-9]+):`) matches := size.FindStringSubmatch(str) E, err := strconv.ParseInt(matches[1], 10, 64) if err != nil { return nil, err } V, err := strconv.ParseInt(matches[2], 10, 64) if err != nil { return nil, err } idx := len(matches[0]) vertices := make(Vertices, V) for i := range vertices { if str[idx] != '(' { return nil, errors.Errorf("expected ( got %v", str[idx:idx+1]) } idx++ parens := 1 labelc := make([]byte, 0, 10) for ; idx < len(str); idx++ { if str[idx] == '\\' { continue } else if str[idx-1] == '\\' { labelc = append(labelc, str[idx]) } else if str[idx] == '(' { parens += 1 labelc = append(labelc, str[idx]) } else if str[idx] == ')' { parens -= 1 if parens > 0 { labelc = append(labelc, str[idx]) } else { idx++ break } } else { labelc = append(labelc, str[idx]) } } label := string(labelc) vertices[i].Idx = i vertices[i].Color = labels.Color(label) } edges := make(Edges, E) for i := range edges { matches := edge.FindStringSubmatch(str[idx:]) src, err := strconv.ParseInt(matches[1], 10, 64) if err != nil { return nil, err } targ, err := strconv.ParseInt(matches[2], 10, 64) if err != nil { return nil, err } labelc := make([]byte, 0, 10) idx += len(matches[0]) for ; idx < len(str); idx++ { if str[idx] == '\\' { continue } else if str[idx-1] == '\\' { labelc = append(labelc, str[idx]) } else if str[idx] == ']' { idx++ break } else { labelc = append(labelc, str[idx]) } } label := string(labelc) edges[i].Src = int(src) edges[i].Targ = int(targ) edges[i].Color = labels.Color(label) } sg := &Builder{V: vertices, E: edges} return sg.Build(), nil }