imstkSpinLock.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include <atomic>

namespace imstk
{
namespace ParallelUtils
{
class SpinLock
{
public:
    SpinLock()
    {
        unlock();
    }

    SpinLock(const SpinLock&)
    {
        unlock();
    }

    void lock()
    {
        while (m_Lock.test_and_set(std::memory_order_acquire)) {}
    }

    void unlock()
    {
        m_Lock.clear(std::memory_order_release);
    }

private:
    std::atomic_flag m_Lock;
};
} // end namespace ParallelUtils
} // end namespace imstk