v_magnetometer.c

/*
 * Copyright (c) 2006-2020, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2020-12-19     zhangsz      add virtual sensor device
 */

#include <rtthread.h>

#ifdef PKG_USING_VIRTUAL_SENSOR_MAG

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

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

enum SENS_MAG_ID
{
    SENS_MAG_01 = 0, //Magnetometer
    SENS_MAG_MAX,
};

#define SENS_BUS_NAME                       "sens_bus"
#define SENS_MAG_01_SENSOR_ID                  (RT_SENSOR_CLASS_MAG + 0x10)

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

static struct sens_mag sens_mag_tbl[SENS_MAG_MAX] =
{
    {V_SENS_MAG_DEV_NAME,        0x00 }, /* Magnetometer */
};

static struct rt_sensor_info mag_info_tbl[SENS_MAG_MAX] =
{
    {RT_SENSOR_CLASS_MAG,   RT_SENSOR_VENDOR_BOSCH,   RT_NULL,    RT_SENSOR_UNIT_MGAUSS,      RT_SENSOR_INTF_SPI,     360,   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_MAG_01:
        chip_id = SENS_MAG_01_SENSOR_ID;
        break;
    default:
        break;
    }

    return chip_id;
}

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

    return RT_EOK;
}

static void* mag_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 mag_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 mag_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 mag_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 mag_sensor_fetch_data(struct rt_sensor_device* sensor, void* buf, rt_size_t size)
{
    struct rt_sensor_data* data = buf;
    rt_int16_t max_range = 0;

    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;
    }

    max_range = mag_info_tbl[SENS_MAG_01].range_max - mag_info_tbl[SENS_MAG_01].range_min;

    for (int i = 0; i < size; i++)
    {
        data->type = RT_SENSOR_CLASS_MAG;
        data->data.mag.x = rand() % max_range + mag_info_tbl[SENS_MAG_01].range_min;
        data->data.mag.y = rand() % max_range + mag_info_tbl[SENS_MAG_01].range_min;
        data->data.mag.z = rand() % max_range + mag_info_tbl[SENS_MAG_01].range_min;
        data->timestamp = rt_sensor_get_ts();

        LOG_D("%s:%d,%d,%d", __func__, data->data.mag.x,
            data->data.mag.y, data->data.mag.z);
        data++;
    }

    return size;
}

static rt_err_t mag_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_mag_tbl[SENS_MAG_01].sens_id;
        break;
    case RT_SENSOR_CTRL_SET_ODR:
        result = mag_sensor_set_odr(sensor, (rt_uint32_t)args & 0xffff);
        break;
    case RT_SENSOR_CTRL_SET_RANGE:
        result = mag_sensor_set_range(sensor, (rt_uint32_t)args);
        break;
    case RT_SENSOR_CTRL_SET_POWER:
        result = mag_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[] =
{
    {mag_sensor_fetch_data, mag_sensor_control},
};

int rt_vd_sens_mag_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_MAG_MAX; index++)
    {
        mag_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 = mag_info_tbl[index].type;
        sensor_dat->info.vendor = mag_info_tbl[index].vendor;
        sensor_dat->info.model = mag_info_tbl[index].model;
        sensor_dat->info.unit = mag_info_tbl[index].unit;
        sensor_dat->info.intf_type = mag_info_tbl[index].intf_type;
        sensor_dat->info.range_max = mag_info_tbl[index].range_max;
        sensor_dat->info.range_min = mag_info_tbl[index].range_min;
        sensor_dat->info.period_min = mag_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_mag_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_mag_init);

#endif