/*
 * 86Box    A hypervisor and IBM PC system emulator that specializes in
 *          running old operating systems and software designed for IBM
 *          PC systems and compatibles from 1981 through fairly recent
 *          system designs based on the PCI bus.
 *
 *          This file is part of the 86Box distribution.
 *
 *          3DFX Voodoo emulation.
 *
 *
 *
 * Authors: Sarah Walker, <https://pcem-emulator.co.uk/>
 *
 *          Copyright 2008-2020 Sarah Walker.
 */
#include <stdarg.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <stdlib.h>
#include <stddef.h>
#include <wchar.h>
#include <math.h>
#define HAVE_STDARG_H
#include <86box/86box.h>
#include "cpu.h"
#include <86box/machine.h>
#include <86box/device.h>
#include <86box/mem.h>
#include <86box/timer.h>
#include <86box/device.h>
#include <86box/plat.h>
#include <86box/thread.h>
#include <86box/video.h>
#include <86box/vid_svga.h>
#include <86box/vid_voodoo_common.h>
#include <86box/vid_voodoo_dither.h>
#include <86box/vid_voodoo_regs.h>
#include <86box/vid_voodoo_render.h>
#include <86box/vid_voodoo_fb.h>

#ifdef ENABLE_VOODOO_FB_LOG
int voodoo_fb_do_log = ENABLE_VOODOO_FB_LOG;

static void
voodoo_fb_log(const char *fmt, ...)
{
    va_list ap;

    if (voodoo_fb_do_log) {
        va_start(ap, fmt);
        pclog_ex(fmt, ap);
        va_end(ap);
    }
}
#else
#    define voodoo_fb_log(fmt, ...)
#endif

uint16_t
voodoo_fb_readw(uint32_t addr, void *priv)
{
    voodoo_t *voodoo = (voodoo_t *) priv;
    int       x;
    int       y;
    uint32_t  read_addr;
    uint16_t  temp;

    if (voodoo->type >= VOODOO_BANSHEE) {
        x = addr & 0xffe;
        y = (addr >> 12) & 0x3ff;
    } else {
        x = addr & 0x7fe;
        y = (addr >> 11) & 0x3ff;
    }

    if (SLI_ENABLED) {
        const voodoo_set_t *set = voodoo->set;

        if (y & 1)
            voodoo = set->voodoos[1];
        else
            voodoo = set->voodoos[0];

        y >>= 1;
    }

    if (voodoo->col_tiled)
        read_addr = voodoo->fb_read_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        read_addr = voodoo->fb_read_offset + x + (y * voodoo->row_width);

    if (read_addr > voodoo->fb_mask)
        return 0xffff;

    temp = *(uint16_t *) (&voodoo->fb_mem[read_addr & voodoo->fb_mask]);

    //        voodoo_fb_log("voodoo_fb_readw : %08X %08X  %i %i  %08X %08X  %08x:%08x %i\n", addr, temp, x, y, read_addr, *(uint32_t *)(&voodoo->fb_mem[4]), cs, pc, fb_reads++);
    return temp;
}
uint32_t
voodoo_fb_readl(uint32_t addr, void *priv)
{
    voodoo_t *voodoo = (voodoo_t *) priv;
    int       x;
    int       y;
    uint32_t  read_addr;
    uint32_t  temp;

    if (voodoo->type >= VOODOO_BANSHEE) {
        x = addr & 0xffe;
        y = (addr >> 12) & 0x3ff;
    } else {
        x = addr & 0x7fe;
        y = (addr >> 11) & 0x3ff;
    }

    if (SLI_ENABLED) {
        const voodoo_set_t *set = voodoo->set;

        if (y & 1)
            voodoo = set->voodoos[1];
        else
            voodoo = set->voodoos[0];

        y >>= 1;
    }

    if (voodoo->col_tiled)
        read_addr = voodoo->fb_read_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        read_addr = voodoo->fb_read_offset + x + (y * voodoo->row_width);

    if (read_addr > voodoo->fb_mask)
        return 0xffffffff;

    temp = *(uint32_t *) (&voodoo->fb_mem[read_addr & voodoo->fb_mask]);

    //        voodoo_fb_log("voodoo_fb_readl : %08X %08x %08X  x=%i y=%i  %08X %08X  %08x:%08x %i ro=%08x rw=%i\n", addr, read_addr, temp, x, y, read_addr, *(uint32_t *)(&voodoo->fb_mem[4]), cs, pc, fb_reads++, voodoo->fb_read_offset, voodoo->row_width);
    return temp;
}

