Index: webrtc/common_audio/swap_queue.h |
diff --git a/webrtc/common_audio/swap_queue.h b/webrtc/common_audio/swap_queue.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..d8bb5c024e1bb10c9b3b259663b213a0d8074b02 |
--- /dev/null |
+++ b/webrtc/common_audio/swap_queue.h |
@@ -0,0 +1,210 @@ |
+/* |
+ * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#ifndef WEBRTC_COMMON_AUDIO_SWAP_QUEUE_H_ |
+#define WEBRTC_COMMON_AUDIO_SWAP_QUEUE_H_ |
+ |
+#include <algorithm> |
+#include <utility> |
+#include <vector> |
+ |
+#include "webrtc/base/checks.h" |
+#include "webrtc/base/criticalsection.h" |
+ |
+namespace webrtc { |
+ |
+namespace internal { |
+ |
+// (Internal; please don't use outside this file.) |
+template <typename T> |
+bool NoopSwapQueueItemVerifierFunction(const T&) { |
+ return true; |
+} |
+ |
+} // namespace internal |
+ |
+// Functor to use when supplying a verifier function for the queue. |
+template <typename T, |
+ bool (*QueueItemVerifierFunction)(const T&) = |
+ internal::NoopSwapQueueItemVerifierFunction> |
+class SwapQueueItemVerifier { |
+ public: |
+ bool operator()(const T& t) const { return QueueItemVerifierFunction(t); } |
+}; |
+ |
+// This class is a fixed-size queue. A producer calls Insert() to insert |
+// an element of type T at the back of the queue, and a consumer calls |
+// Remove() to remove an element from the front of the queue. It's safe |
+// for the producer(s) and the consumer(s) to access the queue |
+// concurrently, from different threads. |
+// |
+// To avoid the construction, copying, and destruction of Ts that a naive |
+// queue implementation would require, for each "full" T passed from |
+// producer to consumer, SwapQueue<T> passes an "empty" T in the other |
+// direction (an "empty" T is one that contains nothing of value for the |
+// consumer). This bidirectional movement is implemented with swap(). |
+// |
+// // Create queue: |
+// Bottle proto(568); // Prepare an empty Bottle. Heap allocates space for |
+// // 568 ml. |
+// SwapQueue<Bottle> q(N, proto); // Init queue with N copies of proto. |
+// // Each copy allocates on the heap. |
+// // Producer pseudo-code: |
+// Bottle b(568); // Prepare an empty Bottle. Heap allocates space for 568 ml. |
+// loop { |
+// b.Fill(amount); // Where amount <= 568 ml. |
+// q.Insert(&b); // Swap our full Bottle for an empty one from q. |
+// } |
+// |
+// // Consumer pseudo-code: |
+// Bottle b(568); // Prepare an empty Bottle. Heap allocates space for 568 ml. |
+// loop { |
+// q.Remove(&b); // Swap our empty Bottle for the next-in-line full Bottle. |
+// Drink(&b); |
+// } |
+// |
+// For a well-behaved Bottle class, there are no allocations in the |
+// producer, since it just fills an empty Bottle that's already large |
+// enough; no deallocations in the consumer, since it returns each empty |
+// Bottle to the queue after having drunk it; and no copies along the |
+// way, since the queue uses swap() everywhere to move full Bottles in |
+// one direction and empty ones in the other. |
+template <typename T, typename QueueItemVerifier = SwapQueueItemVerifier<T>> |
+class SwapQueue { |
+ public: |
+ // Creates a queue of size size and fills it with default constructed Ts. |
+ explicit SwapQueue(size_t size) : queue_(size) { |
+ RTC_DCHECK(VerifyQueueSlots()); |
+ } |
+ |
+ // Same as above and accepts an item verification functor. |
+ SwapQueue(size_t size, const QueueItemVerifier& queue_item_verifier) |
+ : queue_item_verifier_(queue_item_verifier), queue_(size) { |
+ RTC_DCHECK(VerifyQueueSlots()); |
+ } |
+ |
+ // Creates a queue of size size and fills it with copies of prototype. |
+ SwapQueue(size_t size, const T& prototype) : queue_(size, prototype) { |
+ RTC_DCHECK(VerifyQueueSlots()); |
+ } |
+ |
+ // Same as above and accepts an item verification functor. |
+ SwapQueue(size_t size, |
+ const T& prototype, |
+ const QueueItemVerifier& queue_item_verifier) |
+ : queue_item_verifier_(queue_item_verifier), queue_(size, prototype) { |
+ RTC_DCHECK(VerifyQueueSlots()); |
+ } |
+ |
+ // Resets the queue to have zero content wile maintaining the queue size. |
+ void Clear() { |
+ rtc::CritScope cs(&crit_queue_); |
+ next_write_index_ = 0; |
+ next_read_index_ = 0; |
+ num_elements_ = 0; |
+ } |
+ |
+ // Inserts a "full" T at the back of the queue by swapping *input with an |
+ // "empty" T from the queue. |
+ // Returns true if the item was inserted or false if not (the queue was full). |
+ // When specified, the T given in *input must pass the ItemVerifier() test. |
+ // The contents of *input after the call are then also guaranteed to pass the |
+ // ItemVerifier() test. |
+ bool Insert(T* input) WARN_UNUSED_RESULT { |
+ RTC_DCHECK(input); |
+ |
+ rtc::CritScope cs(&crit_queue_); |
+ |
+ RTC_DCHECK(queue_item_verifier_(*input)); |
+ |
+ if (num_elements_ == queue_.size()) { |
+ return false; |
+ } |
+ |
+ using std::swap; |
+ swap(*input, queue_[next_write_index_]); |
+ |
+ ++next_write_index_; |
+ if (next_write_index_ == queue_.size()) { |
+ next_write_index_ = 0; |
+ } |
+ |
+ ++num_elements_; |
+ |
+ RTC_DCHECK_LT(next_write_index_, queue_.size()); |
+ RTC_DCHECK_LE(num_elements_, queue_.size()); |
+ |
+ return true; |
+ } |
+ |
+ // Removes the frontmost "full" T from the queue by swapping it with |
+ // the "empty" T in *output. |
+ // Returns true if an item could be removed or false if not (the queue was |
+ // empty). When specified, The T given in *output must pass the ItemVerifier() |
+ // test and the contents of *output after the call are then also guaranteed to |
+ // pass the ItemVerifier() test. |
+ bool Remove(T* output) WARN_UNUSED_RESULT { |
+ RTC_DCHECK(output); |
+ |
+ rtc::CritScope cs(&crit_queue_); |
+ |
+ RTC_DCHECK(queue_item_verifier_(*output)); |
+ |
+ if (num_elements_ == 0) { |
+ return false; |
+ } |
+ |
+ using std::swap; |
+ swap(*output, queue_[next_read_index_]); |
+ |
+ ++next_read_index_; |
+ if (next_read_index_ == queue_.size()) { |
+ next_read_index_ = 0; |
+ } |
+ |
+ --num_elements_; |
+ |
+ RTC_DCHECK_LT(next_read_index_, queue_.size()); |
+ RTC_DCHECK_LE(num_elements_, queue_.size()); |
+ |
+ return true; |
+ } |
+ |
+ private: |
+ // Verify that the queue slots complies with the ItemVerifier test. |
+ bool VerifyQueueSlots() { |
+ rtc::CritScope cs(&crit_queue_); |
+ for (const auto& v : queue_) { |
+ RTC_DCHECK(queue_item_verifier_(v)); |
+ } |
+ return true; |
+ } |
+ |
+ rtc::CriticalSection crit_queue_; |
+ |
+ // TODO(peah): Change this to use std::function() once we can use C++11 std |
+ // lib. |
+ QueueItemVerifier queue_item_verifier_ GUARDED_BY(crit_queue_); |
+ |
+ // (next_read_index_ + num_elements_) % queue_.size() = |
+ // next_write_index_ |
+ size_t next_write_index_ GUARDED_BY(crit_queue_) = 0; |
+ size_t next_read_index_ GUARDED_BY(crit_queue_) = 0; |
+ size_t num_elements_ GUARDED_BY(crit_queue_) = 0; |
+ |
+ // queue_.size() is constant. |
+ std::vector<T> queue_ GUARDED_BY(crit_queue_); |
+ |
+ RTC_DISALLOW_COPY_AND_ASSIGN(SwapQueue); |
+}; |
+ |
+} // namespace webrtc |
+ |
+#endif // WEBRTC_COMMON_AUDIO_SWAP_QUEUE_H_ |