/* $NetBSD: pcf8574.c,v 1.10 2021/06/21 03:12:54 christos Exp $ */

/*-
 * Copyright (c) 2020 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Julian Coleman.
 *
 * 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.
 */

/*
 * A driver for Philips Semiconductor (NXP) PCF8574/PCF857A GPIO's.
 * Uses device properties to connect pins to the appropriate subsystem.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: pcf8574.c,v 1.10 2021/06/21 03:12:54 christos Exp $");

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/device.h>
#include <sys/kernel.h>

#include <dev/sysmon/sysmonvar.h>
#include <dev/sysmon/sysmon_taskq.h>

#include <dev/i2c/i2cvar.h>
#include <dev/led.h>

#ifdef PCF8574_DEBUG
#define DPRINTF printf
#else
#define DPRINTF if (0) printf
#endif

struct pcf8574_led {
	void *cookie;
	struct led_device *led;
	uint8_t mask, v_on, v_off;
};

struct pcf8574_pin {
	int pin_sensor;
	int pin_active;
	char pin_desc[ENVSYS_DESCLEN];
};

#define PCF8574_NPINS	8
struct pcf8574_softc {
	device_t	sc_dev;
	i2c_tag_t	sc_tag;
	i2c_addr_t	sc_addr;
	uint8_t		sc_state;
	uint8_t		sc_mask;

	uint8_t		sc_alert_mask;
#define	PCF8574_DEFAULT_TIMER	60
	int		sc_callout_time;
	callout_t	sc_timer;

	int		sc_nleds;
	struct pcf8574_led sc_leds[PCF8574_NPINS];
	struct pcf8574_pin sc_pins[PCF8574_NPINS];

	struct sysmon_envsys *sc_sme;
	envsys_data_t	sc_sensor[PCF8574_NPINS];
};

static int	pcf8574_match(device_t, cfdata_t, void *);
static void	pcf8574_attach(device_t, device_t, void *);
static int	pcf8574_detach(device_t, int);

static int	pcf8574_read(struct pcf8574_softc *sc, uint8_t *);
static int	pcf8574_write(struct pcf8574_softc *sc, uint8_t);
static void	pcf8574_attach_led(
			struct pcf8574_softc *, char *, int, int, int);
static int	pcf8574_attach_sysmon(
			struct pcf8574_softc *, char *, int, int, int);
void		pcf8574_refresh(struct sysmon_envsys *, envsys_data_t *);
int		pcf8574_get_led(void *);
void		pcf8574_set_led(void *, int);
static void	pcf8574_timeout(void *);
static void	pcf8574_check(void *);
static void	pcf8574_check_alert(struct pcf8574_softc *, uint8_t, uint8_t);

CFATTACH_DECL_NEW(pcf8574io, sizeof(struct pcf8574_softc),
	pcf8574_match, pcf8574_attach, pcf8574_detach, NULL);

static const struct device_compatible_entry compat_data[] = {
	{ .compat = "i2c-pcf8574" },
	DEVICE_COMPAT_EOL
};

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

	if (iic_use_direct_match(ia, cf, compat_data, &match_result))
		return match_result;

	/* We don't support indirect matches */
	return 0;
}

