arkcompiler_ets_runtime/ecmascript/js_array_iterator.cpp
fangting cdbfa4340e optimize compiler
Issue:#IALVU8
Signed-off-by: fangting <fangting12@huawei.com>
2024-08-27 11:02:23 +08:00

102 lines
4.6 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
* Copyright (c) 2021 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "ecmascript/js_array_iterator.h"
#include "ecmascript/base/typed_array_helper-inl.h"
#include "ecmascript/js_array.h"
namespace panda::ecmascript {
using BuiltinsBase = base::BuiltinsBase;
// 22.1.5.2.1 %ArrayIteratorPrototype%.next ( )
JSTaggedValue JSArrayIterator::Next(EcmaRuntimeCallInfo *argv)
{
ASSERT(argv);
JSThread *thread = argv->GetThread();
[[maybe_unused]] EcmaHandleScope handleScope(thread);
// 1.Let O be the this value.
JSHandle<JSTaggedValue> thisObj(BuiltinsBase::GetThis(argv));
return NextInternal(thread, thisObj);
}
JSTaggedValue JSArrayIterator::NextInternal(JSThread *thread, JSHandle<JSTaggedValue> thisObj)
{
// 2.If Type(O) is not Object, throw a TypeError exception.
// 3.If O does not have all of the internal slots of an TaggedArray Iterator Instance (22.1.5.3), throw a TypeError
// exception.
if (!thisObj->IsJSArrayIterator()) {
THROW_TYPE_ERROR_AND_RETURN(thread, "this value is not an array iterator", JSTaggedValue::Exception());
}
JSHandle<JSArrayIterator> iter(thisObj);
// 4.Let a be O.[[IteratedArrayLike]].
JSHandle<JSTaggedValue> array(thread, iter->GetIteratedArray());
JSHandle<JSTaggedValue> undefinedHandle(thread, JSTaggedValue::Undefined());
// 5.If a is undefined, return CreateIterResultObject(undefined, true).
if (array->IsUndefined()) {
return JSIterator::CreateIterResultObject(thread, undefinedHandle, true).GetTaggedValue();
}
// 6.Let index be O.[[ArrayLikeNextIndex]].
uint32_t index = iter->GetNextIndex();
// 7.Let itemKind be O.[[ArrayLikeIterationKind]].
IterationKind itemKind = iter->GetIterationKind();
uint32_t length;
// 8. If a has a [[TypedArrayName]] internal slot, then
// a. Let len be the value of Os [[ArrayLength]] internal slot.
if (array->IsTypedArray()) {
length = JSHandle<JSTypedArray>::Cast(array)->GetArrayLength();
} else if (array->IsJSArray()) {
length = JSHandle<JSArray>(array)->GetArrayLength();
} else {
// 9.Else
JSHandle<JSTaggedValue> lengthKey = thread->GlobalConstants()->GetHandledLengthString();
auto lengthProperty = JSTaggedValue::GetProperty(thread, array, lengthKey);
RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
auto lengthValue = JSTaggedValue::ToLength(thread, lengthProperty.GetValue());
RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
length = lengthValue.ToUint32();
}
// 10.If index ≥ len, then
if (index >= length) {
// Set O.[[IteratedArrayLike]] to undefined.
// Return CreateIterResultObject(undefined, true).
iter->SetIteratedArray(thread, undefinedHandle);
return JSIterator::CreateIterResultObject(thread, undefinedHandle, true).GetTaggedValue();
}
// 11.Set O.[[ArrayLikeNextIndex]] to index + 1.
iter->SetNextIndex(index + 1);
// 12.If itemKind is key, return CreateIterResultObject(index, false).
JSHandle<JSTaggedValue> key(thread, JSTaggedValue(index));
if (itemKind == IterationKind::KEY) {
return JSIterator::CreateIterResultObject(thread, key, false).GetTaggedValue();
}
JSHandle<JSTaggedValue> value = JSArray::FastGetPropertyByValue(thread, array, index);
RETURN_EXCEPTION_IF_ABRUPT_COMPLETION(thread);
// 15.If itemKind is value, let result be elementValue.
if (itemKind == IterationKind::VALUE) {
return JSIterator::CreateIterResultObject(thread, value, false).GetTaggedValue();
}
// 16. Else
ASSERT_PRINT(itemKind == IterationKind::KEY_AND_VALUE, "itemKind is invalid");
ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
JSHandle<TaggedArray> resultArray(factory->NewTaggedArray(2)); // 2 means the length of array
resultArray->Set(thread, 0, key);
resultArray->Set(thread, 1, value);
JSHandle<JSTaggedValue> keyAndValue(JSArray::CreateArrayFromList(thread, resultArray));
return JSIterator::CreateIterResultObject(thread, keyAndValue, false).GetTaggedValue();
}
} // namespace panda::ecmascript