/*
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 *
 * HDF is dual licensed: you can use it either under the terms of
 * the GPL, or the BSD license, at your option.
 * See the LICENSE file in the root of this repository for complete details.
 */

#include "als_bh1745.h"
#include <securec.h>
#include "osal_mem.h"
#include "osal_time.h"
#include "sensor_als_driver.h"
#include "sensor_config_controller.h"
#include "sensor_device_manager.h"

#define HDF_LOG_TAG    khdf_sensor_als_driver

/* IO config for int-pin and I2C-pin */
#define SENSOR_I2C6_DATA_REG_ADDR 0x114f004c
#define SENSOR_I2C6_CLK_REG_ADDR  0x114f0048
#define SENSOR_I2C_REG_CFG        0x403
#define SENSOR_TIME_INCREASE      0
#define SENSOR_TIME_DECREASE      1
#define SENSOR_GAIN_INCREASE      0

static struct Bh1745DrvData *g_bh1745DrvData = NULL;
static uint32_t g_timeChangeStatus = SENSOR_TIME_DECREASE;

static struct TimeRegAddrValueMap g_timeMap[EXTENDED_ALS_TIME_GROUP_INDEX_MAX] = {
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_0,           BH1745_TIME_160MSEC },
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_1,           BH1745_TIME_320MSEC },
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_2,           BH1745_TIME_640MSEC },
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_3,           BH1745_TIME_1280MSEC },
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_4,           BH1745_TIME_2560MSEC },
    { EXTENDED_ALS_TIME_GROUP_ATTR_VALUE_5,           BH1745_TIME_5120MSEC }
};

static struct GainRegAddrValueMap g_gainMap[EXTENDED_ALS_GAIN_GROUP_INDEX_MAX] = {
    { EXTENDED_ALS_GAIN_GROUP_ATTR_VALUE_0,              BH1745_GAIN_1X },
    { EXTENDED_ALS_GAIN_GROUP_ATTR_VALUE_1,              BH1745_GAIN_2X },
    { EXTENDED_ALS_GAIN_GROUP_ATTR_VALUE_2,              BH1745_GAIN_16X }
};

static uint32_t g_red[BH1745_COEFFICIENT_RED] = {
    BH1745_COEFFICIENT_RED_LEVEL_0,
    BH1745_COEFFICIENT_RED_LEVEL_1
};

static uint32_t g_green[BH1745_COEFFICIENT_GREEN] = {
    BH1745_COEFFICIENT_GREEN_LEVEL_0,
    BH1745_COEFFICIENT_GREEN_LEVEL_1
};

static int32_t DynamicRangCovert(struct SensorCfgData *CfgData, uint32_t *rgbcData)
{
    uint8_t timeItemNum;
    uint8_t regValue;
    int32_t ret;
    uint32_t temp;
    struct SensorRegCfgGroupNode *timeGroupNode = NULL;
    int32_t index = EXTENDED_ALS_TIME_GROUP_INDEX_0;

    timeGroupNode = CfgData->extendedRegCfgGroup[EXTENDED_ALS_TIME_GROUP];
    timeItemNum = timeGroupNode->itemNum;
    if (timeItemNum > EXTENDED_ALS_TIME_GROUP_INDEX_MAX) {
        HDF_LOGE("%s: TimeItemNum out of range ", __func__);
        return HDF_FAILURE;
    }

    ret = ReadSensorRegCfgArray(&CfgData->busCfg, timeGroupNode, index, &regValue, sizeof(regValue));
    CHECK_SENSOR_REGIARRAY_RETURN_VALUE(ret, "read");

    regValue &= timeGroupNode->regCfgItem->mask;
    temp = GetTimeByRegValue(regValue, g_timeMap, timeItemNum);
    index = GetRegGroupIndexByTime(temp, g_timeMap, timeItemNum);
    if (index < 0) {
        HDF_LOGE("%s: Index out of range ", __func__);
        return HDF_FAILURE;
    }

    if (((rgbcData[ALS_R] * BH1745_MULTIPLE_100 > BH1745_TIME_MAX) ||
        (rgbcData[ALS_G] * BH1745_MULTIPLE_100 > BH1745_TIME_MAX)) && (temp >= BH1745_TIME_320MSEC)) {
        g_timeChangeStatus = SENSOR_TIME_DECREASE;
        index = GetRegGroupIndexByTime(temp, g_timeMap, timeItemNum);
        index--;

        ret = WriteSensorRegCfgArray(&CfgData->busCfg, timeGroupNode, index, sizeof(regValue));
        CHECK_SENSOR_REGIARRAY_RETURN_VALUE(ret, "write");
    } else if (((rgbcData[ALS_R] * BH1745_MULTIPLE_100 < BH1745_TIME_MIN) ||
        (rgbcData[ALS_G] * BH1745_MULTIPLE_100 < BH1745_TIME_MIN)) && (g_timeChangeStatus == SENSOR_TIME_DECREASE)) {
        g_timeChangeStatus = SENSOR_TIME_INCREASE;
        index = GetRegGroupIndexByTime(temp, g_timeMap, timeItemNum);
        index++;
        if (index >= timeItemNum) {
            HDF_LOGE("%s: Index out of range ", __func__);
            return HDF_FAILURE;
        }

        ret = WriteSensorRegCfgArray(&CfgData->busCfg, timeGroupNode, index, sizeof(regValue));
        CHECK_SENSOR_REGIARRAY_RETURN_VALUE(ret, "write");
    }

    return HDF_SUCCESS;
}

