/*      $NetBSD: mcp23xxxgpio_i2c.c,v 1.2 2022/01/17 19:36:54 thorpej Exp $     */

/*-
 * Copyright (c) 2022 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Jason R. Thorpe.
 *
 * 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 NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``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 FOUNDATION OR CONTRIBUTORS
 * 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: mcp23xxxgpio_i2c.c,v 1.2 2022/01/17 19:36:54 thorpej Exp $");

/* 
 * Driver for Microchip serial I/O expanders:
 *
 *      MCP23008        8-bit, I2C interface
 *      MCP23017        16-bit, I2C interface
 *      MCP23018        16-bit (open-drain outputs), I2C interface
 *
 * Data sheet:
 *
 *      https://ww1.microchip.com/downloads/en/DeviceDoc/20001952C.pdf
 */

#include <sys/types.h>
#include <sys/device.h>
#include <sys/kernel.h>

#include <dev/ic/mcp23xxxgpioreg.h>
#include <dev/ic/mcp23xxxgpiovar.h>

#include <dev/i2c/i2cvar.h>

struct mcpgpio_i2c_softc {
        struct mcpgpio_softc    sc_mcpgpio;

        i2c_tag_t               sc_i2c;
        i2c_addr_t              sc_addr;
};

#define MCPGPIO_TO_I2C(sc)                                      \
        container_of((sc), struct mcpgpio_i2c_softc, sc_mcpgpio)

static const struct mcpgpio_variant mcp23008 = {
        .name = "MCP23008",
        .type = MCPGPIO_TYPE_23x08,
};

static const struct mcpgpio_variant mcp23017 = {
        .name = "MCP23017",
        .type = MCPGPIO_TYPE_23x17,
};

static const struct mcpgpio_variant mcp23018 = {
        .name = "MCP23018",
        .type = MCPGPIO_TYPE_23x18,
};

static const struct device_compatible_entry compat_data[] = {
        { .compat = "microchip,mcp23008",       .data = &mcp23008 },
        { .compat = "microchip,mcp23017",       .data = &mcp23017 },
        { .compat = "microchip,mcp23018",       .data = &mcp23018 },
        DEVICE_COMPAT_EOL
};

static int
mcpgpio_i2c_lock(struct mcpgpio_softc *sc)
{
        struct mcpgpio_i2c_softc *isc = MCPGPIO_TO_I2C(sc);

        return iic_acquire_bus(isc->sc_i2c, 0);
}

static void
mcpgpio_i2c_unlock(struct mcpgpio_softc *sc)
{
        struct mcpgpio_i2c_softc *isc = MCPGPIO_TO_I2C(sc);

        iic_release_bus(isc->sc_i2c, 0);
}

static int
mcpgpio_i2c_read(struct mcpgpio_softc *sc, unsigned int bank,
    uint8_t reg, uint8_t *valp)
{
        struct mcpgpio_i2c_softc *isc = MCPGPIO_TO_I2C(sc);

        /* Only one chip per hardware address in i2c. */
        KASSERT((bank & ~1U) == 0);

        return iic_exec(isc->sc_i2c, I2C_OP_READ_WITH_STOP, isc->sc_addr,
            &reg, 1, valp, 1, 0);
}

static int
mcpgpio_i2c_write(struct mcpgpio_softc *sc, unsigned int bank,
    uint8_t reg, uint8_t val)
{
        struct mcpgpio_i2c_softc *isc = MCPGPIO_TO_I2C(sc);

        /* Only one chip per hardware address in i2c. */
        KASSERT((bank & ~1U) == 0);

        return iic_exec(isc->sc_i2c, I2C_OP_WRITE_WITH_STOP, isc->sc_addr,
            &reg, 1, &val, 1, 0);
}

static const struct mcpgpio_accessops mcpgpio_i2c_accessops = {
        .lock   =       mcpgpio_i2c_lock,
        .unlock =       mcpgpio_i2c_unlock,
        .read   =       mcpgpio_i2c_read,
        .write  =       mcpgpio_i2c_write,
};

static int
mcpgpio_i2c_match(device_t parent, cfdata_t match, void *aux)
{
        struct i2c_attach_args *ia = aux;
        int match_result;

        if (iic_use_direct_match(ia, match, compat_data, &match_result)) {
                return match_result;
        }
        return 0;
}

static void
mcpgpio_i2c_attach(device_t parent, device_t self, void *aux)
{
        struct mcpgpio_i2c_softc *isc = device_private(self);
        struct mcpgpio_softc *sc = &isc->sc_mcpgpio;
        struct i2c_attach_args *ia = aux;
        const struct device_compatible_entry *dce;

        isc->sc_i2c = ia->ia_tag;
        isc->sc_addr = ia->ia_addr;

        dce = iic_compatible_lookup(ia, compat_data);
        KASSERT(dce != NULL);

        sc->sc_dev = self;
        sc->sc_variant = dce->data;
        sc->sc_iocon = IOCON_SEQOP;
        sc->sc_accessops = &mcpgpio_i2c_accessops;

        aprint_naive("\n");     
        aprint_normal(": %s I/O Expander\n", sc->sc_variant->name);

        mcpgpio_attach(sc);
}

CFATTACH_DECL_NEW(mcpgpio_i2c, sizeof(struct mcpgpio_i2c_softc),
    mcpgpio_i2c_match, mcpgpio_i2c_attach, NULL, NULL);