unilex/example/math_expression/main.go

55 lines
933 B
Go
Raw Normal View History

2021-03-06 22:30:32 +03:00
package main
import (
"fmt"
"github.com/neonxp/unilex"
)
var output []unilex.Lexem = []unilex.Lexem{}
var opPriority = map[string]int{
"^": 3,
"!": 3,
"*": 2,
"/": 2,
"+": 1,
"-": 1,
}
func main() {
l := unilex.New("10 * (20.0 + 30.0)")
go l.Run(lexExpression) // Start lexer
// Read infix expression lexems from lexer and convert them to RPN (reverse polish notation)
rpn := infixToRPNotation(l)
fmt.Println("RPN:", rpn)
// Calculate RPN
result := calculateRPN(rpn)
fmt.Println("Result:", result)
}
func lexExpression(l *unilex.Lexer) unilex.StateFunc {
l.AcceptWhile(" \t")
l.Ignore() // Ignore whitespaces
switch {
case l.Accept("("):
l.Emit("LP")
case l.Accept(")"):
l.Emit("RP")
case unilex.ScanNumber(l):
l.Emit("NUMBER")
case l.Accept("+-*/^!"):
l.Emit("OPERATOR")
case l.Peek() == unilex.EOF:
return nil
default:
return l.Errorf("Unexpected symbol")
}
return lexExpression
}