1 /** 2 * Copyright (c) 2021-2025 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef ES2PANDA_IR_EXPRESSION_BINARY_EXPRESSION_H 17 #define ES2PANDA_IR_EXPRESSION_BINARY_EXPRESSION_H 18 19 #include "checker/checkerContext.h" 20 #include "compiler/core/vReg.h" 21 #include "ir/expression.h" 22 23 namespace ark::es2panda::checker { 24 class ETSAnalyzer; 25 } // namespace ark::es2panda::checker 26 27 namespace ark::es2panda::ir { 28 class BinaryExpression : public Expression { 29 public: 30 BinaryExpression() = delete; 31 ~BinaryExpression() override = default; 32 33 NO_COPY_SEMANTIC(BinaryExpression); 34 NO_MOVE_SEMANTIC(BinaryExpression); 35 BinaryExpression(Expression * const left,Expression * const right,lexer::TokenType const operatorType)36 explicit BinaryExpression(Expression *const left, Expression *const right, lexer::TokenType const operatorType) 37 : Expression(AstNodeType::BINARY_EXPRESSION), left_(left), right_(right), operator_(operatorType) 38 { 39 } 40 41 // NOTE (csabahurton): friend relationship can be removed once there are getters for private fields 42 friend class checker::ETSAnalyzer; 43 Left()44 [[nodiscard]] const Expression *Left() const noexcept 45 { 46 return left_; 47 } 48 Left()49 [[nodiscard]] Expression *Left() noexcept 50 { 51 return left_; 52 } 53 Right()54 [[nodiscard]] const Expression *Right() const noexcept 55 { 56 return right_; 57 } 58 Right()59 [[nodiscard]] Expression *Right() noexcept 60 { 61 return right_; 62 } 63 Result()64 [[nodiscard]] const Expression *Result() const noexcept 65 { 66 return result_; 67 } 68 Result()69 [[nodiscard]] Expression *Result() noexcept 70 { 71 return result_; 72 } 73 OperatorType()74 [[nodiscard]] lexer::TokenType OperatorType() const noexcept 75 { 76 return operator_; 77 } 78 IsLogical()79 [[nodiscard]] bool IsLogical() const noexcept 80 { 81 return operator_ == lexer::TokenType::PUNCTUATOR_NULLISH_COALESCING || 82 operator_ == lexer::TokenType::PUNCTUATOR_LOGICAL_OR || 83 operator_ == lexer::TokenType::PUNCTUATOR_LOGICAL_AND; 84 } 85 IsLogicalExtended()86 [[nodiscard]] bool IsLogicalExtended() const noexcept 87 { 88 return operator_ == lexer::TokenType::PUNCTUATOR_LOGICAL_AND || 89 operator_ == lexer::TokenType::PUNCTUATOR_LOGICAL_OR; 90 } 91 IsBitwise()92 [[nodiscard]] bool IsBitwise() const noexcept 93 { 94 return operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_OR || 95 operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_XOR || 96 operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_AND || 97 operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_AND_EQUAL || 98 operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_OR_EQUAL || 99 operator_ == lexer::TokenType::PUNCTUATOR_BITWISE_XOR_EQUAL; 100 } 101 IsArithmetic()102 [[nodiscard]] bool IsArithmetic() const noexcept 103 { 104 return operator_ == lexer::TokenType::PUNCTUATOR_PLUS || operator_ == lexer::TokenType::PUNCTUATOR_MINUS || 105 operator_ == lexer::TokenType::PUNCTUATOR_MULTIPLY || operator_ == lexer::TokenType::PUNCTUATOR_DIVIDE || 106 operator_ == lexer::TokenType::PUNCTUATOR_MOD || operator_ == lexer::TokenType::PUNCTUATOR_PLUS_EQUAL || 107 operator_ == lexer::TokenType::PUNCTUATOR_MINUS_EQUAL || 108 operator_ == lexer::TokenType::PUNCTUATOR_MULTIPLY_EQUAL || 109 operator_ == lexer::TokenType::PUNCTUATOR_DIVIDE_EQUAL || 110 operator_ == lexer::TokenType::PUNCTUATOR_MOD_EQUAL || IsBitwise(); 111 } 112 SetLeft(Expression * expr)113 void SetLeft(Expression *expr) noexcept 114 { 115 left_ = expr; 116 ES2PANDA_ASSERT(left_); 117 left_->SetParent(this); 118 SetStart(left_->Start()); 119 } 120 SetRight(Expression * expr)121 void SetRight(Expression *expr) noexcept 122 { 123 right_ = expr; 124 ES2PANDA_ASSERT(right_); 125 right_->SetParent(this); 126 SetEnd(right_->End()); 127 } 128 SetResult(Expression * expr)129 void SetResult(Expression *expr) noexcept 130 { 131 result_ = expr; 132 result_->SetParent(this); 133 SetStart(result_->Start()); 134 } 135 SetOperator(lexer::TokenType operatorType)136 void SetOperator(lexer::TokenType operatorType) noexcept 137 { 138 operator_ = operatorType; 139 type_ = AstNodeType::BINARY_EXPRESSION; 140 } 141 OperationType()142 [[nodiscard]] checker::Type *OperationType() noexcept 143 { 144 return operationType_; 145 } 146 SetOperationType(checker::Type * const operationType)147 void SetOperationType(checker::Type *const operationType) noexcept 148 { 149 operationType_ = operationType; 150 } 151 OperationType()152 [[nodiscard]] const checker::Type *OperationType() const noexcept 153 { 154 return operationType_; 155 } 156 157 [[nodiscard]] BinaryExpression *Clone(ArenaAllocator *allocator, AstNode *parent) override; 158 159 void TransformChildren(const NodeTransformer &cb, std::string_view transformationName) override; 160 void Iterate(const NodeTraverser &cb) const override; 161 void Dump(ir::AstDumper *dumper) const override; 162 void Dump(ir::SrcDumper *dumper) const override; 163 void Compile(compiler::PandaGen *pg) const override; 164 void Compile(compiler::ETSGen *etsg) const override; 165 void CompileOperands(compiler::ETSGen *etsg, compiler::VReg lhs) const; 166 checker::Type *Check(checker::TSChecker *checker) override; 167 checker::VerifiedType Check(checker::ETSChecker *checker) override; 168 Accept(ASTVisitorT * v)169 void Accept(ASTVisitorT *v) override 170 { 171 v->Accept(this); 172 } 173 CleanUp()174 void CleanUp() override 175 { 176 AstNode::CleanUp(); 177 operationType_ = nullptr; 178 } 179 180 private: 181 Expression *left_ = nullptr; 182 Expression *right_ = nullptr; 183 Expression *result_ = nullptr; 184 lexer::TokenType operator_; 185 checker::Type *operationType_ {}; 186 }; 187 } // namespace ark::es2panda::ir 188 189 #endif 190