// Copyright (c) 2012- PPSSPP Project / Dolphin Project.

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, version 2.0 or later versions.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License 2.0 for more details.

// A copy of the GPL 2.0 should have been included with the program.
// If not, see http://www.gnu.org/licenses/

// Official git repository and contact information can be found at
// https://github.com/hrydgard/ppsspp and http://www.ppsspp.org/.

#pragma once

#if defined(_MSC_VER)
    #include <cstdlib>
#elif defined(__linux__)
    #include <byteswap.h>
#elif defined(__FreeBSD__)
    #include <sys/endian.h>
#endif

#include "common/common_types.h"

// GCC 4.6+
#if __GNUC__ >= 5 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 6)

#if __BYTE_ORDER__ && (__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__) && !defined(COMMON_LITTLE_ENDIAN)
#define COMMON_LITTLE_ENDIAN 1
#elif __BYTE_ORDER__ && (__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__) && !defined(COMMON_BIG_ENDIAN)
#define COMMON_BIG_ENDIAN 1
#endif

// LLVM/clang
#elif __clang__

#if __LITTLE_ENDIAN__ && !defined(COMMON_LITTLE_ENDIAN)
#define COMMON_LITTLE_ENDIAN 1
#elif __BIG_ENDIAN__ && !defined(COMMON_BIG_ENDIAN)
#define COMMON_BIG_ENDIAN 1
#endif

// MSVC
#elif defined(_MSC_VER) && !defined(COMMON_BIG_ENDIAN) && !defined(COMMON_LITTLE_ENDIAN)

#define COMMON_LITTLE_ENDIAN 1
#endif

// Worst case, default to little endian.
#if !COMMON_BIG_ENDIAN && !COMMON_LITTLE_ENDIAN
#define COMMON_LITTLE_ENDIAN 1
#endif

namespace Common {

inline u8 swap8(u8 _data) {return _data;}
inline u32 swap24(const u8* _data) {return (_data[0] << 16) | (_data[1] << 8) | _data[2];}

#ifdef _MSC_VER
inline u16 swap16(u16 _data) {return _byteswap_ushort(_data);}
inline u32 swap32(u32 _data) {return _byteswap_ulong (_data);}
inline u64 swap64(u64 _data) {return _byteswap_uint64(_data);}
#elif _M_ARM
inline u16 swap16 (u16 _data) { u32 data = _data; __asm__ ("rev16 %0, %1\n" : "=l" (data) : "l" (data)); return (u16)data;}
inline u32 swap32 (u32 _data) {__asm__ ("rev %0, %1\n" : "=l" (_data) : "l" (_data)); return _data;}
inline u64 swap64(u64 _data) {return ((u64)swap32(_data) << 32) | swap32(_data >> 32);}
#elif __linux__
inline u16 swap16(u16 _data) {return bswap_16(_data);}
inline u32 swap32(u32 _data) {return bswap_32(_data);}
inline u64 swap64(u64 _data) {return bswap_64(_data);}
#elif __APPLE__
inline __attribute__((always_inline)) u16 swap16(u16 _data)
{return (_data >> 8) | (_data << 8);}
inline __attribute__((always_inline)) u32 swap32(u32 _data)
{return __builtin_bswap32(_data);}
inline __attribute__((always_inline)) u64 swap64(u64 _data)
{return __builtin_bswap64(_data);}
#elif __FreeBSD__
inline u16 swap16(u16 _data) {return bswap16(_data);}
inline u32 swap32(u32 _data) {return bswap32(_data);}
inline u64 swap64(u64 _data) {return bswap64(_data);}
#else
// Slow generic implementation.
inline u16 swap16(u16 data) {return (data >> 8) | (data << 8);}
inline u32 swap32(u32 data) {return (swap16(data) << 16) | swap16(data >> 16);}
inline u64 swap64(u64 data) {return ((u64)swap32(data) << 32) | swap32(data >> 32);}
#endif

inline float swapf(float f) {
    union {
        float f;
        unsigned int u32;
    } dat1, dat2;

    dat1.f = f;
    dat2.u32 = swap32(dat1.u32);

    return dat2.f;
}

inline double swapd(double f) {
    union  {
        double f;
        unsigned long long u64;
    } dat1, dat2;

    dat1.f = f;
    dat2.u64 = swap64(dat1.u64);

    return dat2.f;
}

inline u16 swap16(const u8* _pData) {return swap16(*(const u16*)_pData);}
inline u32 swap32(const u8* _pData) {return swap32(*(const u32*)_pData);}
inline u64 swap64(const u8* _pData) {return swap64(*(const u64*)_pData);}

template <int count>
void swap(u8*);

template <>
inline void swap<1>(u8* data) { }

template <>
inline void swap<2>(u8* data) {
    *reinterpret_cast<u16*>(data) = swap16(data);
}

template <>
inline void swap<4>(u8* data) {
    *reinterpret_cast<u32*>(data) = swap32(data);
}

template <>
inline void swap<8>(u8* data) {
    *reinterpret_cast<u64*>(data) = swap64(data);
}

}  // Namespace Common


