#include "ibm.h"
#include "io.h"
#include "mem.h"
#include "pic.h"
#include "pit.h"
#include "sound.h"
#include "sound_speaker.h"
#include "t3100e.h"
#include "timer.h"
#include "video.h"
#include "x86.h"
#include "xi8088.h"

#include "keyboard.h"
#include "keyboard_at.h"

#define STAT_PARITY 0x80
#define STAT_RTIMEOUT 0x40
#define STAT_TTIMEOUT 0x20
#define STAT_MFULL 0x20
#define STAT_LOCK 0x10
#define STAT_CD 0x08
#define STAT_SYSFLAG 0x04
#define STAT_IFULL 0x02
#define STAT_OFULL 0x01

#define PS2_REFRESH_TIME (16 * TIMER_USEC)

#define RESET_DELAY_TIME (100 * 10) /*600ms*/

struct {
        int initialised;
        int want60;
        int wantirq, wantirq12;
        uint8_t command;
        uint8_t status;
        uint8_t mem[0x20];
        uint8_t out;
        int out_new, out_delayed;

        int scancode_set;
        int translate;
        int next_is_release;

        uint8_t input_port;
        uint8_t output_port;

        uint8_t key_command;
        int key_wantdata;

        int last_irq;

        void (*mouse_write)(uint8_t val, void *p);
        void *mouse_p;

        pc_timer_t refresh_timer;
        int refresh;

        int is_ps2;

        pc_timer_t send_delay_timer;

        int reset_delay;
} keyboard_at;

/*Translation table taken from https://www.win.tue.nl/~aeb/linux/kbd/scancodes-10.html#ss10.3*/
static uint8_t at_translation[256] = {
        0xff, 0x43, 0x41, 0x3f, 0x3d, 0x3b, 0x3c, 0x58, 0x64, 0x44, 0x42, 0x40, 0x3e, 0x0f, 0x29, 0x59, 0x65, 0x38, 0x2a, 0x70,
        0x1d, 0x10, 0x02, 0x5a, 0x66, 0x71, 0x2c, 0x1f, 0x1e, 0x11, 0x03, 0x5b, 0x67, 0x2e, 0x2d, 0x20, 0x12, 0x05, 0x04, 0x5c,
        0x68, 0x39, 0x2f, 0x21, 0x14, 0x13, 0x06, 0x5d, 0x69, 0x31, 0x30, 0x23, 0x22, 0x15, 0x07, 0x5e, 0x6a, 0x72, 0x32, 0x24,
        0x16, 0x08, 0x09, 0x5f, 0x6b, 0x33, 0x25, 0x17, 0x18, 0x0b, 0x0a, 0x60, 0x6c, 0x34, 0x35, 0x26, 0x27, 0x19, 0x0c, 0x61,
        0x6d, 0x73, 0x28, 0x74, 0x1a, 0x0d, 0x62, 0x6e, 0x3a, 0x36, 0x1c, 0x1b, 0x75, 0x2b, 0x63, 0x76, 0x55, 0x56, 0x77, 0x78,
        0x79, 0x7a, 0x0e, 0x7b, 0x7c, 0x4f, 0x7d, 0x4b, 0x47, 0x7e, 0x7f, 0x6f, 0x52, 0x53, 0x50, 0x4c, 0x4d, 0x48, 0x01, 0x45,
        0x57, 0x4e, 0x51, 0x4a, 0x37, 0x49, 0x46, 0x54, 0x80, 0x81, 0x82, 0x41, 0x54, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b,
        0x8c, 0x8d, 0x8e, 0x8f, 0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f,
        0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf, 0xb0, 0xb1, 0xb2, 0xb3,
        0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf, 0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7,
        0xc8, 0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf, 0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xdb,
        0xdc, 0xdd, 0xde, 0xdf, 0xe0, 0xe1, 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef,
        0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff};
static uint8_t key_ctrl_queue[16];
static int key_ctrl_queue_start = 0, key_ctrl_queue_end = 0;

static uint8_t key_queue[16];
static int key_queue_start = 0, key_queue_end = 0;

uint8_t mouse_queue[16];
int mouse_queue_start = 0, mouse_queue_end = 0;

void keyboard_at_adddata_keyboard(uint8_t val);

