/*
 * Copyright 2001-2014 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors (in chronological order):
 *              Elad Lahav, elad@eldarshany.com
 *              Stefano Ceccherini, burton666@libero.it
 *              Axel Dörfler, axeld@pinc-software.de
 *              Marcus Overhagen, marcus@overhagen.de
 *              Clemens Zeidler, czeidler@gmx.de
 *              John Scipione, jscipione@gmail.com
 */


/*!     PS/2 mouse device driver

        A PS/2 mouse is connected to the IBM 8042 controller, and gets its
        name from the IBM PS/2 personal computer, which was the first to
        use this device. All resources are shared between the keyboard, and
        the mouse, referred to as the "Auxiliary Device".

        I/O:
        ~~~
        The controller has 3 I/O registers:
        1. Status (input), mapped to port 64h
        2. Control (output), mapped to port 64h
        3. Data (input/output), mapped to port 60h

        Data:
        ~~~~
        A packet read from the mouse data port is composed of
        three bytes:
                byte 0: status byte, where
                        - bit 7: Y overflow (1 = true)
                        - bit 6: X overflow (1 = true)
                        - bit 5: MSB of Y offset
                        - bit 4: MSB of X offset
                        - bit 3: Syncronization bit (always 1)
                        - bit 2: Middle button (1 = down)
                        - bit 1: Right button (1 = down)
                        - bit 0: Left button (1 = down)
                byte 1: X position change, since last probed (-127 to +127)
                byte 2: Y position change, since last probed (-127 to +127)

        Intellimouse mice send a four byte packet, where the first three
        bytes are the same as standard mice, and the last one reports the
        Z position, which is, usually, the wheel movement.

        Interrupts:
        ~~~~~~~~~~
        The PS/2 mouse device is connected to interrupt 12.
        The controller uses 3 consecutive interrupts to inform the computer
        that it has new data. On the first the data register holds the status
        byte, on the second the X offset, and on the 3rd the Y offset.
*/


#include <stdlib.h>
#include <string.h>

#include <keyboard_mouse_driver.h>

#include "ps2_service.h"
#include "ps2_standard_mouse.h"


//#define TRACE_PS2_MOUSE
#ifdef TRACE_PS2_MOUSE
#       define TRACE(x...) dprintf(x)
#else
#       define TRACE(x...)
#endif


const char* kStandardMousePath[4] = {
        "input/mouse/ps2/standard_0",
        "input/mouse/ps2/standard_1",
        "input/mouse/ps2/standard_2",
        "input/mouse/ps2/standard_3"
};

const char* kIntelliMousePath[4] = {
        "input/mouse/ps2/intelli_0",
        "input/mouse/ps2/intelli_1",
        "input/mouse/ps2/intelli_2",
        "input/mouse/ps2/intelli_3"
};


//!     Set sampling rate of the ps2 port.
static inline status_t
ps2_set_sample_rate(ps2_dev* dev, uint8 rate)
{
        return ps2_dev_command(dev, PS2_CMD_SET_SAMPLE_RATE, &rate, 1, NULL, 0);
}


//!     Converts a packet received by the mouse to a "movement".
static void
ps2_packet_to_movement(standard_mouse_cookie* cookie, uint8 packet[],
        mouse_movement* pos)
{
        int buttons = packet[0] & 7;
        int xDelta = ((packet[0] & 0x10) ? ~0xff : 0) | packet[1];
        int yDelta = ((packet[0] & 0x20) ? ~0xff : 0) | packet[2];
        int xDeltaWheel = 0;
        int yDeltaWheel = 0;
        bigtime_t currentTime = system_time();

        if (buttons != 0 && cookie->buttons_state == 0) {
                if (cookie->click_last_time + cookie->click_speed > currentTime)
                        cookie->click_count++;
                else
                        cookie->click_count = 1;

                cookie->click_last_time = currentTime;
        }

        cookie->buttons_state = buttons;

        if (cookie->flags & F_MOUSE_TYPE_INTELLIMOUSE) {
                yDeltaWheel = packet[3] & 0x07;
                if (packet[3] & 0x08)
                        yDeltaWheel |= ~0x07;
        }
#if 0
        if (cookie->flags & F_MOUSE_TYPE_2WHEELS) {
                switch (packet[3] & 0x0F) {
                        case 0x01: yDeltaWheel = +1; break; // wheel 1 down
                        case 0x0F: yDeltaWheel = -1; break; // wheel 1 up
                        case 0x02: xDeltaWheel = +1; break; // wheel 2 down
                        case 0x0E: xDeltaWheel = -1; break; // wheel 2 up
                }
        }
#endif

#if 0
        TRACE("packet: %02x %02x %02x %02x: xd %d, yd %d, 0x%x (%d), w-xd %d, "
                "w-yd %d\n", packet[0], packet[1], packet[2], packet[3],
                xDelta, yDelta, buttons, cookie->click_count, xDeltaWheel,
                yDeltaWheel);
#endif

        if (pos != NULL) {
                pos->xdelta = xDelta;
                pos->ydelta = yDelta;
                pos->buttons = buttons;
                pos->clicks = cookie->click_count;
                pos->modifiers = 0;
                pos->timestamp = currentTime;
                pos->wheel_ydelta = yDeltaWheel;
                pos->wheel_xdelta = xDeltaWheel;

                TRACE("ps2: ps2_packet_to_movement xdelta: %d, ydelta: %d, buttons %x, "
                        "clicks: %d, timestamp %" B_PRIdBIGTIME "\n",
                        xDelta, yDelta, buttons, cookie->click_count, currentTime);
        }
}


