/*
 * Copyright (c) 2021 Nordic Semiconductor ASA
 * SPDX-License-Identifier: Apache-2.0
 */

#define DT_DRV_COMPAT bosch_bmx055

#include <zephyr/device.h>
#include <zephyr/drivers/gpio.h>
#include <zephyr/drivers/i2c.h>
#include <zephyr/drivers/sensor.h>

#include <zephyr/logging/log.h>
LOG_MODULE_REGISTER(bmx055, CONFIG_SENSOR_LOG_LEVEL);

// Accelerometer registers
const uint8_t REG_BWG_CHIPID = 0x00;
const uint8_t REG_ACCD_X_LSB = 0x02;
const uint8_t REG_ACCD_X_MSB = 0x03;
const uint8_t REG_ACCD_Y_LSB = 0x04;
const uint8_t REG_ACCD_Y_MSB = 0x05;
const uint8_t REG_ACCD_Z_LSB = 0x06;
const uint8_t REG_ACCD_Z_MSB = 0x07;
const uint8_t REG_ACCD_TEMP = 0x08;
const uint8_t REG_INT_STATUS_0 = 0x09;
const uint8_t REG_INT_STATUS_1 = 0x0A;
const uint8_t REG_INT_STATUS_2 = 0x0B;
const uint8_t REG_INT_STATUS_3 = 0x0C;
const uint8_t REG_FIFO_STATUS = 0x0E;
const uint8_t REG_PMU_RANGE = 0x0F;
const uint8_t REG_PMU_BW = 0x10;
const uint8_t REG_PMU_LPW = 0x11;
const uint8_t REG_PMU_LOW_POWER = 0x12;
const uint8_t REG_ACCD_HBW = 0x13;
const uint8_t REG_BGW_SOFTRESET = 0x14;
const uint8_t REG_INT_EN_0 = 0x16;
const uint8_t REG_INT_EN_1 = 0x17;
const uint8_t REG_INT_EN_2 = 0x18;
const uint8_t REG_INT_MAP_0 = 0x19;
const uint8_t REG_INT_MAP_1 = 0x1A;
const uint8_t REG_INT_MAP_2 = 0x1B;
const uint8_t REG_INT_SRC = 0x1E;
const uint8_t REG_INT_OUT_CTRL = 0x20;
const uint8_t REG_INT_RST_LATCH = 0x21;
const uint8_t REG_INT_0 = 0x22;
const uint8_t REG_INT_1 = 0x23;
const uint8_t REG_INT_2 = 0x24;
const uint8_t REG_INT_3 = 0x25;
const uint8_t REG_INT_4 = 0x26;
const uint8_t REG_INT_5 = 0x27;
const uint8_t REG_INT_6 = 0x28;
const uint8_t REG_INT_7 = 0x29;
const uint8_t REG_INT_8 = 0x2A;
const uint8_t REG_INT_9 = 0x2B;
const uint8_t REG_INT_A = 0x2C;
const uint8_t REG_INT_B = 0x2D;
const uint8_t REG_INT_C = 0x2E;
const uint8_t REG_INT_D = 0x2F;
const uint8_t REG_FIFO_CONFIG_0 = 0x30;
const uint8_t REG_PMU_SELF_TEST = 0x32;
const uint8_t REG_TRIM_NVM_CTRL = 0x33;
const uint8_t REG_BGW_SPI3_WDT = 0x34;
const uint8_t REG_OFC_CTRL = 0x36;
const uint8_t REG_OFC_SETTING = 0x37;
const uint8_t REG_OFC_OFFSET_X = 0x38;
const uint8_t REG_OFC_OFFSET_Y = 0x39;
const uint8_t REG_OFC_OFFSET_Z = 0x3A;
const uint8_t REG_TRIM_GP0 = 0x3B;
const uint8_t REG_TRIM_GP1 = 0x3C;
const uint8_t REG_FIFO_CONFIG_1 = 0x3E;
const uint8_t REG_FIFO_DATA = 0x3F;

struct bmx055_data
{
    int16_t accel_x;
    int16_t accel_y;
    int16_t accel_z;
    int16_t gyro_x;
    int16_t gyro_y;
    int16_t gyro_z;
    int8_t temperature;
};