void keyboard_at_poll() {
        timer_advance_u64(&keyboard_at.send_delay_timer, (100 * TIMER_USEC));

        if (keyboard_at.out_new != -1 && !keyboard_at.last_irq) {
                keyboard_at.wantirq = 0;
                if (keyboard_at.out_new & 0x100) {
                        if (mouse_scan) {
                                //                                pclog("keyboard_at : take IRQ12\n");
                                if (keyboard_at.mem[0] & 0x02)
                                        picint(0x1000);
                                keyboard_at.out = keyboard_at.out_new & 0xff;
                                keyboard_at.out_new = -1;
                                keyboard_at.status |= STAT_OFULL;
                                keyboard_at.status &= ~STAT_IFULL;
                                keyboard_at.status |= STAT_MFULL;
                                keyboard_at.last_irq = 0x1000;
                        } else {
                                //                                pclog("keyboard_at: suppressing IRQ12\n");
                                keyboard_at.out_new = -1;
                        }
                } else {
                        if (keyboard_at.mem[0] & 0x01)
                                picint(2);
                        keyboard_at.out = keyboard_at.out_new & 0xff;
                        keyboard_at.out_new = -1;
                        keyboard_at.status |= STAT_OFULL;
                        keyboard_at.status &= ~STAT_IFULL;
                        keyboard_at.status &= ~STAT_MFULL;
                        //                        pclog("keyboard_at : take IRQ1\n");
                        keyboard_at.last_irq = 2;
                }
        }

        if (keyboard_at.out_new == -1 && !(keyboard_at.status & STAT_OFULL) && key_ctrl_queue_start != key_ctrl_queue_end) {
                keyboard_at.out_new = key_ctrl_queue[key_ctrl_queue_start] | 0x200;
                key_ctrl_queue_start = (key_ctrl_queue_start + 1) & 0xf;
        } else if (!(keyboard_at.status & STAT_OFULL) && keyboard_at.out_new == -1 && keyboard_at.out_delayed != -1) {
                keyboard_at.out_new = keyboard_at.out_delayed;
                keyboard_at.out_delayed = -1;
        } else if (!(keyboard_at.status & STAT_OFULL) && keyboard_at.out_new == -1 && !(keyboard_at.mem[0] & 0x10) &&
                   keyboard_at.out_delayed != -1) {
                keyboard_at.out_new = keyboard_at.out_delayed;
                keyboard_at.out_delayed = -1;
        } else if (!(keyboard_at.status & STAT_OFULL) && keyboard_at.out_new == -1 && /*!(keyboard_at.mem[0] & 0x20) &&*/
                   mouse_queue_start != mouse_queue_end) {
                keyboard_at.out_new = mouse_queue[mouse_queue_start] | 0x100;
                mouse_queue_start = (mouse_queue_start + 1) & 0xf;
        } else if (!(keyboard_at.status & STAT_OFULL) && keyboard_at.out_new == -1 && !(keyboard_at.mem[0] & 0x10) &&
                   key_queue_start != key_queue_end) {
                keyboard_at.out_new = key_queue[key_queue_start];
                key_queue_start = (key_queue_start + 1) & 0xf;
        }

        if (keyboard_at.reset_delay) {
                keyboard_at.reset_delay--;
                if (!keyboard_at.reset_delay)
                        keyboard_at_adddata_keyboard(0xaa);
        }
}

void keyboard_at_adddata(uint8_t val) {
        key_ctrl_queue[key_ctrl_queue_end] = val;
        key_ctrl_queue_end = (key_ctrl_queue_end + 1) & 0xf;

        if (!(keyboard_at.out_new & 0x300)) {
                keyboard_at.out_delayed = keyboard_at.out_new;
                keyboard_at.out_new = -1;
        }
}

