typedarray.cc

#include <vector>
#include "napi.h"
using namespace Napi;

#if defined(NAPI_HAS_CONSTEXPR)
#define NAPI_TYPEDARRAY_NEW(className, env, length, type)                      \
  className::New(env, length)
#define NAPI_TYPEDARRAY_NEW_BUFFER(                                            \
    className, env, length, buffer, bufferOffset, type)                        \
  className::New(env, length, buffer, bufferOffset)
#else
#define NAPI_TYPEDARRAY_NEW(className, env, length, type)                      \
  className::New(env, length, type)
#define NAPI_TYPEDARRAY_NEW_BUFFER(                                            \
    className, env, length, buffer, bufferOffset, type)                        \
  className::New(env, length, buffer, bufferOffset, type)
#endif

namespace {

Value CreateTypedArray(const CallbackInfo& info) {
  std::string arrayType = info[0].As<String>();
  size_t length = info[1].As<Number>().Uint32Value();
  Value buffer = info[2];
  size_t bufferOffset =
      info[3].IsUndefined() ? 0 : info[3].As<Number>().Uint32Value();

  if (arrayType == "int8") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Int8Array, info.Env(), length, napi_int8_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Int8Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_int8_array);
  } else if (arrayType == "uint8") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Uint8Array, info.Env(), length, napi_uint8_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Uint8Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_uint8_array);
  } else if (arrayType == "uint8_clamped") {
    return buffer.IsUndefined()
               ? Uint8Array::New(info.Env(), length, napi_uint8_clamped_array)
               : Uint8Array::New(info.Env(),
                                 length,
                                 buffer.As<ArrayBuffer>(),
                                 bufferOffset,
                                 napi_uint8_clamped_array);
  } else if (arrayType == "int16") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Int16Array, info.Env(), length, napi_int16_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Int16Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_int16_array);
  } else if (arrayType == "uint16") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Uint16Array, info.Env(), length, napi_uint16_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Uint16Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_uint16_array);
  } else if (arrayType == "int32") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Int32Array, info.Env(), length, napi_int32_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Int32Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_int32_array);
  } else if (arrayType == "uint32") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Uint32Array, info.Env(), length, napi_uint32_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Uint32Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_uint32_array);
  } else if (arrayType == "float32") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Float32Array, info.Env(), length, napi_float32_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Float32Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_float32_array);
  } else if (arrayType == "float64") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     Float64Array, info.Env(), length, napi_float64_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(Float64Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_float64_array);
#if (NAPI_VERSION > 5)
  } else if (arrayType == "bigint64") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     BigInt64Array, info.Env(), length, napi_bigint64_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(BigInt64Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_bigint64_array);
  } else if (arrayType == "biguint64") {
    return buffer.IsUndefined()
               ? NAPI_TYPEDARRAY_NEW(
                     BigUint64Array, info.Env(), length, napi_biguint64_array)
               : NAPI_TYPEDARRAY_NEW_BUFFER(BigUint64Array,
                                            info.Env(),
                                            length,
                                            buffer.As<ArrayBuffer>(),
                                            bufferOffset,
                                            napi_biguint64_array);
#endif
  } else {
    Error::New(info.Env(), "Invalid typed-array type.")
        .ThrowAsJavaScriptException();
    return Value();
  }
}

Value CreateInvalidTypedArray(const CallbackInfo& info) {
  return NAPI_TYPEDARRAY_NEW_BUFFER(
      Int8Array, info.Env(), 1, ArrayBuffer(), 0, napi_int8_array);
}