struct bmx055_config
{
    struct i2c_dt_spec bus;
#ifdef CONFIG_INA230_TRIGGER
    bool trig_enabled;
    uint16_t mask;
    const struct gpio_dt_spec alert_gpio;
    uint16_t alert_limit;
#endif /* CONFIG_INA230_TRIGGER */
};

static int bmx055_sample_fetch(const struct device *dev,
                               enum sensor_channel chan)
{
    const struct bmx055_config *config = dev->config;
    struct bmx055_data *data = dev->data;
    int ret;

    ret = i2c_burst_read_dt(&config->bus, REG_ACCD_TEMP, &data->temperature, sizeof(data->temperature));
    if (ret < 0)
    {
        LOG_ERR("Failed to read temperature register!");
        return ret;
    }

    return 0;
}

static int bmx055_channel_get(const struct device *dev,
                              enum sensor_channel chan,
                              struct sensor_value *val)
{
    struct bmx055_data *data = dev->data;

    switch (chan)
    {
    case SENSOR_CHAN_DIE_TEMP:
        // 0.5K/LSB, center temperature is 23C
        val->val1 = 23 + data->temperature / 2;
        val->val2 = 0; // TODO: don't throw out LSB
        break;
    default:
        return -ENOTSUP;
    }

    return 0;
}

static const struct sensor_driver_api bmx055_api = {
    .sample_fetch = &bmx055_sample_fetch,
    .channel_get = &bmx055_channel_get,
};

static int bmx055_init(const struct device *dev)
{
    const struct bmx055_config *const config = dev->config;
    int ret;

    if (!device_is_ready(config->bus.bus))
    {
        LOG_ERR("I2C bus %s is not ready", config->bus.bus->name);
        return -ENODEV;
    }

    uint8_t chip_id;
    ret = i2c_burst_read_dt(&config->bus, REG_BWG_CHIPID, &chip_id, sizeof(chip_id));
    if (ret < 0)
    {
        LOG_ERR("Failed to read chip ID register!");
        return ret;
    }
    const uint8_t CHIP_ID = 0xfa;
    if (chip_id != CHIP_ID)
    {
        LOG_ERR("Chip ID read from %s incorrect. Read 0x%02X, expected 0x%02X", dev->name, chip_id, CHIP_ID);
    }

    //     ret = ina23x_reg_write(&config->bus, INA230_REG_CONFIG, config->config);
    //     if (ret < 0)
    //     {
    //         LOG_ERR("Failed to write configuration register!");
    //         return ret;
    //     }

    //     ret = ina230_calibrate(dev);
    //     if (ret < 0)
    //     {
    //         LOG_ERR("Failed to write calibration register!");
    //         return ret;
    //     }

    // #ifdef CONFIG_INA230_TRIGGER
    //     if (config->trig_enabled)
    //     {
    //         ret = ina230_trigger_mode_init(dev);
    //         if (ret < 0)
    //         {
    //             LOG_ERR("Failed to init trigger mode\n");
    //             return ret;
    //         }

    //         ret = ina23x_reg_write(&config->bus, INA230_REG_ALERT, config->alert_limit);
    //         if (ret < 0)
    //         {
    //             LOG_ERR("Failed to write alert register!");
    //             return ret;
    //         }

    //         ret = ina23x_reg_write(&config->bus, INA230_REG_MASK, config->mask);
    //         if (ret < 0)
    //         {
    //             LOG_ERR("Failed to write mask register!");
    //             return ret;
    //         }
    //     }
    // #endif /* CONFIG_INA230_TRIGGER */

    return 0;
}

#define BMX055_INIT(i)                                            \
    static struct bmx055_data bmx055_data_##i;                    \
                                                                  \
    static const struct bmx055_config bmx055_config_##i = {       \
        .bus = I2C_DT_SPEC_INST_GET(i),                           \
    };                                                            \
                                                                  \
    SENSOR_DEVICE_DT_INST_DEFINE(i, &bmx055_init, NULL,           \
                                 &bmx055_data_##i,                \
                                 &bmx055_config_##i, POST_KERNEL, \
                                 CONFIG_SENSOR_INIT_PRIORITY, &bmx055_api);

DT_INST_FOREACH_STATUS_OKAY(BMX055_INIT)