void keyboard_at_adddata_keyboard(uint8_t val) {
        if (keyboard_at.reset_delay)
                return;
        /*        if (val == 0x1c)
                {
                        key_1c++;
                        if (key_1c == 4)
                                output = 3;
                }*/

        /* Test for T3100E 'Fn' key (Right Alt / Right Ctrl) */
        if (romset == ROM_T3100E && (pcem_key[0xb8] || pcem_key[0x9d])) {
                switch (val) {
                case 0x4f:
                        t3100e_notify_set(0x01);
                        break; /* End */
                case 0x50:
                        t3100e_notify_set(0x02);
                        break; /* Down */
                case 0x51:
                        t3100e_notify_set(0x03);
                        break; /* PgDn */
                case 0x52:
                        t3100e_notify_set(0x04);
                        break; /* Ins */
                case 0x53:
                        t3100e_notify_set(0x05);
                        break; /* Del */
                case 0x54:
                        t3100e_notify_set(0x06);
                        break; /* SysRQ */
                case 0x45:
                        t3100e_notify_set(0x07);
                        break; /* NumLock */
                case 0x46:
                        t3100e_notify_set(0x08);
                        break; /* ScrLock */
                case 0x47:
                        t3100e_notify_set(0x09);
                        break; /* Home */
                case 0x48:
                        t3100e_notify_set(0x0A);
                        break; /* Up */
                case 0x49:
                        t3100e_notify_set(0x0B);
                        break; /* PgUp */
                case 0x4A:
                        t3100e_notify_set(0x0C);
                        break; /* Keypad -*/
                case 0x4B:
                        t3100e_notify_set(0x0D);
                        break; /* Left */
                case 0x4C:
                        t3100e_notify_set(0x0E);
                        break; /* KP 5 */
                case 0x4D:
                        t3100e_notify_set(0x0F);
                        break; /* Right */
                }
        }
        if (keyboard_at.translate) {
                if (val == 0xf0) {
                        keyboard_at.next_is_release = 1;
                        return;
                } else {
                        val = at_translation[val];
                        if (keyboard_at.next_is_release)
                                val |= 0x80;
                        keyboard_at.next_is_release = 0;
                }
        }
        key_queue[key_queue_end] = val;
        key_queue_end = (key_queue_end + 1) & 0xf;
        //        pclog("keyboard_at : %02X added to key queue\n", val);
        return;
}

void keyboard_at_adddata_mouse(uint8_t val) {
        mouse_queue[mouse_queue_end] = val;
        mouse_queue_end = (mouse_queue_end + 1) & 0xf;
        //        pclog("keyboard_at : %02X added to mouse queue\n", val);
        return;
}