Value GetTypedArrayType(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  switch (array.TypedArrayType()) {
    case napi_int8_array:
      return String::New(info.Env(), "int8");
    case napi_uint8_array:
      return String::New(info.Env(), "uint8");
    case napi_uint8_clamped_array:
      return String::New(info.Env(), "uint8_clamped");
    case napi_int16_array:
      return String::New(info.Env(), "int16");
    case napi_uint16_array:
      return String::New(info.Env(), "uint16");
    case napi_int32_array:
      return String::New(info.Env(), "int32");
    case napi_uint32_array:
      return String::New(info.Env(), "uint32");
    case napi_float32_array:
      return String::New(info.Env(), "float32");
    case napi_float64_array:
      return String::New(info.Env(), "float64");
#if (NAPI_VERSION > 5)
    case napi_bigint64_array:
      return String::New(info.Env(), "bigint64");
    case napi_biguint64_array:
      return String::New(info.Env(), "biguint64");
#endif
    default:
      return String::New(info.Env(), "invalid");
  }
}

template <typename type>
bool TypedArrayDataIsEquivalent(TypedArrayOf<type> arr,
                                TypedArrayOf<type> inputArr) {
  if (arr.ElementLength() != inputArr.ElementLength()) {
    return false;
  }
  std::vector<type> bufferContent(arr.Data(), arr.Data() + arr.ElementLength());
  std::vector<type> inputContent(inputArr.Data(),
                                 inputArr.Data() + inputArr.ElementLength());
  if (bufferContent != inputContent) {
    return false;
  }
  return true;
}

Value CheckBufferContent(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();

  switch (array.TypedArrayType()) {
    case napi_int8_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<int8_t>(info[0].As<Int8Array>(),
                                             info[1].As<Int8Array>()));

      break;
    case napi_uint8_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<uint8_t>(info[0].As<Uint8Array>(),
                                              info[1].As<Uint8Array>()));

    case napi_uint8_clamped_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<uint8_t>(info[0].As<Uint8Array>(),
                                              info[1].As<Uint8Array>()));

    case napi_int16_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<int16_t>(info[0].As<Int16Array>(),
                                              info[1].As<Int16Array>()));

    case napi_uint16_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<uint16_t>(info[0].As<Uint16Array>(),
                                               info[1].As<Uint16Array>()));

    case napi_int32_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<int32_t>(info[0].As<Int32Array>(),
                                              info[1].As<Int32Array>()));

    case napi_uint32_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<uint32_t>(info[0].As<Uint32Array>(),
                                               info[1].As<Uint32Array>()));

    case napi_float32_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<float>(info[0].As<Float32Array>(),
                                            info[1].As<Float32Array>()));

    case napi_float64_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<double>(info[0].As<Float64Array>(),
                                             info[1].As<Float64Array>()));

#if (NAPI_VERSION > 5)
    case napi_bigint64_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<int64_t>(info[0].As<BigInt64Array>(),
                                              info[1].As<BigInt64Array>()));

    case napi_biguint64_array:
      return Boolean::New(
          info.Env(),
          TypedArrayDataIsEquivalent<uint64_t>(info[0].As<BigUint64Array>(),
                                               info[1].As<BigUint64Array>()));

#endif
    default:
      return Boolean::New(info.Env(), false);
  }
}

Value GetTypedArrayLength(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  return Number::New(info.Env(), static_cast<double>(array.ElementLength()));
}

Value GetTypedArraySize(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  return Number::New(info.Env(), static_cast<double>(array.ElementSize()));
}

Value GetTypedArrayByteOffset(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  return Number::New(info.Env(), static_cast<double>(array.ByteOffset()));
}

Value GetTypedArrayByteLength(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  return Number::New(info.Env(), static_cast<double>(array.ByteLength()));
}

Value GetTypedArrayBuffer(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  return array.ArrayBuffer();
}

