string.cpp 26 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291
  1. /*
  2. * Copyright 2010-2021 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/file.h>
  8. #include <bx/hash.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. if (0 == max)
  159. {
  160. return _lhsMax == _rhsMax ? 0 : _lhsMax > _rhsMax ? 1 : -1;
  161. }
  162. return fn(*_lhs) - fn(*_rhs);
  163. }
  164. int32_t strCmp(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  165. {
  166. return strCmp<toNoop>(
  167. _lhs.getPtr()
  168. , min(_lhs.getLength(), _max)
  169. , _rhs.getPtr()
  170. , min(_rhs.getLength(), _max)
  171. );
  172. }
  173. int32_t strCmpI(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  174. {
  175. return strCmp<toLower>(
  176. _lhs.getPtr()
  177. , min(_lhs.getLength(), _max)
  178. , _rhs.getPtr()
  179. , min(_rhs.getLength(), _max)
  180. );
  181. }
  182. inline int32_t strCmpV(const char* _lhs, int32_t _lhsMax, const char* _rhs, int32_t _rhsMax)
  183. {
  184. int32_t max = min(_lhsMax, _rhsMax);
  185. int32_t ii = 0;
  186. int32_t idx = 0;
  187. bool zero = true;
  188. for (
  189. ; 0 < max && _lhs[ii] == _rhs[ii]
  190. ; ++ii, --max
  191. )
  192. {
  193. const uint8_t ch = _lhs[ii];
  194. if ('\0' == ch
  195. || '\0' == _rhs[ii])
  196. {
  197. break;
  198. }
  199. if (!isNumeric(ch) )
  200. {
  201. idx = ii+1;
  202. zero = true;
  203. }
  204. else if ('0' != ch)
  205. {
  206. zero = false;
  207. }
  208. }
  209. if (0 == max)
  210. {
  211. return _lhsMax == _rhsMax ? 0 : _lhs[ii] - _rhs[ii];
  212. }
  213. if ('0' != _lhs[idx]
  214. && '0' != _rhs[idx])
  215. {
  216. int32_t jj = 0;
  217. for (jj = ii
  218. ; 0 < max && isNumeric(_lhs[jj])
  219. ; ++jj, --max
  220. )
  221. {
  222. if (!isNumeric(_rhs[jj]) )
  223. {
  224. return 1;
  225. }
  226. }
  227. if (isNumeric(_rhs[jj]))
  228. {
  229. return -1;
  230. }
  231. }
  232. else if (zero
  233. && idx < ii
  234. && (isNumeric(_lhs[ii]) || isNumeric(_rhs[ii]) ) )
  235. {
  236. return (_lhs[ii] - '0') - (_rhs[ii] - '0');
  237. }
  238. return 0 == max && _lhsMax == _rhsMax ? 0 : _lhs[ii] - _rhs[ii];
  239. }
  240. int32_t strCmpV(const StringView& _lhs, const StringView& _rhs, int32_t _max)
  241. {
  242. return strCmpV(
  243. _lhs.getPtr()
  244. , min(_lhs.getLength(), _max)
  245. , _rhs.getPtr()
  246. , min(_rhs.getLength(), _max)
  247. );
  248. }
  249. int32_t strLen(const char* _str, int32_t _max)
  250. {
  251. if (NULL == _str)
  252. {
  253. return 0;
  254. }
  255. const char* ptr = _str;
  256. for (; 0 < _max && *ptr != '\0'; ++ptr, --_max) {};
  257. return int32_t(ptr - _str);
  258. }
  259. int32_t strLen(const StringView& _str, int32_t _max)
  260. {
  261. return strLen(_str.getPtr(), min(_str.getLength(), _max) );
  262. }
  263. inline int32_t strCopy(char* _dst, int32_t _dstSize, const char* _src, int32_t _num)
  264. {
  265. BX_ASSERT(NULL != _dst, "_dst can't be NULL!");
  266. BX_ASSERT(NULL != _src, "_src can't be NULL!");
  267. BX_ASSERT(0 < _dstSize, "_dstSize can't be 0!");
  268. const int32_t len = strLen(_src, _num);
  269. const int32_t max = _dstSize-1;
  270. const int32_t num = (len < max ? len : max);
  271. memCopy(_dst, _src, num);
  272. _dst[num] = '\0';
  273. return num;
  274. }
  275. int32_t strCopy(char* _dst, int32_t _dstSize, const StringView& _str, int32_t _num)
  276. {
  277. return strCopy(_dst, _dstSize, _str.getPtr(), min(_str.getLength(), _num) );
  278. }
  279. inline int32_t strCat(char* _dst, int32_t _dstSize, const char* _src, int32_t _num)
  280. {
  281. BX_ASSERT(NULL != _dst, "_dst can't be NULL!");
  282. BX_ASSERT(NULL != _src, "_src can't be NULL!");
  283. BX_ASSERT(0 < _dstSize, "_dstSize can't be 0!");
  284. const int32_t max = _dstSize;
  285. const int32_t len = strLen(_dst, max);
  286. return strCopy(&_dst[len], max-len, _src, _num);
  287. }
  288. int32_t strCat(char* _dst, int32_t _dstSize, const StringView& _str, int32_t _num)
  289. {
  290. return strCat(_dst, _dstSize, _str.getPtr(), min(_str.getLength(), _num) );
  291. }
  292. inline const char* strFindUnsafe(const char* _str, int32_t _len, char _ch)
  293. {
  294. for (int32_t ii = 0; ii < _len; ++ii)
  295. {
  296. if (_str[ii] == _ch)
  297. {
  298. return &_str[ii];
  299. }
  300. }
  301. return NULL;
  302. }
  303. inline const char* strFind(const char* _str, int32_t _max, char _ch)
  304. {
  305. return strFindUnsafe(_str, strLen(_str, _max), _ch);
  306. }
  307. StringView strFind(const StringView& _str, char _ch)
  308. {
  309. const char* ptr = strFindUnsafe(_str.getPtr(), _str.getLength(), _ch);
  310. if (NULL == ptr)
  311. {
  312. return StringView(_str.getTerm(), _str.getTerm() );
  313. }
  314. return StringView(ptr, ptr+1);
  315. }
  316. inline const char* strRFindUnsafe(const char* _str, int32_t _len, char _ch)
  317. {
  318. for (int32_t ii = _len-1; 0 <= ii; --ii)
  319. {
  320. if (_str[ii] == _ch)
  321. {
  322. return &_str[ii];
  323. }
  324. }
  325. return NULL;
  326. }
  327. StringView strRFind(const StringView& _str, char _ch)
  328. {
  329. const char* ptr = strRFindUnsafe(_str.getPtr(), _str.getLength(), _ch);
  330. if (NULL == ptr)
  331. {
  332. return StringView(_str.getTerm(), _str.getTerm() );
  333. }
  334. return StringView(ptr, ptr+1);
  335. }
  336. template<CharFn fn>
  337. inline const char* strFind(const char* _str, int32_t _strMax, const char* _find, int32_t _findMax)
  338. {
  339. const char* ptr = _str;
  340. int32_t stringLen = _strMax;
  341. const int32_t findLen = _findMax;
  342. for (; stringLen >= findLen; ++ptr, --stringLen)
  343. {
  344. // Find start of the string.
  345. while (fn(*ptr) != fn(*_find) )
  346. {
  347. ++ptr;
  348. --stringLen;
  349. // Search pattern lenght can't be longer than the string.
  350. if (findLen > stringLen)
  351. {
  352. return NULL;
  353. }
  354. }
  355. if (0 == strCmp<fn>(ptr, _findMax, _find, _findMax) )
  356. {
  357. return ptr;
  358. }
  359. }
  360. return NULL;
  361. }
  362. StringView strFind(const StringView& _str, const StringView& _find, int32_t _num)
  363. {
  364. int32_t len = min(_find.getLength(), _num);
  365. const char* ptr = strFind<toNoop>(
  366. _str.getPtr()
  367. , _str.getLength()
  368. , _find.getPtr()
  369. , len
  370. );
  371. if (NULL == ptr)
  372. {
  373. return StringView(_str.getTerm(), _str.getTerm() );
  374. }
  375. return StringView(ptr, len);
  376. }
  377. StringView strFindI(const StringView& _str, const StringView& _find, int32_t _num)
  378. {
  379. int32_t len = min(_find.getLength(), _num);
  380. const char* ptr = strFind<toLower>(
  381. _str.getPtr()
  382. , _str.getLength()
  383. , _find.getPtr()
  384. , len
  385. );
  386. if (NULL == ptr)
  387. {
  388. return StringView(_str.getTerm(), _str.getTerm() );
  389. }
  390. return StringView(ptr, len);
  391. }
  392. StringView strLTrim(const StringView& _str, const StringView& _chars)
  393. {
  394. const char* ptr = _str.getPtr();
  395. const char* chars = _chars.getPtr();
  396. const uint32_t charsLen = _chars.getLength();
  397. for (uint32_t ii = 0, len = _str.getLength(); ii < len; ++ii)
  398. {
  399. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]) )
  400. {
  401. return StringView(ptr + ii, len-ii);
  402. }
  403. }
  404. return _str;
  405. }
  406. StringView strLTrimSpace(const StringView& _str)
  407. {
  408. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  409. {
  410. if (!isSpace(*ptr) )
  411. {
  412. return StringView(ptr, term);
  413. }
  414. }
  415. return StringView(_str.getTerm(), _str.getTerm() );
  416. }
  417. StringView strLTrimNonSpace(const StringView& _str)
  418. {
  419. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  420. {
  421. if (isSpace(*ptr) )
  422. {
  423. return StringView(ptr, term);
  424. }
  425. }
  426. return StringView(_str.getTerm(), _str.getTerm() );
  427. }
  428. StringView strRTrim(const StringView& _str, const StringView& _chars)
  429. {
  430. if (!_str.isEmpty() )
  431. {
  432. const char* ptr = _str.getPtr();
  433. const char* chars = _chars.getPtr();
  434. const uint32_t charsLen = _chars.getLength();
  435. for (int32_t len = _str.getLength(), ii = len - 1; 0 <= ii; --ii)
  436. {
  437. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]))
  438. {
  439. return StringView(ptr, ii + 1);
  440. }
  441. }
  442. }
  443. return _str;
  444. }
  445. StringView strRTrimSpace(const StringView& _str)
  446. {
  447. if (!_str.isEmpty() )
  448. {
  449. const char* ptr = _str.getPtr();
  450. for (int32_t len = _str.getLength(), ii = len - 1; 0 <= ii; --ii)
  451. {
  452. if (!isSpace(ptr[ii]) )
  453. {
  454. return StringView(ptr, ii + 1);
  455. }
  456. }
  457. }
  458. return _str;
  459. }
  460. StringView strTrim(const StringView& _str, const StringView& _chars)
  461. {
  462. return strLTrim(strRTrim(_str, _chars), _chars);
  463. }
  464. StringView strTrimSpace(const StringView& _str)
  465. {
  466. return strLTrimSpace(strRTrimSpace(_str) );
  467. }
  468. constexpr uint32_t kFindStep = 1024;
  469. StringView strFindNl(const StringView& _str)
  470. {
  471. StringView str(_str);
  472. // This method returns the character past the \n, so
  473. // there is no need to look for he \r which preceedes it.
  474. StringView eol = strFind(str, "\n");
  475. if (!eol.isEmpty() )
  476. {
  477. return StringView(eol.getTerm(), str.getTerm() );
  478. }
  479. return StringView(_str.getTerm(), _str.getTerm() );
  480. }
  481. StringView strFindEol(const StringView& _str)
  482. {
  483. StringView str(_str);
  484. for (; str.getPtr() != _str.getTerm()
  485. ; str = StringView(min(str.getPtr() + kFindStep, _str.getTerm() ), min(str.getPtr() + kFindStep*2, _str.getTerm() ) )
  486. )
  487. {
  488. StringView eol = strFind(str, "\r\n");
  489. if (!eol.isEmpty() )
  490. {
  491. return StringView(eol.getPtr(), _str.getTerm() );
  492. }
  493. eol = strFind(str, '\n');
  494. if (!eol.isEmpty() )
  495. {
  496. return StringView(eol.getPtr(), _str.getTerm() );
  497. }
  498. }
  499. return StringView(_str.getTerm(), _str.getTerm() );
  500. }
  501. static const char* strSkipWord(const char* _str, int32_t _max)
  502. {
  503. for (char ch = *_str++; 0 < _max && (isAlphaNum(ch) || '_' == ch); ch = *_str++, --_max) {};
  504. return _str-1;
  505. }
  506. StringView strWord(const StringView& _str)
  507. {
  508. const char* ptr = _str.getPtr();
  509. const char* term = strSkipWord(ptr, _str.getLength() );
  510. return StringView(ptr, term);
  511. }
  512. StringView strFindBlock(const StringView& _str, char _open, char _close)
  513. {
  514. const char* curr = _str.getPtr();
  515. const char* term = _str.getTerm();
  516. const char* start = NULL;
  517. int32_t count = 0;
  518. for (char ch = *curr; curr != term && count >= 0; ch = *(++curr) )
  519. {
  520. if (ch == _open)
  521. {
  522. if (0 == count)
  523. {
  524. start = curr;
  525. }
  526. ++count;
  527. }
  528. else if (ch == _close)
  529. {
  530. --count;
  531. if (NULL == start)
  532. {
  533. break;
  534. }
  535. if (0 == count)
  536. {
  537. return StringView(start, curr+1);
  538. }
  539. }
  540. }
  541. return StringView(term, term);
  542. }
  543. StringView normalizeEolLf(char* _out, int32_t _size, const StringView& _str)
  544. {
  545. const char* start = _out;
  546. const char* end = _out;
  547. if (0 < _size)
  548. {
  549. const char* curr = _str.getPtr();
  550. const char* term = _str.getTerm();
  551. end = _out + _size;
  552. for (char ch = *curr; curr != term && _out < end; ch = *(++curr) )
  553. {
  554. if ('\r' != ch)
  555. {
  556. *_out++ = ch;
  557. }
  558. }
  559. end = _out;
  560. }
  561. return StringView(start, end);
  562. }
  563. StringView findIdentifierMatch(const StringView& _str, const StringView& _word)
  564. {
  565. const int32_t len = _word.getLength();
  566. StringView ptr = strFind(_str, _word);
  567. for (; !ptr.isEmpty(); ptr = strFind(StringView(ptr.getPtr() + len, _str.getTerm() ), _word) )
  568. {
  569. char ch = ptr.getPtr() != _str.getPtr() ? *(ptr.getPtr() - 1) : ' ';
  570. if (isAlphaNum(ch) || '_' == ch)
  571. {
  572. continue;
  573. }
  574. ch = *(ptr.getPtr() + len);
  575. if (isAlphaNum(ch) || '_' == ch)
  576. {
  577. continue;
  578. }
  579. return ptr;
  580. }
  581. return StringView(_str.getTerm(), _str.getTerm() );
  582. }
  583. StringView findIdentifierMatch(const StringView& _str, const char** _words, int32_t _num)
  584. {
  585. int32_t ii = 0;
  586. for (StringView word = *_words; ii < _num && !word.isEmpty(); ++ii, ++_words, word = *_words)
  587. {
  588. StringView match = findIdentifierMatch(_str, word);
  589. if (!match.isEmpty() )
  590. {
  591. return match;
  592. }
  593. }
  594. return StringView(_str.getTerm(), _str.getTerm() );
  595. }
  596. namespace
  597. {
  598. struct Param
  599. {
  600. Param()
  601. : width(0)
  602. , base(10)
  603. , prec(INT32_MAX)
  604. , fill(' ')
  605. , bits(0)
  606. , left(false)
  607. , upper(false)
  608. , spec(false)
  609. , sign(false)
  610. {
  611. }
  612. int32_t width;
  613. int32_t base;
  614. int32_t prec;
  615. char fill;
  616. uint8_t bits;
  617. bool left;
  618. bool upper;
  619. bool spec;
  620. bool sign;
  621. };
  622. static int32_t write(WriterI* _writer, const char* _str, int32_t _len, const Param& _param, Error* _err)
  623. {
  624. int32_t size = 0;
  625. int32_t len = (int32_t)strLen(_str, _len);
  626. if (_param.width > 0)
  627. {
  628. len = min(_param.width, len);
  629. }
  630. const bool hasMinus = (NULL != _str && '-' == _str[0]);
  631. const bool hasSign = _param.sign || hasMinus;
  632. char sign = hasSign ? hasMinus ? '-' : '+' : '\0';
  633. const char* str = _str;
  634. if (hasMinus)
  635. {
  636. str++;
  637. len--;
  638. }
  639. int32_t padding = _param.width > len ? _param.width - len - hasSign: 0;
  640. if (!_param.left)
  641. {
  642. if (' ' != _param.fill
  643. && '\0' != sign)
  644. {
  645. size += write(_writer, sign, _err);
  646. sign = '\0';
  647. }
  648. size += writeRep(_writer, _param.fill, max(0, padding), _err);
  649. }
  650. if ('\0' != sign)
  651. {
  652. size += write(_writer, sign, _err);
  653. }
  654. if (NULL == _str)
  655. {
  656. size += write(_writer, "(null)", 6, _err);
  657. }
  658. else if (_param.upper)
  659. {
  660. for (int32_t ii = 0; ii < len; ++ii)
  661. {
  662. size += write(_writer, toUpper(str[ii]), _err);
  663. }
  664. }
  665. else
  666. {
  667. size += write(_writer, str, len, _err);
  668. }
  669. if (_param.left)
  670. {
  671. size += writeRep(_writer, _param.fill, padding, _err);
  672. }
  673. return size;
  674. }
  675. static int32_t write(WriterI* _writer, char _ch, const Param& _param, Error* _err)
  676. {
  677. return write(_writer, &_ch, 1, _param, _err);
  678. }
  679. static int32_t write(WriterI* _writer, const char* _str, const Param& _param, Error* _err)
  680. {
  681. return write(_writer, _str, _param.prec, _param, _err);
  682. }
  683. static int32_t write(WriterI* _writer, const StringView& _str, const Param& _param, Error* _err)
  684. {
  685. return write(_writer, _str.getPtr(), min(_param.prec, _str.getLength() ), _param, _err);
  686. }
  687. static int32_t write(WriterI* _writer, int32_t _i, const Param& _param, Error* _err)
  688. {
  689. char str[33];
  690. int32_t len = toString(str, sizeof(str), _i, _param.base);
  691. if (len == 0)
  692. {
  693. return 0;
  694. }
  695. return write(_writer, str, len, _param, _err);
  696. }
  697. static int32_t write(WriterI* _writer, int64_t _i, const Param& _param, Error* _err)
  698. {
  699. char str[33];
  700. int32_t len = toString(str, sizeof(str), _i, _param.base);
  701. if (len == 0)
  702. {
  703. return 0;
  704. }
  705. return write(_writer, str, len, _param, _err);
  706. }
  707. static int32_t write(WriterI* _writer, uint32_t _u, const Param& _param, Error* _err)
  708. {
  709. char str[33];
  710. int32_t len = toString(str, sizeof(str), _u, _param.base);
  711. if (len == 0)
  712. {
  713. return 0;
  714. }
  715. return write(_writer, str, len, _param, _err);
  716. }
  717. static int32_t write(WriterI* _writer, uint64_t _u, const Param& _param, Error* _err)
  718. {
  719. char str[33];
  720. int32_t len = toString(str, sizeof(str), _u, _param.base);
  721. if (len == 0)
  722. {
  723. return 0;
  724. }
  725. return write(_writer, str, len, _param, _err);
  726. }
  727. static int32_t write(WriterI* _writer, double _d, const Param& _param, Error* _err)
  728. {
  729. char str[1024];
  730. int32_t len = toString(str, sizeof(str), _d);
  731. if (len == 0)
  732. {
  733. return 0;
  734. }
  735. if (_param.upper)
  736. {
  737. toUpperUnsafe(str, len);
  738. }
  739. const char* dot = strFind(str, INT32_MAX, '.');
  740. if (NULL != dot)
  741. {
  742. const int32_t prec = INT32_MAX == _param.prec ? 6 : _param.prec;
  743. const int32_t precLen = int32_t(
  744. dot
  745. + uint32_min(prec + _param.spec, 1)
  746. + prec
  747. - str
  748. );
  749. if (precLen > len)
  750. {
  751. for (int32_t ii = len; ii < precLen; ++ii)
  752. {
  753. str[ii] = '0';
  754. }
  755. str[precLen] = '\0';
  756. }
  757. len = precLen;
  758. }
  759. return write(_writer, str, len, _param, _err);
  760. }
  761. static int32_t write(WriterI* _writer, const void* _ptr, const Param& _param, Error* _err)
  762. {
  763. char str[35] = "0x";
  764. int32_t len = toString(str + 2, sizeof(str) - 2, uint32_t(uintptr_t(_ptr) ), 16);
  765. if (len == 0)
  766. {
  767. return 0;
  768. }
  769. len += 2;
  770. return write(_writer, str, len, _param, _err);
  771. }
  772. } // anonymous namespace
  773. int32_t write(WriterI* _writer, const StringView& _format, va_list _argList, Error* _err)
  774. {
  775. MemoryReader reader(_format.getPtr(), _format.getLength() );
  776. int32_t size = 0;
  777. while (_err->isOk() )
  778. {
  779. char ch = '\0';
  780. Error err;
  781. read(&reader, ch, &err);
  782. if (!_err->isOk()
  783. || !err.isOk() )
  784. {
  785. break;
  786. }
  787. else if ('%' == ch)
  788. {
  789. // %[Flags][Width][.Precision][Leegth]Type
  790. read(&reader, ch, &err);
  791. Param param;
  792. // Reference(s):
  793. // - Flags field
  794. // https://en.wikipedia.org/wiki/Printf_format_string#Flags_field
  795. //
  796. while (err.isOk()
  797. && ( ' ' == ch
  798. || '-' == ch
  799. || '+' == ch
  800. || '0' == ch
  801. || '#' == ch)
  802. )
  803. {
  804. switch (ch)
  805. {
  806. default:
  807. case ' ': param.fill = ' '; break;
  808. case '-': param.left = true; break;
  809. case '+': param.sign = true; break;
  810. case '0': param.fill = '0'; break;
  811. case '#': param.spec = true; break;
  812. }
  813. read(&reader, ch, &err);
  814. }
  815. if (param.left)
  816. {
  817. param.fill = ' ';
  818. }
  819. // Reference(s):
  820. // - Width field
  821. // https://en.wikipedia.org/wiki/Printf_format_string#Width_field
  822. //
  823. if ('*' == ch)
  824. {
  825. read(&reader, ch, &err);
  826. param.width = va_arg(_argList, int32_t);
  827. if (0 > param.width)
  828. {
  829. param.left = true;
  830. param.width = -param.width;
  831. }
  832. }
  833. else
  834. {
  835. while (err.isOk()
  836. && isNumeric(ch) )
  837. {
  838. param.width = param.width * 10 + ch - '0';
  839. read(&reader, ch, &err);
  840. }
  841. }
  842. // Reference(s):
  843. // - Precision field
  844. // https://en.wikipedia.org/wiki/Printf_format_string#Precision_field
  845. if ('.' == ch)
  846. {
  847. read(&reader, ch, &err);
  848. if ('*' == ch)
  849. {
  850. read(&reader, ch, &err);
  851. param.prec = va_arg(_argList, int32_t);
  852. }
  853. else
  854. {
  855. param.prec = 0;
  856. while (err.isOk()
  857. && isNumeric(ch) )
  858. {
  859. param.prec = param.prec * 10 + ch - '0';
  860. read(&reader, ch, &err);
  861. }
  862. }
  863. }
  864. // Reference(s):
  865. // - Length field
  866. // https://en.wikipedia.org/wiki/Printf_format_string#Length_field
  867. while (err.isOk()
  868. && ( 'h' == ch
  869. || 'I' == ch
  870. || 'l' == ch
  871. || 'j' == ch
  872. || 't' == ch
  873. || 'z' == ch)
  874. )
  875. {
  876. switch (ch)
  877. {
  878. default: break;
  879. case 'j': param.bits = sizeof(intmax_t )*8; break;
  880. case 't': param.bits = sizeof(ptrdiff_t)*8; break;
  881. case 'z': param.bits = sizeof(size_t )*8; break;
  882. case 'h': case 'I': case 'l':
  883. switch (ch)
  884. {
  885. case 'h': param.bits = sizeof(short int)*8; break;
  886. case 'l': param.bits = sizeof(long int )*8; break;
  887. default: break;
  888. }
  889. read(&reader, ch, &err);
  890. switch (ch)
  891. {
  892. case 'h': param.bits = sizeof(signed char )*8; break;
  893. case 'l': param.bits = sizeof(long long int)*8; break;
  894. case '3':
  895. case '6':
  896. read(&reader, ch, &err);
  897. switch (ch)
  898. {
  899. case '2': param.bits = sizeof(int32_t)*8; break;
  900. case '4': param.bits = sizeof(int64_t)*8; break;
  901. default: break;
  902. }
  903. break;
  904. default: seek(&reader, -1); break;
  905. }
  906. break;
  907. }
  908. read(&reader, ch, &err);
  909. }
  910. if (!err.isOk() )
  911. {
  912. break;
  913. }
  914. // Reference(s):
  915. // - Type field
  916. // https://en.wikipedia.org/wiki/Printf_format_string#Type_field
  917. switch (ch)
  918. {
  919. case 'c':
  920. size += write(_writer, char(va_arg(_argList, int32_t) ), param, _err);
  921. break;
  922. case 's':
  923. size += write(_writer, va_arg(_argList, const char*), param, _err);
  924. break;
  925. case 'S':
  926. size += write(_writer, *va_arg(_argList, const StringView*), param, _err);
  927. break;
  928. case 'o':
  929. param.base = 8;
  930. switch (param.bits)
  931. {
  932. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  933. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  934. }
  935. break;
  936. case 'i':
  937. case 'd':
  938. param.base = 10;
  939. switch (param.bits)
  940. {
  941. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  942. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  943. };
  944. break;
  945. case 'e':
  946. case 'E':
  947. case 'f':
  948. case 'F':
  949. case 'g':
  950. case 'G':
  951. param.upper = isUpper(ch);
  952. size += write(_writer, va_arg(_argList, double), param, _err);
  953. break;
  954. case 'p':
  955. size += write(_writer, va_arg(_argList, void*), param, _err);
  956. break;
  957. case 'x':
  958. case 'X':
  959. param.base = 16;
  960. param.upper = isUpper(ch);
  961. switch (param.bits)
  962. {
  963. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  964. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  965. }
  966. break;
  967. case 'u':
  968. param.base = 10;
  969. switch (param.bits)
  970. {
  971. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  972. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  973. }
  974. break;
  975. case 'n':
  976. *va_arg(_argList, int32_t*) = size;
  977. break;
  978. default:
  979. size += write(_writer, ch, _err);
  980. break;
  981. }
  982. }
  983. else
  984. {
  985. size += write(_writer, ch, _err);
  986. }
  987. }
  988. return size;
  989. }
  990. int32_t write(WriterI* _writer, Error* _err, const StringView* _format, ...)
  991. {
  992. va_list argList;
  993. va_start(argList, _format);
  994. int32_t total = write(_writer, *_format, argList, _err);
  995. va_end(argList);
  996. return total;
  997. }
  998. int32_t write(WriterI* _writer, Error* _err, const char* _format, ...)
  999. {
  1000. va_list argList;
  1001. va_start(argList, _format);
  1002. int32_t total = write(_writer, _format, argList, _err);
  1003. va_end(argList);
  1004. return total;
  1005. }
  1006. int32_t vsnprintf(char* _out, int32_t _max, const char* _format, va_list _argList)
  1007. {
  1008. if (1 < _max)
  1009. {
  1010. StaticMemoryBlockWriter writer(_out, uint32_t(_max) );
  1011. Error err;
  1012. va_list argListCopy;
  1013. va_copy(argListCopy, _argList);
  1014. int32_t size = write(&writer, _format, argListCopy, &err);
  1015. va_end(argListCopy);
  1016. if (err.isOk() )
  1017. {
  1018. size += write(&writer, '\0', &err);
  1019. return size - 1 /* size without '\0' terminator */;
  1020. }
  1021. else
  1022. {
  1023. _out[_max-1] = '\0';
  1024. }
  1025. }
  1026. Error err;
  1027. SizerWriter sizer;
  1028. va_list argListCopy;
  1029. va_copy(argListCopy, _argList);
  1030. int32_t total = write(&sizer, _format, argListCopy, &err);
  1031. va_end(argListCopy);
  1032. return total;
  1033. }
  1034. int32_t snprintf(char* _out, int32_t _max, const char* _format, ...)
  1035. {
  1036. va_list argList;
  1037. va_start(argList, _format);
  1038. int32_t total = vsnprintf(_out, _max, _format, argList);
  1039. va_end(argList);
  1040. return total;
  1041. }
  1042. int32_t vprintf(const char* _format, va_list _argList)
  1043. {
  1044. Error err;
  1045. va_list argListCopy;
  1046. va_copy(argListCopy, _argList);
  1047. int32_t total = write(getStdOut(), _format, argListCopy, &err);
  1048. va_end(argListCopy);
  1049. return total;
  1050. }
  1051. int32_t printf(const char* _format, ...)
  1052. {
  1053. va_list argList;
  1054. va_start(argList, _format);
  1055. int32_t total = vprintf(_format, argList);
  1056. va_end(argList);
  1057. return total;
  1058. }
  1059. static const char s_units[] = { 'B', 'k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y' };
  1060. template<uint32_t Kilo, char KiloCh0, char KiloCh1, CharFn fn>
  1061. inline int32_t prettify(char* _out, int32_t _count, uint64_t _value)
  1062. {
  1063. uint8_t idx = 0;
  1064. double value = double(_value);
  1065. while (_value != (_value&0x7ff)
  1066. && idx < BX_COUNTOF(s_units) )
  1067. {
  1068. _value /= Kilo;
  1069. value *= 1.0/double(Kilo);
  1070. ++idx;
  1071. }
  1072. return snprintf(_out, _count, "%0.2f %c%c%c", value
  1073. , fn(s_units[idx])
  1074. , idx > 0 ? KiloCh0 : '\0'
  1075. , KiloCh1
  1076. );
  1077. }
  1078. int32_t prettify(char* _out, int32_t _count, uint64_t _value, Units::Enum _units)
  1079. {
  1080. if (Units::Kilo == _units)
  1081. {
  1082. return prettify<1000, 'B', '\0', toNoop>(_out, _count, _value);
  1083. }
  1084. return prettify<1024, 'i', 'B', toUpper>(_out, _count, _value);
  1085. }
  1086. } // namespace bx