void keyboard_at_write(uint16_t port, uint8_t val, void *priv) {
        //        pclog("keyboard_at : write %04X %02X %i  %02X\n", port, val, keyboard_at.key_wantdata, ram[8]);
        if (romset == ROM_XI8088 && port == 0x63)
                port = 0x61;
        /*        if (ram[8] == 0xc3)
                {
                        output = 3;
                }*/
        switch (port) {
        case 0x60:
                if (keyboard_at.want60) {
                        /*Write to controller*/
                        keyboard_at.want60 = 0;
                        switch (keyboard_at.command) {
                        case 0x60:
                        case 0x61:
                        case 0x62:
                        case 0x63:
                        case 0x64:
                        case 0x65:
                        case 0x66:
                        case 0x67:
                        case 0x68:
                        case 0x69:
                        case 0x6a:
                        case 0x6b:
                        case 0x6c:
                        case 0x6d:
                        case 0x6e:
                        case 0x6f:
                        case 0x70:
                        case 0x71:
                        case 0x72:
                        case 0x73:
                        case 0x74:
                        case 0x75:
                        case 0x76:
                        case 0x77:
                        case 0x78:
                        case 0x79:
                        case 0x7a:
                        case 0x7b:
                        case 0x7c:
                        case 0x7d:
                        case 0x7e:
                        case 0x7f:
                                keyboard_at.mem[keyboard_at.command & 0x1f] = val;
                                if (keyboard_at.command == 0x60) {
                                        if ((val & 1) && (keyboard_at.status & STAT_OFULL))
                                                keyboard_at.wantirq = 1;
                                        if (!(val & 1) && keyboard_at.wantirq)
                                                keyboard_at.wantirq = 0;
                                        mouse_scan = !(val & 0x20);
                                        keyboard_at.translate = val & 0x40;
                                }
                                break;

                        case 0xb6: /* T3100e - set colour/mono switch */
                                if (romset == ROM_T3100E)
                                        t3100e_mono_set(val);
                                break;
                        case 0xcb: /*AMI - set keyboard mode*/
                                break;

                        case 0xcf: /*??? - sent by MegaPC BIOS*/
                                break;

                        case 0xd1: /*Write output port*/
                                   //                                pclog("Write output port - %02X %02X %04X:%04X\n",
                                   //                                keyboard_at.output_port, val, CS, pc);
                                if ((keyboard_at.output_port ^ val) & 0x02) /*A20 enable change*/
                                {
                                        mem_a20_key = val & 0x02;
                                        mem_a20_recalc();
                                        //                                        pclog("Rammask change to %08X %02X\n", rammask,
                                        //                                        val & 0x02);
                                        flushmmucache();
                                }
                                keyboard_at.output_port = val;
                                break;

                        case 0xd2: /*Write to keyboard output buffer*/
                                keyboard_at_adddata(val);
                                break;

                        case 0xd3: /*Write to mouse output buffer*/
                                keyboard_at_adddata_mouse(val);
                                break;

                        case 0xd4: /*Write to mouse*/
                                if (keyboard_at.mouse_write) {
                                        keyboard_at.mouse_write(val, keyboard_at.mouse_p);
                                        /*Implicitly enable mouse*/
                                        mouse_scan = 1;
                                        keyboard_at.mem[0] &= ~0x20;
                                }
                                break;

                        default:
                                pclog("Bad AT keyboard controller 0060 write %02X command %02X\n", val, keyboard_at.command);
                                //                                dumpregs();
                                //                                exit(-1);
                        }
                } else {
                        /*Write to keyboard*/
                        keyboard_at.mem[0] &= ~0x10;
                        if (keyboard_at.key_wantdata) {
                                keyboard_at.key_wantdata = 0;
                                switch (keyboard_at.key_command) {
                                case 0xed: /*Set/reset LEDs*/
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                case 0xf0: /*Set scancode set*/
                                        switch (val) {
                                        case 0: /*Read current set*/
                                                keyboard_at_adddata_keyboard(0xfa);
                                                switch (keyboard_at.scancode_set) {
                                                case SCANCODE_SET_1:
                                                        keyboard_at_adddata_keyboard(0x01);
                                                        break;
                                                case SCANCODE_SET_2:
                                                        keyboard_at_adddata_keyboard(0x02);
                                                        break;
                                                case SCANCODE_SET_3:
                                                        keyboard_at_adddata_keyboard(0x03);
                                                        break;
                                                }
                                                break;
                                        case 1:
                                                keyboard_at.scancode_set = SCANCODE_SET_1;
                                                keyboard_set_scancode_set(SCANCODE_SET_1);
                                                keyboard_at_adddata_keyboard(0xfa);
                                                break;
                                        case 2:
                                                keyboard_at.scancode_set = SCANCODE_SET_2;
                                                keyboard_set_scancode_set(SCANCODE_SET_2);
                                                keyboard_at_adddata_keyboard(0xfa);
                                                break;
                                        case 3:
                                                keyboard_at.scancode_set = SCANCODE_SET_3;
                                                keyboard_set_scancode_set(SCANCODE_SET_3);
                                                keyboard_at_adddata_keyboard(0xfa);
                                                break;
                                        default:
                                                keyboard_at_adddata_keyboard(0xfe);
                                                break;
                                        }
                                        break;

                                case 0xf3: /*Set typematic rate/delay*/
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                default:
                                        pclog("Bad AT keyboard 0060 write %02X command %02X\n", val, keyboard_at.key_command);
                                        //                                        dumpregs();
                                        //                                        exit(-1);
                                }
                        } else {
                                keyboard_at.key_command = val;
                                switch (val) {
                                case 0x05: /*??? - sent by NT 4.0*/
                                        keyboard_at_adddata_keyboard(0xfe);
                                        break;

                                case 0xed: /*Set/reset LEDs*/
                                        keyboard_at.key_wantdata = 1;
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                case 0xee: /*Diagnostic echo*/
                                        keyboard_at_adddata_keyboard(0xee);
                                        break;

                                case 0xf0: /*Set scancode set*/
                                        keyboard_at.key_wantdata = 1;
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                case 0xf2: /*Read ID*/
                                        keyboard_at_adddata_keyboard(0xfa);
                                        keyboard_at_adddata_keyboard(0xab);
                                        keyboard_at_adddata_keyboard(0x83);
                                        break;

                                case 0xf3: /*Set typematic rate/delay*/
                                        keyboard_at.key_wantdata = 1;
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                case 0xf4: /*Enable keyboard*/
                                        keyboard_scan = 1;
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;
                                case 0xf5: /*Disable keyboard*/
                                        keyboard_scan = 0;
                                        keyboard_at_adddata_keyboard(0xfa);
                                        break;

                                case 0xff:                                   /*Reset*/
                                        key_queue_start = key_queue_end = 0; /*Clear key queue*/
                                        keyboard_at_adddata_keyboard(0xfa);
                                        keyboard_at.reset_delay = RESET_DELAY_TIME;
                                        break;

                                default:
                                        pclog("Bad AT keyboard command %02X\n", val);
                                        keyboard_at_adddata_keyboard(0xfe);
                                        //                                        dumpregs();
                                        //                                        exit(-1);
                                }
                        }
                }
                break;

        case 0x61:
                ppi.pb = val;

                speaker_update();
                speaker_gated = val & 1;
                speaker_enable = val & 2;
                if (speaker_enable)
                        was_speaker_enable = 1;
                pit_set_gate(&pit, 2, val & 1);

                if (romset == ROM_XI8088) {
                        if (val & 0x04)
                                xi8088_turbo_set(1);
                        else
                                xi8088_turbo_set(0);
                }
                break;

        case 0x64:
                keyboard_at.want60 = 0;
                keyboard_at.command = val;
                /*New controller command*/
                switch (val) {
                case 0x09: /* GRID1520 - set backlight timeout */
                        if (romset != ROM_GRID1520)
                                pclog("Bad AT keyboard controller command %02X\n", val);
                        break;
                case 0x20:
                case 0x21:
                case 0x22:
                case 0x23:
                case 0x24:
                case 0x25:
                case 0x26:
                case 0x27:
                case 0x28:
                case 0x29:
                case 0x2a:
                case 0x2b:
                case 0x2c:
                case 0x2d:
                case 0x2e:
                case 0x2f:
                case 0x30:
                case 0x31:
                case 0x32:
                case 0x33:
                case 0x34:
                case 0x35:
                case 0x36:
                case 0x37:
                case 0x38:
                case 0x39:
                case 0x3a:
                case 0x3b:
                case 0x3c:
                case 0x3d:
                case 0x3e:
                case 0x3f:
                        keyboard_at_adddata(keyboard_at.mem[val & 0x1f]);
                        break;

                case 0x60:
                case 0x61:
                case 0x62:
                case 0x63:
                case 0x64:
                case 0x65:
                case 0x66:
                case 0x67:
                case 0x68:
                case 0x69:
                case 0x6a:
                case 0x6b:
                case 0x6c:
                case 0x6d:
                case 0x6e:
                case 0x6f:
                case 0x70:
                case 0x71:
                case 0x72:
                case 0x73:
                case 0x74:
                case 0x75:
                case 0x76:
                case 0x77:
                case 0x78:
                case 0x79:
                case 0x7a:
                case 0x7b:
                case 0x7c:
                case 0x7d:
                case 0x7e:
                case 0x7f:
                        keyboard_at.want60 = 1;
                        break;

                case 0xa1: /*AMI - get controlled version*/
                        break;

                case 0xa7: /*Disable mouse port*/
                        mouse_scan = 0;
                        keyboard_at.mem[0] |= 0x20;
                        break;

                case 0xa8: /*Enable mouse port*/
                        mouse_scan = 1;
                        keyboard_at.mem[0] &= ~0x20;
                        break;

                case 0xa9:                         /*Test mouse port*/
                        keyboard_at_adddata(0x00); /*no error*/
                        break;

                case 0xaa: /*Self-test*/
                        if (!keyboard_at.initialised) {
                                keyboard_at.initialised = 1;
                                key_ctrl_queue_start = key_ctrl_queue_end = 0;
                                keyboard_at.status &= ~STAT_OFULL;
                        }
                        /* T3100e and Samsung SPC-6000A expects STAT_IFULL to be set immediately
                         * after sending 0xAA */
                        if (romset == ROM_T3100E || romset == ROM_SPC6000A)
                                keyboard_at.status |= STAT_IFULL;
                        keyboard_at.status |= STAT_SYSFLAG;
                        keyboard_at.mem[0] |= 0x04;
                        keyboard_at_adddata(0x55);
                        /*Self-test also resets the output port, enabling A20*/
                        if (!(keyboard_at.output_port & 0x02)) {
                                mem_a20_key = 2;
                                mem_a20_recalc();
                                //                                pclog("Rammask change to %08X %02X\n", rammask, val & 0x02);
                                flushmmucache();
                        }
                        keyboard_at.output_port = 0xcf;
                        break;

                case 0xab:                         /*Interface test*/
                        keyboard_at_adddata(0x00); /*no error*/
                        break;

                case 0xad: /*Disable keyboard*/
                        keyboard_at.mem[0] |= 0x10;
                        break;

                case 0xae: /*Enable keyboard*/
                        keyboard_at.mem[0] &= ~0x10;
                        break;

                case 0xb0: /* T3100e: Turbo on */
                        if (romset == ROM_T3100E)
                                t3100e_turbo_set(1);
                        break;

                case 0xb1: /* T3100e: Turbo off */
                        if (romset == ROM_T3100E)
                                t3100e_turbo_set(0);
                        break;

                case 0xb2: /* T3100e: Select external display */
                        if (romset == ROM_T3100E)
                                t3100e_display_set(0x00);
                        break;

                case 0xb3: /* T3100e: Select internal display */
                        if (romset == ROM_T3100E)
                                t3100e_display_set(0x01);
                        break;

                case 0xb4: /* T3100e: Get configuration / status */
                        if (romset == ROM_T3100E)
                                keyboard_at_adddata(t3100e_config_get());
                        break;

                case 0xb5: /* T3100e: Get colour / mono byte */
                        if (romset == ROM_T3100E)
                                keyboard_at_adddata(t3100e_mono_get());
                        break;

                case 0xb6: /* T3100e: Set colour / mono byte */
                        if (romset == ROM_T3100E)
                                keyboard_at.want60 = 1;
                        break;

                case 0xba:
                        if (romset == ROM_ENDEAVOR || romset == ROM_ZAPPA || romset == ROM_ITAUTEC_INFOWAYM)
                                keyboard_at_adddata(0);
                        break;

                        /* T3100e commands not implemented:
                         * 0xB7: Emulate PS/2 keyboard
                         * 0xB8: Emulate AT keyboard */

                case 0xbb: /* T3100e: Read 'Fn' key.
                                      Return it for right Ctrl and right
                                      Alt; on the real T3100e, these keystrokes
                                      could only be generated using 'Fn'. */
                        if (romset == ROM_T3100E) {
                                if (pcem_key[0xb8] || /* Right Alt */
                                    pcem_key[0x9d])   /* Right Ctrl */
                                {
                                        keyboard_at_adddata(0x04);
                                } else
                                        keyboard_at_adddata(0x00);
                        }
                        break;

                case 0xbc: /* T3100e: Reset Fn+Key notification */
                        if (romset == ROM_T3100E)
                                t3100e_notify_set(0x00);
                        break;

                case 0xc0: /*Read input port*/
                        /* The T3100e returns all bits set except bit 6 which
                         * is set by t3100e_mono_set() */
                        if (romset == ROM_T3100E)
                                keyboard_at.input_port = (t3100e_mono_get() & 1) ? 0xFF : 0xBF;

                        if (romset == ROM_ENDEAVOR || romset == ROM_ZAPPA || romset == ROM_ITAUTEC_INFOWAYM)
                                keyboard_at_adddata(keyboard_at.input_port | 4 | 0x40);
                        else
                                keyboard_at_adddata(keyboard_at.input_port | 4);

                        keyboard_at.input_port = ((keyboard_at.input_port + 1) & 3) | (keyboard_at.input_port & 0xfc);
                        break;

                case 0xc9: /*AMI - block P22 and P23 ??? */
                        break;

                case 0xca:                                 /*AMI - read keyboard mode*/
                        if (romset == ROM_GA686BX)         /*TODO*/
                                keyboard_at_adddata(0x01); /*PS2 mode*/
                        else
                                keyboard_at_adddata(0x00); /*ISA mode*/
                        break;

                case 0xcb: /*AMI - set keyboard mode*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xcf: /*??? - sent by MegaPC BIOS*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xd0: /*Read output port*/
                        keyboard_at_adddata(keyboard_at.output_port);
                        break;

                case 0xd1: /*Write output port*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xd2: /*Write keyboard output buffer*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xd3: /*Write mouse output buffer*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xd4: /*Write to mouse*/
                        keyboard_at.want60 = 1;
                        break;

                case 0xe0: /*Read test inputs*/
                        keyboard_at_adddata(0x00);
                        break;

                        // AWARD BIOS: called after turbo ON/OFF in the
                        // Hyundai Super-286TR and probably other AWARD 286
                        // bioses. Related to the Turbo LEDs. Exact function
                        // can't be confirmed because the system in question
                        // has no such leds.
                        // case 0xe8: /* Turbo ON, always followed by 0xba */
                        // case 0xe9: /* Turbo OFF */
                        // break;

                case 0xef: /*??? - sent by AMI486*/
                        break;

                case 0xf0:
                case 0xf2:
                case 0xf4:
                case 0xf6:
                case 0xf8:
                case 0xfa:
                case 0xfc:
                case 0xfe:              /*Pulse output port - pin 0 selected - x86 reset*/
                        softresetx86(); /*Pulse reset!*/
                        cpu_set_edx();
                        break;

                case 0xff: /*Pulse output port - but no pins selected - sent by MegaPC BIOS*/
                        break;

                default:
                        pclog("Bad AT keyboard controller command %02X\n", val);
                        //                        dumpregs();
                        //                        exit(-1);
                }
        }
}

