/*
 * 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.
 */

#ifndef ECMASCRIPT_JS_STABLE_ARRAY_H
#define ECMASCRIPT_JS_STABLE_ARRAY_H

#include "ecmascript/base/array_helper.h"
#include "ecmascript/base/typed_array_helper.h"
#include "ecmascript/js_array.h"
#include "ecmascript/js_dataview.h"
#include "ecmascript/js_hclass.h"
#include "ecmascript/js_typed_array.h"
#include "ecmascript/js_tagged_value_wrapper.h"

namespace panda::ecmascript {
class JSStableArray {
public:
    enum class ComparisonType: uint8_t {
        STRICT_EQUAL,
        SAME_VALUE_ZERO,
    };

    enum class IndexOfReturnType: uint8_t {
        TAGGED_FOUND_INDEX,
        TAGGED_FOUND_OR_NOT,
        UNTAGGED_FOUND_INDEX,
        UNTAGGED_FOUND_OR_NOT,
    };

    struct IndexOfOptions {
        ComparisonType compType = ComparisonType::STRICT_EQUAL;
        IndexOfReturnType returnType = IndexOfReturnType::TAGGED_FOUND_INDEX;
        bool reversedOrder = false;
        bool holeAsUndefined = false;
    };

    static JSTaggedValue Push(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Push(JSHandle<JSSharedArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Pop(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Pop(JSHandle<JSSharedArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Splice(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv, uint32_t start,
                                uint32_t insertCount, uint32_t actualDeleteCount,
                                JSHandle<JSObject> newArrayHandle, uint32_t len);
    static JSTaggedValue Splice(JSHandle<JSSharedArray> receiver, EcmaRuntimeCallInfo *argv, uint32_t start,
                                uint32_t insertCount, uint32_t actualDeleteCount,
                                JSHandle<JSObject> newArrayHandle, uint32_t len);
    static JSTaggedValue Shift(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Shift(JSHandle<JSSharedArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue Join(JSHandle<JSTaggedValue> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue HandleFindIndexOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                                 JSHandle<JSTaggedValue> callbackFnHandle,
                                                 JSHandle<JSTaggedValue> thisArgHandle, uint32_t &k);
    static JSTaggedValue HandleFindLastIndexOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                                     JSHandle<JSTaggedValue> callbackFnHandle,
                                                     JSHandle<JSTaggedValue> thisArgHandle, int64_t &k);
    static JSTaggedValue HandleEveryOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                             JSHandle<JSTaggedValue> callbackFnHandle,
                                             JSHandle<JSTaggedValue> thisArgHandle, uint32_t &k);
    static JSTaggedValue HandleSomeOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                             JSHandle<JSTaggedValue> callbackFnHandle,
                                             JSHandle<JSTaggedValue> thisArgHandle, uint32_t &k);
    static JSTaggedValue HandleforEachOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                               JSHandle<JSTaggedValue> callbackFnHandle,
                                               JSHandle<JSTaggedValue> thisArgHandle, uint32_t len, uint32_t &k);
    static JSTaggedValue Includes(JSThread *thread, JSHandle<JSTaggedValue> receiver,
                                 JSHandle<JSTaggedValue> searchElement, uint32_t from, uint32_t len);
    static JSTaggedValue IndexOf(JSThread *thread, JSHandle<JSTaggedValue> receiver,
                                 JSHandle<JSTaggedValue> searchElement, uint32_t from, uint32_t len);
    static JSTaggedValue LastIndexOf(JSThread *thread, JSHandle<JSTaggedValue> receiver,
                                 JSHandle<JSTaggedValue> searchElement, uint32_t from, uint32_t len);
    static JSTaggedValue Filter(JSHandle<JSObject> newArrayHandle, JSHandle<JSObject> thisObjHandle,
                                 EcmaRuntimeCallInfo *argv, uint32_t &k, uint32_t &toIndex);
    static JSTaggedValue Map(JSHandle<JSObject> newArrayHandle, JSHandle<JSObject> thisObjHandle,
                                 EcmaRuntimeCallInfo *argv, uint32_t &k, uint32_t len);
    static JSTaggedValue Reverse(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                 int64_t &lower, uint32_t len);
    static JSTaggedValue FastReverse(JSThread *thread, JSHandle<TaggedArray> elements,
                                     int64_t &lower, uint32_t len, ElementsKind kind);
    static JSTaggedValue Concat(JSThread *thread, JSHandle<JSObject> newArrayHandle,
                                JSHandle<JSObject> thisObjHandle, int64_t &k, int64_t &n);
    template<base::TypedArrayKind typedArrayKind = base::TypedArrayKind::NON_SHARED>
    static JSTaggedValue FastCopyFromArrayToTypedArray(JSThread *thread, JSHandle<JSTypedArray> &target,
                                                       DataViewType targetType, uint64_t targetOffset,
                                                       uint32_t srcLength, JSHandle<JSObject> &obj);
    template<base::TypedArrayKind typedArrayKind = base::TypedArrayKind::NON_SHARED>
    static JSTaggedValue CopyArrayToTypedArrayForCtor(JSThread *thread, JSHandle<JSTypedArray> &target,
                                                      DataViewType targetType, uint64_t targetOffset,
                                                      uint32_t srcLength, JSHandle<JSObject> &obj);
    static JSTaggedValue At(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue At(JSHandle<JSSharedArray> receiver, EcmaRuntimeCallInfo *argv);
    static JSTaggedValue With(JSThread *thread, JSHandle<JSArray> receiver,
                              int64_t insertCount, int64_t index, JSHandle<JSTaggedValue> value);
    static JSTaggedValue ToSpliced(JSHandle<JSArray> receiver, EcmaRuntimeCallInfo *argv,
                                   int64_t argc, int64_t actualStart, int64_t actualSkipCount, int64_t insertCount);
    static JSTaggedValue ToReversed(JSThread *thread, JSHandle<JSArray> receiver, int64_t insertCount);
    static void Reduce(JSThread *thread, JSHandle<JSObject> thisObjHandle, JSHandle<JSTaggedValue> callbackFnHandle,
                       JSMutableHandle<JSTaggedValue> accumulator, int64_t &k, int64_t len);
    static JSTaggedValue Slice(JSThread *thread, JSHandle<JSObject> thisObjHandle, int64_t &k, int64_t &count);
    static JSHandle<TaggedArray> SortIndexedProperties(JSThread *thread, const JSHandle<JSTaggedValue> &thisObj,
                                                       int64_t len, const JSHandle<JSTaggedValue> &callbackFnHandle,
                                                       base::HolesType holes);
    static JSTaggedValue Sort(JSThread *thread, const JSHandle<JSTaggedValue> &thisObjVal,
                              const JSHandle<JSTaggedValue> &callbackFnHandle);
    static JSTaggedValue CopySortedListToReceiver(JSThread *thread, const JSHandle<JSTaggedValue> &thisObjVal,
                                                  JSHandle<TaggedArray> sortedList, uint32_t len);
    static JSTaggedValue Fill(JSThread *thread, const JSHandle<JSObject> &thisObj,
                              const JSHandle<JSTaggedValue> &value,
                              int64_t start, int64_t end);
    static JSTaggedValue HandleFindLastOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                                JSHandle<JSTaggedValue> callbackFnHandle,
                                                JSHandle<JSTaggedValue> thisArgHandle,
                                                JSMutableHandle<JSTaggedValue> &kValue, int64_t &k);
    static JSTaggedValue HandleReduceRightOfStable(JSThread *thread, JSHandle<JSObject> thisObjHandle,
                                                   JSHandle<JSTaggedValue> callbackFnHandle,
                                                   JSMutableHandle<JSTaggedValue> &accumulator,
                                                   JSHandle<JSTaggedValue> thisArgHandle, int64_t &k);

private:
    template <RBMode mode = RBMode::DEFAULT_RB, class Predicate>
    static const JSTaggedType *IndexOfElements(JSThread *thread, Span<const JSTaggedType> rawElements,
                                               IndexOfOptions options, Predicate predicate);
    static const JSTaggedType *IndexOfUndefined(JSThread *thread, Span<const JSTaggedType> elements,
                                                IndexOfOptions options, bool isMutant);
    static const JSTaggedType *IndexOfTaggedZero(JSThread *thread, Span<const JSTaggedType> taggedElements,
                                                 IndexOfOptions options);
    static const JSTaggedType *IndexOfInt(JSThread *thread, Span<const JSTaggedType> elements,
                                          JSTaggedValue searchElement, IndexOfOptions options, bool isMutantInt32Array);
    static const JSTaggedType *IndexOfDouble(JSThread *thread, Span<const JSTaggedType> elements,
                                             JSTaggedValue searchElement, IndexOfOptions options,
                                             bool isMutantDoubleArray);
    static const JSTaggedType *IndexOfObjectAddress(JSThread *thread, Span<const JSTaggedType> elements,
                                                    JSTaggedValue searchElement, IndexOfOptions options);
    static const JSTaggedType *IndexOfString(JSThread *thread, Span<const JSTaggedType> elements,
                                             JSTaggedValue searchElement, IndexOfOptions options);
    static const JSTaggedType *IndexOfBigInt(JSThread *thread, Span<const JSTaggedType> elements,
                                             JSTaggedValue searchElement, IndexOfOptions options);
    static JSTaggedValue IndexOfDispatch(JSThread *thread, JSHandle<JSTaggedValue> receiver,
                                         JSHandle<JSTaggedValue> searchElementHandle, uint32_t from, uint32_t len,
                                         IndexOfOptions options);

    static JSTaggedValue UpdateArrayCapacity(JSHandle<JSObject> &thisObjHandle, uint32_t &len,
                                             uint32_t &insertCount, uint32_t &actualDeleteCount,
                                             JSHandle<JSArray> &receiver, uint32_t &start,
                                             JSThread *thread, bool &needTransition,
                                             JSHandle<JSTaggedValue> &holeHandle,
                                             EcmaRuntimeCallInfo *argv, JSHandle<JSTaggedValue> &thisObjVal,
                                             JSHandle<JSTaggedValue> &lengthKey);
    static void HandleArray(JSHandle<JSObject> &newArrayHandle, uint32_t &actualDeleteCount,
                            JSThread *thread, uint32_t &start, JSHandle<JSObject> &thisObjHandle,
                            JSHandle<JSTaggedValue> &holeHandle);

    // Allocate object larger than 256 need liner search in the free object list,
    // so try to use tree string when the join result is larger than 256.
    static constexpr size_t TREE_STRING_THRESHOLD = 256;
    static constexpr size_t NUM_2 = 2;

    // When Array length is no more than 64, use array (stack memory) instead of vector to store the elements.
    static constexpr int64_t USE_STACK_MEMORY_THRESHOLD = 64;
    inline static bool WorthUseTreeString(uint32_t sepLength, size_t allocateLength, uint32_t len);
    template <typename Container>
    static void ProcessElements(JSThread *thread, JSHandle<JSTaggedValue> receiverValue, uint32_t len,
                                Container &arrElements, bool &isOneByte, uint64_t &allocateLength);
    template <typename Container>
    static JSTaggedValue DoStableArrayJoin(JSThread *thread, JSHandle<JSTaggedValue> receiverValue, uint32_t len,
                                           Container &arrElements, bool &isOneByte, uint32_t sep,
                                           uint32_t sepLength, JSHandle<EcmaString> sepStringHandle);
    template <typename Container>
    static JSTaggedValue JoinUseTreeString(const JSThread *thread, JSHandle<JSTaggedValue> receiverValue,
                                           JSHandle<EcmaString> sepStringHandle, uint32_t sepLength,
                                           Container &arrElements, uint32_t elemNum);
};
}  // namespace panda::ecmascript
#endif  // ECMASCRIPT_JS_STABLE_ARRAY_H