1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183 |
- import (
- "os.odin";
- "mem.odin";
- "utf8.odin";
- "types.odin";
- "strconv.odin";
- "raw.odin";
- )
- _BUFFER_SIZE :: 1<<12;
- StringBuffer :: union {
- []u8,
- [dynamic]u8,
- }
- FmtInfo :: struct {
- minus: bool;
- plus: bool;
- space: bool;
- zero: bool;
- hash: bool;
- width_set: bool;
- prec_set: bool;
- width: int;
- prec: int;
- indent: int;
- reordered: bool;
- good_arg_index: bool;
- buf: ^StringBuffer;
- arg: any; // Temporary
- }
- string_buffer_data :: proc(buf: ^StringBuffer) -> []u8 {
- match b in buf {
- case []u8: return b[..];
- case [dynamic]u8: return b[..];
- }
- return nil;
- }
- string_buffer_data :: proc(buf: StringBuffer) -> []u8 {
- match b in buf {
- case []u8: return b[..];
- case [dynamic]u8: return b[..];
- }
- return nil;
- }
- to_string :: proc(buf: StringBuffer) -> string {
- return string(string_buffer_data(buf));
- }
- write_string :: proc(buf: ^StringBuffer, s: string) {
- write_bytes(buf, cast([]u8)s);
- }
- write_bytes :: proc(buf: ^StringBuffer, data: []u8) {
- match b in buf {
- case []u8: append(b, ...data);
- case [dynamic]u8: append(b, ...data);
- }
- }
- write_byte :: proc(buf: ^StringBuffer, data: u8) {
- match b in buf {
- case []u8: append(b, data);
- case [dynamic]u8: append(b, data);
- }
- }
- write_rune :: proc(buf: ^StringBuffer, r: rune) {
- if r < utf8.RUNE_SELF {
- write_byte(buf, u8(r));
- return;
- }
- b, n := utf8.encode_rune(r);
- write_bytes(buf, b[..n]);
- }
- write_int :: proc(buf: ^StringBuffer, i: i128, base: int) {
- b: [129]u8;
- s := strconv.append_bits(b[..0], u128(i), base, true, 128, strconv.digits, 0);
- write_string(buf, s);
- }
- write_int :: proc(buf: ^StringBuffer, i: i64, base: int) {
- b: [129]u8;
- s := strconv.append_bits(b[..0], u128(i), base, true, 64, strconv.digits, 0);
- write_string(buf, s);
- }
- fprint :: proc(fd: os.Handle, args: ...any) -> int {
- data: [_BUFFER_SIZE]u8;
- buf := StringBuffer(data[..0]);
- sbprint(&buf, ...args);
- res := string_buffer_data(buf);
- os.write(fd, res);
- return len(res);
- }
- fprintln :: proc(fd: os.Handle, args: ...any) -> int {
- data: [_BUFFER_SIZE]u8;
- buf := StringBuffer(data[..0]);
- sbprintln(&buf, ...args);
- res := string_buffer_data(buf);
- os.write(fd, res);
- return len(res);
- }
- fprintf :: proc(fd: os.Handle, fmt: string, args: ...any) -> int {
- data: [_BUFFER_SIZE]u8;
- buf := StringBuffer(data[..0]);
- sbprintf(&buf, fmt, ...args);
- res := string_buffer_data(buf);
- os.write(fd, res);
- return len(res);
- }
- // print* procedures return the number of bytes written
- print :: proc(args: ...any) -> int { return fprint(os.stdout, ...args); }
- print_err :: proc(args: ...any) -> int { return fprint(os.stderr, ...args); }
- println :: proc(args: ...any) -> int { return fprintln(os.stdout, ...args); }
- println_err :: proc(args: ...any) -> int { return fprintln(os.stderr, ...args); }
- printf :: proc(fmt: string, args: ...any) -> int { return fprintf(os.stdout, fmt, ...args); }
- printf_err :: proc(fmt: string, args: ...any) -> int { return fprintf(os.stderr, fmt, ...args); }
- // aprint* procedures return a string that was allocated with the current context
- // They must be freed accordingly
- aprint :: proc(args: ...any) -> string {
- buf := StringBuffer(make([dynamic]u8));
- sbprint(&buf, ...args);
- return to_string(buf);
- }
- aprintln :: proc(args: ...any) -> string {
- buf := StringBuffer(make([dynamic]u8));
- sbprintln(&buf, ...args);
- return to_string(buf);
- }
- aprintf :: proc(fmt: string, args: ...any) -> string {
- buf := StringBuffer(make([dynamic]u8));
- sbprintf(&buf, fmt, ...args);
- return to_string(buf);
- }
- // bprint* procedures return a string using a buffer from an array
- bprint :: proc(buf: []u8, args: ...any) -> string {
- sb := StringBuffer(buf[..0..len(buf)]);
- return sbprint(&sb, ...args);
- }
- bprintln :: proc(buf: []u8, args: ...any) -> string {
- sb := StringBuffer(buf[..0..len(buf)]);
- return sbprintln(&sb, ...args);
- }
- bprintf :: proc(buf: []u8, fmt: string, args: ...any) -> string {
- sb := StringBuffer(buf[..0..len(buf)]);
- return sbprintf(&sb, fmt, ...args);
- }
- fprint_type :: proc(fd: os.Handle, info: ^TypeInfo) {
- data: [_BUFFER_SIZE]u8;
- buf := StringBuffer(data[..0]);
- write_type(&buf, info);
- os.write(fd, string_buffer_data(buf));
- }
- write_type :: proc(buf: ^StringBuffer, ti: ^TypeInfo) {
- if ti == nil do return;
- using TypeInfo;
- match info in ti.variant {
- case Named:
- write_string(buf, info.name);
- case Integer:
- match {
- case ti == type_info_of(int): write_string(buf, "int");
- case ti == type_info_of(uint): write_string(buf, "uint");
- case:
- if info.signed do write_byte(buf, 'i');
- else do write_byte(buf, 'u');
- write_int(buf, i64(8*ti.size), 10);
- }
- case Rune:
- write_string(buf, "rune");
- case Float:
- match ti.size {
- case 2: write_string(buf, "f16");
- case 4: write_string(buf, "f32");
- case 8: write_string(buf, "f64");
- }
- case Complex:
- match ti.size {
- case 4: write_string(buf, "complex32");
- case 8: write_string(buf, "complex64");
- case 16: write_string(buf, "complex128");
- }
- case String: write_string(buf, "string");
- case Boolean: write_string(buf, "bool");
- case Any:
- write_string(buf, "any");
- case Pointer:
- if info.elem == nil {
- write_string(buf, "rawptr");
- } else {
- write_string(buf, "^");
- write_type(buf, info.elem);
- }
- case Procedure:
- write_string(buf, "proc");
- if info.params == nil {
- write_string(buf, "()");
- } else {
- t := info.params.variant.(Tuple);
- write_string(buf, "(");
- for t, i in t.types {
- if i > 0 do write_string(buf, ", ");
- write_type(buf, t);
- }
- write_string(buf, ")");
- }
- if info.results != nil {
- write_string(buf, " -> ");
- write_type(buf, info.results);
- }
- case Tuple:
- count := len(info.names);
- if count != 1 do write_string(buf, "(");
- for name, i in info.names {
- if i > 0 do write_string(buf, ", ");
- t := info.types[i];
- if len(name) > 0 {
- write_string(buf, name);
- write_string(buf, ": ");
- }
- write_type(buf, t);
- }
- if count != 1 do write_string(buf, ")");
- case Array:
- write_string(buf, "[");
- fi := FmtInfo{buf = buf};
- write_int(buf, i64(info.count), 10);
- write_string(buf, "]");
- write_type(buf, info.elem);
- case DynamicArray:
- write_string(buf, "[dynamic]");
- write_type(buf, info.elem);
- case Slice:
- write_string(buf, "[]");
- write_type(buf, info.elem);
- case Vector:
- write_string(buf, "[vector ");
- write_int(buf, i64(info.count), 10);
- write_string(buf, "]");
- write_type(buf, info.elem);
- case Map:
- write_string(buf, "map[");
- write_type(buf, info.key);
- write_byte(buf, ']');
- write_type(buf, info.value);
- case Struct:
- write_string(buf, "struct ");
- if info.is_packed do write_string(buf, "#packed ");
- if info.is_ordered do write_string(buf, "#ordered ");
- if info.is_raw_union do write_string(buf, "#raw_union ");
- if info.custom_align {
- write_string(buf, "#align ");
- write_int(buf, i64(ti.align), 10);
- write_byte(buf, ' ');
- }
- write_byte(buf, '{');
- for name, i in info.names {
- if i > 0 do write_string(buf, ", ");
- write_string(buf, name);
- write_string(buf, ": ");
- write_type(buf, info.types[i]);
- }
- write_byte(buf, '}');
- case Union:
- write_string(buf, "union {");
- for variant, i in info.variants {
- if i > 0 do write_string(buf, ", ");
- write_type(buf, variant);
- }
- write_string(buf, "}");
- case Enum:
- write_string(buf, "enum ");
- write_type(buf, info.base);
- write_string(buf, " {");
- for name, i in info.names {
- if i > 0 do write_string(buf, ", ");
- write_string(buf, name);
- }
- write_string(buf, "}");
- case BitField:
- write_string(buf, "bit_field ");
- if ti.align != 1 {
- write_string(buf, "#align ");
- write_int(buf, i64(ti.align), 10);
- write_rune(buf, ' ');
- }
- write_string(buf, " {");
- for name, i in info.names {
- if i > 0 do write_string(buf, ", ");
- write_string(buf, name);
- write_string(buf, ": ");
- write_int(buf, i64(info.bits[i]), 10);
- }
- write_string(buf, "}");
- }
- }
- _parse_int :: proc(s: string, offset: int) -> (result: int, offset: int, ok: bool) {
- is_digit :: proc(r: rune) -> bool #inline {
- return '0' <= r && r <= '9';
- }
- result := 0;
- ok := true;
- i := 0;
- for i < len(s[offset..]) {
- c := rune(s[offset+i]);
- if !is_digit(c) do break;
- i += 1;
- result *= 10;
- result += int(c)-'0';
- }
- return result, offset+i, i != 0;
- }
- _arg_number :: proc(fi: ^FmtInfo, arg_index: int, format: string, offset, arg_count: int) -> (index, offset: int, ok: bool) {
- parse_arg_number :: proc(format: string) -> (int, int, bool) {
- if len(format) < 3 do return 0, 1, false;
- for i in 1...len(format) {
- if format[i] == ']' {
- width, new_index, ok := _parse_int(format, 1);
- if !ok || new_index != i {
- return 0, i+1, false;
- }
- return width-1, i+1, true;
- }
- }
- return 0, 1, false;
- }
- if len(format) <= offset || format[offset] != '[' {
- return arg_index, offset, false;
- }
- fi.reordered = true;
- index, width, ok := parse_arg_number(format[offset..]);
- if ok && 0 <= index && index < arg_count {
- return index, offset+width, true;
- }
- fi.good_arg_index = false;
- return arg_index, offset+width, false;
- }
- int_from_arg :: proc(args: []any, arg_index: int) -> (int, int, bool) {
- num := 0;
- new_arg_index := arg_index;
- ok := true;
- if arg_index < len(args) {
- arg := args[arg_index];
- arg.type_info = type_info_base(arg.type_info);
- match i in arg {
- case int: num = i;
- case i8: num = int(i);
- case i16: num = int(i);
- case i32: num = int(i);
- case i64: num = int(i);
- case u8: num = int(i);
- case u16: num = int(i);
- case u32: num = int(i);
- case u64: num = int(i);
- case:
- ok = false;
- }
- }
- return num, new_arg_index, ok;
- }
- fmt_bad_verb :: proc(using fi: ^FmtInfo, verb: rune) {
- assert(verb != 'v');
- write_string(buf, "%!");
- write_rune(buf, verb);
- write_byte(buf, '(');
- if arg.type_info != nil {
- write_type(buf, arg.type_info);
- write_byte(buf, '=');
- fmt_value(fi, arg, 'v');
- } else {
- write_string(buf, "<nil>");
- }
- write_byte(buf, ')');
- }
- fmt_bool :: proc(using fi: ^FmtInfo, b: bool, verb: rune) {
- match verb {
- case 't', 'v':
- s := "false";
- if b do s = "true";
- write_string(buf, s);
- case:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_write_padding :: proc(fi: ^FmtInfo, width: int) {
- if width <= 0 do return;
- pad_byte: u8 = '0';
- if fi.space do pad_byte = ' ';
- for _ in 0..width {
- write_byte(fi.buf, pad_byte);
- }
- }
- _fmt_int :: proc(fi: ^FmtInfo, u: u128, base: int, is_signed: bool, bit_size: int, digits: string) {
- _, neg := strconv.is_integer_negative(u128(u), is_signed, bit_size);
- BUF_SIZE :: 256;
- if fi.width_set || fi.prec_set {
- width := fi.width + fi.prec + 3; // 3 extra bytes for sign and prefix
- if width > BUF_SIZE {
- // TODO(bill):????
- panic("_fmt_int: buffer overrun. Width and precision too big");
- }
- }
- prec := 0;
- if fi.prec_set {
- prec = fi.prec;
- if prec == 0 && u == 0 {
- prev_zero := fi.zero;
- fi.zero = false;
- fmt_write_padding(fi, fi.width);
- fi.zero = prev_zero;
- return;
- }
- } else if fi.zero && fi.width_set {
- prec = fi.width;
- if neg || fi.plus || fi.space {
- // There needs to be space for the "sign"
- prec -= 1;
- }
- }
- match base {
- case 2, 8, 10, 12, 16:
- break;
- case:
- panic("_fmt_int: unknown base, whoops");
- }
- buf: [256]u8;
- start := 0;
- flags: strconv.IntFlag;
- if fi.hash && !fi.zero do flags |= strconv.IntFlag.Prefix;
- if fi.plus do flags |= strconv.IntFlag.Plus;
- if fi.space do flags |= strconv.IntFlag.Space;
- s := strconv.append_bits(buf[start..start], u128(u), base, is_signed, bit_size, digits, flags);
- if fi.hash && fi.zero {
- c: u8;
- match base {
- case 2: c = 'b';
- case 8: c = 'o';
- case 10: c = 'd';
- case 12: c = 'z';
- case 16: c = 'x';
- }
- if c != 0 {
- write_byte(fi.buf, '0');
- write_byte(fi.buf, c);
- }
- }
- prev_zero := fi.zero;
- defer fi.zero = prev_zero;
- fi.zero = false;
- _pad(fi, s);
- }
- __DIGITS_LOWER := "0123456789abcdefx";
- __DIGITS_UPPER := "0123456789ABCDEFX";
- fmt_rune :: proc(fi: ^FmtInfo, r: rune, verb: rune) {
- match verb {
- case 'c', 'r', 'v':
- write_rune(fi.buf, r);
- case:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_int :: proc(fi: ^FmtInfo, u: u128, is_signed: bool, bit_size: int, verb: rune) {
- match verb {
- case 'v': _fmt_int(fi, u, 10, is_signed, bit_size, __DIGITS_LOWER);
- case 'b': _fmt_int(fi, u, 2, is_signed, bit_size, __DIGITS_LOWER);
- case 'o': _fmt_int(fi, u, 8, is_signed, bit_size, __DIGITS_LOWER);
- case 'd': _fmt_int(fi, u, 10, is_signed, bit_size, __DIGITS_LOWER);
- case 'z': _fmt_int(fi, u, 12, is_signed, bit_size, __DIGITS_LOWER);
- case 'x': _fmt_int(fi, u, 16, is_signed, bit_size, __DIGITS_LOWER);
- case 'X': _fmt_int(fi, u, 16, is_signed, bit_size, __DIGITS_UPPER);
- case 'c', 'r':
- fmt_rune(fi, rune(u), verb);
- case 'U':
- r := rune(u);
- if r < 0 || r > utf8.MAX_RUNE {
- fmt_bad_verb(fi, verb);
- } else {
- write_string(fi.buf, "U+");
- _fmt_int(fi, u, 16, false, bit_size, __DIGITS_UPPER);
- }
- case:
- fmt_bad_verb(fi, verb);
- }
- }
- _pad :: proc(fi: ^FmtInfo, s: string) {
- if !fi.width_set {
- write_string(fi.buf, s);
- return;
- }
- width := fi.width - utf8.rune_count(s);
- if fi.minus { // right pad
- write_string(fi.buf, s);
- fmt_write_padding(fi, width);
- } else { // left pad
- fmt_write_padding(fi, width);
- write_string(fi.buf, s);
- }
- }
- fmt_float :: proc(fi: ^FmtInfo, v: f64, bit_size: int, verb: rune) {
- match verb {
- // case 'e', 'E', 'f', 'F', 'g', 'G', 'v':
- // case 'f', 'F', 'v':
- case 'f', 'F', 'v':
- prec: int = 3;
- if fi.prec_set do prec = fi.prec;
- buf: [386]u8;
- str := strconv.append_float(buf[1..1], v, 'f', prec, bit_size);
- str = string(buf[...len(str)]);
- if str[1] == '+' || str[1] == '-' {
- str = str[1..];
- } else {
- str[0] = '+';
- }
- if fi.space && !fi.plus && str[0] == '+' {
- str[0] = ' ';
- }
- if len(str) > 1 && str[1] == 'N' && str[1] == 'I' {
- write_string(fi.buf, str);
- return;
- }
- if fi.plus || str[0] != '+' {
- if fi.zero && fi.width_set && fi.width > len(str) {
- write_byte(fi.buf, str[0]);
- fmt_write_padding(fi, fi.width - len(str));
- write_string(fi.buf, str[1..]);
- } else {
- _pad(fi, str);
- }
- } else {
- _pad(fi, str[1..]);
- }
- case:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_string :: proc(fi: ^FmtInfo, s: string, verb: rune) {
- match verb {
- case 's', 'v':
- write_string(fi.buf, s);
- case 'x', 'X':
- space := fi.space;
- fi.space = false;
- defer fi.space = space;
- for i in 0..len(s) {
- if i > 0 && space do write_byte(fi.buf, ' ');
- char_set := __DIGITS_UPPER;
- if verb == 'x' do char_set = __DIGITS_LOWER;
- _fmt_int(fi, u128(s[i]), 16, false, 8, char_set);
- }
- case:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_pointer :: proc(fi: ^FmtInfo, p: rawptr, verb: rune) {
- match verb {
- case 'p', 'v':
- // Okay
- case:
- fmt_bad_verb(fi, verb);
- return;
- }
- u := u128(uint(p));
- if !fi.hash || verb == 'v' {
- write_string(fi.buf, "0x");
- }
- _fmt_int(fi, u, 16, false, 8*size_of(rawptr), __DIGITS_UPPER);
- }
- enum_value_to_string :: proc(v: any) -> (string, bool) {
- v.type_info = type_info_base(v.type_info);
- using TypeInfo;
- match e in v.type_info.variant {
- case: return "", false;
- case Enum:
- get_str :: proc(i: $T, e: Enum) -> (string, bool) {
- if types.is_string(e.base) {
- for val, idx in e.values {
- if v, ok := val.(T); ok && v == i {
- return e.names[idx], true;
- }
- }
- } else if len(e.values) == 0 {
- return "", true;
- } else {
- for val, idx in e.values {
- if v, ok := val.(T); ok && v == i {
- return e.names[idx], true;
- }
- }
- }
- return "", false;
- }
- a := any{v.data, type_info_base(e.base)};
- match v in a {
- case rune: return get_str(v, e);
- case i8: return get_str(v, e);
- case i16: return get_str(v, e);
- case i32: return get_str(v, e);
- case i64: return get_str(v, e);
- case i128: return get_str(v, e);
- case int: return get_str(v, e);
- case u8: return get_str(v, e);
- case u16: return get_str(v, e);
- case u32: return get_str(v, e);
- case u64: return get_str(v, e);
- case u128: return get_str(v, e);
- case uint: return get_str(v, e);
- case f32: return get_str(v, e);
- case f64: return get_str(v, e);
- }
- }
- return "", false;
- }
- string_to_enum_value :: proc(T: type, s: string) -> (T, bool) {
- ti := type_info_base(type_info_of(T));
- if e, ok := ti.variant.(TypeInfo.Enum); ok {
- for str, idx in e.names {
- if s == str {
- // NOTE(bill): Unsafe cast
- ptr := cast(^T)&e.values[idx];
- return ptr^, true;
- }
- }
- }
- return T{}, false;
- }
- fmt_enum :: proc(fi: ^FmtInfo, v: any, verb: rune) {
- if v.type_info == nil || v.data == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- using TypeInfo;
- match e in v.type_info.variant {
- case: fmt_bad_verb(fi, verb);
- case Enum:
- match verb {
- case: fmt_bad_verb(fi, verb);
- case 'd', 'f':
- fmt_arg(fi, any{v.data, type_info_base(e.base)}, verb);
- case 's', 'v':
- str, ok := enum_value_to_string(v);
- if !ok do str = "!%(BAD ENUM VALUE)";
- write_string(fi.buf, str);
- }
- }
- }
- fmt_value :: proc(fi: ^FmtInfo, v: any, verb: rune) {
- if v.data == nil || v.type_info == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- using TypeInfo;
- match info in v.type_info.variant {
- case Named:
- match b in info.base.variant {
- case Struct:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- if b.is_raw_union {
- write_string(fi.buf, info.name);
- write_string(fi.buf, "{}");
- return;
- }
- write_string(fi.buf, info.name);
- write_byte(fi.buf, '{');
- defer write_byte(fi.buf, '}');
- hash := fi.hash; defer fi.hash = hash;
- indent := fi.indent; defer fi.indent -= 1;
- fi.hash = false;
- fi.indent += 1;
- if hash do write_byte(fi.buf, '\n');
- for _, i in b.names {
- if !hash && i > 0 do write_string(fi.buf, ", ");
- if hash {
- for in 0..fi.indent {
- write_byte(fi.buf, '\t');
- }
- }
- write_string(fi.buf, b.names[i]);
- write_string(fi.buf, " = ");
- if t := b.types[i]; types.is_any(t) {
- write_string(fi.buf, "any{}");
- } else {
- data := cast(^u8)v.data + b.offsets[i];
- fmt_arg(fi, any{rawptr(data), t}, 'v');
- }
- if hash do write_string(fi.buf, ",\n");
- }
- case:
- fmt_value(fi, any{v.data, info.base}, verb);
- }
- case Boolean: fmt_arg(fi, v, verb);
- case Integer: fmt_arg(fi, v, verb);
- case Rune: fmt_arg(fi, v, verb);
- case Float: fmt_arg(fi, v, verb);
- case Complex: fmt_arg(fi, v, verb);
- case String: fmt_arg(fi, v, verb);
- case Pointer:
- if v.type_info == type_info_of(^TypeInfo) {
- write_type(fi.buf, (cast(^^TypeInfo)v.data)^);
- } else {
- fmt_pointer(fi, (cast(^rawptr)v.data)^, verb);
- }
- case Array:
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- for i in 0..info.count {
- if i > 0 do write_string(fi.buf, ", ");
- data := cast(^u8)v.data + i*info.elem_size;
- fmt_arg(fi, any{rawptr(data), info.elem}, verb);
- }
- case DynamicArray:
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- array := cast(^raw.DynamicArray)v.data;
- for i in 0..array.len {
- if i > 0 do write_string(fi.buf, ", ");
- data := cast(^u8)array.data + i*info.elem_size;
- fmt_arg(fi, any{rawptr(data), info.elem}, verb);
- }
- case Slice:
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- slice := cast(^[]u8)v.data;
- for _, i in slice {
- if i > 0 do write_string(fi.buf, ", ");
- data := &slice[0] + i*info.elem_size;
- fmt_arg(fi, any{rawptr(data), info.elem}, verb);
- }
- case Vector:
- write_byte(fi.buf, '<');
- defer write_byte(fi.buf, '>');
- for i in 0..info.count {
- if i > 0 do write_string(fi.buf, ", ");
- data := cast(^u8)v.data + i*info.elem_size;
- fmt_value(fi, any{rawptr(data), info.elem}, verb);
- }
- case Map:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_string(fi.buf, "map[");
- defer write_byte(fi.buf, ']');
- entries := &((cast(^raw.DynamicMap)v.data).entries);
- gs := type_info_base(info.generated_struct).variant.(Struct);
- ed := type_info_base(gs.types[1]).variant.(DynamicArray);
- entry_type := ed.elem.variant.(Struct);
- entry_size := ed.elem_size;
- for i in 0..entries.len {
- if i > 0 do write_string(fi.buf, ", ");
- data := cast(^u8)entries.data + i*entry_size;
- header := cast(^__MapEntryHeader)data;
- if types.is_string(info.key) {
- write_string(fi.buf, header.key.str);
- } else {
- fi := FmtInfo{buf = fi.buf};
- fmt_arg(&fi, any{rawptr(&header.key.hash), info.key}, 'v');
- }
- write_string(fi.buf, "=");
- value := data + entry_type.offsets[2];
- fmt_arg(fi, any{rawptr(value), info.value}, 'v');
- }
- case Struct:
- if info.is_raw_union {
- write_string(fi.buf, "(raw_union)");
- return;
- }
- write_byte(fi.buf, '{');
- defer write_byte(fi.buf, '}');
- hash := fi.hash; defer fi.hash = hash;
- indent := fi.indent; defer fi.indent -= 1;
- fi.hash = false;
- fi.indent += 1;
- if hash do write_byte(fi.buf, '\n');
- for _, i in info.names {
- if !hash && i > 0 do write_string(fi.buf, ", ");
- if hash {
- for in 0..fi.indent {
- write_byte(fi.buf, '\t');
- }
- }
- write_string(fi.buf, info.names[i]);
- write_string(fi.buf, " = ");
- if t := info.types[i]; types.is_any(t) {
- write_string(fi.buf, "any{}");
- } else {
- data := cast(^u8)v.data + info.offsets[i];
- fmt_arg(fi, any{rawptr(data), t}, 'v');
- }
- if hash do write_string(fi.buf, ",\n");
- }
- case Union:
- data := cast(^u8)v.data;
- tipp := cast(^^TypeInfo)(data + info.tag_offset);
- if data == nil || tipp == nil {
- write_string(fi.buf, "(union)");
- } else {
- ti := tipp^;
- fmt_arg(fi, any{data, ti}, verb);
- }
- case Enum:
- fmt_enum(fi, v, verb);
- case Procedure:
- write_type(fi.buf, v.type_info);
- write_string(fi.buf, " @ ");
- fmt_pointer(fi, (cast(^rawptr)v.data)^, 'p');
- }
- }
- fmt_complex :: proc(fi: ^FmtInfo, c: complex128, bits: int, verb: rune) {
- match verb {
- case 'f', 'F', 'v':
- r, i := real(c), imag(c);
- fmt_float(fi, r, bits/2, verb);
- if !fi.plus && i >= 0 {
- write_rune(fi.buf, '+');
- }
- fmt_float(fi, i, bits/2, verb);
- write_rune(fi.buf, 'i');
- case:
- fmt_bad_verb(fi, verb);
- return;
- }
- }
- _u128_to_lo_hi :: proc(a: u128) -> (lo, hi: u64) { return u64(a), u64(a>>64); }
- _i128_to_lo_hi :: proc(a: u128) -> (lo: u64 hi: i64) { return u64(a), i64(a>>64); }
- do_foo :: proc(fi: ^FmtInfo, f: f64) {
- fmt_string(fi, "Hellope$%!", 'v');
- }
- fmt_arg :: proc(fi: ^FmtInfo, arg: any, verb: rune) {
- if arg == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- fi.arg = arg;
- if verb == 'T' {
- ti := arg.type_info;
- match a in arg {
- case ^TypeInfo: ti = a;
- }
- write_type(fi.buf, ti);
- return;
- }
- base_arg := arg;
- base_arg.type_info = type_info_base(base_arg.type_info);
- match a in base_arg {
- case any: fmt_arg(fi, a, verb);
- case bool: fmt_bool(fi, a, verb);
- case rune: fmt_rune(fi, a, verb);
- case f32: fmt_float(fi, f64(a), 32, verb);
- case f64: fmt_float(fi, a, 64, verb);
- case complex64: fmt_complex(fi, complex128(a), 64, verb);
- case complex128: fmt_complex(fi, a, 128, verb);
- case int: fmt_int(fi, u128(a), true, 8*size_of(int), verb);
- case i8: fmt_int(fi, u128(a), true, 8, verb);
- case i16: fmt_int(fi, u128(a), true, 16, verb);
- case i32: fmt_int(fi, u128(a), true, 32, verb);
- case i64: fmt_int(fi, u128(a), true, 64, verb);
- case i128: fmt_int(fi, u128(a), true, 128, verb);
- case uint: fmt_int(fi, u128(a), false, 8*size_of(uint), verb);
- case u8: fmt_int(fi, u128(a), false, 8, verb);
- case u16: fmt_int(fi, u128(a), false, 16, verb);
- case u32: fmt_int(fi, u128(a), false, 32, verb);
- case u64: fmt_int(fi, u128(a), false, 64, verb);
- case u128: fmt_int(fi, u128(a), false, 128, verb);
- case string: fmt_string(fi, a, verb);
- case: fmt_value(fi, arg, verb);
- }
- }
- sbprint :: proc(buf: ^StringBuffer, args: ...any) -> string {
- fi: FmtInfo;
- prev_string := false;
- fi.buf = buf;
- for arg, i in args {
- is_string := arg != nil && types.is_string(arg.type_info);
- if i > 0 && !is_string && !prev_string {
- write_byte(buf, ' ');
- }
- fmt_value(&fi, args[i], 'v');
- prev_string = is_string;
- }
- return to_string(buf^);
- }
- sbprintln :: proc(buf: ^StringBuffer, args: ...any) -> string {
- fi: FmtInfo;
- fi.buf = buf;
- for arg, i in args {
- if i > 0 do write_byte(buf, ' ');
- fmt_value(&fi, args[i], 'v');
- }
- write_byte(buf, '\n');
- return to_string(buf^);
- }
- sbprintf :: proc(b: ^StringBuffer, fmt: string, args: ...any) -> string {
- fi: FmtInfo;
- arg_index: int = 0;
- end := len(fmt);
- was_prev_index := false;
- for i := 0; i < end; /**/ {
- fi = FmtInfo{buf = b, good_arg_index = true};
- prev_i := i;
- for i < end && fmt[i] != '%' {
- i += 1;
- }
- if i > prev_i {
- write_string(b, fmt[prev_i..i]);
- }
- if i >= end {
- break;
- }
- // Process a "verb"
- i += 1;
- prefix_loop: for ; i < end; i += 1 {
- match fmt[i] {
- case '+':
- fi.plus = true;
- case '-':
- fi.minus = true;
- fi.zero = false;
- case ' ':
- fi.space = true;
- case '#':
- fi.hash = true;
- case '0':
- fi.zero = !fi.minus;
- case:
- break prefix_loop;
- }
- }
- arg_index, i, was_prev_index = _arg_number(&fi, arg_index, fmt, i, len(args));
- // Width
- if i < end && fmt[i] == '*' {
- i += 1;
- fi.width, arg_index, fi.width_set = int_from_arg(args, arg_index);
- if !fi.width_set {
- write_string(b, "%!(BAD WIDTH)");
- }
- if fi.width < 0 {
- fi.width = -fi.width;
- fi.minus = true;
- fi.zero = false;
- }
- was_prev_index = false;
- } else {
- fi.width, i, fi.width_set = _parse_int(fmt, i);
- if was_prev_index && fi.width_set { // %[6]2d
- fi.good_arg_index = false;
- }
- }
- // Precision
- if i < end && fmt[i] == '.' {
- i += 1;
- if was_prev_index { // %[6].2d
- fi.good_arg_index = false;
- }
- if i < end && fmt[i] == '*' {
- arg_index, i, was_prev_index = _arg_number(&fi, arg_index, fmt, i, len(args));
- i += 1;
- fi.prec, arg_index, fi.prec_set = int_from_arg(args, arg_index);
- if fi.prec < 0 {
- fi.prec = 0;
- fi.prec_set = false;
- }
- if !fi.prec_set {
- write_string(fi.buf, "%!(BAD PRECISION)");
- }
- was_prev_index = false;
- } else {
- fi.prec, i, fi.prec_set = _parse_int(fmt, i);
- if !fi.prec_set {
- // fi.prec_set = true;
- // fi.prec = 0;
- }
- }
- }
- if !was_prev_index {
- arg_index, i, was_prev_index = _arg_number(&fi, arg_index, fmt, i, len(args));
- }
- if i >= end {
- write_string(b, "%!(NO VERB)");
- break;
- }
- verb, w := utf8.decode_rune(fmt[i..]);
- i += w;
- if verb == '%' {
- write_byte(b, '%');
- } else if !fi.good_arg_index {
- write_string(b, "%!(BAD ARGUMENT NUMBER)");
- } else if arg_index >= len(args) {
- write_string(b, "%!(MISSING ARGUMENT)");
- } else {
- fmt_arg(&fi, args[arg_index], verb);
- arg_index += 1;
- }
- }
- if !fi.reordered && arg_index < len(args) {
- write_string(b, "%!(EXTRA ");
- for arg, index in args[arg_index..] {
- if index > 0 do write_string(b, ", ");
- if arg == nil do write_string(b, "<nil>");
- else do fmt_arg(&fi, args[index], 'v');
- }
- write_string(b, ")");
- }
- return to_string(b^);
- }
|