/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * 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 ART_RUNTIME_UTILS_DEX_CACHE_ARRAYS_LAYOUT_INL_H_
#define ART_RUNTIME_UTILS_DEX_CACHE_ARRAYS_LAYOUT_INL_H_

#include "dex_cache_arrays_layout.h"

#include <android-base/logging.h>

#include "base/bit_utils.h"
#include "dex/primitive.h"
#include "gc_root.h"
#include "globals.h"
#include "mirror/dex_cache.h"

namespace art {

inline DexCacheArraysLayout::DexCacheArraysLayout(PointerSize pointer_size,
                                                  const DexFile::Header& header,
                                                  uint32_t num_call_sites)
    : pointer_size_(pointer_size),
      /* types_offset_ is always 0u, so it's constexpr */
      methods_offset_(
          RoundUp(types_offset_ + TypesSize(header.type_ids_size_), MethodsAlignment())),
      strings_offset_(
          RoundUp(methods_offset_ + MethodsSize(header.method_ids_size_), StringsAlignment())),
      fields_offset_(
          RoundUp(strings_offset_ + StringsSize(header.string_ids_size_), FieldsAlignment())),
      method_types_offset_(
          RoundUp(fields_offset_ + FieldsSize(header.field_ids_size_), MethodTypesAlignment())),
    call_sites_offset_(
        RoundUp(method_types_offset_ + MethodTypesSize(header.proto_ids_size_),
                MethodTypesAlignment())),
      size_(RoundUp(call_sites_offset_ + CallSitesSize(num_call_sites), Alignment())) {
}

inline DexCacheArraysLayout::DexCacheArraysLayout(PointerSize pointer_size, const DexFile* dex_file)
    : DexCacheArraysLayout(pointer_size, dex_file->GetHeader(), dex_file->NumCallSiteIds()) {
}

inline size_t DexCacheArraysLayout::Alignment() const {
  return Alignment(pointer_size_);
}

inline constexpr size_t DexCacheArraysLayout::Alignment(PointerSize pointer_size) {
  // mirror::Type/String/MethodTypeDexCacheType alignment is 8,
  // i.e. higher than or equal to the pointer alignment.
  static_assert(alignof(mirror::TypeDexCacheType) == 8,
                "Expecting alignof(ClassDexCacheType) == 8");
  static_assert(alignof(mirror::StringDexCacheType) == 8,
                "Expecting alignof(StringDexCacheType) == 8");
  static_assert(alignof(mirror::MethodTypeDexCacheType) == 8,
                "Expecting alignof(MethodTypeDexCacheType) == 8");
  // This is the same as alignof({Field,Method}DexCacheType) for the given pointer size.
  return 2u * static_cast<size_t>(pointer_size);
}

template <typename T>
constexpr PointerSize GcRootAsPointerSize() {
  static_assert(sizeof(GcRoot<T>) == 4U, "Unexpected GcRoot size");
  return PointerSize::k32;
}

inline size_t DexCacheArraysLayout::TypeOffset(dex::TypeIndex type_idx) const {
  return types_offset_ + ElementOffset(PointerSize::k64,
                                       type_idx.index_ % mirror::DexCache::kDexCacheTypeCacheSize);
}

inline size_t DexCacheArraysLayout::TypesSize(size_t num_elements) const {
  size_t cache_size = mirror::DexCache::kDexCacheTypeCacheSize;
  if (num_elements < cache_size) {
    cache_size = num_elements;
  }
  return PairArraySize(GcRootAsPointerSize<mirror::Class>(), cache_size);
}

inline size_t DexCacheArraysLayout::TypesAlignment() const {
  return alignof(GcRoot<mirror::Class>);
}

inline size_t DexCacheArraysLayout::MethodOffset(uint32_t method_idx) const {
  return methods_offset_ + ElementOffset(pointer_size_, method_idx);
}

inline size_t DexCacheArraysLayout::MethodsSize(size_t num_elements) const {
  size_t cache_size = mirror::DexCache::kDexCacheMethodCacheSize;
  if (num_elements < cache_size) {
    cache_size = num_elements;
  }
  return PairArraySize(pointer_size_, cache_size);
}

inline size_t DexCacheArraysLayout::MethodsAlignment() const {
  return 2u * static_cast<size_t>(pointer_size_);
}

inline size_t DexCacheArraysLayout::StringOffset(uint32_t string_idx) const {
  uint32_t string_hash = string_idx % mirror::DexCache::kDexCacheStringCacheSize;
  return strings_offset_ + ElementOffset(PointerSize::k64, string_hash);
}

inline size_t DexCacheArraysLayout::StringsSize(size_t num_elements) const {
  size_t cache_size = mirror::DexCache::kDexCacheStringCacheSize;
  if (num_elements < cache_size) {
    cache_size = num_elements;
  }
  return PairArraySize(GcRootAsPointerSize<mirror::String>(), cache_size);
}

inline size_t DexCacheArraysLayout::StringsAlignment() const {
  static_assert(alignof(mirror::StringDexCacheType) == 8,
                "Expecting alignof(StringDexCacheType) == 8");
  return alignof(mirror::StringDexCacheType);
}

inline size_t DexCacheArraysLayout::FieldOffset(uint32_t field_idx) const {
  uint32_t field_hash = field_idx % mirror::DexCache::kDexCacheFieldCacheSize;
  return fields_offset_ + 2u * static_cast<size_t>(pointer_size_) * field_hash;
}

inline size_t DexCacheArraysLayout::FieldsSize(size_t num_elements) const {
  size_t cache_size = mirror::DexCache::kDexCacheFieldCacheSize;
  if (num_elements < cache_size) {
    cache_size = num_elements;
  }
  return PairArraySize(pointer_size_, cache_size);
}

inline size_t DexCacheArraysLayout::FieldsAlignment() const {
  return 2u * static_cast<size_t>(pointer_size_);
}

inline size_t DexCacheArraysLayout::MethodTypesSize(size_t num_elements) const {
  size_t cache_size = mirror::DexCache::kDexCacheMethodTypeCacheSize;
  if (num_elements < cache_size) {
    cache_size = num_elements;
  }

  return ArraySize(PointerSize::k64, cache_size);
}

inline size_t DexCacheArraysLayout::MethodTypesAlignment() const {
  static_assert(alignof(mirror::MethodTypeDexCacheType) == 8,
                "Expecting alignof(MethodTypeDexCacheType) == 8");
  return alignof(mirror::MethodTypeDexCacheType);
}

inline size_t DexCacheArraysLayout::CallSitesSize(size_t num_elements) const {
  return ArraySize(GcRootAsPointerSize<mirror::CallSite>(), num_elements);
}

inline size_t DexCacheArraysLayout::CallSitesAlignment() const {
  return alignof(GcRoot<mirror::CallSite>);
}

inline size_t DexCacheArraysLayout::ElementOffset(PointerSize element_size, uint32_t idx) {
  return static_cast<size_t>(element_size) * idx;
}

inline size_t DexCacheArraysLayout::ArraySize(PointerSize element_size, uint32_t num_elements) {
  return static_cast<size_t>(element_size) * num_elements;
}

inline size_t DexCacheArraysLayout::PairArraySize(PointerSize element_size, uint32_t num_elements) {
  return 2u * static_cast<size_t>(element_size) * num_elements;
}

}  // namespace art

#endif  // ART_RUNTIME_UTILS_DEX_CACHE_ARRAYS_LAYOUT_INL_H_