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
}
