]> git.lizzy.rs Git - rust.git/commitdiff
Update and add test case
authorJared Roesch <roeschinc@gmail.com>
Sun, 26 Jul 2015 05:17:43 +0000 (22:17 -0700)
committerJared Roesch <roeschinc@gmail.com>
Tue, 4 Aug 2015 23:05:07 +0000 (16:05 -0700)
Test case from here: https://github.com/freebroccolo/rust/commit/9e93fef3c0e61836a8b56f727eb7a2e94bb4ca09

src/librustc_typeck/astconv.rs
src/libsyntax/ext/base.rs
src/test/run-pass/type-macros.rs [new file with mode: 0644]

index cfd681e51f4e1c18569468f0f40f41a45e44b2ed..99f375c32868e295398c9b453ebc7b507ae569fe 100644 (file)
@@ -1663,7 +1663,7 @@ pub fn ast_ty_to_ty<'tcx>(this: &AstConv<'tcx>,
             this.ty_infer(None, None, None, ast_ty.span)
         }
         ast::TyMac(_) => {
-            tcx.sess.span_bug(m.span, "unexpanded type macro found conversion")
+            tcx.sess.span_bug(ast_ty.span, "unexpanded type macro found conversion")
         }
     };
 
index 28c7ead20bcaef7adac840bf867980e5a3c84d7f..1b5dcf64833466c6e8230400a348498aae54330d 100644 (file)
@@ -416,19 +416,22 @@ pub fn raw_pat(sp: Span) -> ast::Pat {
 
     pub fn raw_ty(sp: Span) -> P<ast::Ty> {
         P(ast::Ty {
-            id: ast:DUMMY_NODE_ID,
+            id: ast::DUMMY_NODE_ID,
             node: ast::TyInfer,
             span: sp
         })
+    }
 }
 
 impl MacResult for DummyResult {
     fn make_expr(self: Box<DummyResult>) -> Option<P<ast::Expr>> {
         Some(DummyResult::raw_expr(self.span))
     }
+
     fn make_pat(self: Box<DummyResult>) -> Option<P<ast::Pat>> {
         Some(P(DummyResult::raw_pat(self.span)))
     }
+
     fn make_items(self: Box<DummyResult>) -> Option<SmallVector<P<ast::Item>>> {
         // this code needs a comment... why not always just return the Some() ?
         if self.expr_only {
@@ -437,6 +440,7 @@ fn make_items(self: Box<DummyResult>) -> Option<SmallVector<P<ast::Item>>> {
             Some(SmallVector::zero())
         }
     }
+
     fn make_impl_items(self: Box<DummyResult>) -> Option<SmallVector<P<ast::ImplItem>>> {
         if self.expr_only {
             None
@@ -444,6 +448,7 @@ fn make_impl_items(self: Box<DummyResult>) -> Option<SmallVector<P<ast::ImplItem
             Some(SmallVector::zero())
         }
     }
+
     fn make_stmts(self: Box<DummyResult>) -> Option<SmallVector<P<ast::Stmt>>> {
         Some(SmallVector::one(P(
             codemap::respan(self.span,
diff --git a/src/test/run-pass/type-macros.rs b/src/test/run-pass/type-macros.rs
new file mode 100644 (file)
index 0000000..3ca3be8
--- /dev/null
@@ -0,0 +1,82 @@
+// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use std::ops::*;
+
+#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
+struct Nil; // empty HList
+#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
+struct Cons<H, T: HList>(H, T); // cons cell of HList
+
+// trait to classify valid HLists
+trait HList {}
+impl HList for Nil {}
+impl<H, T: HList> HList for Cons<H, T> {}
+
+// term-level macro for HLists
+macro_rules! hlist {
+        {} => { Nil };
+            { $head:expr } => { Cons($head, Nil) };
+                { $head:expr, $($tail:expr),* } => { Cons($head, hlist!($($tail),*)) };
+}
+
+// type-level macro for HLists
+macro_rules! HList {
+        {} => { Nil };
+            { $head:ty } => { Cons<$head, Nil> };
+                { $head:ty, $($tail:ty),* } => { Cons<$head, HList!($($tail),*)> };
+}
+
+// nil case for HList append
+impl<Ys: HList> Add<Ys> for Nil {
+        type Output = Ys;
+
+            fn add(self, rhs: Ys) -> Ys {
+                        rhs
+                                }
+}
+
+// cons case for HList append
+impl<Rec: HList + Sized, X, Xs: HList, Ys: HList> Add<Ys> for Cons<X, Xs> where
+    Xs: Add<Ys, Output = Rec>,
+{
+        type Output = Cons<X, Rec>;
+
+            fn add(self, rhs: Ys) -> Cons<X, Rec> {
+                        Cons(self.0, self.1 + rhs)
+                                }
+}
+
+// type macro Expr allows us to expand the + operator appropriately
+macro_rules! Expr {
+        { ( $($LHS:tt)+ ) } => { Expr!($($LHS)+) };
+            { HList ! [ $($LHS:tt)* ] + $($RHS:tt)+ } => { <Expr!(HList![$($LHS)*]) as Add<Expr!($($RHS)+)>>::Output };
+                { $LHS:tt + $($RHS:tt)+ } => { <Expr!($LHS) as Add<Expr!($($RHS)+)>>::Output };
+                    { $LHS:ty } => { $LHS };
+}
+
+// test demonstrating term level `xs + ys` and type level `Expr!(Xs + Ys)`
+fn main() {
+    fn aux<Xs: HList, Ys: HList>(xs: Xs, ys: Ys) -> Expr!(Xs + Ys)
+        where Xs: Add<Ys> {
+            xs + ys
+        }
+
+    let xs: HList![&str, bool, Vec<u64>] = hlist!["foo", false, vec![]];
+    let ys: HList![u64, [u8; 3], ()] = hlist![0, [0, 1, 2], ()];
+
+    // demonstrate recursive expansion of Expr!
+    let zs: Expr!((HList![&str] + HList![bool] + HList![Vec<u64>]) +
+                  (HList![u64] + HList![[u8; 3], ()]) +
+                  HList![])
+        = aux(xs, ys);
+    assert_eq!(zs, hlist!["foo", false, vec![], 0, [0, 1, 2], ()])
+}
+