#ifndef __JABYENGINE_IOPORT_HPP__
#define __JABYENGINE_IOPORT_HPP__
#include "../../Auxiliary/bits.hpp"

struct ClearBitValue {
    size_t bit;

    constexpr ClearBitValue(size_t bit) : bit(bit) {
    }
};

template<typename T>
struct Bit {
    typedef T ValueType;

    size_t value;

    constexpr Bit(size_t value) : value(value) {
    }

    constexpr operator size_t() const {
        return this->value;
    }

    constexpr ClearBitValue operator!() const {
        return ClearBitValue(this->value);
    }
};

template<typename T>
struct BitRange {
    typedef T ValueType;

    size_t begin;
    size_t length;

    static constexpr BitRange<T> from_to(size_t start, size_t end) {
        return {start, (end - start + 1)};
    }
};

template<typename T>
struct BitRangeValue {
    T           value;
    BitRange<T> range;

    constexpr BitRangeValue(BitRange<T> range, T value) : value(value), range(range) {
    }
};

template<typename T>
static constexpr __always_inline BitRangeValue<T> operator<<(const BitRange<T>& range, T value) {
    return BitRangeValue(range, value);
}

template<typename T>
class __no_align IOPort {
private:
    T value = 0;

public:
    constexpr IOPort() = default;
    constexpr IOPort(T value) : value(value) {
    }

    //Accesssing bits
    template<typename S>
    constexpr IOPort<T>& set_bit(S bit) {
        this->value = bit::set(this->value, static_cast<size_t>(bit));
        return *this;
    }

    template<typename S>
    constexpr void set_bit(S bit) volatile {
        this->value = bit::set(this->value, static_cast<size_t>(bit));
    }

    template<typename S>
    constexpr IOPort<T>& clear_bit(S bit) {
        this->value = bit::clear(this->value, static_cast<size_t>(bit));
        return *this;
    }

    template<typename S>
    constexpr void clear_bit(S bit) volatile {
        this->value = bit::clear(this->value, static_cast<size_t>(bit));
    }

    template<typename S>
    constexpr bool is_bit_set(S bit) {
        return bit::is_set(this->value, static_cast<size_t>(bit));
    }

    template<typename S>
    constexpr bool is_bit_set(S bit) volatile {
        return bit::is_set(this->value, static_cast<size_t>(bit));
    }

    //Accessing values
    template<typename S>
    constexpr IOPort<T>& set_value(S value, const BitRange<S>& range) {
        this->value = bit::value::set_normalized(this->value, static_cast<T>(value), range.begin, range.length);
        return *this;
    }

    template<typename S>
    constexpr void set_value(S value, const BitRange<S>& range) volatile {
        this->value = bit::value::set_normalized(this->value, static_cast<T>(value), range.begin, range.length);
    }

    template<typename S>
    constexpr IOPort<T>& clear_value(const BitRange<S>& range) {
        this->value = bit::value::clear_normalized(this->value, range.begin, range.length);
        return *this;
    }

    template<typename S>
    constexpr void clear_value(const BitRange<S>& range) volatile {
        this->value = bit::value::clear_normalized(this->value, range.begin, range.length);
    }

    template<typename S>
    constexpr S get_value(const BitRange<S>& range) {
        return static_cast<S>(bit::value::get_normalized(this->value, range.begin, range.length));
    }

    template<typename S>
    constexpr S get_value(const BitRange<S>& range) volatile {
        return static_cast<S>(bit::value::get_normalized(this->value, range.begin, range.length));
    }

    //For easy access
    constexpr T read() const {
        return const_cast<volatile IOPort<T>*>(this)->value;
    }

    constexpr void write(T value) {
        const_cast<volatile IOPort<T>*>(this)->value = value;
    }

    //For raw access
    constexpr operator T() const {
        return this->value;
    }

    constexpr operator T() const volatile {
        return this->value;
    }

    constexpr IOPort<T>& operator=(T value) {
        this->value = value;
        return *this;
    }

    constexpr void operator=(T value) volatile {
        this->value = value;
    }

    // For easier constructing
    constexpr IOPort<T>& operator|(const BitRangeValue<T>& value) {
        this->set_value(value.value, value.range);
        return *this;
    }

    constexpr IOPort<T>& operator|(const Bit<T>& bit) {
        this->set_bit(bit.value);
        return *this;
    }

    constexpr IOPort<T>& operator|(const ClearBitValue& value) {
        this->clear_bit(value.bit);
        return *this;
    }
};

struct __no_align ubus32_t {
    IOPort<uint16_t> low;
    IOPort<uint16_t> high; 

    constexpr ubus32_t(uint32_t value) {
        *this = value;
    }

    constexpr operator uint32_t() const {
        return ((this->high << 16) | this->low);
    }

    constexpr operator uint32_t() const volatile {
        return ((this->high << 16) | this->low);
    }

    constexpr ubus32_t& operator=(uint32_t value) {
        this->low  = (value & 0xFFFF);
        this->high = (value >> 16);

        return *this;
    }

    constexpr void operator=(uint32_t value) volatile {
        this->low  = (value & 0xFFFF);
        this->high = (value >> 16);
    }
};
static constexpr uintptr_t IO_Base_Mask = 0xF0000000;
static constexpr uintptr_t IO_Base_Adr  = 0x10000000;

#define __declare_io_port_global(type, name, adr) static __always_inline auto& name = *reinterpret_cast<IOPort<type>*>((IO_Base_Adr + (adr & ~IO_Base_Mask)))
#define __io_port_inherit(name) \
    using IOPort::operator=; \
    constexpr name() = default; \
    constexpr name(IOPort value) : IOPort(value) { \
    }

#endif //!__JABYENGINE_IOPORT_HPP__