root/trunk/src/shared/Util.cpp @ 2

Revision 2, 11.8 kB (checked in by yumileroy, 17 years ago)

[svn] * Proper SVN structure

Original author: Neo2003
Date: 2008-10-02 16:23:55-05:00
Line 
1/*
2 * Copyright (C) 2005-2008 MaNGOS <http://www.mangosproject.org/>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17 */
18
19#include "Util.h"
20
21#include "sockets/socket_include.h"
22#include "utf8cpp/utf8.h"
23#include "mersennetwister/MersenneTwister.h"
24#include "zthread/ThreadLocal.h"
25
26typedef ZThread::ThreadLocal<MTRand> MTRandTSS;
27
28/* NOTE: Not sure if static initialization is ok for TSS objects ,
29 * as I see zthread uses custom implementation of the TSS
30 * ,and in the consturctor there is no code ,so I suppose its ok
31 * If its not ok ,change it to use singleton.
32 */
33static MTRandTSS mtRand;
34
35int32 irand (int32 min, int32 max)
36{
37  return int32 (mtRand.get ().randInt (max - min)) + min;
38}
39
40uint32 urand (uint32 min, uint32 max)
41{
42  return mtRand.get ().randInt (max - min) + min;
43}
44
45int32 rand32 ()
46{
47  return mtRand.get ().randInt ();
48}
49
50double rand_norm(void)
51{
52  return mtRand.get ().randExc ();
53}
54
55double rand_chance (void)
56{
57  return mtRand.get ().randExc (100.0);
58}
59
60Tokens StrSplit(const std::string &src, const std::string &sep)
61{
62    Tokens r;
63    std::string s;
64    for (std::string::const_iterator i = src.begin(); i != src.end(); i++)
65    {
66        if (sep.find(*i) != std::string::npos)
67        {
68            if (s.length()) r.push_back(s);
69            s = "";
70        }
71        else
72        {
73            s += *i;
74        }
75    }
76    if (s.length()) r.push_back(s);
77    return r;
78}
79
80void stripLineInvisibleChars(std::string &str)
81{
82    static std::string invChars = " \t\7";
83
84    size_t wpos = 0;
85
86    bool space = false;
87    for(size_t pos = 0; pos < str.size(); ++pos)
88    {
89        if(invChars.find(str[pos])!=std::string::npos)
90        {
91            if(!space)
92            {
93                str[wpos++] = ' ';
94                space = true;
95            }
96        }
97        else
98        {
99            if(wpos!=pos)
100                str[wpos++] = str[pos];
101            else
102                ++wpos;
103            space = false;
104        }
105    }
106
107    if(wpos < str.size())
108        str.erase(wpos,str.size());
109}
110
111std::string secsToTimeString(uint32 timeInSecs, bool shortText, bool hoursOnly)
112{
113    uint32 secs    = timeInSecs % MINUTE;
114    uint32 minutes = timeInSecs % HOUR / MINUTE;
115    uint32 hours   = timeInSecs % DAY  / HOUR;
116    uint32 days    = timeInSecs / DAY;
117
118    std::ostringstream ss;
119    if(days)
120        ss << days << (shortText ? "d" : " Day(s) ");
121    if(hours || hoursOnly)
122        ss << hours << (shortText ? "h" : " Hour(s) ");
123    if(!hoursOnly)
124    {
125        if(minutes)
126            ss << minutes << (shortText ? "m" : " Minute(s) ");
127        if(secs || (!days && !hours && !minutes) )
128            ss << secs << (shortText ? "s" : " Second(s).");
129    }
130
131    return ss.str();
132}
133
134uint32 TimeStringToSecs(std::string timestring)
135{
136    uint32 secs       = 0;
137    uint32 buffer     = 0;
138    uint32 multiplier = 0;
139
140    for(std::string::iterator itr = timestring.begin(); itr != timestring.end(); itr++ )
141    {
142        if(isdigit(*itr))
143        {
144            std::string str;                                //very complicated typecast char->const char*; is there no better way?
145            str += *itr;
146            const char* tmp = str.c_str();
147
148            buffer*=10;
149            buffer+=atoi(tmp);
150        }
151        else
152        {
153            switch(*itr)
154            {
155                case 'd': multiplier = DAY;     break;
156                case 'h': multiplier = HOUR;    break;
157                case 'm': multiplier = MINUTE;  break;
158                case 's': multiplier = 1;       break;
159                default : return 0;                         //bad format
160            }
161            buffer*=multiplier;
162            secs+=buffer;
163            buffer=0;
164        }
165    }
166
167    return secs;
168}
169
170std::string TimeToTimestampStr(time_t t)
171{
172    tm* aTm = localtime(&t);
173    //       YYYY   year
174    //       MM     month (2 digits 01-12)
175    //       DD     day (2 digits 01-31)
176    //       HH     hour (2 digits 00-23)
177    //       MM     minutes (2 digits 00-59)
178    //       SS     seconds (2 digits 00-59)
179    char buf[20];
180    snprintf(buf,20,"%04d-%02d-%02d_%02d-%02d-%02d",aTm->tm_year+1900,aTm->tm_mon+1,aTm->tm_mday,aTm->tm_hour,aTm->tm_min,aTm->tm_sec);
181    return std::string(buf);
182}
183
184/// Check if the string is a valid ip address representation
185bool IsIPAddress(char const* ipaddress)
186{
187    if(!ipaddress)
188        return false;
189
190    // Let the big boys do it.
191    // Drawback: all valid ip address formats are recognized e.g.: 12.23,121234,0xABCD)
192    return inet_addr(ipaddress) != INADDR_NONE;
193}
194
195/// create PID file
196uint32 CreatePIDFile(std::string filename)
197{
198    FILE * pid_file = fopen (filename.c_str(), "w" );
199    if (pid_file == NULL)
200        return 0;
201
202#ifdef WIN32
203    DWORD pid = GetCurrentProcessId();
204#else
205    pid_t pid = getpid();
206#endif
207
208    fprintf(pid_file, "%d", pid );
209    fclose(pid_file);
210
211    return (uint32)pid;
212}
213
214size_t utf8length(std::string& utf8str)
215{
216    try
217    {
218        return utf8::distance(utf8str.c_str(),utf8str.c_str()+utf8str.size());
219    }
220    catch(std::exception)
221    {
222        utf8str = "";
223        return 0;
224    }
225}
226
227void utf8truncate(std::string& utf8str,size_t len)
228{
229    try
230    {
231        size_t wlen = utf8::distance(utf8str.c_str(),utf8str.c_str()+utf8str.size());
232        if(wlen <= len)
233            return;
234
235        std::wstring wstr;
236        wstr.resize(wlen);
237        utf8::utf8to16(utf8str.c_str(),utf8str.c_str()+utf8str.size(),&wstr[0]);
238        wstr.resize(len);
239        char* oend = utf8::utf16to8(wstr.c_str(),wstr.c_str()+wstr.size(),&utf8str[0]);
240        utf8str.resize(oend-(&utf8str[0]));                 // remove unused tail
241    }
242    catch(std::exception)
243    {
244        utf8str = "";
245    }
246}
247
248bool Utf8toWStr(char const* utf8str, size_t csize, wchar_t* wstr, size_t& wsize)
249{
250    try
251    {
252        size_t len = utf8::distance(utf8str,utf8str+csize);
253        if(len > wsize)
254        {
255            wsize = 0;
256            wstr = L"";
257            return false;
258        }
259
260        wsize = len;
261        utf8::utf8to16(utf8str,utf8str+csize,wstr);
262        wstr[len] = L'\0';
263    }
264    catch(std::exception)
265    {
266        wsize = 0;
267        wstr = L"";
268        return false;
269    }
270
271    return true;
272}
273
274bool Utf8toWStr(std::string utf8str, std::wstring& wstr)
275{
276    try
277    {
278        size_t len = utf8::distance(utf8str.c_str(),utf8str.c_str()+utf8str.size());
279        wstr.resize(len);
280
281        utf8::utf8to16(utf8str.c_str(),utf8str.c_str()+utf8str.size(),&wstr[0]);
282    }
283    catch(std::exception)
284    {
285        wstr = L"";
286        return false;
287    }
288
289    return true;
290}
291
292bool WStrToUtf8(wchar_t* wstr, size_t size, std::string& utf8str)
293{
294    try
295    {
296        std::string utf8str2;
297        utf8str2.resize(size*4);                            // allocate for most long case
298
299        char* oend = utf8::utf16to8(wstr,wstr+size,&utf8str2[0]);
300        utf8str2.resize(oend-(&utf8str2[0]));               // remove unused tail
301        utf8str = utf8str2;
302    }
303    catch(std::exception)
304    {
305        utf8str = "";
306        return false;
307    }
308
309    return true;
310}
311
312bool WStrToUtf8(std::wstring wstr, std::string& utf8str)
313{
314    try
315    {
316        std::string utf8str2;
317        utf8str2.resize(wstr.size()*4);                     // allocate for most long case
318
319        char* oend = utf8::utf16to8(wstr.c_str(),wstr.c_str()+wstr.size(),&utf8str2[0]);
320        utf8str2.resize(oend-(&utf8str2[0]));                // remove unused tail
321        utf8str = utf8str2;
322    }
323    catch(std::exception)
324    {
325        utf8str = "";
326        return false;
327    }
328
329    return true;
330}
331
332typedef wchar_t const* const* wstrlist;
333
334std::wstring GetMainPartOfName(std::wstring wname, uint32 declension)
335{
336    // supported only Cyrillic cases
337    if(wname.size() < 1 || !isCyrillicCharacter(wname[0]) || declension > 5)
338        return wname;
339
340    // Important: end length must be <= MAX_INTERNAL_PLAYER_NAME-MAX_PLAYER_NAME (3 currently)
341
342    static wchar_t const a_End[]    = { wchar_t(1), wchar_t(0x0430),wchar_t(0x0000)};
343    static wchar_t const o_End[]    = { wchar_t(1), wchar_t(0x043E),wchar_t(0x0000)};
344    static wchar_t const ya_End[]   = { wchar_t(1), wchar_t(0x044F),wchar_t(0x0000)};
345    static wchar_t const ie_End[]   = { wchar_t(1), wchar_t(0x0435),wchar_t(0x0000)};
346    static wchar_t const i_End[]    = { wchar_t(1), wchar_t(0x0438),wchar_t(0x0000)};
347    static wchar_t const yeru_End[] = { wchar_t(1), wchar_t(0x044B),wchar_t(0x0000)};
348    static wchar_t const u_End[]    = { wchar_t(1), wchar_t(0x0443),wchar_t(0x0000)};
349    static wchar_t const yu_End[]   = { wchar_t(1), wchar_t(0x044E),wchar_t(0x0000)};
350    static wchar_t const oj_End[]   = { wchar_t(2), wchar_t(0x043E),wchar_t(0x0439),wchar_t(0x0000)};
351    static wchar_t const ie_j_End[] = { wchar_t(2), wchar_t(0x0435),wchar_t(0x0439),wchar_t(0x0000)};
352    static wchar_t const io_j_End[] = { wchar_t(2), wchar_t(0x0451),wchar_t(0x0439),wchar_t(0x0000)};
353    static wchar_t const o_m_End[]  = { wchar_t(2), wchar_t(0x043E),wchar_t(0x043C),wchar_t(0x0000)};
354    static wchar_t const io_m_End[] = { wchar_t(2), wchar_t(0x0451),wchar_t(0x043C),wchar_t(0x0000)};
355    static wchar_t const ie_m_End[] = { wchar_t(2), wchar_t(0x0435),wchar_t(0x043C),wchar_t(0x0000)};
356    static wchar_t const soft_End[] = { wchar_t(1), wchar_t(0x044C),wchar_t(0x0000)};
357    static wchar_t const j_End[]    = { wchar_t(1), wchar_t(0x0439),wchar_t(0x0000)};
358
359    static wchar_t const* const dropEnds[6][8] = {
360        { &a_End[1],  &o_End[1],    &ya_End[1],   &ie_End[1],  &soft_End[1], &j_End[1],    NULL,       NULL },
361        { &a_End[1],  &ya_End[1],   &yeru_End[1], &i_End[1],   NULL,         NULL,         NULL,       NULL },
362        { &ie_End[1], &u_End[1],    &yu_End[1],   &i_End[1],   NULL,         NULL,         NULL,       NULL },
363        { &u_End[1],  &yu_End[1],   &o_End[1],    &ie_End[1],  &soft_End[1], &ya_End[1],   &a_End[1],  NULL },
364        { &oj_End[1], &io_j_End[1], &ie_j_End[1], &o_m_End[1], &io_m_End[1], &ie_m_End[1], &yu_End[1], NULL },
365        { &ie_End[1], &i_End[1],    NULL,         NULL,        NULL,         NULL,         NULL,       NULL }
366    };
367
368    for(wchar_t const * const* itr = &dropEnds[declension][0]; *itr; ++itr)
369    {
370        size_t len = size_t((*itr)[-1]);                    // get length from string size field
371
372        if(wname.substr(wname.size()-len,len)==*itr)
373            return wname.substr(0,wname.size()-len);
374    }
375
376    return wname;
377}
378
379bool utf8ToConsole(std::string utf8str, std::string& conStr)
380{
381#if PLATFORM == PLATFORM_WINDOWS
382    std::wstring wstr;
383    if(!Utf8toWStr(utf8str,wstr))
384        return false;
385
386    conStr.resize(wstr.size());
387    CharToOemBuffW(&wstr[0],&conStr[0],wstr.size());
388#else
389    // not implemented yet
390    conStr = utf8str;
391#endif
392
393    return true;
394}
395
396bool consoleToUtf8(std::string conStr,std::string& utf8str)
397{
398#if PLATFORM == PLATFORM_WINDOWS
399    std::wstring wstr;
400    wstr.resize(conStr.size());
401    OemToCharBuffW(&conStr[0],&wstr[0],conStr.size());
402
403    return WStrToUtf8(wstr,utf8str);
404#else
405    // not implemented yet
406    utf8str = conStr;
407    return true;
408#endif
409}
410
411bool Utf8FitTo(std::string str, std::wstring search)
412{
413    std::wstring temp;
414
415    if(!Utf8toWStr(str,temp))
416        return false;
417
418    // converting to lower case
419    wstrToLower( temp );
420
421    if(temp.find(search) == std::wstring::npos)
422        return false;
423
424    return true;
425}
Note: See TracBrowser for help on using the browser.