1 /* 2 * Copyright (c) 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 #ifndef ECMASCRIPT_COMPILER_STRING_BUILDER_OPTIMIZER_H 17 #define ECMASCRIPT_COMPILER_STRING_BUILDER_OPTIMIZER_H 18 19 #include "ecmascript/compiler/argument_accessor.h" 20 #include "ecmascript/compiler/bytecode_circuit_builder.h" 21 #include "ecmascript/compiler/circuit_builder-inl.h" 22 #include "ecmascript/compiler/combined_pass_visitor.h" 23 #include "ecmascript/compiler/graph_linearizer.h" 24 25 namespace panda::ecmascript::kungfu { 26 class StringBuilderOptimizer { 27 public: StringBuilderOptimizer(Circuit * circuit,bool enableLog,const std::string & name,CompilationConfig * cmpCfg,Chunk * chunk)28 StringBuilderOptimizer(Circuit* circuit, 29 bool enableLog, 30 const std::string& name, 31 CompilationConfig* cmpCfg, 32 Chunk* chunk) 33 : circuit_(circuit), 34 acc_(circuit), 35 builder_(circuit, cmpCfg), 36 graphLinearizer_(circuit, enableLog, name, chunk, true, true), 37 dependEntry_(circuit->GetDependRoot()), 38 status_(chunk), 39 stringBuilders_(chunk), 40 toVisit_(chunk), 41 ends_(chunk), 42 concatGates_(chunk) {} 43 44 ~StringBuilderOptimizer() = default; 45 46 void Run(); 47 private: 48 static constexpr int INVALID_ID = -1; 49 enum State { 50 UNVISITED = 0, 51 BEGIN_STRING_BUILDER, 52 IN_STRING_BUILDER, 53 PENDING_PHI, 54 CONFIRMED_IN_STRING_BUILDER, 55 END_STRING_BUILDER, 56 END_STRING_BUILDER_LOOP_PHI, 57 INVALID_OPT, 58 NUM_OF_STATE, 59 }; 60 61 struct Status { 62 int id {INVALID_ID}; 63 State state {State::INVALID_OPT}; 64 }; 65 GetStatus(GateRef gate)66 Status& GetStatus(GateRef gate) 67 { 68 ASSERT(acc_.GetId(gate) < status_.size()); 69 return status_[acc_.GetId(gate)]; 70 } GetStatus(GateRef gate)71 const Status& GetStatus(GateRef gate) const 72 { 73 ASSERT(acc_.GetId(gate) < status_.size()); 74 return status_[acc_.GetId(gate)]; 75 } 76 void SetStatus(GateRef gate, State state, int id = INVALID_ID) 77 { 78 status_[acc_.GetId(gate)] = Status{id, state}; 79 } UpdateStatus(GateRef gate,State state)80 void UpdateStatus(GateRef gate, State state) 81 { 82 int id = state == State::INVALID_OPT ? INVALID_ID : GetStatus(gate).id; 83 status_[acc_.GetId(gate)] = Status{id, state}; 84 } 85 86 struct StringBuilder { 87 GateRef start {Circuit::NullGate()}; 88 int id {INVALID_ID}; 89 bool has_loop_phi {false}; 90 }; 91 92 void VisitGraph(); 93 void FindBuilderBegin(GateRef gate); 94 void FindInBuilder(GateRef gate); 95 void FinalizeStringBuilders(); 96 void VisitGateUse(GateRef use); 97 void StatusTransfer(GateRef gate); 98 bool HasConcatOrPhiUse(GateRef gate); 99 bool CheckStringAddUses(GateRef gate, GateId stringAddId); 100 bool CheckPhiUses(GateRef gate, GateId stringAddId); 101 bool IsLiteralString(GateRef gate); 102 bool IsLoopHeader(GateRef gate); 103 bool LoopContains(GateRef loopPhi, GateRef gate); 104 bool PhiInputsAreConcatsOrPhi(GateRef phi); 105 106 Circuit *circuit_; 107 GateAccessor acc_; 108 CircuitBuilder builder_; 109 GraphLinearizer graphLinearizer_; 110 GateRef dependEntry_; 111 112 uint32_t stringBuilderCount_ {0}; 113 size_t currentIndex_ {0}; 114 115 ChunkVector<Status> status_; 116 ChunkVector<StringBuilder> stringBuilders_; 117 ChunkVector<GateRef> toVisit_; 118 ChunkVector<GateRef> ends_; 119 ChunkVector<GateRef> concatGates_; 120 }; 121 } // panda::ecmascript::kungfu 122 #endif // ECMASCRIPT_COMPILER_STRING_BUILDER_OPTIMIZER_H