aboutsummaryrefslogtreecommitdiff
path: root/src/parser/Statement.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/parser/Statement.hs')
-rw-r--r--src/parser/Statement.hs11
1 files changed, 7 insertions, 4 deletions
diff --git a/src/parser/Statement.hs b/src/parser/Statement.hs
index ca16eca..889f24f 100644
--- a/src/parser/Statement.hs
+++ b/src/parser/Statement.hs
@@ -9,19 +9,22 @@ import Parser.Expr
data Statement
= Evaluation Expr
+ | PolynomEvaluation Expr Expr
| VariableDeclaration String Expr
| FunctionDeclaration String String Expr
statementP :: Parser Statement
-statementP = functionDeclarationP <|> variableDeclarationP <|> evaluationP
+statementP = functionDeclarationP <|> variableDeclarationP <|> polynomEvaluationP <|> evaluationP
where
functionDeclarationP = FunctionDeclaration
- <$> alphaStringP
- <*> parenthesis alphaStringP
+ <$> labelP
+ <*> parenthesis labelP
<*> (char '=' *> exprP)
variableDeclarationP = VariableDeclaration
- <$> alphaStringP
+ <$> labelP
<*> (char '=' *> exprP)
+ polynomEvaluationP = PolynomEvaluation <$> exprP <*> (char '=' *> exprP <* char '?')
+
evaluationP = Evaluation <$> exprP <* char '=' <* char '?'