2015-07-01 01:22:42 +00:00
|
|
|
/* Format strings for OctaSTD.
|
|
|
|
*
|
|
|
|
* This file is part of OctaSTD. See COPYING.md for futher information.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef OCTA_FORMAT_HH
|
|
|
|
#define OCTA_FORMAT_HH
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <ctype.h>
|
|
|
|
#include <assert.h>
|
|
|
|
|
|
|
|
#include "octa/algorithm.hh"
|
|
|
|
#include "octa/string.hh"
|
|
|
|
|
|
|
|
namespace octa {
|
|
|
|
|
2015-07-02 18:02:59 +00:00
|
|
|
enum FormatFlags {
|
|
|
|
FMT_FLAG_DASH = 1 << 0,
|
|
|
|
FMT_FLAG_ZERO = 1 << 1,
|
|
|
|
FMT_FLAG_SPACE = 1 << 2,
|
|
|
|
FMT_FLAG_PLUS = 1 << 3,
|
|
|
|
FMT_FLAG_HASH = 1 << 4
|
|
|
|
};
|
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
namespace detail {
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline int parse_fmt_flags(const char *&fmt, int ret) {
|
|
|
|
while (*fmt) {
|
|
|
|
switch (*fmt) {
|
|
|
|
case '-': ret |= FMT_FLAG_DASH; ++fmt; break;
|
|
|
|
case '+': ret |= FMT_FLAG_PLUS; ++fmt; break;
|
|
|
|
case '#': ret |= FMT_FLAG_HASH; ++fmt; break;
|
|
|
|
case '0': ret |= FMT_FLAG_ZERO; ++fmt; break;
|
|
|
|
case ' ': ret |= FMT_FLAG_SPACE; ++fmt; break;
|
|
|
|
default: goto retflags;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
retflags:
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline octa::Size read_digits(const char *&fmt, char *buf) {
|
|
|
|
octa::Size ret = 0;
|
|
|
|
for (; isdigit(*fmt); ++ret)
|
|
|
|
*buf++ = *fmt++;
|
|
|
|
*buf = '\0';
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
struct FormatSpec {
|
|
|
|
FormatSpec(): p_fmt(nullptr) {}
|
|
|
|
FormatSpec(const char *fmt): p_fmt(fmt) {}
|
2015-07-02 01:00:05 +00:00
|
|
|
|
2015-07-02 18:02:59 +00:00
|
|
|
int width = 0;
|
|
|
|
int precision = 0;
|
|
|
|
|
|
|
|
bool has_width = false;
|
|
|
|
bool has_precision = false;
|
|
|
|
|
|
|
|
bool arg_width = false;
|
|
|
|
bool arg_precision = false;
|
|
|
|
|
|
|
|
int flags = 0;
|
|
|
|
|
|
|
|
char spec = '\0';
|
|
|
|
|
|
|
|
octa::byte index = 0;
|
|
|
|
|
|
|
|
template<typename R>
|
|
|
|
bool read_until_spec(R &writer, octa::Size *wret) {
|
|
|
|
octa::Size written = 0;
|
|
|
|
if (!p_fmt) return false;
|
|
|
|
while (*p_fmt) {
|
|
|
|
if (*p_fmt == '%') {
|
|
|
|
++p_fmt;
|
|
|
|
if (*p_fmt == '%') goto plain;
|
|
|
|
bool r = read_spec();
|
|
|
|
if (wret) *wret = written;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
plain:
|
|
|
|
++written;
|
|
|
|
writer.put(*p_fmt++);
|
|
|
|
}
|
|
|
|
if (wret) *wret = 0;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename R>
|
|
|
|
void write_ws(R &writer, octa::Size n, bool left) const {
|
|
|
|
if (left == bool(flags & FMT_FLAG_DASH)) return;
|
|
|
|
for (int w = width - int(n); --w >= 0; writer.put(' '));
|
|
|
|
}
|
|
|
|
|
|
|
|
const char *rest() const {
|
|
|
|
return p_fmt;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
bool read_spec() {
|
|
|
|
octa::Size ndig = octa::detail::read_digits(p_fmt, p_buf);
|
|
|
|
|
|
|
|
bool havepos = false;
|
|
|
|
index = 0;
|
|
|
|
/* parse index */
|
|
|
|
if (*p_fmt == '$') {
|
|
|
|
if (ndig <= 0) return false; /* no pos given */
|
|
|
|
int idx = atoi(p_buf);
|
|
|
|
if (idx <= 0 || idx > 255) return false; /* bad index */
|
|
|
|
index = octa::byte(idx);
|
|
|
|
++p_fmt;
|
|
|
|
havepos = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* parse flags */
|
|
|
|
flags = 0;
|
|
|
|
octa::Size skipd = 0;
|
|
|
|
if (havepos || !ndig) {
|
|
|
|
flags = octa::detail::parse_fmt_flags(p_fmt, 0);
|
|
|
|
} else {
|
|
|
|
for (octa::Size i = 0; i < ndig; ++i) {
|
|
|
|
if (p_buf[i] != '0') break;
|
|
|
|
++skipd;
|
|
|
|
}
|
|
|
|
if (skipd) flags = FMT_FLAG_ZERO;
|
|
|
|
if (skipd == ndig)
|
|
|
|
flags = octa::detail::parse_fmt_flags(p_fmt, flags);
|
2015-07-02 01:00:05 +00:00
|
|
|
}
|
|
|
|
|
2015-07-02 18:02:59 +00:00
|
|
|
/* parse width */
|
|
|
|
width = 0;
|
|
|
|
has_width = false;
|
|
|
|
arg_width = false;
|
|
|
|
if (!havepos && ndig && (ndig - skipd)) {
|
|
|
|
width = atoi(p_buf + skipd);
|
|
|
|
has_width = true;
|
|
|
|
} else if (octa::detail::read_digits(p_fmt, p_buf)) {
|
|
|
|
width = atoi(p_buf);
|
|
|
|
has_width = true;
|
|
|
|
} else if (*p_fmt == '*') {
|
|
|
|
arg_width = true;
|
|
|
|
++p_fmt;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* parse precision */
|
|
|
|
precision = 0;
|
|
|
|
has_precision = false;
|
|
|
|
arg_precision = false;
|
|
|
|
if (*p_fmt != '.') goto fmtchar;
|
|
|
|
++p_fmt;
|
|
|
|
|
|
|
|
if (octa::detail::read_digits(p_fmt, p_buf)) {
|
|
|
|
precision = atoi(p_buf);
|
|
|
|
has_precision = true;
|
|
|
|
} else if (*p_fmt == '*') {
|
|
|
|
arg_precision = true;
|
|
|
|
++p_fmt;
|
|
|
|
} else return false;
|
|
|
|
|
|
|
|
fmtchar:
|
|
|
|
spec = *p_fmt++;
|
|
|
|
if (spec != 's') return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
const char *p_fmt;
|
|
|
|
char p_buf[32];
|
|
|
|
};
|
|
|
|
|
|
|
|
namespace detail {
|
2015-07-01 17:51:39 +00:00
|
|
|
/* C string */
|
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
template<typename R>
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write_val(R &writer, const FormatSpec &fl,
|
2015-07-02 01:00:05 +00:00
|
|
|
const char *val) {
|
|
|
|
octa::Size n = strlen(val);
|
2015-07-02 18:02:59 +00:00
|
|
|
fl.write_ws(writer, n, true);
|
|
|
|
writer.put_n(val, n);
|
|
|
|
fl.write_ws(writer, n, false);
|
|
|
|
return n;
|
2015-07-01 01:29:42 +00:00
|
|
|
}
|
2015-07-01 01:22:42 +00:00
|
|
|
|
2015-07-01 17:51:39 +00:00
|
|
|
/* OctaSTD string */
|
|
|
|
|
2015-07-01 19:09:02 +00:00
|
|
|
template<typename R, typename A>
|
|
|
|
static inline octa::Size
|
2015-07-02 18:02:59 +00:00
|
|
|
write_val(R &writer, const FormatSpec &fl, const octa::AnyString<A> &val) {
|
2015-07-02 01:00:05 +00:00
|
|
|
octa::Size n = val.size();
|
2015-07-02 18:02:59 +00:00
|
|
|
fl.write_ws(writer, n, true);
|
|
|
|
writer.put_n(val.data(), n);
|
|
|
|
fl.write_ws(writer, n, false);
|
2015-07-02 01:00:05 +00:00
|
|
|
return n;
|
2015-07-01 01:29:42 +00:00
|
|
|
}
|
2015-07-01 01:22:42 +00:00
|
|
|
|
2015-07-01 17:51:39 +00:00
|
|
|
/* A character */
|
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
template<typename R>
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write_val(R &writer, const FormatSpec &fl,
|
2015-07-02 01:00:05 +00:00
|
|
|
char val) {
|
2015-07-02 18:02:59 +00:00
|
|
|
fl.write_ws(writer, 1, true);
|
2015-07-01 01:29:42 +00:00
|
|
|
writer.put(val);
|
2015-07-02 18:02:59 +00:00
|
|
|
fl.write_ws(writer, 1, false);
|
|
|
|
return 1;
|
2015-07-02 01:00:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* boolean */
|
|
|
|
|
|
|
|
template<typename R>
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write_val(R &writer, const FormatSpec &fl,
|
2015-07-02 01:00:05 +00:00
|
|
|
bool val) {
|
2015-07-02 18:02:59 +00:00
|
|
|
return write_val(writer, fl, ("false\0true") + (6 * val));
|
2015-07-01 17:51:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Integer helpers */
|
|
|
|
|
|
|
|
template<typename R>
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write_u(R &writer, const FormatSpec &fl,
|
|
|
|
bool neg, octa::Uintmax val) {
|
2015-07-01 17:51:39 +00:00
|
|
|
char buf[20];
|
|
|
|
octa::Size n = 0;
|
|
|
|
for (; val; val /= 10) buf[n++] = '0' + val % 10;
|
2015-07-02 18:02:59 +00:00
|
|
|
bool lsgn = !!(fl.flags & FMT_FLAG_PLUS);
|
|
|
|
bool lsp = !!(fl.flags & FMT_FLAG_SPACE);
|
|
|
|
bool sign = !!(neg + lsgn + lsp);
|
|
|
|
fl.write_ws(writer, n + sign, true);
|
|
|
|
if (sign) writer.put(neg ? '-' : *((" \0+") + lsgn * 2));
|
2015-07-01 17:51:39 +00:00
|
|
|
for (int i = int(n - 1); i >= 0; --i) {
|
|
|
|
writer.put(buf[i]);
|
|
|
|
}
|
2015-07-02 18:02:59 +00:00
|
|
|
fl.write_ws(writer, n + sign, false);
|
|
|
|
return n + sign;
|
2015-07-01 17:51:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template<typename R, typename T, bool = octa::IsIntegral<T>::value,
|
|
|
|
bool = octa::IsSigned<T>::value>
|
|
|
|
struct WriteIntVal {
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write(R &writer, const FormatSpec &fl,
|
2015-07-02 01:00:05 +00:00
|
|
|
const T &val) {
|
|
|
|
return octa::detail::write_val(writer, fl,
|
|
|
|
octa::to_string(val));
|
2015-07-01 17:51:39 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
template<typename R, typename T> struct WriteIntVal<R, T, true, true> {
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write(R &writer, const FormatSpec &fl, T val) {
|
|
|
|
return octa::detail::write_u(writer, fl, val < 0,
|
|
|
|
(val < 0) ? -val : val);
|
2015-07-01 17:51:39 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
template<typename R, typename T> struct WriteIntVal<R, T, true, false> {
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write(R &writer, const FormatSpec &fl, T val) {
|
|
|
|
return octa::detail::write_u(writer, fl, false, val);
|
2015-07-01 17:51:39 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Integers or generic value */
|
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
template<typename R, typename T>
|
2015-07-02 18:02:59 +00:00
|
|
|
static inline octa::Size write_val(R &writer, const FormatSpec &fl,
|
2015-07-02 01:00:05 +00:00
|
|
|
const T &val) {
|
|
|
|
return octa::detail::WriteIntVal<R, T>::write(writer, fl, val);
|
2015-07-01 01:29:42 +00:00
|
|
|
}
|
2015-07-01 01:22:42 +00:00
|
|
|
|
2015-07-01 17:51:39 +00:00
|
|
|
/* Writer entry point */
|
|
|
|
|
2015-07-02 01:00:05 +00:00
|
|
|
template<typename R, typename T>
|
|
|
|
static inline octa::Size write_idx(R &writer, octa::Size idx,
|
2015-07-02 18:02:59 +00:00
|
|
|
const FormatSpec &fl, const T &v) {
|
|
|
|
if (idx) {
|
|
|
|
assert(false && "not enough format args");
|
|
|
|
return -1;
|
|
|
|
}
|
2015-07-02 01:00:05 +00:00
|
|
|
return octa::detail::write_val(writer, fl, v);
|
2015-07-01 01:22:42 +00:00
|
|
|
}
|
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
template<typename R, typename T, typename ...A>
|
2015-07-02 01:00:05 +00:00
|
|
|
static inline octa::Size write_idx(R &writer, octa::Size idx,
|
2015-07-02 18:02:59 +00:00
|
|
|
const FormatSpec &fl, const T &v,
|
2015-07-01 17:51:39 +00:00
|
|
|
const A &...args) {
|
2015-07-02 01:00:05 +00:00
|
|
|
if (idx) return octa::detail::write_idx(writer, idx - 1, fl,
|
|
|
|
args...);
|
|
|
|
return octa::detail::write_val(writer, fl, v);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
static inline int conv_idx_num(const T &v, octa::EnableIf<
|
|
|
|
octa::IsIntegral<T>::value, bool
|
|
|
|
> = true) {
|
|
|
|
return int(v);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
static inline int conv_idx_num(const T &, octa::EnableIf<
|
|
|
|
!octa::IsIntegral<T>::value, bool
|
|
|
|
> = true) {
|
|
|
|
assert(false && "argument is not an integer");
|
2015-07-02 18:02:59 +00:00
|
|
|
return -2;
|
2015-07-02 01:00:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
static inline int read_idx_num(octa::Size idx, const T &v) {
|
2015-07-02 18:02:59 +00:00
|
|
|
if (idx) {
|
|
|
|
assert(false && "not enough format args");
|
|
|
|
return -2;
|
|
|
|
}
|
2015-07-02 01:00:05 +00:00
|
|
|
return conv_idx_num(v);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename T, typename ...A>
|
|
|
|
static inline int read_idx_num(octa::Size idx, const T &v, const A &...args) {
|
|
|
|
if (idx) return octa::detail::read_idx_num(idx - 1, args...);
|
|
|
|
return conv_idx_num(v);
|
2015-07-01 01:29:42 +00:00
|
|
|
}
|
2015-07-02 01:00:05 +00:00
|
|
|
|
2015-07-01 01:29:42 +00:00
|
|
|
} /* namespace detail */
|
2015-07-01 01:22:42 +00:00
|
|
|
|
|
|
|
template<typename R, typename ...A>
|
2015-07-01 17:51:39 +00:00
|
|
|
static inline octa::Size formatted_write(R writer, octa::Size &fmtn,
|
|
|
|
const char *fmt, const A &...args) {
|
2015-07-02 18:02:59 +00:00
|
|
|
octa::Size argidx = 1, written = 0, retn = 0, twr = 0;
|
|
|
|
FormatSpec spec(fmt);
|
|
|
|
while (spec.read_until_spec(writer, &twr)) {
|
|
|
|
written += twr;
|
|
|
|
octa::Size argpos = spec.index;
|
|
|
|
if (!argpos) argpos = argidx++;
|
|
|
|
octa::Size wl = octa::detail::write_idx(writer, argpos - 1, spec, args...);
|
|
|
|
written += octa::max(spec.width, int(wl));
|
2015-07-01 01:22:42 +00:00
|
|
|
}
|
2015-07-02 18:02:59 +00:00
|
|
|
fmtn = retn;
|
2015-07-01 17:51:39 +00:00
|
|
|
return written;
|
|
|
|
}
|
|
|
|
|
2015-07-01 19:09:02 +00:00
|
|
|
template<typename R, typename AL, typename ...A>
|
|
|
|
octa::Size formatted_write(R writer, octa::Size &fmtn,
|
|
|
|
const octa::AnyString<AL> &fmt,
|
2015-07-01 17:51:39 +00:00
|
|
|
const A &...args) {
|
|
|
|
return formatted_write(writer, fmtn, fmt.data(), args...);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename R, typename ...A>
|
|
|
|
octa::Size formatted_write(R writer, const char *fmt, const A &...args) {
|
|
|
|
octa::Size fmtn = 0;
|
|
|
|
return formatted_write(writer, fmtn, fmt, args...);
|
2015-07-01 01:22:42 +00:00
|
|
|
}
|
|
|
|
|
2015-07-01 19:09:02 +00:00
|
|
|
template<typename R, typename AL, typename ...A>
|
|
|
|
octa::Size formatted_write(R writer, const octa::AnyString<AL> &fmt,
|
2015-07-01 17:51:39 +00:00
|
|
|
const A &...args) {
|
|
|
|
octa::Size fmtn = 0;
|
|
|
|
return formatted_write(writer, fmtn, fmt.data(), args...);
|
2015-07-01 01:22:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
} /* namespace octa */
|
|
|
|
|
|
|
|
#endif
|