mirror of
				https://github.com/PabloMK7/citra.git
				synced 2025-10-31 05:40:04 +00:00 
			
		
		
		
	Merge pull request #4645 from FearlessTobi/port-2116
Port yuzu-emu/yuzu#2116: "threadsafe_queue: Remove NeedSize template parameter"
This commit is contained in:
		
						commit
						93275d752b
					
				
					 2 changed files with 17 additions and 20 deletions
				
			
		|  | @ -7,18 +7,17 @@ | |||
| // a simple lockless thread-safe,
 | ||||
| // single reader, single writer queue
 | ||||
| 
 | ||||
| #include <algorithm> | ||||
| #include <atomic> | ||||
| #include <condition_variable> | ||||
| #include <cstddef> | ||||
| #include <mutex> | ||||
| #include "common/common_types.h" | ||||
| #include <utility> | ||||
| 
 | ||||
| namespace Common { | ||||
| template <typename T, bool NeedSize = true> | ||||
| template <typename T> | ||||
| class SPSCQueue { | ||||
| public: | ||||
|     SPSCQueue() : size(0) { | ||||
|     SPSCQueue() { | ||||
|         write_ptr = read_ptr = new ElementPtr(); | ||||
|     } | ||||
|     ~SPSCQueue() { | ||||
|  | @ -26,13 +25,12 @@ public: | |||
|         delete read_ptr; | ||||
|     } | ||||
| 
 | ||||
|     u32 Size() const { | ||||
|         static_assert(NeedSize, "using Size() on FifoQueue without NeedSize"); | ||||
|     std::size_t Size() const { | ||||
|         return size.load(); | ||||
|     } | ||||
| 
 | ||||
|     bool Empty() const { | ||||
|         return !read_ptr->next.load(); | ||||
|         return Size() == 0; | ||||
|     } | ||||
| 
 | ||||
|     T& Front() const { | ||||
|  | @ -48,14 +46,14 @@ public: | |||
|         ElementPtr* new_ptr = new ElementPtr(); | ||||
|         write_ptr->next.store(new_ptr, std::memory_order_release); | ||||
|         write_ptr = new_ptr; | ||||
|         if (NeedSize) | ||||
|             size++; | ||||
|         ++size; | ||||
| 
 | ||||
|         cv.notify_one(); | ||||
|     } | ||||
| 
 | ||||
|     void Pop() { | ||||
|         if (NeedSize) | ||||
|             size--; | ||||
|         --size; | ||||
| 
 | ||||
|         ElementPtr* tmpptr = read_ptr; | ||||
|         // advance the read pointer
 | ||||
|         read_ptr = tmpptr->next.load(); | ||||
|  | @ -68,8 +66,7 @@ public: | |||
|         if (Empty()) | ||||
|             return false; | ||||
| 
 | ||||
|         if (NeedSize) | ||||
|             size--; | ||||
|         --size; | ||||
| 
 | ||||
|         ElementPtr* tmpptr = read_ptr; | ||||
|         read_ptr = tmpptr->next.load(std::memory_order_acquire); | ||||
|  | @ -101,7 +98,7 @@ private: | |||
|     // and a pointer to the next ElementPtr
 | ||||
|     class ElementPtr { | ||||
|     public: | ||||
|         ElementPtr() : next(nullptr) {} | ||||
|         ElementPtr() = default; | ||||
|         ~ElementPtr() { | ||||
|             ElementPtr* next_ptr = next.load(); | ||||
| 
 | ||||
|  | @ -110,12 +107,12 @@ private: | |||
|         } | ||||
| 
 | ||||
|         T current; | ||||
|         std::atomic<ElementPtr*> next; | ||||
|         std::atomic<ElementPtr*> next{nullptr}; | ||||
|     }; | ||||
| 
 | ||||
|     ElementPtr* write_ptr; | ||||
|     ElementPtr* read_ptr; | ||||
|     std::atomic<u32> size; | ||||
|     std::atomic_size_t size{0}; | ||||
|     std::mutex cv_mutex; | ||||
|     std::condition_variable cv; | ||||
| }; | ||||
|  | @ -123,10 +120,10 @@ private: | |||
| // a simple thread-safe,
 | ||||
| // single reader, multiple writer queue
 | ||||
| 
 | ||||
| template <typename T, bool NeedSize = true> | ||||
| template <typename T> | ||||
| class MPSCQueue { | ||||
| public: | ||||
|     u32 Size() const { | ||||
|     std::size_t Size() const { | ||||
|         return spsc_queue.Size(); | ||||
|     } | ||||
| 
 | ||||
|  | @ -162,7 +159,7 @@ public: | |||
|     } | ||||
| 
 | ||||
| private: | ||||
|     SPSCQueue<T, NeedSize> spsc_queue; | ||||
|     SPSCQueue<T> spsc_queue; | ||||
|     std::mutex write_lock; | ||||
| }; | ||||
| } // namespace Common
 | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue