tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Inline zero_encode implementation to speed up it 25% more.

Browse Source
This commit is contained in:
levlam 2023-07-26 15:07:15 +03:00
parent c50afa3820
commit 65710fcec6
1 changed files with 50 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

92
tdutils/td/utils/misc.cpp
View File

@ -203,34 +203,14 @@ string buffer_to_hex(Slice buffer) {
return res; return res;
} }
namespace { string zero_encode(Slice data) {
template <class F>
string x_decode(Slice s, F &&f) {
auto buffer = StackAllocator::alloc(1024); auto buffer = StackAllocator::alloc(1024);
auto res = StringBuilder(buffer.as_slice(), true); auto res = StringBuilder(buffer.as_slice(), true);
for (size_t n = s.size(), i = 0; i < n; i++) { for (size_t n = data.size(), i = 0; i < n; i++) {
if (i + 1 < n && f(s[i])) { res.push_back(data[i]);
for (int cnt = static_cast<unsigned char>(s[i + 1]); cnt > 0; cnt--) { if (data[i] == 0) {
res.push_back(s[i]);
}
i++;
continue;
}
res.push_back(s[i]);
}
return res.as_cslice().str();
}
template <class F>
string x_encode(Slice s, F &&f) {
auto buffer = StackAllocator::alloc(1024);
auto res = StringBuilder(buffer.as_slice(), true);
for (size_t n = s.size(), i = 0; i < n; i++) {
res.push_back(s[i]);
if (f(s[i])) {
unsigned char cnt = 1; unsigned char cnt = 1;
while (cnt < 250 && i + cnt < n && s[i + cnt] == s[i]) { while (cnt < 250 && i + cnt < n && data[i + cnt] == data[i]) {
cnt++; cnt++;
} }
res.push_back(static_cast<char>(cnt)); res.push_back(static_cast<char>(cnt));
@ -240,27 +220,55 @@ string x_encode(Slice s, F &&f) {
return res.as_cslice().str(); return res.as_cslice().str();
} }
bool is_zero(unsigned char c) { string zero_decode(Slice data) {
return c == 0; auto buffer = StackAllocator::alloc(1024);
auto res = StringBuilder(buffer.as_slice(), true);
for (size_t n = data.size(), i = 0; i < n; i++) {
if (data[i] == 0 && i + 1 < n) {
for (int cnt = static_cast<unsigned char>(data[i + 1]); cnt > 0; cnt--) {
res.push_back(data[i]);
}
i++;
continue;
}
res.push_back(data[i]);
}
return res.as_cslice().str();
} }
bool is_zero_or_one(unsigned char c) { string zero_one_encode(Slice data) {
return c == 0 || c == 0xff; auto buffer = StackAllocator::alloc(1024);
auto res = StringBuilder(buffer.as_slice(), true);
for (size_t n = data.size(), i = 0; i < n; i++) {
res.push_back(data[i]);
auto c = static_cast<unsigned char>(data[i]);
if (c == 0 || c == 0xff) {
unsigned char cnt = 1;
while (cnt < 250 && i + cnt < n && data[i + cnt] == data[i]) {
cnt++;
}
res.push_back(static_cast<char>(cnt));
i += cnt - 1;
}
}
return res.as_cslice().str();
} }
} // namespace string zero_one_decode(Slice data) {
auto buffer = StackAllocator::alloc(1024);
std::string zero_encode(Slice data) { auto res = StringBuilder(buffer.as_slice(), true);
return x_encode(data, is_zero); for (size_t n = data.size(), i = 0; i < n; i++) {
} auto c = static_cast<unsigned char>(data[i]);
std::string zero_decode(Slice data) { if ((c == 0 || c == 0xff) && i + 1 < n) {
return x_decode(data, is_zero); for (int cnt = static_cast<unsigned char>(data[i + 1]); cnt > 0; cnt--) {
} res.push_back(data[i]);
std::string zero_one_encode(Slice data) { }
return x_encode(data, is_zero_or_one); i++;
} continue;
std::string zero_one_decode(Slice data) { }
return x_decode(data, is_zero_or_one); res.push_back(data[i]);
}
return res.as_cslice().str();
} }
} // namespace td } // namespace td