#pragma once
#include <assert.h>

#define S 1
#define S16 2
#define S8 3

enum
{
    TASK_TYPE = 0xfc0,
    TASK_FLAGS = 0xfc4,
    TASK_UCODE_BOOT = 0xfc8,
    TASK_UCODE_BOOT_SIZE = 0xfcc,
    TASK_UCODE = 0xfd0,
    TASK_UCODE_SIZE = 0xfd4,
    TASK_UCODE_DATA = 0xfd8,
    TASK_UCODE_DATA_SIZE = 0xfdc,
    TASK_DRAM_STACK = 0xfe0,
    TASK_DRAM_STACK_SIZE = 0xfe4,
    TASK_OUTPUT_BUFF = 0xfe8,
    TASK_OUTPUT_BUFF_SIZE = 0xfec,
    TASK_DATA_PTR = 0xff0,
    TASK_DATA_SIZE = 0xff4,
    TASK_YIELD_DATA_PTR = 0xff8,
    TASK_YIELD_DATA_SIZE = 0xffc
};

static inline unsigned int align(unsigned int x, unsigned amount)
{
    --amount;
    return (x + amount) & ~amount;
}

static inline uint8_t* u8(const unsigned char* buffer, unsigned address)
{
    return (uint8_t*)(buffer + (address ^ S8));
}

static inline uint16_t* u16(const unsigned char* buffer, unsigned address)
{
    assert((address & 1) == 0);
    return (uint16_t*)(buffer + (address ^ S16));
}

static inline uint32_t* u32(const unsigned char* buffer, unsigned address)
{
    assert((address & 3) == 0);
    return (uint32_t*)(buffer + address);
}

void load_u8 (uint8_t*  dst, const unsigned char* buffer, unsigned address, size_t count);
void load_u16(uint16_t* dst, const unsigned char* buffer, unsigned address, size_t count);
void load_u32(uint32_t* dst, const unsigned char* buffer, unsigned address, size_t count);
void store_u16(unsigned char* buffer, unsigned address, const uint16_t* src, size_t count);
void store_u32(unsigned char* buffer, unsigned address, const uint32_t* src, size_t count);

static inline uint32_t* dmem_u32(CHle * hle, uint16_t address)
{
    return u32(hle->dmem(), address & 0xfff);
}

static inline void dmem_store_u32(CHle * hle, const uint32_t* src, uint16_t address, size_t count)
{
    store_u32(hle->dmem(), address & 0xfff, src, count);
}

// Convenient functions DRAM access
static inline uint8_t* dram_u8(CHle * hle, uint32_t address)
{
    return u8(hle->dram(), address & 0xffffff);
}

static inline uint16_t* dram_u16(CHle * hle, uint32_t address)
{
    return u16(hle->dram(), address & 0xffffff);
}

static inline uint32_t* dram_u32(CHle * hle, uint32_t address)
{
    return u32(hle->dram(), address & 0xffffff);
}

static inline void dram_load_u8(CHle * hle, uint8_t* dst, uint32_t address, size_t count)
{
    load_u8(dst, hle->dram(), address & 0xffffff, count);
}

static inline void dram_load_u16(CHle * hle, uint16_t* dst, uint32_t address, size_t count)
{
    load_u16(dst, hle->dram(), address & 0xffffff, count);
}

static inline void dram_load_u32(CHle * hle, uint32_t* dst, uint32_t address, size_t count)
{
    load_u32(dst, hle->dram(), address & 0xffffff, count);
}

static inline void dram_store_u16(CHle * hle, const uint16_t* src, uint32_t address, size_t count)
{
    store_u16(hle->dram(), address & 0xffffff, src, count);
}

static inline void dram_store_u32(CHle * hle, const uint32_t* src, uint32_t address, size_t count)
{
    store_u32(hle->dram(), address & 0xffffff, src, count);
}