static void
pcf8574_attach(device_t parent, device_t self, void *aux)
{
	struct pcf8574_softc *sc = device_private(self);
	struct i2c_attach_args *ia = aux;
	prop_dictionary_t dict = device_properties(self);
	prop_array_t pins;
	prop_dictionary_t pin;
	int i, num, def, envc = 0;
	char name[32];
	const char *nptr = NULL, *spptr;
	bool ok = TRUE, act;

	sc->sc_tag = ia->ia_tag;
	sc->sc_addr = ia->ia_addr;
	sc->sc_dev = self;

	sc->sc_sme = NULL;
#ifdef PCF8574_DEBUG
	sc->sc_callout_time = 60;	/* watch for changes when debugging */
#else
	sc->sc_callout_time = 0;
#endif

	/*
	 * The PCF8574 requires input pins to be written with the value 1,
	 * and then read.  Assume that all pins are input initially.
	 * We'll use the mask when we write, because we have to write a
	 * value for every pin, and we don't want to change input pins.
	 */
	sc->sc_mask = 0xff;

	/* Try a read, and fail if this component isn't present */
	if (pcf8574_read(sc, &sc->sc_state)) {
		aprint_normal(": read failed\n");
		return;
	}

#ifdef PCF8574_DEBUG
	aprint_normal(": GPIO: state = 0x%02x\n", sc->sc_state);
#else
	aprint_normal(": GPIO\n");
#endif

	pins = prop_dictionary_get(dict, "pins");
	if (pins == NULL)
		return;

	for (i = 0; i < prop_array_count(pins); i++) {
		pin = prop_array_get(pins, i);
		ok &= prop_dictionary_get_string(pin, "name", &nptr);
		ok &= prop_dictionary_get_uint32(pin, "pin", &num);
		ok &= prop_dictionary_get_bool(pin, "active_high", &act);
		/* optional default state */
		def = -1;
		prop_dictionary_get_int32(pin, "default_state", &def);
		if (!ok)
			continue;
		/* Extract pin type from the name */
		spptr = strstr(nptr, " ");
		if (spptr == NULL)
			continue;
		spptr += 1;
		strncpy(name, spptr, 31);
		sc->sc_pins[i].pin_active = act;
		if (!strncmp(nptr, "LED ", 4)) {
			sc->sc_mask &= ~(1 << num);
			pcf8574_attach_led(sc, name, num, act, def);
		}
		if (!strncmp(nptr, "INDICATOR ", 10)) {
			if (pcf8574_attach_sysmon(sc, name, envc, num, act))
				return;
			envc++;
		}
		if (!strncmp(nptr, "ALERT ", 6)) {
			u_int64_t alert_time;

			if (pcf8574_attach_sysmon(sc, name, envc, num, act))
				return;

			/* Adjust our timeout if the alert times out. */
			if (def != -1)
				alert_time = def / 2;
			else
				alert_time = PCF8574_DEFAULT_TIMER;
			
			if (sc->sc_callout_time == 0 ||
			    alert_time < sc->sc_callout_time)
				sc->sc_callout_time = alert_time;

			sc->sc_alert_mask |= (1 << num);
			envc++;
		}
	}

	if (sc->sc_sme != NULL) {
		sc->sc_sme->sme_name = device_xname(self);
		sc->sc_sme->sme_cookie = sc;
		sc->sc_sme->sme_refresh = pcf8574_refresh;
		if (sysmon_envsys_register(sc->sc_sme)) {
			aprint_error_dev(self,
			    "unable to register with sysmon\n");
			sysmon_envsys_destroy(sc->sc_sme);
			sc->sc_sme = NULL;
			return;
		}
	}

	if (sc->sc_callout_time) {
		callout_init(&sc->sc_timer, CALLOUT_MPSAFE);
		callout_reset(&sc->sc_timer, hz * sc->sc_callout_time,
		    pcf8574_timeout, sc);
	}
}

static int
pcf8574_detach(device_t self, int flags)
{
	struct pcf8574_softc *sc = device_private(self);
	int i;

	if (sc->sc_callout_time) {
		callout_halt(&sc->sc_timer, NULL);
		callout_destroy(&sc->sc_timer);
		sc->sc_callout_time = 0;
	}

	if (sc->sc_sme != NULL) {
		sysmon_envsys_unregister(sc->sc_sme);
		sc->sc_sme = NULL;
	}

	for (i = 0; i < sc->sc_nleds; i++)
		led_detach(sc->sc_leds[i].led);

	return 0;
}

static int
pcf8574_read(struct pcf8574_softc *sc, uint8_t *val)
{
	int err = 0;

	if ((err = iic_acquire_bus(sc->sc_tag, 0)) != 0)
		return err;
	err = iic_exec(sc->sc_tag, I2C_OP_READ_WITH_STOP,
	    sc->sc_addr, NULL, 0, val, 1, 0);
	iic_release_bus(sc->sc_tag, 0);
	return err;
}

static int
pcf8574_write(struct pcf8574_softc *sc, uint8_t val)
{
	int err = 0;

	if ((err = iic_acquire_bus(sc->sc_tag, 0)) != 0)
		return err;
	err = iic_exec(sc->sc_tag, I2C_OP_WRITE_WITH_STOP, sc->sc_addr,
	    &val, 1, NULL, 0, 0);
	iic_release_bus(sc->sc_tag, 0);
	return err;
}

static void
pcf8574_attach_led(struct pcf8574_softc *sc, char *name, int pin, int act,
	int def)
{
	struct pcf8574_led *l;

	l = &sc->sc_leds[sc->sc_nleds];
	l->cookie = sc;
	l->mask = 1 << pin;
	l->v_on = act ? l->mask : 0;
	l->v_off = act ? 0 : l->mask;
	led_attach(name, l, pcf8574_get_led, pcf8574_set_led);
	if (def != -1)
		pcf8574_set_led(l, def);
	DPRINTF("%s: attached LED: %02x %02x %02x def %d\n",
	    device_xname(sc->sc_dev), l->mask, l->v_on, l->v_off, def);
	sc->sc_nleds++;
}

