#pragma once
#include "antlr4-common.h"
#include <mutex>
#include <shared_mutex>
#include <utility>
#if ANTLR4CPP_USING_ABSEIL
#include "absl/base/call_once.h"
#include "absl/base/thread_annotations.h"
#include "absl/synchronization/mutex.h"
#define ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS ABSL_NO_THREAD_SAFETY_ANALYSIS
#else
#define ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS
#endif
namespace antlr4::internal {
class ANTLR4CPP_PUBLIC Mutex final {
public:
Mutex() = default;
Mutex(const Mutex&) = delete;
Mutex(Mutex&&) = delete;
Mutex& operator=(const Mutex&) = delete;
Mutex& operator=(Mutex&&) = delete;
void lock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
bool try_lock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
void unlock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
private:
#if ANTLR4CPP_USING_ABSEIL
absl::Mutex _impl;
#else
std::mutex _impl;
#endif
};
template <typename Mutex>
using UniqueLock = std::unique_lock<Mutex>;
class ANTLR4CPP_PUBLIC SharedMutex final {
public:
SharedMutex() = default;
SharedMutex(const SharedMutex&) = delete;
SharedMutex(SharedMutex&&) = delete;
SharedMutex& operator=(const SharedMutex&) = delete;
SharedMutex& operator=(SharedMutex&&) = delete;
void lock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
bool try_lock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
void unlock() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
void lock_shared() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
bool try_lock_shared() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
void unlock_shared() ANTLR4CPP_NO_THREAD_SAFTEY_ANALYSIS;
private:
#if ANTLR4CPP_USING_ABSEIL
absl::Mutex _impl;
#else
std::shared_mutex _impl;
#endif
};
template <typename Mutex>
using SharedLock = std::shared_lock<Mutex>;
class OnceFlag;
template <typename Callable, typename... Args>
void call_once(OnceFlag &onceFlag, Callable &&callable, Args&&... args);
class ANTLR4CPP_PUBLIC OnceFlag final {
public:
constexpr OnceFlag() = default;
OnceFlag(const OnceFlag&) = delete;
OnceFlag(OnceFlag&&) = delete;
OnceFlag& operator=(const OnceFlag&) = delete;
OnceFlag& operator=(OnceFlag&&) = delete;
private:
template <typename Callable, typename... Args>
friend void call_once(OnceFlag &onceFlag, Callable &&callable, Args&&... args);
#if ANTLR4CPP_USING_ABSEIL
absl::once_flag _impl;
#else
std::once_flag _impl;
#endif
};
template <typename Callable, typename... Args>
void call_once(OnceFlag &onceFlag, Callable &&callable, Args&&... args) {
#if ANTLR4CPP_USING_ABSEIL
absl::call_once(onceFlag._impl, std::forward<Callable>(callable), std::forward<Args>(args)...);
#else
std::call_once(onceFlag._impl, std::forward<Callable>(callable), std::forward<Args>(args)...);
#endif
}
}