diff options
Diffstat (limited to 'lib/Parser/Parser_Function.cc')
| -rw-r--r-- | lib/Parser/Parser_Function.cc | 129 |
1 files changed, 0 insertions, 129 deletions
diff --git a/lib/Parser/Parser_Function.cc b/lib/Parser/Parser_Function.cc deleted file mode 100644 index bdb4ee7..0000000 --- a/lib/Parser/Parser_Function.cc +++ /dev/null @@ -1,129 +0,0 @@ -#include "cps/Parser.h" -#include "cps/FunctionAST.h" - -using namespace cps; - -std::vector<std::pair<std::string, std::string>> Parser::ParsePrototypeArgs() { - std::vector<std::pair<std::string, std::string>> Args; - if (CurTok != '(') return Args; - getNextToken(); - - while (CurTok != ')') { - if (CurTok == -1000 || CurTok == -1001) getNextToken(); - - if (CurTok != tok_identifier) { - fprintf(stderr, "Error: Expected argument name\n"); - return Args; - } - std::string Name = Lex.IdentifierStr; - getNextToken(); - - if (CurTok != tok_colon) { - fprintf(stderr, "Error: Expected ':' after argument name\n"); - return Args; - } - getNextToken(); - - std::string Type = "INTEGER"; - if (CurTok == tok_integer_kw) { - getNextToken(); - } else if (CurTok == tok_identifier) { - Type = Lex.IdentifierStr; - getNextToken(); - } else { - fprintf(stderr, "Error: Expected type (e.g. INTEGER) for argument '%s'\n", Name.c_str()); - return Args; - } - - Args.push_back({Name, Type}); - - if (CurTok == ')') break; - if (CurTok != ',') { - fprintf(stderr, "Error: Expected ',' or ')'\n"); - return Args; - } - getNextToken(); - } - getNextToken(); - return Args; -} - - -std::unique_ptr<StmtAST> Parser::ParseFunction() { - getNextToken(); - std::string Name = Lex.IdentifierStr; - getNextToken(); - - auto Args = ParsePrototypeArgs(); - - std::string RetType = "INTEGER"; - if (CurTok == tok_returns) { - getNextToken(); - getNextToken(); - } - - auto Proto = std::make_unique<PrototypeAST>(Name, std::move(Args), RetType); - - std::vector<std::unique_ptr<StmtAST>> Body; - while (CurTok != tok_endfunction && CurTok != tok_eof) { - auto Stmt = ParseStatement(); - if (Stmt) Body.push_back(std::move(Stmt)); - else if (CurTok != tok_eof && CurTok != tok_endfunction) getNextToken(); - } - getNextToken(); - - return std::make_unique<FunctionDefAST>(std::move(Proto), std::move(Body)); -} - -std::unique_ptr<StmtAST> Parser::ParseProcedure() { - getNextToken(); - std::string Name = Lex.IdentifierStr; - getNextToken(); - - auto Args = ParsePrototypeArgs(); - auto Proto = std::make_unique<PrototypeAST>(Name, std::move(Args), "VOID"); - - std::vector<std::unique_ptr<StmtAST>> Body; - while (CurTok != tok_endprocedure && CurTok != tok_eof) { - auto Stmt = ParseStatement(); - if (Stmt) Body.push_back(std::move(Stmt)); - else if (CurTok != tok_eof && CurTok != tok_endprocedure) getNextToken(); - } - getNextToken(); - - return std::make_unique<FunctionDefAST>(std::move(Proto), std::move(Body)); -} - -std::unique_ptr<StmtAST> Parser::ParseCallStmt() { - getNextToken(); - std::string Callee = Lex.IdentifierStr; - getNextToken(); - - std::vector<std::unique_ptr<ExprAST>> Args; - if (CurTok == '(') { - getNextToken(); - while (CurTok != ')') { - auto Arg = ParseExpression(); - if (Arg) Args.push_back(std::move(Arg)); - - if (CurTok == ')') break; - if (CurTok != ',') { - fprintf(stderr, "Error: Expected ',' or ')' in call\n"); - return nullptr; - } - getNextToken(); - } - getNextToken(); - } - - return std::make_unique<CallStmtAST>(Callee, std::move(Args)); -} - -std::unique_ptr<StmtAST> Parser::ParseReturnStmt() { - getNextToken(); - std::unique_ptr<ExprAST> Expr = nullptr; - if (CurTok != tok_endif && CurTok != tok_else && CurTok != tok_endfunction && CurTok != tok_endprocedure) { - Expr = ParseExpression(); - } - return std::make_unique<ReturnStmtAST>(std::move(Expr)); -} |
