/*      $NetBSD: nbpiic.c,v 1.7 2025/09/15 13:23:01 thorpej Exp $ */
/*
 * Copyright (c) 2011 KIYOHARA Takashi
 * All rights reserved.
 *
 * 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 THE AUTHOR ``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 THE AUTHOR 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: nbpiic.c,v 1.7 2025/09/15 13:23:01 thorpej Exp $");

#include <sys/param.h>
#include <sys/device.h>
#include <sys/mutex.h>

#include <machine/platid.h>
#include <machine/platid_mask.h>

#include <arm/xscale/pxa2x0var.h>
#include <arm/xscale/pxa2x0_i2c.h>

#include <hpcarm/dev/nbppconvar.h>

#include <dev/i2c/i2cvar.h>

#include "nbppcon.h"

#define NBPIIC_SLAVE_ADDR       0x70

struct nbpiic_softc {
        struct pxa2x0_i2c_softc sc_pxa_i2c;

        struct i2c_controller sc_i2c;
        kmutex_t sc_lock;

        device_t sc_pcon;
        char sc_buf[16];
        int sc_idx;
};


static int pxaiic_match(device_t, cfdata_t, void *);
static void pxaiic_attach(device_t, device_t, void *);

static int nbpiic_intr(void *);
int nbpiic_poll(void *, int, char *);

/* functions for i2c_controller */
static int nbpiic_exec(void *cookie, i2c_op_t, i2c_addr_t, const void *, size_t,
    void *, size_t, int);

CFATTACH_DECL_NEW(pxaiic, sizeof(struct nbpiic_softc),
    pxaiic_match, pxaiic_attach, NULL, NULL);


/* ARGSUSED */
static int
pxaiic_match(device_t parent, cfdata_t match, void *aux)
{
        struct pxaip_attach_args *pxa = aux;

        if (strcmp(pxa->pxa_name, match->cf_name) != 0 ||
            !platid_match(&platid, &platid_mask_MACH_PSIONTEKLOGIX_NETBOOK_PRO))
                return 0;

        pxa->pxa_size = PXA2X0_I2C_SIZE;
        return 1;
}

/* ARGSUSED */
static void
pxaiic_attach(device_t parent, device_t self, void *aux)
{
        struct nbpiic_softc *sc = device_private(self);
        struct pxaip_attach_args *pxa = aux;
        void *ih;

        aprint_normal("\n");
        aprint_naive("\n");

        sc->sc_pxa_i2c.sc_dev = self;
        sc->sc_pxa_i2c.sc_iot = pxa->pxa_iot;
        sc->sc_pxa_i2c.sc_addr = pxa->pxa_addr;
        sc->sc_pxa_i2c.sc_size = pxa->pxa_size;
        sc->sc_pxa_i2c.sc_isar = NBPIIC_SLAVE_ADDR;
        sc->sc_pxa_i2c.sc_stat = PI2C_STAT_INIT;
        if (pxa2x0_i2c_attach_sub(&sc->sc_pxa_i2c)) {
                aprint_error_dev(self, "unable to attach PXA I2C\n");
                return;
        }

        /*
         * Initialize mutex with IPL_HIGH.  Keyboard was connected to us.
         * This is orthogonal to the lock held at the i2c layer; this
         * is just to interlock us with the keyboard interrupt.
         */
        mutex_init(&sc->sc_lock, MUTEX_DEFAULT, IPL_HIGH);

        ih = pxa2x0_intr_establish(pxa->pxa_intr, IPL_HIGH, nbpiic_intr, sc);
        if (ih == NULL) {
                aprint_error_dev(self, "intr_establish failed\n");
                return;
        }

        /* Initialize i2c_controller  */
        iic_tag_init(&sc->sc_i2c);
        sc->sc_i2c.ic_cookie = sc;
        sc->sc_i2c.ic_exec = nbpiic_exec;

        pxa2x0_i2c_open(&sc->sc_pxa_i2c);

        iicbus_attach(self, &sc->sc_i2c);

        sc->sc_pcon = device_find_by_xname("nbppcon0");

        /* Don't close I2C-bus.  We are slave device. */
}

