string.cpp 25 KB

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