string.cpp 24 KB

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