static int
nbpiic_intr(void *arg)
{
        struct nbpiic_softc *sc = arg;
        struct pxa2x0_i2c_softc *pi2c = &sc->sc_pxa_i2c;
        int handled, len;

        handled = pxa2x0_i2c_intr_sub(pi2c, &len, &sc->sc_buf[sc->sc_idx]);
        if (len != 0)
                sc->sc_idx += len;
        if (pi2c->sc_stat == PI2C_STAT_STOP) {
#if NNBPPCON > 0
                nbppcon_intr(sc->sc_pcon, sc->sc_idx, sc->sc_buf);
#endif
                sc->sc_idx = 0;
                pi2c->sc_stat = PI2C_STAT_INIT;
        }

        return handled;
}

int
nbpiic_poll(void *cookie, int buflen, char *buf)
{
        struct nbpiic_softc *sc = cookie;
        int rv;

        if (sc->sc_idx > 0) {
                if (buflen < sc->sc_idx) {
                        (void) pxa2x0_i2c_poll(&sc->sc_pxa_i2c,
                            sizeof(sc->sc_buf) - sc->sc_idx, buf + sc->sc_idx,
                            I2C_F_READ);
                        sc->sc_idx = 0;
                } else
                        memcpy(buf, sc->sc_buf, sc->sc_idx);
        }
        rv = pxa2x0_i2c_poll(&sc->sc_pxa_i2c,
            buflen - sc->sc_idx, buf + sc->sc_idx, I2C_F_READ);
        sc->sc_idx = 0;

        return rv;
}

static int
nbpiic_exec(void *cookie, i2c_op_t op, i2c_addr_t addr, const void *vcmd,
            size_t cmdlen, void *vbuf, size_t buflen, int flags)
{
        struct nbpiic_softc *sc = cookie;
        int rv = -1;
        const u_char cmd = *(const u_char *)vcmd; 

        mutex_enter(&sc->sc_lock);

        if (I2C_OP_READ_P(op) && (cmdlen == 0) && (buflen == 1))
                rv = pxa2x0_i2c_read(&sc->sc_pxa_i2c, addr, (u_char *)vbuf);

        if ((I2C_OP_READ_P(op)) && (cmdlen == 1) && (buflen == 1)) {
                rv = pxa2x0_i2c_write(&sc->sc_pxa_i2c, addr, cmd);
                if (rv == 0)
                        rv = pxa2x0_i2c_read(&sc->sc_pxa_i2c,
                            addr, (u_char *)vbuf);
        }

        if ((I2C_OP_READ_P(op)) && (cmdlen == 1) && (buflen == 2)) {
printf("%s: read cmdlen=1, buflen=2: Ooops, maybe error...\n", __func__);
                rv = pxa2x0_i2c_write(&sc->sc_pxa_i2c, addr, cmd);
                if (rv == 0)
                        rv = pxa2x0_i2c_read(&sc->sc_pxa_i2c,
                            addr, &((u_char *)vbuf)[0]);
                if (rv == 0)
                        rv = pxa2x0_i2c_read(&sc->sc_pxa_i2c,
                            addr, &((u_char *)vbuf)[1]);
        }

        if ((I2C_OP_WRITE_P(op)) && (cmdlen == 0) && (buflen == 1))
                rv = pxa2x0_i2c_write(&sc->sc_pxa_i2c, addr, *(u_char *)vbuf);

        if ((I2C_OP_WRITE_P(op)) && (cmdlen == 1) && (buflen == 1)) {
                u_short v = (cmd << 8) | ((u_char *)vbuf)[0];

                rv = pxa2x0_i2c_write_2(&sc->sc_pxa_i2c, addr, v);
        }

        if ((I2C_OP_WRITE_P(op)) && (cmdlen == 1) && (buflen == 2)) {
printf("%s: write cmdlen=1, buflen=2: Ooops, maybe error...\n", __func__);
                rv = pxa2x0_i2c_write(&sc->sc_pxa_i2c, addr, cmd);
                if (rv == 0) {
                        u_short v =
                            (((u_char *)vbuf)[0] << 8) | ((u_char *)vbuf)[1];

                        rv = pxa2x0_i2c_write_2(&sc->sc_pxa_i2c, addr, v);
                }
        }

        /* Handle quick_read/quick_write ops - XXX Untested XXX */
        if ((cmdlen == 0) && (buflen == 0))
                rv = pxa2x0_i2c_quick(&sc->sc_pxa_i2c, addr,
                        I2C_OP_READ_P(op) ? 1 : 0);

        mutex_exit(&sc->sc_lock);

        return rv;
}