1 /**
2 * Copyright (c) 2021-2024 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 #include "blockStatement.h"
17
18 #include "compiler/core/pandagen.h"
19 #include "compiler/core/regScope.h"
20 #include "compiler/core/ETSGen.h"
21 #include "checker/TSchecker.h"
22 #include "checker/ETSchecker.h"
23 #include "ir/astDump.h"
24 #include "ir/srcDump.h"
25
26 namespace ark::es2panda::ir {
TransformChildren(const NodeTransformer &cb, std::string_view const transformationName)27 void BlockStatement::TransformChildren(const NodeTransformer &cb, std::string_view const transformationName)
28 {
29 for (auto *&it : statements_) {
30 if (auto *transformedNode = cb(it); it != transformedNode) {
31 it->SetTransformedNode(transformationName, transformedNode);
32 it = transformedNode->AsStatement();
33 }
34 }
35 }
36
Clone(ArenaAllocator *const allocator, AstNode *const parent)37 AstNode *BlockStatement::Clone(ArenaAllocator *const allocator, AstNode *const parent)
38 {
39 ArenaVector<Statement *> statements(allocator->Adapter());
40
41 for (auto *statement : this->statements_) {
42 statements.push_back(statement->Clone(allocator, parent)->AsStatement());
43 }
44
45 auto retVal = util::NodeAllocator::ForceSetParent<ir::BlockStatement>(allocator, allocator, std::move(statements));
46 retVal->SetParent(parent);
47
48 return retVal;
49 }
50
Iterate(const NodeTraverser &cb) const51 void BlockStatement::Iterate(const NodeTraverser &cb) const
52 {
53 // This will survive pushing element to the back of statements_ in the process
54 // NOLINTNEXTLINE(modernize-loop-convert)
55 for (size_t ix = 0; ix < statements_.size(); ix++) {
56 cb(statements_[ix]);
57 }
58 }
59
Dump(ir::AstDumper *dumper) const60 void BlockStatement::Dump(ir::AstDumper *dumper) const
61 {
62 dumper->Add({{"type", IsProgram() ? "Program" : "BlockStatement"}, {"statements", statements_}});
63 }
64
Dump(ir::SrcDumper *dumper) const65 void BlockStatement::Dump(ir::SrcDumper *dumper) const
66 {
67 // NOTE(nsizov): trailing blocks
68 if (Parent() != nullptr && (Parent()->IsBlockStatement() || Parent()->IsCallExpression())) {
69 dumper->Add("{");
70 if (!statements_.empty()) {
71 dumper->IncrIndent();
72 dumper->Endl();
73 }
74 }
75 for (auto statement : statements_) {
76 statement->Dump(dumper);
77 if (statement != statements_.back()) {
78 dumper->Endl();
79 }
80 }
81 if (Parent() != nullptr && (Parent()->IsBlockStatement() || Parent()->IsCallExpression())) {
82 if (!statements_.empty()) {
83 dumper->DecrIndent();
84 dumper->Endl();
85 }
86 dumper->Add("}");
87 }
88 }
89
Compile([[maybe_unused]] compiler::PandaGen *pg) const90 void BlockStatement::Compile([[maybe_unused]] compiler::PandaGen *pg) const
91 {
92 pg->GetAstCompiler()->Compile(this);
93 }
94
Compile([[maybe_unused]] compiler::ETSGen *etsg) const95 void BlockStatement::Compile([[maybe_unused]] compiler::ETSGen *etsg) const
96 {
97 etsg->GetAstCompiler()->Compile(this);
98 }
99
Check([[maybe_unused]] checker::TSChecker *checker)100 checker::Type *BlockStatement::Check([[maybe_unused]] checker::TSChecker *checker)
101 {
102 return checker->GetAnalyzer()->Check(this);
103 }
104
Check([[maybe_unused]] checker::ETSChecker *checker)105 checker::Type *BlockStatement::Check([[maybe_unused]] checker::ETSChecker *checker)
106 {
107 return checker->GetAnalyzer()->Check(this);
108 }
109 } // namespace ark::es2panda::ir
110