uint8_t keyboard_at_read(uint16_t port, void *priv) {
        uint8_t temp = 0xff;

        if (romset != ROM_IBMAT && romset != ROM_IBMXT286)
                cycles -= ISA_CYCLES(8);
        //        if (port != 0x61) pclog("keyboard_at : read %04X ", port);
        if (romset == ROM_XI8088 && port == 0x63)
                port = 0x61;
        switch (port) {
        case 0x60:
                temp = keyboard_at.out;
                keyboard_at.status &= ~(STAT_OFULL /* | STAT_MFULL*/);
                picintc(keyboard_at.last_irq);
                keyboard_at.last_irq = 0;
                break;

        case 0x61:
                temp = ppi.pb & ~0xe0;
                if (ppispeakon)
                        temp |= 0x20;
                if (keyboard_at.is_ps2) {
                        if (keyboard_at.refresh)
                                temp |= 0x10;
                        else
                                temp &= ~0x10;
                }
                if (romset == ROM_XI8088) {
                        if (xi8088_turbo_get())
                                temp |= 0x04;
                        else
                                temp &= ~0x04;
                }
                break;

        case 0x64:
                temp = keyboard_at.status & ~4;
                if (keyboard_at.mem[0] & 0x04)
                        temp |= 0x04;
                keyboard_at.status &= ~(STAT_RTIMEOUT /* | STAT_TTIMEOUT*/);
                break;
        }
        //        if (port != 0x61) pclog("%02X  %08X\n", temp, rammask);
        return temp;
}

