tracyLocks.h

Go to the documentation of this file.

#pragma once

#include "vrfutil/profiler.h"
#include "vrfutil/vrfProfileColors.h"


const char DtScopedLockDefaultName[] = "DtTracyScopedLock_waiting";
template<typename T_Mutex, char const* T_LockName = DtScopedLockDefaultName>
class DtTracyScopedLock
{
public:
   DtTracyScopedLock(T_Mutex& m)
      : lock()
   {
      if (!lock.try_acquire(m))
      {
         DtPROFILEC(T_LockName, DtVrfProfileColor::WaitingOnLock);
         lock.acquire(m);
      }
   };

   void release()
   {
      lock.release();
   }

protected:
   typename T_Mutex::scoped_lock lock;
};


const char DtScopedRwLockDefaultName[] = "DtTracyScopedRwLock_waiting";
template<typename T_Mutex, char const* T_LockName = DtScopedRwLockDefaultName>
class DtTracyScopedRwLock
{
public:
   DtTracyScopedRwLock(T_Mutex& m, bool writeLock) 
      : lock() 
   {
      acquire(m, writeLock);
   };

   void acquire(T_Mutex& m, bool writeLock = true)
   {
      if (!lock.try_acquire(m, writeLock))
      {
         DtPROFILEC(T_LockName, tracy::Color::Red);
         lock.acquire(m, writeLock);
      }
   }

   bool upgrade_to_writer()
   {
      DtPROFILEC(T_LockName, tracy::Color::Red);
      return lock.upgrade_to_writer();
   }

   void release()
   {
      lock.release();
   }

   bool downgrade_to_reader()
   {
      return lock.downgrade_to_reader();
   }

protected:
   typename T_Mutex::scoped_lock lock;
};