You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.4 KiB
85 lines
2.4 KiB
package grammar
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/brettlangdon/gython/symbol"
|
|
"github.com/brettlangdon/gython/token"
|
|
)
|
|
|
|
type Node interface {
|
|
ID() symbol.SymbolID
|
|
Name() string
|
|
Repr() []interface{}
|
|
}
|
|
|
|
type TokenNode struct {
|
|
Token *token.Token
|
|
}
|
|
|
|
func NewTokenNode(tok *token.Token) *TokenNode {
|
|
return &TokenNode{
|
|
Token: tok,
|
|
}
|
|
}
|
|
func (node *TokenNode) atomChild() {}
|
|
func (node *TokenNode) atomExpressionChild() {}
|
|
func (node *TokenNode) comparisonChild() {}
|
|
func (node *TokenNode) expressionStatementChild() {}
|
|
func (node *TokenNode) factorChild() {}
|
|
func (node *TokenNode) fileInputChild() {}
|
|
func (node *TokenNode) shiftExpressionChild() {}
|
|
func (node *TokenNode) simpleStatementChild() {}
|
|
func (node *TokenNode) trailerChild() {}
|
|
func (node *TokenNode) ID() symbol.SymbolID { return 0 }
|
|
func (node *TokenNode) Name() string { return token.TokenNames[node.Token.ID] }
|
|
func (node *TokenNode) Repr() []interface{} {
|
|
parts := make([]interface{}, 0)
|
|
parts = append(parts, node.Name())
|
|
literal := fmt.Sprintf("%#v", node.Token.Literal)
|
|
return append(parts, literal)
|
|
}
|
|
|
|
type BaseNode struct {
|
|
id symbol.SymbolID
|
|
child Node
|
|
}
|
|
|
|
func (node *BaseNode) initBaseNode(id symbol.SymbolID) { node.id = id }
|
|
func (node *BaseNode) ID() symbol.SymbolID { return node.id }
|
|
func (node *BaseNode) Name() string { return symbol.SymbolNames[node.ID()] }
|
|
func (node *BaseNode) Repr() (parts []interface{}) { return append(parts, node.Name()) }
|
|
|
|
type ParentNode struct {
|
|
BaseNode
|
|
child Node
|
|
}
|
|
|
|
func (node *ParentNode) SetChild(n Node) { node.child = n }
|
|
func (node *ParentNode) Child() Node { return node.child }
|
|
func (node *ParentNode) Repr() (parts []interface{}) {
|
|
parts = node.BaseNode.Repr()
|
|
child := node.Child()
|
|
if child != nil {
|
|
parts = append(parts, child.Repr())
|
|
}
|
|
return parts
|
|
}
|
|
|
|
type ListNode struct {
|
|
BaseNode
|
|
children []Node
|
|
}
|
|
|
|
func (node *ListNode) initListNode() { node.children = make([]Node, 0) }
|
|
func (node *ListNode) Length() int { return len(node.children) }
|
|
func (node *ListNode) Children() []Node { return node.children }
|
|
func (node *ListNode) Append(n Node) { node.children = append(node.children, n) }
|
|
func (node *ListNode) Repr() (parts []interface{}) {
|
|
parts = node.BaseNode.Repr()
|
|
children := node.Children()
|
|
for _, child := range children {
|
|
parts = append(parts, child.Repr())
|
|
}
|
|
return parts
|
|
}
|