template <typename T, typename F>
struct swap_struct_t {
    typedef swap_struct_t<T, F> swapped_t;

protected:
    T value = T();

    static T swap(T v) {
        return F::swap(v);
    }
public:
    T const swap() const {
        return swap(value);

    }
    swap_struct_t() = default;
    swap_struct_t(const T &v): value(swap(v)) {}

    template <typename S>
    swapped_t& operator=(const S &source) {
        value = swap((T)source);
        return *this;
    }

    operator s8() const { return (s8)swap(); }
    operator u8() const { return (u8)swap(); }
    operator s16() const { return (s16)swap(); }
    operator u16() const { return (u16)swap(); }
    operator s32() const { return (s32)swap(); }
    operator u32() const { return (u32)swap(); }
    operator s64() const { return (s64)swap(); }
    operator u64() const { return (u64)swap(); }
    operator float() const { return (float)swap(); }
    operator double() const { return (double)swap(); }

    // +v
    swapped_t operator +() const {
        return +swap();
    }
    // -v
    swapped_t operator -() const {
        return -swap();
    }

    // v / 5
    swapped_t operator/(const swapped_t &i) const {
        return swap() / i.swap();
    }
    template <typename S>
    swapped_t operator/(const S &i) const {
        return swap() / i;
    }

    // v * 5
    swapped_t operator*(const swapped_t &i) const {
        return swap() * i.swap();
    }
    template <typename S>
    swapped_t operator*(const S &i) const {
        return swap() * i;
    }

    // v + 5
    swapped_t operator+(const swapped_t &i) const {
        return swap() + i.swap();
    }
    template <typename S>
    swapped_t operator+(const S &i) const {
        return swap() + (T)i;
    }
    // v - 5
    swapped_t operator-(const swapped_t &i) const {
        return swap() - i.swap();
    }
    template <typename S>
    swapped_t operator-(const S &i) const {
        return swap() - (T)i;
    }

    // v += 5
    swapped_t& operator+=(const swapped_t &i) {
        value = swap(swap() + i.swap());
        return *this;
    }
    template <typename S>
    swapped_t& operator+=(const S &i) {
        value = swap(swap() + (T)i);
        return *this;
    }
    // v -= 5
    swapped_t& operator-=(const swapped_t &i) {
        value = swap(swap() - i.swap());
        return *this;
    }
    template <typename S>
    swapped_t& operator-=(const S &i) {
        value = swap(swap() - (T)i);
        return *this;
    }

    // ++v
    swapped_t& operator++() {
        value = swap(swap()+1);
        return *this;
    }
    // --v
    swapped_t& operator--()  {
        value = swap(swap()-1);
        return *this;
    }

    // v++
    swapped_t operator++(int) {
        swapped_t old = *this;
        value = swap(swap()+1);
        return old;
    }
    // v--
    swapped_t operator--(int) {
        swapped_t old = *this;
        value = swap(swap()-1);
        return old;
    }
    // Comparaison
    // v == i
    bool operator==(const swapped_t &i) const {
        return swap() == i.swap();
    }
    template <typename S>
    bool operator==(const S &i) const {
        return swap() == i;
    }