//!     Read a mouse event from the mouse events chain buffer.
static status_t
standard_mouse_read_event(standard_mouse_cookie* cookie,
        mouse_movement* movement)
{
        uint8 packet[PS2_MAX_PACKET_SIZE];
        status_t status;

        TRACE("ps2: standard_mouse_read_event\n");

        status = acquire_sem_etc(cookie->standard_mouse_sem, 1, B_CAN_INTERRUPT, 0);
        TRACE("ps2: standard_mouse_read_event acquired\n");
        if (status < B_OK)
                return status;

        if (!cookie->dev->active) {
                TRACE("ps2: standard_mouse_read_event: Error device no longer "
                        "active\n");
                return B_ERROR;
        }

        if (packet_buffer_read(cookie->standard_mouse_buffer, packet,
                cookie->dev->packet_size) != cookie->dev->packet_size) {
                TRACE("ps2: error copying buffer\n");
                return B_ERROR;
        }

        if (!(packet[0] & 8))
                panic("ps2: got broken data from packet_buffer_read\n");

        ps2_packet_to_movement(cookie, packet, movement);
        return B_OK;
}


//      #pragma mark - Interrupt handler functions


void
standard_mouse_disconnect(ps2_dev* dev)
{
        // the mouse device might not be opened at this point
        INFO("ps2: ps2_standard_mouse_disconnect %s\n", dev->name);
        if (dev->flags & PS2_FLAG_OPEN)
                release_sem(((standard_mouse_cookie*)dev->cookie)->standard_mouse_sem);
}


/*!     Interrupt handler for the mouse device. Called whenever the I/O
        controller generates an interrupt for the PS/2 mouse. Reads mouse
        information from the data port, and stores it, so it can be accessed
        by read() operations. The full data is obtained using 3 consecutive
        calls to the handler, each holds a different byte on the data port.
*/
int32
standard_mouse_handle_int(ps2_dev* dev)
{
        standard_mouse_cookie* cookie = (standard_mouse_cookie*)dev->cookie;
        const uint8 data = dev->history[0].data;

        TRACE("ps2: standard mouse: %1x\t%1x\t%1x\t%1x\t%1x\t%1x\t%1x\t%1x\n",
                data >> 7 & 1, data >> 6 & 1, data >> 5 & 1,
                data >> 4 & 1, data >> 3 & 1, data >> 2 & 1,
                data >> 1 & 1, data >> 0 & 1);

        if (cookie->packet_index == 0 && (data & 8) == 0) {
                INFO("ps2: bad mouse data, trying resync\n");
                return B_HANDLED_INTERRUPT;
        }

        // Workarounds for active multiplexing keyboard controllers
        // that lose data or send them to the wrong port.
        if (cookie->packet_index == 0 && (data & 0xc0) != 0) {
                INFO("ps2: strange mouse data, x/y overflow, trying resync\n");
                return B_HANDLED_INTERRUPT;
        }

        cookie->buffer[cookie->packet_index++] = data;

        if (cookie->packet_index != dev->packet_size) {
                // packet not yet complete
                return B_HANDLED_INTERRUPT;
        }

        // complete packet is assembled

        cookie->packet_index = 0;
        if (packet_buffer_write(cookie->standard_mouse_buffer,
                cookie->buffer, dev->packet_size) != dev->packet_size) {
                // buffer is full, drop new data
                return B_HANDLED_INTERRUPT;
        }

        release_sem_etc(cookie->standard_mouse_sem, 1, B_DO_NOT_RESCHEDULE);

        return B_INVOKE_SCHEDULER;
}


