StringBuffer.h 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
  1. #ifndef STRINGBUFFER_H
  2. #define STRINGBUFFER_H
  3. #include <cstring>
  4. #include <iostream>
  5. #include "utils/Types.h"
  6. // ignore buffer overflow warnings with snprintf
  7. #pragma GCC diagnostic push
  8. #pragma GCC diagnostic ignored "-Wformat-truncation"
  9. template<int N>
  10. class StringBuffer final {
  11. int length;
  12. Hash hash;
  13. char data[N];
  14. void addToHash(char c) {
  15. hash = static_cast<Hash>(2120251889) * hash + static_cast<Hash>(c);
  16. }
  17. void addToHash(int from, int to) {
  18. for(int i = from; i < to; i++) {
  19. addToHash(data[i]);
  20. }
  21. }
  22. public:
  23. StringBuffer() : length(0), hash(0) {
  24. data[0] = '\0';
  25. }
  26. template<typename T>
  27. StringBuffer(const T& t) : StringBuffer() {
  28. append(t);
  29. }
  30. bool operator==(const char* str) const {
  31. return strcmp(data, str) == 0;
  32. }
  33. bool operator==(const StringBuffer& other) const {
  34. return length == other.length && other == data;
  35. }
  36. bool operator!=(const char* str) const {
  37. return !((*this) == str);
  38. }
  39. bool operator!=(const StringBuffer& other) const {
  40. return !((*this) == other);
  41. }
  42. operator const char*() const {
  43. return data;
  44. }
  45. char operator[](int index) const {
  46. return data[index];
  47. }
  48. int getLength() const {
  49. return length;
  50. }
  51. StringBuffer& append(char c) {
  52. if(length < N - 1) {
  53. data[length++] = c;
  54. addToHash(c);
  55. data[length] = '\0';
  56. }
  57. return *this;
  58. }
  59. StringBuffer& append(signed char c) {
  60. return append(static_cast<char>(c));
  61. }
  62. StringBuffer& append(unsigned char c) {
  63. return append(static_cast<char>(c));
  64. }
  65. StringBuffer& append(const char* str) {
  66. return appendString<const char*>(str);
  67. }
  68. StringBuffer& append(const signed char* str) {
  69. return appendString<const char*>(str);
  70. }
  71. StringBuffer& append(const unsigned char* str) {
  72. return appendString<const unsigned char*>(str);
  73. }
  74. template<typename T>
  75. StringBuffer& append(const char* format, const T& t) {
  76. int left = N - length;
  77. int written = snprintf(data + length, left, format, t);
  78. int oldLength = length;
  79. if(written < left) {
  80. length += written;
  81. } else {
  82. length = N - 1;
  83. }
  84. addToHash(oldLength, length);
  85. return *this;
  86. }
  87. StringBuffer& append(signed short s) {
  88. return append("%hd", s);
  89. }
  90. StringBuffer& append(unsigned short s) {
  91. return append("%hu", s);
  92. }
  93. StringBuffer& append(signed int i) {
  94. return append("%d", i);
  95. }
  96. StringBuffer& append(unsigned int i) {
  97. return append("%u", i);
  98. }
  99. StringBuffer& append(float f) {
  100. return append("%.2f", f);
  101. }
  102. StringBuffer& append(bool b) {
  103. return b ? append("true") : append("false");
  104. }
  105. template<typename T>
  106. StringBuffer& append(const T& t) {
  107. t.toString(*this);
  108. return *this;
  109. }
  110. template<int L>
  111. void toString(StringBuffer<L>& s) const {
  112. if(reinterpret_cast<const void*>(this) ==
  113. reinterpret_cast<const void*>(&s)) {
  114. return;
  115. }
  116. s.append(data);
  117. }
  118. StringBuffer& appendUnicode(unsigned int c) {
  119. if(c < (1 << 7)) {
  120. append(static_cast<char>(c & 0x7F));
  121. } else if(c < (1 << 11)) {
  122. append(static_cast<char>(((c >> 6) & 0x1F) | 0xC0));
  123. append(static_cast<char>(((c >> 0) & 0x3F) | 0x80));
  124. } else if(c < (1 << 16)) {
  125. append(static_cast<char>(((c >> 12) & 0x0F) | 0xE0));
  126. append(static_cast<char>(((c >> 6) & 0x3F) | 0x80));
  127. append(static_cast<char>(((c >> 0) & 0x3F) | 0x80));
  128. } else if(c < (1 << 21)) {
  129. append(static_cast<char>(((c >> 18) & 0x07) | 0xF0));
  130. append(static_cast<char>(((c >> 12) & 0x3F) | 0x80));
  131. append(static_cast<char>(((c >> 6) & 0x3F) | 0x80));
  132. append(static_cast<char>(((c >> 0) & 0x3F) | 0x80));
  133. }
  134. return *this;
  135. }
  136. StringBuffer& clear() {
  137. length = 0;
  138. hash = 0;
  139. data[0] = '\0';
  140. return *this;
  141. }
  142. Hash hashCode() const {
  143. return hash;
  144. }
  145. void print() const {
  146. std::cout << data;
  147. }
  148. void printLine() const {
  149. std::cout << data << '\n';
  150. }
  151. private:
  152. template<typename T>
  153. StringBuffer& appendString(T str) {
  154. for(int i = 0; length < N - 1 && str[i] != '\0'; length++, i++) {
  155. data[length] = str[i];
  156. addToHash(str[i]);
  157. }
  158. data[length] = '\0';
  159. return *this;
  160. }
  161. };
  162. template<int N>
  163. bool operator==(const char* str, const StringBuffer<N> buffer) {
  164. return buffer == str;
  165. }
  166. template<int N>
  167. bool operator!=(const char* str, const StringBuffer<N> buffer) {
  168. return buffer != str;
  169. }
  170. #pragma GCC diagnostic pop
  171. #endif