string.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197
  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 StringView strFindEol(const StringView& _str)
  452. {
  453. StringView str(_str);
  454. for (; str.getPtr() != _str.getTerm()
  455. ; str = StringView(str.getPtr()+1024, min(str.getPtr()+1024, _str.getTerm() ) )
  456. )
  457. {
  458. const char* eol = strFind(str, "\r\n");
  459. if (NULL != eol)
  460. {
  461. return StringView(eol, _str.getTerm() );
  462. }
  463. eol = strFind(str, "\n");
  464. if (NULL != eol)
  465. {
  466. return StringView(eol, _str.getTerm() );
  467. }
  468. }
  469. return StringView(_str.getTerm(), _str.getTerm() );
  470. }
  471. const char* strws(const char* _str)
  472. {
  473. for (; isSpace(*_str); ++_str) {};
  474. return _str;
  475. }
  476. const StringView strSkipSpace(const StringView& _str)
  477. {
  478. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  479. {
  480. if (!isSpace(*ptr) )
  481. {
  482. return StringView(ptr, term);
  483. }
  484. }
  485. return StringView(_str.getTerm(), _str.getTerm() );
  486. }
  487. const char* strnws(const char* _str)
  488. {
  489. for (; !isSpace(*_str); ++_str) {};
  490. return _str;
  491. }
  492. const StringView strSkipNonSpace(const StringView& _str)
  493. {
  494. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  495. {
  496. if (isSpace(*ptr) )
  497. {
  498. return StringView(ptr, term);
  499. }
  500. }
  501. return StringView(_str.getTerm(), _str.getTerm() );
  502. }
  503. const char* strSkipWord(const char* _str, int32_t _max)
  504. {
  505. for (char ch = *_str++; 0 < _max && (isAlphaNum(ch) || '_' == ch); ch = *_str++, --_max) {};
  506. return _str-1;
  507. }
  508. StringView strWord(const StringView& _str)
  509. {
  510. const char* ptr = _str.getPtr();
  511. const char* term = strSkipWord(ptr, _str.getLength() );
  512. return StringView(ptr, term);
  513. }
  514. const char* strmb(const char* _str, char _open, char _close)
  515. {
  516. int count = 0;
  517. for (char ch = *_str++; ch != '\0' && count >= 0; ch = *_str++)
  518. {
  519. if (ch == _open)
  520. {
  521. count++;
  522. }
  523. else if (ch == _close)
  524. {
  525. count--;
  526. if (0 == count)
  527. {
  528. return _str-1;
  529. }
  530. }
  531. }
  532. return NULL;
  533. }
  534. void eolLF(char* _out, int32_t _size, const char* _str)
  535. {
  536. if (0 < _size)
  537. {
  538. char* end = _out + _size - 1;
  539. for (char ch = *_str++; ch != '\0' && _out < end; ch = *_str++)
  540. {
  541. if ('\r' != ch)
  542. {
  543. *_out++ = ch;
  544. }
  545. }
  546. *_out = '\0';
  547. }
  548. }
  549. const char* findIdentifierMatch(const char* _str, const char* _word)
  550. {
  551. int32_t len = strLen(_word);
  552. const char* ptr = strFind(_str, _word);
  553. for (; NULL != ptr; ptr = strFind(ptr + len, _word) )
  554. {
  555. if (ptr != _str)
  556. {
  557. char ch = *(ptr - 1);
  558. if (isAlphaNum(ch) || '_' == ch)
  559. {
  560. continue;
  561. }
  562. }
  563. char ch = ptr[len];
  564. if (isAlphaNum(ch) || '_' == ch)
  565. {
  566. continue;
  567. }
  568. return ptr;
  569. }
  570. return ptr;
  571. }
  572. const char* findIdentifierMatch(const char* _str, const char* _words[])
  573. {
  574. for (const char* word = *_words; NULL != word; ++_words, word = *_words)
  575. {
  576. const char* match = findIdentifierMatch(_str, word);
  577. if (NULL != match)
  578. {
  579. return match;
  580. }
  581. }
  582. return NULL;
  583. }
  584. namespace
  585. {
  586. struct Param
  587. {
  588. Param()
  589. : width(0)
  590. , base(10)
  591. , prec(6)
  592. , fill(' ')
  593. , bits(0)
  594. , left(false)
  595. , upper(false)
  596. , spec(false)
  597. , sign(false)
  598. {
  599. }
  600. int32_t width;
  601. uint32_t base;
  602. uint32_t prec;
  603. char fill;
  604. uint8_t bits;
  605. bool left;
  606. bool upper;
  607. bool spec;
  608. bool sign;
  609. };
  610. static int32_t write(WriterI* _writer, const char* _str, int32_t _len, const Param& _param, Error* _err)
  611. {
  612. int32_t size = 0;
  613. int32_t len = (int32_t)strLen(_str, _len);
  614. int32_t padding = _param.width > len ? _param.width - len : 0;
  615. bool sign = _param.sign && len > 1 && _str[0] != '-';
  616. padding = padding > 0 ? padding - sign : 0;
  617. if (!_param.left)
  618. {
  619. size += writeRep(_writer, _param.fill, padding, _err);
  620. }
  621. if (NULL == _str)
  622. {
  623. size += write(_writer, "(null)", 6, _err);
  624. }
  625. else if (_param.upper)
  626. {
  627. for (int32_t ii = 0; ii < len; ++ii)
  628. {
  629. size += write(_writer, toUpper(_str[ii]), _err);
  630. }
  631. }
  632. else if (sign)
  633. {
  634. size += write(_writer, '+', _err);
  635. size += write(_writer, _str, len, _err);
  636. }
  637. else
  638. {
  639. size += write(_writer, _str, len, _err);
  640. }
  641. if (_param.left)
  642. {
  643. size += writeRep(_writer, _param.fill, padding, _err);
  644. }
  645. return size;
  646. }
  647. static int32_t write(WriterI* _writer, char _ch, const Param& _param, Error* _err)
  648. {
  649. return write(_writer, &_ch, 1, _param, _err);
  650. }
  651. static int32_t write(WriterI* _writer, const char* _str, const Param& _param, Error* _err)
  652. {
  653. return write(_writer, _str, INT32_MAX, _param, _err);
  654. }
  655. static int32_t write(WriterI* _writer, int32_t _i, const Param& _param, Error* _err)
  656. {
  657. char str[33];
  658. int32_t len = toString(str, sizeof(str), _i, _param.base);
  659. if (len == 0)
  660. {
  661. return 0;
  662. }
  663. return write(_writer, str, len, _param, _err);
  664. }
  665. static int32_t write(WriterI* _writer, int64_t _i, const Param& _param, Error* _err)
  666. {
  667. char str[33];
  668. int32_t len = toString(str, sizeof(str), _i, _param.base);
  669. if (len == 0)
  670. {
  671. return 0;
  672. }
  673. return write(_writer, str, len, _param, _err);
  674. }
  675. static int32_t write(WriterI* _writer, uint32_t _u, const Param& _param, Error* _err)
  676. {
  677. char str[33];
  678. int32_t len = toString(str, sizeof(str), _u, _param.base);
  679. if (len == 0)
  680. {
  681. return 0;
  682. }
  683. return write(_writer, str, len, _param, _err);
  684. }
  685. static int32_t write(WriterI* _writer, uint64_t _u, const Param& _param, Error* _err)
  686. {
  687. char str[33];
  688. int32_t len = toString(str, sizeof(str), _u, _param.base);
  689. if (len == 0)
  690. {
  691. return 0;
  692. }
  693. return write(_writer, str, len, _param, _err);
  694. }
  695. static int32_t write(WriterI* _writer, double _d, const Param& _param, Error* _err)
  696. {
  697. char str[1024];
  698. int32_t len = toString(str, sizeof(str), _d);
  699. if (len == 0)
  700. {
  701. return 0;
  702. }
  703. if (_param.upper)
  704. {
  705. toUpperUnsafe(str, len);
  706. }
  707. const char* dot = strFind(str, INT32_MAX, '.');
  708. if (NULL != dot)
  709. {
  710. const int32_t precLen = int32_t(
  711. dot
  712. + uint32_min(_param.prec + _param.spec, 1)
  713. + _param.prec
  714. - str
  715. );
  716. if (precLen > len)
  717. {
  718. for (int32_t ii = len; ii < precLen; ++ii)
  719. {
  720. str[ii] = '0';
  721. }
  722. str[precLen] = '\0';
  723. }
  724. len = precLen;
  725. }
  726. return write(_writer, str, len, _param, _err);
  727. }
  728. static int32_t write(WriterI* _writer, const void* _ptr, const Param& _param, Error* _err)
  729. {
  730. char str[35] = "0x";
  731. int32_t len = toString(str + 2, sizeof(str) - 2, uint32_t(uintptr_t(_ptr) ), 16);
  732. if (len == 0)
  733. {
  734. return 0;
  735. }
  736. len += 2;
  737. return write(_writer, str, len, _param, _err);
  738. }
  739. } // anonymous namespace
  740. int32_t write(WriterI* _writer, const char* _format, va_list _argList, Error* _err)
  741. {
  742. MemoryReader reader(_format, uint32_t(strLen(_format) ) );
  743. int32_t size = 0;
  744. while (_err->isOk() )
  745. {
  746. char ch = '\0';
  747. Error err;
  748. read(&reader, ch, &err);
  749. if (!_err->isOk()
  750. || !err.isOk() )
  751. {
  752. break;
  753. }
  754. else if ('%' == ch)
  755. {
  756. // %[flags][width][.precision][length sub-specifier]specifier
  757. read(&reader, ch);
  758. Param param;
  759. // flags
  760. while (' ' == ch
  761. || '-' == ch
  762. || '+' == ch
  763. || '0' == ch
  764. || '#' == ch)
  765. {
  766. switch (ch)
  767. {
  768. default:
  769. case ' ': param.fill = ' '; break;
  770. case '-': param.left = true; break;
  771. case '+': param.sign = true; break;
  772. case '0': param.fill = '0'; break;
  773. case '#': param.spec = true; break;
  774. }
  775. read(&reader, ch);
  776. }
  777. if (param.left)
  778. {
  779. param.fill = ' ';
  780. }
  781. // width
  782. if ('*' == ch)
  783. {
  784. read(&reader, ch);
  785. param.width = va_arg(_argList, int32_t);
  786. if (0 > param.width)
  787. {
  788. param.left = true;
  789. param.width = -param.width;
  790. }
  791. }
  792. else
  793. {
  794. while (isNumeric(ch) )
  795. {
  796. param.width = param.width * 10 + ch - '0';
  797. read(&reader, ch);
  798. }
  799. }
  800. // .precision
  801. if ('.' == ch)
  802. {
  803. read(&reader, ch);
  804. if ('*' == ch)
  805. {
  806. read(&reader, ch);
  807. param.prec = va_arg(_argList, int32_t);
  808. }
  809. else
  810. {
  811. param.prec = 0;
  812. while (isNumeric(ch) )
  813. {
  814. param.prec = param.prec * 10 + ch - '0';
  815. read(&reader, ch);
  816. }
  817. }
  818. }
  819. // length sub-specifier
  820. while ('h' == ch
  821. || 'I' == ch
  822. || 'l' == ch
  823. || 'j' == ch
  824. || 't' == ch
  825. || 'z' == ch)
  826. {
  827. switch (ch)
  828. {
  829. default: break;
  830. case 'j': param.bits = sizeof(intmax_t )*8; break;
  831. case 't': param.bits = sizeof(size_t )*8; break;
  832. case 'z': param.bits = sizeof(ptrdiff_t)*8; break;
  833. case 'h': case 'I': case 'l':
  834. switch (ch)
  835. {
  836. case 'h': param.bits = sizeof(short int)*8; break;
  837. case 'l': param.bits = sizeof(long int )*8; break;
  838. default: break;
  839. }
  840. read(&reader, ch);
  841. switch (ch)
  842. {
  843. case 'h': param.bits = sizeof(signed char )*8; break;
  844. case 'l': param.bits = sizeof(long long int)*8; break;
  845. case '3':
  846. case '6':
  847. read(&reader, ch);
  848. switch (ch)
  849. {
  850. case '2': param.bits = sizeof(int32_t)*8; break;
  851. case '4': param.bits = sizeof(int64_t)*8; break;
  852. default: break;
  853. }
  854. break;
  855. default: seek(&reader, -1); break;
  856. }
  857. break;
  858. }
  859. read(&reader, ch);
  860. }
  861. // specifier
  862. switch (toLower(ch) )
  863. {
  864. case 'c':
  865. size += write(_writer, char(va_arg(_argList, int32_t) ), param, _err);
  866. break;
  867. case 's':
  868. size += write(_writer, va_arg(_argList, const char*), param, _err);
  869. break;
  870. case 'o':
  871. param.base = 8;
  872. switch (param.bits)
  873. {
  874. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  875. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  876. }
  877. break;
  878. case 'i':
  879. case 'd':
  880. param.base = 10;
  881. switch (param.bits)
  882. {
  883. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  884. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  885. };
  886. break;
  887. case 'e':
  888. case 'f':
  889. case 'g':
  890. param.upper = isUpper(ch);
  891. size += write(_writer, va_arg(_argList, double), param, _err);
  892. break;
  893. case 'p':
  894. size += write(_writer, va_arg(_argList, void*), param, _err);
  895. break;
  896. case 'x':
  897. param.base = 16;
  898. param.upper = isUpper(ch);
  899. switch (param.bits)
  900. {
  901. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  902. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  903. }
  904. break;
  905. case 'u':
  906. param.base = 10;
  907. switch (param.bits)
  908. {
  909. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  910. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  911. }
  912. break;
  913. default:
  914. size += write(_writer, ch, _err);
  915. break;
  916. }
  917. }
  918. else
  919. {
  920. size += write(_writer, ch, _err);
  921. }
  922. }
  923. size += write(_writer, '\0', _err);
  924. return size;
  925. }
  926. int32_t write(WriterI* _writer, Error* _err, const char* _format, ...)
  927. {
  928. va_list argList;
  929. va_start(argList, _format);
  930. int32_t total = write(_writer, _format, argList, _err);
  931. va_end(argList);
  932. return total;
  933. }
  934. int32_t vsnprintfRef(char* _out, int32_t _max, const char* _format, va_list _argList)
  935. {
  936. if (1 < _max)
  937. {
  938. StaticMemoryBlockWriter writer(_out, uint32_t(_max-1) );
  939. _out[_max-1] = '\0';
  940. Error err;
  941. va_list argListCopy;
  942. va_copy(argListCopy, _argList);
  943. int32_t size = write(&writer, _format, argListCopy, &err);
  944. va_end(argListCopy);
  945. if (err.isOk() )
  946. {
  947. return size - 1 /* size without '\0' terminator */;
  948. }
  949. }
  950. Error err;
  951. SizerWriter sizer;
  952. va_list argListCopy;
  953. va_copy(argListCopy, _argList);
  954. int32_t size = write(&sizer, _format, argListCopy, &err);
  955. va_end(argListCopy);
  956. return size - 1 /* size without '\0' terminator */;
  957. }
  958. int32_t vsnprintf(char* _out, int32_t _max, const char* _format, va_list _argList)
  959. {
  960. va_list argList;
  961. va_copy(argList, _argList);
  962. int32_t total = 0;
  963. #if BX_CRT_NONE
  964. total = vsnprintfRef(_out, _max, _format, argList);
  965. #elif BX_CRT_MSVC
  966. int32_t len = -1;
  967. if (NULL != _out)
  968. {
  969. va_list argListCopy;
  970. va_copy(argListCopy, _argList);
  971. len = ::vsnprintf_s(_out, _max, size_t(-1), _format, argListCopy);
  972. va_end(argListCopy);
  973. }
  974. total = -1 == len ? ::_vscprintf(_format, argList) : len;
  975. #else
  976. total = ::vsnprintf(_out, _max, _format, argList);
  977. #endif // BX_COMPILER_MSVC
  978. va_end(argList);
  979. return total;
  980. }
  981. int32_t snprintf(char* _out, int32_t _max, const char* _format, ...)
  982. {
  983. va_list argList;
  984. va_start(argList, _format);
  985. int32_t total = vsnprintf(_out, _max, _format, argList);
  986. va_end(argList);
  987. return total;
  988. }
  989. static const char s_units[] = { 'B', 'k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y' };
  990. template<uint32_t Kilo, char KiloCh0, char KiloCh1, CharFn fn>
  991. inline int32_t prettify(char* _out, int32_t _count, uint64_t _value)
  992. {
  993. uint8_t idx = 0;
  994. double value = double(_value);
  995. while (_value != (_value&0x7ff)
  996. && idx < BX_COUNTOF(s_units) )
  997. {
  998. _value /= Kilo;
  999. value *= 1.0/double(Kilo);
  1000. ++idx;
  1001. }
  1002. return snprintf(_out, _count, "%0.2f %c%c%c", value
  1003. , fn(s_units[idx])
  1004. , idx > 0 ? KiloCh0 : '\0'
  1005. , KiloCh1
  1006. );
  1007. }
  1008. int32_t prettify(char* _out, int32_t _count, uint64_t _value, Units::Enum _units)
  1009. {
  1010. if (Units::Kilo == _units)
  1011. {
  1012. return prettify<1000, 'B', '\0', toNoop>(_out, _count, _value);
  1013. }
  1014. return prettify<1024, 'i', 'B', toUpper>(_out, _count, _value);
  1015. }
  1016. } // namespace bx