From a9e14c89903c7cadad581393388d92a6e6ba8283 Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf.pld@gmail.com>
Date: Sun, 10 Dec 2017 21:56:19 +0100
Subject: [PATCH] Add standard lock events to shared locking.

---
 client/TracyLock.hpp | 47 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/client/TracyLock.hpp b/client/TracyLock.hpp
index cc304861..b16c4adb 100644
--- a/client/TracyLock.hpp
+++ b/client/TracyLock.hpp
@@ -205,17 +205,64 @@ public:
 
     tracy_force_inline void lock_shared()
     {
+        const auto thread = GetThreadHandle();
+        {
+            Magic magic;
+            auto& token = s_token.ptr;
+            auto& tail = token->get_tail_index();
+            auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+            item->hdr.type = QueueType::LockWait;
+            item->lockWait.id = m_id;
+            item->lockWait.thread = thread;
+            item->lockWait.time = Profiler::GetTime();
+            tail.store( magic + 1, std::memory_order_release );
+        }
+
         m_lockable.lock_shared();
+
+        {
+            Magic magic;
+            auto& token = s_token.ptr;
+            auto& tail = token->get_tail_index();
+            auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+            item->hdr.type = QueueType::LockObtain;
+            item->lockObtain.id = m_id;
+            item->lockObtain.thread = thread;
+            item->lockObtain.time = Profiler::GetTime();
+            tail.store( magic + 1, std::memory_order_release );
+        }
     }
 
     tracy_force_inline void unlock_shared()
     {
         m_lockable.unlock_shared();
+
+        Magic magic;
+        auto& token = s_token.ptr;
+        auto& tail = token->get_tail_index();
+        auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+        item->hdr.type = QueueType::LockRelease;
+        item->lockRelease.id = m_id;
+        item->lockRelease.thread = GetThreadHandle();
+        item->lockRelease.time = Profiler::GetTime();
+        tail.store( magic + 1, std::memory_order_release );
     }
 
     tracy_force_inline bool try_lock_shared()
     {
         const auto ret = m_lockable.try_lock_shared();
+        if( ret )
+        {
+            Magic magic;
+            auto& token = s_token.ptr;
+            auto& tail = token->get_tail_index();
+            auto item = token->enqueue_begin<moodycamel::CanAlloc>( magic );
+            item->hdr.type = QueueType::LockObtain;
+            item->lockObtain.id = (uint64_t)&m_lockable;
+            item->lockObtain.thread = GetThreadHandle();
+            item->lockObtain.time = Profiler::GetTime();
+            tail.store( magic + 1, std::memory_order_release );
+        }
         return ret;
     }