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 #include "ifStatement.h"
17
18 #include "compiler/base/condition.h"
19 #include "compiler/core/pandagen.h"
20 #include "ir/astDump.h"
21 #include "ir/statements/blockStatement.h"
22 #include "ir/expression.h"
23 #include "typescript/checker.h"
24
25 namespace panda::es2panda::ir {
26
Iterate(const NodeTraverser &cb) const27 void IfStatement::Iterate(const NodeTraverser &cb) const
28 {
29 cb(test_);
30 cb(consequent_);
31
32 if (alternate_) {
33 cb(alternate_);
34 }
35 }
36
Dump(ir::AstDumper *dumper) const37 void IfStatement::Dump(ir::AstDumper *dumper) const
38 {
39 dumper->Add({{"type", "IfStatement"},
40 {"test", test_},
41 {"consequent", consequent_},
42 {"alternate", AstDumper::Nullable(alternate_)}});
43 }
44
Compile(compiler::PandaGen *pg) const45 void IfStatement::Compile(compiler::PandaGen *pg) const
46 {
47 auto *consequentEnd = pg->AllocLabel();
48 compiler::Label *statementEnd = consequentEnd;
49
50 compiler::Condition::Compile(pg, test_, consequentEnd);
51 consequent_->Compile(pg);
52
53 if (alternate_) {
54 statementEnd = pg->AllocLabel();
55 pg->Branch(pg->Insns().back()->Node(), statementEnd);
56
57 pg->SetLabel(this, consequentEnd);
58 alternate_->Compile(pg);
59 }
60
61 pg->SetLabel(this, statementEnd);
62 }
63
Check(checker::Checker *checker) const64 checker::Type *IfStatement::Check(checker::Checker *checker) const
65 {
66 checker::Type *testType = test_->Check(checker);
67 checker->CheckTruthinessOfType(testType, Start());
68 checker->CheckTestingKnownTruthyCallableOrAwaitableType(test_, testType, consequent_);
69
70 consequent_->Check(checker);
71
72 if (alternate_) {
73 alternate_->Check(checker);
74 }
75
76 return nullptr;
77 }
78
UpdateSelf(const NodeUpdater &cb, binder::Binder *binder)79 void IfStatement::UpdateSelf(const NodeUpdater &cb, binder::Binder *binder)
80 {
81 test_ = std::get<ir::AstNode *>(cb(test_))->AsExpression();
82 consequent_ = UpdateIfStatementChildStatement(cb, binder, consequent_);
83
84 if (alternate_) {
85 alternate_ = UpdateIfStatementChildStatement(cb, binder, alternate_);
86 }
87 }
88
UpdateIfStatementChildStatement(const NodeUpdater &cb, const binder::Binder *binder, Statement *statement) const89 Statement *IfStatement::UpdateIfStatementChildStatement(const NodeUpdater &cb,
90 const binder::Binder *binder,
91 Statement *statement) const
92 {
93 auto newStatement = cb(statement);
94 if (std::holds_alternative<ir::AstNode *>(newStatement)) {
95 return std::get<ir::AstNode *>(newStatement)->AsStatement();
96 }
97
98 ASSERT(std::holds_alternative<std::vector<ir::AstNode *>>(newStatement));
99 ArenaVector<ir::Statement *> statements(binder->Allocator()->Adapter());
100 auto newStatements = std::get<std::vector<ir::AstNode *>>(newStatement);
101 for (auto *it : newStatements) {
102 statements.push_back(it->AsStatement());
103 }
104 return binder->Allocator()->New<ir::BlockStatement>(nullptr, std::move(statements));
105 }
106
107 } // namespace panda::es2panda::ir
108