CommonLibs/plog/Util.h

#pragma once
#include <cassert>
#include <cstring>
#include <cstdio>
#include <sstream>
#include <fcntl.h>
#include <sys/stat.h>

#ifndef PLOG_ENABLE_WCHAR_INPUT
#   ifdef _WIN32
#       define PLOG_ENABLE_WCHAR_INPUT 1
#   else
#       define PLOG_ENABLE_WCHAR_INPUT 0
#   endif
#endif

#ifdef _WIN32
#   include <plog/WinApi.h>
#   include <time.h>
#   include <sys/timeb.h>
#   include <io.h>
#   include <share.h>
#else
#   include <unistd.h>
#   include <sys/syscall.h>
#   include <sys/time.h>
#   include <pthread.h>
#   if PLOG_ENABLE_WCHAR_INPUT
#       include <iconv.h>
#   endif
#endif

#ifdef _WIN32
#   define _PLOG_NSTR(x)   L##x
#   define PLOG_NSTR(x)    _PLOG_NSTR(x)
#else
#   define PLOG_NSTR(x)    x
#endif

namespace plog
{
    namespace util
    {
#ifdef _WIN32
        typedef std::wstring nstring;
        typedef std::wstringstream nstringstream;
        typedef wchar_t nchar;
#else
        typedef std::string nstring;
        typedef std::stringstream nstringstream;
        typedef char nchar;
#endif

        inline void localtime_s(struct tm* t, const time_t* time)
        {
#if defined(_WIN32) && defined(__BORLANDC__)
            ::localtime_s(time, t);
#elif defined(_WIN32) && defined(__MINGW32__) && !defined(__MINGW64_VERSION_MAJOR)
            *t = *::localtime(time);
#elif defined(_WIN32)
            ::localtime_s(t, time);
#else
            ::localtime_r(time, t);
#endif
        }

#ifdef _WIN32
        typedef timeb Time;

        inline void ftime(Time* t)
        {
            ::ftime(t);
        }
#else
        struct Time
        {
            time_t time;
            unsigned short millitm;
        };

        inline void ftime(Time* t)
        {
            timeval tv;
            ::gettimeofday(&tv, NULL);

            t->time = tv.tv_sec;
            t->millitm = static_cast<unsigned short>(tv.tv_usec / 1000);
        }
#endif

        inline unsigned int gettid()
        {
#ifdef _WIN32
            return GetCurrentThreadId();
#elif defined(__unix__)
            return static_cast<unsigned int>(::syscall(__NR_gettid));
#elif defined(__APPLE__)
            uint64_t tid64;
            pthread_threadid_np(NULL, &tid64);
            return static_cast<unsigned int>(tid64);
#endif
        }

#if PLOG_ENABLE_WCHAR_INPUT && !defined(_WIN32)
        inline std::string toNarrow(const wchar_t* wstr)
        {
            size_t wlen = ::wcslen(wstr);
            std::string str(wlen * sizeof(wchar_t), 0);

            if (!str.empty())
            {
                const char* in = reinterpret_cast<const char*>(&wstr[0]);
                char* out = &str[0];
                size_t inBytes = wlen * sizeof(wchar_t);
                size_t outBytes = str.size();

                iconv_t cd = ::iconv_open("UTF-8", "WCHAR_T");
                ::iconv(cd, const_cast<char**>(&in), &inBytes, &out, &outBytes);
                ::iconv_close(cd);

                str.resize(str.size() - outBytes);
            }

            return str;
        }
#endif

#ifdef _WIN32
        inline std::wstring toWide(const char* str)
        {
            size_t len = ::strlen(str);
            std::wstring wstr(len, 0);

            if (!wstr.empty())
            {
                int wlen = MultiByteToWideChar(codePage::kActive, 0, str, static_cast<int>(len), &wstr[0], static_cast<int>(wstr.size()));
                wstr.resize(wlen);
            }

            return wstr;
        }

        inline std::string toUTF8(const std::wstring& wstr)
        {
            std::string str(wstr.size() * sizeof(wchar_t), 0);

            if (!str.empty())
            {
                int len = WideCharToMultiByte(codePage::kUTF8, 0, wstr.c_str(), static_cast<int>(wstr.size()), &str[0], static_cast<int>(str.size()), 0, 0);
                str.resize(len);
            }

            return str;
        }
#endif

