| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | // Copyright (c) 2021-2025 ChilliBits. All rights reserved. | ||
| 2 | |||
| 3 | #pragma once | ||
| 4 | |||
| 5 | #include <irgenerator/LLVMExprResult.h> | ||
| 6 | #include <symboltablebuilder/QualType.h> | ||
| 7 | |||
| 8 | #include <llvm/IR/IRBuilder.h> | ||
| 9 | #include <llvm/IR/Value.h> | ||
| 10 | |||
| 11 | namespace spice::compiler { | ||
| 12 | |||
| 13 | // Forward declarations | ||
| 14 | class SourceFile; | ||
| 15 | class IRGenerator; | ||
| 16 | class StdFunctionManager; | ||
| 17 | class SymbolTableEntry; | ||
| 18 | struct CodeLoc; | ||
| 19 | |||
| 20 | // Typedefs | ||
| 21 | using ResolverFct = const std::function<llvm::Value *()>; | ||
| 22 | |||
| 23 | #define COMB(en1, en2) ((en1) | ((en2) << 16)) | ||
| 24 | |||
| 25 | /** | ||
| 26 | * Helper class for the IRGenerator to resolve operator instructions for each valid operator/type combination | ||
| 27 | */ | ||
| 28 | class OpRuleConversionManager { | ||
| 29 | public: | ||
| 30 | // Constructors | ||
| 31 | OpRuleConversionManager(SourceFile *sourceFile, IRGenerator *irGenerator); | ||
| 32 | |||
| 33 | // Public methods | ||
| 34 | LLVMExprResult getPlusEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 35 | size_t opIdx); | ||
| 36 | LLVMExprResult getMinusEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 37 | QualType rhsSTy, size_t opIdx); | ||
| 38 | LLVMExprResult getMulEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 39 | size_t opIdx); | ||
| 40 | LLVMExprResult getDivEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 41 | size_t opIdx); | ||
| 42 | LLVMExprResult getRemEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 43 | LLVMExprResult getSHLEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 44 | LLVMExprResult getSHREqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 45 | LLVMExprResult getAndEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 46 | LLVMExprResult getOrEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 47 | LLVMExprResult getXorEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 48 | LLVMExprResult getBitwiseOrInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 49 | size_t opIdx); | ||
| 50 | LLVMExprResult getBitwiseXorInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 51 | QualType rhsSTy); | ||
| 52 | LLVMExprResult getBitwiseAndInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 53 | QualType rhsSTy, size_t opIdx); | ||
| 54 | LLVMExprResult getEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 55 | size_t opIdx); | ||
| 56 | LLVMExprResult getNotEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 57 | size_t opIdx); | ||
| 58 | LLVMExprResult getLessInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 59 | LLVMExprResult getGreaterInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy); | ||
| 60 | LLVMExprResult getLessEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 61 | QualType rhsSTy); | ||
| 62 | LLVMExprResult getGreaterEqualInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 63 | QualType rhsSTy); | ||
| 64 | LLVMExprResult getShiftLeftInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 65 | size_t opIdx); | ||
| 66 | LLVMExprResult getShiftRightInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 67 | QualType rhsSTy, size_t opIdx); | ||
| 68 | LLVMExprResult getPlusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 69 | size_t opIdx); | ||
| 70 | LLVMExprResult getMinusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 71 | size_t opIdx); | ||
| 72 | LLVMExprResult getMulInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 73 | size_t opIdx); | ||
| 74 | LLVMExprResult getDivInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy, | ||
| 75 | size_t opIdx); | ||
| 76 | LLVMExprResult getRemInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, LLVMExprResult &rhs, | ||
| 77 | QualType rhsSTy) const; | ||
| 78 | LLVMExprResult getPrefixMinusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy) const; | ||
| 79 | LLVMExprResult getPrefixPlusPlusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy) const; | ||
| 80 | LLVMExprResult getPrefixMinusMinusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy) const; | ||
| 81 | LLVMExprResult getPrefixNotInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy) const; | ||
| 82 | LLVMExprResult getPrefixBitwiseNotInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy) const; | ||
| 83 | LLVMExprResult getPostfixPlusPlusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, size_t opIdx); | ||
| 84 | LLVMExprResult getPostfixMinusMinusInst(const ASTNode *node, LLVMExprResult &lhs, QualType lhsSTy, size_t opIdx); | ||
| 85 | LLVMExprResult getCastInst(const ASTNode *node, QualType lhsSTy, LLVMExprResult &rhs, QualType rhsSTy) const; | ||
| 86 | |||
| 87 | // Operator overloading | ||
| 88 | bool callsOverloadedOpFct(const ASTNode *node, size_t opIdx) const; | ||
| 89 | template <size_t N> | ||
| 90 | LLVMExprResult callOperatorOverloadFct(const ASTNode *node, const std::array<ResolverFct, N * 2> &opV, size_t opIdx); | ||
| 91 | |||
| 92 | private: | ||
| 93 | // Members | ||
| 94 | llvm::LLVMContext &context; | ||
| 95 | llvm::IRBuilder<> &builder; | ||
| 96 | IRGenerator *irGenerator; | ||
| 97 | const StdFunctionManager &stdFunctionManager; | ||
| 98 | |||
| 99 | // Private methods | ||
| 100 | [[nodiscard]] llvm::Value *generateIToFp(const QualType &srcSTy, llvm::Value *srcV, llvm::Type *tgtT) const; | ||
| 101 | [[nodiscard]] llvm::Value *generateSHR(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 102 | llvm::Value *rhsV) const; | ||
| 103 | [[nodiscard]] llvm::Value *generateLT(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 104 | llvm::Value *rhsV) const; | ||
| 105 | [[nodiscard]] llvm::Value *generateGT(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 106 | llvm::Value *rhsV) const; | ||
| 107 | [[nodiscard]] llvm::Value *generateLE(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 108 | llvm::Value *rhsV) const; | ||
| 109 | [[nodiscard]] llvm::Value *generateGE(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 110 | llvm::Value *rhsV) const; | ||
| 111 | [[nodiscard]] llvm::Value *generateDiv(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 112 | llvm::Value *rhsV) const; | ||
| 113 | [[nodiscard]] llvm::Value *generateRem(const QualType &lhsSTy, const QualType &rhsSTy, llvm::Value *lhsV, | ||
| 114 | llvm::Value *rhsV) const; | ||
| 115 | 15388 | [[nodiscard]] static uint32_t getTypeCombination(const QualType &lhsTy, const QualType &rhsTy) { | |
| 116 | 15388 | return COMB(lhsTy.getSuperType(), rhsTy.getSuperType()); | |
| 117 | } | ||
| 118 | }; | ||
| 119 | |||
| 120 | } // namespace spice::compiler | ||
| 121 |