static int32_t CalLux(struct SensorCfgData *CfgData, struct AlsReportData *reportData, uint32_t *rgbcData)
{
    int32_t ret;
    uint32_t time;
    uint32_t gain;
    uint8_t regValue;
    uint32_t index = 1;
    uint32_t luxTemp;
    uint8_t itemNum;
    struct SensorRegCfgGroupNode *groupNode = NULL;
    int32_t timeIndex = EXTENDED_ALS_TIME_GROUP_INDEX_0;
    int32_t gainIndex = EXTENDED_ALS_GAIN_GROUP_INDEX_0;

    if (rgbcData[ALS_G] <= 0) {
        HDF_LOGE("%s: RgbcData out of range ", __func__);
        return HDF_FAILURE;
    }

    if (BH1745_MULTIPLE_100 * rgbcData[ALS_C] / rgbcData[ALS_G] < BH1745_COEFFICIENT_JUDGE) {
        index = 0;
    }

    luxTemp = g_red[index] * rgbcData[ALS_R] + g_green[index] * rgbcData[ALS_G];
    groupNode = CfgData->extendedRegCfgGroup[EXTENDED_ALS_TIME_GROUP];
    itemNum = groupNode->itemNum;
    if (itemNum > EXTENDED_ALS_TIME_GROUP_INDEX_MAX) {
        HDF_LOGE("%s: ItemNum out of range ", __func__);
        return HDF_FAILURE;
    }

    ret = ReadSensorRegCfgArray(&CfgData->busCfg, groupNode, timeIndex, &regValue, sizeof(regValue));
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Failed to read sensor register array ", __func__);
        return HDF_FAILURE;
    }
    regValue &= groupNode->regCfgItem->mask;
    time = GetTimeByRegValue(regValue, g_timeMap, itemNum);

    regValue = 0;
    groupNode = CfgData->extendedRegCfgGroup[EXTENDED_ALS_GAIN_GROUP];
    itemNum = groupNode->itemNum;

    ret = ReadSensorRegCfgArray(&CfgData->busCfg, groupNode, gainIndex, &regValue, sizeof(regValue));
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Failed to read sensor register array ", __func__);
        return HDF_FAILURE;
    }
    regValue &= groupNode->regCfgItem->mask;
    gain = GetGainByRegValue(regValue, g_gainMap, itemNum);
    if (gain == SENSOR_GAIN_INCREASE || time == SENSOR_TIME_INCREASE) {
        return HDF_FAILURE;
    } else {
        reportData->als = ((luxTemp / gain) * BH1745_GAIN_16X / time) * BH1745_TIME_160MSEC;
    }

    return HDF_SUCCESS;
}

static int32_t RawDataConvert(struct SensorCfgData *CfgData, struct AlsReportData *reportData, uint32_t *rgbcData)
{
    int ret;

    ret = CalLux(CfgData, reportData, rgbcData);
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Failed to calculate sensor brightness ", __func__);
        return HDF_FAILURE;
    }

    reportData->als = (reportData->als > 0) ? reportData->als : 0;

    ret = DynamicRangCovert(CfgData, rgbcData);
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "DynamicRangCovert");

    return HDF_SUCCESS;
}

