string.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299
  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/allocator.h>
  6. #include <bx/file.h>
  7. #include <bx/hash.h>
  8. #include <bx/string.h>
  9. namespace bx
  10. {
  11. inline bool isInRange(char _ch, char _from, char _to)
  12. {
  13. return unsigned(_ch - _from) <= unsigned(_to-_from);
  14. }
  15. bool isSpace(char _ch)
  16. {
  17. return ' ' == _ch
  18. || '\t' == _ch
  19. || '\n' == _ch
  20. || '\v' == _ch
  21. || '\f' == _ch
  22. || '\r' == _ch
  23. ;
  24. }
  25. bool isUpper(char _ch)
  26. {
  27. return isInRange(_ch, 'A', 'Z');
  28. }
  29. bool isLower(char _ch)
  30. {
  31. return isInRange(_ch, 'a', 'z');
  32. }
  33. bool isAlpha(char _ch)
  34. {
  35. return isLower(_ch) || isUpper(_ch);
  36. }
  37. bool isNumeric(char _ch)
  38. {
  39. return isInRange(_ch, '0', '9');
  40. }
  41. bool isAlphaNum(char _ch)
  42. {
  43. return false
  44. || isAlpha(_ch)
  45. || isNumeric(_ch)
  46. ;
  47. }
  48. bool isHexNum(char _ch)
  49. {
  50. return false
  51. || isInRange(toLower(_ch), 'a', 'f')
  52. || isNumeric(_ch)
  53. ;
  54. }
  55. bool isPrint(char _ch)
  56. {
  57. return isInRange(_ch, ' ', '~');
  58. }
  59. typedef bool (*CharTestFn)(char _ch);
  60. template<CharTestFn fn>
  61. inline bool isCharTest(const StringView& _str)
  62. {
  63. bool result = true;
  64. for (const char* ptr = _str.getPtr(), *term = _str.getTerm()
  65. ; ptr != term && result
  66. ; ++ptr
  67. )
  68. {
  69. result &= fn(*ptr);
  70. }
  71. return result;
  72. }
  73. bool isSpace(const StringView& _str)
  74. {
  75. return isCharTest<isSpace>(_str);
  76. }
  77. bool isUpper(const StringView& _str)
  78. {
  79. return isCharTest<isUpper>(_str);
  80. }
  81. bool isLower(const StringView& _str)
  82. {
  83. return isCharTest<isLower>(_str);
  84. }
  85. bool isAlpha(const StringView& _str)
  86. {
  87. return isCharTest<isAlpha>(_str);
  88. }
  89. bool isNumeric(const StringView& _str)
  90. {
  91. return isCharTest<isNumeric>(_str);
  92. }
  93. bool isAlphaNum(const StringView& _str)
  94. {
  95. return isCharTest<isAlphaNum>(_str);
  96. }
  97. bool isHexNum(const StringView& _str)
  98. {
  99. return isCharTest<isHexNum>(_str);
  100. }
  101. bool isPrint(const StringView& _str)
  102. {
  103. return isCharTest<isPrint>(_str);
  104. }
  105. char toLower(char _ch)
  106. {
  107. return _ch + (isUpper(_ch) ? 0x20 : 0);
  108. }
  109. void toLowerUnsafe(char* _inOutStr, int32_t _len)
  110. {
  111. for (int32_t ii = 0; ii < _len; ++ii)
  112. {
  113. _inOutStr[ii] = toLower(_inOutStr[ii]);
  114. }
  115. }
  116. void toLower(char* _inOutStr, int32_t _max)
  117. {
  118. const int32_t len = strLen(_inOutStr, _max);
  119. toLowerUnsafe(_inOutStr, len);
  120. }
  121. char toUpper(char _ch)
  122. {
  123. return _ch - (isLower(_ch) ? 0x20 : 0);
  124. }
  125. void toUpperUnsafe(char* _inOutStr, int32_t _len)
  126. {
  127. for (int32_t ii = 0; ii < _len; ++ii)
  128. {
  129. _inOutStr[ii] = toUpper(_inOutStr[ii]);
  130. }
  131. }
  132. void toUpper(char* _inOutStr, int32_t _max)
  133. {
  134. const int32_t len = strLen(_inOutStr, _max);
  135. toUpperUnsafe(_inOutStr, len);
  136. }
  137. typedef char (*CharFn)(char _ch);
  138. inline char toNoop(char _ch)
  139. {
  140. return _ch;
  141. }
  142. template<CharFn fn>
  143. inline int32_t strCmp(const char* _lhs, int32_t _lhsMax, const char* _rhs, int32_t _rhsMax)
  144. {
  145. int32_t max = min(_lhsMax, _rhsMax);
  146. for (
  147. ; 0 < max && fn(*_lhs) == fn(*_rhs)
  148. ; ++_lhs, ++_rhs, --max
  149. )
  150. {
  151. if (*_lhs == '\0'
  152. || *_rhs == '\0')
  153. {
  154. break;
  155. }
  156. }
  157. if (0 == max)
  158. {
  159. return _lhsMax == _rhsMax ? 0 : _lhsMax > _rhsMax ? 1 : -1;
  160. }
  161. return 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_ASSERT(NULL != _dst, "_dst can't be NULL!");
  265. BX_ASSERT(NULL != _src, "_src can't be NULL!");
  266. BX_ASSERT(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_ASSERT(NULL != _dst, "_dst can't be NULL!");
  281. BX_ASSERT(NULL != _src, "_src can't be NULL!");
  282. BX_ASSERT(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. StringView strFind(const StringView& _str, char _ch)
  307. {
  308. const char* ptr = strFindUnsafe(_str.getPtr(), _str.getLength(), _ch);
  309. if (NULL == ptr)
  310. {
  311. return StringView(_str.getTerm(), _str.getTerm() );
  312. }
  313. return StringView(ptr, ptr+1);
  314. }
  315. inline const char* strRFindUnsafe(const char* _str, int32_t _len, char _ch)
  316. {
  317. for (int32_t ii = _len-1; 0 <= ii; --ii)
  318. {
  319. if (_str[ii] == _ch)
  320. {
  321. return &_str[ii];
  322. }
  323. }
  324. return NULL;
  325. }
  326. StringView strRFind(const StringView& _str, char _ch)
  327. {
  328. const char* ptr = strRFindUnsafe(_str.getPtr(), _str.getLength(), _ch);
  329. if (NULL == ptr)
  330. {
  331. return StringView(_str.getTerm(), _str.getTerm() );
  332. }
  333. return StringView(ptr, ptr+1);
  334. }
  335. template<CharFn fn>
  336. inline const char* strFind(const char* _str, int32_t _strMax, const char* _find, int32_t _findMax)
  337. {
  338. const char* ptr = _str;
  339. int32_t stringLen = _strMax;
  340. const int32_t findLen = _findMax;
  341. for (; stringLen >= findLen; ++ptr, --stringLen)
  342. {
  343. // Find start of the string.
  344. while (fn(*ptr) != fn(*_find) )
  345. {
  346. ++ptr;
  347. --stringLen;
  348. // Search pattern lenght can't be longer than the string.
  349. if (findLen > stringLen)
  350. {
  351. return NULL;
  352. }
  353. }
  354. if (0 == strCmp<fn>(ptr, _findMax, _find, _findMax) )
  355. {
  356. return ptr;
  357. }
  358. }
  359. return NULL;
  360. }
  361. StringView strFind(const StringView& _str, const StringView& _find, int32_t _num)
  362. {
  363. int32_t len = min(_find.getLength(), _num);
  364. const char* ptr = strFind<toNoop>(
  365. _str.getPtr()
  366. , _str.getLength()
  367. , _find.getPtr()
  368. , len
  369. );
  370. if (NULL == ptr)
  371. {
  372. return StringView(_str.getTerm(), _str.getTerm() );
  373. }
  374. return StringView(ptr, len);
  375. }
  376. StringView strFindI(const StringView& _str, const StringView& _find, int32_t _num)
  377. {
  378. int32_t len = min(_find.getLength(), _num);
  379. const char* ptr = strFind<toLower>(
  380. _str.getPtr()
  381. , _str.getLength()
  382. , _find.getPtr()
  383. , len
  384. );
  385. if (NULL == ptr)
  386. {
  387. return StringView(_str.getTerm(), _str.getTerm() );
  388. }
  389. return StringView(ptr, len);
  390. }
  391. StringView strLTrim(const StringView& _str, const StringView& _chars)
  392. {
  393. const char* ptr = _str.getPtr();
  394. const char* chars = _chars.getPtr();
  395. const uint32_t charsLen = _chars.getLength();
  396. for (uint32_t ii = 0, len = _str.getLength(); ii < len; ++ii)
  397. {
  398. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]) )
  399. {
  400. return StringView(ptr + ii, len-ii);
  401. }
  402. }
  403. return _str;
  404. }
  405. StringView strLTrimSpace(const StringView& _str)
  406. {
  407. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  408. {
  409. if (!isSpace(*ptr) )
  410. {
  411. return StringView(ptr, term);
  412. }
  413. }
  414. return StringView(_str.getTerm(), _str.getTerm() );
  415. }
  416. StringView strLTrimNonSpace(const StringView& _str)
  417. {
  418. for (const char* ptr = _str.getPtr(), *term = _str.getTerm(); ptr != term; ++ptr)
  419. {
  420. if (isSpace(*ptr) )
  421. {
  422. return StringView(ptr, term);
  423. }
  424. }
  425. return StringView(_str.getTerm(), _str.getTerm() );
  426. }
  427. StringView strRTrim(const StringView& _str, const StringView& _chars)
  428. {
  429. if (!_str.isEmpty() )
  430. {
  431. const char* ptr = _str.getPtr();
  432. const char* chars = _chars.getPtr();
  433. const uint32_t charsLen = _chars.getLength();
  434. for (int32_t len = _str.getLength(), ii = len - 1; 0 <= ii; --ii)
  435. {
  436. if (NULL == strFindUnsafe(chars, charsLen, ptr[ii]))
  437. {
  438. return StringView(ptr, ii + 1);
  439. }
  440. }
  441. }
  442. return _str;
  443. }
  444. StringView strRTrimSpace(const StringView& _str)
  445. {
  446. if (!_str.isEmpty() )
  447. {
  448. const char* ptr = _str.getPtr();
  449. for (int32_t len = _str.getLength(), ii = len - 1; 0 <= ii; --ii)
  450. {
  451. if (!isSpace(ptr[ii]) )
  452. {
  453. return StringView(ptr, ii + 1);
  454. }
  455. }
  456. }
  457. return _str;
  458. }
  459. StringView strTrim(const StringView& _str, const StringView& _chars)
  460. {
  461. return strLTrim(strRTrim(_str, _chars), _chars);
  462. }
  463. StringView strTrimSpace(const StringView& _str)
  464. {
  465. return strLTrimSpace(strRTrimSpace(_str) );
  466. }
  467. constexpr uint32_t kFindStep = 1024;
  468. StringView strFindNl(const StringView& _str)
  469. {
  470. StringView str(_str);
  471. // This method returns the character past the \n, so
  472. // there is no need to look for he \r which preceedes it.
  473. StringView eol = strFind(str, "\n");
  474. if (!eol.isEmpty() )
  475. {
  476. return StringView(eol.getTerm(), str.getTerm() );
  477. }
  478. return StringView(_str.getTerm(), _str.getTerm() );
  479. }
  480. StringView strFindEol(const StringView& _str)
  481. {
  482. StringView str(_str);
  483. for (; str.getPtr() != _str.getTerm()
  484. ; str = StringView(min(str.getPtr() + kFindStep, _str.getTerm() ), min(str.getPtr() + kFindStep*2, _str.getTerm() ) )
  485. )
  486. {
  487. StringView eol = strFind(str, "\r\n");
  488. if (!eol.isEmpty() )
  489. {
  490. return StringView(eol.getPtr(), _str.getTerm() );
  491. }
  492. eol = strFind(str, '\n');
  493. if (!eol.isEmpty() )
  494. {
  495. return StringView(eol.getPtr(), _str.getTerm() );
  496. }
  497. }
  498. return StringView(_str.getTerm(), _str.getTerm() );
  499. }
  500. static const char* strSkipWord(const char* _str, int32_t _max)
  501. {
  502. for (char ch = *_str++; 0 < _max && (isAlphaNum(ch) || '_' == ch); ch = *_str++, --_max) {};
  503. return _str-1;
  504. }
  505. StringView strWord(const StringView& _str)
  506. {
  507. const char* ptr = _str.getPtr();
  508. const char* term = strSkipWord(ptr, _str.getLength() );
  509. return StringView(ptr, term);
  510. }
  511. StringView strFindBlock(const StringView& _str, char _open, char _close)
  512. {
  513. const char* curr = _str.getPtr();
  514. const char* term = _str.getTerm();
  515. const char* start = NULL;
  516. int32_t count = 0;
  517. for (char ch = *curr; curr != term && count >= 0; ch = *(++curr) )
  518. {
  519. if (ch == _open)
  520. {
  521. if (0 == count)
  522. {
  523. start = curr;
  524. }
  525. ++count;
  526. }
  527. else if (ch == _close)
  528. {
  529. --count;
  530. if (NULL == start)
  531. {
  532. break;
  533. }
  534. if (0 == count)
  535. {
  536. return StringView(start, curr+1);
  537. }
  538. }
  539. }
  540. return StringView(term, term);
  541. }
  542. StringView normalizeEolLf(char* _out, int32_t _size, const StringView& _str)
  543. {
  544. const char* start = _out;
  545. const char* end = _out;
  546. if (0 < _size)
  547. {
  548. const char* curr = _str.getPtr();
  549. const char* term = _str.getTerm();
  550. end = _out + _size;
  551. for (char ch = *curr; curr != term && _out < end; ch = *(++curr) )
  552. {
  553. if ('\r' != ch)
  554. {
  555. *_out++ = ch;
  556. }
  557. }
  558. end = _out;
  559. }
  560. return StringView(start, end);
  561. }
  562. StringView findIdentifierMatch(const StringView& _str, const StringView& _word)
  563. {
  564. const int32_t len = _word.getLength();
  565. StringView ptr = strFind(_str, _word);
  566. for (; !ptr.isEmpty(); ptr = strFind(StringView(ptr.getPtr() + len, _str.getTerm() ), _word) )
  567. {
  568. char ch = ptr.getPtr() != _str.getPtr() ? *(ptr.getPtr() - 1) : ' ';
  569. if (isAlphaNum(ch) || '_' == ch)
  570. {
  571. continue;
  572. }
  573. ch = *(ptr.getPtr() + len);
  574. if (isAlphaNum(ch) || '_' == ch)
  575. {
  576. continue;
  577. }
  578. return ptr;
  579. }
  580. return StringView(_str.getTerm(), _str.getTerm() );
  581. }
  582. StringView findIdentifierMatch(const StringView& _str, const char** _words, int32_t _num)
  583. {
  584. int32_t ii = 0;
  585. for (StringView word = *_words; ii < _num && !word.isEmpty(); ++ii, ++_words, word = *_words)
  586. {
  587. StringView match = findIdentifierMatch(_str, word);
  588. if (!match.isEmpty() )
  589. {
  590. return match;
  591. }
  592. }
  593. return StringView(_str.getTerm(), _str.getTerm() );
  594. }
  595. namespace
  596. {
  597. struct Param
  598. {
  599. Param()
  600. : width(0)
  601. , base(10)
  602. , prec(INT32_MAX)
  603. , fill(' ')
  604. , bits(0)
  605. , left(false)
  606. , upper(false)
  607. , spec(false)
  608. , sign(false)
  609. {
  610. }
  611. int32_t width;
  612. int32_t base;
  613. int32_t prec;
  614. char fill;
  615. uint8_t bits;
  616. bool left;
  617. bool upper;
  618. bool spec;
  619. bool sign;
  620. };
  621. static int32_t write(WriterI* _writer, const char* _str, int32_t _len, const Param& _param, Error* _err)
  622. {
  623. int32_t size = 0;
  624. int32_t len = (int32_t)strLen(_str, _len);
  625. if (_param.width > 0)
  626. {
  627. len = min(_param.width, len);
  628. }
  629. const bool hasMinus = (NULL != _str && '-' == _str[0]);
  630. const bool hasSign = _param.sign || hasMinus;
  631. char sign = hasSign ? hasMinus ? '-' : '+' : '\0';
  632. const char* str = _str;
  633. if (hasMinus)
  634. {
  635. str++;
  636. len--;
  637. }
  638. int32_t padding = _param.width > len ? _param.width - len - hasSign: 0;
  639. if (!_param.left)
  640. {
  641. if (' ' != _param.fill
  642. && '\0' != sign)
  643. {
  644. size += write(_writer, sign, _err);
  645. sign = '\0';
  646. }
  647. size += writeRep(_writer, _param.fill, max(0, padding), _err);
  648. }
  649. if ('\0' != sign)
  650. {
  651. size += write(_writer, sign, _err);
  652. }
  653. if (NULL == _str)
  654. {
  655. size += write(_writer, "(null)", 6, _err);
  656. }
  657. else if (_param.upper)
  658. {
  659. for (int32_t ii = 0; ii < len; ++ii)
  660. {
  661. size += write(_writer, toUpper(str[ii]), _err);
  662. }
  663. }
  664. else
  665. {
  666. size += write(_writer, str, len, _err);
  667. }
  668. if (_param.left)
  669. {
  670. size += writeRep(_writer, _param.fill, padding, _err);
  671. }
  672. return size;
  673. }
  674. static int32_t write(WriterI* _writer, char _ch, const Param& _param, Error* _err)
  675. {
  676. return write(_writer, &_ch, 1, _param, _err);
  677. }
  678. static int32_t write(WriterI* _writer, const char* _str, const Param& _param, Error* _err)
  679. {
  680. return write(_writer, _str, _param.prec, _param, _err);
  681. }
  682. static int32_t write(WriterI* _writer, const StringView& _str, const Param& _param, Error* _err)
  683. {
  684. return write(_writer, _str.getPtr(), min(_param.prec, _str.getLength() ), _param, _err);
  685. }
  686. static int32_t write(WriterI* _writer, int32_t _i, const Param& _param, Error* _err)
  687. {
  688. char str[33];
  689. int32_t len = toString(str, sizeof(str), _i, _param.base);
  690. if (len == 0)
  691. {
  692. return 0;
  693. }
  694. return write(_writer, str, len, _param, _err);
  695. }
  696. static int32_t write(WriterI* _writer, int64_t _i, const Param& _param, Error* _err)
  697. {
  698. char str[33];
  699. int32_t len = toString(str, sizeof(str), _i, _param.base);
  700. if (len == 0)
  701. {
  702. return 0;
  703. }
  704. return write(_writer, str, len, _param, _err);
  705. }
  706. static int32_t write(WriterI* _writer, uint32_t _u, const Param& _param, Error* _err)
  707. {
  708. char str[33];
  709. int32_t len = toString(str, sizeof(str), _u, _param.base);
  710. if (len == 0)
  711. {
  712. return 0;
  713. }
  714. return write(_writer, str, len, _param, _err);
  715. }
  716. static int32_t write(WriterI* _writer, uint64_t _u, const Param& _param, Error* _err)
  717. {
  718. char str[33];
  719. int32_t len = toString(str, sizeof(str), _u, _param.base);
  720. if (len == 0)
  721. {
  722. return 0;
  723. }
  724. return write(_writer, str, len, _param, _err);
  725. }
  726. static int32_t write(WriterI* _writer, double _d, const Param& _param, Error* _err)
  727. {
  728. char str[1024];
  729. int32_t len = toString(str, sizeof(str), _d);
  730. if (len == 0)
  731. {
  732. return 0;
  733. }
  734. const char* dot = strFind(str, INT32_MAX, '.');
  735. if (NULL != dot)
  736. {
  737. const int32_t prec = INT32_MAX == _param.prec ? 6 : _param.prec;
  738. const char* strEnd = str + len;
  739. const char* exponent = strFind(str, INT32_MAX, 'e');
  740. const char* fracEnd = NULL != exponent ? exponent : strEnd;
  741. char* fracBegin = &str[dot - str + min(prec + _param.spec, 1)];
  742. const int32_t curPrec = int32_t(fracEnd - fracBegin);
  743. // Move exponent to its final location after trimming or adding extra 0s.
  744. if (fracEnd != strEnd)
  745. {
  746. const int32_t exponentLen = int32_t(strEnd - fracEnd);
  747. char* finalExponentPtr = &fracBegin[prec];
  748. memMove(finalExponentPtr, fracEnd, exponentLen);
  749. finalExponentPtr[exponentLen] = '\0';
  750. len = int32_t(&finalExponentPtr[exponentLen] - str);
  751. }
  752. else
  753. {
  754. len = (int32_t)(fracBegin + prec - str);
  755. }
  756. if (curPrec < prec)
  757. {
  758. for (int32_t ii = curPrec; ii < prec; ++ii)
  759. {
  760. fracBegin[ii] = '0';
  761. }
  762. }
  763. }
  764. return write(_writer, str, len, _param, _err);
  765. }
  766. static int32_t write(WriterI* _writer, const void* _ptr, const Param& _param, Error* _err)
  767. {
  768. char str[35] = "0x";
  769. int32_t len = toString(str + 2, sizeof(str) - 2, uint32_t(uintptr_t(_ptr) ), 16);
  770. if (len == 0)
  771. {
  772. return 0;
  773. }
  774. len += 2;
  775. return write(_writer, str, len, _param, _err);
  776. }
  777. } // anonymous namespace
  778. int32_t write(WriterI* _writer, const StringView& _format, va_list _argList, Error* _err)
  779. {
  780. MemoryReader reader(_format.getPtr(), _format.getLength() );
  781. int32_t size = 0;
  782. while (_err->isOk() )
  783. {
  784. char ch = '\0';
  785. Error err;
  786. read(&reader, ch, &err);
  787. if (!_err->isOk()
  788. || !err.isOk() )
  789. {
  790. break;
  791. }
  792. else if ('%' == ch)
  793. {
  794. // %[Flags][Width][.Precision][Leegth]Type
  795. read(&reader, ch, &err);
  796. Param param;
  797. // Reference(s):
  798. // - Flags field
  799. // https://en.wikipedia.org/wiki/Printf_format_string#Flags_field
  800. //
  801. while (err.isOk()
  802. && ( ' ' == ch
  803. || '-' == ch
  804. || '+' == ch
  805. || '0' == ch
  806. || '#' == ch)
  807. )
  808. {
  809. switch (ch)
  810. {
  811. default:
  812. case ' ': param.fill = ' '; break;
  813. case '-': param.left = true; break;
  814. case '+': param.sign = true; break;
  815. case '0': param.fill = '0'; break;
  816. case '#': param.spec = true; break;
  817. }
  818. read(&reader, ch, &err);
  819. }
  820. if (param.left)
  821. {
  822. param.fill = ' ';
  823. }
  824. // Reference(s):
  825. // - Width field
  826. // https://en.wikipedia.org/wiki/Printf_format_string#Width_field
  827. //
  828. if ('*' == ch)
  829. {
  830. read(&reader, ch, &err);
  831. param.width = va_arg(_argList, int32_t);
  832. if (0 > param.width)
  833. {
  834. param.left = true;
  835. param.width = -param.width;
  836. }
  837. }
  838. else
  839. {
  840. while (err.isOk()
  841. && isNumeric(ch) )
  842. {
  843. param.width = param.width * 10 + ch - '0';
  844. read(&reader, ch, &err);
  845. }
  846. }
  847. // Reference(s):
  848. // - Precision field
  849. // https://en.wikipedia.org/wiki/Printf_format_string#Precision_field
  850. if ('.' == ch)
  851. {
  852. read(&reader, ch, &err);
  853. if ('*' == ch)
  854. {
  855. read(&reader, ch, &err);
  856. param.prec = va_arg(_argList, int32_t);
  857. }
  858. else
  859. {
  860. param.prec = 0;
  861. while (err.isOk()
  862. && isNumeric(ch) )
  863. {
  864. param.prec = param.prec * 10 + ch - '0';
  865. read(&reader, ch, &err);
  866. }
  867. }
  868. }
  869. // Reference(s):
  870. // - Length field
  871. // https://en.wikipedia.org/wiki/Printf_format_string#Length_field
  872. while (err.isOk()
  873. && ( 'h' == ch
  874. || 'I' == ch
  875. || 'l' == ch
  876. || 'j' == ch
  877. || 't' == ch
  878. || 'z' == ch)
  879. )
  880. {
  881. switch (ch)
  882. {
  883. default: break;
  884. case 'j': param.bits = sizeof(intmax_t )*8; break;
  885. case 't': param.bits = sizeof(ptrdiff_t)*8; break;
  886. case 'z': param.bits = sizeof(size_t )*8; break;
  887. case 'h': case 'I': case 'l':
  888. switch (ch)
  889. {
  890. case 'h': param.bits = sizeof(short int)*8; break;
  891. case 'l': param.bits = sizeof(long int )*8; break;
  892. default: break;
  893. }
  894. read(&reader, ch, &err);
  895. switch (ch)
  896. {
  897. case 'h': param.bits = sizeof(signed char )*8; break;
  898. case 'l': param.bits = sizeof(long long int)*8; break;
  899. case '3':
  900. case '6':
  901. read(&reader, ch, &err);
  902. switch (ch)
  903. {
  904. case '2': param.bits = sizeof(int32_t)*8; break;
  905. case '4': param.bits = sizeof(int64_t)*8; break;
  906. default: break;
  907. }
  908. break;
  909. default: seek(&reader, -1); break;
  910. }
  911. break;
  912. }
  913. read(&reader, ch, &err);
  914. }
  915. if (!err.isOk() )
  916. {
  917. break;
  918. }
  919. // Reference(s):
  920. // - Type field
  921. // https://en.wikipedia.org/wiki/Printf_format_string#Type_field
  922. switch (ch)
  923. {
  924. case 'c':
  925. size += write(_writer, char(va_arg(_argList, int32_t) ), param, _err);
  926. break;
  927. case 's':
  928. size += write(_writer, va_arg(_argList, const char*), param, _err);
  929. break;
  930. case 'S':
  931. size += write(_writer, *va_arg(_argList, const StringView*), param, _err);
  932. break;
  933. case 'o':
  934. param.base = 8;
  935. switch (param.bits)
  936. {
  937. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  938. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  939. }
  940. break;
  941. case 'i':
  942. case 'd':
  943. param.base = 10;
  944. switch (param.bits)
  945. {
  946. default: size += write(_writer, va_arg(_argList, int32_t), param, _err); break;
  947. case 64: size += write(_writer, va_arg(_argList, int64_t), param, _err); break;
  948. };
  949. break;
  950. case 'e':
  951. case 'E':
  952. case 'f':
  953. case 'F':
  954. case 'g':
  955. case 'G':
  956. param.upper = isUpper(ch);
  957. size += write(_writer, va_arg(_argList, double), param, _err);
  958. break;
  959. case 'p':
  960. size += write(_writer, va_arg(_argList, void*), param, _err);
  961. break;
  962. case 'x':
  963. case 'X':
  964. param.base = 16;
  965. param.upper = isUpper(ch);
  966. switch (param.bits)
  967. {
  968. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  969. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  970. }
  971. break;
  972. case 'u':
  973. param.base = 10;
  974. switch (param.bits)
  975. {
  976. default: size += write(_writer, va_arg(_argList, uint32_t), param, _err); break;
  977. case 64: size += write(_writer, va_arg(_argList, uint64_t), param, _err); break;
  978. }
  979. break;
  980. case 'n':
  981. *va_arg(_argList, int32_t*) = size;
  982. break;
  983. default:
  984. size += write(_writer, ch, _err);
  985. break;
  986. }
  987. }
  988. else
  989. {
  990. size += write(_writer, ch, _err);
  991. }
  992. }
  993. return size;
  994. }
  995. int32_t write(WriterI* _writer, Error* _err, const StringView* _format, ...)
  996. {
  997. va_list argList;
  998. va_start(argList, _format);
  999. int32_t total = write(_writer, *_format, argList, _err);
  1000. va_end(argList);
  1001. return total;
  1002. }
  1003. int32_t write(WriterI* _writer, Error* _err, const char* _format, ...)
  1004. {
  1005. va_list argList;
  1006. va_start(argList, _format);
  1007. int32_t total = write(_writer, _format, argList, _err);
  1008. va_end(argList);
  1009. return total;
  1010. }
  1011. int32_t vsnprintf(char* _out, int32_t _max, const char* _format, va_list _argList)
  1012. {
  1013. if (1 < _max)
  1014. {
  1015. StaticMemoryBlockWriter writer(_out, uint32_t(_max) );
  1016. Error err;
  1017. va_list argListCopy;
  1018. va_copy(argListCopy, _argList);
  1019. int32_t size = write(&writer, _format, argListCopy, &err);
  1020. va_end(argListCopy);
  1021. if (err.isOk() )
  1022. {
  1023. size += write(&writer, '\0', &err);
  1024. return size - 1 /* size without '\0' terminator */;
  1025. }
  1026. else
  1027. {
  1028. _out[_max-1] = '\0';
  1029. }
  1030. }
  1031. Error err;
  1032. SizerWriter sizer;
  1033. va_list argListCopy;
  1034. va_copy(argListCopy, _argList);
  1035. int32_t total = write(&sizer, _format, argListCopy, &err);
  1036. va_end(argListCopy);
  1037. return total;
  1038. }
  1039. int32_t snprintf(char* _out, int32_t _max, const char* _format, ...)
  1040. {
  1041. va_list argList;
  1042. va_start(argList, _format);
  1043. int32_t total = vsnprintf(_out, _max, _format, argList);
  1044. va_end(argList);
  1045. return total;
  1046. }
  1047. int32_t vprintf(const char* _format, va_list _argList)
  1048. {
  1049. Error err;
  1050. va_list argListCopy;
  1051. va_copy(argListCopy, _argList);
  1052. int32_t total = write(getStdOut(), _format, argListCopy, &err);
  1053. va_end(argListCopy);
  1054. return total;
  1055. }
  1056. int32_t printf(const char* _format, ...)
  1057. {
  1058. va_list argList;
  1059. va_start(argList, _format);
  1060. int32_t total = vprintf(_format, argList);
  1061. va_end(argList);
  1062. return total;
  1063. }
  1064. static const char s_units[] = { 'B', 'k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y' };
  1065. template<uint32_t Kilo, char KiloCh0, char KiloCh1, CharFn fn>
  1066. inline int32_t prettify(char* _out, int32_t _count, uint64_t _value)
  1067. {
  1068. uint8_t idx = 0;
  1069. double value = double(_value);
  1070. while (_value != (_value&0x7ff)
  1071. && idx < BX_COUNTOF(s_units) )
  1072. {
  1073. _value /= Kilo;
  1074. value *= 1.0/double(Kilo);
  1075. ++idx;
  1076. }
  1077. return snprintf(_out, _count, "%0.2f %c%c%c", value
  1078. , fn(s_units[idx])
  1079. , idx > 0 ? KiloCh0 : '\0'
  1080. , KiloCh1
  1081. );
  1082. }
  1083. int32_t prettify(char* _out, int32_t _count, uint64_t _value, Units::Enum _units)
  1084. {
  1085. if (Units::Kilo == _units)
  1086. {
  1087. return prettify<1000, 'B', '\0', toNoop>(_out, _count, _value);
  1088. }
  1089. return prettify<1024, 'i', 'B', toUpper>(_out, _count, _value);
  1090. }
  1091. } // namespace bx