upd: [imp] fmt performance optimization

This commit is contained in:
mutouyun 2022-12-03 17:59:03 +08:00
parent f2e981c538
commit d258e00710

View File

@ -3,6 +3,10 @@
#include <cstdio> // std::snprintf #include <cstdio> // std::snprintf
#include <iomanip> // std::put_time #include <iomanip> // std::put_time
#include <sstream> // std::ostringstream #include <sstream> // std::ostringstream
#include <array>
#include <cstring> // std::memcpy, std::strncat
#include <algorithm> // std::min
#include <initializer_list>
#include "libimp/codecvt.h" #include "libimp/codecvt.h"
@ -20,17 +24,46 @@ span<char const> normalize(span<char const> a) {
return a.first(a.size() - (a.back() == '\0' ? 1 : 0)); return a.first(a.size() - (a.back() == '\0' ? 1 : 0));
} }
std::string as_string(span<char const> a) { span<char> smem_cpy(span<char> sbuf, span<char const> a) noexcept {
if (a.empty()) return {}; if (sbuf.empty()) return {};
a = normalize(a); a = normalize(a);
return std::string(a.data(), a.size()); auto sz = (std::min)(sbuf.size() - 1, a.size());
if (sz != 0) std::memcpy(sbuf.data(), a.data(), sz);
return sbuf.first(sz);
} }
std::string fmt_of(span<char const> fstr, span<char const> s) { span<char> sbuf_cpy(span<char> sbuf, span<char const> a) noexcept {
return '%' + as_string(fstr) + as_string(s); sbuf = smem_cpy(sbuf, a);
*sbuf.end() = '\0';
return sbuf;
} }
std::string fmt_of_unsigned(span<char const> fstr, span<char const> l) { span<char> sbuf_cat(span<char> sbuf, std::initializer_list<span<char const>> args) noexcept {
std::size_t remain = sbuf.size();
for (auto s : args) {
remain -= smem_cpy(sbuf.last(remain), s).size();
}
auto sz = sbuf.size() - remain;
sbuf[sz] = '\0';
return sbuf.first(sz);
}
span<char> local_fmt_str() noexcept {
thread_local std::array<char, 512> sbuf;
return sbuf;
}
char const *as_cstr(span<char const> a) {
if (a.empty()) return "";
if (a.back() == '\0') return a.data();
return sbuf_cpy(local_fmt_str(), a).data();
}
span<char> fmt_of(span<char const> fstr, span<char const> s) {
return sbuf_cat(local_fmt_str(), {"%", fstr, s});
}
span<char> fmt_of_unsigned(span<char const> fstr, span<char const> l) {
if (fstr.empty()) { if (fstr.empty()) {
return fmt_of(l, "u"); return fmt_of(l, "u");
} }
@ -39,12 +72,12 @@ std::string fmt_of_unsigned(span<char const> fstr, span<char const> l) {
case 'o': case 'o':
case 'x': case 'x':
case 'X': case 'X':
case 'u': return fmt_of(fstr.first(fstr.size() - 1), l) + fstr.back(); case 'u': return sbuf_cat(local_fmt_str(), {"%", fstr.first(fstr.size() - 1), l, fstr.last(1)});
default : return fmt_of(fstr, l) + 'u'; default : return sbuf_cat(local_fmt_str(), {"%", fstr, l, "u"});
} }
} }
std::string fmt_of_signed(span<char const> fstr, span<char const> l) { span<char> fmt_of_signed(span<char const> fstr, span<char const> l) {
if (fstr.empty()) { if (fstr.empty()) {
return fmt_of(l, "d"); return fmt_of(l, "d");
} }
@ -54,11 +87,11 @@ std::string fmt_of_signed(span<char const> fstr, span<char const> l) {
case 'x': case 'x':
case 'X': case 'X':
case 'u': return fmt_of_unsigned(fstr, l); case 'u': return fmt_of_unsigned(fstr, l);
default : return fmt_of(fstr, l) + 'd'; default : return sbuf_cat(local_fmt_str(), {"%", fstr, l, "d"});
} }
} }
std::string fmt_of_float(span<char const> fstr, span<char const> l) { span<char> fmt_of_float(span<char const> fstr, span<char const> l) {
if (fstr.empty()) { if (fstr.empty()) {
return fmt_of(l, "f"); return fmt_of(l, "f");
} }
@ -67,22 +100,22 @@ std::string fmt_of_float(span<char const> fstr, span<char const> l) {
case 'e': case 'e':
case 'E': case 'E':
case 'g': case 'g':
case 'G': return fmt_of(fstr.first(fstr.size() - 1), l) + fstr.back(); case 'G': return sbuf_cat(local_fmt_str(), {"%", fstr.first(fstr.size() - 1), l, fstr.last(1)});
default : return fmt_of(fstr, l) + 'f'; default : return sbuf_cat(local_fmt_str(), {"%", fstr, l, "f"});
} }
} }
template <typename A /*a fundamental or pointer type*/> template <typename A /*a fundamental or pointer type*/>
std::string sprintf(std::string const &sfmt, A a) { std::string sprintf(span<char const> sfmt, A a) {
char sbuf[2048]; std::array<char, 512> sbuf;
auto sz = std::snprintf(sbuf, sizeof(sbuf), sfmt.c_str(), a); auto sz = std::snprintf(sbuf.data(), sbuf.size(), sfmt.data(), a);
if (sz <= 0) return {}; if (sz <= 0) return {};
if (sz < sizeof(sbuf)) { if (sz < sbuf.size()) {
return sbuf; return std::string(sbuf.data(), sz);
} }
std::string des; std::string des;
des.resize(sz + 1); des.resize(sz + 1);
if (std::snprintf(&des[0], des.size(), sfmt.c_str(), a) < 0) { if (std::snprintf(&des[0], des.size(), sfmt.data(), a) < 0) {
return {}; return {};
} }
des.pop_back(); // remove the terminated null character des.pop_back(); // remove the terminated null character
@ -190,7 +223,7 @@ std::string to_string(std::tm const &a, span<char const> fstr) noexcept {
} }
LIBIMP_TRY { LIBIMP_TRY {
std::ostringstream ss; std::ostringstream ss;
ss << std::put_time(&a, as_string(fstr).c_str()); ss << std::put_time(&a, as_cstr(fstr));
return ss.str(); return ss.str();
} LIBIMP_CATCH(...) { } LIBIMP_CATCH(...) {
return {}; return {};