From 1b524f75401a6318fbaf3fee0bfcbe59348f2e51 Mon Sep 17 00:00:00 2001 From: _ov4 Date: Mon, 2 Feb 2026 10:08:22 +0000 Subject: Added STRING --- include/cps/AST.h | 7 +++++++ include/cps/CodeGen.h | 3 +++ include/cps/Lexer.h | 9 +++++++++ include/cps/Parser.h | 3 +++ include/cps/StringHandler.h | 38 ++++++++++++++++++++++++++++++++++++++ 5 files changed, 60 insertions(+) create mode 100644 include/cps/StringHandler.h (limited to 'include/cps') diff --git a/include/cps/AST.h b/include/cps/AST.h index de57024..f07e937 100644 --- a/include/cps/AST.h +++ b/include/cps/AST.h @@ -32,6 +32,13 @@ public: bool getVal() const { return Val; } }; +class StringExprAST : public ExprAST { + std::string Val; +public: + StringExprAST(const std::string &Val) : Val(Val) {} + const std::string &getVal() const { return Val; } +}; + class VariableExprAST : public ExprAST { std::string Name; public: diff --git a/include/cps/CodeGen.h b/include/cps/CodeGen.h index df9eef4..cb8cecb 100644 --- a/include/cps/CodeGen.h +++ b/include/cps/CodeGen.h @@ -10,6 +10,7 @@ #include "cps/RealHandler.h" #include "cps/BooleanHandler.h" #include "cps/ArithmeticHandler.h" +#include "cps/StringHandler.h" #include #include @@ -32,6 +33,7 @@ class CodeGen { std::unique_ptr RealHelper; std::unique_ptr BoolHandler; std::unique_ptr ArithHandler; + std::unique_ptr StrHandler; llvm::FunctionCallee PrintfFunc; llvm::FunctionCallee ScanfFunc; @@ -42,6 +44,7 @@ class CodeGen { llvm::Value *ScanfFormatStr; // %lld llvm::Value *ScanfFloatFormatStr; // %lf + llvm::Value *ScanfStringFormatStr; // %s llvm::Value *TrueStr; // "TRUE" llvm::Value *FalseStr; // "FALSE" diff --git a/include/cps/Lexer.h b/include/cps/Lexer.h index 97ccd83..c36599f 100644 --- a/include/cps/Lexer.h +++ b/include/cps/Lexer.h @@ -41,6 +41,14 @@ enum Token { tok_or = -26, tok_not = -27, + tok_string_kw = -30, + tok_length = -31, + tok_mid = -32, + tok_right = -33, + tok_lcase = -34, + tok_ucase = -35, + tok_string_literal = -36, + tok_identifier = -100, tok_number_int = -101, tok_number_real = -102, @@ -71,6 +79,7 @@ class Lexer { int CurrentLine = 1; public: std::string IdentifierStr; + std::string StringVal; int64_t NumVal; double RealVal; diff --git a/include/cps/Parser.h b/include/cps/Parser.h index db1f243..877c684 100644 --- a/include/cps/Parser.h +++ b/include/cps/Parser.h @@ -4,6 +4,7 @@ #include #include #include +#include namespace cps { @@ -24,6 +25,8 @@ class Parser { std::unique_ptr ParseNumberExpr(); std::unique_ptr ParseIdentifierExpr(); std::unique_ptr ParseParenExpr(); + + std::unique_ptr ParseStringBuiltin(const std::string &FuncName); std::unique_ptr ParseStatement(); std::unique_ptr ParseIfStmt(); diff --git a/include/cps/StringHandler.h b/include/cps/StringHandler.h new file mode 100644 index 0000000..87710fb --- /dev/null +++ b/include/cps/StringHandler.h @@ -0,0 +1,38 @@ +#pragma once +#include "llvm/IR/IRBuilder.h" +#include "llvm/IR/LLVMContext.h" +#include "llvm/IR/Module.h" +#include +#include + +namespace cps { + +class StringHandler { + llvm::LLVMContext &Context; + llvm::IRBuilder<> &Builder; + llvm::Module &Module; + std::map &NamedValues; + + llvm::FunctionCallee MallocFunc; + llvm::FunctionCallee FreeFunc; + llvm::FunctionCallee StrLenFunc; + llvm::FunctionCallee MemCpyFunc; + llvm::FunctionCallee ToUpperFunc; + llvm::FunctionCallee ToLowerFunc; + +public: + StringHandler(llvm::LLVMContext &Ctx, llvm::IRBuilder<> &B, llvm::Module &M, + std::map &NV); + + void setupExternalFunctions(); + void emitDeclare(const std::string &Name); + llvm::Value *createLiteral(const std::string &Val); + + llvm::Value *emitLength(llvm::Value *Str); + llvm::Value *emitMid(llvm::Value *Str, llvm::Value *Start, llvm::Value *Len); + llvm::Value *emitRight(llvm::Value *Str, llvm::Value *Len); + llvm::Value *emitLCase(llvm::Value *Str); + llvm::Value *emitUCase(llvm::Value *Str); +}; + +} -- cgit v1.3.1-4-g156e