static int
pcf8574_attach_sysmon(struct pcf8574_softc *sc, char *name, int envc, int pin,
	int act)
{
	int ret;

	if (sc->sc_sme == NULL) {
		sc->sc_sme = sysmon_envsys_create();
		sc->sc_sme->sme_events_timeout = 0;
	}

	strlcpy(sc->sc_pins[pin].pin_desc, name,
	    sizeof(sc->sc_pins[pin].pin_desc));
	/* envsys sensor # to pin # mapping */
	sc->sc_pins[envc].pin_sensor = pin;
	sc->sc_sensor[envc].state = ENVSYS_SINVALID;
	sc->sc_sensor[envc].units = ENVSYS_INDICATOR;
	strlcpy(sc->sc_sensor[envc].desc, name,
	    sizeof(sc->sc_sensor[envc].desc));
	ret = sysmon_envsys_sensor_attach(sc->sc_sme, &sc->sc_sensor[envc]);
	if (ret) {
		sysmon_envsys_destroy(sc->sc_sme);
		sc->sc_sme = NULL;
		aprint_error_dev(sc->sc_dev,
		    "unable to attach pin %d at sysmon\n", pin);
		return ret;
	}
	DPRINTF("%s: added sysmon: pin %d sensor %d (%s)\n",
	    device_xname(sc->sc_dev), pin, envc, name);
	return 0;
}

void
pcf8574_refresh(struct sysmon_envsys *sme, envsys_data_t *edata)
{
	struct pcf8574_softc *sc = sme->sme_cookie;
	int pin = sc->sc_pins[edata->sensor].pin_sensor;
	int act = sc->sc_pins[pin].pin_active;
	u_int8_t prev_state = sc->sc_state;

	pcf8574_read(sc, &sc->sc_state);
	if (act)
		edata->value_cur = sc->sc_state & 1 << pin ? TRUE : FALSE;
	else
		edata->value_cur = sc->sc_state & 1 << pin ? FALSE : TRUE;
	edata->state = ENVSYS_SVALID;

	/* We read all the pins, so check for alerts on any pin now */
	if (sc->sc_state != prev_state) {
		DPRINTF("%s: (refresh) status change: 0x%02x > 0x%02x\n",
                    device_xname(sc->sc_dev), prev_state, sc->sc_state);
		pcf8574_check_alert(sc, prev_state, sc->sc_state);
	}
}

int
pcf8574_get_led(void *cookie)
{
	struct pcf8574_led *l = cookie;
	struct pcf8574_softc *sc = l->cookie;

	return ((sc->sc_state & l->mask) == l->v_on);
}

void
pcf8574_set_led(void *cookie, int val)
{
	struct pcf8574_led *l = cookie;
	struct pcf8574_softc *sc = l->cookie;
	uint32_t newstate;	

	newstate = sc->sc_state & ~l->mask;
	newstate |= val ? l->v_on : l->v_off;
	DPRINTF("%s: set LED: %02x -> %02x, %02x %02x %02x\n",
	    device_xname(sc->sc_dev), sc->sc_state, newstate, l->mask, l->v_on, l->v_off);
	if (newstate != sc->sc_state) {
		pcf8574_write(sc, newstate | sc->sc_mask);
		pcf8574_read(sc, &sc->sc_state);
	}
}

static void
pcf8574_timeout(void *v)
{
	struct pcf8574_softc *sc = v;

	sysmon_task_queue_sched(0, pcf8574_check, sc);
	callout_reset(&sc->sc_timer, hz * sc->sc_callout_time,
	    pcf8574_timeout, sc);
}

static void
pcf8574_check(void *v)
{
	struct pcf8574_softc *sc = v;
	uint8_t prev_state = sc->sc_state;

	pcf8574_read(sc, &sc->sc_state);
	if (sc->sc_state != prev_state) {
		DPRINTF("%s: (check) status change: 0x%02x > 0x%02x\n",
		    device_xname(sc->sc_dev), prev_state, sc->sc_state);
		pcf8574_check_alert(sc, prev_state, sc->sc_state);
	}
}


static void
pcf8574_check_alert(struct pcf8574_softc *sc, uint8_t prev_state,
	uint8_t curr_state)
{
	int i;
	uint8_t pin_chg;

	for (i = 0; i < PCF8574_NPINS; i++) {
		pin_chg = (sc->sc_state & 1 << i) ^ (prev_state & 1 << i);
		if (pin_chg & sc->sc_alert_mask) {
			if (sc->sc_pins[i].pin_active)
				printf("%s: Alert: %s = %s\n",
				    device_xname(sc->sc_dev),
				    sc->sc_pins[i].pin_desc,
				    sc->sc_state & 1 << i ?  "True" : "False");
			else
				printf("%s: Alert: %s = %s\n",
				    device_xname(sc->sc_dev),
				    sc->sc_pins[i].pin_desc,
				    sc->sc_state & 1 << i ?  "False" : "True");
		}
	}
}