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_CLASS_DECLARATION_H
17 #define ES2PANDA_IR_STATEMENT_CLASS_DECLARATION_H
18 
19 #include <ir/base/classDefinition.h>
20 #include <ir/statement.h>
21 
22 namespace panda::es2panda::compiler {
23 class PandaGen;
24 }  // namespace panda::es2panda::compiler
25 
26 namespace panda::es2panda::checker {
27 class Checker;
28 class Type;
29 }  // namespace panda::es2panda::checker
30 
31 namespace panda::es2panda::ir {
32 
33 class ClassDeclaration : public Statement {
34 public:
ClassDeclaration(ClassDefinition *def, ArenaVector<Decorator *> &&decorators, ArenaVector<Annotation *> &&annotations, bool isAnnoDecl)35     explicit ClassDeclaration(ClassDefinition *def, ArenaVector<Decorator *> &&decorators,
36                               ArenaVector<Annotation *> &&annotations, bool isAnnoDecl)
37         : Statement(AstNodeType::CLASS_DECLARATION), def_(def), decorators_(std::move(decorators)),
38           isAnnotationDecl_(isAnnoDecl)
39     {
40         def_->Ctor()->SetAnnotations(std::move(annotations));
41     }
42 
Definition() const43     const ClassDefinition *Definition() const
44     {
45         return def_;
46     }
47 
Definition()48     ClassDefinition *Definition()
49     {
50         return def_;
51     }
52 
Decorators() const53     const ArenaVector<Decorator *> &Decorators() const
54     {
55         return decorators_;
56     }
57 
HasDecorators() const58     bool HasDecorators() const
59     {
60         return !decorators_.empty();
61     }
62 
IsAnnotationDecl() const63     bool IsAnnotationDecl() const
64     {
65         return isAnnotationDecl_;
66     }
67 
68     void Iterate(const NodeTraverser &cb) const override;
69     void Dump(ir::AstDumper *dumper) const override;
70     void Compile(compiler::PandaGen *pg) const override;
71     checker::Type *Check([[maybe_unused]] checker::Checker *checker) const override;
72     void UpdateSelf(const NodeUpdater &cb, [[maybe_unused]] binder::Binder *binder) override;
73 
74 private:
75     ClassDefinition *def_;
76     ArenaVector<Decorator *> decorators_;
77     bool isAnnotationDecl_ = false;
78 };
79 
80 }  // namespace panda::es2panda::ir
81 
82 #endif
83