vsensor/v_spo2.c

/*
 * Copyright (c) 2006-2020, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2021-04-01     stackRyan    add spo2 virtual sensor
 */

#include <rtthread.h>

#ifdef PKG_USING_VIRTUAL_SENSOR_SPO2

#include "sensor.h"
#include <stdlib.h>

#define DBG_TAG    "v_spo2"
#ifdef PKG_USING_VIRTUAL_SENSOR_DBG
    #define DBG_LVL    DBG_LOG
#else
    #define DBG_LVL    DBG_INFO
#endif
#include <rtdbg.h>

enum SENS_SPO2_ID
{
    SENS_SPO2_01 = 0, //SPO2
    SENS_SPO2_MAX,
};

#define SENS_BUS_NAME                   "sens_bus"
#define SENS_SPO2_01_SENSOR_ID            (RT_SENSOR_CLASS_SPO2 + 0x10)

struct sens_spo2
{
    char* dev_name;
    rt_uint8_t sens_id;
};

static struct sens_spo2 sens_spo2_tbl[SENS_SPO2_MAX] =
{
    {V_SENS_SPO2_DEV_NAME,    0x00 }, /* SPO2 */
};

static struct rt_sensor_info spo2_info_tbl[SENS_SPO2_MAX] =
{
    {RT_SENSOR_CLASS_SPO2,  RT_SENSOR_VENDOR_UNKNOWN,   RT_NULL,    RT_SENSOR_UNIT_PERMILLAGE,      RT_SENSOR_INTF_UART,     100,   0,   1 },
};

static rt_uint8_t sensor_get_id(rt_uint8_t sens_index)
{
    rt_uint8_t chip_id = 0x00;

    switch (sens_index)
    {
    case SENS_SPO2_01:
        chip_id = SENS_SPO2_01_SENSOR_ID;
        break;
    default:
        break;
    }

    return chip_id;
}

static int sensor_init(rt_uint8_t index)
{
    sens_spo2_tbl[index].sens_id = sensor_get_id(index);

    return RT_EOK;
}

static void* spo2_sensor_create(struct rt_sensor_intf* intf, rt_uint8_t index)
{
    if (sensor_init(index) != RT_EOK)
    {
        LOG_E("%s:error!", __func__);
    }

    return 0;
}

static rt_err_t spo2_sensor_set_odr(rt_sensor_t sensor, rt_uint16_t odr)
{
    LOG_D("%s:odr=%d", __func__, odr);
    return RT_EOK;
}

static rt_err_t spo2_sensor_set_range(rt_sensor_t sensor, rt_uint16_t range)
{
    LOG_D("%s:range=%d", __func__, range);
    return RT_EOK;
}

static rt_err_t spo2_sensor_set_power(rt_sensor_t sensor, rt_uint8_t power)
{
    rt_int8_t rslt = 0;
    LOG_D("%s:power=%d", __func__, power);
    return rslt;
}

static rt_size_t spo2_sensor_fetch_data(struct rt_sensor_device* sensor, void* buf, rt_size_t size)
{
    struct rt_sensor_data* data = buf;

    if (size < 1)
    {
        LOG_E("%s:read size err! size=%d", __func__, size);
        return 0;
    }

    if (buf == RT_NULL)
    {
        LOG_E("%s:read buf is NULL!", __func__);
        return 0;
    }

    for (int i = 0; i < size; i++)
    {
        data->type = RT_SENSOR_CLASS_SPO2;
        data->data.spo2 = 90 + rand() % 10;
        data->timestamp = rt_sensor_get_ts();
        LOG_D("%s:[%d]", __func__, data->data.spo2);
        data++;
    }

    return size;
}

static rt_err_t spo2_sensor_control(struct rt_sensor_device* sensor, int cmd, void* args)
{
    rt_err_t result = RT_EOK;

    switch (cmd)
    {
    case RT_SENSOR_CTRL_GET_ID:
        *(rt_uint8_t*)args = sens_spo2_tbl[SENS_SPO2_01].sens_id;
        break;
    case RT_SENSOR_CTRL_SET_ODR:
        result = spo2_sensor_set_odr(sensor, (rt_uint32_t)args & 0xffff);
        break;
    case RT_SENSOR_CTRL_SET_RANGE:
        result = spo2_sensor_set_range(sensor, (rt_uint32_t)args);
        break;
    case RT_SENSOR_CTRL_SET_POWER:
        result = spo2_sensor_set_power(sensor, (rt_uint32_t)args & 0xff);
        break;
    case RT_SENSOR_CTRL_SELF_TEST:
        /* TODO */
        result = -RT_EINVAL;
        break;
    default:
        return -RT_EINVAL;
    }
    return result;
}

static struct rt_sensor_ops sensor_ops[] =
{
    {spo2_sensor_fetch_data, spo2_sensor_control},
};

int rt_vd_sens_spo2_init(void)
{
    rt_int8_t result;
    rt_uint8_t index = 0;
    rt_sensor_t sensor_dat = RT_NULL;
    struct rt_sensor_config cfg;

    cfg.intf.dev_name = SENS_BUS_NAME;
    cfg.intf.user_data = RT_NULL;
    cfg.irq_pin.pin = RT_PIN_NONE;

    for (index = 0; index < SENS_SPO2_MAX; index++)
    {
        spo2_sensor_create(&cfg.intf, index);
        sensor_dat = rt_calloc(1, sizeof(struct rt_sensor_device));
        if (sensor_dat == RT_NULL)
        {
            LOG_E("%s:rt_calloc err!", __func__);
            return -RT_ERROR;
        }

        sensor_dat->info.type = spo2_info_tbl[index].type;
        sensor_dat->info.vendor = spo2_info_tbl[index].vendor;
        sensor_dat->info.model = spo2_info_tbl[index].model;
        sensor_dat->info.unit = spo2_info_tbl[index].unit;
        sensor_dat->info.intf_type = spo2_info_tbl[index].intf_type;
        sensor_dat->info.range_max = spo2_info_tbl[index].range_max;
        sensor_dat->info.range_min = spo2_info_tbl[index].range_min;
        sensor_dat->info.period_min = spo2_info_tbl[index].period_min;

        rt_memcpy(&sensor_dat->config, &cfg, sizeof(struct rt_sensor_config));
        sensor_dat->ops = &sensor_ops[index];

        result = rt_hw_sensor_register(sensor_dat, sens_spo2_tbl[index].dev_name, RT_DEVICE_FLAG_RDWR, RT_NULL);
        if (result != RT_EOK)
        {
            LOG_E("%s:device register err code: %d", __func__, result);
            rt_free(sensor_dat);
            return -RT_ERROR;
        }
    }

    return RT_EOK;
}

INIT_DEVICE_EXPORT(rt_vd_sens_spo2_init);

#endif