438 lines
13 KiB
C
438 lines
13 KiB
C
|
/*
|
||
|
* Copyright (c) 2006-2023, RT-Thread Development Team
|
||
|
*
|
||
|
* SPDX-License-Identifier: Apache-2.0
|
||
|
*
|
||
|
* Change Logs:
|
||
|
* Date Author Notes
|
||
|
* 2018-05-07 aozima the first version
|
||
|
* 2022-05-14 Stanley Lwin add pwm function
|
||
|
* 2022-07-25 liYony fix complementary outputs and add usage information in finsh
|
||
|
* 2022-08-31 liYony Add complementary output section to framework for management
|
||
|
* 2022-09-24 qiyu Add dead-time and phase configuration
|
||
|
*/
|
||
|
|
||
|
#include <rtdevice.h>
|
||
|
|
||
|
static rt_err_t _pwm_control(rt_device_t dev, int cmd, void *args)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_device_pwm *pwm = (struct rt_device_pwm *)dev;
|
||
|
struct rt_pwm_configuration *configuration = (struct rt_pwm_configuration *)args;
|
||
|
|
||
|
switch (cmd)
|
||
|
{
|
||
|
case PWMN_CMD_ENABLE:
|
||
|
configuration->complementary = RT_TRUE;
|
||
|
break;
|
||
|
case PWMN_CMD_DISABLE:
|
||
|
configuration->complementary = RT_FALSE;
|
||
|
break;
|
||
|
default:
|
||
|
if(pwm->ops->control)
|
||
|
result = pwm->ops->control(pwm, cmd, args);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
|
||
|
/*
|
||
|
pos: channel
|
||
|
void *buffer: rt_uint32_t pulse[size]
|
||
|
size : number of pulse, only set to sizeof(rt_uint32_t).
|
||
|
*/
|
||
|
static rt_ssize_t _pwm_read(rt_device_t dev, rt_off_t pos, void *buffer, rt_size_t size)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_device_pwm *pwm = (struct rt_device_pwm *)dev;
|
||
|
rt_uint32_t *pulse = (rt_uint32_t *)buffer;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
configuration.channel = (pos > 0) ? (pos) : (-pos);
|
||
|
|
||
|
if (pwm->ops->control)
|
||
|
{
|
||
|
result = pwm->ops->control(pwm, PWM_CMD_GET, &configuration);
|
||
|
if (result != RT_EOK)
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
*pulse = configuration.pulse;
|
||
|
}
|
||
|
|
||
|
return size;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
pos: channel
|
||
|
void *buffer: rt_uint32_t pulse[size]
|
||
|
size : number of pulse, only set to sizeof(rt_uint32_t).
|
||
|
*/
|
||
|
static rt_ssize_t _pwm_write(rt_device_t dev, rt_off_t pos, const void *buffer, rt_size_t size)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_device_pwm *pwm = (struct rt_device_pwm *)dev;
|
||
|
rt_uint32_t *pulse = (rt_uint32_t *)buffer;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
configuration.channel = (pos > 0) ? (pos) : (-pos);
|
||
|
|
||
|
if (pwm->ops->control)
|
||
|
{
|
||
|
result = pwm->ops->control(pwm, PWM_CMD_GET, &configuration);
|
||
|
if (result != RT_EOK)
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
configuration.pulse = *pulse;
|
||
|
|
||
|
result = pwm->ops->control(pwm, PWM_CMD_SET, &configuration);
|
||
|
if (result != RT_EOK)
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return size;
|
||
|
}
|
||
|
|
||
|
#ifdef RT_USING_DEVICE_OPS
|
||
|
static const struct rt_device_ops pwm_device_ops =
|
||
|
{
|
||
|
RT_NULL,
|
||
|
RT_NULL,
|
||
|
RT_NULL,
|
||
|
_pwm_read,
|
||
|
_pwm_write,
|
||
|
_pwm_control
|
||
|
};
|
||
|
#endif /* RT_USING_DEVICE_OPS */
|
||
|
|
||
|
rt_err_t rt_device_pwm_register(struct rt_device_pwm *device, const char *name, const struct rt_pwm_ops *ops, const void *user_data)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
|
||
|
rt_memset(device, 0, sizeof(struct rt_device_pwm));
|
||
|
|
||
|
#ifdef RT_USING_DEVICE_OPS
|
||
|
device->parent.ops = &pwm_device_ops;
|
||
|
#else
|
||
|
device->parent.init = RT_NULL;
|
||
|
device->parent.open = RT_NULL;
|
||
|
device->parent.close = RT_NULL;
|
||
|
device->parent.read = _pwm_read;
|
||
|
device->parent.write = _pwm_write;
|
||
|
device->parent.control = _pwm_control;
|
||
|
#endif /* RT_USING_DEVICE_OPS */
|
||
|
|
||
|
device->parent.type = RT_Device_Class_PWM;
|
||
|
device->ops = ops;
|
||
|
device->parent.user_data = (void *)user_data;
|
||
|
|
||
|
result = rt_device_register(&device->parent, name, RT_DEVICE_FLAG_RDWR);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_enable(struct rt_device_pwm *device, int channel)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
/* Make it is positive num forever */
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
|
||
|
/* If channel is a positive number (0 ~ n), it means using normal output pin.
|
||
|
* If channel is a negative number (0 ~ -n), it means using complementary output pin. */
|
||
|
if(channel > 0)
|
||
|
{
|
||
|
result = rt_device_control(&device->parent, PWMN_CMD_DISABLE, &configuration);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
result = rt_device_control(&device->parent, PWMN_CMD_ENABLE, &configuration);
|
||
|
}
|
||
|
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_ENABLE, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_disable(struct rt_device_pwm *device, int channel)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
/* Make it is positive num forever */
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
|
||
|
/* If channel is a positive number (0 ~ n), it means using normal output pin.
|
||
|
* If channel is a negative number (0 ~ -n), it means using complementary output pin. */
|
||
|
if(channel > 0)
|
||
|
{
|
||
|
result = rt_device_control(&device->parent, PWMN_CMD_DISABLE, &configuration);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
result = rt_device_control(&device->parent, PWMN_CMD_ENABLE, &configuration);
|
||
|
}
|
||
|
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_DISABLE, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_set(struct rt_device_pwm *device, int channel, rt_uint32_t period, rt_uint32_t pulse)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
configuration.period = period;
|
||
|
configuration.pulse = pulse;
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_SET, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_set_period(struct rt_device_pwm *device, int channel, rt_uint32_t period)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
configuration.period = period;
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_SET_PERIOD, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_set_pulse(struct rt_device_pwm *device, int channel, rt_uint32_t pulse)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
configuration.pulse = pulse;
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_SET_PULSE, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
rt_err_t rt_pwm_set_dead_time(struct rt_device_pwm *device, int channel, rt_uint32_t dead_time)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
configuration.dead_time = dead_time;
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_SET_DEAD_TIME, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
|
||
|
rt_err_t rt_pwm_set_phase(struct rt_device_pwm *device, int channel, rt_uint32_t phase)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
struct rt_pwm_configuration configuration = {0};
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
configuration.channel = (channel > 0) ? (channel) : (-channel);
|
||
|
configuration.phase = phase;
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_SET_PHASE, &configuration);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
rt_err_t rt_pwm_get(struct rt_device_pwm *device, struct rt_pwm_configuration *cfg)
|
||
|
{
|
||
|
rt_err_t result = RT_EOK;
|
||
|
|
||
|
if (!device)
|
||
|
{
|
||
|
return -RT_EIO;
|
||
|
}
|
||
|
|
||
|
result = rt_device_control(&device->parent, PWM_CMD_GET, cfg);
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
#ifdef RT_USING_FINSH
|
||
|
#include <stdlib.h>
|
||
|
#include <string.h>
|
||
|
#include <finsh.h>
|
||
|
|
||
|
static int pwm(int argc, char **argv)
|
||
|
{
|
||
|
rt_err_t result = -RT_ERROR;
|
||
|
char *result_str;
|
||
|
static struct rt_device_pwm *pwm_device = RT_NULL;
|
||
|
struct rt_pwm_configuration cfg = {0};
|
||
|
|
||
|
if(argc > 1)
|
||
|
{
|
||
|
if(!strcmp(argv[1], "probe"))
|
||
|
{
|
||
|
if(argc == 3)
|
||
|
{
|
||
|
pwm_device = (struct rt_device_pwm *)rt_device_find(argv[2]);
|
||
|
result_str = (pwm_device == RT_NULL) ? "failure" : "success";
|
||
|
rt_kprintf("probe %s %s\n", argv[2], result_str);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("pwm probe <device name> - probe pwm by name\n");
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if(pwm_device == RT_NULL)
|
||
|
{
|
||
|
rt_kprintf("Please using 'pwm probe <device name>' first.\n");
|
||
|
return -RT_ERROR;
|
||
|
}
|
||
|
if(!strcmp(argv[1], "enable"))
|
||
|
{
|
||
|
if(argc == 3)
|
||
|
{
|
||
|
result = rt_pwm_enable(pwm_device, atoi(argv[2]));
|
||
|
result_str = (result == RT_EOK) ? "success" : "failure";
|
||
|
rt_kprintf("%s channel %d is enabled %s \n", pwm_device->parent.parent.name, atoi(argv[2]), result_str);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("pwm enable <channel> - enable pwm channel\n");
|
||
|
rt_kprintf(" e.g. MSH >pwm enable 1 - PWM_CH1 nomal\n");
|
||
|
rt_kprintf(" e.g. MSH >pwm enable -1 - PWM_CH1N complememtary\n");
|
||
|
}
|
||
|
}
|
||
|
else if(!strcmp(argv[1], "disable"))
|
||
|
{
|
||
|
if(argc == 3)
|
||
|
{
|
||
|
result = rt_pwm_disable(pwm_device, atoi(argv[2]));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("pwm disable <channel> - disable pwm channel\n");
|
||
|
}
|
||
|
}
|
||
|
else if(!strcmp(argv[1], "get"))
|
||
|
{
|
||
|
cfg.channel = atoi(argv[2]);
|
||
|
result = rt_pwm_get(pwm_device, &cfg);
|
||
|
if(result == RT_EOK)
|
||
|
{
|
||
|
rt_kprintf("Info of device [%s] channel [%d]:\n",pwm_device, atoi(argv[2]));
|
||
|
rt_kprintf("period : %d\n", cfg.period);
|
||
|
rt_kprintf("pulse : %d\n", cfg.pulse);
|
||
|
rt_kprintf("Duty cycle : %d%%\n",(int)(((double)(cfg.pulse)/(cfg.period)) * 100));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("Get info of device: [%s] error.\n", pwm_device);
|
||
|
}
|
||
|
}
|
||
|
else if (!strcmp(argv[1], "set"))
|
||
|
{
|
||
|
if(argc == 5)
|
||
|
{
|
||
|
result = rt_pwm_set(pwm_device, atoi(argv[2]), atoi(argv[3]), atoi(argv[4]));
|
||
|
rt_kprintf("pwm info set on %s at channel %d\n",pwm_device,(rt_base_t)atoi(argv[2]));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("Set info of device: [%s] error\n", pwm_device);
|
||
|
rt_kprintf("Usage: pwm set <channel> <period> <pulse>\n");
|
||
|
}
|
||
|
}
|
||
|
else if(!strcmp(argv[1], "phase"))
|
||
|
{
|
||
|
if(argc == 4)
|
||
|
{
|
||
|
result = rt_pwm_set_phase(pwm_device, atoi(argv[2]),atoi(argv[3]));
|
||
|
result_str = (result == RT_EOK) ? "success" : "failure";
|
||
|
rt_kprintf("%s phase is set %d \n", pwm_device->parent.parent.name, (rt_base_t)atoi(argv[3]));
|
||
|
}
|
||
|
}
|
||
|
else if(!strcmp(argv[1], "dead_time"))
|
||
|
{
|
||
|
if(argc == 4)
|
||
|
{
|
||
|
result = rt_pwm_set_dead_time(pwm_device, atoi(argv[2]),atoi(argv[3]));
|
||
|
result_str = (result == RT_EOK) ? "success" : "failure";
|
||
|
rt_kprintf("%s dead_time is set %d \n", pwm_device->parent.parent.name, (rt_base_t)atoi(argv[3]));
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("Usage: \n");
|
||
|
rt_kprintf("pwm probe <device name> - probe pwm by name\n");
|
||
|
rt_kprintf("pwm enable <channel> - enable pwm channel\n");
|
||
|
rt_kprintf("pwm disable <channel> - disable pwm channel\n");
|
||
|
rt_kprintf("pwm get <channel> - get pwm channel info\n");
|
||
|
rt_kprintf("pwm set <channel> <period> <pulse> - set pwm channel info\n");
|
||
|
rt_kprintf("pwm phase <channel> <phase> - set pwm phase\n");
|
||
|
rt_kprintf("pwm dead_time <channel> <dead_time> - set pwm dead time\n");
|
||
|
result = -RT_ERROR;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("Usage: \n");
|
||
|
rt_kprintf("pwm probe <device name> - probe pwm by name\n");
|
||
|
rt_kprintf("pwm enable <channel> - enable pwm channel\n");
|
||
|
rt_kprintf("pwm disable <channel> - disable pwm channel\n");
|
||
|
rt_kprintf("pwm get <channel> - get pwm channel info\n");
|
||
|
rt_kprintf("pwm set <channel> <period> <pulse> - set pwm channel info\n");
|
||
|
rt_kprintf("pwm phase <channel> <phase> - set pwm phase\n");
|
||
|
rt_kprintf("pwm dead_time <channel> <dead_time> - set pwm dead time\n");
|
||
|
result = -RT_ERROR;
|
||
|
}
|
||
|
|
||
|
return RT_EOK;
|
||
|
}
|
||
|
MSH_CMD_EXPORT(pwm, pwm [option]);
|
||
|
|
||
|
#endif /* RT_USING_FINSH */
|