-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathtree.go
47 lines (45 loc) · 889 Bytes
/
tree.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package peg
import(
//"fmt"
)
type (
ExpressionTree struct {
Name string
Children []*ExpressionTree
Value int
}
)
// collapse the expression tree (removing anything that isn't a NonTerminal or
// a Terminal)
func (this *ExpressionTree) collapse() {
ncs := make([]*ExpressionTree, 0)
for _, c := range this.Children {
c.collapse()
if c.Name == "" && c.Value == -1 {
for _, gc := range c.Children {
ncs = append(ncs, gc)
}
} else {
ncs = append(ncs, c)
}
}
this.Children = ncs
}
func (this *ExpressionTree) toString(indent int) string {
str := ""
for i := 0; i < indent; i++ {
str += " "
}
if this.Value > -1 {
str += "<" + string(this.Value) + ">"
} else {
str += this.Name
}
for _, t := range this.Children {
str += "\n" + t.toString(indent + 2)
}
return str
}
func (this *ExpressionTree) String() string {
return this.toString(0)
}