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_ETS_NEW_ARRAY_INSTANCE_EXPRESSION_H 17 #define ES2PANDA_IR_ETS_NEW_ARRAY_INSTANCE_EXPRESSION_H 18 19 #include "ir/expression.h" 20 #include "checker/types/type.h" 21 22 namespace ark::es2panda::checker { 23 class ETSAnalyzer; 24 class Signature; 25 } // namespace ark::es2panda::checker 26 27 namespace ark::es2panda::compiler { 28 class ETSCompiler; 29 } // namespace ark::es2panda::compiler 30 31 namespace ark::es2panda::ir { 32 33 class ETSNewArrayInstanceExpression : public Expression { 34 public: 35 ETSNewArrayInstanceExpression() = delete; 36 ~ETSNewArrayInstanceExpression() override = default; 37 38 NO_COPY_SEMANTIC(ETSNewArrayInstanceExpression); 39 NO_MOVE_SEMANTIC(ETSNewArrayInstanceExpression); 40 ETSNewArrayInstanceExpression(ir::TypeNode * const typeReference,ir::Expression * const dimension)41 explicit ETSNewArrayInstanceExpression(ir::TypeNode *const typeReference, ir::Expression *const dimension) 42 : Expression(AstNodeType::ETS_NEW_ARRAY_INSTANCE_EXPRESSION), 43 typeReference_(typeReference), 44 dimension_(dimension) 45 { 46 } 47 TypeReference()48 [[nodiscard]] ir::TypeNode *TypeReference() noexcept 49 { 50 return typeReference_; 51 } 52 TypeReference()53 [[nodiscard]] ir::TypeNode const *TypeReference() const noexcept 54 { 55 return typeReference_; 56 } 57 Dimension()58 [[nodiscard]] ir::Expression *Dimension() noexcept 59 { 60 return dimension_; 61 } 62 Dimension()63 [[nodiscard]] ir::Expression const *Dimension() const noexcept 64 { 65 return dimension_; 66 } 67 Signature()68 [[nodiscard]] checker::Signature *Signature() const noexcept 69 { 70 return defaultConstructorSignature_; 71 } 72 Signature()73 [[nodiscard]] checker::Signature *Signature() noexcept 74 { 75 return defaultConstructorSignature_; 76 } 77 SetPreferredType(checker::Type * preferredType)78 void SetPreferredType(checker::Type *preferredType) noexcept 79 { 80 preferredType_ = preferredType; 81 } 82 GetPreferredType()83 [[nodiscard]] checker::Type *GetPreferredType() noexcept 84 { 85 return preferredType_; 86 } 87 GetPreferredType()88 [[nodiscard]] checker::Type const *GetPreferredType() const noexcept 89 { 90 return preferredType_; 91 } 92 SetDimension(ir::Expression * dimension)93 void SetDimension(ir::Expression *dimension) noexcept 94 { 95 dimension_ = dimension; 96 if (dimension_ != nullptr) { 97 dimension_->SetParent(this); 98 } 99 } 100 SetSignature(checker::Signature * signature)101 void SetSignature(checker::Signature *signature) noexcept 102 { 103 defaultConstructorSignature_ = signature; 104 } 105 106 [[nodiscard]] ETSNewArrayInstanceExpression *Clone(ArenaAllocator *allocator, AstNode *parent) override; 107 108 void TransformChildren(const NodeTransformer &cb, std::string_view transformationName) override; 109 void Iterate(const NodeTraverser &cb) const override; 110 void Dump(ir::AstDumper *dumper) const override; 111 void Dump(ir::SrcDumper *dumper) const override; 112 void Compile(compiler::PandaGen *pg) const override; 113 void Compile(compiler::ETSGen *etsg) const override; 114 checker::Type *Check(checker::TSChecker *checker) override; 115 checker::VerifiedType Check(checker::ETSChecker *checker) override; 116 void SetPreferredTypeBasedOnFuncParam(checker::ETSChecker *checker, checker::Type *param, 117 checker::TypeRelationFlag flags); 118 Accept(ASTVisitorT * v)119 void Accept(ASTVisitorT *v) override 120 { 121 v->Accept(this); 122 } 123 CleanUp()124 void CleanUp() override 125 { 126 AstNode::CleanUp(); 127 defaultConstructorSignature_ = nullptr; 128 } 129 130 void ClearPreferredType(); 131 132 private: 133 ir::TypeNode *typeReference_; 134 ir::Expression *dimension_; 135 checker::Signature *defaultConstructorSignature_ {}; 136 checker::Type *preferredType_ {}; 137 }; 138 } // namespace ark::es2panda::ir 139 140 #endif 141