// ArduinoJson - https://arduinojson.org

// Copyright © 2014-2025, Benoit BLANCHON

// MIT License

#pragma once

#include <ArduinoJson/Variant/VariantCompare.hpp>

#include <ArduinoJson/Variant/VariantImpl.hpp>

ARDUINOJSON_BEGIN_PRIVATE_NAMESPACE

inline VariantImpl::iterator VariantImpl::at(size_t index) const {

  if (!isArray())

    return iterator();

  auto it = createIterator();

  while (!it.done() && index) {

    it.move(resources_);

    --index;

  }

  return it;

}

inline VariantData* VariantImpl::addNewElement(VariantData* data,

                                               ResourceManager* resources) {

  ARDUINOJSON_ASSERT(data != nullptr);

  ARDUINOJSON_ASSERT(data->isArray());

  ARDUINOJSON_ASSERT(resources != nullptr);

  auto slot = resources->allocVariant();

  if (!slot)

    return nullptr;

  addElement(slot, data, resources);

  return slot.ptr();

}

inline VariantData* VariantImpl::getOrAddElement(size_t index) {

  auto it = createIterator();

  while (!it.done() && index > 0) {

    it.move(resources_);

    index--;

  }

  if (it.done())

    index++;

  VariantData* element = it.data();

  while (index > 0) {

    element = addNewElement();

    if (!element)

      return nullptr;

    index--;

  }

  return element;

}

inline VariantData* VariantImpl::getElement(size_t index) const {

  return at(index).data();

}

inline void VariantImpl::removeElement(size_t index) {

  removeElement(at(index));

}

// Returns the size (in bytes) of an array with n elements.

constexpr size_t sizeofArray(size_t n) {

  return n * sizeof(VariantData);

}

ARDUINOJSON_END_PRIVATE_NAMESPACE