//      #pragma mark - probe_standard_mouse()


status_t
probe_standard_mouse(ps2_dev* dev)
{
        status_t status;
        uint8 deviceId = 0;

        // get device id
        status = ps2_dev_command(dev, PS2_CMD_GET_DEVICE_ID, NULL, 0,
                &deviceId, 1);
        if (status != B_OK) {
                INFO("ps2: probe_mouse get device id failed\n");
                return B_ERROR;
        }

        TRACE("ps2: probe_mouse device id: %2x\n", deviceId);

        // check for MS Intellimouse
        if (deviceId == 0) {
                uint8 alternate_device_id;
                status  = ps2_set_sample_rate(dev, 200);
                status |= ps2_set_sample_rate(dev, 100);
                status |= ps2_set_sample_rate(dev, 80);
                status |= ps2_dev_command(dev, PS2_CMD_GET_DEVICE_ID, NULL, 0,
                        &alternate_device_id, 1);
                if (status == 0) {
                        TRACE("ps2: probe_mouse alternate device id: %2x\n",
                                alternate_device_id);
                        deviceId = alternate_device_id;
                }
        }

        if (deviceId == PS2_DEV_ID_STANDARD
                || deviceId == PS2_DEV_ID_TOUCHPAD_RICATECH) {
                INFO("ps2: probe_mouse Standard PS/2 mouse found\n");
                dev->name = kStandardMousePath[dev->idx];
                dev->packet_size = PS2_PACKET_STANDARD;
        } else if (deviceId == PS2_DEV_ID_INTELLIMOUSE) {
                dev->name = kIntelliMousePath[dev->idx];
                dev->packet_size = PS2_PACKET_INTELLIMOUSE;
                INFO("ps2: probe_mouse Extended PS/2 mouse found\n");
        } else {
                INFO("ps2: probe_mouse Error unknown device id.\n");
                return B_ERROR;
        }

        return B_OK;
}


//      #pragma mark - Device functions


status_t
standard_mouse_open(const char* name, uint32 flags, void** _cookie)
{
        standard_mouse_cookie* cookie;
        status_t status;
        ps2_dev* dev = NULL;
        int i;

        TRACE("ps2: standard_mouse_open %s\n", name);

        for (dev = NULL, i = 0; i < PS2_DEVICE_COUNT; i++) {
                if (0 == strcmp(ps2_device[i].name, name)) {
                        dev = &ps2_device[i];
                        break;
                }
#if 0
                if (0 == strcmp(g_passthrough_dev.name, name)) {
                        dev = &g_passthrough_dev;
                        isSynapticsPTDevice = true;
                        break;
                }
#endif
        }

        if (dev == NULL) {
                TRACE("ps2: dev = NULL\n");
                return B_ERROR;
        }

        if (atomic_or(&dev->flags, PS2_FLAG_OPEN) & PS2_FLAG_OPEN)
                return B_BUSY;

        cookie = (standard_mouse_cookie*)malloc(sizeof(standard_mouse_cookie));
        if (cookie == NULL)
                goto err1;

        *_cookie = cookie;
        memset(cookie, 0, sizeof(*cookie));

        cookie->dev = dev;
        dev->cookie = cookie;
        dev->disconnect = &standard_mouse_disconnect;
        dev->handle_int = &standard_mouse_handle_int;

        if (strstr(dev->name, "standard") != NULL)
                cookie->flags = F_MOUSE_TYPE_STANDARD;

        if (strstr(dev->name, "intelli") != NULL)
                cookie->flags = F_MOUSE_TYPE_INTELLIMOUSE;

        cookie->standard_mouse_buffer
                = create_packet_buffer(MOUSE_HISTORY_SIZE * dev->packet_size);
        if (cookie->standard_mouse_buffer == NULL) {
                TRACE("ps2: can't allocate mouse actions buffer\n");
                goto err2;
        }

        // create the mouse semaphore, used for synchronization between
        // the interrupt handler and the read operation
        cookie->standard_mouse_sem = create_sem(0, "ps2_standard_mouse_sem");
        if (cookie->standard_mouse_sem < 0) {
                TRACE("ps2: failed creating PS/2 mouse semaphore!\n");
                goto err3;
        }

        status = ps2_dev_command(dev, PS2_CMD_ENABLE, NULL, 0, NULL, 0);
        if (status < B_OK) {
                INFO("ps2: cannot enable mouse %s\n", name);
                goto err4;
        }

        atomic_or(&dev->flags, PS2_FLAG_ENABLED);


        TRACE("ps2: standard_mouse_open %s success\n", name);
        return B_OK;

err4:
        delete_sem(cookie->standard_mouse_sem);
err3:
        delete_packet_buffer(cookie->standard_mouse_buffer);
err2:
        free(cookie);
err1:
        atomic_and(&dev->flags, ~PS2_FLAG_OPEN);

        TRACE("ps2: standard_mouse_open %s failed\n", name);
        return B_ERROR;
}


