1 /* 2 * Copyright (c) 2023 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 PANDA_BYTECODE_OPT_CODEGEN_H 17 #define PANDA_BYTECODE_OPT_CODEGEN_H 18 19 #include "assembler/assembly-function.h" 20 #include "assembler/assembly-ins.h" 21 #include "ins_create_api.h" 22 #include "ir_interface.h" 23 #include "compiler/optimizer/pass.h" 24 #include "compiler/optimizer/ir/basicblock.h" 25 #include "compiler/optimizer/ir/graph.h" 26 #include "compiler/optimizer/ir/graph_visitor.h" 27 #include "utils/logger.h" 28 #include "common.h" 29 30 namespace panda::bytecodeopt { 31 32 using compiler::BasicBlock; 33 using compiler::Inst; 34 using compiler::Opcode; 35 36 void DoLdaObj(compiler::Register reg, std::vector<pandasm::Ins> &result); 37 void DoLda(compiler::Register reg, std::vector<pandasm::Ins> &result); 38 void DoLda64(compiler::Register reg, std::vector<pandasm::Ins> &result); 39 void DoStaObj(compiler::Register reg, std::vector<pandasm::Ins> &result); 40 void DoSta(compiler::Register reg, std::vector<pandasm::Ins> &result); 41 void DoSta64(compiler::Register reg, std::vector<pandasm::Ins> &result); 42 void DoLdaDyn(compiler::Register reg, std::vector<pandasm::Ins> &result); 43 void DoStaDyn(compiler::Register reg, std::vector<pandasm::Ins> &result); 44 45 // NOLINTNEXTLINE(fuchsia-multiple-inheritance) 46 class BytecodeGen : public compiler::Optimization, public compiler::GraphVisitor { 47 public: BytecodeGen(compiler::Graph * graph,pandasm::Function * function,const BytecodeOptIrInterface * iface)48 explicit BytecodeGen(compiler::Graph *graph, pandasm::Function *function, const BytecodeOptIrInterface *iface) 49 : compiler::Optimization(graph), function_(function), irInterface_(iface) 50 { 51 } 52 ~BytecodeGen() override = default; 53 NO_COPY_SEMANTIC(BytecodeGen); 54 NO_MOVE_SEMANTIC(BytecodeGen); 55 56 bool RunImpl() override; GetPassName()57 const char *GetPassName() const override 58 { 59 return "BytecodeGen"; 60 } GetEncodedInstructions()61 std::vector<pandasm::Ins> GetEncodedInstructions() const 62 { 63 return res_; 64 } 65 66 void Reserve(size_t resSize = 0) 67 { 68 if (resSize > 0) { 69 result_.reserve(resSize); 70 } 71 } 72 GetStatus()73 bool GetStatus() const 74 { 75 return success_; 76 } 77 GetResult()78 const std::vector<pandasm::Ins> &GetResult() const 79 { 80 return result_; 81 } 82 GetResult()83 std::vector<pandasm::Ins> &&GetResult() 84 { 85 return std::move(result_); 86 } 87 LabelName(uint32_t id)88 static std::string LabelName(uint32_t id) 89 { 90 return "label_" + std::to_string(id); 91 } 92 EmitLabel(const std::string & label)93 void EmitLabel(const std::string &label) 94 { 95 pandasm::Ins l; 96 l.label = label; 97 l.setLabel = true; 98 result_.emplace_back(l); 99 } 100 101 void EmitJump(const BasicBlock *bb); 102 103 void EncodeSpillFillData(const compiler::SpillFillData &sf); 104 void EncodeSta(compiler::Register reg, compiler::DataType::Type type); 105 void AddLineNumber(const Inst *inst, size_t idx); 106 void AddColumnNumber(const Inst *inst, uint32_t idx); 107 void AddLineAndColumnNumber(const Inst *inst, size_t idx); 108 GetBlocksToVisit()109 const ArenaVector<BasicBlock *> &GetBlocksToVisit() const override 110 { 111 return GetGraph()->GetBlocksRPO(); 112 } 113 static void VisitSpillFill(GraphVisitor *v, Inst *inst); 114 static void VisitConstant(GraphVisitor *v, Inst *inst); 115 static void VisitCallStatic(GraphVisitor *visitor, Inst *inst); 116 static void VisitCallVirtual(GraphVisitor *visitor, Inst *inst); 117 static void VisitInitObject(GraphVisitor *visitor, Inst *inst); 118 static void VisitCatchPhi(GraphVisitor *visitor, Inst *inst); 119 120 static void VisitIf(GraphVisitor *v, Inst *instBase); 121 static void VisitIfImm(GraphVisitor *v, Inst *instBase); 122 static void VisitCast(GraphVisitor *v, Inst *instBase); 123 static void IfImmZero(GraphVisitor *v, Inst *instBase); 124 static void IfImmNonZero(GraphVisitor *v, Inst *instBase); 125 static void IfImm64(GraphVisitor *v, Inst *instBase); 126 static void VisitIntrinsic(GraphVisitor *v, Inst *instBase); 127 static void CallHandler(GraphVisitor *visitor, Inst *inst); 128 static void VisitStoreObject(GraphVisitor *v, Inst *instBase); 129 static void VisitStoreStatic(GraphVisitor *v, Inst *instBase); 130 static void VisitLoadObject(GraphVisitor *v, Inst *instBase); 131 static void VisitLoadStatic(GraphVisitor *v, Inst *instBase); 132 static void VisitLoadString(GraphVisitor *v, Inst *instBase); 133 static void VisitReturn(GraphVisitor *v, Inst *instBase); 134 135 static void VisitCastValueToAnyType(GraphVisitor *v, Inst *instBase); 136 137 static void VisitEcma(GraphVisitor *v, Inst *instBase); 138 static void IfEcma(GraphVisitor *v, compiler::IfInst *inst); 139 140 #include "generated/codegen_visitors.inc" 141 142 #include "generated/insn_selection.h" 143 VisitDefault(Inst * inst)144 void VisitDefault(Inst *inst) override 145 { 146 LOG(ERROR, BYTECODE_OPTIMIZER) << "Opcode " << compiler::GetOpcodeString(inst->GetOpcode()) 147 << " not yet implemented in codegen"; 148 success_ = false; 149 } 150 151 #include "compiler/optimizer/ir/visitor.inc" 152 153 private: 154 void AppendCatchBlock(uint32_t typeId, const compiler::BasicBlock *tryBegin, const compiler::BasicBlock *tryEnd, 155 const compiler::BasicBlock *catchBegin, const compiler::BasicBlock *catchEnd = nullptr); 156 void VisitTryBegin(const compiler::BasicBlock *bb); 157 158 private: 159 pandasm::Function *function_; 160 const BytecodeOptIrInterface *irInterface_; 161 162 std::vector<pandasm::Ins> res_; 163 std::vector<pandasm::Function::CatchBlock> catchBlocks_; 164 165 bool success_ {true}; 166 std::vector<pandasm::Ins> result_; 167 }; 168 169 } // namespace panda::bytecodeopt 170 171 #endif // PANDA_BYTECODE_OPT_CODEGEN_H 172