void keyboard_at_reset() {
        keyboard_at.initialised = 0;
        keyboard_at.status = STAT_LOCK | STAT_CD;
        keyboard_at.mem[0] = 0x11;
        keyboard_at.wantirq = 0;
        keyboard_at.output_port = 0xcf;
        if (romset == ROM_XI8088)
                keyboard_at.input_port = (video_is_mda()) ? 0xb0 : 0xf0;
        else
                keyboard_at.input_port = (video_is_mda()) ? 0xf0 : 0xb0;
        keyboard_at.out_new = -1;
        keyboard_at.out_delayed = -1;
        keyboard_at.last_irq = 0;

        keyboard_at.key_wantdata = 0;

        keyboard_scan = 1;
}

static void at_refresh(void *p) {
        keyboard_at.refresh = !keyboard_at.refresh;
        timer_advance_u64(&keyboard_at.refresh_timer, PS2_REFRESH_TIME);
}

void keyboard_at_init() {
        // return;
        memset(&keyboard_at, 0, sizeof(keyboard_at));
        io_sethandler(0x0060, 0x0005, keyboard_at_read, NULL, NULL, keyboard_at_write, NULL, NULL, NULL);
        keyboard_at_reset();
        keyboard_send = keyboard_at_adddata_keyboard;
        keyboard_poll = keyboard_at_poll;
        keyboard_at.mouse_write = NULL;
        keyboard_at.mouse_p = NULL;
        keyboard_at.is_ps2 = 0;
        keyboard_set_scancode_set(SCANCODE_SET_2);
        keyboard_at.scancode_set = SCANCODE_SET_2;

        timer_add(&keyboard_at.send_delay_timer, keyboard_at_poll, NULL, 1);
}

void keyboard_at_set_mouse(void (*mouse_write)(uint8_t val, void *p), void *p) {
        keyboard_at.mouse_write = mouse_write;
        keyboard_at.mouse_p = p;
}

void keyboard_at_init_ps2() {
        timer_add(&keyboard_at.refresh_timer, at_refresh, NULL, 1);
        keyboard_at.is_ps2 = 1;
}
