/*
    exptools.cpp
    Copyright (C) 2007-2009 somebody
    Copyright (C) 2009 yellow wood goblin

    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, either version 3 of the License, or
    (at your option) any later version.

    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 for more details.

    You should have received a copy of the GNU General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "exptools.h"

#define _PSRAM 0x08060000  // an offset into PSRAM to write to so stuff doesn't get lost...

void cExpansion::OpenNorWrite(void) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9c40000 = 0x1500;
    *(vu16*)0x9fc0000 = 0x1500;
}

void cExpansion::CloseNorWrite(void) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9c40000 = 0xd200;
    *(vu16*)0x9fc0000 = 0x1500;
}

void cExpansion::SetRompage(u16 page) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9880000 = page;
    *(vu16*)0x9fc0000 = 0x1500;
}

void cExpansion::SetRampage(u16 page) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9c00000 = page;
    *(vu16*)0x9fc0000 = 0x1500;
    iRamPage = page;
}

u16 cExpansion::Rampage(void) {
    return iRamPage;
}

void cExpansion::SetSerialMode(void) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9a40000 = 0xe200;
    *(vu16*)0x9fc0000 = 0x1500;
}

void cExpansion::SetShake(u16 data) {
    *(vu16*)0x9fe0000 = 0xd200;
    *(vu16*)0x8000000 = 0x1500;
    *(vu16*)0x8020000 = 0xd200;
    *(vu16*)0x8040000 = 0x1500;
    *(vu16*)0x9e20000 = data;
    *(vu16*)0x9fc0000 = 0x1500;
}

void cExpansion::EnableBrowser(void) {
    for (u32 i = 0; i < 0x100; i += 4) {
        *(vu32*)(0x9000000 + i) = 0xffffffff;
        *(vu32*)(0x8000000 + i) = 0xffffffff;
    }
    *(vu32*)0x90000b0 = 0xffff;
    *(vu32*)0x90000b4 = 0x24242400;
    *(vu32*)0x90000b8 = 0xffffffff;
    *(vu32*)0x90000bc = 0x7fffffff;
    *(vu32*)0x901fffc = 0x7fffffff;
    *(vu16*)0x9240002 = 1;
}

void cExpansion::Block_Erase(u32 blockAdd) {
    vu16 v1, v2;
    u32 Address;
    u32 loop;
    u32 off = 0;
    if ((blockAdd >= 0x1000000) && (iId == 0x227E2202)) {
        off = 0x1000000;
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xF0;
        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xF0;
    } else
        off = 0;
    Address = blockAdd;
    *((vu16*)(FlashBase + 0x555 * 2)) = 0xF0;
    *((vu16*)(FlashBase + 0x1555 * 2)) = 0xF0;
    if ((blockAdd == 0) || (blockAdd == 0x1FC0000) || (blockAdd == 0xFC0000) ||
        (blockAdd == 0x1000000)) {
        for (loop = 0; loop < 0x40000; loop += 0x8000) {
            *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x555 * 2)) = 0x80;
            *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
            *((vu16*)(FlashBase + Address + loop)) = 0x30;

            *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0x80;
            *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
            *((vu16*)(FlashBase + Address + loop + 0x2000)) = 0x30;

            *((vu16*)(FlashBase + off + 0x2555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x22AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x2555 * 2)) = 0x80;
            *((vu16*)(FlashBase + off + 0x2555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x22AA * 2)) = 0x55;
            *((vu16*)(FlashBase + Address + loop + 0x4000)) = 0x30;

            *((vu16*)(FlashBase + off + 0x3555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x32AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x3555 * 2)) = 0x80;
            *((vu16*)(FlashBase + off + 0x3555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x32AA * 2)) = 0x55;
            *((vu16*)(FlashBase + Address + loop + 0x6000)) = 0x30;
            do {
                v1 = *((vu16*)(FlashBase + Address + loop));
                v2 = *((vu16*)(FlashBase + Address + loop));
            } while (v1 != v2);
            do {
                v1 = *((vu16*)(FlashBase + Address + loop + 0x2000));
                v2 = *((vu16*)(FlashBase + Address + loop + 0x2000));
            } while (v1 != v2);
            do {
                v1 = *((vu16*)(FlashBase + Address + loop + 0x4000));
                v2 = *((vu16*)(FlashBase + Address + loop + 0x4000));
            } while (v1 != v2);
            do {
                v1 = *((vu16*)(FlashBase + Address + loop + 0x6000));
                v2 = *((vu16*)(FlashBase + Address + loop + 0x6000));
            } while (v1 != v2);
        }
    } else {
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0x80;
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
        *((vu16*)(FlashBase + Address)) = 0x30;

        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0x80;
        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
        *((vu16*)(FlashBase + Address + 0x2000)) = 0x30;

        do {
            v1 = *((vu16*)(FlashBase + Address));
            v2 = *((vu16*)(FlashBase + Address));
        } while (v1 != v2);
        do {
            v1 = *((vu16*)(FlashBase + Address + 0x2000));
            v2 = *((vu16*)(FlashBase + Address + 0x2000));
        } while (v1 != v2);

        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0x80;
        *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
        *((vu16*)(FlashBase + Address + 0x20000)) = 0x30;

        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0x80;
        *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
        *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
        *((vu16*)(FlashBase + Address + 0x2000 + 0x20000)) = 0x30;

        do {
            v1 = *((vu16*)(FlashBase + Address + 0x20000));
            v2 = *((vu16*)(FlashBase + Address + 0x20000));
        } while (v1 != v2);
        do {
            v1 = *((vu16*)(FlashBase + Address + 0x2000 + 0x20000));
            v2 = *((vu16*)(FlashBase + Address + 0x2000 + 0x20000));
        } while (v1 != v2);
    }
}

void cExpansion::WriteNorFlash(u32 address, const u8* buffer, u32 size) {
    vu16 v1, v2;
    register u32 loopwrite;
    vu16* buf = (vu16*)buffer;
    u32 size2, lop;
    u32 mapaddress;
    u32 j;
    v1 = 0;
    v2 = 1;
    u32 off = 0;
    if ((address >= 0x1000000) && (iId == 0x227E2202)) {
        off = 0x1000000;
    } else
        off = 0;
    if (size > 0x4000) {
        size2 = size >> 1;
        lop = 2;
    } else {
        size2 = size;
        lop = 1;
    }
    mapaddress = address;
    for (j = 0; j < lop; j++) {
        if (j != 0) {
            mapaddress += 0x4000;
            buf = (vu16*)(buffer + 0x4000);
        }
        for (loopwrite = 0; loopwrite < (size2 >> 2); loopwrite++) {
            *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x2AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x555 * 2)) = 0xA0;
            *((vu16*)(FlashBase + mapaddress + loopwrite * 2)) = buf[loopwrite];

            *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xAA;
            *((vu16*)(FlashBase + off + 0x12AA * 2)) = 0x55;
            *((vu16*)(FlashBase + off + 0x1555 * 2)) = 0xA0;
            *((vu16*)(FlashBase + mapaddress + 0x2000 + loopwrite * 2)) = buf[0x1000 + loopwrite];
            do {
                v1 = *((vu16*)(FlashBase + mapaddress + loopwrite * 2));
                v2 = *((vu16*)(FlashBase + mapaddress + loopwrite * 2));
            } while (v1 != v2);
            do {
                v1 = *((vu16*)(FlashBase + mapaddress + 0x2000 + loopwrite * 2));
                v2 = *((vu16*)(FlashBase + mapaddress + 0x2000 + loopwrite * 2));
            } while (v1 != v2);
        }
    }
}

void cExpansion::WritePSRAM(u32 address, const u8* buffer, u32 size) {
    u16* addr = (u16*)(address + _PSRAM);
    u16* pData = (u16*)buffer;
    for (u32 i = 0; i < size; i += 2) {
        addr[i >> 1] = pData[i >> 1];
    }
}

void cExpansion::WriteSram(uint32 address, const u8* data, uint32 size) {
    for (u32 i = 0; i < size; i++) *(u8*)(address + i) = data[i];
}

void cExpansion::ReadSram(uint32 address, u8* data, uint32 size) {
    u16* pData = (u16*)data;
    for (u32 i = 0; i < size; i += 2) {
        pData[i >> 1] = *(u8*)(address + i) + (*(u8*)(address + i + 1) * 0x100);
    }
}

void cExpansion::SoftReset(void) {
    CloseNorWrite();
    SetRompage(0);
    SetRampage(16);
    SetShake(8);
}

void cExpansion::ReadNorFlashID(void) {
    vu16 id1, id2;
    *((vu16*)(FlashBase + 0x555 * 2)) = 0xAA;
    *((vu16*)(FlashBase + 0x2AA * 2)) = 0x55;
    *((vu16*)(FlashBase + 0x555 * 2)) = 0x90;

    *((vu16*)(FlashBase + 0x1555 * 2)) = 0xAA;
    *((vu16*)(FlashBase + 0x12AA * 2)) = 0x55;
    *((vu16*)(FlashBase + 0x1555 * 2)) = 0x90;

    id1 = *((vu16*)(FlashBase + 0x2));
    id2 = *((vu16*)(FlashBase + 0x2002));
    if ((id1 != 0x227E) || (id2 != 0x227E)) return;

    id1 = *((vu16*)(FlashBase + 0xE * 2));
    id2 = *((vu16*)(FlashBase + 0x100e * 2));
    if (id1 == 0x2218 && id2 == 0x2218)  // H6H6
    {
        iId = 0x227E2218;
        return;
    }
    if (id1 == 0x2202 && id2 == 0x2202)  // VZ064
    {
        iId = 0x227E2202;
        return;
    }
    if (id1 == 0x2202 && id2 == 0x2220)  // VZ064
    {
        iId = 0x227E2202;
        return;
    }
    if (id1 == 0x2202 && id2 == 0x2215)  // VZ064
    {
        iId = 0x227E2202;
        return;
    }
}

void cExpansion::ChipReset(void) {
    *((vu16*)(FlashBase)) = 0xF0;
    *((vu16*)(FlashBase + 0x1000 * 2)) = 0xF0;
    if (iId == 0x227E2202) {
        *((vu16*)(FlashBase + 0x1000000)) = 0xF0;
        *((vu16*)(FlashBase + 0x1000000 + 0x1000 * 2)) = 0xF0;
    }
}