static inline uint16_t
do_dither(voodoo_params_t *params, rgba8_t col, int x, int y)
{
    int r;
    int g;
    int b;

    if (dither) {
        if (dither2x2) {
            r = dither_rb2x2[col.r][y & 1][x & 1];
            g = dither_g2x2[col.g][y & 1][x & 1];
            b = dither_rb2x2[col.b][y & 1][x & 1];
        } else {
            r = dither_rb[col.r][y & 3][x & 3];
            g = dither_g[col.g][y & 3][x & 3];
            b = dither_rb[col.b][y & 3][x & 3];
        }
    } else {
        r = col.r >> 3;
        g = col.g >> 2;
        b = col.b >> 3;
    }

    return b | (g << 5) | (r << 11);
}

void
voodoo_fb_writew(uint32_t addr, uint16_t val, void *priv)
{
    voodoo_t              *voodoo = (voodoo_t *) priv;
    const voodoo_params_t *params = &voodoo->params;
    int                    x;
    int                    y;
    uint32_t               write_addr;
    uint32_t               write_addr_aux;
    rgba8_t                colour_data;
    uint16_t               depth_data;
    uint8_t                alpha_data;
    int                    write_mask = 0;

    colour_data.r = colour_data.g = colour_data.b = colour_data.a = 0;

    depth_data = voodoo->params.zaColor & 0xffff;
    alpha_data = voodoo->params.zaColor >> 24;

#if 0
    while (!RB_EMPTY)
        thread_reset_event(voodoo->not_full_event);
#endif

#if 0
    voodoo_fb_log("voodoo_fb_writew : %08X %04X\n", addr, val);
#endif

    switch (voodoo->lfbMode & LFB_FORMAT_MASK) {
        case LFB_FORMAT_RGB565:
            colour_data = rgb565[val];
            alpha_data  = 0xff;
            write_mask  = LFB_WRITE_COLOUR;
            break;
        case LFB_FORMAT_RGB555:
            colour_data = argb1555[val];
            alpha_data  = 0xff;
            write_mask  = LFB_WRITE_COLOUR;
            break;
        case LFB_FORMAT_ARGB1555:
            colour_data = argb1555[val];
            alpha_data  = colour_data.a;
            write_mask  = LFB_WRITE_COLOUR;
            break;
        case LFB_FORMAT_DEPTH:
            depth_data = val;
            write_mask = LFB_WRITE_DEPTH;
            break;

        default:
            fatal("voodoo_fb_writew : bad LFB format %08X\n", voodoo->lfbMode);
    }

    if (voodoo->type >= VOODOO_BANSHEE) {
        x = addr & 0xffe;
        y = (addr >> 12) & 0x3ff;
    } else {
        x = addr & 0x7fe;
        y = (addr >> 11) & 0x3ff;
    }

    if (SLI_ENABLED) {
        if ((!(voodoo->initEnable & INITENABLE_SLI_MASTER_SLAVE) && (y & 1)) || ((voodoo->initEnable & INITENABLE_SLI_MASTER_SLAVE) && !(y & 1)))
            return;
        y >>= 1;
    }

    if (voodoo->fb_write_offset == voodoo->params.front_offset && y < 2048)
        voodoo->dirty_line[y] = 1;

    if (voodoo->col_tiled)
        write_addr = voodoo->fb_write_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        write_addr = voodoo->fb_write_offset + x + (y * voodoo->row_width);
    if (voodoo->aux_tiled)
        write_addr_aux = voodoo->params.aux_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        write_addr_aux = voodoo->params.aux_offset + x + (y * voodoo->row_width);

    //        voodoo_fb_log("fb_writew %08x %i %i %i %08x\n", addr, x, y, voodoo->row_width, write_addr);

    if (voodoo->lfbMode & 0x100) {
        {
            rgba8_t  write_data = colour_data;
            uint16_t new_depth  = depth_data;
            int      colbfog_r  = 0;
            int      colbfog_g  = 0;
            int      colbfog_b  = 0;

            if (params->fbzMode & FBZ_DEPTH_ENABLE) {
                uint16_t old_depth = *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]);

                DEPTH_TEST(new_depth);
            }

            if ((params->fbzMode & FBZ_CHROMAKEY) && write_data.r == params->chromaKey_r && write_data.g == params->chromaKey_g && write_data.b == params->chromaKey_b)
                goto skip_pixel;

            colbfog_r = write_data.r;
            colbfog_g = write_data.g;
            colbfog_b = write_data.b;

            if (params->fogMode & FOG_ENABLE) {
                int32_t z       = new_depth << 12;
                int64_t w_depth = (int64_t) (int32_t) new_depth;
                int32_t ia      = alpha_data << 12;

                APPLY_FOG(write_data.r, write_data.g, write_data.b, z, ia, w_depth);
            }

            if (params->alphaMode & 1)
                ALPHA_TEST(alpha_data);

            if (params->alphaMode & (1 << 4)) {
                uint16_t dat = *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]);
                int      dest_r;
                int      dest_g;
                int      dest_b;
                int      dest_a;

                dest_r = (dat >> 8) & 0xf8;
                dest_g = (dat >> 3) & 0xfc;
                dest_b = (dat << 3) & 0xf8;
                dest_r |= (dest_r >> 5);
                dest_g |= (dest_g >> 6);
                dest_b |= (dest_b >> 5);
                dest_a = 0xff;

                ALPHA_BLEND(write_data.r, write_data.g, write_data.b, alpha_data);
            }

            if (params->fbzMode & FBZ_RGB_WMASK)
                *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]) = do_dither(&voodoo->params, write_data, x >> 1, y);
            if (params->fbzMode & FBZ_DEPTH_WMASK)
                *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]) = new_depth;