    // v != i
    bool operator!=(const swapped_t &i) const {
        return swap() != i.swap();
    }
    template <typename S>
    bool operator!=(const S &i) const {
        return swap() != i;
    }

    // v > i
    bool operator>(const swapped_t &i) const {
        return swap() > i.swap();
    }
    template <typename S>
    bool operator>(const S &i) const {
        return swap() > i;
    }

    // v < i
    bool operator<(const swapped_t &i) const {
        return swap() < i.swap();
    }
    template <typename S>
    bool operator<(const S &i) const {
        return swap() < i;
    }

    // v >= i
    bool operator>=(const swapped_t &i) const {
        return swap() >= i.swap();
    }
    template <typename S>
    bool operator>=(const S &i) const {
        return swap() >= i;
    }

    // v <= i
    bool operator<=(const swapped_t &i) const {
        return swap() <= i.swap();
    }
    template <typename S>
    bool operator<=(const S &i) const {
        return swap() <= i;
    }

    // logical
    swapped_t operator !() const {
        return !swap();
    }

    // bitmath
    swapped_t operator ~() const {
        return ~swap();
    }

    swapped_t operator &(const swapped_t &b) const {
        return swap() & b.swap();
    }
    template <typename S>
    swapped_t operator &(const S &b) const {
        return swap() & b;
    }
    swapped_t& operator &=(const swapped_t &b) {
        value = swap(swap() & b.swap());
        return *this;
    }
    template <typename S>
    swapped_t& operator &=(const S b) {
        value = swap(swap() & b);
        return *this;
    }

    swapped_t operator |(const swapped_t &b) const {
        return swap() | b.swap();
    }
    template <typename S>
    swapped_t operator |(const S &b) const {
        return swap() | b;
    }
    swapped_t& operator |=(const swapped_t &b) {
        value = swap(swap() | b.swap());
        return *this;
    }
    template <typename S>
    swapped_t& operator |=(const S &b) {
        value = swap(swap() | b);
        return *this;
    }

    swapped_t operator ^(const swapped_t &b) const {
        return swap() ^ b.swap();
    }
    template <typename S>
    swapped_t operator ^(const S &b) const {
        return swap() ^ b;
    }
    swapped_t& operator ^=(const swapped_t &b) {
        value = swap(swap() ^ b.swap());
        return *this;
    }
    template <typename S>
    swapped_t& operator ^=(const S &b) {
        value = swap(swap() ^ b);
        return *this;
    }

    template <typename S>
    swapped_t operator <<(const S &b) const {
        return swap() << b;
    }
    template <typename S>
    swapped_t& operator <<=(const S &b) const {
        value = swap(swap() << b);
        return *this;
    }

    template <typename S>
    swapped_t operator >>(const S &b) const {
        return swap() >> b;
    }
    template <typename S>
    swapped_t& operator >>=(const S &b) const {
        value = swap(swap() >> b);
        return *this;
    }

    // Member
    /** todo **/


