byte_queue.c 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174
  1. #include "byte_queue.h"
  2. static u16 byte_queue_add(const byte_queue_t *queue, u16 value1, u16 value2)
  3. {
  4. return (value1 + value2) % queue->size;
  5. }
  6. static u16 byte_queue_tail_add(const byte_queue_t *queue, u16 value)
  7. {
  8. return byte_queue_add(queue, queue->tail, value);
  9. }
  10. static u16 byte_queue_head_add(const byte_queue_t *queue, u16 value)
  11. {
  12. return byte_queue_add(queue, queue->head, value);
  13. }
  14. u8 *byte_queue_head(byte_queue_t *queue)
  15. {
  16. return queue->buff + queue->head;
  17. }
  18. bool byte_queue_empty(const byte_queue_t *queue)
  19. {
  20. return (bool) (queue->head == queue->tail);
  21. }
  22. bool byte_queue_readable(const byte_queue_t *queue)
  23. {
  24. return (bool) (queue->head != queue->tail);
  25. }
  26. bool byte_queue_full(const byte_queue_t *queue)
  27. {
  28. return (bool) (byte_queue_tail_add(queue, 1) == queue->head);
  29. }
  30. bool byte_queue_writeable(const byte_queue_t *queue)
  31. {
  32. return (bool) (byte_queue_tail_add(queue, 1) != queue->head);
  33. }
  34. u16 byte_queue_get_used(const byte_queue_t *queue)
  35. {
  36. if (queue->head <= queue->tail) {
  37. return queue->tail - queue->head;
  38. }
  39. return queue->size - (queue->head - queue->tail) - 1;
  40. }
  41. u16 byte_queue_get_free(const byte_queue_t *queue)
  42. {
  43. if (queue->tail < queue->head) {
  44. return queue->head - queue->tail;
  45. }
  46. return queue->size - (queue->tail - queue->head) - 1;
  47. }
  48. void byte_queue_reset(byte_queue_t *queue)
  49. {
  50. queue->head = queue->tail = 0;
  51. }
  52. void byte_queue_init(byte_queue_t *queue, u8 *buff, u16 size)
  53. {
  54. queue->buff = buff;
  55. queue->size = size;
  56. queue->head = queue->tail = 0;
  57. }
  58. u16 byte_queue_write(byte_queue_t *queue, const u8 *buff, u16 size)
  59. {
  60. const u8 *buff_bak = buff;
  61. const u8 *buff_end;
  62. for (buff_end = buff + size; buff < buff_end; buff++) {
  63. u16 tail = byte_queue_tail_add(queue, 1);
  64. if (tail == queue->head) {
  65. return buff - buff_bak;
  66. }
  67. queue->buff[queue->tail] = *buff;
  68. queue->tail = tail;
  69. }
  70. return size;
  71. }
  72. u16 byte_queue_write_byte(byte_queue_t *queue, u8 byte)
  73. {
  74. return byte_queue_write(queue, &byte, 1);
  75. }
  76. u16 byte_queue_read(byte_queue_t *queue, u8 *buff, u16 size)
  77. {
  78. u8 *buff_bak = buff;
  79. u8 *buff_end;
  80. for (buff_end = buff + size; buff < buff_end; buff++) {
  81. if (queue->head == queue->tail) {
  82. return buff - buff_bak;
  83. }
  84. *buff = queue->buff[queue->head];
  85. queue->head = byte_queue_head_add(queue, 1);
  86. }
  87. return size;
  88. }
  89. void byte_queue_fill(byte_queue_t *queue, u8 *buff, u16 size)
  90. {
  91. while (size > 0) {
  92. u16 length = byte_queue_read(queue, buff, size);
  93. size -= length;
  94. buff += size;
  95. }
  96. }
  97. void byte_queue_skip(byte_queue_t *queue, u16 length)
  98. {
  99. queue->head = byte_queue_head_add(queue, length);
  100. }
  101. u16 byte_queue_peek(byte_queue_t *queue)
  102. {
  103. if (queue->tail < queue->head) {
  104. return queue->size - queue->head;
  105. } else {
  106. return queue->tail - queue->head;
  107. }
  108. }
  109. // ================================================================================
  110. void byte_queue_alloc_init(byte_queue_t *queue, u8 *buff, u8 size)
  111. {
  112. queue->buff = buff;
  113. queue->size = size;
  114. byte_queue_alloc_reset(queue);
  115. }
  116. void byte_queue_alloc_reset(byte_queue_t *queue)
  117. {
  118. u8 *buff = queue->buff;
  119. u8 size = queue->size;
  120. u8 index;
  121. for (index = 0; index < size; index++) {
  122. buff[index] = index;
  123. }
  124. queue->tail = size - 1;
  125. queue->head = 0;
  126. }
  127. u8 byte_queue_alloc(byte_queue_t *queue)
  128. {
  129. u8 index;
  130. if (byte_queue_read(queue, &index, 1) > 0) {
  131. return index;
  132. }
  133. return 0xFF;
  134. }
  135. void byte_queue_free(byte_queue_t *queue, u8 index)
  136. {
  137. byte_queue_write(queue, &index, 1);
  138. }