/*	$OpenBSD: mvtemp.c,v 1.5 2023/04/18 08:35:02 patrick Exp $	*/
/*
 * Copyright (c) 2018 Mark Kettenis <kettenis@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

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

#include <machine/bus.h>
#include <machine/fdt.h>

#include <dev/ofw/openfirm.h>
#include <dev/ofw/ofw_misc.h>
#include <dev/ofw/fdt.h>

/* Registers */
#define TEMP_STAT			0x0000
#define TEMP_CTRL0			0x0000
#define  TEMP_CTRL0_TSEN_TC_TRIM_MASK	0x7
#define  TEMP_CTRL0_TSEN_TC_TRIM_VAL	0x3
#define  TEMP_CTRL0_TSEN_START		(1 << 0)
#define  TEMP_CTRL0_TSEN_RESET		(1 << 1)
#define  TEMP_CTRL0_TSEN_ENABLE		(1 << 2)
#define  TEMP_CTRL0_TSEN_AVG_BYPASS	(1 << 6)
#define  TEMP_CTRL0_TSEN_OSR_MAX	(0x3 << 24)
#define TEMP_CTRL1			0x0004
#define  TMEP_CTRL1_TSEN_START		(1 << 0)
#define  TEMP_CTRL1_TSEN_SW_RESET	(1 << 7)
#define  TEMP_CTRL1_TSEN_HW_RESETN	(1 << 8)
#define  TEMP_CTRL1_TSEN_AVG_MASK	0x7

#define REG_CTRL0			0
#define REG_CTRL1			1
#define REG_STAT			2
#define REG_MAX				3

struct mvtemp_softc {
	struct device		sc_dev;
	bus_space_tag_t		sc_iot;
	bus_space_handle_t	sc_stat_ioh;
	bus_space_handle_t	sc_ctrl_ioh;
	struct regmap		*sc_rm;

	uint32_t		sc_stat_valid;
	int32_t			(*sc_calc_temp)(uint32_t);
	bus_size_t		sc_offs[REG_MAX];

	struct ksensor		sc_sensor;
	struct ksensordev	sc_sensordev;
};

int	mvtemp_match(struct device *, void *, void *);
void	mvtemp_attach(struct device *, struct device *, void *);

const struct cfattach	mvtemp_ca = {
	sizeof (struct mvtemp_softc), mvtemp_match, mvtemp_attach
};

struct cfdriver mvtemp_cd = {
	NULL, "mvtemp", DV_DULL
};

struct mvtemp_compat {
	const char *compat;
	uint32_t stat_valid;
	void	(*init)(struct mvtemp_softc *);
	int32_t	(*calc_temp)(uint32_t);
	bus_size_t offs[REG_MAX];
};

void	mvtemp_380_init(struct mvtemp_softc *);
void	mvtemp_ap806_init(struct mvtemp_softc *);
int32_t mvtemp_ap806_calc_temp(uint32_t);
void	mvtemp_cp110_init(struct mvtemp_softc *);
int32_t mvtemp_cp110_calc_temp(uint32_t);

const struct mvtemp_compat mvtemp_compat[] = {
	{
		"marvell,armada-ap806-thermal", (1 << 16),
		mvtemp_ap806_init, mvtemp_ap806_calc_temp,
		{ 0x84, 0x88, 0x8c },
	},
	{
		"marvell,armada-cp110-thermal", (1 << 10),
		mvtemp_cp110_init, mvtemp_cp110_calc_temp,
		{ 0x70, 0x74, 0x78 },
	},
	{
		"marvell,armada380-thermal", (1 << 10),
		mvtemp_380_init, mvtemp_cp110_calc_temp,
		{ 0x70, 0x74, 0x78 },
	},
};

void	mvtemp_refresh_sensors(void *);

int
mvtemp_match(struct device *parent, void *match, void *aux)
{
	struct fdt_attach_args *faa = aux;
	int i;

	for (i = 0; i < nitems(mvtemp_compat); i++) {
		if (OF_is_compatible(faa->fa_node, mvtemp_compat[i].compat))
			return 1;
	}

	return 0;
}