        inline std::string processFuncName(const char* func)
        {
#if (defined(_WIN32) && !defined(__MINGW32__)) || defined(__OBJC__)
            return std::string(func);
#else
            const char* funcBegin = func;
            const char* funcEnd = ::strchr(funcBegin, '(');

            if (!funcEnd)
            {
                return std::string(func);
            }

            for (const char* i = funcEnd - 1; i >= funcBegin; --i) // search backwards for the first space char
            {
                if (*i == ' ')
                {
                    funcBegin = i + 1;
                    break;
                }
            }

            return std::string(funcBegin, funcEnd);
#endif
        }

        inline const nchar* findExtensionDot(const nchar* fileName)
        {
#ifdef _WIN32
            return std::wcsrchr(fileName, L'.');
#else
            return std::strrchr(fileName, '.');
#endif
        }

        inline void splitFileName(const nchar* fileName, nstring& fileNameNoExt, nstring& fileExt)
        {
            const nchar* dot = findExtensionDot(fileName);

            if (dot)
            {
                fileNameNoExt.assign(fileName, dot);
                fileExt.assign(dot + 1);
            }
            else
            {
                fileNameNoExt.assign(fileName);
                fileExt.clear();
            }
        }

        class NonCopyable
        {
        protected:
            NonCopyable()
            {
            }

        private:
            NonCopyable(const NonCopyable&);
            NonCopyable& operator=(const NonCopyable&);
        };

        class File : NonCopyable
        {
        public:
            File() : m_file(-1)
            {
            }

            File(const nchar* fileName) : m_file(-1)
            {
                open(fileName);
            }

            ~File()
            {
                close();
            }

            off_t open(const nchar* fileName)
            {
#if defined(_WIN32) && (defined(__BORLANDC__) || defined(__MINGW32__))
                m_file = ::_wsopen(fileName, _O_CREAT | _O_WRONLY | _O_BINARY, SH_DENYWR, _S_IREAD | _S_IWRITE);
#elif defined(_WIN32)
                ::_wsopen_s(&m_file, fileName, _O_CREAT | _O_WRONLY | _O_BINARY, _SH_DENYWR, _S_IREAD | _S_IWRITE);
#else
                m_file = ::open(fileName, O_CREAT | O_WRONLY, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
#endif
                return seek(0, SEEK_END);
            }

            int write(const void* buf, size_t count)
            {
#ifdef _WIN32
                return m_file != -1 ? ::_write(m_file, buf, static_cast<unsigned int>(count)) : -1;
#else
                return m_file != -1 ? static_cast<int>(::write(m_file, buf, count)) : -1;
#endif
            }

            template<class CharType>
            int write(const std::basic_string<CharType>& str)
            {
                return write(str.data(), str.size() * sizeof(CharType));
            }

            off_t seek(off_t offset, int whence)
            {
#ifdef _WIN32
                return m_file != -1 ? ::_lseek(m_file, offset, whence) : -1;
#else
                return m_file != -1 ? ::lseek(m_file, offset, whence) : -1;
#endif
            }

            void close()
            {
                if (m_file != -1)
                {
#ifdef _WIN32
                    ::_close(m_file);
#else
                    ::close(m_file);
#endif
                    m_file = -1;
                }
            }

            static int unlink(const nchar* fileName)
            {
#ifdef _WIN32
                return ::_wunlink(fileName);
#else
                return ::unlink(fileName);
#endif
            }

            static int rename(const nchar* oldFilename, const nchar* newFilename)
            {
#ifdef _WIN32
                return MoveFileW(oldFilename, newFilename);
#else
                return ::rename(oldFilename, newFilename);
#endif
            }

        private:
            int m_file;
        };

        class Mutex : NonCopyable
        {
        public:
            Mutex()
            {
#ifdef _WIN32
                InitializeCriticalSection(&m_sync);
#else
                ::pthread_mutex_init(&m_sync, 0);
#endif
            }

            ~Mutex()
            {
#ifdef _WIN32
                DeleteCriticalSection(&m_sync);
#else
                ::pthread_mutex_destroy(&m_sync);
#endif
            }

            friend class MutexLock;

