12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058 |
- #import "os.odin";
- #import "mem.odin";
- #import "utf8.odin";
- #import "types.odin";
- #import "strconv.odin";
- _BUFFER_SIZE :: 1<<12;
- write_string :: proc(buf: ^[]byte, s: string) {
- append(buf, ..cast([]byte)s);
- }
- write_byte :: proc(buf: ^[]byte, b: byte) {
- append(buf, b);
- }
- write_rune :: proc(buf: ^[]byte, r: rune) {
- if r < utf8.RUNE_SELF {
- write_byte(buf, cast(byte)r);
- return;
- }
- b, n := utf8.encode_rune(r);
- append(buf, ..b[..n]);
- }
- Fmt_Info :: struct {
- minus: bool,
- plus: bool,
- space: bool,
- zero: bool,
- hash: bool,
- width_set: bool,
- prec_set: bool,
- width: int,
- prec: int,
- reordered: bool,
- good_arg_index: bool,
- buf: ^[]byte,
- arg: any, // Temporary
- }
- fprint :: proc(fd: os.Handle, args: ..any) -> int {
- data: [_BUFFER_SIZE]byte;
- buf := data[..0];
- bprint(^buf, ..args);
- os.write(fd, buf[..buf.count]);
- return buf.count;
- }
- fprintln :: proc(fd: os.Handle, args: ..any) -> int {
- data: [_BUFFER_SIZE]byte;
- buf := data[..0];
- bprintln(^buf, ..args);
- os.write(fd, buf[..buf.count]);
- return buf.count;
- }
- fprintf :: proc(fd: os.Handle, fmt: string, args: ..any) -> int {
- data: [_BUFFER_SIZE]byte;
- buf := data[..0];
- bprintf(^buf, fmt, ..args);
- os.write(fd, buf[..buf.count]);
- return buf.count;
- }
- print :: proc(args: ..any) -> int {
- return fprint(os.stdout, ..args);
- }
- println :: proc(args: ..any) -> int {
- return fprintln(os.stdout, ..args);
- }
- printf :: proc(fmt: string, args: ..any) -> int {
- return fprintf(os.stdout, fmt, ..args);
- }
- fprint_type :: proc(fd: os.Handle, info: ^Type_Info) {
- data: [_BUFFER_SIZE]byte;
- buf := data[..0];
- write_type(^buf, info);
- os.write(fd, buf[..buf.count]);
- }
- write_type :: proc(buf: ^[]byte, ti: ^Type_Info) {
- if ti == nil {
- return;
- }
- using Type_Info;
- match info in ti {
- case Named:
- write_string(buf, info.name);
- case Integer:
- match {
- case ti == type_info(int): write_string(buf, "int");
- case ti == type_info(uint): write_string(buf, "uint");
- default:
- write_string(buf, info.signed ? "i" : "u");
- fi := Fmt_Info{buf = buf};
- fmt_int(^fi, cast(u64)(8*info.size), false, 64, 'd');
- }
- case Float:
- match info.size {
- case 4: write_string(buf, "f32");
- case 8: write_string(buf, "f64");
- }
- case String: write_string(buf, "string");
- case Boolean: write_string(buf, "bool");
- 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 := union_cast(^Tuple)info.params;
- write_string(buf, "(");
- for type, i in t.types {
- if i > 0 { write_string(buf, ", "); }
- write_type(buf, type);
- }
- write_string(buf, ")");
- }
- if info.results != nil {
- write_string(buf, " -> ");
- write_type(buf, info.results);
- }
- case Tuple:
- count := info.names.count;
- if count != 1 { write_string(buf, "("); }
- for name, i in info.names {
- if i > 0 { write_string(buf, ", "); }
- type := info.types[i];
- if name.count > 0 {
- write_string(buf, name);
- write_string(buf, ": ");
- }
- write_type(buf, type);
- }
- if count != 1 { write_string(buf, ")"); }
- case Array:
- write_string(buf, "[");
- fi := Fmt_Info{buf = buf};
- fmt_int(^fi, cast(u64)info.count, false, 64, 'd');
- write_string(buf, "]");
- write_type(buf, info.elem);
- case Dynamic_Array:
- write_string(buf, "[..]");
- write_type(buf, info.elem);
- case Slice:
- write_string(buf, "[");
- write_string(buf, "]");
- write_type(buf, info.elem);
- case Vector:
- write_string(buf, "[vector ");
- fi := Fmt_Info{buf = buf};
- fmt_int(^fi, cast(u64)info.count, false, 64, 'd');
- 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.packed { write_string(buf, "#packed "); }
- if info.ordered { write_string(buf, "#ordered "); }
- if info.custom_align {
- write_string(buf, "#align ");
- fi := Fmt_Info{buf = buf};
- fmt_int(^fi, cast(u64)info.align, false, 64, 'd');
- write_byte(buf, ' ');
- }
- write_byte(buf, '{');
- for name, i in info.names {
- if i > 0 {
- 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 {");
- cf := info.common_fields;
- total_count := 0;
- for name, i in cf.names {
- if i > 0 {
- write_string(buf, ", ");
- }
- write_string(buf, name);
- write_string(buf, ": ");
- write_type(buf, cf.types[i]);
- total_count++;
- }
- for name, i in info.variant_names {
- if total_count > 0 || i > 0 {
- write_string(buf, ", ");
- }
- write_string(buf, name);
- write_byte(buf, '{');
- defer write_byte(buf, '}');
- variant_type := type_info_base(info.variant_types[i]);
- variant := union_cast(^Struct)variant_type;
- vc := variant.names.count-cf.names.count;
- for j in 0..vc {
- if j > 0 {
- write_string(buf, ", ");
- }
- index := j + cf.names.count;
- write_string(buf, variant.names[index]);
- write_string(buf, ": ");
- write_type(buf, variant.types[index]);
- }
- }
- write_string(buf, "}");
- case Raw_Union:
- write_string(buf, "raw_union {");
- for name, i in info.names {
- if i > 0 {
- write_string(buf, ", ");
- }
- write_string(buf, name);
- write_string(buf, ": ");
- write_type(buf, info.types[i]);
- }
- 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 {
- write_string(buf, ", ");
- }
- write_string(buf, name);
- }
- write_string(buf, "}");
- }
- }
- bprint :: proc(buf: ^[]byte, args: ..any) -> int {
- fi: Fmt_Info;
- fi.buf = buf;
- prev_string := false;
- for arg, i in args {
- is_string := arg.data != 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 buf.count;
- }
- bprintln :: proc(buf: ^[]byte, args: ..any) -> int {
- fi: Fmt_Info;
- fi.buf = buf;
- for arg, i in args {
- if i > 0 {
- write_byte(buf, ' ');
- }
- fmt_value(^fi, args[i], 'v');
- }
- write_byte(buf, '\n');
- return buf.count;
- }
- sprint :: proc(buf: []byte, args: ..any) -> string {
- count := bprint(^buf, ..args);
- return cast(string)buf[..count];
- }
- sprintln :: proc(buf: []byte, args: ..any) -> string {
- count := bprintln(^buf, ..args);
- return cast(string)buf[..count];
- }
- sprintf :: proc(buf: []byte, fmt: string, args: ..any) -> string {
- count := bprintf(^buf, fmt, ..args);
- return cast(string)buf[..count];
- }
- 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 o in offset..s.count {
- c := cast(rune)s[offset+i];
- if !is_digit(c) {
- break;
- }
- i++;
- result *= 10;
- result += cast(int)(c - '0');
- }
- return result, offset+i, i != 0;
- }
- _arg_number :: proc(fi: ^Fmt_Info,
- arg_index: int,
- format: string,
- offset: int,
- arg_count: int,
- ) -> (index: int, offset: int, ok: bool) {
- parse_arg_number :: proc(format: string) -> (int, int, bool) {
- if format.count < 3 {
- return 0, 1, false;
- }
- for i in 1..format.count {
- 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 format.count <= 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 < args.count {
- arg := args[arg_index];
- arg.type_info = type_info_base(arg.type_info);
- match i in arg {
- case int: num = i;
- case i8: num = cast(int)i;
- case i16: num = cast(int)i;
- case i32: num = cast(int)i;
- case i64: num = cast(int)i;
- case u8: num = cast(int)i;
- case u16: num = cast(int)i;
- case u32: num = cast(int)i;
- case u64: num = cast(int)i;
- default:
- ok = false;
- }
- }
- return num, new_arg_index, ok;
- }
- fmt_bad_verb :: proc(using fi: ^Fmt_Info, 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: ^Fmt_Info, b: bool, verb: rune) {
- match verb {
- case 't', 'v':
- write_string(buf, b ? "true" : "false");
- default:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_write_padding :: proc(fi: ^Fmt_Info, width: int) {
- if width <= 0 {
- return;
- }
- pad_byte: byte = ' ';
- if fi.zero {
- pad_byte = '0';
- }
- count := min(width, fi.buf.capacity-fi.buf.count);
- for _ in 0..count {
- append(fi.buf, pad_byte);
- }
- }
- is_integer_negative :: proc(u: u64, is_signed: bool, bit_size: int) -> (unsigned: u64, neg: bool) {
- neg := false;
- if is_signed {
- match bit_size {
- case 8:
- i := cast(i8)u;
- neg = i < 0;
- if neg { i = -i; }
- u = cast(u64)i;
- case 16:
- i := cast(i16)u;
- neg = i < 0;
- if neg { i = -i; }
- u = cast(u64)i;
- case 32:
- i := cast(i32)u;
- neg = i < 0;
- if neg { i = -i; }
- u = cast(u64)i;
- case 64:
- i := cast(i64)u;
- neg = i < 0;
- if neg { i = -i; }
- u = cast(u64)i;
- default:
- panic("is_integer_negative: Unknown integer size");
- }
- }
- return u, neg;
- }
- _write_int :: proc(fi: ^Fmt_Info, u: u64, base: int, is_signed: bool, bit_size: int, digits: string) {
- _, neg := is_integer_negative(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("_write_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--;
- }
- }
- match base {
- case 2, 8, 10, 12, 16:
- break;
- default:
- panic("_write_int: unknown base, whoops");
- }
- buf: [256]byte;
- flags: strconv.Int_Flag;
- if fi.hash { flags |= strconv.Int_Flag.PREFIX; }
- if fi.plus { flags |= strconv.Int_Flag.PLUS; }
- if fi.space { flags |= strconv.Int_Flag.SPACE; }
- s := strconv.append_bits(buf[..0], u, base, is_signed, bit_size, digits, flags);
- prev_zero := fi.zero;
- defer fi.zero = prev_zero;
- fi.zero = false;
- _pad(fi, s);
- }
- immutable __DIGITS_LOWER := "0123456789abcdefx";
- immutable __DIGITS_UPPER := "0123456789ABCDEFX";
- fmt_rune :: proc(fi: ^Fmt_Info, r: rune) {
- write_rune(fi.buf, r);
- }
- fmt_int :: proc(fi: ^Fmt_Info, u: u64, is_signed: bool, bit_size: int, verb: rune) {
- match verb {
- case 'v': _write_int(fi, u, 10, is_signed, bit_size, __DIGITS_LOWER);
- case 'b': _write_int(fi, u, 2, is_signed, bit_size, __DIGITS_LOWER);
- case 'o': _write_int(fi, u, 8, is_signed, bit_size, __DIGITS_LOWER);
- case 'd': _write_int(fi, u, 10, is_signed, bit_size, __DIGITS_LOWER);
- case 'x': _write_int(fi, u, 16, is_signed, bit_size, __DIGITS_LOWER);
- case 'X': _write_int(fi, u, 16, is_signed, bit_size, __DIGITS_UPPER);
- case 'c', 'r':
- fmt_rune(fi, cast(rune)u);
- case 'U':
- r := cast(rune)u;
- if r < 0 || r > utf8.MAX_RUNE {
- fmt_bad_verb(fi, verb);
- } else {
- write_string(fi.buf, "U+");
- _write_int(fi, u, 16, false, bit_size, __DIGITS_UPPER);
- }
- default:
- fmt_bad_verb(fi, verb);
- }
- }
- _pad :: proc(fi: ^Fmt_Info, s: string) {
- if !fi.width_set || fi.width == 0 {
- 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: ^Fmt_Info, 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 {
- prec = fi.prec;
- }
- buf: [128]byte;
- str := strconv.append_float(buf[1..1], v, 'f', prec, bit_size);
- str = cast(string)buf[..str.count+1];
- if str[1] == '+' || str[1] == '-' {
- str = str[1..];
- } else {
- str[0] = '+';
- }
- if fi.space && !fi.plus && str[0] == '+' {
- str[0] = ' ';
- }
- if 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 > str.count {
- write_byte(fi.buf, str[0]);
- fmt_write_padding(fi, fi.width - str.count);
- write_string(fi.buf, str[1..]);
- } else {
- _pad(fi, str);
- }
- } else {
- _pad(fi, str[1..]);
- }
- default:
- fmt_bad_verb(fi, verb);
- return;
- }
- }
- fmt_string :: proc(fi: ^Fmt_Info, s: string, verb: rune) {
- match verb {
- case 's', 'v':
- write_string(fi.buf, s);
- default:
- fmt_bad_verb(fi, verb);
- }
- }
- fmt_pointer :: proc(fi: ^Fmt_Info, p: rawptr, verb: rune) {
- match verb {
- case 'p', 'v':
- // Okay
- default:
- fmt_bad_verb(fi, verb);
- return;
- }
- u := cast(u64)cast(uint)p;
- if !fi.hash || verb == 'v' {
- write_string(fi.buf, "0x");
- }
- _write_int(fi, u, 16, false, 8*size_of(rawptr), __DIGITS_UPPER);
- }
- fmt_enum :: proc(fi: ^Fmt_Info, v: any, verb: rune) {
- if v.type_info == nil || v.data == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- using Type_Info;
- match e in v.type_info {
- default:
- fmt_bad_verb(fi, verb);
- return;
- case Enum:
- match verb {
- case 'd', 'f':
- fmt_arg(fi, any{type_info_base(e.base), v.data}, verb);
- case 's', 'v':
- i: i64;
- f: f64;
- ok := false;
- a := any{type_info_base(e.base), v.data};
- match v in a {
- case i8: i = cast(i64)v;
- case i16: i = cast(i64)v;
- case i32: i = cast(i64)v;
- case i64: i = cast(i64)v;
- case int: i = cast(i64)v;
- case u8: i = cast(i64)v;
- case u16: i = cast(i64)v;
- case u32: i = cast(i64)v;
- case u64: i = cast(i64)v;
- case uint: i = cast(i64)v;
- case f32: f = cast(f64)v; i = transmute(i64)f;
- case f64: f = cast(f64)v; i = transmute(i64)f;
- }
- if types.is_string(e.base) {
- for val, idx in e.values {
- if val.i == i {
- write_string(fi.buf, e.names[idx]);
- ok = true;
- break;
- }
- }
- } else if e.values.count == 0 {
- write_string(fi.buf, "");
- ok = true;
- } else {
- for val, idx in e.values {
- if val.i == i {
- write_string(fi.buf, e.names[idx]);
- ok = true;
- break;
- }
- }
- }
- if !ok {
- write_string(fi.buf, "!%(BAD ENUM VALUE)");
- }
- default:
- fmt_bad_verb(fi, verb);
- return;
- }
- }
- }
- fmt_value :: proc(fi: ^Fmt_Info, v: any, verb: rune) {
- if v.data == nil || v.type_info == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- using Type_Info;
- match info in v.type_info {
- case Named:
- match b in info.base {
- case Struct:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_string(fi.buf, info.name);
- write_byte(fi.buf, '{');
- for _, i in b.names {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- write_string(fi.buf, b.names[i]);
- write_string(fi.buf, " = ");
- data := cast(^byte)v.data + b.offsets[i];
- fmt_arg(fi, any{b.types[i], cast(rawptr)data}, 'v');
- }
- write_byte(fi.buf, '}');
- default:
- fmt_value(fi, any{info.base, v.data}, verb);
- }
- case Boolean: fmt_arg(fi, v, verb);
- case Float: fmt_arg(fi, v, verb);
- case Integer: fmt_arg(fi, v, verb);
- case String: fmt_arg(fi, v, verb);
- case Pointer:
- if v.type_info == type_info(^Type_Info) {
- write_type(fi.buf, (cast(^^Type_Info)v.data)^);
- } else {
- fmt_pointer(fi, (cast(^rawptr)v.data)^, verb);
- }
- case Array:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- for i in 0..info.count {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- data := cast(^byte)v.data + i*info.elem_size;
- fmt_arg(fi, any{info.elem, cast(rawptr)data}, 'v');
- }
- case Dynamic_Array:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- array := cast(^Raw_Dynamic_Array)v.data;
- for i in 0..array.count {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- data := cast(^byte)array.data + i*info.elem_size;
- fmt_arg(fi, any{info.elem, cast(rawptr)data}, 'v');
- }
- case Map:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_string(fi.buf, "map[");
- defer write_byte(fi.buf, ']');
- entries := ^(cast(^Raw_Dynamic_Map)v.data).entries;
- gs := union_cast(^Struct)type_info_base(info.generated_struct);
- ed := union_cast(^Dynamic_Array)type_info_base(gs.types[1]);
- entry_type := union_cast(^Struct)ed.elem;
- entry_size := ed.elem_size;
- for i in 0..entries.count {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- data := cast(^byte)entries.data + i*entry_size;
- header := cast(^__Map_Entry_Header)data;
- if types.is_string(info.key) {
- write_string(fi.buf, header.key.str);
- } else {
- fi := Fmt_Info{buf = fi.buf};
- fmt_arg(^fi, any{info.key, cast(rawptr)^header.key.hash}, 'v');
- }
- write_string(fi.buf, "=");
- value := data + entry_type.offsets[2];
- fmt_arg(fi, any{info.value, cast(rawptr)value}, 'v');
- }
- case Slice:
- if verb != 'v' {
- fmt_bad_verb(fi, verb);
- return;
- }
- write_byte(fi.buf, '[');
- defer write_byte(fi.buf, ']');
- slice := cast(^[]byte)v.data;
- for i in 0..slice.count {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- data := slice.data + i*info.elem_size;
- fmt_arg(fi, any{info.elem, cast(rawptr)data}, 'v');
- }
- case Vector:
- write_byte(fi.buf, '<');
- defer write_byte(fi.buf, '>');
- for i in 0..info.count {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- data := cast(^byte)v.data + i*info.elem_size;
- fmt_value(fi, any{info.elem, cast(rawptr)data}, 'v');
- }
- case Struct:
- write_byte(fi.buf, '{');
- defer write_byte(fi.buf, '}');
- for _, i in info.names {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- write_string(fi.buf, info.names[i]);
- write_string(fi.buf, " = ");
- data := cast(^byte)v.data + info.offsets[i];
- fmt_value(fi, any{info.types[i], cast(rawptr)data}, 'v');
- }
- case Union:
- write_byte(fi.buf, '{');
- defer write_byte(fi.buf, '}');
- cf := info.common_fields;
- for _, i in cf.names {
- if i > 0 {
- write_string(fi.buf, ", ");
- }
- write_string(fi.buf, cf.names[i]);
- write_string(fi.buf, " = ");
- data := cast(^byte)v.data + cf.offsets[i];
- fmt_value(fi, any{cf.types[i], cast(rawptr)data}, 'v');
- }
- case Raw_Union:
- write_string(fi.buf, "(raw_union)");
- 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_arg :: proc(fi: ^Fmt_Info, arg: any, verb: rune) {
- if arg.data == nil || arg.type_info == nil {
- write_string(fi.buf, "<nil>");
- return;
- }
- fi.arg = arg;
- if verb == 'T' {
- ti := arg.type_info;
- match a in arg {
- case ^Type_Info: 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 bool: fmt_bool(fi, a, verb);
- case f32: fmt_float(fi, cast(f64)a, 32, verb);
- case f64: fmt_float(fi, a, 64, verb);
- case int: fmt_int(fi, cast(u64)a, true, 8*size_of(int), verb);
- case i8: fmt_int(fi, cast(u64)a, true, 8, verb);
- case i16: fmt_int(fi, cast(u64)a, true, 16, verb);
- case i32: fmt_int(fi, cast(u64)a, true, 32, verb);
- case i64: fmt_int(fi, cast(u64)a, true, 64, verb);
- case uint: fmt_int(fi, cast(u64)a, false, 8*size_of(uint), verb);
- case u8: fmt_int(fi, cast(u64)a, false, 8, verb);
- case u16: fmt_int(fi, cast(u64)a, false, 16, verb);
- case u32: fmt_int(fi, cast(u64)a, false, 32, verb);
- case u64: fmt_int(fi, cast(u64)a, false, 64, verb);
- case string: fmt_string(fi, a, verb);
- default: fmt_value(fi, arg, verb);
- }
- }
- bprintf :: proc(b: ^[]byte, fmt: string, args: ..any) -> int {
- fi := Fmt_Info{};
- end := fmt.count;
- arg_index := 0;
- was_prev_index := false;
- for i := 0; i < end; {
- fi = Fmt_Info{buf = b, good_arg_index = true};
- prev_i := i;
- for i < end && fmt[i] != '%' {
- i++;
- }
- if i > prev_i {
- write_string(b, fmt[prev_i..i]);
- }
- if i >= end {
- break;
- }
- // Process a "verb"
- i++;
- #label prefix_loop
- for ; i < end; i++ {
- 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;
- default:
- break prefix_loop;
- }
- }
- arg_index, i, was_prev_index = _arg_number(^fi, arg_index, fmt, i, args.count);
- // Width
- if i < end && fmt[i] == '*' {
- i++;
- 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++;
- 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, args.count);
- i++;
- 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, args.count);
- }
- 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 >= args.count {
- write_string(b, "%!(MISSING ARGUMENT)");
- } else {
- fmt_arg(^fi, args[arg_index], verb);
- arg_index++;
- }
- }
- if !fi.reordered && arg_index < args.count {
- write_string(b, "%!(EXTRA ");
- for arg, index in args[arg_index..] {
- if index > 0 {
- write_string(b, ", ");
- }
- if arg.data == nil || arg.type_info == nil {
- write_string(b, "<nil>");
- } else {
- fmt_arg(^fi, args[index], 'v');
- }
- }
- write_string(b, ")");
- }
- return b.count;
- }
|