string.cpp 22 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147
  1. /*
  2. * Copyright 2010-2018 Branimir Karadzic. All rights reserved.
  3. * License: https://github.com/bkaradzic/bx#license-bsd-2-clause
  4. */
  5. #include "bx_p.h"
  6. #include <bx/allocator.h>
  7. #include <bx/hash.h>
  8. #include <bx/readerwriter.h>
  9. #include <bx/string.h>
  10. #if !BX_CRT_NONE
  11. # include <stdio.h> // vsnprintf
  12. #endif // !BX_CRT_NONE
  13. namespace bx
  14. {
  15. inline bool isInRange(char _ch, char _from, char _to)
  16. {
  17. return unsigned(_ch - _from) <= unsigned(_to-_from);
  18. }
  19. bool isSpace(char _ch)
  20. {
  21. return ' ' == _ch
  22. || '\t' == _ch
  23. || '\n' == _ch
  24. || '\v' == _ch
  25. || '\f' == _ch
  26. || '\r' == _ch
  27. ;
  28. }
  29. bool isUpper(char _ch)
  30. {
  31. return isInRange(_ch, 'A', 'Z');
  32. }
  33. bool isLower(char _ch)
  34. {
  35. return isInRange(_ch, 'a', 'z');
  36. }
  37. bool isAlpha(char _ch)
  38. {
  39. return isLower(_ch) || isUpper(_ch);
  40. }
  41. bool isNumeric(char _ch)
  42. {
  43. return isInRange(_ch, '0', '9');
  44. }
  45. bool isAlphaNum(char _ch)
  46. {
  47. return false
  48. || isAlpha(_ch)
  49. || isNumeric(_ch)
  50. ;
  51. }
  52. bool isHexNum(char _ch)
  53. {
  54. return false
  55. || isInRange(toLower(_ch), 'a', 'f')
  56. || isNumeric(_ch)
  57. ;
  58. }
  59. bool isPrint(char _ch)
  60. {
  61. return isInRange(_ch, ' ', '~');
  62. }
  63. typedef bool (*CharTestFn)(char _ch);
  64. template<CharTestFn fn>
  65. inline bool isCharTest(const StringView& _str)
  66. {
  67. bool result = true;
  68. for (const char* ptr = _str.getPtr(), *term = _str.getTerm()
  69. ; ptr != term && result
  70. ; ++ptr
  71. )
  72. {
  73. result &= fn(*ptr);
  74. }
  75. return result;
  76. }
  77. bool isSpace(const StringView& _str)
  78. {
  79. return isCharTest<isSpace>(_str);
  80. }
  81. bool isUpper(const StringView& _str)
  82. {
  83. return isCharTest<isUpper>(_str);
  84. }
  85. bool isLower(const StringView& _str)
  86. {
  87. return isCharTest<isLower>(_str);
  88. }
  89. bool isAlpha(const StringView& _str)
  90. {
  91. return isCharTest<isAlpha>(_str);
  92. }
  93. bool isNumeric(const StringView& _str)
  94. {
  95. return isCharTest<isNumeric>(_str);
  96. }
  97. bool isAlphaNum(const StringView& _str)
  98. {
  99. return isCharTest<isAlphaNum>(_str);
  100. }
  101. bool isHexNum(const StringView& _str)
  102. {
  103. return isCharTest<isHexNum>(_str);
  104. }
  105. bool isPrint(const StringView& _str)
  106. {
  107. return isCharTest<isPrint>(_str);
  108. }
  109. char toLower(char _ch)
  110. {
  111. return _ch + (isUpper(_ch) ? 0x20 : 0);
  112. }
  113. void toLowerUnsafe(char* _inOutStr, int32_t _len)
  114. {
  115. for (int32_t ii = 0; ii < _len; ++ii)
  116. {
  117. *_inOutStr = toLower(*_inOutStr);
  118. }
  119. }
  120. void toLower(char* _inOutStr, int32_t _max)
  121. {
  122. const int32_t len = strLen(_inOutStr, _max);
  123. toLowerUnsafe(_inOutStr, len);
  124. }
  125. char toUpper(char _ch)
  126. {
  127. return _ch - (isLower(_ch) ? 0x20 : 0);
  128. }
  129. void toUpperUnsafe(char* _inOutStr, int32_t _len)
  130. {
  131. for (int32_t ii = 0; ii < _len; ++ii)
  132. {
  133. *_inOutStr = toUpper(*_inOutStr);
  134. }
  135. }
  136. void toUpper(char* _inOutStr, int32_t _max)
  137. {
  138. const int32_t len = strLen(_inOutStr, _max);
  139. toUpperUnsafe(_inOutStr, len);
  140. }
  141. typedef char (*CharFn)(char _ch);
  142. inline char toNoop(char _ch)
  143. {
  144. return _ch;
  145. }
  146. template<CharFn fn>
  147. inline int32_t strCmp(const char* _lhs, int32_t _lhsMax, const char* _rhs, int32_t _rhsMax)
  148. {
  149. int32_t max = min(_lhsMax, _rhsMax);
  150. for (
  151. ; 0 < max && fn(*_lhs) == fn(*_rhs)
  152. ; ++_lhs, ++_rhs, --max
  153. )
  154. {
  155. if (*_lhs == '\0'
  156. || *_rhs == '\0')
  157. {
  158. break;
  159. }
  160. }
  161. return 0 == max && _lhsMax == _rhsMax ? 0 : fn(*_lhs) - fn(*_rhs);
  162. }
  163. int32_t strCmp(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  164. {
  165. return strCmp<toNoop>(
  166. _lhs.getPtr()
  167. , min(_lhs.getLength(), _max)
  168. , _rhs.getPtr()
  169. , min(_rhs.getLength(), _max)
  170. );
  171. }
  172. int32_t strCmpI(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  173. {
  174. return strCmp<toLower>(
  175. _lhs.getPtr()
  176. , min(_lhs.getLength(), _max)
  177. , _rhs.getPtr()
  178. , min(_rhs.getLength(), _max)
  179. );
  180. }
  181. inline int32_t strCmpV(const char* _lhs, int32_t _lhsMax, const char* _rhs, int32_t _rhsMax)
  182. {
  183. int32_t max = min(_lhsMax, _rhsMax);
  184. int32_t ii = 0;
  185. int32_t idx = 0;
  186. bool zero = true;
  187. for (
  188. ; 0 < max && _lhs[ii] == _rhs[ii]
  189. ; ++ii, --max
  190. )
  191. {
  192. const uint8_t ch = _lhs[ii];
  193. if ('\0' == ch
  194. || '\0' == _rhs[ii])
  195. {
  196. break;
  197. }
  198. if (!isNumeric(ch) )
  199. {
  200. idx = ii+1;
  201. zero = true;
  202. }
  203. else if ('0' != ch)
  204. {
  205. zero = false;
  206. }
  207. }
  208. if (0 == max)
  209. {
  210. return _lhsMax == _rhsMax ? 0 : _lhs[ii] - _rhs[ii];
  211. }
  212. if ('0' != _lhs[idx]
  213. && '0' != _rhs[idx])
  214. {
  215. int32_t jj = 0;
  216. for (jj = ii
  217. ; 0 < max && isNumeric(_lhs[jj])
  218. ; ++jj, --max
  219. )
  220. {
  221. if (!isNumeric(_rhs[jj]) )
  222. {
  223. return 1;
  224. }
  225. }
  226. if (isNumeric(_rhs[jj]))
  227. {
  228. return -1;
  229. }
  230. }
  231. else if (zero
  232. && idx < ii
  233. && (isNumeric(_lhs[ii]) || isNumeric(_rhs[ii]) ) )
  234. {
  235. return (_lhs[ii] - '0') - (_rhs[ii] - '0');
  236. }
  237. return 0 == max && _lhsMax == _rhsMax ? 0 : _lhs[ii] - _rhs[ii];
  238. }
  239. int32_t strCmpV(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  240. {
  241. return strCmpV(
  242. _lhs.getPtr()
  243. , min(_lhs.getLength(), _max)
  244. , _rhs.getPtr()
  245. , min(_rhs.getLength(), _max)
  246. );
  247. }
  248. int32_t strLen(const char* _str, int32_t _max)
  249. {
  250. if (NULL == _str)
  251. {
  252. return 0;
  253. }
  254. const char* ptr = _str;
  255. for (; 0 < _max && *ptr != '\0'; ++ptr, --_max) {};
  256. return int32_t(ptr - _str);
  257. }
  258. int32_t strLen(const StringView& _str, int32_t _max)
  259. {
  260. return strLen(_str.getPtr(), min(_str.getLength(), _max) );
  261. }
  262. inline int32_t strCopy(char* _dst, int32_t _dstSize, const char* _src, int32_t _num)
  263. {
  264. BX_CHECK(NULL != _dst, "_dst can't be NULL!");
  265. BX_CHECK(NULL != _src, "_src can't be NULL!");
  266. BX_CHECK(0 < _dstSize, "_dstSize can't be 0!");
  267. const int32_t len = strLen(_src, _num);
  268. const int32_t max = _dstSize-1;
  269. const int32_t num = (len < max ? len : max);
  270. memCopy(_dst, _src, num);
  271. _dst[num] = '\0';
  272. return num;
  273. }
  274. int32_t strCopy(char* _dst, int32_t _dstSize, const StringView& _str, int32_t _num)
  275. {
  276. return strCopy(_dst, _dstSize, _str.getPtr(), min(_str.getLength(), _num) );
  277. }
  278. inline int32_t strCat(char* _dst, int32_t _dstSize, const char* _src, int32_t _num)
  279. {
  280. BX_CHECK(NULL != _dst, "_dst can't be NULL!");
  281. BX_CHECK(NULL != _src, "_src can't be NULL!");
  282. BX_CHECK(0 < _dstSize, "_dstSize can't be 0!");
  283. const int32_t max = _dstSize;
  284. const int32_t len = strLen(_dst, max);
  285. return strCopy(&_dst[len], max-len, _src, _num);
  286. }
  287. int32_t strCat(char* _dst, int32_t _dstSize, const StringView& _str, int32_t _num)
  288. {
  289. return strCat(_dst, _dstSize, _str.getPtr(), min(_str.getLength(), _num) );
  290. }
  291. inline const char* strFindUnsafe(const char* _str, int32_t _len, char _ch)
  292. {
  293. for (int32_t ii = 0; ii < _len; ++ii)
  294. {
  295. if (_str[ii] == _ch)
  296. {
  297. return &_str[ii];
  298. }
  299. }
  300. return NULL;
  301. }
  302. inline const char* strFind(const char* _str, int32_t _max, char _ch)
  303. {
  304. return strFindUnsafe(_str, strLen(_str, _max), _ch);
  305. }
  306. const char* strFind(const StringView& _str, char _ch)
  307. {
  308. return strFind(_str.getPtr(), _str.getLength(), _ch);
  309. }
  310. inline const char* strRFindUnsafe(const char* _str, int32_t _len, char _ch)
  311. {
  312. for (int32_t ii = _len; 0 <= ii; --ii)
  313. {
  314. if (_str[ii] == _ch)
  315. {
  316. return &_str[ii];
  317. }
  318. }
  319. return NULL;
  320. }
  321. inline const char* strRFind(const char* _str, int32_t _max, char _ch)
  322. {
  323. return strRFindUnsafe(_str, strLen(_str, _max), _ch);
  324. }
  325. const char* strRFind(const StringView& _str, char _ch)
  326. {
  327. return strRFind(_str.getPtr(), _str.getLength(), _ch);
  328. }
  329. template<CharFn fn>
  330. inline const char* strFind(const char* _str, int32_t _strMax, const char* _find, int32_t _findMax)
  331. {
  332. const char* ptr = _str;
  333. int32_t stringLen = strLen(_str, _strMax);
  334. const int32_t findLen = strLen(_find, _findMax);
  335. for (; stringLen >= findLen; ++ptr, --stringLen)
  336. {
  337. // Find start of the string.
  338. while (fn(*ptr) != fn(*_find) )
  339. {
  340. ++ptr;
  341. --stringLen;
  342. // Search pattern lenght can't be longer than the string.
  343. if (findLen > stringLen)
  344. {
  345. return NULL;
  346. }
  347. }
  348. // Set pointers.
  349. const char* string = ptr;
  350. const char* search = _find;
  351. // Start comparing.
  352. while (fn(*string++) == fn(*search++) )
  353. {
  354. // If end of the 'search' string is reached, all characters match.
  355. if ('\0' == *search)
  356. {
  357. return ptr;
  358. }
  359. }
  360. }
  361. return NULL;
  362. }
  363. const char* strFind(const StringView& _str, const StringView& _find, int32_t _num)
  364. {
  365. return strFind<toNoop>(
  366. _str.getPtr()
  367. , _str.getLength()
  368. , _find.getPtr()
  369. , min(_find.getLength(), _num)
  370. );
  371. }
  372. const char* strFindI(const StringView& _str, const StringView& _find, int32_t _num)
  373. {
  374. return strFind<toLower>(
  375. _str.getPtr()
  376. , _str.getLength()
  377. , _find.getPtr()
  378. , min(_find.getLength(), _num)
  379. );
  380. }
  381. StringView strLTrim(const StringView& _str, const StringView& _chars)
  382. {
  383. const char* ptr = _str.getPtr();
  384. const char* chars = _chars.getPtr();
  385. const uint32_t charsLen = _chars.getLength();
  386. for (uint32_t ii = 0, len = _str.getLength(); ii < len; ++ii)
  387. {
  388. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]) )
  389. {
  390. return StringView(ptr + ii, len-ii);
  391. }
  392. }
  393. return StringView();
  394. }
  395. StringView strRTrim(const StringView& _str, const StringView& _chars)
  396. {
  397. if (_str.isEmpty() )
  398. {
  399. return StringView();
  400. }
  401. const char* ptr = _str.getPtr();
  402. const char* chars = _chars.getPtr();
  403. const uint32_t charsLen = _chars.getLength();
  404. for (int32_t len = _str.getLength(), ii = len-1; 0 <= ii; --ii)
  405. {
  406. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]) )
  407. {
  408. return StringView(ptr, ii+1);
  409. }
  410. }
  411. return StringView();
  412. }
  413. StringView strTrim(const StringView& _str, const StringView& _chars)
  414. {
  415. return strLTrim(strRTrim(_str, _chars), _chars);
  416. }
  417. const char* strnl(const char* _str)
  418. {
  419. for (; '\0' != *_str; _str += strLen(_str, 1024) )
  420. {
  421. const char* eol = strFind(StringView(_str, 1024), "\r\n");
  422. if (NULL != eol)
  423. {
  424. return eol + 2;
  425. }
  426. eol = strFind(StringView(_str, 1024), "\n");
  427. if (NULL != eol)
  428. {
  429. return eol + 1;
  430. }
  431. }
  432. return _str;
  433. }
  434. const char* streol(const char* _str)
  435. {
  436. for (; '\0' != *_str; _str += strLen(_str, 1024) )
  437. {
  438. const char* eol = strFind(StringView(_str, 1024), "\r\n");
  439. if (NULL != eol)
  440. {
  441. return eol;
  442. }
  443. eol = strFind(StringView(_str, 1024), "\n");
  444. if (NULL != eol)
  445. {
  446. return eol;
  447. }
  448. }
  449. return _str;
  450. }
  451. const char* strws(const char* _str)
  452. {
  453. for (; isSpace(*_str); ++_str) {};
  454. return _str;
  455. }
  456. const char* strnws(const char* _str)
  457. {
  458. for (; !isSpace(*_str); ++_str) {};
  459. return _str;
  460. }
  461. const char* strSkipWord(const char* _str, int32_t _max)
  462. {
  463. for (char ch = *_str++; 0 < _max && (isAlphaNum(ch) || '_' == ch); ch = *_str++, --_max) {};
  464. return _str-1;
  465. }
  466. StringView strWord(const StringView& _str)
  467. {
  468. const char* ptr = _str.getPtr();
  469. const char* term = strSkipWord(ptr, _str.getLength() );
  470. return StringView(ptr, term);
  471. }
  472. const char* strmb(const char* _str, char _open, char _close)
  473. {
  474. int count = 0;
  475. for (char ch = *_str++; ch != '\0' && count >= 0; ch = *_str++)
  476. {
  477. if (ch == _open)
  478. {
  479. count++;
  480. }
  481. else if (ch == _close)
  482. {
  483. count--;
  484. if (0 == count)
  485. {
  486. return _str-1;
  487. }
  488. }
  489. }
  490. return NULL;
  491. }
  492. void eolLF(char* _out, int32_t _size, const char* _str)
  493. {
  494. if (0 < _size)
  495. {
  496. char* end = _out + _size - 1;
  497. for (char ch = *_str++; ch != '\0' && _out < end; ch = *_str++)
  498. {
  499. if ('\r' != ch)
  500. {
  501. *_out++ = ch;
  502. }
  503. }
  504. *_out = '\0';
  505. }
  506. }
  507. const char* findIdentifierMatch(const char* _str, const char* _word)
  508. {
  509. int32_t len = strLen(_word);
  510. const char* ptr = strFind(_str, _word);
  511. for (; NULL != ptr; ptr = strFind(ptr + len, _word) )
  512. {
  513. if (ptr != _str)
  514. {
  515. char ch = *(ptr - 1);
  516. if (isAlphaNum(ch) || '_' == ch)
  517. {
  518. continue;
  519. }
  520. }
  521. char ch = ptr[len];
  522. if (isAlphaNum(ch) || '_' == ch)
  523. {
  524. continue;
  525. }
  526. return ptr;
  527. }
  528. return ptr;
  529. }
  530. const char* findIdentifierMatch(const char* _str, const char* _words[])
  531. {
  532. for (const char* word = *_words; NULL != word; ++_words, word = *_words)
  533. {
  534. const char* match = findIdentifierMatch(_str, word);
  535. if (NULL != match)
  536. {
  537. return match;
  538. }
  539. }
  540. return NULL;
  541. }
  542. namespace
  543. {
  544. struct Param
  545. {
  546. Param()
  547. : width(0)
  548. , base(10)
  549. , prec(6)
  550. , fill(' ')
  551. , bits(0)
  552. , left(false)
  553. , upper(false)
  554. , spec(false)
  555. , sign(false)
  556. {
  557. }
  558. int32_t width;
  559. uint32_t base;
  560. uint32_t prec;
  561. char fill;
  562. uint8_t bits;
  563. bool left;
  564. bool upper;
  565. bool spec;
  566. bool sign;
  567. };
  568. static int32_t write(WriterI* _writer, const char* _str, int32_t _len, const Param& _param, Error* _err)
  569. {
  570. int32_t size = 0;
  571. int32_t len = (int32_t)strLen(_str, _len);
  572. int32_t padding = _param.width > len ? _param.width - len : 0;
  573. bool sign = _param.sign && len > 1 && _str[0] != '-';
  574. padding = padding > 0 ? padding - sign : 0;
  575. if (!_param.left)
  576. {
  577. size += writeRep(_writer, _param.fill, padding, _err);
  578. }
  579. if (NULL == _str)
  580. {
  581. size += write(_writer, "(null)", 6, _err);
  582. }
  583. else if (_param.upper)
  584. {
  585. for (int32_t ii = 0; ii < len; ++ii)
  586. {
  587. size += write(_writer, toUpper(_str[ii]), _err);
  588. }
  589. }
  590. else if (sign)
  591. {
  592. size += write(_writer, '+', _err);
  593. size += write(_writer, _str, len, _err);
  594. }
  595. else
  596. {
  597. size += write(_writer, _str, len, _err);
  598. }
  599. if (_param.left)
  600. {
  601. size += writeRep(_writer, _param.fill, padding, _err);
  602. }
  603. return size;
  604. }
  605. static int32_t write(WriterI* _writer, char _ch, const Param& _param, Error* _err)
  606. {
  607. return write(_writer, &_ch, 1, _param, _err);
  608. }
  609. static int32_t write(WriterI* _writer, const char* _str, const Param& _param, Error* _err)
  610. {
  611. return write(_writer, _str, INT32_MAX, _param, _err);
  612. }
  613. static int32_t write(WriterI* _writer, int32_t _i, const Param& _param, Error* _err)
  614. {
  615. char str[33];
  616. int32_t len = toString(str, sizeof(str), _i, _param.base);
  617. if (len == 0)
  618. {
  619. return 0;
  620. }
  621. return write(_writer, str, len, _param, _err);
  622. }
  623. static int32_t write(WriterI* _writer, int64_t _i, const Param& _param, Error* _err)
  624. {
  625. char str[33];
  626. int32_t len = toString(str, sizeof(str), _i, _param.base);
  627. if (len == 0)
  628. {
  629. return 0;
  630. }
  631. return write(_writer, str, len, _param, _err);
  632. }
  633. static int32_t write(WriterI* _writer, uint32_t _u, const Param& _param, Error* _err)
  634. {
  635. char str[33];
  636. int32_t len = toString(str, sizeof(str), _u, _param.base);
  637. if (len == 0)
  638. {
  639. return 0;
  640. }
  641. return write(_writer, str, len, _param, _err);
  642. }
  643. static int32_t write(WriterI* _writer, uint64_t _u, const Param& _param, Error* _err)
  644. {
  645. char str[33];
  646. int32_t len = toString(str, sizeof(str), _u, _param.base);
  647. if (len == 0)
  648. {
  649. return 0;
  650. }
  651. return write(_writer, str, len, _param, _err);
  652. }
  653. static int32_t write(WriterI* _writer, double _d, const Param& _param, Error* _err)
  654. {
  655. char str[1024];
  656. int32_t len = toString(str, sizeof(str), _d);
  657. if (len == 0)
  658. {
  659. return 0;
  660. }
  661. if (_param.upper)
  662. {
  663. toUpperUnsafe(str, len);
  664. }
  665. const char* dot = strFind(str, INT32_MAX, '.');
  666. if (NULL != dot)
  667. {
  668. const int32_t precLen = int32_t(
  669. dot
  670. + uint32_min(_param.prec + _param.spec, 1)
  671. + _param.prec
  672. - str
  673. );
  674. if (precLen > len)
  675. {
  676. for (int32_t ii = len; ii < precLen; ++ii)
  677. {
  678. str[ii] = '0';
  679. }
  680. str[precLen] = '\0';
  681. }
  682. len = precLen;
  683. }
  684. return write(_writer, str, len, _param, _err);
  685. }
  686. static int32_t write(WriterI* _writer, const void* _ptr, const Param& _param, Error* _err)
  687. {
  688. char str[35] = "0x";
  689. int32_t len = toString(str + 2, sizeof(str) - 2, uint32_t(uintptr_t(_ptr) ), 16);
  690. if (len == 0)
  691. {
  692. return 0;
  693. }
  694. len += 2;
  695. return write(_writer, str, len, _param, _err);
  696. }
  697. } // anonymous namespace
  698. int32_t write(WriterI* _writer, const char* _format, va_list _argList, Error* _err)
  699. {
  700. MemoryReader reader(_format, uint32_t(strLen(_format) ) );
  701. int32_t size = 0;
  702. while (_err->isOk() )
  703. {
  704. char ch = '\0';
  705. Error err;
  706. read(&reader, ch, &err);
  707. if (!_err->isOk()
  708. || !err.isOk() )
  709. {
  710. break;
  711. }
  712. else if ('%' == ch)
  713. {
  714. // %[flags][width][.precision][length sub-specifier]specifier
  715. read(&reader, ch);
  716. Param param;
  717. // flags
  718. while (' ' == ch
  719. || '-' == ch
  720. || '+' == ch
  721. || '0' == ch
  722. || '#' == ch)
  723. {
  724. switch (ch)
  725. {
  726. default:
  727. case ' ': param.fill = ' '; break;
  728. case '-': param.left = true; break;
  729. case '+': param.sign = true; break;
  730. case '0': param.fill = '0'; break;
  731. case '#': param.spec = true; break;
  732. }
  733. read(&reader, ch);
  734. }
  735. if (param.left)
  736. {
  737. param.fill = ' ';
  738. }
  739. // width
  740. if ('*' == ch)
  741. {
  742. read(&reader, ch);
  743. param.width = va_arg(_argList, int32_t);
  744. if (0 > param.width)
  745. {
  746. param.left = true;
  747. param.width = -param.width;
  748. }
  749. }
  750. else
  751. {
  752. while (isNumeric(ch) )
  753. {
  754. param.width = param.width * 10 + ch - '0';
  755. read(&reader, ch);
  756. }
  757. }
  758. // .precision
  759. if ('.' == ch)
  760. {
  761. read(&reader, ch);
  762. if ('*' == ch)
  763. {
  764. read(&reader, ch);
  765. param.prec = va_arg(_argList, int32_t);
  766. }
  767. else
  768. {
  769. param.prec = 0;
  770. while (isNumeric(ch) )
  771. {
  772. param.prec = param.prec * 10 + ch - '0';
  773. read(&reader, ch);
  774. }
  775. }
  776. }
  777. // length sub-specifier
  778. while ('h' == ch
  779. || 'I' == ch
  780. || 'l' == ch
  781. || 'j' == ch
  782. || 't' == ch
  783. || 'z' == ch)
  784. {
  785. switch (ch)
  786. {
  787. default: break;
  788. case 'j': param.bits = sizeof(intmax_t )*8; break;
  789. case 't': param.bits = sizeof(size_t )*8; break;
  790. case 'z': param.bits = sizeof(ptrdiff_t)*8; break;
  791. case 'h': case 'I': case 'l':
  792. switch (ch)
  793. {
  794. case 'h': param.bits = sizeof(short int)*8; break;
  795. case 'l': param.bits = sizeof(long int )*8; break;
  796. default: break;
  797. }
  798. read(&reader, ch);
  799. switch (ch)
  800. {
  801. case 'h': param.bits = sizeof(signed char )*8; break;
  802. case 'l': param.bits = sizeof(long long int)*8; break;
  803. case '3':
  804. case '6':
  805. read(&reader, ch);
  806. switch (ch)
  807. {
  808. case '2': param.bits = sizeof(int32_t)*8; break;
  809. case '4': param.bits = sizeof(int64_t)*8; break;
  810. default: break;
  811. }
  812. break;
  813. default: seek(&reader, -1); break;
  814. }
  815. break;
  816. }
  817. read(&reader, ch);
  818. }
  819. // specifier
  820. switch (toLower(ch) )
  821. {
  822. case 'c':
  823. size += write(_writer, char(va_arg(_argList, int32_t) ), param, _err);
  824. break;
  825. case 's':
  826. size += write(_writer, va_arg(_argList, const char*), param, _err);
  827. break;
  828. case 'o':
  829. param.base = 8;
  830. switch (param.bits)
  831. {
  832. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  833. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  834. }
  835. break;
  836. case 'i':
  837. case 'd':
  838. param.base = 10;
  839. switch (param.bits)
  840. {
  841. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  842. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  843. };
  844. break;
  845. case 'e':
  846. case 'f':
  847. case 'g':
  848. param.upper = isUpper(ch);
  849. size += write(_writer, va_arg(_argList, double), param, _err);
  850. break;
  851. case 'p':
  852. size += write(_writer, va_arg(_argList, void*), param, _err);
  853. break;
  854. case 'x':
  855. param.base = 16;
  856. param.upper = isUpper(ch);
  857. switch (param.bits)
  858. {
  859. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  860. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  861. }
  862. break;
  863. case 'u':
  864. param.base = 10;
  865. switch (param.bits)
  866. {
  867. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  868. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  869. }
  870. break;
  871. default:
  872. size += write(_writer, ch, _err);
  873. break;
  874. }
  875. }
  876. else
  877. {
  878. size += write(_writer, ch, _err);
  879. }
  880. }
  881. size += write(_writer, '\0', _err);
  882. return size;
  883. }
  884. int32_t write(WriterI* _writer, Error* _err, const char* _format, ...)
  885. {
  886. va_list argList;
  887. va_start(argList, _format);
  888. int32_t total = write(_writer, _format, argList, _err);
  889. va_end(argList);
  890. return total;
  891. }
  892. int32_t vsnprintfRef(char* _out, int32_t _max, const char* _format, va_list _argList)
  893. {
  894. if (1 < _max)
  895. {
  896. StaticMemoryBlockWriter writer(_out, uint32_t(_max-1) );
  897. _out[_max-1] = '\0';
  898. Error err;
  899. va_list argListCopy;
  900. va_copy(argListCopy, _argList);
  901. int32_t size = write(&writer, _format, argListCopy, &err);
  902. va_end(argListCopy);
  903. if (err.isOk() )
  904. {
  905. return size;
  906. }
  907. }
  908. Error err;
  909. SizerWriter sizer;
  910. va_list argListCopy;
  911. va_copy(argListCopy, _argList);
  912. int32_t size = write(&sizer, _format, argListCopy, &err);
  913. va_end(argListCopy);
  914. return size - 1 /* size without '\0' terminator */;
  915. }
  916. int32_t vsnprintf(char* _out, int32_t _max, const char* _format, va_list _argList)
  917. {
  918. va_list argList;
  919. va_copy(argList, _argList);
  920. int32_t total = 0;
  921. #if BX_CRT_NONE
  922. total = vsnprintfRef(_out, _max, _format, argList);
  923. #elif BX_CRT_MSVC
  924. int32_t len = -1;
  925. if (NULL != _out)
  926. {
  927. va_list argListCopy;
  928. va_copy(argListCopy, _argList);
  929. len = ::vsnprintf_s(_out, _max, size_t(-1), _format, argListCopy);
  930. va_end(argListCopy);
  931. }
  932. total = -1 == len ? ::_vscprintf(_format, argList) : len;
  933. #else
  934. total = ::vsnprintf(_out, _max, _format, argList);
  935. #endif // BX_COMPILER_MSVC
  936. va_end(argList);
  937. return total;
  938. }
  939. int32_t snprintf(char* _out, int32_t _max, const char* _format, ...)
  940. {
  941. va_list argList;
  942. va_start(argList, _format);
  943. int32_t total = vsnprintf(_out, _max, _format, argList);
  944. va_end(argList);
  945. return total;
  946. }
  947. static const char s_units[] = { 'B', 'k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y' };
  948. template<uint32_t Kilo, char KiloCh0, char KiloCh1, CharFn fn>
  949. inline int32_t prettify(char* _out, int32_t _count, uint64_t _value)
  950. {
  951. uint8_t idx = 0;
  952. double value = double(_value);
  953. while (_value != (_value&0x7ff)
  954. && idx < BX_COUNTOF(s_units) )
  955. {
  956. _value /= Kilo;
  957. value *= 1.0/double(Kilo);
  958. ++idx;
  959. }
  960. return snprintf(_out, _count, "%0.2f %c%c%c", value
  961. , fn(s_units[idx])
  962. , idx > 0 ? KiloCh0 : '\0'
  963. , KiloCh1
  964. );
  965. }
  966. int32_t prettify(char* _out, int32_t _count, uint64_t _value, Units::Enum _units)
  967. {
  968. if (Units::Kilo == _units)
  969. {
  970. return prettify<1000, 'B', '\0', toNoop>(_out, _count, _value);
  971. }
  972. return prettify<1024, 'i', 'B', toUpper>(_out, _count, _value);
  973. }
  974. } // namespace bx