        private:
            void lock()
            {
#ifdef _WIN32
                EnterCriticalSection(&m_sync);
#else
                ::pthread_mutex_lock(&m_sync);
#endif
            }

            void unlock()
            {
#ifdef _WIN32
                LeaveCriticalSection(&m_sync);
#else
                ::pthread_mutex_unlock(&m_sync);
#endif
            }

        private:
#ifdef _WIN32
            CRITICAL_SECTION m_sync;
#else
            pthread_mutex_t m_sync;
#endif
        };

        class MutexLock : NonCopyable
        {
        public:
            MutexLock(Mutex& mutex) : m_mutex(mutex)
            {
                m_mutex.lock();
            }

            ~MutexLock()
            {
                m_mutex.unlock();
            }

        private:
            Mutex& m_mutex;
        };

        template<class T>
        class Singleton : NonCopyable
        {
        public:
            Singleton()
            {
                assert(!m_instance);
                m_instance = static_cast<T*>(this);
            }

            ~Singleton()
            {
                assert(m_instance);
                m_instance = 0;
            }

            static T* getInstance()
            {
                return m_instance;
            }

        private:
            static T* m_instance;
        };

        template<class T>
        T* Singleton<T>::m_instance = NULL;
    }
}
Revision:	1096
Committed:	Sat Dec 30 14:40:33 2017 UTC (7 years, 9 months ago) by s10k
Content type:	text/x-chdr
File size:	9751 byte(s)
Log Message:	Added zlib, quazip, basicxmlsyntaxhighlighter, conditionalsemaphore and linenumberdisplay libraries. zlib and quazip are pre-compiled, but you can compile them yourself, just delete the dll files (or equivalent binary files to your OS)
#	Content
1	#pragma once
2	#include <cassert>
3	#include <cstring>
4	#include <cstdio>
5	#include <sstream>
6	#include <fcntl.h>
7	#include <sys/stat.h>
8
9	#ifndef PLOG_ENABLE_WCHAR_INPUT
10	# ifdef _WIN32
11	# define PLOG_ENABLE_WCHAR_INPUT 1
12	# else
13	# define PLOG_ENABLE_WCHAR_INPUT 0
14	# endif
15	#endif
16
17	#ifdef _WIN32
18	# include <plog/WinApi.h>
19	# include <time.h>
20	# include <sys/timeb.h>
21	# include <io.h>
22	# include <share.h>
23	#else
24	# include <unistd.h>
25	# include <sys/syscall.h>
26	# include <sys/time.h>
27	# include <pthread.h>
28	# if PLOG_ENABLE_WCHAR_INPUT
29	# include <iconv.h>
30	# endif
31	#endif
32
33	#ifdef _WIN32
34	# define _PLOG_NSTR(x) L##x
35	# define PLOG_NSTR(x) _PLOG_NSTR(x)
36	#else
37	# define PLOG_NSTR(x) x
38	#endif
39
40	namespace plog
41	{
42	namespace util
43	{
44	#ifdef _WIN32
45	typedef std::wstring nstring;
46	typedef std::wstringstream nstringstream;
47	typedef wchar_t nchar;
48	#else
49	typedef std::string nstring;
50	typedef std::stringstream nstringstream;
51	typedef char nchar;
52	#endif
53
54	inline void localtime_s(struct tm* t, const time_t* time)
55	{
56	#if defined(_WIN32) && defined(__BORLANDC__)
57	::localtime_s(time, t);
58	#elif defined(_WIN32) && defined(__MINGW32__) && !defined(__MINGW64_VERSION_MAJOR)
59	t = ::localtime(time);
60	#elif defined(_WIN32)
61	::localtime_s(t, time);
62	#else
63	::localtime_r(time, t);
64	#endif
65	}
66
67	#ifdef _WIN32
68	typedef timeb Time;
69
70	inline void ftime(Time* t)
71	{
72	::ftime(t);
73	}
74	#else
75	struct Time
76	{
77	time_t time;
78	unsigned short millitm;
79	};
80
81	inline void ftime(Time* t)
82	{
83	timeval tv;
84	::gettimeofday(&tv, NULL);
85
86	t->time = tv.tv_sec;
87	t->millitm = static_cast<unsigned short>(tv.tv_usec / 1000);
88	}
89	#endif
90
91	inline unsigned int gettid()
92	{
93	#ifdef _WIN32
94	return GetCurrentThreadId();
95	#elif defined(__unix__)
96	return static_cast<unsigned int>(::syscall(__NR_gettid));
97	#elif defined(__APPLE__)
98	uint64_t tid64;
99	pthread_threadid_np(NULL, &tid64);
100	return static_cast<unsigned int>(tid64);
101	#endif
102	}
103
104	#if PLOG_ENABLE_WCHAR_INPUT && !defined(_WIN32)
105	inline std::string toNarrow(const wchar_t* wstr)
106	{
107	size_t wlen = ::wcslen(wstr);
108	std::string str(wlen * sizeof(wchar_t), 0);
109
110	if (!str.empty())
111	{
112	const char* in = reinterpret_cast<const char*>(&wstr[0]);
113	char* out = &str[0];
114	size_t inBytes = wlen * sizeof(wchar_t);
115	size_t outBytes = str.size();
116
117	iconv_t cd = ::iconv_open("UTF-8", "WCHAR_T");
118	::iconv(cd, const_cast<char**>(&in), &inBytes, &out, &outBytes);
119	::iconv_close(cd);
120
121	str.resize(str.size() - outBytes);
122	}
123
124	return str;
125	}
126	#endif
127
128	#ifdef _WIN32
129	inline std::wstring toWide(const char* str)
130	{
131	size_t len = ::strlen(str);
132	std::wstring wstr(len, 0);
133
134	if (!wstr.empty())
135	{
136	int wlen = MultiByteToWideChar(codePage::kActive, 0, str, static_cast<int>(len), &wstr[0], static_cast<int>(wstr.size()));
137	wstr.resize(wlen);
138	}
139
140	return wstr;
141	}
142
143	inline std::string toUTF8(const std::wstring& wstr)
144	{
145	std::string str(wstr.size() * sizeof(wchar_t), 0);
146
147	if (!str.empty())
148	{
149	int len = WideCharToMultiByte(codePage::kUTF8, 0, wstr.c_str(), static_cast<int>(wstr.size()), &str[0], static_cast<int>(str.size()), 0, 0);
150	str.resize(len);
151	}
152
153	return str;
154	}
155	#endif
156
157	inline std::string processFuncName(const char* func)
158	{
159	#if (defined(_WIN32) && !defined(__MINGW32__)) \|\| defined(__OBJC__)
160	return std::string(func);
161	#else
162	const char* funcBegin = func;
163	const char* funcEnd = ::strchr(funcBegin, '(');
164
165	if (!funcEnd)
166	{
167	return std::string(func);
168	}
169
170	for (const char* i = funcEnd - 1; i >= funcBegin; --i) // search backwards for the first space char
171	{
172	if (*i == ' ')
173	{
174	funcBegin = i + 1;
175	break;
176	}
177	}
178
179	return std::string(funcBegin, funcEnd);
180	#endif
181	}
182
183	inline const nchar* findExtensionDot(const nchar* fileName)
184	{
185	#ifdef _WIN32
186	return std::wcsrchr(fileName, L'.');
187	#else
188	return std::strrchr(fileName, '.');
189	#endif
190	}
191
192	inline void splitFileName(const nchar* fileName, nstring& fileNameNoExt, nstring& fileExt)
193	{
194	const nchar* dot = findExtensionDot(fileName);
195
196	if (dot)
197	{
198	fileNameNoExt.assign(fileName, dot);
199	fileExt.assign(dot + 1);
200	}
201	else
202	{
203	fileNameNoExt.assign(fileName);
204	fileExt.clear();
205	}
206	}
207
208	class NonCopyable
209	{
210	protected:
211	NonCopyable()
212	{
213	}
214
215	private:
216	NonCopyable(const NonCopyable&);
217	NonCopyable& operator=(const NonCopyable&);
218	};
219
220	class File : NonCopyable
221	{
222	public:
223	File() : m_file(-1)
224	{
225	}
226
227	File(const nchar* fileName) : m_file(-1)
228	{
229	open(fileName);
230	}
231
232	~File()
233	{
234	close();
235	}
236
237	off_t open(const nchar* fileName)
238	{
239	#if defined(_WIN32) && (defined(__BORLANDC__) \|\| defined(__MINGW32__))
240	m_file = ::_wsopen(fileName, _O_CREAT \| _O_WRONLY \| _O_BINARY, SH_DENYWR, _S_IREAD \| _S_IWRITE);
241	#elif defined(_WIN32)
242	::_wsopen_s(&m_file, fileName, _O_CREAT \| _O_WRONLY \| _O_BINARY, _SH_DENYWR, _S_IREAD \| _S_IWRITE);
243	#else
244	m_file = ::open(fileName, O_CREAT \| O_WRONLY, S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IROTH);
245	#endif
246	return seek(0, SEEK_END);
247	}
248
249	int write(const void* buf, size_t count)
250	{
251	#ifdef _WIN32
252	return m_file != -1 ? ::_write(m_file, buf, static_cast<unsigned int>(count)) : -1;
253	#else
254	return m_file != -1 ? static_cast<int>(::write(m_file, buf, count)) : -1;
255	#endif
256	}
257
258	template<class CharType>
259	int write(const std::basic_string<CharType>& str)
260	{
261	return write(str.data(), str.size() * sizeof(CharType));
262	}
263
264	off_t seek(off_t offset, int whence)
265	{
266	#ifdef _WIN32
267	return m_file != -1 ? ::_lseek(m_file, offset, whence) : -1;
268	#else
269	return m_file != -1 ? ::lseek(m_file, offset, whence) : -1;
270	#endif
271	}
272
273	void close()
274	{
275	if (m_file != -1)
276	{
277	#ifdef _WIN32
278	::_close(m_file);
279	#else
280	::close(m_file);
281	#endif
282	m_file = -1;
283	}
284	}
285
286	static int unlink(const nchar* fileName)
287	{
288	#ifdef _WIN32
289	return ::_wunlink(fileName);
290	#else
291	return ::unlink(fileName);
292	#endif
293	}
294
295	static int rename(const nchar* oldFilename, const nchar* newFilename)
296	{
297	#ifdef _WIN32
298	return MoveFileW(oldFilename, newFilename);
299	#else
300	return ::rename(oldFilename, newFilename);
301	#endif
302	}
303
304	private:
305	int m_file;
306	};
307
308	class Mutex : NonCopyable
309	{
310	public:
311	Mutex()
312	{
313	#ifdef _WIN32
314	InitializeCriticalSection(&m_sync);
315	#else
316	::pthread_mutex_init(&m_sync, 0);
317	#endif
318	}
319
320	~Mutex()
321	{
322	#ifdef _WIN32
323	DeleteCriticalSection(&m_sync);
324	#else
325	::pthread_mutex_destroy(&m_sync);
326	#endif
327	}
328
329	friend class MutexLock;
330
331	private:
332	void lock()
333	{
334	#ifdef _WIN32
335	EnterCriticalSection(&m_sync);
336	#else
337	::pthread_mutex_lock(&m_sync);
338	#endif
339	}
340
341	void unlock()
342	{
343	#ifdef _WIN32
344	LeaveCriticalSection(&m_sync);
345	#else
346	::pthread_mutex_unlock(&m_sync);
347	#endif
348	}
349
350	private:
351	#ifdef _WIN32
352	CRITICAL_SECTION m_sync;
353	#else
354	pthread_mutex_t m_sync;
355	#endif
356	};
357
358	class MutexLock : NonCopyable
359	{
360	public:
361	MutexLock(Mutex& mutex) : m_mutex(mutex)
362	{
363	m_mutex.lock();
364	}
365
366	~MutexLock()
367	{
368	m_mutex.unlock();
369	}
370
371	private:
372	Mutex& m_mutex;
373	};
374
375	template<class T>
376	class Singleton : NonCopyable
377	{
378	public:
379	Singleton()
380	{
381	assert(!m_instance);
382	m_instance = static_cast<T*>(this);
383	}
384
385	~Singleton()
386	{
387	assert(m_instance);
388	m_instance = 0;
389	}
390
391	static T* getInstance()
392	{
393	return m_instance;
394	}
395
396	private:
397	static T* m_instance;
398	};
399
400	template<class T>
401	T* Singleton<T>::m_instance = NULL;
402	}
403	}