#pragma once
#include <types.hpp>
#include <cstring>
#include <functional>
#include <bit>
#include <algorithm>
#include <vector>
#include <concepts>

namespace ircolib {
static inline std::vector<u8> IntegralToBuffer(const std::integral auto &val) {
  std::vector<u8> ret{};
  ret.resize(sizeof(val));

  memcpy(ret.data(), &val, sizeof(val));

  return ret;
}

static inline constexpr bool IsInsideRange(const std::integral auto& addr,
                                                 const std::integral auto& start,
                                                 const std::integral auto& end) {
  return addr >= start && addr <= end;
}

template <typename T>
static constexpr inline T ReadAccess(const u8 *data, const u32 index);
template <typename T>
static constexpr inline T ReadAccess(const std::vector<u8> &data, const u32 index);
template <typename T, size_t Size>
static constexpr inline T ReadAccess(const std::array<u8, Size> &data, const u32 index);

template <typename T>
static constexpr inline void WriteAccess(u8 *data, const u32 index, const T val);
template <typename T>
static constexpr inline void WriteAccess(std::vector<u8> &data, const u32 index, const T val);
template <typename T, size_t Size>
static constexpr inline void WriteAccess(std::array<u8, Size> &data, const u32 index, const T val);

template <>
constexpr inline u64 ReadAccess(const u8 *data, const u32 index) {
  u32 hi = *reinterpret_cast<const u32 *>(&data[index + 0]);
  u32 lo = *reinterpret_cast<const u32 *>(&data[index + 4]);
  const auto& result = static_cast<u64>(hi) << 32 | static_cast<u64>(lo);
  return result;
}

template <typename T>
static constexpr inline T ReadAccess(const u8 *data, const u32 index) {
  return *reinterpret_cast<const T *>(&data[index]);
}

template <>
constexpr inline u64 ReadAccess(const std::vector<u8> &data, const u32 index) {
  u32 hi = *reinterpret_cast<const u32 *>(&data[index + 0]);
  u32 lo = *reinterpret_cast<const u32 *>(&data[index + 4]);
  return (static_cast<u64>(hi) << 32) | static_cast<u64>(lo);
}

template <typename T>
static constexpr inline T ReadAccess(const std::vector<u8> &data, const u32 index) {
  return *reinterpret_cast<const T *>(&data[index]);
}

template <size_t Size>
constexpr inline u64 ReadAccess(const std::array<u8, Size> &data, const u32 index) {
  u32 hi = *reinterpret_cast<const u32 *>(&data[index + 0]);
  u32 lo = *reinterpret_cast<const u32 *>(&data[index + 4]);
  return static_cast<u64>(hi) << 32 | static_cast<u64>(lo);
}

template <typename T, size_t Size>
static constexpr inline T ReadAccess(const std::array<u8, Size> &data, const u32 index) {
  return *reinterpret_cast<const T *>(&data[index]);
}

template <size_t Size>
constexpr inline void WriteAccess(std::array<u8, Size> &data, const u32 index, const u64 val) {
  const u32 hi = val >> 32;
  const u32 lo = val;

  *reinterpret_cast<u32 *>(&data[index + 0]) = hi;
  *reinterpret_cast<u32 *>(&data[index + 4]) = lo;
}

template <typename T, size_t Size>
static constexpr inline void WriteAccess(std::array<u8, Size> &data, const u32 index, const T val) {
  *reinterpret_cast<T *>(&data[index]) = val;
}

template <>
constexpr inline void WriteAccess(std::vector<u8> &data, const u32 index, const u64 val) {
  const u32 hi = val >> 32;
  const u32 lo = val;

  *reinterpret_cast<u32 *>(&data[index + 0]) = hi;
  *reinterpret_cast<u32 *>(&data[index + 4]) = lo;
}

template <typename T>
static constexpr inline void WriteAccess(std::vector<u8> &data, const u32 index, const T val) {
  *reinterpret_cast<T *>(&data[index]) = val;
}

template <>
constexpr inline void WriteAccess(u8 *data, const u32 index, const u64 val) {
  const u32 hi = val >> 32;
  const u32 lo = val;

  *reinterpret_cast<u32 *>(&data[index + 0]) = hi;
  *reinterpret_cast<u32 *>(&data[index + 4]) = lo;
}

template <typename T>
static constexpr inline void WriteAccess(u8 *data, const u32 index, const T val) {
  *reinterpret_cast<T *>(&data[index]) = val;
}

template <typename T>
static constexpr inline void SwapBuffer(std::vector<u8> &data) {
  for (size_t i = 0; i < data.size(); i += sizeof(T)) {
    const T original = *reinterpret_cast<T *>(&data[i]);
    *reinterpret_cast<T *>(&data[i]) = std::byteswap(original);
  }
}

template <typename T, size_t Size>
static constexpr inline void SwapBuffer(std::array<u8, Size> &data) {
  for (size_t i = 0; i < data.size(); i += sizeof(T)) {
    const T original = *reinterpret_cast<T *>(&data[i]);
    *reinterpret_cast<T *>(&data[i]) = std::byteswap(original);
  }
}

#ifdef _WIN32
inline void *aligned_alloc(const size_t alignment, const size_t size) { return _aligned_malloc(size, alignment); }

inline void aligned_free(void *ptr) { _aligned_free(ptr); }
#else
inline void *aligned_alloc(const size_t alignment, const size_t size) {
  return std::aligned_alloc(alignment, size);
}

inline void aligned_free(void *ptr) { std::free(ptr); }
#endif
} // namespace Util