Value GetTypedArrayElement(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  size_t index = info[1].As<Number>().Uint32Value();
  switch (array.TypedArrayType()) {
    case napi_int8_array:
      return Number::New(info.Env(), array.As<Int8Array>()[index]);
    case napi_uint8_array:
      return Number::New(info.Env(), array.As<Uint8Array>()[index]);
    case napi_uint8_clamped_array:
      return Number::New(info.Env(), array.As<Uint8Array>()[index]);
    case napi_int16_array:
      return Number::New(info.Env(), array.As<Int16Array>()[index]);
    case napi_uint16_array:
      return Number::New(info.Env(), array.As<Uint16Array>()[index]);
    case napi_int32_array:
      return Number::New(info.Env(), array.As<Int32Array>()[index]);
    case napi_uint32_array:
      return Number::New(info.Env(), array.As<Uint32Array>()[index]);
    case napi_float32_array:
      return Number::New(info.Env(), array.As<Float32Array>()[index]);
    case napi_float64_array:
      return Number::New(info.Env(), array.As<Float64Array>()[index]);
#if (NAPI_VERSION > 5)
    case napi_bigint64_array:
      return BigInt::New(info.Env(), array.As<BigInt64Array>()[index]);
    case napi_biguint64_array:
      return BigInt::New(info.Env(), array.As<BigUint64Array>()[index]);
#endif
    default:
      Error::New(info.Env(), "Invalid typed-array type.")
          .ThrowAsJavaScriptException();
      return Value();
  }
}

void SetTypedArrayElement(const CallbackInfo& info) {
  TypedArray array = info[0].As<TypedArray>();
  size_t index = info[1].As<Number>().Uint32Value();
  Value value = info[2];
  switch (array.TypedArrayType()) {
    case napi_int8_array:
      array.As<Int8Array>()[index] =
          static_cast<int8_t>(value.As<Number>().Int32Value());
      break;
    case napi_uint8_array:
      array.As<Uint8Array>()[index] =
          static_cast<uint8_t>(value.As<Number>().Uint32Value());
      break;
    case napi_uint8_clamped_array:
      array.As<Uint8Array>()[index] =
          static_cast<uint8_t>(value.As<Number>().Uint32Value());
      break;
    case napi_int16_array:
      array.As<Int16Array>()[index] =
          static_cast<int16_t>(value.As<Number>().Int32Value());
      break;
    case napi_uint16_array:
      array.As<Uint16Array>()[index] =
          static_cast<uint16_t>(value.As<Number>().Uint32Value());
      break;
    case napi_int32_array:
      array.As<Int32Array>()[index] = value.As<Number>().Int32Value();
      break;
    case napi_uint32_array:
      array.As<Uint32Array>()[index] = value.As<Number>().Uint32Value();
      break;
    case napi_float32_array:
      array.As<Float32Array>()[index] = value.As<Number>().FloatValue();
      break;
    case napi_float64_array:
      array.As<Float64Array>()[index] = value.As<Number>().DoubleValue();
      break;
#if (NAPI_VERSION > 5)
    case napi_bigint64_array: {
      bool lossless;
      array.As<BigInt64Array>()[index] =
          value.As<BigInt>().Int64Value(&lossless);
      break;
    }
    case napi_biguint64_array: {
      bool lossless;
      array.As<BigUint64Array>()[index] =
          value.As<BigInt>().Uint64Value(&lossless);
      break;
    }
#endif
    default:
      Error::New(info.Env(), "Invalid typed-array type.")
          .ThrowAsJavaScriptException();
  }
}

}  // end anonymous namespace

Object InitTypedArray(Env env) {
  Object exports = Object::New(env);

  exports["createTypedArray"] = Function::New(env, CreateTypedArray);
  exports["createInvalidTypedArray"] =
      Function::New(env, CreateInvalidTypedArray);
  exports["getTypedArrayType"] = Function::New(env, GetTypedArrayType);
  exports["getTypedArrayLength"] = Function::New(env, GetTypedArrayLength);
  exports["getTypedArraySize"] = Function::New(env, GetTypedArraySize);
  exports["getTypedArrayByteOffset"] =
      Function::New(env, GetTypedArrayByteOffset);
  exports["getTypedArrayByteLength"] =
      Function::New(env, GetTypedArrayByteLength);
  exports["getTypedArrayBuffer"] = Function::New(env, GetTypedArrayBuffer);
  exports["getTypedArrayElement"] = Function::New(env, GetTypedArrayElement);
  exports["setTypedArrayElement"] = Function::New(env, SetTypedArrayElement);
  exports["checkBufferContent"] = Function::New(env, CheckBufferContent);

  return exports;
}