/*      $NetBSD: netwalker_btn.c,v 1.4 2021/08/07 16:18:50 thorpej Exp $        */

/*
* Copyright (c) 2014  Genetec Corporation.  All rights reserved.
* Written by Hashimoto Kenichi for Genetec Corporation.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY GENETEC CORPORATION ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL GENETEC CORPORATION
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: netwalker_btn.c,v 1.4 2021/08/07 16:18:50 thorpej Exp $");

#include "opt_imxspi.h"
#include "opt_mousebtn.h"

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/device.h>
#include <sys/lock.h>
#include <sys/callout.h>
#include <sys/gpio.h>
#include <sys/bus.h>
#include <sys/mutex.h>

#include <dev/wscons/wsconsio.h>
#include <dev/wscons/wsmousevar.h>

#include <dev/gpio/gpiovar.h>

#define GPIO1_BASE              160

#define MOUSEBTN_PIN_LEFT       0
#define MOUSEBTN_PIN_RIGHT      1
#define MOUSEBTN_NPINS          2

#define POLLRATE (hz/10)

struct mousebtn_softc
{
       device_t sc_dev;
       void *sc_gpio;
       void *sc_intr[MOUSEBTN_NPINS];

       struct gpio_pinmap sc_map;
       int sc_map_pins[MOUSEBTN_NPINS];

       int sc_buttons;

       struct callout sc_c;
       kmutex_t sc_lock;

       int sc_enabled;
       struct device *sc_wsmousedev;
};

static int mousebtn_match(device_t, cfdata_t, void *);
static void mousebtn_attach(device_t, device_t, void *);
static int mousebtn_detach(device_t, int);

CFATTACH_DECL_NEW(netwalker_btn, sizeof(struct mousebtn_softc),
   mousebtn_match, mousebtn_attach, mousebtn_detach, NULL);

static void mousebtn_poll(void *);
static int mousebtn_intr(void *);

static int mousebtn_enable(void *v);
static void mousebtn_disable(void *v);
static int mousebtn_ioctl(void *, u_long, void *, int, struct lwp *);

static bool mousebtn_resume(device_t, const pmf_qual_t *);
static bool mousebtn_suspend(device_t, const pmf_qual_t *);

static const struct wsmouse_accessops mousebtn_accessops = {
       .enable  = mousebtn_enable,
       .ioctl   = mousebtn_ioctl,
       .disable = mousebtn_disable
};

static int
mousebtn_match(device_t parent, cfdata_t cf, void * aux)
{
       struct gpio_attach_args *ga = aux;

       if (strcmp(ga->ga_dvname, cf->cf_name))
               return 0;
       if (ga->ga_offset == -1)
               return 0;

       /* check that we have enough pins */
       if (gpio_npins(ga->ga_mask) != MOUSEBTN_NPINS) {
               aprint_debug("%s: invalid pin mask 0x%02x\n", cf->cf_name,
                   ga->ga_mask);
               return 0;
       }

       return 1;
}

static void
mousebtn_attach(device_t parent, device_t self, void *aux)
{
       struct mousebtn_softc *sc = device_private(self);
       struct gpio_attach_args *ga = aux;
       int caps;
       struct wsmousedev_attach_args a;

       sc->sc_dev = self;
       sc->sc_gpio = ga->ga_gpio;

       /* map pins */
       sc->sc_map.pm_map = sc->sc_map_pins;
       if (gpio_pin_map(sc->sc_gpio, ga->ga_offset, ga->ga_mask,
               &sc->sc_map)) {
               aprint_error(": couldn't map the pins\n");
               return;
       }

       /* configure left pin */
       caps = gpio_pin_caps(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_LEFT);
       if (!(caps & GPIO_PIN_INPUT)) {
               aprint_error(": pin is unable to read input\n");
               gpio_pin_unmap(sc->sc_gpio, &sc->sc_map);
               return;
       }
       gpio_pin_ctl(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_LEFT,
           GPIO_PIN_INPUT);

       /* configure right pin */
       caps = gpio_pin_caps(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_RIGHT);
       if (!(caps & GPIO_PIN_INPUT)) {
               aprint_error(": pin is unable to read input\n");
               gpio_pin_unmap(sc->sc_gpio, &sc->sc_map);
               return;
       }
       gpio_pin_ctl(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_RIGHT,
           GPIO_PIN_INPUT);

       /* interrupt settings */
       sc->sc_intr[0] = intr_establish(GPIO1_BASE + ga->ga_offset,
           IPL_VM, IST_EDGE_BOTH, mousebtn_intr, sc);
       if (sc->sc_intr[0] == NULL) {
               aprint_error(": couldn't establish interrupt\n");
               return;
       }
       sc->sc_intr[1] = intr_establish(GPIO1_BASE + ga->ga_offset + 1,
            IPL_VM, IST_EDGE_BOTH, mousebtn_intr, sc);
       if (sc->sc_intr[1] == NULL) {
               aprint_error(": couldn't establish interrupt\n");
               intr_disestablish(sc->sc_intr[0]);
               return;
       }

       aprint_normal(": NetWalker mouse button\n");
       aprint_naive(": NetWalker mouse button\n");

       mutex_init(&sc->sc_lock, MUTEX_DEFAULT, IPL_NONE);
       callout_init(&sc->sc_c, 0);

       a.accessops = &mousebtn_accessops;
       a.accesscookie = sc;

       sc->sc_buttons = 0;
       sc->sc_wsmousedev = config_found(self, &a, wsmousedevprint, CFARGS_NONE);
}

