• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2022-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 "ecmascript/tagged_tree.h"
17 
18 #include "ecmascript/interpreter/interpreter.h"
19 #include "ecmascript/js_function.h"
20 
21 namespace panda::ecmascript {
22 template<typename Derived>
Create(const JSThread * thread,int numberOfElements)23 JSHandle<Derived> TaggedTree<Derived>::Create(const JSThread *thread, int numberOfElements)
24 {
25     ASSERT_PRINT(numberOfElements > 0, "size must be a non-negative integer");
26     ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
27     int capacity = numberOfElements > MIN_CAPACITY ? numberOfElements : MIN_CAPACITY;
28     int length = ELEMENTS_START_INDEX + capacity * (Derived::ENTRY_SIZE);
29 
30     auto tree = JSHandle<Derived>::Cast(factory->NewTaggedArray(length));
31     tree->SetNumberOfElements(thread, 0);
32     tree->SetNumberOfDeletedElements(thread, 0);
33     tree->SetRootEntries(thread, -1);
34     tree->SetCompare(thread, JSTaggedValue::Hole());
35     tree->SetCapacity(thread, capacity);
36     return tree;
37 }
38 
39 template<typename Derived>
InsertRebalance(const JSThread * thread,int index)40 void TaggedTree<Derived>::InsertRebalance(const JSThread *thread, int index)
41 {
42     while (IsValidIndex(index) && GetColor(GetParent(index)) == TreeColor::RED) {
43         if (IsLeft(GetParent(index))) {
44             int bro = GetLeftBrother(GetParent(index));
45             if (GetColor(bro)) {
46                 SetColor(thread, GetParent(index), TreeColor::BLACK);
47                 SetColor(thread, bro, TreeColor::BLACK);
48                 SetColor(thread, GetParent(GetParent(index)), TreeColor::RED);
49                 index = GetParent(GetParent(index));
50             } else {
51                 if (IsRight(index)) {
52                     index = GetParent(index);
53                     LeftRotate(thread, index);
54                 }
55                 SetColor(thread, GetParent(index), TreeColor::BLACK);
56                 SetColor(thread, GetParent(GetParent(index)), TreeColor::RED);
57                 RightRotate(thread, GetParent(GetParent(index)));
58             }
59         } else {
60             int bro = GetRightBrother(GetParent(index));
61             if (GetColor(bro)) {
62                 SetColor(thread, GetParent(index), TreeColor::BLACK);
63                 SetColor(thread, bro, TreeColor::BLACK);
64                 SetColor(thread, GetParent(GetParent(index)), TreeColor::RED);
65                 index = GetParent(GetParent(index));
66             } else {
67                 if (IsLeft(index)) {
68                     index = GetParent(index);
69                     RightRotate(thread, index);
70                 }
71                 SetColor(thread, GetParent(index), TreeColor::BLACK);
72                 SetColor(thread, GetParent(GetParent(index)), TreeColor::RED);
73                 LeftRotate(thread, GetParent(GetParent(index)));
74             }
75         }
76     }
77     SetColor(thread, GetRootEntries(), TreeColor::BLACK);
78 }
79 
80 template<typename Derived>
LeftRotate(const JSThread * thread,int index)81 void TaggedTree<Derived>::LeftRotate(const JSThread *thread, int index)
82 {
83     if (index >= 0) {
84         int right = GetRightChild(index).GetInt();
85         JSTaggedValue leftOfRight = GetLeftChild(right);
86         SetRightChild(thread, index, leftOfRight);
87         if (!leftOfRight.IsHole()) {
88             SetParent(thread, leftOfRight.GetInt(), JSTaggedValue(index));
89         }
90         int parentOfIndex = GetParent(index);
91         SetParent(thread, right, JSTaggedValue(parentOfIndex));
92         if (parentOfIndex < 0) {
93             SetRootEntries(thread, right);
94         } else {
95             JSTaggedValue left = GetLeftChild(parentOfIndex);
96             if (!left.IsHole() && left.GetInt() == index) { // change to isleft
97                 SetLeftChild(thread, parentOfIndex, JSTaggedValue(right));
98             } else {
99                 SetRightChild(thread, parentOfIndex, JSTaggedValue(right));
100             }
101         }
102         SetLeftChild(thread, right, JSTaggedValue(index));
103         SetParent(thread, index, JSTaggedValue(right));
104     }
105 }
106 
107 template<typename Derived>
RightRotate(const JSThread * thread,int index)108 void TaggedTree<Derived>::RightRotate(const JSThread *thread, int index)
109 {
110     if (index >= 0) {
111         int left = GetLeftChild(index).GetInt();
112         JSTaggedValue rightOfLeft = GetRightChild(left);
113         SetLeftChild(thread, index, rightOfLeft);
114         if (!rightOfLeft.IsHole()) {
115             SetParent(thread, rightOfLeft.GetInt(), JSTaggedValue(index));
116         }
117         int parentOfIndex = GetParent(index);
118         SetParent(thread, left, JSTaggedValue(parentOfIndex));
119         if (parentOfIndex < 0) {
120             SetRootEntries(thread, left);
121         } else {
122             JSTaggedValue right = GetRightChild(parentOfIndex);
123             if (!right.IsHole() && right.GetInt() == index) { // change to isright
124                 SetRightChild(thread, parentOfIndex, JSTaggedValue(left));
125             } else {
126                 SetLeftChild(thread, parentOfIndex, JSTaggedValue(left));
127             }
128         }
129         SetRightChild(thread, left, JSTaggedValue(index));
130         SetParent(thread, index, JSTaggedValue(left));
131     }
132 }
133 
134 template<typename Derived>
AdjustTaggedTree(const JSThread * thread,const JSHandle<Derived> & tree,int len)135 JSHandle<Derived> TaggedTree<Derived>::AdjustTaggedTree(const JSThread *thread, const JSHandle<Derived> &tree, int len)
136 {
137     JSMutableHandle<Derived> newTree(thread, JSTaggedValue::Undefined());
138     ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
139     if (tree->NumberOfDeletedElements() == 0) {
140         newTree.Update(factory->ExtendArray(JSHandle<TaggedArray>::Cast(tree), len).GetTaggedValue());
141         return newTree;
142     }
143 
144     uint32_t elements = tree->NumberOfElements();
145     newTree.Update(factory->NewTaggedArray(len).GetTaggedValue());
146     newTree->SetNumberOfElements(thread, elements);
147     newTree->SetNumberOfDeletedElements(thread, 0);
148 
149     newTree->SetRootEntries(thread, 0);
150     CQueue<int> entries;
151     entries.push(tree->GetRootEntries());
152     int index = 0;
153     newTree->SetParent(thread, index, JSTaggedValue(-1));
154     int child = 1;
155     while (!entries.empty()) {
156         int parent = entries.front();
157         JSTaggedValue left = tree->GetLeftChild(parent);
158         if (!left.IsHole()) {
159             entries.push(left.GetInt());
160             newTree->SetLeftChild(thread, index, JSTaggedValue(child));
161             newTree->SetParent(thread, child++, JSTaggedValue(index));
162         }
163         JSTaggedValue right = tree->GetRightChild(parent);
164         if (!right.IsHole()) {
165             entries.push(right.GetInt());
166             newTree->SetRightChild(thread, index, JSTaggedValue(child));
167             newTree->SetParent(thread, child++, JSTaggedValue(index));
168         }
169         tree->CopyEntry(thread, parent, newTree, index);
170         entries.pop();
171         index++;
172     }
173     return newTree;
174 }
175 
176 template<typename Derived>
Transplant(const JSThread * thread,int dst,int src)177 void TaggedTree<Derived>::Transplant(const JSThread *thread, int dst, int src)
178 {
179     int parent = GetParent(dst);
180     if (parent < 0) {
181         SetRootEntries(thread, src);
182     } else if (IsLeft(dst)) {
183         JSTaggedValue child = src < 0 ? JSTaggedValue::Hole() : JSTaggedValue(src);
184         SetLeftChild(thread, parent, child);
185     } else {
186         JSTaggedValue child = src < 0 ? JSTaggedValue::Hole() : JSTaggedValue(src);
187         SetRightChild(thread, parent, child);
188     }
189     SetParent(thread, src, JSTaggedValue(parent));
190 }
191 
192 template<typename Derived>
Remove(const JSThread * thread,const JSHandle<Derived> & tree,int entry)193 void TaggedTree<Derived>::Remove(const JSThread *thread, const JSHandle<Derived> &tree, int entry)
194 {
195     int successor = entry;
196     if (!tree->GetLeftChild(entry).IsHole() && !tree->GetRightChild(entry).IsHole()) {
197         successor = tree->GetSuccessor(entry);
198         tree->CopyData(thread, entry, successor);
199     }
200     JSTaggedValue left = tree->GetLeftChild(successor);
201     JSTaggedValue right = tree->GetRightChild(successor);
202     int child = left.IsHole() ? (right.IsHole() ? -1 : right.GetInt()) : left.GetInt();
203     if (child < 0) {
204         if (tree->GetColor(successor) == TreeColor::BLACK) {
205             tree->DeleteRebalance(thread, successor);
206         }
207     }
208     tree->Transplant(thread, successor, child);
209 
210     if (child >= 0) {
211         if (tree->GetColor(successor) == TreeColor::BLACK) {
212             tree->DeleteRebalance(thread, child);
213         }
214     }
215     tree->RemoveEntry(thread, successor);
216     ASSERT(tree->NumberOfElements() > 0);
217     tree->SetNumberOfElements(thread, tree->NumberOfElements() - 1);
218     tree->SetNumberOfDeletedElements(thread, tree->NumberOfDeletedElements() + 1);
219 }
220 
221 template<typename Derived>
DeleteRebalance(const JSThread * thread,int index)222 void TaggedTree<Derived>::DeleteRebalance(const JSThread *thread, int index)
223 {
224     while (index != GetRootEntries() && GetColor(index) == TreeColor::BLACK) {
225         if (IsLeft(index)) {
226             int bro = GetLeftBrother(index);
227             if (GetColor(bro)) {
228                 SetColor(thread, bro, TreeColor::BLACK);
229                 SetColor(thread, GetParent(index), TreeColor::RED);
230                 LeftRotate(thread, GetParent(index));
231                 bro = GetLeftBrother(index);
232             }
233             if (GetColor(GetLeftChildIndex(bro)) == TreeColor::BLACK &&
234                 GetColor(GetRightChildIndex(bro)) == TreeColor::BLACK) {
235                 SetColor(thread, bro, TreeColor::RED);
236                 index = GetParent(index);
237             } else {
238                 if (GetColor(GetRightChildIndex(bro)) == TreeColor::BLACK) {
239                     SetColor(thread, GetLeftChildIndex(bro), TreeColor::BLACK);
240                     SetColor(thread, bro, TreeColor::RED);
241                     RightRotate(thread, bro);
242                     bro = GetLeftBrother(index);
243                 }
244                 SetColor(thread, bro, GetColor(GetParent(index)));
245                 SetColor(thread, GetParent(index), TreeColor::BLACK);
246                 SetColor(thread, GetRightChildIndex(bro), TreeColor::BLACK);
247                 LeftRotate(thread, GetParent(index));
248                 index = GetRootEntries();
249             }
250         } else {
251             int bro = GetRightBrother(index);
252             if (GetColor(bro)) {
253                 SetColor(thread, bro, TreeColor::BLACK);
254                 SetColor(thread, GetParent(index), TreeColor::RED);
255                 RightRotate(thread, GetParent(index));
256                 bro = GetRightBrother(index);
257             }
258             if (GetColor(GetRightChildIndex(bro)) == TreeColor::BLACK &&
259                 GetColor(GetLeftChildIndex(bro)) == TreeColor::BLACK) {
260                 SetColor(thread, bro, TreeColor::RED);
261                 index = GetParent(index);
262             } else {
263                 if (GetColor(GetLeftChildIndex(bro)) == TreeColor::BLACK) {
264                     SetColor(thread, GetRightChildIndex(bro), TreeColor::BLACK);
265                     SetColor(thread, bro, TreeColor::RED);
266                     LeftRotate(thread, bro);
267                     bro = GetRightBrother(index);
268                 }
269                 SetColor(thread, bro, GetColor(GetParent(index)));
270                 SetColor(thread, GetParent(index), TreeColor::BLACK);
271                 SetColor(thread, GetLeftChildIndex(bro), TreeColor::BLACK);
272                 RightRotate(thread, GetParent(index));
273                 index = GetRootEntries();
274             }
275         }
276     }
277     SetColor(thread, index, TreeColor::BLACK);
278 }
279 
280 template<typename Derived>
GetPreDecessor(int entry) const281 int TaggedTree<Derived>::GetPreDecessor(int entry) const
282 {
283     int child = GetLeftChildIndex(entry);
284     if (child >= 0) {
285         return GetMaximum(child);
286     }
287     int parent = GetParent(entry);
288     while (parent >= 0 && (GetLeftChildIndex(parent) == entry)) {
289         entry = parent;
290         parent = GetParent(entry);
291     }
292     return parent;
293 }
294 
295 template<typename Derived>
GetSuccessor(int entry) const296 int TaggedTree<Derived>::GetSuccessor(int entry) const
297 {
298     int child = GetRightChildIndex(entry);
299     if (child >= 0) {
300         return GetMinimum(child);
301     }
302     int parent = GetParent(entry);
303     while (parent >= 0 && (GetRightChildIndex(parent) == entry)) {
304         entry = parent;
305         parent = GetParent(entry);
306     }
307     return parent;
308 }
309 
310 template<typename Derived>
FindEntry(JSThread * thread,const JSHandle<Derived> & tree,const JSHandle<JSTaggedValue> & key)311 int TaggedTree<Derived>::FindEntry(JSThread *thread, const JSHandle<Derived> &tree, const JSHandle<JSTaggedValue> &key)
312 {
313     int parentIndex = tree->GetRootEntries();
314     JSMutableHandle<JSTaggedValue> parentKey(thread, tree->GetKey(parentIndex));
315     ComparisonResult res;
316     while (!parentKey->IsHole()) {
317         res = EntryCompare(thread, key, parentKey, tree);
318         RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, 0);
319         if (res == ComparisonResult::EQUAL) {
320             return parentIndex;
321         } else if (res == ComparisonResult::LESS) {
322             JSTaggedValue child = tree->GetLeftChild(parentIndex);
323             if (child.IsHole()) {
324                 break;
325             }
326             parentIndex = child.GetInt();
327         } else {
328             JSTaggedValue child = tree->GetRightChild(parentIndex);
329             if (child.IsHole()) {
330                 break;
331             }
332             parentIndex = child.GetInt();
333         }
334         parentKey.Update(tree->GetKey(parentIndex));
335     }
336     return -1;
337 }
338 
339 template<typename Derived>
EntryCompare(JSThread * thread,const JSHandle<JSTaggedValue> valueX,const JSHandle<JSTaggedValue> valueY,JSHandle<Derived> tree)340 ComparisonResult TaggedTree<Derived>::EntryCompare(JSThread *thread, const JSHandle<JSTaggedValue> valueX,
341     const JSHandle<JSTaggedValue> valueY, JSHandle<Derived> tree)
342 {
343     JSTaggedValue fn = tree->GetCompare();
344     if (fn.IsHole()) {
345         return OrdinayEntryCompare(thread, valueX, valueY);
346     }
347     if (valueX->IsUndefined()) {
348         return valueY->IsUndefined() ? ComparisonResult::EQUAL : ComparisonResult::GREAT;
349     }
350     if (valueY->IsUndefined()) {
351         return ComparisonResult::LESS;
352     }
353     if (valueX->IsNull()) {
354         return valueY->IsNull() ? ComparisonResult::EQUAL : ComparisonResult::GREAT;
355     }
356     if (valueY->IsNull()) {
357         return ComparisonResult::LESS;
358     }
359     JSHandle<JSTaggedValue> compareFn(thread, fn);
360     JSHandle<JSTaggedValue> thisArgHandle = thread->GlobalConstants()->GetHandledUndefined();
361     const uint32_t argsLength = 2;
362     JSHandle<JSTaggedValue> undefined = thread->GlobalConstants()->GetHandledUndefined();
363     EcmaRuntimeCallInfo *info =
364         EcmaInterpreter::NewRuntimeCallInfo(thread, compareFn, thisArgHandle, undefined, argsLength);
365     RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, ComparisonResult::UNDEFINED);
366     info->SetCallArg(valueX.GetTaggedValue(), valueY.GetTaggedValue());
367     JSTaggedValue callResult = JSFunction::Call(info);
368     RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, ComparisonResult::UNDEFINED);
369     int compareResult = -1;
370     if (callResult.IsBoolean()) {
371         // if callResult is true, compareResult = -1.
372         if (callResult.IsFalse()) {
373             info = EcmaInterpreter::NewRuntimeCallInfo(thread, compareFn, thisArgHandle, undefined, argsLength);
374             info->SetCallArg(valueY.GetTaggedValue(), valueX.GetTaggedValue());
375             callResult = JSFunction::Call(info);
376             RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, ComparisonResult::UNDEFINED);
377             compareResult = callResult.IsTrue() ? 1 : 0;
378         }
379     } else if (callResult.IsInt()) {
380         compareResult = callResult.GetInt();
381     } else {
382         JSTaggedNumber v = JSTaggedValue::ToNumber(thread, JSHandle<JSTaggedValue>(thread, callResult));
383         RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, ComparisonResult::UNDEFINED);
384         double value = v.GetNumber();
385         if (std::isnan(value)) {
386             THROW_TYPE_ERROR_AND_RETURN(thread, "CompareFn has illegal return value", ComparisonResult::UNDEFINED);
387         }
388         compareResult = static_cast<int>(value);
389     }
390     return compareResult > 0 ? ComparisonResult::GREAT :
391                             (compareResult < 0 ? ComparisonResult::LESS : ComparisonResult::EQUAL);
392 }
393 
394 template<typename Derived>
GetSortArray(const JSThread * thread,const JSHandle<Derived> & tree)395 JSHandle<TaggedArray> TaggedTree<Derived>::GetSortArray(const JSThread *thread, const JSHandle<Derived> &tree)
396 {
397     JSHandle<TaggedArray> sortArray = thread->GetEcmaVM()->GetFactory()->NewTaggedArray(tree->NumberOfElements());
398     CStack<int> entries;
399     int index = tree->GetRootEntries();
400     int aid = 0;
401     while (index >= 0 || !entries.empty()) {
402         while (index >= 0) {
403             entries.emplace(index);
404             index = tree->GetLeftChildIndex(index);
405         }
406         if (!entries.empty()) {
407             sortArray->Set(thread, aid++, JSTaggedValue(entries.top()));
408             index = tree->GetRightChildIndex(entries.top());
409             entries.pop();
410         }
411     }
412     return sortArray;
413 }
414 
415 template<typename Derived>
Insert(JSThread * thread,JSHandle<Derived> & tree,const JSHandle<JSTaggedValue> & key,const JSHandle<JSTaggedValue> & value)416 JSHandle<Derived> TaggedTree<Derived>::Insert(JSThread *thread, JSHandle<Derived> &tree,
417                                               const JSHandle<JSTaggedValue> &key, const JSHandle<JSTaggedValue> &value)
418 {
419     ASSERT(IsKey(key.GetTaggedValue()));
420     JSMutableHandle<JSTaggedValue> parentKey(thread, tree->GetRootKey());
421     if (parentKey->IsHole()) {
422         tree->SetRoot(thread, 0, key.GetTaggedValue(), value.GetTaggedValue());
423         tree->SetNumberOfElements(thread, tree->NumberOfElements() + 1);
424         return tree;
425     }
426 
427     JSHandle<Derived> newTree = GrowCapacity(thread, tree);
428     int parentIndex = newTree->GetRootEntries();
429     ComparisonResult res;
430     while (!parentKey->IsHole()) {
431         res = EntryCompare(thread, key, parentKey, tree);
432         RETURN_VALUE_IF_ABRUPT_COMPLETION(thread, JSHandle<Derived>(thread, JSTaggedValue::Exception()));
433         if (res == ComparisonResult::EQUAL) {
434             newTree->SetValue(thread, parentIndex, value.GetTaggedValue());
435             return tree;
436         } else if (res == ComparisonResult::LESS) {
437             JSTaggedValue child = newTree->GetLeftChild(parentIndex);
438             if (child.IsHole()) {
439                 break;
440             }
441             parentIndex = child.GetInt();
442         } else {
443             JSTaggedValue child = newTree->GetRightChild(parentIndex);
444             if (child.IsHole()) {
445                 break;
446             }
447             parentIndex = child.GetInt();
448         }
449         parentKey.Update(newTree->GetKey(parentIndex));
450     }
451 
452     uint32_t entry = newTree->NumberOfElements() + newTree->NumberOfDeletedElements();
453     if (res != ComparisonResult::LESS) {
454         newTree->InsertRightEntry(thread, parentIndex, entry, key.GetTaggedValue(), value.GetTaggedValue());
455     } else {
456         newTree->InsertLeftEntry(thread, parentIndex, entry, key.GetTaggedValue(), value.GetTaggedValue());
457     }
458     newTree->SetNumberOfElements(thread, newTree->NumberOfElements() + 1);
459     newTree->InsertRebalance(thread, entry);
460     return newTree;
461 }
462 
463 template<typename Derived>
GrowCapacity(const JSThread * thread,JSHandle<Derived> & tree)464 JSHandle<Derived> TaggedTree<Derived>::GrowCapacity(const JSThread *thread, JSHandle<Derived> &tree)
465 {
466     uint32_t nof = tree->NumberOfElements() + tree->NumberOfDeletedElements();
467     int oldCapacity = tree->Capacity();
468     if (static_cast<int>(nof + 1) <= oldCapacity) {
469         return tree;
470     }
471 
472     int newCapacity = ComputeCapacity(oldCapacity);
473     int length = ELEMENTS_START_INDEX + newCapacity * (Derived::ENTRY_SIZE);
474     JSHandle<Derived> newTree = AdjustTaggedTree(thread, tree, length);
475     newTree->SetCapacity(thread, newCapacity);
476     return newTree;
477 }
478 
479 template<typename Derived>
GetLowerKey(JSThread * thread,const JSHandle<Derived> & tree,const JSHandle<JSTaggedValue> & key)480 JSTaggedValue TaggedTree<Derived>::GetLowerKey(JSThread *thread, const JSHandle<Derived> &tree,
481                                                const JSHandle<JSTaggedValue> &key)
482 {
483     int parentIndex = tree->GetRootEntries();
484     JSMutableHandle<JSTaggedValue> parentKey(thread, tree->GetKey(parentIndex));
485     int resultIndex = -1;
486     ComparisonResult res;
487     while (parentIndex >= 0) {
488         res = EntryCompare(thread, key, parentKey, tree);
489         RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
490         if (res == ComparisonResult::GREAT) {
491             resultIndex = parentIndex;
492             parentIndex = tree->GetRightChildIndex(parentIndex);
493         } else {
494             parentIndex = tree->GetLeftChildIndex(parentIndex);
495         }
496         parentKey.Update(tree->GetKey(parentIndex));
497     }
498     JSTaggedValue lowerKey = tree->GetKey(resultIndex);
499     return tree->Transform(lowerKey);
500 }
501 
502 template<typename Derived>
GetHigherKey(JSThread * thread,const JSHandle<Derived> & tree,const JSHandle<JSTaggedValue> & key)503 JSTaggedValue TaggedTree<Derived>::GetHigherKey(JSThread *thread, const JSHandle<Derived> &tree,
504                                                 const JSHandle<JSTaggedValue> &key)
505 {
506     int parentIndex = tree->GetRootEntries();
507     JSMutableHandle<JSTaggedValue> parentKey(thread, tree->GetKey(parentIndex));
508     int resultIndex = -1;
509     ComparisonResult res;
510     while (parentIndex >= 0) {
511         res = EntryCompare(thread, key, parentKey, tree);
512         RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
513         if (res == ComparisonResult::LESS) {
514             resultIndex = parentIndex;
515             parentIndex = tree->GetLeftChildIndex(parentIndex);
516         } else {
517             parentIndex = tree->GetRightChildIndex(parentIndex);
518         }
519         parentKey.Update(tree->GetKey(parentIndex));
520     }
521     JSTaggedValue lowerKey = tree->GetKey(resultIndex);
522     return tree->Transform(lowerKey);
523 }
524 
525 template<typename Derived>
Shrink(const JSThread * thread,const JSHandle<Derived> & tree)526 JSHandle<Derived> TaggedTree<Derived>::Shrink(const JSThread *thread, const JSHandle<Derived> &tree)
527 {
528     int oldCapacity = static_cast<int>(tree->Capacity());
529     if (static_cast<int>(tree->NumberOfElements()) >= (oldCapacity + 1) / 4) { // 4: quarter
530         return tree;
531     }
532     uint32_t newCapacity = static_cast<uint32_t>(oldCapacity - 1) >> 1;
533     if (newCapacity < static_cast<uint32_t>(Derived::MIN_SHRINK_CAPACITY)) {
534         return tree;
535     }
536 
537     int length = ELEMENTS_START_INDEX + static_cast<int>(newCapacity) * (Derived::ENTRY_SIZE);
538     JSHandle<Derived> newTree = AdjustTaggedTree(thread, tree, length);
539     JSTaggedValue fn = tree->GetCompare();
540     JSHandle<JSTaggedValue> compareFn = JSHandle<JSTaggedValue>(thread, fn);
541     if (!compareFn->IsUndefined() && !compareFn->IsNull()) {
542         newTree->SetCompare(thread, compareFn.GetTaggedValue());
543     }
544     newTree->SetCapacity(thread, newCapacity);
545     return newTree;
546 }
547 
548 // TaggedTreeMap
Create(const JSThread * thread,int numberOfElements)549 JSTaggedValue TaggedTreeMap::Create(const JSThread *thread, int numberOfElements)
550 {
551     return RBTree::Create(thread, numberOfElements).GetTaggedValue();
552 }
553 
GetArrayFromMap(const JSThread * thread,const JSHandle<TaggedTreeMap> & map)554 JSHandle<TaggedArray> TaggedTreeMap::GetArrayFromMap(const JSThread *thread, const JSHandle<TaggedTreeMap> &map)
555 {
556     return RBTree::GetSortArray(thread, map);
557 }
558 
Set(JSThread * thread,JSHandle<TaggedTreeMap> & obj,const JSHandle<JSTaggedValue> & key,const JSHandle<JSTaggedValue> & value)559 JSTaggedValue TaggedTreeMap::Set(JSThread *thread, JSHandle<TaggedTreeMap> &obj,
560                                  const JSHandle<JSTaggedValue> &key, const JSHandle<JSTaggedValue> &value)
561 {
562     return RBTree::Insert(thread, obj, key, value).GetTaggedValue();
563 }
564 
Delete(JSThread * thread,const JSHandle<TaggedTreeMap> & map,int entry)565 JSTaggedValue TaggedTreeMap::Delete(JSThread *thread, const JSHandle<TaggedTreeMap> &map, int entry)
566 {
567     RBTree::Remove(thread, map, entry);
568     return RBTree::Shrink(thread, map).GetTaggedValue();
569 }
570 
HasValue(const JSThread * thread,JSTaggedValue value) const571 bool TaggedTreeMap::HasValue([[maybe_unused]] const JSThread *thread, JSTaggedValue value) const
572 {
573     int root = GetRootEntries();
574     if (root < 0) {
575         return false;
576     }
577 
578     CQueue<int> entries;
579     entries.push(root);
580     while (!entries.empty()) {
581         int parent = entries.front();
582         if (JSTaggedValue::SameValue(GetValue(parent), value)) {
583             return true;
584         }
585         int left = GetLeftChildIndex(parent);
586         if (left >= 0) {
587             entries.push(left);
588         }
589         int right = GetRightChildIndex(parent);
590         if (right >= 0) {
591             entries.push(right);
592         }
593         entries.pop();
594     }
595     return false;
596 }
597 
SetAll(JSThread * thread,JSHandle<TaggedTreeMap> & dst,const JSHandle<TaggedTreeMap> & src)598 JSTaggedValue TaggedTreeMap::SetAll(JSThread *thread, JSHandle<TaggedTreeMap> &dst, const JSHandle<TaggedTreeMap> &src)
599 {
600     CQueue<int> entries;
601     entries.push(src->GetRootEntries());
602     JSHandle<TaggedTreeMap> map = dst;
603     while (!entries.empty()) {
604         int parent = entries.front();
605         map = Insert(thread, map, JSHandle<JSTaggedValue>(thread, src->GetKey(parent)),
606                            JSHandle<JSTaggedValue>(thread, src->GetValue(parent)));
607         RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
608         int left = src->GetLeftChildIndex(parent);
609         if (left >= 0) {
610             entries.push(left);
611         }
612         int right = src->GetRightChildIndex(parent);
613         if (right >= 0) {
614             entries.push(right);
615         }
616         entries.pop();
617     }
618     return map.GetTaggedValue();
619 }
620 
GetLowerKey(JSThread * thread,const JSHandle<TaggedTreeMap> & map,const JSHandle<JSTaggedValue> & key)621 JSTaggedValue TaggedTreeMap::GetLowerKey(JSThread *thread, const JSHandle<TaggedTreeMap> &map,
622                                          const JSHandle<JSTaggedValue> &key)
623 {
624     return RBTree::GetLowerKey(thread, map, key);
625 }
626 
GetHigherKey(JSThread * thread,const JSHandle<TaggedTreeMap> & map,const JSHandle<JSTaggedValue> & key)627 JSTaggedValue TaggedTreeMap::GetHigherKey(JSThread *thread, const JSHandle<TaggedTreeMap> &map,
628                                           const JSHandle<JSTaggedValue> &key)
629 {
630     return RBTree::GetHigherKey(thread, map, key);
631 }
632 
FindEntry(JSThread * thread,const JSHandle<TaggedTreeMap> & map,const JSHandle<JSTaggedValue> & key)633 int TaggedTreeMap::FindEntry(JSThread *thread, const JSHandle<TaggedTreeMap> &map, const JSHandle<JSTaggedValue> &key)
634 {
635     return RBTree::FindEntry(thread, map, key);
636 }
637 
638 // TaggedTreeSet
Create(const JSThread * thread,int numberOfElements)639 JSTaggedValue TaggedTreeSet::Create(const JSThread *thread, int numberOfElements)
640 {
641     return RBTree::Create(thread, numberOfElements).GetTaggedValue();
642 }
643 
GetArrayFromSet(const JSThread * thread,const JSHandle<TaggedTreeSet> & set)644 JSHandle<TaggedArray> TaggedTreeSet::GetArrayFromSet(const JSThread *thread, const JSHandle<TaggedTreeSet> &set)
645 {
646     return RBTree::GetSortArray(thread, set);
647 }
648 
Add(JSThread * thread,JSHandle<TaggedTreeSet> & obj,const JSHandle<JSTaggedValue> & value)649 JSTaggedValue TaggedTreeSet::Add(JSThread *thread, JSHandle<TaggedTreeSet> &obj, const JSHandle<JSTaggedValue> &value)
650 {
651     return RBTree::Insert(thread, obj, value, value).GetTaggedValue();
652 }
653 
Delete(JSThread * thread,const JSHandle<TaggedTreeSet> & set,int entry)654 JSTaggedValue TaggedTreeSet::Delete(JSThread *thread, const JSHandle<TaggedTreeSet> &set, int entry)
655 {
656     RBTree::Remove(thread, set, entry);
657     return RBTree::Shrink(thread, set).GetTaggedValue();
658 }
659 
GetLowerKey(JSThread * thread,const JSHandle<TaggedTreeSet> & set,const JSHandle<JSTaggedValue> & key)660 JSTaggedValue TaggedTreeSet::GetLowerKey(JSThread *thread, const JSHandle<TaggedTreeSet> &set,
661                                          const JSHandle<JSTaggedValue> &key)
662 {
663     return RBTree::GetLowerKey(thread, set, key);
664 }
665 
GetHigherKey(JSThread * thread,const JSHandle<TaggedTreeSet> & set,const JSHandle<JSTaggedValue> & key)666 JSTaggedValue TaggedTreeSet::GetHigherKey(JSThread *thread, const JSHandle<TaggedTreeSet> &set,
667                                           const JSHandle<JSTaggedValue> &key)
668 {
669     return RBTree::GetHigherKey(thread, set, key);
670 }
671 
FindEntry(JSThread * thread,const JSHandle<TaggedTreeSet> & set,const JSHandle<JSTaggedValue> & key)672 int TaggedTreeSet::FindEntry(JSThread *thread, const JSHandle<TaggedTreeSet> &set, const JSHandle<JSTaggedValue> &key)
673 {
674     return RBTree::FindEntry(thread, set, key);
675 }
676 }  // namespace panda::ecmascript
677