1 /** 2 * Copyright (c) 2021 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_STATEMENT_FOR_UPDATE_STATEMENT_H 17 #define ES2PANDA_IR_STATEMENT_FOR_UPDATE_STATEMENT_H 18 19 #include "ir/statements/loopStatement.h" 20 21 namespace panda::es2panda::varbinder { 22 class LoopScope; 23 } // namespace panda::es2panda::varbinder 24 25 namespace panda::es2panda::ir { 26 class Expression; 27 28 class ForUpdateStatement : public LoopStatement { 29 public: ForUpdateStatement(AstNode * init,Expression * test,Expression * update,Statement * body)30 explicit ForUpdateStatement(AstNode *init, Expression *test, Expression *update, Statement *body) 31 : LoopStatement(AstNodeType::FOR_UPDATE_STATEMENT), init_(init), test_(test), update_(update), body_(body) 32 { 33 } 34 Init()35 AstNode *Init() 36 { 37 return init_; 38 } 39 Init()40 const AstNode *Init() const 41 { 42 return init_; 43 } 44 Test()45 Expression *Test() 46 { 47 return test_; 48 } 49 Test()50 const Expression *Test() const 51 { 52 return test_; 53 } 54 Update()55 Expression *Update() 56 { 57 return update_; 58 } 59 Update()60 const Expression *Update() const 61 { 62 return update_; 63 } 64 Body()65 Statement *Body() 66 { 67 return body_; 68 } 69 Body()70 const Statement *Body() const 71 { 72 return body_; 73 } 74 75 void TransformChildren(const NodeTransformer &cb) override; SetReturnType(checker::ETSChecker * checker,checker::Type * type)76 void SetReturnType(checker::ETSChecker *checker, checker::Type *type) override 77 { 78 if (body_ != nullptr) { 79 body_->SetReturnType(checker, type); 80 } 81 } 82 83 void Iterate(const NodeTraverser &cb) const override; 84 void Dump(ir::AstDumper *dumper) const override; 85 void Dump(ir::SrcDumper *dumper) const override; 86 void Compile(compiler::PandaGen *pg) const override; 87 void Compile(compiler::ETSGen *etsg) const override; 88 checker::Type *Check(checker::TSChecker *checker) override; 89 checker::Type *Check(checker::ETSChecker *checker) override; 90 Accept(ASTVisitorT * v)91 void Accept(ASTVisitorT *v) override 92 { 93 v->Accept(this); 94 } 95 96 private: 97 AstNode *init_; 98 Expression *test_; 99 Expression *update_; 100 Statement *body_; 101 }; 102 } // namespace panda::es2panda::ir 103 104 #endif 105