#include <iostream>
#include "Section.hpp"

Section::Section(byte *dataBlocks, size_t dataBlocksLength, byte *dataLight, byte *dataSky, byte bitsPerBlock,
                 std::vector<unsigned short> palette) {
    m_dataBlocksLen = dataBlocksLength;
    m_dataBlocks = new byte[m_dataBlocksLen];
    std::copy(dataBlocks, dataBlocks + m_dataBlocksLen, m_dataBlocks);

    m_dataLight = new byte[2048];
    std::copy(dataLight, dataLight + 2048, m_dataLight);

    if (dataSky) {
        m_dataSkyLight = new byte[2048];
        std::copy(dataSky, dataSky + 2048, m_dataSkyLight);
    }

    m_palette = palette;
    m_bitsPerBlock = bitsPerBlock;
}

Section::~Section() {
    delete[] m_dataBlocks;
    m_dataBlocksLen = 0;
    m_dataBlocks = nullptr;
    delete[] m_dataLight;
    m_dataLight = nullptr;
    delete[] m_dataSkyLight;
    m_dataSkyLight = nullptr;
}

Block &Section::GetBlock(Vector pos) {
    if (m_dataBlocks != nullptr) {
        std::mutex parseMutex;
        std::unique_lock<std::mutex> parseLocker(parseMutex);
        parseWaiter.wait(parseLocker);
        while (m_dataBlocks != nullptr) {
            parseWaiter.wait(parseLocker);
        }
    }
    return m_blocks[pos.GetY() * 256 + pos.GetZ() * 16 + pos.GetX()];
}

void Section::Parse() {
    if (m_dataBlocks == nullptr)
        return;

    long long *longArray = reinterpret_cast<long long *>(m_dataBlocks);
    for (int i = 0; i < m_dataBlocksLen / 8; i++)
        endswap(&longArray[i]);
    std::vector<unsigned short> blocks;
    blocks.reserve(4096);
    int bitPos = 0;
    unsigned short t = 0;
    for (int i = 0; i < m_dataBlocksLen; i++) {
        for (int j = 0; j < 8; j++) {
            t |= (m_dataBlocks[i] & 0x01) ? 0x80 : 0x00;
            t >>= 1;
            m_dataBlocks[i] >>= 1;
            bitPos++;
            if (bitPos >= m_bitsPerBlock) {
                bitPos = 0;
                t >>= m_bitsPerBlock - 1;
                blocks.push_back(t);
                t = 0;
            }
        }
    }

    std::vector<byte> light;
    light.reserve(4096);
    for (int i = 0; i < 2048; i++) {
        byte t = m_dataLight[i];
        byte first = t & 0b11110000;
        byte second = t >> 4;
        light.push_back(first);
        light.push_back(second);
    }
    for (int i = 0; i < 4096; i++) {
        unsigned short blockId = m_palette.size() > 0 ? m_palette[blocks[i]] : blocks[i];
        Block block(blockId, light[i]);
        m_blocks.push_back(block);
    }
    if ((light.size() + blocks.size()) / 2 != 4096) {
        throw 118;
    }
    delete[] m_dataBlocks;
    m_dataBlocksLen = 0;
    m_dataBlocks = nullptr;
    delete[] m_dataLight;
    m_dataLight = nullptr;
    delete[] m_dataSkyLight;
    m_dataSkyLight = nullptr;
    parseWaiter.notify_all();
}

Section &Section::operator=(Section other) {
    other.swap(*this);
    return *this;
}

void Section::swap(Section &other) {
    std::swap(other.m_dataBlocksLen, m_dataBlocksLen);
    std::swap(other.m_dataBlocks, m_dataBlocks);
    std::swap(other.m_dataLight, m_dataLight);
    std::swap(other.m_dataSkyLight, m_dataSkyLight);
    std::swap(other.m_blocks, m_blocks);
    std::swap(other.m_palette, m_palette);
    std::swap(other.m_bitsPerBlock, m_bitsPerBlock);
}

Section::Section(const Section &other) {
    m_dataBlocksLen = other.m_dataBlocksLen;
    m_dataBlocks = new byte[m_dataBlocksLen];
    std::copy(other.m_dataBlocks, other.m_dataBlocks + m_dataBlocksLen, m_dataBlocks);

    m_dataLight = new byte[2048];
    std::copy(other.m_dataLight, other.m_dataLight + 2048, m_dataLight);

    if (other.m_dataSkyLight) {
        m_dataSkyLight = new byte[2048];
        std::copy(other.m_dataSkyLight, other.m_dataSkyLight + 2048, m_dataSkyLight);
    }

    m_palette = other.m_palette;
    m_bitsPerBlock = other.m_bitsPerBlock;
}