skip_pixel:
            return;
        }
    } else {
        if (write_mask & LFB_WRITE_COLOUR)
            *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]) = do_dither(&voodoo->params, colour_data, x >> 1, y);
        if (write_mask & LFB_WRITE_DEPTH)
            *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]) = depth_data;
    }
}

void
voodoo_fb_writel(uint32_t addr, uint32_t val, void *priv)
{
    voodoo_t              *voodoo = (voodoo_t *) priv;
    const voodoo_params_t *params = &voodoo->params;
    int                    x;
    int                    y;
    uint32_t               write_addr;
    uint32_t               write_addr_aux;
    rgba8_t                colour_data[2];
    uint16_t               depth_data[2];
    uint8_t                alpha_data[2];
    int                    write_mask = 0;
    int                    count = 1;

    depth_data[0] = depth_data[1] = voodoo->params.zaColor & 0xffff;
    alpha_data[0] = alpha_data[1] = voodoo->params.zaColor >> 24;
#if 0
    while (!RB_EMPTY)
        thread_reset_event(voodoo->not_full_event);
#endif

#if 0
    voodoo_fb_log("voodoo_fb_writel : %08X %08X\n", addr, val);
#endif

    switch (voodoo->lfbMode & LFB_FORMAT_MASK) {
        case LFB_FORMAT_RGB565:
            colour_data[0] = rgb565[val & 0xffff];
            colour_data[1] = rgb565[val >> 16];
            write_mask     = LFB_WRITE_COLOUR;
            count          = 2;
            break;
        case LFB_FORMAT_RGB555:
            colour_data[0] = argb1555[val & 0xffff];
            colour_data[1] = argb1555[val >> 16];
            write_mask     = LFB_WRITE_COLOUR;
            count          = 2;
            break;
        case LFB_FORMAT_ARGB1555:
            colour_data[0] = argb1555[val & 0xffff];
            alpha_data[0]  = colour_data[0].a;
            colour_data[1] = argb1555[val >> 16];
            alpha_data[1]  = colour_data[1].a;
            write_mask     = LFB_WRITE_COLOUR;
            count          = 2;
            break;

        case LFB_FORMAT_ARGB8888:
            colour_data[0].b = val & 0xff;
            colour_data[0].g = (val >> 8) & 0xff;
            colour_data[0].r = (val >> 16) & 0xff;
            alpha_data[0]    = (val >> 24) & 0xff;
            write_mask       = LFB_WRITE_COLOUR;
            addr >>= 1;
            break;

        case LFB_FORMAT_XRGB8888:
            colour_data[0].b = val & 0xff;
            colour_data[0].g = (val >> 8) & 0xff;
            colour_data[0].r = (val >> 16) & 0xff;
            alpha_data[0]    = 0xff;
            write_mask       = LFB_WRITE_COLOUR;
            addr >>= 1;
            break;

        case LFB_FORMAT_DEPTH_RGB565:
            colour_data[0] = rgb565[val & 0xffff];
            depth_data[0]  = val >> 16;
            write_mask     = LFB_WRITE_BOTH;
            count          = 1;
            break;
        case LFB_FORMAT_DEPTH_RGB555:
            colour_data[0] = argb1555[val & 0xffff];
            depth_data[0]  = val >> 16;
            write_mask     = LFB_WRITE_BOTH;
            count          = 1;
            break;
        case LFB_FORMAT_DEPTH_ARGB1555:
            colour_data[0] = argb1555[val & 0xffff];
            alpha_data[0]  = colour_data[0].a;
            depth_data[0]  = val >> 16;
            write_mask     = LFB_WRITE_BOTH;
            count          = 1;
            break;

        case LFB_FORMAT_DEPTH:
            depth_data[0] = val;
            depth_data[1] = val >> 16;
            write_mask    = LFB_WRITE_DEPTH;
            count         = 2;
            break;

        default:
            fatal("voodoo_fb_writel : bad LFB format %08X\n", voodoo->lfbMode);
    }

    if (voodoo->type >= VOODOO_BANSHEE) {
        x = addr & 0xffe;
        y = (addr >> 12) & 0x3ff;
    } else {
        x = addr & 0x7fe;
        y = (addr >> 11) & 0x3ff;
    }

    if (SLI_ENABLED) {
        if ((!(voodoo->initEnable & INITENABLE_SLI_MASTER_SLAVE) && (y & 1)) || ((voodoo->initEnable & INITENABLE_SLI_MASTER_SLAVE) && !(y & 1)))
            return;
        y >>= 1;
    }

    if (voodoo->fb_write_offset == voodoo->params.front_offset && y < 2048)
        voodoo->dirty_line[y] = 1;

    if (voodoo->col_tiled)
        write_addr = voodoo->fb_write_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        write_addr = voodoo->fb_write_offset + x + (y * voodoo->row_width);
    if (voodoo->aux_tiled)
        write_addr_aux = voodoo->params.aux_offset + (x & 127) + (x >> 7) * 128 * 32 + (y & 31) * 128 + (y >> 5) * voodoo->row_width;
    else
        write_addr_aux = voodoo->params.aux_offset + x + (y * voodoo->row_width);

#if 0
    voodoo_fb_log("fb_writel %08x x=%i y=%i rw=%i %08x wo=%08x\n", addr, x, y, voodoo->row_width, write_addr, voodoo->fb_write_offset);
#endif

    if (voodoo->lfbMode & 0x100) {
        for (int c = 0; c < count; c++) {
            rgba8_t  write_data = colour_data[c];
            uint16_t new_depth  = depth_data[c];
            int      colbfog_r  = 0;
            int      colbfog_g  = 0;
            int      colbfog_b  = 0;

            if (params->fbzMode & FBZ_DEPTH_ENABLE) {
                uint16_t old_depth = *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]);

                DEPTH_TEST(new_depth);
            }

            if ((params->fbzMode & FBZ_CHROMAKEY) && write_data.r == params->chromaKey_r && write_data.g == params->chromaKey_g && write_data.b == params->chromaKey_b)
                goto skip_pixel;

            colbfog_r = write_data.r;
            colbfog_g = write_data.g;
            colbfog_b = write_data.b;

            if (params->fogMode & FOG_ENABLE) {
                int32_t z       = new_depth << 12;
                int64_t w_depth = new_depth;
                int32_t ia      = alpha_data[c] << 12;

                APPLY_FOG(write_data.r, write_data.g, write_data.b, z, ia, w_depth);
            }

            if (params->alphaMode & 1)
                ALPHA_TEST(alpha_data[c]);

            if (params->alphaMode & (1 << 4)) {
                uint16_t dat = *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]);
                int      dest_r;
                int      dest_g;
                int      dest_b;
                int      dest_a;

                dest_r = (dat >> 8) & 0xf8;
                dest_g = (dat >> 3) & 0xfc;
                dest_b = (dat << 3) & 0xf8;
                dest_r |= (dest_r >> 5);
                dest_g |= (dest_g >> 6);
                dest_b |= (dest_b >> 5);
                dest_a = 0xff;

                ALPHA_BLEND(write_data.r, write_data.g, write_data.b, alpha_data[c]);
            }

            if (params->fbzMode & FBZ_RGB_WMASK)
                *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]) = do_dither(&voodoo->params, write_data, (x >> 1) + c, y);
            if (params->fbzMode & FBZ_DEPTH_WMASK)
                *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]) = new_depth;

skip_pixel:
            write_addr += 2;
            write_addr_aux += 2;
        }
    } else {
        for (int c = 0; c < count; c++) {
            if (write_mask & LFB_WRITE_COLOUR)
                *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]) =
                              do_dither(&voodoo->params, colour_data[c], (x >> 1) + c, y);
            if (write_mask & LFB_WRITE_DEPTH)
                *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]) = depth_data[c];
            if (write_mask & LFB_WRITE_BOTH) {
                *(uint16_t *) (&voodoo->fb_mem[write_addr & voodoo->fb_mask]) =
                              do_dither(&voodoo->params, colour_data[c], (x >> 1) + c, y);
                *(uint16_t *) (&voodoo->fb_mem[write_addr_aux & voodoo->fb_mask]) = depth_data[c];
            }

            write_addr += 2;
            write_addr_aux += 2;
        }
    }
}