summaryrefslogtreecommitdiff
path: root/lib/std/fmtfuncs.myr
diff options
context:
space:
mode:
Diffstat (limited to 'lib/std/fmtfuncs.myr')
-rw-r--r--lib/std/fmtfuncs.myr54
1 files changed, 54 insertions, 0 deletions
diff --git a/lib/std/fmtfuncs.myr b/lib/std/fmtfuncs.myr
new file mode 100644
index 0000000..abe5e75
--- /dev/null
+++ b/lib/std/fmtfuncs.myr
@@ -0,0 +1,54 @@
+use "alloc.use"
+use "bigint.use"
+use "bitset.use"
+use "fmt.use"
+use "introspect.use"
+use "strbuf.use"
+use "varargs.use"
+
+pkg std =
+;;
+
+const __init__ = {
+ var bigint : bigint#
+ var bitset : bitset#
+
+ fmtinstall(typeof(bigint), bigfmt, [][:])
+ fmtinstall(typeof(bitset), bsfmt, [][:])
+}
+
+const bigfmt = {sb, ap, opts
+ var a, n, buf
+
+ a = vanext(ap)
+ /*
+ allocate a buffer guaranteed to be big enough.
+ that's
+ 2 + floor(nbits/(log_2(10)))
+ or
+ 2 + a.dig.len * 32/3.32...
+ or
+ 2 + a.dig.len * 10
+ plus one for the - sign.
+ */
+ buf = slalloc(3 + a.dig.len * 10)
+ n = bigbfmt(buf, a, 10)
+ sbputs(sb, buf[:n])
+ slfree(buf)
+}
+
+const bsfmt = {sb, ap, opts
+ var i, bs, sep
+
+ bs = vanext(ap)
+ sep = ""
+ sbputs(sb, "bitset{")
+ for i = 0; i < bsmax(bs); i++
+ if bshas(bs, i)
+ sbfmt(sb, sep)
+ sbfmt(sb, "{}", i)
+ sep = ", "
+ ;;
+ ;;
+ sbputs(sb, "}")
+}