DoubleEndedQueue.sol 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. // SPDX-License-Identifier: MIT
  2. // OpenZeppelin Contracts (last updated v4.9.0) (utils/structs/DoubleEndedQueue.sol)
  3. pragma solidity ^0.8.20;
  4. /**
  5. * @dev A sequence of items with the ability to efficiently push and pop items (i.e. insert and remove) on both ends of
  6. * the sequence (called front and back). Among other access patterns, it can be used to implement efficient LIFO and
  7. * FIFO queues. Storage use is optimized, and all operations are O(1) constant time. This includes {clear}, given that
  8. * the existing queue contents are left in storage.
  9. *
  10. * The struct is called `Bytes32Deque`. Other types can be cast to and from `bytes32`. This data structure can only be
  11. * used in storage, and not in memory.
  12. * ```solidity
  13. * DoubleEndedQueue.Bytes32Deque queue;
  14. * ```
  15. */
  16. library DoubleEndedQueue {
  17. /**
  18. * @dev An operation (e.g. {front}) couldn't be completed due to the queue being empty.
  19. */
  20. error QueueEmpty();
  21. /**
  22. * @dev A push operation couldn't be completed due to the queue being full.
  23. */
  24. error QueueFull();
  25. /**
  26. * @dev An operation (e.g. {at}) couldn't be completed due to an index being out of bounds.
  27. */
  28. error QueueOutOfBounds();
  29. /**
  30. * @dev Indices are signed integers because the queue can grow in any direction. They are 128 bits so begin and end
  31. * are packed in a single storage slot for efficient access. Since the items are added one at a time we can safely
  32. * assume that these 128-bit indices will not overflow, and use unchecked arithmetic.
  33. *
  34. * Struct members have an underscore prefix indicating that they are "private" and should not be read or written to
  35. * directly. Use the functions provided below instead. Modifying the struct manually may violate assumptions and
  36. * lead to unexpected behavior.
  37. *
  38. * Indices are in the range [begin, end) which means the first item is at data[begin] and the last item is at
  39. * data[end - 1].
  40. */
  41. struct Bytes32Deque {
  42. uint128 _begin;
  43. uint128 _end;
  44. mapping(uint128 index => bytes32) _data;
  45. }
  46. /**
  47. * @dev Inserts an item at the end of the queue.
  48. */
  49. function pushBack(Bytes32Deque storage deque, bytes32 value) internal {
  50. unchecked {
  51. uint128 backIndex = deque._end;
  52. if (backIndex + 1 == deque._begin) revert QueueFull();
  53. deque._data[backIndex] = value;
  54. deque._end = backIndex + 1;
  55. }
  56. }
  57. /**
  58. * @dev Removes the item at the end of the queue and returns it.
  59. *
  60. * Reverts with `QueueEmpty` if the queue is empty.
  61. */
  62. function popBack(Bytes32Deque storage deque) internal returns (bytes32 value) {
  63. unchecked {
  64. uint128 backIndex = deque._end;
  65. if (backIndex == deque._begin) revert QueueEmpty();
  66. --backIndex;
  67. value = deque._data[backIndex];
  68. delete deque._data[backIndex];
  69. deque._end = backIndex;
  70. }
  71. }
  72. /**
  73. * @dev Inserts an item at the beginning of the queue.
  74. */
  75. function pushFront(Bytes32Deque storage deque, bytes32 value) internal {
  76. unchecked {
  77. uint128 frontIndex = deque._begin - 1;
  78. if (frontIndex == deque._end) revert QueueFull();
  79. deque._data[frontIndex] = value;
  80. deque._begin = frontIndex;
  81. }
  82. }
  83. /**
  84. * @dev Removes the item at the beginning of the queue and returns it.
  85. *
  86. * Reverts with `QueueEmpty` if the queue is empty.
  87. */
  88. function popFront(Bytes32Deque storage deque) internal returns (bytes32 value) {
  89. unchecked {
  90. uint128 frontIndex = deque._begin;
  91. if (frontIndex == deque._end) revert QueueEmpty();
  92. value = deque._data[frontIndex];
  93. delete deque._data[frontIndex];
  94. deque._begin = frontIndex + 1;
  95. }
  96. }
  97. /**
  98. * @dev Returns the item at the beginning of the queue.
  99. *
  100. * Reverts with `QueueEmpty` if the queue is empty.
  101. */
  102. function front(Bytes32Deque storage deque) internal view returns (bytes32 value) {
  103. if (empty(deque)) revert QueueEmpty();
  104. return deque._data[deque._begin];
  105. }
  106. /**
  107. * @dev Returns the item at the end of the queue.
  108. *
  109. * Reverts with `QueueEmpty` if the queue is empty.
  110. */
  111. function back(Bytes32Deque storage deque) internal view returns (bytes32 value) {
  112. if (empty(deque)) revert QueueEmpty();
  113. unchecked {
  114. return deque._data[deque._end - 1];
  115. }
  116. }
  117. /**
  118. * @dev Return the item at a position in the queue given by `index`, with the first item at 0 and last item at
  119. * `length(deque) - 1`.
  120. *
  121. * Reverts with `QueueOutOfBounds` if the index is out of bounds.
  122. */
  123. function at(Bytes32Deque storage deque, uint256 index) internal view returns (bytes32 value) {
  124. if (index >= length(deque)) revert QueueOutOfBounds();
  125. // By construction, length is a uint128, so the check above ensures that index can be safely downcast to uint128.
  126. unchecked {
  127. return deque._data[deque._begin + uint128(index)];
  128. }
  129. }
  130. /**
  131. * @dev Resets the queue back to being empty.
  132. *
  133. * NOTE: The current items are left behind in storage. This does not affect the functioning of the queue, but misses
  134. * out on potential gas refunds.
  135. */
  136. function clear(Bytes32Deque storage deque) internal {
  137. deque._begin = 0;
  138. deque._end = 0;
  139. }
  140. /**
  141. * @dev Returns the number of items in the queue.
  142. */
  143. function length(Bytes32Deque storage deque) internal view returns (uint256) {
  144. unchecked {
  145. return uint256(deque._end - deque._begin);
  146. }
  147. }
  148. /**
  149. * @dev Returns true if the queue is empty.
  150. */
  151. function empty(Bytes32Deque storage deque) internal view returns (bool) {
  152. return deque._end == deque._begin;
  153. }
  154. }