    // Arithmetics
    template <typename S, typename T2, typename F2>
    friend S operator+(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend S operator-(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend S operator/(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend S operator*(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend S operator%(const S &p, const swapped_t v);

    // Arithmetics + assignements
    template <typename S, typename T2, typename F2>
    friend S operator+=(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend S operator-=(const S &p, const swapped_t v);

    // Bitmath
    template <typename S, typename T2, typename F2>
    friend S operator&(const S &p, const swapped_t v);

    // Comparison
    template <typename S, typename T2, typename F2>
    friend bool operator<(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend bool operator>(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend bool operator<=(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend bool operator>=(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend bool operator!=(const S &p, const swapped_t v);

    template <typename S, typename T2, typename F2>
    friend bool operator==(const S &p, const swapped_t v);
};


// Arithmetics
template <typename S, typename T, typename F>
S operator+(const S &i, const swap_struct_t<T, F> v) {
    return i + v.swap();
}

template <typename S, typename T, typename F>
S operator-(const S &i, const swap_struct_t<T, F> v) {
    return i - v.swap();
}

template <typename S, typename T, typename F>
S operator/(const S &i, const swap_struct_t<T, F> v) {
    return i / v.swap();
}

template <typename S, typename T, typename F>
S operator*(const S &i, const swap_struct_t<T, F> v) {
    return i * v.swap();
}

template <typename S, typename T, typename F>
S operator%(const S &i, const swap_struct_t<T, F> v) {
    return i % v.swap();
}

// Arithmetics + assignements
template <typename S, typename T, typename F>
S &operator+=(S &i, const swap_struct_t<T, F> v) {
    i += v.swap();
    return i;
}

template <typename S, typename T, typename F>
S &operator-=(S &i, const swap_struct_t<T, F> v) {
    i -= v.swap();
    return i;
}

// Logical
template <typename S, typename T, typename F>
S operator&(const S &i, const swap_struct_t<T, F> v) {
    return i & v.swap();
}

template <typename S, typename T, typename F>
S operator&(const swap_struct_t<T, F> v, const S &i) {
    return (S)(v.swap() & i);
}


// Comparaison
template <typename S, typename T, typename F>
bool operator<(const S &p, const swap_struct_t<T, F> v) {
    return p < v.swap();
}
template <typename S, typename T, typename F>
bool operator>(const S &p, const swap_struct_t<T, F> v) {
    return p > v.swap();
}
template <typename S, typename T, typename F>
bool operator<=(const S &p, const swap_struct_t<T, F> v) {
    return p <= v.swap();
}
template <typename S, typename T, typename F>
bool operator>=(const S &p, const swap_struct_t<T, F> v) {
    return p >= v.swap();
}
template <typename S, typename T, typename F>
bool operator!=(const S &p, const swap_struct_t<T, F> v) {
    return p != v.swap();
}
template <typename S, typename T, typename F>
bool operator==(const S &p, const swap_struct_t<T, F> v) {
    return p == v.swap();
}

template <typename T>
struct swap_64_t {
    static T swap(T x) {
        return (T)Common::swap64(*(u64 *)&x);
    }
};

template <typename T>
struct swap_32_t {
    static T swap(T x) {
        return (T)Common::swap32(*(u32 *)&x);
    }
};

template <typename T>
struct swap_16_t {
    static T swap(T x) {
        return (T)Common::swap16(*(u16 *)&x);
    }
};

template <typename T>
struct swap_float_t {
    static T swap(T x) {
        return (T)Common::swapf(*(float *)&x);
    }
};

template <typename T>
struct swap_double_t {
    static T swap(T x) {
        return (T)Common::swapd(*(double *)&x);
    }
};

#if COMMON_LITTLE_ENDIAN
typedef u32 u32_le;
typedef u16 u16_le;
typedef u64 u64_le;

typedef s32 s32_le;
typedef s16 s16_le;
typedef s64 s64_le;

typedef float float_le;
typedef double double_le;

typedef swap_struct_t<u64, swap_64_t<u64> > u64_be;
typedef swap_struct_t<s64, swap_64_t<s64> > s64_be;

typedef swap_struct_t<u32, swap_32_t<u32> > u32_be;
typedef swap_struct_t<s32, swap_32_t<s32> > s32_be;

typedef swap_struct_t<u16, swap_16_t<u16> > u16_be;
typedef swap_struct_t<s16, swap_16_t<s16> > s16_be;

typedef swap_struct_t<float, swap_float_t<float> > float_be;
typedef swap_struct_t<double, swap_double_t<double> > double_be;
#else

typedef swap_struct_t<u64, swap_64_t<u64> > u64_le;
typedef swap_struct_t<s64, swap_64_t<s64> > s64_le;

typedef swap_struct_t<u32, swap_32_t<u32> > u32_le;
typedef swap_struct_t<s32, swap_32_t<s32> > s32_le;

typedef swap_struct_t<u16, swap_16_t<u16> > u16_le;
typedef swap_struct_t< s16, swap_16_t<s16> > s16_le;

typedef swap_struct_t<float, swap_float_t<float> > float_le;
typedef swap_struct_t<double, swap_double_t<double> > double_le;

typedef u32 u32_be;
typedef u16 u16_be;
typedef u64 u64_be;

typedef s32 s32_be;
typedef s16 s16_be;
typedef s64 s64_be;

typedef float float_be;
typedef double double_be;

#endif
