/*      $OpenBSD: bmc150.c,v 1.2 2022/04/06 18:59:28 naddy Exp $        */

/*
 * Copyright (c) 2017 Mark Kettenis
 *
 * 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 <dev/i2c/i2cvar.h>

#define BGW_CHIPID              0x00
#define  BGW_CHIPID_BMA222E     0xf8
#define  BGW_CHIPID_BMA250E     0xf9
#define  BGW_CHIPID_BMC150      0xfa
#define  BGW_CHIPID_BMI055      0xfa
#define  BGW_CHIPID_BMA255      0xfa
#define  BGW_CHIPID_BMA280      0xfb
#define ACCD_X_LSB              0x02
#define ACCD_X_MSB              0x03
#define ACCD_Y_LSB              0x04
#define ACCD_Y_MSB              0x05
#define ACCD_Z_LSB              0x06
#define ACCD_Z_MSB              0x07
#define ACCD_TEMP               0x08

#define BGW_NUM_SENSORS         4

#define BGW_SENSOR_XACCEL       0
#define BGW_SENSOR_YACCEL       1
#define BGW_SENSOR_ZACCEL       2
#define BGW_SENSOR_TEMP         3

struct bgw_softc {
        struct device sc_dev;
        i2c_tag_t sc_tag;
        i2c_addr_t sc_addr;

        uint8_t sc_temp0;

        struct ksensor sc_sensor[BGW_NUM_SENSORS];
        struct ksensordev sc_sensordev;
};

int     bgw_match(struct device *, void *, void *);
void    bgw_attach(struct device *, struct device *, void *);

const struct cfattach bgw_ca = {
        sizeof(struct bgw_softc), bgw_match, bgw_attach
};

struct cfdriver bgw_cd = {
        NULL, "bgw", DV_DULL
};

void    bgw_refresh(void *);

int
bgw_match(struct device *parent, void *match, void *aux)
{
        struct i2c_attach_args *ia = aux;

        if (strcmp(ia->ia_name, "BSBA0150") == 0 ||
            strcmp(ia->ia_name, "BMC150A") == 0 ||
            strcmp(ia->ia_name, "BMI055A") == 0 ||
            strcmp(ia->ia_name, "BMA0255") == 0 ||
            strcmp(ia->ia_name, "BMA250E") == 0 ||
            strcmp(ia->ia_name, "BMA222E") == 0 ||
            strcmp(ia->ia_name, "BMA0280") == 0 ||
            strcmp(ia->ia_name, "BOSC0200") == 0)
                return 1;
        return 0;
}

void
bgw_attach(struct device *parent, struct device *self, void *aux)
{
        struct bgw_softc *sc = (struct bgw_softc *)self;
        struct i2c_attach_args *ia = aux;
        uint8_t cmd, data;
        int i;

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

        iic_acquire_bus(sc->sc_tag, 0);
        cmd = BGW_CHIPID;
        if (iic_exec(sc->sc_tag, I2C_OP_READ_WITH_STOP,
            sc->sc_addr, &cmd, sizeof cmd, &data, sizeof data, 0)) {
                iic_release_bus(sc->sc_tag, 0);
                printf(": can't read chip ID\n");
                return;
        }
        iic_release_bus(sc->sc_tag, 0);

        switch (data) {
        case BGW_CHIPID_BMA222E:
                sc->sc_temp0 = 23;
                printf(": BMA222E");
                break;
        case BGW_CHIPID_BMA250E:
                sc->sc_temp0 = 23;
                printf(": BMA250E");
                break;
        case BGW_CHIPID_BMC150:
                sc->sc_temp0 = 24;
                printf(": BMC150");
                break;
        case BGW_CHIPID_BMA280:
                sc->sc_temp0 = 23;
                printf(": BMA280");
                break;
        }

        sc->sc_sensor[BGW_SENSOR_XACCEL].type = SENSOR_INTEGER;
        strlcpy(sc->sc_sensor[BGW_SENSOR_XACCEL].desc, "X_ACCEL",
            sizeof(sc->sc_sensor[BGW_SENSOR_XACCEL].desc));
        sc->sc_sensor[BGW_SENSOR_YACCEL].type = SENSOR_INTEGER;
        strlcpy(sc->sc_sensor[BGW_SENSOR_YACCEL].desc, "Y_ACCEL",
            sizeof(sc->sc_sensor[BGW_SENSOR_YACCEL].desc));
        sc->sc_sensor[BGW_SENSOR_ZACCEL].type = SENSOR_INTEGER;
        strlcpy(sc->sc_sensor[BGW_SENSOR_ZACCEL].desc, "Z_ACCEL",
            sizeof(sc->sc_sensor[BGW_SENSOR_ZACCEL].desc));
        sc->sc_sensor[BGW_SENSOR_TEMP].type = SENSOR_TEMP;

        strlcpy(sc->sc_sensordev.xname, sc->sc_dev.dv_xname,
            sizeof(sc->sc_sensordev.xname));
        for (i = 0; i < BGW_NUM_SENSORS; i++)
                sensor_attach(&sc->sc_sensordev, &sc->sc_sensor[i]);
        sensordev_install(&sc->sc_sensordev);

        if (sensor_task_register(sc, bgw_refresh, 5) == NULL) {
                printf(", unable to register update task\n");
                return;
        }

        printf("\n");
}

void
bgw_refresh(void *arg)
{
        struct bgw_softc *sc = arg;
        uint8_t cmd, data[7];
        uint8_t lsb;
        int8_t msb;
        int8_t temp;
        int i;

        iic_acquire_bus(sc->sc_tag, 0);
        cmd = ACCD_X_LSB;
        if (iic_exec(sc->sc_tag, I2C_OP_READ_WITH_STOP,
            sc->sc_addr, &cmd, sizeof cmd, &data, sizeof data, 0)) {
                iic_release_bus(sc->sc_tag, 0);
                for (i = 0; i < BGW_NUM_SENSORS; i++)
                        sc->sc_sensor[i].flags |= SENSOR_FINVALID;
                return;
        }
        iic_release_bus(sc->sc_tag, 0);

        lsb = data[ACCD_X_LSB - ACCD_X_LSB];
        msb = data[ACCD_X_MSB - ACCD_X_LSB];
        sc->sc_sensor[BGW_SENSOR_XACCEL].value = (msb << 4) | (lsb >> 4);
        sc->sc_sensor[BGW_SENSOR_XACCEL].flags &= ~SENSOR_FINVALID;

        lsb = data[ACCD_Y_LSB - ACCD_X_LSB];
        msb = data[ACCD_Y_MSB - ACCD_X_LSB];
        sc->sc_sensor[BGW_SENSOR_YACCEL].value = (msb << 4) | (lsb >> 4);
        sc->sc_sensor[BGW_SENSOR_YACCEL].flags &= ~SENSOR_FINVALID;

        lsb = data[ACCD_Z_LSB - ACCD_X_LSB];
        msb = data[ACCD_Z_MSB - ACCD_X_LSB];
        sc->sc_sensor[BGW_SENSOR_ZACCEL].value = (msb << 4) | (lsb >> 4);
        sc->sc_sensor[BGW_SENSOR_ZACCEL].flags &= ~SENSOR_FINVALID;

        temp = data[ACCD_TEMP - ACCD_X_LSB];
        sc->sc_sensor[BGW_SENSOR_TEMP].value =
            273150000 + (sc->sc_temp0 + temp) * 1000000;
        sc->sc_sensor[BGW_SENSOR_TEMP].flags &= ~SENSOR_FINVALID;
}