/* * Copyright (c) 2006-2023, RT-Thread Development Team * * SPDX-License-Identifier: Apache-2.0 * * Change Logs: * Date Author Notes * 2023-02-13 zhkag first version * 2023-04-03 xqyjlj fix cputimer in multithreading */ #include #include #include static rt_list_t _cputimer_list = RT_LIST_OBJECT_INIT(_cputimer_list); static struct rt_cputimer *_cputimer_nowtimer = RT_NULL; static void _cputime_sleep_timeout(void *parameter) { struct rt_semaphore *sem; sem = (struct rt_semaphore *)parameter; rt_sem_release(sem); } static void _cputime_timeout_callback(void *parameter) { struct rt_cputimer *timer; timer = (struct rt_cputimer *)parameter; rt_base_t level; level = rt_hw_interrupt_disable(); _cputimer_nowtimer = RT_NULL; rt_list_remove(&(timer->row)); rt_hw_interrupt_enable(level); timer->timeout_func(&(timer->sem)); } static void _set_next_timeout() { struct rt_cputimer *t; if (&_cputimer_list != _cputimer_list.prev) { t = rt_list_entry((&_cputimer_list)->next, struct rt_cputimer, row); if (_cputimer_nowtimer != RT_NULL) { if (t != _cputimer_nowtimer && t->timeout_tick < _cputimer_nowtimer->timeout_tick) { _cputimer_nowtimer = t; clock_cpu_settimeout(t->timeout_tick, _cputime_timeout_callback, t); } } else { _cputimer_nowtimer = t; clock_cpu_settimeout(t->timeout_tick, _cputime_timeout_callback, t); } } else { _cputimer_nowtimer = NULL; } } void rt_cputimer_init(rt_cputimer_t timer, const char *name, void (*timeout)(void *parameter), void *parameter, rt_uint64_t tick, rt_uint8_t flag) { /* parameter check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(timeout != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); /* set flag */ timer->parent.flag = flag; /* set deactivated */ timer->parent.flag &= ~RT_TIMER_FLAG_ACTIVATED; timer->timeout_func = timeout; timer->parameter = parameter; timer->timeout_tick = tick + clock_cpu_gettime(); timer->init_tick = tick; rt_list_init(&(timer->row)); rt_sem_init(&(timer->sem), "cputime", 0, RT_IPC_FLAG_PRIO); } rt_err_t rt_cputimer_delete(rt_cputimer_t timer) { rt_base_t level; /* parameter check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); /* disable interrupt */ level = rt_hw_interrupt_disable(); rt_list_remove(&timer->row); /* stop timer */ timer->parent.flag &= ~RT_TIMER_FLAG_ACTIVATED; /* enable interrupt */ rt_hw_interrupt_enable(level); _set_next_timeout(); return RT_EOK; } rt_err_t rt_cputimer_start(rt_cputimer_t timer) { rt_list_t *timer_list; rt_base_t level; /* parameter check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); /* stop timer firstly */ level = rt_hw_interrupt_disable(); /* remove timer from list */ rt_list_remove(&timer->row); /* change status of timer */ timer->parent.flag &= ~RT_TIMER_FLAG_ACTIVATED; timer_list = &_cputimer_list; for (; timer_list != _cputimer_list.prev; timer_list = timer_list->next) { struct rt_cputimer *t; rt_list_t *p = timer_list->next; t = rt_list_entry(p, struct rt_cputimer, row); if ((t->timeout_tick - timer->timeout_tick) == 0) { continue; } else if ((t->timeout_tick - timer->timeout_tick) < 0x7fffffffffffffff) { break; } } rt_list_insert_after(timer_list, &(timer->row)); timer->parent.flag |= RT_TIMER_FLAG_ACTIVATED; _set_next_timeout(); /* enable interrupt */ rt_hw_interrupt_enable(level); return RT_EOK; } rt_err_t rt_cputimer_stop(rt_cputimer_t timer) { rt_base_t level; /* disable interrupt */ level = rt_hw_interrupt_disable(); /* timer check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); if (!(timer->parent.flag & RT_TIMER_FLAG_ACTIVATED)) { rt_hw_interrupt_enable(level); return -RT_ERROR; } rt_list_remove(&timer->row); /* change status */ timer->parent.flag &= ~RT_TIMER_FLAG_ACTIVATED; _set_next_timeout(); /* enable interrupt */ rt_hw_interrupt_enable(level); return RT_EOK; } rt_err_t rt_cputimer_control(rt_cputimer_t timer, int cmd, void *arg) { rt_base_t level; /* parameter check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); level = rt_hw_interrupt_disable(); switch (cmd) { case RT_TIMER_CTRL_GET_TIME: *(rt_uint64_t *)arg = timer->init_tick; break; case RT_TIMER_CTRL_SET_TIME: RT_ASSERT((*(rt_uint64_t *)arg) < 0x7fffffffffffffff); timer->init_tick = *(rt_uint64_t *)arg; break; case RT_TIMER_CTRL_SET_ONESHOT: timer->parent.flag &= ~RT_TIMER_FLAG_PERIODIC; break; case RT_TIMER_CTRL_SET_PERIODIC: timer->parent.flag |= RT_TIMER_FLAG_PERIODIC; break; case RT_TIMER_CTRL_GET_STATE: if (timer->parent.flag & RT_TIMER_FLAG_ACTIVATED) { /*timer is start and run*/ *(rt_uint32_t *)arg = RT_TIMER_FLAG_ACTIVATED; } else { /*timer is stop*/ *(rt_uint32_t *)arg = RT_TIMER_FLAG_DEACTIVATED; } break; case RT_TIMER_CTRL_GET_REMAIN_TIME: *(rt_uint64_t *)arg = timer->timeout_tick; break; case RT_TIMER_CTRL_GET_FUNC: arg = (void *)timer->timeout_func; break; case RT_TIMER_CTRL_SET_FUNC: timer->timeout_func = (void (*)(void *))arg; break; case RT_TIMER_CTRL_GET_PARM: *(void **)arg = timer->parameter; break; case RT_TIMER_CTRL_SET_PARM: timer->parameter = arg; break; default: break; } rt_hw_interrupt_enable(level); return RT_EOK; } rt_err_t rt_cputimer_detach(rt_cputimer_t timer) { rt_base_t level; /* parameter check */ RT_ASSERT(timer != RT_NULL); RT_ASSERT(clock_cpu_issettimeout() != RT_FALSE); /* disable interrupt */ level = rt_hw_interrupt_disable(); rt_list_remove(&timer->row); /* stop timer */ timer->parent.flag &= ~RT_TIMER_FLAG_ACTIVATED; _set_next_timeout(); /* enable interrupt */ rt_hw_interrupt_enable(level); rt_sem_detach(&(timer->sem)); return RT_EOK; } rt_err_t rt_cputime_sleep(rt_uint64_t tick) { rt_base_t level; struct rt_cputimer cputimer; if (!clock_cpu_issettimeout()) { rt_int32_t ms = clock_cpu_millisecond(tick); return rt_thread_delay(rt_tick_from_millisecond(ms)); } if (tick == 0) { return -RT_EINVAL; } rt_cputimer_init(&cputimer, "cputime_sleep", _cputime_sleep_timeout, &(cputimer.sem), tick, RT_TIMER_FLAG_ONE_SHOT | RT_TIMER_FLAG_SOFT_TIMER); /* disable interrupt */ level = rt_hw_interrupt_disable(); rt_cputimer_start(&cputimer); /* reset the timeout of thread timer and start it */ rt_hw_interrupt_enable(level); rt_sem_take_interruptible(&(cputimer.sem), RT_WAITING_FOREVER); rt_cputimer_detach(&cputimer); return RT_EOK; } rt_err_t rt_cputime_ndelay(rt_uint64_t ns) { uint64_t unit = clock_cpu_getres(); return rt_cputime_sleep(ns * (1000UL * 1000) / unit); } rt_err_t rt_cputime_udelay(rt_uint64_t us) { return rt_cputime_ndelay(us * 1000); } rt_err_t rt_cputime_mdelay(rt_uint64_t ms) { return rt_cputime_ndelay(ms * 1000000); }