// Copyright 2024-2025, Amlal El Mahrouss (amlal@nekernel.org)
// Distributed under the Apache Software License, Version 2.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.apache.org/licenses/LICENSE-2.0)
// Official repository: https://github.com/ne-foss-org/neboot

#ifndef __NB_FDT_H__
#define __NB_FDT_H__

#include <include/boot.h>

#ifdef __USE_SPARSE__
#define FDT_FORCE __attribute__((force))
#define FDT_BITWISE __attribute__((bitwise))
#else
#define FDT_FORCE
#define FDT_BITWISE
#endif

typedef uint16_t FDT_BITWISE fdt16_t;
typedef uint32_t FDT_BITWISE fdt32_t;
typedef uint64_t FDT_BITWISE fdt64_t;

#define EXTRACT_BYTE(x, n) ((unsigned long long) ((uint8_t*) &x)[n])
#define CPU_TO_FDT16(x) ((EXTRACT_BYTE(x, 0) << 8) | EXTRACT_BYTE(x, 1))
#define CPU_TO_FDT32(x)                                                                  \
  ((EXTRACT_BYTE(x, 0) << 24) | (EXTRACT_BYTE(x, 1) << 16) | (EXTRACT_BYTE(x, 2) << 8) | \
   EXTRACT_BYTE(x, 3))
#define CPU_TO_FDT64(x)                                                                   \
  ((EXTRACT_BYTE(x, 0) << 56) | (EXTRACT_BYTE(x, 1) << 48) | (EXTRACT_BYTE(x, 2) << 40) | \
   (EXTRACT_BYTE(x, 3) << 32) | (EXTRACT_BYTE(x, 4) << 24) | (EXTRACT_BYTE(x, 5) << 16) | \
   (EXTRACT_BYTE(x, 6) << 8) | EXTRACT_BYTE(x, 7))

static inline uint16_t fdt16_to_cpu(fdt16_t x) {
  return (FDT_FORCE uint16_t) CPU_TO_FDT16(x);
}

static inline fdt16_t cpu_to_fdt16(uint16_t x) {
  return (FDT_FORCE fdt16_t) CPU_TO_FDT16(x);
}

static inline uint32_t fdt32_to_cpu(fdt32_t x) {
  return (FDT_FORCE uint32_t) CPU_TO_FDT32(x);
}

static inline fdt32_t cpu_to_fdt32(uint32_t x) {
  return (FDT_FORCE fdt32_t) CPU_TO_FDT32(x);
}

static inline uint64_t fdt64_to_cpu(fdt64_t x) {
  return (FDT_FORCE uint64_t) CPU_TO_FDT64(x);
}

static inline fdt64_t cpu_to_fdt64(uint64_t x) {
  return (FDT_FORCE fdt64_t) CPU_TO_FDT64(x);
}

#undef CPU_TO_FDT64
#undef CPU_TO_FDT32
#undef CPU_TO_FDT16
#undef EXTRACT_BYTE

#endif  //!__NB_FDT_H__