#include <stdlib.h>
#include "ibm.h"
#include "keyboard_at.h"
#include "mouse.h"
#include "mouse_ps2.h"
#include "plat-mouse.h"

int mouse_scan = 0;

enum
{
        MOUSE_STREAM,
        MOUSE_REMOTE,
        MOUSE_ECHO
};

#define MOUSE_ENABLE 0x20
#define MOUSE_SCALE  0x10

typedef struct mouse_ps2_t
{
        int mode;
        
        uint8_t flags;
        uint8_t resolution;
        uint8_t sample_rate;
        
        uint8_t command;
        
        int cd;
        
        int x, y, z, b;
        
        int is_intellimouse;
        int intellimouse_mode;
        
        uint8_t last_data[6];
} mouse_ps2_t;

void mouse_ps2_write(uint8_t val, void *p)
{
        mouse_ps2_t *mouse = (mouse_ps2_t *)p;
        
        if (mouse->cd)
        {
                mouse->cd = 0;
                switch (mouse->command)
                {
                        case 0xe8: /*Set mouse resolution*/
                        mouse->resolution = val;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                        
                        case 0xf3: /*Set sample rate*/
                        mouse->sample_rate = val;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                        
//                        default:
//                        fatal("mouse_ps2 : Bad data write %02X for command %02X\n", val, mouse->command);
                }
        }
        else
        {
                uint8_t temp;
                
                mouse->command = val;
                switch (mouse->command)
                {
                        case 0xe6: /*Set scaling to 1:1*/
                        mouse->flags &= ~MOUSE_SCALE;
                        keyboard_at_adddata_mouse(0xfa);
                        break;

                        case 0xe7: /*Set scaling to 2:1*/
                        mouse->flags |= MOUSE_SCALE;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                        
                        case 0xe8: /*Set mouse resolution*/
                        mouse->cd = 1;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                        
                        case 0xe9: /*Status request*/
                        keyboard_at_adddata_mouse(0xfa);
                        temp = mouse->flags;
                        if (mouse_buttons & 1)
                                temp |= 1;
                        if (mouse_buttons & 2)
                                temp |= 2;
                        if (mouse_buttons & 4)
                                temp |= 3;
                        keyboard_at_adddata_mouse(temp);
                        keyboard_at_adddata_mouse(mouse->resolution);
                        keyboard_at_adddata_mouse(mouse->sample_rate);
                        break;
                        
                        case 0xf2: /*Read ID*/
                        keyboard_at_adddata_mouse(0xfa);
                        if (mouse->intellimouse_mode)
                                keyboard_at_adddata_mouse(0x03);
                        else
                                keyboard_at_adddata_mouse(0x00);
                        break;
                        
                        case 0xf3: /*Set sample rate*/
                        mouse->cd = 1;
                        keyboard_at_adddata_mouse(0xfa);
                        break;

                        case 0xf4: /*Enable*/
                        mouse->flags |= MOUSE_ENABLE;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                                                
                        case 0xf5: /*Disable*/
                        mouse->flags &= ~MOUSE_ENABLE;
                        keyboard_at_adddata_mouse(0xfa);
                        break;
                        
                        case 0xff: /*Reset*/
                        mouse->mode  = MOUSE_STREAM;
                        mouse->flags = 0;
                        mouse->intellimouse_mode = 0;
                        keyboard_at_adddata_mouse(0xfa);
                        keyboard_at_adddata_mouse(0xaa);
                        keyboard_at_adddata_mouse(0x00);
                        break;
                        
//                        default:
//                        fatal("mouse_ps2 : Bad command %02X\n", val, mouse->command);
                }
        }
        
        if (mouse->is_intellimouse)
        {
                int c;
        
                for (c = 0; c < 5; c++)        
                        mouse->last_data[c] = mouse->last_data[c+1];
                        
                mouse->last_data[5] = val;
        
                if (mouse->last_data[0] == 0xf3 && mouse->last_data[1] == 0xc8 &&
                    mouse->last_data[2] == 0xf3 && mouse->last_data[3] == 0x64 &&
                    mouse->last_data[4] == 0xf3 && mouse->last_data[5] == 0x50)
                        mouse->intellimouse_mode = 1;
        }
}

uint8_t mouse_ps2_poll(int x, int y, int z, int b, void *p)
{
        mouse_ps2_t *mouse = (mouse_ps2_t *)p;
        uint8_t packet[3] = {0x08, 0, 0};
        
        if (!x && !y && !z && b == mouse->b)
                return(0xff);

        if (!mouse_scan)
                return(0xff);
                
        mouse->x += x;
        mouse->y -= y;
        mouse->z -= z;
        if (mouse->mode == MOUSE_STREAM && (mouse->flags & MOUSE_ENABLE) &&
            ((mouse_queue_end - mouse_queue_start) & 0xf) < 13)
        {
                mouse->b = b;
               // pclog("Send packet : %i %i\n", ps2_x, ps2_y);
                if (mouse->x > 255)
                        mouse->x = 255;
                if (mouse->x < -256)
                        mouse->x = -256;
                if (mouse->y > 255)
                        mouse->y = 255;
                if (mouse->y < -256)
                        mouse->y = -256;
                if (mouse->z < -8)
                        mouse->z = -8;
                if (mouse->z > 7)
                        mouse->z = 7;
                if (mouse->x < 0)
                        packet[0] |= 0x10;
                if (mouse->y < 0)
                        packet[0] |= 0x20;
                if (mouse_buttons & 1)
                        packet[0] |= 1;
                if (mouse_buttons & 2)
                        packet[0] |= 2;
                if ((mouse_buttons & 4) && (mouse_get_type(mouse_type) & MOUSE_TYPE_3BUTTON))
                        packet[0] |= 4;
                packet[1] = mouse->x & 0xff;
                packet[2] = mouse->y & 0xff;

                keyboard_at_adddata_mouse(packet[0]);
                keyboard_at_adddata_mouse(packet[1]);
                keyboard_at_adddata_mouse(packet[2]);
                if (mouse->intellimouse_mode)
                        keyboard_at_adddata_mouse(mouse->z);

                mouse->x = mouse->y = mouse->z = 0;                
        }

	return(0);
}

void *mouse_ps2_init()
{
        mouse_ps2_t *mouse = (mouse_ps2_t *)malloc(sizeof(mouse_ps2_t));
        memset(mouse, 0, sizeof(mouse_ps2_t));
        
//        mouse_poll  = mouse_ps2_poll;
//        mouse_write = mouse_ps2_write;
        mouse->cd = 0;
        mouse->flags = 0;
        mouse->mode = MOUSE_STREAM;
        
        keyboard_at_set_mouse(mouse_ps2_write, mouse);
        
        return mouse;
}

void *mouse_intellimouse_init()
{
        mouse_ps2_t *mouse = mouse_ps2_init();

        mouse->is_intellimouse = 1;
                
        return mouse;
}

void mouse_ps2_close(void *p)
{
        mouse_ps2_t *mouse = (mouse_ps2_t *)p;
        
        free(mouse);
}

mouse_t mouse_ps2_2_button =
{
        "2-button mouse (PS/2)",
        "ps2",
        MOUSE_TYPE_PS2,
        mouse_ps2_init,
        mouse_ps2_close,
        mouse_ps2_poll
};
mouse_t mouse_intellimouse =
{
        "Microsoft Intellimouse (PS/2)",
        "intellimouse",
        MOUSE_TYPE_PS2 | MOUSE_TYPE_3BUTTON,
        mouse_intellimouse_init,
        mouse_ps2_close,
        mouse_ps2_poll
};