status_t
standard_mouse_close(void* _cookie)
{
        standard_mouse_cookie* cookie = (standard_mouse_cookie*)_cookie;

        TRACE("ps2: standard_mouse_close %s enter\n", cookie->dev->name);

        ps2_dev_command_timeout(cookie->dev, PS2_CMD_DISABLE, NULL, 0, NULL, 0,
                150000);

        delete_packet_buffer(cookie->standard_mouse_buffer);
        delete_sem(cookie->standard_mouse_sem);

        atomic_and(&cookie->dev->flags, ~PS2_FLAG_OPEN);
        atomic_and(&cookie->dev->flags, ~PS2_FLAG_ENABLED);

        TRACE("ps2: standard_mouse_close %s done\n", cookie->dev->name);
        return B_OK;
}


status_t
standard_mouse_freecookie(void* _cookie)
{
        standard_mouse_cookie* cookie = (standard_mouse_cookie*)_cookie;
        free(cookie);
        return B_OK;
}


static status_t
standard_mouse_read(void* cookie, off_t pos, void* buffer, size_t* _length)
{
        *_length = 0;
        return B_NOT_ALLOWED;
}


static status_t
standard_mouse_write(void* cookie, off_t pos, const void* buffer,
        size_t* _length)
{
        *_length = 0;
        return B_NOT_ALLOWED;
}


status_t
standard_mouse_ioctl(void* _cookie, uint32 op, void* buffer, size_t length)
{
        standard_mouse_cookie* cookie = (standard_mouse_cookie*)_cookie;

        switch (op) {
                case MS_NUM_EVENTS:
                {
                        int32 count;
                        TRACE("ps2: ioctl MS_NUM_EVENTS\n");
                        get_sem_count(cookie->standard_mouse_sem, &count);
                        return count;
                }

                case MS_READ:
                {
                        mouse_movement movement;
                        status_t status;
                        TRACE("ps2: ioctl MS_READ\n");
                        if ((status = standard_mouse_read_event(cookie, &movement)) < B_OK)
                                return status;
//                      TRACE("%s %d %d %d %d\n", cookie->dev->name,
//                              movement.xdelta, movement.ydelta, movement.buttons,
//                              movement.clicks);
                        return user_memcpy(buffer, &movement, sizeof(movement));
                }

                case MS_SET_TYPE:
                        TRACE("ps2: ioctl MS_SET_TYPE not implemented\n");
                        return B_BAD_VALUE;

                case MS_SET_MAP:
                        TRACE("ps2: ioctl MS_SET_MAP (set mouse mapping) not "
                                "implemented\n");
                        return B_BAD_VALUE;

                case MS_GET_ACCEL:
                        TRACE("ps2: ioctl MS_GET_ACCEL (get mouse acceleration) not "
                                "implemented\n");
                        return B_BAD_VALUE;

                case MS_SET_ACCEL:
                        TRACE("ps2: ioctl MS_SET_ACCEL (set mouse acceleration) not "
                                "implemented\n");
                        return B_BAD_VALUE;

                case MS_SET_CLICKSPEED:
                        TRACE("ps2: ioctl MS_SETCLICK (set click speed)\n");
                        return user_memcpy(&cookie->click_speed, buffer, sizeof(bigtime_t));

                default:
                        TRACE("ps2: ioctl unknown mouse opcode: %" B_PRIu32 "\n", op);
                        return B_DEV_INVALID_IOCTL;
        }
}


device_hooks gStandardMouseDeviceHooks = {
        standard_mouse_open,
        standard_mouse_close,
        standard_mouse_freecookie,
        standard_mouse_ioctl,
        standard_mouse_read,
        standard_mouse_write,
};