添加rtthread相关代码
This commit is contained in:
		
							
								
								
									
										296
									
								
								riscv/rtthread/components/drivers/audio/audio_pipe.c
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										296
									
								
								riscv/rtthread/components/drivers/audio/audio_pipe.c
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,296 @@ | ||||
| /* | ||||
|  * Copyright (c) 2006-2023, RT-Thread Development Team | ||||
|  * | ||||
|  * SPDX-License-Identifier: Apache-2.0 | ||||
|  * | ||||
|  * Change Logs: | ||||
|  * Date           Author       Notes | ||||
|  * 2012-09-30     Bernard      first version. | ||||
|  */ | ||||
|  | ||||
| #include <rthw.h> | ||||
| #include <rtdevice.h> | ||||
| #include "audio_pipe.h" | ||||
|  | ||||
| static void _rt_pipe_resume_writer(struct rt_audio_pipe *pipe) | ||||
| { | ||||
|     if (!rt_list_isempty(&pipe->suspended_write_list)) | ||||
|     { | ||||
|         rt_thread_t thread; | ||||
|  | ||||
|         RT_ASSERT(pipe->flag & RT_PIPE_FLAG_BLOCK_WR); | ||||
|  | ||||
|         /* get suspended thread */ | ||||
|         thread = RT_THREAD_LIST_NODE_ENTRY(pipe->suspended_write_list.next); | ||||
|  | ||||
|         /* resume the write thread */ | ||||
|         rt_thread_resume(thread); | ||||
|  | ||||
|         rt_schedule(); | ||||
|     } | ||||
| } | ||||
|  | ||||
| static rt_ssize_t rt_pipe_read(rt_device_t dev, | ||||
|                               rt_off_t    pos, | ||||
|                               void       *buffer, | ||||
|                               rt_size_t   size) | ||||
| { | ||||
|     rt_base_t level; | ||||
|     rt_thread_t thread; | ||||
|     struct rt_audio_pipe *pipe; | ||||
|     rt_size_t read_nbytes; | ||||
|  | ||||
|     pipe = (struct rt_audio_pipe *)dev; | ||||
|     RT_ASSERT(pipe != RT_NULL); | ||||
|  | ||||
|     if (!(pipe->flag & RT_PIPE_FLAG_BLOCK_RD)) | ||||
|     { | ||||
|         level = rt_hw_interrupt_disable(); | ||||
|         read_nbytes = rt_ringbuffer_get(&(pipe->ringbuffer), (rt_uint8_t *)buffer, size); | ||||
|  | ||||
|         /* if the ringbuffer is empty, there won't be any writer waiting */ | ||||
|         if (read_nbytes) | ||||
|             _rt_pipe_resume_writer(pipe); | ||||
|  | ||||
|         rt_hw_interrupt_enable(level); | ||||
|  | ||||
|         return read_nbytes; | ||||
|     } | ||||
|  | ||||
|     thread = rt_thread_self(); | ||||
|  | ||||
|     /* current context checking */ | ||||
|     RT_DEBUG_NOT_IN_INTERRUPT; | ||||
|  | ||||
|     do | ||||
|     { | ||||
|         level = rt_hw_interrupt_disable(); | ||||
|         read_nbytes = rt_ringbuffer_get(&(pipe->ringbuffer), (rt_uint8_t *)buffer, size); | ||||
|         if (read_nbytes == 0) | ||||
|         { | ||||
|             rt_thread_suspend(thread); | ||||
|             /* waiting on suspended read list */ | ||||
|             rt_list_insert_before(&(pipe->suspended_read_list), | ||||
|                                   &RT_THREAD_LIST_NODE(thread)); | ||||
|             rt_hw_interrupt_enable(level); | ||||
|  | ||||
|             rt_schedule(); | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             _rt_pipe_resume_writer(pipe); | ||||
|             rt_hw_interrupt_enable(level); | ||||
|             break; | ||||
|         } | ||||
|     } | ||||
|     while (read_nbytes == 0); | ||||
|  | ||||
|     return read_nbytes; | ||||
| } | ||||
|  | ||||
| static void _rt_pipe_resume_reader(struct rt_audio_pipe *pipe) | ||||
| { | ||||
|     if (pipe->parent.rx_indicate) | ||||
|         pipe->parent.rx_indicate(&pipe->parent, | ||||
|                                  rt_ringbuffer_data_len(&pipe->ringbuffer)); | ||||
|  | ||||
|     if (!rt_list_isempty(&pipe->suspended_read_list)) | ||||
|     { | ||||
|         rt_thread_t thread; | ||||
|  | ||||
|         RT_ASSERT(pipe->flag & RT_PIPE_FLAG_BLOCK_RD); | ||||
|  | ||||
|         /* get suspended thread */ | ||||
|         thread = RT_THREAD_LIST_NODE_ENTRY(pipe->suspended_read_list.next); | ||||
|  | ||||
|         /* resume the read thread */ | ||||
|         rt_thread_resume(thread); | ||||
|  | ||||
|         rt_schedule(); | ||||
|     } | ||||
| } | ||||
|  | ||||
| static rt_ssize_t rt_pipe_write(rt_device_t dev, | ||||
|                                rt_off_t    pos, | ||||
|                                const void *buffer, | ||||
|                                rt_size_t   size) | ||||
| { | ||||
|     rt_base_t level; | ||||
|     rt_thread_t thread; | ||||
|     struct rt_audio_pipe *pipe; | ||||
|     rt_size_t write_nbytes; | ||||
|  | ||||
|     pipe = (struct rt_audio_pipe *)dev; | ||||
|     RT_ASSERT(pipe != RT_NULL); | ||||
|  | ||||
|     if ((pipe->flag & RT_PIPE_FLAG_FORCE_WR) || | ||||
|             !(pipe->flag & RT_PIPE_FLAG_BLOCK_WR)) | ||||
|     { | ||||
|         level = rt_hw_interrupt_disable(); | ||||
|  | ||||
|         if (pipe->flag & RT_PIPE_FLAG_FORCE_WR) | ||||
|             write_nbytes = rt_ringbuffer_put_force(&(pipe->ringbuffer), | ||||
|                                                    (const rt_uint8_t *)buffer, size); | ||||
|         else | ||||
|             write_nbytes = rt_ringbuffer_put(&(pipe->ringbuffer), | ||||
|                                              (const rt_uint8_t *)buffer, size); | ||||
|  | ||||
|         _rt_pipe_resume_reader(pipe); | ||||
|  | ||||
|         rt_hw_interrupt_enable(level); | ||||
|  | ||||
|         return write_nbytes; | ||||
|     } | ||||
|  | ||||
|     thread = rt_thread_self(); | ||||
|  | ||||
|     /* current context checking */ | ||||
|     RT_DEBUG_NOT_IN_INTERRUPT; | ||||
|  | ||||
|     do | ||||
|     { | ||||
|         level = rt_hw_interrupt_disable(); | ||||
|         write_nbytes = rt_ringbuffer_put(&(pipe->ringbuffer), (const rt_uint8_t *)buffer, size); | ||||
|         if (write_nbytes == 0) | ||||
|         { | ||||
|             /* pipe full, waiting on suspended write list */ | ||||
|             rt_thread_suspend(thread); | ||||
|             /* waiting on suspended read list */ | ||||
|             rt_list_insert_before(&(pipe->suspended_write_list), | ||||
|                                   &RT_THREAD_LIST_NODE(thread)); | ||||
|             rt_hw_interrupt_enable(level); | ||||
|  | ||||
|             rt_schedule(); | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             _rt_pipe_resume_reader(pipe); | ||||
|             rt_hw_interrupt_enable(level); | ||||
|             break; | ||||
|         } | ||||
|     } | ||||
|     while (write_nbytes == 0); | ||||
|  | ||||
|     return write_nbytes; | ||||
| } | ||||
|  | ||||
| static rt_err_t rt_pipe_control(rt_device_t dev, int cmd, void *args) | ||||
| { | ||||
|     struct rt_audio_pipe *pipe; | ||||
|  | ||||
|     pipe = (struct rt_audio_pipe *)dev; | ||||
|  | ||||
|     if (cmd == PIPE_CTRL_GET_SPACE && args) | ||||
|         *(rt_size_t *)args = rt_ringbuffer_space_len(&pipe->ringbuffer); | ||||
|     return RT_EOK; | ||||
| } | ||||
|  | ||||
| #ifdef RT_USING_DEVICE_OPS | ||||
| const static struct rt_device_ops audio_pipe_ops = | ||||
| { | ||||
|     RT_NULL, | ||||
|     RT_NULL, | ||||
|     RT_NULL, | ||||
|     rt_pipe_read, | ||||
|     rt_pipe_write, | ||||
|     rt_pipe_control | ||||
| }; | ||||
| #endif | ||||
|  | ||||
| /** | ||||
|  * This function will initialize a pipe device and put it under control of | ||||
|  * resource management. | ||||
|  * | ||||
|  * @param pipe the pipe device | ||||
|  * @param name the name of pipe device | ||||
|  * @param flag the attribute of the pipe device | ||||
|  * @param buf  the buffer of pipe device | ||||
|  * @param size the size of pipe device buffer | ||||
|  * | ||||
|  * @return the operation status, RT_EOK on successful | ||||
|  */ | ||||
| rt_err_t rt_audio_pipe_init(struct rt_audio_pipe *pipe, | ||||
|                             const char *name, | ||||
|                             rt_int32_t flag, | ||||
|                             rt_uint8_t *buf, | ||||
|                             rt_size_t size) | ||||
| { | ||||
|     RT_ASSERT(pipe); | ||||
|     RT_ASSERT(buf); | ||||
|  | ||||
|     /* initialize suspended list */ | ||||
|     rt_list_init(&pipe->suspended_read_list); | ||||
|     rt_list_init(&pipe->suspended_write_list); | ||||
|  | ||||
|     /* initialize ring buffer */ | ||||
|     rt_ringbuffer_init(&pipe->ringbuffer, buf, size); | ||||
|  | ||||
|     pipe->flag = flag; | ||||
|  | ||||
|     /* create pipe */ | ||||
|     pipe->parent.type    = RT_Device_Class_Pipe; | ||||
| #ifdef RT_USING_DEVICE_OPS | ||||
|     pipe->parent.ops     = &audio_pipe_ops; | ||||
| #else | ||||
|     pipe->parent.init    = RT_NULL; | ||||
|     pipe->parent.open    = RT_NULL; | ||||
|     pipe->parent.close   = RT_NULL; | ||||
|     pipe->parent.read    = rt_pipe_read; | ||||
|     pipe->parent.write   = rt_pipe_write; | ||||
|     pipe->parent.control = rt_pipe_control; | ||||
| #endif | ||||
|  | ||||
|     return rt_device_register(&(pipe->parent), name, RT_DEVICE_FLAG_RDWR); | ||||
| } | ||||
|  | ||||
| /** | ||||
|  * This function will detach a pipe device from resource management | ||||
|  * | ||||
|  * @param pipe the pipe device | ||||
|  * | ||||
|  * @return the operation status, RT_EOK on successful | ||||
|  */ | ||||
| rt_err_t rt_audio_pipe_detach(struct rt_audio_pipe *pipe) | ||||
| { | ||||
|     return rt_device_unregister(&pipe->parent); | ||||
| } | ||||
|  | ||||
| #ifdef RT_USING_HEAP | ||||
| rt_err_t rt_audio_pipe_create(const char *name, rt_int32_t flag, rt_size_t size) | ||||
| { | ||||
|     rt_uint8_t *rb_memptr = RT_NULL; | ||||
|     struct rt_audio_pipe *pipe = RT_NULL; | ||||
|  | ||||
|     /* get aligned size */ | ||||
|     size = RT_ALIGN(size, RT_ALIGN_SIZE); | ||||
|     pipe = (struct rt_audio_pipe *)rt_calloc(1, sizeof(struct rt_audio_pipe)); | ||||
|     if (pipe == RT_NULL) | ||||
|         return -RT_ENOMEM; | ||||
|  | ||||
|     /* create ring buffer of pipe */ | ||||
|     rb_memptr = (rt_uint8_t *)rt_malloc(size); | ||||
|     if (rb_memptr == RT_NULL) | ||||
|     { | ||||
|         rt_free(pipe); | ||||
|         return -RT_ENOMEM; | ||||
|     } | ||||
|  | ||||
|     return rt_audio_pipe_init(pipe, name, flag, rb_memptr, size); | ||||
| } | ||||
|  | ||||
| void rt_audio_pipe_destroy(struct rt_audio_pipe *pipe) | ||||
| { | ||||
|     if (pipe == RT_NULL) | ||||
|         return; | ||||
|  | ||||
|     /* un-register pipe device */ | ||||
|     rt_audio_pipe_detach(pipe); | ||||
|  | ||||
|     /* release memory */ | ||||
|     rt_free(pipe->ringbuffer.buffer_ptr); | ||||
|     rt_free(pipe); | ||||
|  | ||||
|     return; | ||||
| } | ||||
|  | ||||
| #endif /* RT_USING_HEAP */ | ||||
		Reference in New Issue
	
	Block a user