AsyncLatch.hpp

Go to the documentation of this file.

//
//  CeresEngine - A game development framework
//
//  Created by Rogiel Sulzbach.
//  Copyright (c) 2018-2022 Rogiel Sulzbach. All rights reserved.
//
 
#pragma once
 
#include "AsyncManualResetEvent.hpp"
 
#include "CeresEngine/DataTypes.hpp"
#include "CeresEngine/Macros.hpp"
 
#include "CeresEngine/Foundation/Container/Atomic.hpp"
 
namespace CeresEngine {
 
    class AsyncLatch {
    public:
        AsyncLatch(std::ptrdiff_t initialCount) noexcept : mCount(initialCount), mEvent(initialCount <= 0) {}
 
        bool is_ready() const noexcept { return mEvent.is_set(); }
 
        void count_down(std::ptrdiff_t n = 1) noexcept {
            if(mCount.fetch_sub(n, std::memory_order_acq_rel) <= n) {
                mEvent.set();
            }
        }
 
        auto operator co_await() const noexcept { return mEvent.operator co_await(); }
 
    private:
        Atomic<std::ptrdiff_t> mCount;
        AsyncManualResetEvent mEvent;
    };
} // namespace CeresEngine