2018-08-23 00:24:55 +08:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <string>
|
|
|
|
#include <vector>
|
2019-06-05 03:21:04 +08:00
|
|
|
#include <deque>
|
2019-02-09 13:30:38 +08:00
|
|
|
#include <array>
|
2018-08-23 00:24:55 +08:00
|
|
|
#include <unordered_map>
|
2018-11-05 10:19:00 +08:00
|
|
|
#include <unordered_set>
|
2018-08-23 00:24:55 +08:00
|
|
|
#include <functional>
|
2018-10-31 13:20:44 +08:00
|
|
|
#include <algorithm>
|
2018-11-05 10:19:00 +08:00
|
|
|
#include <regex>
|
2018-10-31 13:20:44 +08:00
|
|
|
#include <memory>
|
2018-10-10 19:03:15 +08:00
|
|
|
#include <optional>
|
2018-08-23 00:24:55 +08:00
|
|
|
#include <thread>
|
2018-08-23 23:53:23 +08:00
|
|
|
#include <mutex>
|
2019-01-24 02:59:34 +08:00
|
|
|
#include <shared_mutex>
|
2019-02-09 13:30:38 +08:00
|
|
|
#include <atomic>
|
2020-02-17 08:58:09 +08:00
|
|
|
#include <filesystem>
|
2018-09-21 09:59:07 +08:00
|
|
|
#include <cstdint>
|
2018-12-27 12:56:42 +08:00
|
|
|
#include <cassert>
|
2019-02-19 12:12:12 +08:00
|
|
|
|
2019-06-01 02:48:07 +08:00
|
|
|
#ifdef _WIN64
|
|
|
|
constexpr bool x64 = true;
|
|
|
|
#else
|
|
|
|
constexpr bool x64 = false;
|
|
|
|
#endif
|
|
|
|
|
2020-01-19 14:25:57 +08:00
|
|
|
template <typename T, typename... Xs>
|
|
|
|
struct ArrayImpl { using type = std::tuple<T, Xs...>[]; };
|
|
|
|
template <typename T>
|
|
|
|
struct ArrayImpl<T> { using type = T[]; };
|
|
|
|
template <typename... Ts>
|
|
|
|
using Array = typename ArrayImpl<Ts...>::type;
|
2019-02-19 12:12:12 +08:00
|
|
|
|
2020-02-17 08:58:09 +08:00
|
|
|
template <auto F> using Functor = std::integral_constant<std::remove_reference_t<decltype(F)>, F>;
|
2019-08-07 17:18:04 +08:00
|
|
|
|
|
|
|
template <typename V>
|
|
|
|
struct Identity { V operator()(V v) const { return v; } };
|
|
|
|
|
|
|
|
struct PermissivePointer
|
|
|
|
{
|
|
|
|
template <typename T>
|
|
|
|
operator T*() { return (T*)p; }
|
|
|
|
void* p;
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename HandleCloser = Functor<CloseHandle>>
|
|
|
|
class AutoHandle
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
AutoHandle(HANDLE h) : h(h) {}
|
|
|
|
operator HANDLE() { return h.get(); }
|
|
|
|
PHANDLE operator&() { static_assert(sizeof(*this) == sizeof(HANDLE)); assert(!h); return (PHANDLE)this; }
|
|
|
|
operator bool() { return h.get() != NULL && h.get() != INVALID_HANDLE_VALUE; }
|
|
|
|
|
|
|
|
private:
|
|
|
|
struct HandleCleaner { void operator()(void* h) { if (h != INVALID_HANDLE_VALUE) HandleCloser()(PermissivePointer{ h }); } };
|
|
|
|
std::unique_ptr<void, HandleCleaner> h;
|
|
|
|
};
|
|
|
|
|
2019-06-10 13:49:11 +08:00
|
|
|
template<typename T, typename M = std::mutex>
|
2019-06-05 11:12:45 +08:00
|
|
|
class Synchronized
|
2019-02-19 12:12:12 +08:00
|
|
|
{
|
|
|
|
public:
|
|
|
|
template <typename... Args>
|
2019-06-05 11:12:45 +08:00
|
|
|
Synchronized(Args&&... args) : contents(std::forward<Args>(args)...) {}
|
|
|
|
|
|
|
|
struct Locker
|
2019-02-19 12:12:12 +08:00
|
|
|
{
|
2019-06-10 13:49:11 +08:00
|
|
|
T* operator->() { return &contents; }
|
2019-06-05 11:12:45 +08:00
|
|
|
std::unique_lock<M> lock;
|
2019-06-10 13:49:11 +08:00
|
|
|
T& contents;
|
2019-06-05 11:12:45 +08:00
|
|
|
};
|
|
|
|
|
2019-06-10 13:49:11 +08:00
|
|
|
Locker Acquire() { return { std::unique_lock(m), contents }; }
|
2019-06-05 11:12:45 +08:00
|
|
|
Locker operator->() { return Acquire(); }
|
2019-02-19 12:12:12 +08:00
|
|
|
|
|
|
|
private:
|
2019-06-10 13:49:11 +08:00
|
|
|
T contents;
|
|
|
|
M m;
|
2019-02-19 12:12:12 +08:00
|
|
|
};
|
|
|
|
|
2019-06-09 19:33:26 +08:00
|
|
|
static struct
|
2019-06-04 05:58:30 +08:00
|
|
|
{
|
|
|
|
BYTE DUMMY[100];
|
|
|
|
template <typename T>
|
|
|
|
operator T*() { static_assert(sizeof(T) < sizeof(DUMMY)); return (T*)DUMMY; }
|
|
|
|
} DUMMY;
|
|
|
|
|
2019-06-13 16:01:29 +08:00
|
|
|
template <typename T>
|
|
|
|
inline auto FormatArg(T arg) { return arg; }
|
|
|
|
|
|
|
|
template <typename C>
|
|
|
|
inline auto FormatArg(const std::basic_string<C>& arg) { return arg.c_str(); }
|
|
|
|
|
2019-06-02 01:59:37 +08:00
|
|
|
#pragma warning(push)
|
|
|
|
#pragma warning(disable: 4996)
|
|
|
|
template <typename... Args>
|
2019-06-29 18:13:26 +08:00
|
|
|
inline std::string FormatString(const char* format, const Args&... args)
|
2019-06-02 01:59:37 +08:00
|
|
|
{
|
2019-06-13 16:01:29 +08:00
|
|
|
std::string buffer(snprintf(nullptr, 0, format, FormatArg(args)...), '\0');
|
|
|
|
sprintf(buffer.data(), format, FormatArg(args)...);
|
2019-06-02 01:59:37 +08:00
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename... Args>
|
2019-06-29 18:13:26 +08:00
|
|
|
inline std::wstring FormatString(const wchar_t* format, const Args&... args)
|
2019-06-02 01:59:37 +08:00
|
|
|
{
|
2019-06-13 16:01:29 +08:00
|
|
|
std::wstring buffer(_snwprintf(nullptr, 0, format, FormatArg(args)...), L'\0');
|
|
|
|
_swprintf(buffer.data(), format, FormatArg(args)...);
|
2019-06-02 01:59:37 +08:00
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
#pragma warning(pop)
|
|
|
|
|
2020-02-28 15:34:34 +08:00
|
|
|
inline std::optional<std::wstring> StringToWideString(const std::string& text, UINT encoding)
|
|
|
|
{
|
|
|
|
std::vector<wchar_t> buffer(text.size() + 1);
|
|
|
|
if (int length = MultiByteToWideChar(encoding, 0, text.c_str(), text.size() + 1, buffer.data(), buffer.size()))
|
|
|
|
return std::wstring(buffer.data(), length - 1);
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
|
|
|
inline std::wstring StringToWideString(const std::string& text)
|
|
|
|
{
|
|
|
|
std::vector<wchar_t> buffer(text.size() + 1);
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, text.c_str(), -1, buffer.data(), buffer.size());
|
|
|
|
return buffer.data();
|
|
|
|
}
|
|
|
|
|
|
|
|
inline std::string WideStringToString(const std::wstring& text)
|
|
|
|
{
|
|
|
|
std::vector<char> buffer((text.size() + 1) * 4);
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, text.c_str(), -1, buffer.data(), buffer.size(), nullptr, nullptr);
|
|
|
|
return buffer.data();
|
|
|
|
}
|
|
|
|
|
2020-02-12 14:34:03 +08:00
|
|
|
template <typename... Args>
|
|
|
|
inline void TEXTRACTOR_MESSAGE(const wchar_t* format, const Args&... args) { MessageBoxW(NULL, FormatString(format, args...).c_str(), L"Textractor", MB_OK); }
|
2019-09-11 06:53:55 +08:00
|
|
|
|
2019-02-19 12:12:12 +08:00
|
|
|
#ifdef _DEBUG
|
2019-06-13 16:01:29 +08:00
|
|
|
#define TEST(...) static auto _ = CreateThread(nullptr, 0, [](auto) { __VA_ARGS__; return 0UL; }, NULL, 0, nullptr);
|
2019-02-19 12:12:12 +08:00
|
|
|
#else
|
|
|
|
#define TEST(...)
|
|
|
|
#endif
|