diff --git a/src/libcore/reflect.rs b/src/libcore/reflect.rs index 12b02b1b8d219..28c6031263ba6 100644 --- a/src/libcore/reflect.rs +++ b/src/libcore/reflect.rs @@ -32,12 +32,12 @@ pub trait MovePtr { /// Helper function for alignment calculation. #[inline(always)] -fn align(size: uint, align: uint) -> uint { +pub fn align(size: uint, align: uint) -> uint { ((size + align) - 1u) & !(align - 1u) } /// Adaptor to wrap around visitors implementing MovePtr. -struct MovePtrAdaptor { +pub struct MovePtrAdaptor { inner: V } pub fn MovePtrAdaptor(v: V) -> MovePtrAdaptor { diff --git a/src/libcore/repr.rs b/src/libcore/repr.rs index 46bef4dd68aea..d3ed4f31515f5 100644 --- a/src/libcore/repr.rs +++ b/src/libcore/repr.rs @@ -17,27 +17,18 @@ More runtime type reflection #[forbid(deprecated_mode)]; #[forbid(deprecated_pattern)]; -use dvec::DVec; use io::{Writer, WriterUtil}; use libc::c_void; -use sys::TypeDesc; use to_str::ToStr; use cast::transmute; use intrinsic::{TyDesc, TyVisitor, visit_tydesc}; -use reflect::{MovePtr, MovePtrAdaptor}; +use reflect::{MovePtr, MovePtrAdaptor, align}; use vec::UnboxedVecRepr; use vec::raw::{VecRepr, SliceRepr}; pub use managed::raw::BoxRepr; -use managed::raw::BoxHeaderRepr; /// Helpers -/// Helper function for alignment calculation. -#[inline(always)] -fn align(size: uint, align: uint) -> uint { - ((size + align) - 1) & !(align - 1) -} - trait EscapedCharWriter { fn write_escaped_char(ch: char); } @@ -130,11 +121,11 @@ impl char : Repr { // New implementation using reflect::MovePtr -struct ReprVisitor { +pub struct ReprVisitor { mut ptr: *c_void, writer: @Writer } -fn ReprVisitor(ptr: *c_void, writer: @Writer) -> ReprVisitor { +pub fn ReprVisitor(ptr: *c_void, writer: @Writer) -> ReprVisitor { ReprVisitor { ptr: ptr, writer: writer } } @@ -458,7 +449,7 @@ impl ReprVisitor : TyVisitor { fn visit_closure_ptr(_ck: uint) -> bool { true } } -pub fn write_repr2(writer: @Writer, object: &T) { +pub fn write_repr(writer: @Writer, object: &T) { let ptr = ptr::to_unsafe_ptr(object) as *c_void; let tydesc = intrinsic::get_tydesc::(); let mut u = ReprVisitor(ptr, writer); @@ -467,545 +458,57 @@ pub fn write_repr2(writer: @Writer, object: &T) { } #[test] -fn test_repr2() { - repr::write_repr2(io::stdout(), &10); - io::println(""); - repr::write_repr2(io::stdout(), &true); - io::println(""); - repr::write_repr2(io::stdout(), &false); - io::println(""); - repr::write_repr2(io::stdout(), &1.234); - io::println(""); - repr::write_repr2(io::stdout(), &(&"hello")); - io::println(""); - repr::write_repr2(io::stdout(), &(@"hello")); - io::println(""); - repr::write_repr2(io::stdout(), &(~"he\u10f3llo")); - io::println(""); - repr::write_repr2(io::stdout(), &(@10)); - io::println(""); - repr::write_repr2(io::stdout(), &(@mut 10)); - io::println(""); - repr::write_repr2(io::stdout(), &(~10)); - io::println(""); - repr::write_repr2(io::stdout(), &(~mut 10)); - io::println(""); - repr::write_repr2(io::stdout(), &(&10)); - io::println(""); - let mut x = 10; - repr::write_repr2(io::stdout(), &(&mut x)); - io::println(""); - repr::write_repr2(io::stdout(), &(ptr::to_unsafe_ptr(&10) as *int)); - io::println(""); - repr::write_repr2(io::stdout(), &(@[1,2,3,4,5,6,7,8])); - io::println(""); - repr::write_repr2(io::stdout(), &(@[1u8,2u8,3u8,4u8])); - io::println(""); - repr::write_repr2(io::stdout(), &(@["hi", "there"])); - io::println(""); - repr::write_repr2(io::stdout(), &(~["hi", "there"])); - io::println(""); - repr::write_repr2(io::stdout(), &(&["hi", "there"])); - io::println(""); - repr::write_repr2(io::stdout(), &({a:10, b:1.234})); - io::println(""); - repr::write_repr2(io::stdout(), &(@{a:10, b:1.234})); - io::println(""); - repr::write_repr2(io::stdout(), &(~{a:10, b:1.234})); - io::println(""); -} - - -// Old non-factored implementation, transitional... - -#[deriving_eq] -enum EnumVisitState { - PreVariant, // We're before the variant we're interested in. - InVariant, // We're inside the variant we're interested in. - PostVariant, // We're after the variant we're interested in. - Degenerate // This is a degenerate enum (exactly 1 variant) -} - -struct EnumState { - end_ptr: *c_void, - state: EnumVisitState -} - -/// XXX: This should not use a boxed writer! -struct ReprPrinter { - mut ptr: *c_void, - writer: @Writer, // XXX: This should not use a boxed trait. - enum_stack: DVec -} - -/// FIXME (issue #3462): This is horrible. -struct ReprPrinterWrapper { - printer: @ReprPrinter -} +fn test_repr() { -impl ReprPrinter { - #[inline(always)] - fn align(n: uint) { - unsafe { - self.ptr = transmute(align(self.ptr as uint, n)); + fn exact_test(t: &T, e:&str) { + let s : &str = io::with_str_writer(|w| repr::write_repr(w, t)); + if s != e { + error!("expected '%s', got '%s'", + e, s); } + assert s == e; } - #[inline(always)] - fn bump(n: uint) { - unsafe { - self.ptr = transmute(self.ptr as uint + n); - } - } - #[inline(always)] - fn log_simple() -> bool { - unsafe { - self.align(sys::min_align_of::()); - let value_addr: &T = transmute(copy self.ptr); - value_addr.write_repr(self.writer); - self.bump(sys::size_of::()); - true - } - } -} - -impl ReprPrinterWrapper { - fn visit_estr() -> bool { - unsafe { - self.printer.writer.write_char('"'); - let vec_repr_ptr: **VecRepr = transmute(copy self.printer.ptr); - let vec_repr = *vec_repr_ptr; - let data_ptr = ptr::to_unsafe_ptr(&(*vec_repr).unboxed.data); - let slice: &str = transmute((data_ptr, (*vec_repr).unboxed.fill)); - for str::chars_each(slice) |ch| { - self.printer.writer.write_escaped_char(ch); - } - self.printer.writer.write_char('"'); - let ptr_size = sys::size_of::<*c_void>(); - self.printer.ptr = transmute(self.printer.ptr as uint + ptr_size); - true - } - } - - fn visit_self_describing_heap_alloc(mtbl: uint) -> bool { - unsafe { - if mtbl != 1 { self.printer.writer.write_str("mut "); } - let box_ptr: **BoxRepr = transmute(copy self.printer.ptr); - let box = *box_ptr; - self.printer.ptr = transmute(&(*box).data); - intrinsic::visit_tydesc((*box).header.type_desc, - self as @TyVisitor); - let box_size = sys::size_of::<*BoxRepr>(); - self.printer.ptr = transmute(box_ptr as uint + box_size); - true - } - } + exact_test(&10, "10"); + exact_test(&true, "true"); + exact_test(&false, "false"); + exact_test(&1.234, "1.2340"); + exact_test(&(&"hello"), "\"hello\""); + exact_test(&(@"hello"), "@\"hello\""); + exact_test(&(~"he\u10f3llo"), "~\"he\\u10f3llo\""); - fn visit_ptr_contents(mtbl: uint, inner: *TyDesc) -> bool { - unsafe { - if mtbl != 1 { self.printer.writer.write_str("mut "); } - let data_ptr: **c_void = transmute(copy self.printer.ptr); - if *data_ptr == ptr::null() { - self.printer.writer.write_str("null"); - } else { - self.printer.ptr = *data_ptr; - intrinsic::visit_tydesc(inner, self as @TyVisitor); - } - let ptr_size = sys::size_of::<*c_void>(); - self.printer.ptr = transmute(data_ptr as uint + ptr_size); - true - } - } - - fn visit_evec(mtbl: uint, inner: *TyDesc) -> bool { - unsafe { - self.printer.writer.write_char('['); - self.printer.align(sys::min_align_of::<*c_void>()); - let vec_repr_ptr: **VecRepr = transmute(copy self.printer.ptr); - let old_ptr = self.printer.ptr as uint; - let vec_repr: *VecRepr = *vec_repr_ptr; - self.printer.ptr = transmute(&(*vec_repr).unboxed.data); - let end_ptr: *c_void = transmute(self.printer.ptr as uint + - (*vec_repr).unboxed.fill); - let sys_tydesc: *TypeDesc = transmute(copy inner); - let alignment = (*sys_tydesc).align; - let mut first = true; - loop { - self.printer.align(alignment); - if self.printer.ptr >= end_ptr { break; } - if first { - self.printer.writer.write_char(' '); - if mtbl != 1 { self.printer.writer.write_str("mut "); } - } else { - self.printer.writer.write_str(", "); - } - intrinsic::visit_tydesc(inner, self as @TyVisitor); - first = false; - } - if !first { - self.printer.writer.write_char(' '); - } else if mtbl != 1 { - self.printer.writer.write_str("mut"); - } - self.printer.writer.write_char(']'); - self.printer.ptr = transmute(old_ptr + sys::size_of::()); - true - } - } -} - -impl ReprPrinterWrapper : TyVisitor { - fn visit_bot() -> bool { - self.printer.bump(1); - self.printer.writer.write_str("fail"); - true - } - - fn visit_nil() -> bool { self.printer.log_simple::<()>() } - fn visit_bool() -> bool { self.printer.log_simple::() } - - // Numbers - - fn visit_int() -> bool { self.printer.log_simple::() } - fn visit_i8() -> bool { self.printer.log_simple::() } - fn visit_i16() -> bool { self.printer.log_simple::() } - fn visit_i32() -> bool { self.printer.log_simple::() } - fn visit_i64() -> bool { self.printer.log_simple::() } - - fn visit_uint() -> bool { self.printer.log_simple::() } - fn visit_u8() -> bool { self.printer.log_simple::() } - fn visit_u16() -> bool { self.printer.log_simple::() } - fn visit_u32() -> bool { self.printer.log_simple::() } - fn visit_u64() -> bool { self.printer.log_simple::() } - - fn visit_float() -> bool { self.printer.log_simple::() } - fn visit_f32() -> bool { self.printer.log_simple::() } - fn visit_f64() -> bool { self.printer.log_simple::() } - - fn visit_char() -> bool { self.printer.log_simple::() } - fn visit_str() -> bool { true } - - // Strings - - fn visit_estr_box() -> bool { - self.printer.writer.write_char('@'); - self.visit_estr() - } - fn visit_estr_uniq() -> bool { - self.printer.writer.write_char('~'); - self.visit_estr() - } - fn visit_estr_slice() -> bool { - unsafe { - self.printer.writer.write_char('"'); - let slice_ptr: *&str = transmute(copy self.printer.ptr); - let slice = *slice_ptr; - for str::chars_each(slice) |ch| { - self.printer.writer.write_escaped_char(ch); - } - self.printer.writer.write_char('"'); - let slice_sz = sys::size_of::<(*char,uint)>(); - self.printer.ptr = transmute(self.printer.ptr as uint + slice_sz); - true - } - } - fn visit_estr_fixed(_n: uint, _sz: uint, _align: uint) -> bool { true } - - // Pointers - - fn visit_box(mtbl: uint, _inner: *TyDesc) -> bool { - self.printer.writer.write_char('@'); - self.visit_self_describing_heap_alloc(mtbl) - } - fn visit_uniq(mtbl: uint, _inner: *TyDesc) -> bool { - self.printer.writer.write_char('~'); - self.visit_self_describing_heap_alloc(mtbl) - } - fn visit_ptr(mtbl: uint, inner: *TyDesc) -> bool { - self.printer.writer.write_char('*'); - self.visit_ptr_contents(mtbl, inner) - } - fn visit_rptr(mtbl: uint, inner: *TyDesc) -> bool { - self.printer.writer.write_char('&'); - self.visit_ptr_contents(mtbl, inner) - } - - // Vectors - - fn visit_vec(_mtbl: uint, _inner: *TyDesc) -> bool { true } - fn visit_unboxed_vec(_mtbl: uint, _inner: *TyDesc) -> bool { true } - fn visit_evec_box(mtbl: uint, inner: *TyDesc) -> bool { - self.printer.writer.write_char('@'); - self.visit_evec(mtbl, inner) - } - fn visit_evec_uniq(mtbl: uint, inner: *TyDesc) -> bool { - self.printer.writer.write_char('~'); - self.visit_evec(mtbl, inner) - } - fn visit_evec_slice(mtbl: uint, inner: *TyDesc) -> bool { - unsafe { - self.printer.writer.write_str("&["); - self.printer.align(sys::min_align_of::<(*c_void,uint)>()); - let slice_ptr: *(*c_void,uint) = transmute(copy self.printer.ptr); - let (data, fill) = *slice_ptr; - self.printer.ptr = data; - let end_ptr: *c_void = transmute(self.printer.ptr as uint + fill); - let sys_tydesc: *TypeDesc = transmute(copy inner); - let alignment = (*sys_tydesc).align; - let mut first = true; - loop { - self.printer.align(alignment); - if self.printer.ptr >= end_ptr { break; } - if first { - self.printer.writer.write_char(' '); - if mtbl != 1 { self.printer.writer.write_str("mut "); } - } else { - self.printer.writer.write_str(", "); - } - intrinsic::visit_tydesc(inner, self as @TyVisitor); - first = false; - } - if !first { - self.printer.writer.write_char(' '); - } else if mtbl != 1 { - self.printer.writer.write_str("mut"); - } - self.printer.writer.write_char(']'); - let slice_size = sys::size_of::<(uint, *c_void)>(); - self.printer.ptr = transmute(slice_ptr as uint + slice_size); - true - } - } - fn visit_evec_fixed(n: uint, sz: uint, align: uint, mtbl: uint, - inner: *TyDesc) -> bool { - unsafe { - self.printer.writer.write_char('['); - self.printer.align(align); - let end_ptr: *c_void = transmute(self.printer.ptr as uint + sz); - for uint::range(0, n) |i| { - self.printer.align(align); - if i == 0 { - self.printer.writer.write_char(' '); - if mtbl != 1 { self.printer.writer.write_str("mut "); } - } else { - self.printer.writer.write_str(", "); - } - intrinsic::visit_tydesc(inner, self as @TyVisitor); - } - if n > 0 { - self.printer.writer.write_char(' '); - } else if mtbl != 1 { - self.printer.writer.write_str("mut"); - } - self.printer.writer.write_char(']'); - self.printer.ptr = end_ptr; - true - } - } - - // Records - - fn visit_enter_rec(_n_fields: uint, _sz: uint, align: uint) -> bool { - self.printer.writer.write_char('{'); - self.printer.align(align); - true - } - fn visit_rec_field(i: uint, name: &str, mtbl: uint, inner: *TyDesc) -> - bool { - if i != 0 { - self.printer.writer.write_str(", "); - } else { - self.printer.writer.write_char(' '); - } - if mtbl != 1 { self.printer.writer.write_str("mut "); } - self.printer.writer.write_str(name); - self.printer.writer.write_str(": "); - intrinsic::visit_tydesc(inner, self as @TyVisitor); - true - } - fn visit_leave_rec(n_fields: uint, _sz: uint, _align: uint) -> bool { - if n_fields > 0 { self.printer.writer.write_char(' '); } - self.printer.writer.write_char('}'); - true - } - - // Structs - - fn visit_enter_class(_n_fields: uint, _sz: uint, align: uint) -> bool { - self.printer.writer.write_char('{'); - self.printer.align(align); - true - } - fn visit_class_field(i: uint, name: &str, mtbl: uint, inner: *TyDesc) -> - bool { - if i != 0 { - self.printer.writer.write_str(", "); - } else { - self.printer.writer.write_char(' '); - } - if mtbl != 1 { self.printer.writer.write_str("mut "); } - self.printer.writer.write_str(name); - self.printer.writer.write_str(": "); - intrinsic::visit_tydesc(inner, self as @TyVisitor); - true - } - fn visit_leave_class(n_fields: uint, _sz: uint, _align: uint) -> bool { - if n_fields > 0 { self.printer.writer.write_char(' '); } - self.printer.writer.write_char('}'); - true - } - - // Tuples - - fn visit_enter_tup(_n_fields: uint, _sz: uint, align: uint) -> bool { - self.printer.writer.write_char('('); - self.printer.align(align); - true - } - fn visit_tup_field(i: uint, inner: *TyDesc) -> bool { - if i != 0 { self.printer.writer.write_str(", "); } - intrinsic::visit_tydesc(inner, self as @TyVisitor); - true - } - fn visit_leave_tup(_n_fields: uint, _sz: uint, _align: uint) -> bool { - self.printer.writer.write_char(')'); - true - } - - // Enums - - fn visit_enter_enum(n_variants: uint, sz: uint, align: uint) -> bool { - unsafe { - self.printer.align(align); - - // Write in the location of the end of this enum. - let end_ptr = transmute(self.printer.ptr as uint + sz); - let state = if n_variants == 1 { Degenerate } else { PreVariant }; - let new_state = EnumState { end_ptr: end_ptr, state: state }; - self.printer.enum_stack.push(new_state); - - true - } - } - - fn visit_enter_enum_variant(_variant: uint, - disr_val: int, - _n_fields: uint, - name: &str) -> bool { - unsafe { - let stack = &self.printer.enum_stack; - let mut enum_state = stack.last(); - match enum_state.state { - PreVariant => { - let disr_ptr = self.printer.ptr as *int; - if *disr_ptr == disr_val { - enum_state.state = InVariant; - self.printer.writer.write_str(name); - self.printer.bump(sys::size_of::()); - stack.set_elt(stack.len() - 1, enum_state); - } - } - Degenerate => { - self.printer.writer.write_str(name); - } - InVariant | PostVariant => {} - } - true - } - } - - fn visit_enum_variant_field(i: uint, inner: *TyDesc) -> bool { - match self.printer.enum_stack.last().state { - InVariant | Degenerate => { - if i == 0 { - self.printer.writer.write_char('('); - } else { - self.printer.writer.write_str(", "); - } - - intrinsic::visit_tydesc(inner, self as @TyVisitor); - } - PreVariant | PostVariant => {} - } - true - } - - fn visit_leave_enum_variant(_variant: uint, - _disr_val: int, - n_fields: uint, - _name: &str) -> bool { - let stack = &self.printer.enum_stack; - let mut enum_state = stack.last(); - match enum_state.state { - InVariant => { - if n_fields >= 1 { self.printer.writer.write_char(')'); } - enum_state.state = PostVariant; - stack.set_elt(stack.len() - 1, enum_state); - } - Degenerate => { - if n_fields >= 1 { self.printer.writer.write_char(')'); } - } - PreVariant | PostVariant => {} - } - true - } - - fn visit_leave_enum(_n_variants: uint, _sz: uint, _align: uint) -> bool { - self.printer.ptr = self.printer.enum_stack.pop().end_ptr; - true - } - - // Functions - - fn visit_enter_fn(_purity: uint, proto: uint, - _n_inputs: uint, _retstyle: uint) -> bool { - self.printer.align(sys::min_align_of::<(uint,uint)>()); - match proto { - 2u => self.printer.writer.write_char('~'), - 3u => self.printer.writer.write_char('@'), - 4u => self.printer.writer.write_char('&'), - _ => {} - } - self.printer.writer.write_str("fn"); - true - } - - fn visit_fn_input(_i: uint, _mode: uint, _inner: *TyDesc) -> bool { true } - fn visit_fn_output(_retstyle: uint, _inner: *TyDesc) -> bool { true } - fn visit_leave_fn(_purity: uint, _proto: uint, - _n_inputs: uint, _retstyle: uint) -> bool { true } - - // Others - - fn visit_trait() -> bool { self.printer.writer.write_str("@trait"); true } - fn visit_var() -> bool { true } - fn visit_var_integral() -> bool { true } - fn visit_param(_i: uint) -> bool { true } - fn visit_self() -> bool { true } - fn visit_type() -> bool { true } - fn visit_opaque_box() -> bool { - self.printer.writer.write_char('@'); - self.visit_self_describing_heap_alloc(1) - } - fn visit_constr(_inner: *TyDesc) -> bool { true } - fn visit_closure_ptr(_ck: uint) -> bool { true } -} - -pub fn write_repr(writer: @Writer, object: &T) { - unsafe { - let ptr = ptr::to_unsafe_ptr(object) as *c_void; - let tydesc = sys::get_type_desc::(); - let tydesc = cast::transmute(move tydesc); - - let repr_printer = @ReprPrinter { - ptr: ptr, - writer: writer, - enum_stack: DVec() - }; - - let wrapper = ReprPrinterWrapper { printer: repr_printer }; - intrinsic::visit_tydesc(tydesc, wrapper as @TyVisitor); - } + // FIXME #4210: the mut fields are a bit off here. + exact_test(&(@10), "@10"); + exact_test(&(@mut 10), "@10"); + exact_test(&(~10), "~10"); + exact_test(&(~mut 10), "~mut 10"); + exact_test(&(&10), "&10"); + let mut x = 10; + exact_test(&(&mut x), "&mut 10"); + + exact_test(&(@[1,2,3,4,5,6,7,8]), + "@[1, 2, 3, 4, 5, 6, 7, 8]"); + exact_test(&(@[1u8,2u8,3u8,4u8]), + "@[1, 2, 3, 4]"); + exact_test(&(@["hi", "there"]), + "@[\"hi\", \"there\"]"); + exact_test(&(~["hi", "there"]), + "~[\"hi\", \"there\"]"); + exact_test(&(&["hi", "there"]), + "&[\"hi\", \"there\"]"); + exact_test(&({a:10, b:1.234}), + "{a: 10, b: 1.2340}"); + exact_test(&(@{a:10, b:1.234}), + "@{a: 10, b: 1.2340}"); + exact_test(&(~{a:10, b:1.234}), + "~{a: 10, b: 1.2340}"); + exact_test(&(10_u8, ~"hello"), + "(10, ~\"hello\")"); + exact_test(&(10_u16, ~"hello"), + "(10, ~\"hello\")"); + exact_test(&(10_u32, ~"hello"), + "(10, ~\"hello\")"); + exact_test(&(10_u64, ~"hello"), + "(10, ~\"hello\")"); } -