static int
mousebtn_detach(device_t self, int flags)
{
       struct mousebtn_softc *sc = device_private(self);

       if (sc->sc_intr[0] != NULL)
               intr_disestablish(sc->sc_intr[0]);
       if (sc->sc_intr[1] != NULL)
               intr_disestablish(sc->sc_intr[1]);

       gpio_pin_unmap(sc->sc_gpio, &sc->sc_map);

       return 0;
}

static void
mousebtn_poll(void *arg)
{
       struct mousebtn_softc *sc = (struct mousebtn_softc *)arg;
       uint32_t buttons = 0;
       int s;
       int left;
       int right;

       mutex_enter(&sc->sc_lock);

       left = !gpio_pin_read(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_LEFT);
       right = !gpio_pin_read(sc->sc_gpio, &sc->sc_map, MOUSEBTN_PIN_RIGHT);
       buttons = (right << 2) | left;

       if (sc->sc_buttons != buttons) {
               s = spltty();
               wsmouse_input(sc->sc_wsmousedev, buttons, 0, 0, 0, 0,
                   WSMOUSE_INPUT_DELTA);
               sc->sc_buttons = buttons;
               splx(s);
       }

       mutex_exit(&sc->sc_lock);

#if defined(MOUSEBTN_POLLING)
       if (sc->sc_enabled)
               callout_reset(&sc->sc_c, POLLRATE, mousebtn_poll, sc);
#endif
       return;
}

static int
mousebtn_intr(void *v)
{
       struct mousebtn_softc *sc = v;

       if (sc->sc_enabled)
               callout_reset(&sc->sc_c, POLLRATE, mousebtn_poll, sc);

       return 1;
}


int
mousebtn_ioctl(void *v, u_long cmd, void *data, int flag, struct lwp *l)
{
       struct wsmouse_id *id;

       switch (cmd) {
       case WSMOUSEIO_GTYPE:
               *(u_int *)data = WSMOUSE_TYPE_PS2;
               return 0;
       case WSMOUSEIO_GETID:
               id = (struct wsmouse_id *)data;
               if (id->type != WSMOUSE_ID_TYPE_UIDSTR)
                       return EINVAL;
               strlcpy(id->data, "NetWalker Mouse Button", WSMOUSE_ID_MAXLEN);
               id->length = strlen(id->data);
               return 0;
       }

       return EPASSTHROUGH;
}

int
mousebtn_enable(void *v)
{
       struct mousebtn_softc *sc = (struct mousebtn_softc *)v;

       if (sc->sc_enabled)
               return EBUSY;

       if (!pmf_device_register(sc->sc_dev, mousebtn_suspend, mousebtn_resume))
               aprint_error_dev(sc->sc_dev,
                   "couldn't establish power handler\n");

       sc->sc_enabled = 1;
#if defined(MOUSEBTN_POLLING)
       callout_reset(&sc->sc_c, POLLRATE, mousebtn_poll, sc);
#endif

       return 0;
}

void
mousebtn_disable(void *v)
{
       struct mousebtn_softc *sc = (struct mousebtn_softc *)v;

       if (!sc->sc_enabled)
               return;

       pmf_device_deregister(sc->sc_dev);

       sc->sc_enabled = 0;

       return;
}

static bool
mousebtn_suspend(device_t dv, const pmf_qual_t *qual)
{
       struct mousebtn_softc *sc = device_private(dv);

#if defined(MOUSEBTN_POLLING)
       callout_stop(&sc->sc_c);
#endif
       sc->sc_enabled = 0;

       return true;
}

static bool
mousebtn_resume(device_t dv, const pmf_qual_t *qual)
{
       struct mousebtn_softc *sc = device_private(dv);

       sc->sc_enabled = 1;
#if defined(MOUSEBTN_POLLING)
       callout_reset(&sc->sc_c, POLLRATE, mousebtn_poll, sc);
#endif
       return true;
}