void
mvtemp_attach(struct device *parent, struct device *self, void *aux)
{
	struct mvtemp_softc *sc = (struct mvtemp_softc *)self;
	struct fdt_attach_args *faa = aux;
	int i;

	if (faa->fa_nreg >= 2) {
		sc->sc_iot = faa->fa_iot;
		if (bus_space_map(sc->sc_iot, faa->fa_reg[0].addr,
		    faa->fa_reg[0].size, 0, &sc->sc_stat_ioh)) {
			printf(": can't map registers\n");
			return;
		}
		if (bus_space_map(sc->sc_iot, faa->fa_reg[1].addr,
		    faa->fa_reg[1].size, 0, &sc->sc_ctrl_ioh)) {
			bus_space_unmap(sc->sc_iot, sc->sc_stat_ioh,
			    faa->fa_reg[0].size);
			printf(": can't map registers\n");
			return;
		}
	} else {
		sc->sc_rm = regmap_bynode(OF_parent(faa->fa_node));
		if (sc->sc_rm == NULL) {
			printf(": no registers\n");
			return;
		}
	}

	printf("\n");

	for (i = 0; i < nitems(mvtemp_compat); i++) {
		if (OF_is_compatible(faa->fa_node, mvtemp_compat[i].compat)) {
			break;
		}
	}
	KASSERT(i < nitems(mvtemp_compat));

	/* Legacy offsets */
	sc->sc_offs[REG_CTRL0] = TEMP_CTRL0;
	sc->sc_offs[REG_CTRL1] = TEMP_CTRL1;
	sc->sc_offs[REG_STAT] = TEMP_STAT;

	/* Syscon offsets */
	if (sc->sc_rm != NULL) {
		sc->sc_offs[REG_CTRL0] = mvtemp_compat[i].offs[REG_CTRL0];
		sc->sc_offs[REG_CTRL1] = mvtemp_compat[i].offs[REG_CTRL1];
		sc->sc_offs[REG_STAT] = mvtemp_compat[i].offs[REG_STAT];
	}

	mvtemp_compat[i].init(sc);
	sc->sc_stat_valid = mvtemp_compat[i].stat_valid;
	sc->sc_calc_temp = mvtemp_compat[i].calc_temp;

	/* Register sensors. */
	strlcpy(sc->sc_sensordev.xname, sc->sc_dev.dv_xname,
	    sizeof(sc->sc_sensordev.xname));
	sc->sc_sensor.type = SENSOR_TEMP;
	sc->sc_sensor.flags = SENSOR_FINVALID;
	sensor_attach(&sc->sc_sensordev, &sc->sc_sensor);
	sensordev_install(&sc->sc_sensordev);
	sensor_task_register(sc, mvtemp_refresh_sensors, 5);
}

uint32_t
mvtemp_read(struct mvtemp_softc *sc, int reg)
{
	if (sc->sc_rm != NULL)
		return regmap_read_4(sc->sc_rm, sc->sc_offs[reg]);
	else if (reg == REG_STAT)
		return bus_space_read_4(sc->sc_iot, sc->sc_stat_ioh, sc->sc_offs[reg]);
	else
		return bus_space_read_4(sc->sc_iot, sc->sc_ctrl_ioh, sc->sc_offs[reg]);
}

void
mvtemp_write(struct mvtemp_softc *sc, int reg, uint32_t val)
{
	if (sc->sc_rm != NULL)
		regmap_write_4(sc->sc_rm, sc->sc_offs[reg], val);
	else if (reg == REG_STAT)
		bus_space_write_4(sc->sc_iot, sc->sc_stat_ioh, sc->sc_offs[reg], val);
	else
		bus_space_write_4(sc->sc_iot, sc->sc_ctrl_ioh, sc->sc_offs[reg], val);
}

/* 380 */

void
mvtemp_380_init(struct mvtemp_softc *sc)
{
	uint32_t ctrl;

	ctrl = mvtemp_read(sc, REG_CTRL1);
	ctrl |= TEMP_CTRL1_TSEN_HW_RESETN;
	ctrl &= ~TEMP_CTRL1_TSEN_SW_RESET;
	mvtemp_write(sc, REG_CTRL1, ctrl);

	ctrl = mvtemp_read(sc, REG_CTRL0);
	ctrl &= ~TEMP_CTRL0_TSEN_TC_TRIM_MASK;
	ctrl |= TEMP_CTRL0_TSEN_TC_TRIM_VAL;
	mvtemp_write(sc, REG_CTRL0, ctrl);
}

/* AP806 */

void
mvtemp_ap806_init(struct mvtemp_softc *sc)
{
	uint32_t ctrl;

	ctrl = mvtemp_read(sc, REG_CTRL0);
	ctrl &= ~TEMP_CTRL0_TSEN_RESET;
	ctrl |= TEMP_CTRL0_TSEN_START | TEMP_CTRL0_TSEN_ENABLE;
	ctrl |= TEMP_CTRL0_TSEN_OSR_MAX;
	ctrl &= ~TEMP_CTRL0_TSEN_AVG_BYPASS;
	mvtemp_write(sc, REG_CTRL0, ctrl);
}

int32_t
mvtemp_ap806_calc_temp(uint32_t stat)
{
	stat = ((stat & 0x3ff) ^ 0x200) - 0x200;
	return (stat * 423000) + 150000000 + 273150000;
}

/* CP110 */

void
mvtemp_cp110_init(struct mvtemp_softc *sc)
{
	uint32_t ctrl;

	mvtemp_380_init(sc);

	ctrl = mvtemp_read(sc, REG_CTRL0);
	ctrl |= TEMP_CTRL0_TSEN_OSR_MAX;
	mvtemp_write(sc, REG_CTRL0, ctrl);

	ctrl = mvtemp_read(sc, REG_CTRL1);
	ctrl |= TMEP_CTRL1_TSEN_START;
	ctrl &= ~TEMP_CTRL1_TSEN_AVG_MASK;
	mvtemp_write(sc, REG_CTRL1, ctrl);
}

int32_t
mvtemp_cp110_calc_temp(uint32_t stat)
{
	return ((stat & 0x3ff) * 476100) - 279100000 + 273150000;
}

void
mvtemp_refresh_sensors(void *arg)
{
	struct mvtemp_softc *sc = arg;
	int32_t stat, temp;

	stat = mvtemp_read(sc, REG_STAT);
	temp = sc->sc_calc_temp(stat);
	sc->sc_sensor.value = temp;
	if ((stat & sc->sc_stat_valid) && temp >= 0)
		sc->sc_sensor.flags &= ~SENSOR_FINVALID;
	else
		sc->sc_sensor.flags |= SENSOR_FINVALID;
}