-- |
-- Module      : Endian
-- License     : BSD-3-Clause
-- Copyright   : (c) 2026 Olivier Chéron
--
-- Endianness utilities
--
{-# LANGUAGE CPP #-}
module Endian
    ( B.BE, B.LE, fromLE, toLE
    ) where

#include "MachDeps.h"

-- Little-endian conversion in `memory` / `ram` is avoided at compile
-- time only for AMD/Intel, here we will short circuit on ARM too
#if (defined(arm_HOST_ARCH) || defined(aarch64_HOST_ARCH)) \
    && !defined(WORDS_BIGENDIAN)
#define MLKEM_FORCE_LITTLE_ENDIAN_ARCH 1
#endif

#ifdef MLKEM_FORCE_LITTLE_ENDIAN_ARCH
#else
import Data.Memory.Endian (ByteSwap)
#endif

import qualified Data.Memory.Endian as B

#ifdef MLKEM_FORCE_LITTLE_ENDIAN_ARCH
fromLE :: B.LE a -> a
fromLE = B.unLE  -- unwrap constructor with no byte swapping
#else
fromLE :: ByteSwap a => B.LE a -> a
fromLE :: forall a. ByteSwap a => LE a -> a
fromLE = LE a -> a
forall a. ByteSwap a => LE a -> a
B.fromLE  -- byte swap if necessary
#endif

#ifdef MLKEM_FORCE_LITTLE_ENDIAN_ARCH
toLE :: a -> B.LE a
toLE = B.LE  -- wrap constructor with no byte swapping
#else
toLE :: ByteSwap a => a -> B.LE a
toLE :: forall a. ByteSwap a => a -> LE a
toLE = a -> LE a
forall a. ByteSwap a => a -> LE a
B.toLE  -- byte swap if necessary
#endif