static int32_t ReadBh1745RawData(struct SensorCfgData *data, struct AlsData *rawData, uint64_t *timestamp)
{
    uint8_t status = 0;
    uint8_t reg[ALS_LIGHT_BUTT];
    OsalTimespec time;

    (void)memset_s(&time, sizeof(time), 0, sizeof(time));
    (void)memset_s(reg, sizeof(reg), 0, sizeof(reg));

    CHECK_NULL_PTR_RETURN_VALUE(data, HDF_ERR_INVALID_PARAM);

    if (OsalGetTime(&time) != HDF_SUCCESS) {
        HDF_LOGE("%s: Get time failed", __func__);
        return HDF_FAILURE;
    }
    *timestamp = time.sec * SENSOR_SECOND_CONVERT_NANOSECOND + time.usec * SENSOR_CONVERT_UNIT; /* unit nanosecond */

    int32_t ret = ReadSensor(&data->busCfg, BH1745_MODECONTROL3_ADDR, &status, sizeof(uint8_t));
    if (!(status & BH1745_ALS_DATA_READY_MASK) || (ret != HDF_SUCCESS)) {
        HDF_LOGE("%s: data status [%u] ret [%d]", __func__, status, ret);
        return HDF_FAILURE;
    }

    ret = ReadSensor(&data->busCfg, BH1745_ALS_R_LSB_ADDR, &reg[ALS_R_LSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_R_MSB_ADDR, &reg[ALS_R_MSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_G_LSB_ADDR, &reg[ALS_G_LSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_G_MSB_ADDR, &reg[ALS_G_MSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_B_LSB_ADDR, &reg[ALS_B_LSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_B_MSB_ADDR, &reg[ALS_B_MSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_C_LSB_ADDR, &reg[ALS_C_LSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    ret = ReadSensor(&data->busCfg, BH1745_ALS_C_MSB_ADDR, &reg[ALS_C_MSB], sizeof(uint8_t));
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "read data");

    rawData->red = (uint16_t)(SENSOR_DATA_SHIFT_LEFT(reg[ALS_R_MSB], SENSOR_DATA_WIDTH_8_BIT) |
        reg[ALS_R_LSB]);
    rawData->green = (uint16_t)(SENSOR_DATA_SHIFT_LEFT(reg[ALS_G_MSB], SENSOR_DATA_WIDTH_8_BIT) |
        reg[ALS_G_LSB]);
    rawData->blue = (uint16_t)(SENSOR_DATA_SHIFT_LEFT(reg[ALS_B_MSB], SENSOR_DATA_WIDTH_8_BIT) |
        reg[ALS_B_LSB]);
    rawData->clear = (uint16_t)(SENSOR_DATA_SHIFT_LEFT(reg[ALS_C_MSB], SENSOR_DATA_WIDTH_8_BIT) |
        reg[ALS_C_LSB]);

    return HDF_SUCCESS;
}

static int32_t ReadBh1745Data(struct SensorCfgData *data, struct SensorReportEvent *event)
{
    int32_t ret;
    struct AlsData rawData = { 0, 0, 0, 0 };
    uint32_t tmp[ALS_LIGHT_NUM];
    static struct AlsReportData reportData;

    ret = ReadBh1745RawData(data, &rawData, &event->timestamp);
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: BH1745 read raw data failed", __func__);
        return HDF_FAILURE;
    }

    event->sensorId = SENSOR_TAG_AMBIENT_LIGHT;
    event->option = 0;
    event->mode = SENSOR_WORK_MODE_REALTIME;

    tmp[ALS_R] = rawData.red;
    tmp[ALS_G] = rawData.green;
    tmp[ALS_B] = rawData.blue;
    tmp[ALS_C] = rawData.clear;

    ret = RawDataConvert(data, &reportData, tmp);
    CHECK_PARSER_RESULT_RETURN_VALUE(ret, "RawDataConvert");

    event->dataLen = sizeof(reportData.als);
    event->data = (uint8_t *)&reportData.als;

    return ret;
}

static int32_t InitBh1745(struct SensorCfgData *data)
{
    int32_t ret;

    CHECK_NULL_PTR_RETURN_VALUE(data, HDF_ERR_INVALID_PARAM);
    ret = SetSensorRegCfgArray(&data->busCfg, data->regCfgGroup[SENSOR_INIT_GROUP]);
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: BH1745 sensor init config failed", __func__);
        return HDF_FAILURE;
    }

    return HDF_SUCCESS;
}

static int32_t InitAlsPreConfig(void)
{
    if (SetSensorPinMux(SENSOR_I2C6_DATA_REG_ADDR, SENSOR_ADDR_WIDTH_4_BYTE, SENSOR_I2C_REG_CFG) != HDF_SUCCESS) {
        HDF_LOGE("%s: Data write mux pin failed", __func__);
        return HDF_FAILURE;
    }
    if (SetSensorPinMux(SENSOR_I2C6_CLK_REG_ADDR, SENSOR_ADDR_WIDTH_4_BYTE, SENSOR_I2C_REG_CFG) != HDF_SUCCESS) {
        HDF_LOGE("%s: Clk write mux pin failed", __func__);
        return HDF_FAILURE;
    }

    return HDF_SUCCESS;
}

static int32_t DispatchBH1745(struct HdfDeviceIoClient *client,
    int cmd, struct HdfSBuf *data, struct HdfSBuf *reply)
{
    (void)client;
    (void)cmd;
    (void)data;
    (void)reply;

    return HDF_SUCCESS;
}

static int32_t Bh1745BindDriver(struct HdfDeviceObject *device)
{
    CHECK_NULL_PTR_RETURN_VALUE(device, HDF_ERR_INVALID_PARAM);

    struct Bh1745DrvData *drvData = (struct Bh1745DrvData *)OsalMemCalloc(sizeof(*drvData));
    if (drvData == NULL) {
        HDF_LOGE("%s: Malloc Bh1745 drv data fail", __func__);
        return HDF_ERR_MALLOC_FAIL;
    }

    drvData->ioService.Dispatch = DispatchBH1745;
    drvData->device = device;
    device->service = &drvData->ioService;
    g_bh1745DrvData = drvData;

    return HDF_SUCCESS;
}

static int32_t Bh1745InitDriver(struct HdfDeviceObject *device)
{
    int32_t ret;
    struct AlsOpsCall ops;

    CHECK_NULL_PTR_RETURN_VALUE(device, HDF_ERR_INVALID_PARAM);
    struct Bh1745DrvData *drvData = (struct Bh1745DrvData *)device->service;
    CHECK_NULL_PTR_RETURN_VALUE(drvData, HDF_ERR_INVALID_PARAM);

    ret = InitAlsPreConfig();
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Init  BH1745 bus mux config", __func__);
        return HDF_FAILURE;
    }

    drvData->sensorCfg = AlsCreateCfgData(device->property);
    if (drvData->sensorCfg == NULL || drvData->sensorCfg->root == NULL) {
        HDF_LOGD("%s: Creating alscfg failed because detection failed", __func__);
        return HDF_ERR_NOT_SUPPORT;
    }

    ops.Init = NULL;
    ops.ReadData = ReadBh1745Data;
    ret = AlsRegisterChipOps(&ops);
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Register BH1745 als failed", __func__);
        return HDF_FAILURE;
    }

    ret = InitBh1745(drvData->sensorCfg);
    if (ret != HDF_SUCCESS) {
        HDF_LOGE("%s: Init BH1745 als failed", __func__);
        return HDF_FAILURE;
    }

    return HDF_SUCCESS;
}

static void Bh1745ReleaseDriver(struct HdfDeviceObject *device)
{
    CHECK_NULL_PTR_RETURN(device);

    struct Bh1745DrvData *drvData = (struct Bh1745DrvData *)device->service;
    CHECK_NULL_PTR_RETURN(drvData);

    if (drvData->sensorCfg != NULL) {
        AlsReleaseCfgData(drvData->sensorCfg);
        drvData->sensorCfg = NULL;
    }
    OsalMemFree(drvData);
}

struct HdfDriverEntry g_alsBh1745DevEntry = {
    .moduleVersion = 1,
    .moduleName = "HDF_SENSOR_ALS_BH1745",
    .Bind = Bh1745BindDriver,
    .Init = Bh1745InitDriver,
    .Release = Bh1745ReleaseDriver,
};

HDF_INIT(g_alsBh1745DevEntry);