1 // Copyright 2013-2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
11 //! Some code that abstracts away much of the boilerplate of writing
12 //! `derive` instances for traits. Among other things it manages getting
13 //! access to the fields of the 4 different sorts of structs and enum
14 //! variants, as well as creating the method and impl ast instances.
16 //! Supported features (fairly exhaustive):
18 //! - Methods taking any number of parameters of any type, and returning
19 //! any type, other than vectors, bottom and closures.
20 //! - Generating `impl`s for types with type parameters and lifetimes
21 //! (e.g. `Option<T>`), the parameters are automatically given the
22 //! current trait as a bound. (This includes separate type parameters
23 //! and lifetimes for methods.)
24 //! - Additional bounds on the type parameters (`TraitDef.additional_bounds`)
26 //! The most important thing for implementers is the `Substructure` and
27 //! `SubstructureFields` objects. The latter groups 5 possibilities of the
30 //! - `Struct`, when `Self` is a struct (including tuple structs, e.g
31 //! `struct T(i32, char)`).
32 //! - `EnumMatching`, when `Self` is an enum and all the arguments are the
33 //! same variant of the enum (e.g. `Some(1)`, `Some(3)` and `Some(4)`)
34 //! - `EnumNonMatchingCollapsed` when `Self` is an enum and the arguments
35 //! are not the same variant (e.g. `None`, `Some(1)` and `None`).
36 //! - `StaticEnum` and `StaticStruct` for static methods, where the type
37 //! being derived upon is either an enum or struct respectively. (Any
38 //! argument with type Self is just grouped among the non-self
41 //! In the first two cases, the values from the corresponding fields in
42 //! all the arguments are grouped together. For `EnumNonMatchingCollapsed`
43 //! this isn't possible (different variants have different fields), so the
44 //! fields are inaccessible. (Previous versions of the deriving infrastructure
45 //! had a way to expand into code that could access them, at the cost of
46 //! generating exponential amounts of code; see issue #15375). There are no
47 //! fields with values in the static cases, so these are treated entirely
50 //! The non-static cases have `Option<ident>` in several places associated
51 //! with field `expr`s. This represents the name of the field it is
52 //! associated with. It is only not `None` when the associated field has
53 //! an identifier in the source code. For example, the `x`s in the
57 //! struct A { x : i32 }
67 //! The `i32`s in `B` and `C0` don't have an identifier, so the
68 //! `Option<ident>`s would be `None` for them.
70 //! In the static cases, the structure is summarised, either into the just
71 //! spans of the fields or a list of spans and the field idents (for tuple
72 //! structs and record structs, respectively), or a list of these, for
73 //! enums (one for each variant). For empty struct and empty enum
74 //! variants, it is represented as a count of 0.
76 //! # "`cs`" functions
78 //! The `cs_...` functions ("combine substructure) are designed to
79 //! make life easier by providing some pre-made recipes for common
80 //! threads; mostly calling the function being derived on all the
81 //! arguments and then combining them back together in some way (or
82 //! letting the user chose that). They are not meant to be the only
83 //! way to handle the structures that this code creates.
87 //! The following simplified `PartialEq` is used for in-code examples:
91 //! fn eq(&self, other: &Self);
93 //! impl PartialEq for i32 {
94 //! fn eq(&self, other: &i32) -> bool {
100 //! Some examples of the values of `SubstructureFields` follow, using the
101 //! above `PartialEq`, `A`, `B` and `C`.
105 //! When generating the `expr` for the `A` impl, the `SubstructureFields` is
108 //! Struct(vec![FieldInfo {
109 //! span: <span of x>
110 //! name: Some(<ident of x>),
111 //! self_: <expr for &self.x>,
112 //! other: vec![<expr for &other.x]
116 //! For the `B` impl, called with `B(a)` and `B(b)`,
119 //! Struct(vec![FieldInfo {
120 //! span: <span of `i32`>,
122 //! self_: <expr for &a>
123 //! other: vec![<expr for &b>]
129 //! When generating the `expr` for a call with `self == C0(a)` and `other
130 //! == C0(b)`, the SubstructureFields is
133 //! EnumMatching(0, <ast::Variant for C0>,
135 //! span: <span of i32>
137 //! self_: <expr for &a>,
138 //! other: vec![<expr for &b>]
142 //! For `C1 {x}` and `C1 {x}`,
145 //! EnumMatching(1, <ast::Variant for C1>,
147 //! span: <span of x>
148 //! name: Some(<ident of x>),
149 //! self_: <expr for &self.x>,
150 //! other: vec![<expr for &other.x>]
154 //! For `C0(a)` and `C1 {x}` ,
157 //! EnumNonMatchingCollapsed(
158 //! vec![<ident of self>, <ident of __arg_1>],
159 //! &[<ast::Variant for C0>, <ast::Variant for C1>],
160 //! &[<ident for self index value>, <ident of __arg_1 index value>])
163 //! It is the same for when the arguments are flipped to `C1 {x}` and
164 //! `C0(a)`; the only difference is what the values of the identifiers
165 //! <ident for self index value> and <ident of __arg_1 index value> will
166 //! be in the generated code.
168 //! `EnumNonMatchingCollapsed` deliberately provides far less information
169 //! than is generally available for a given pair of variants; see #15375
174 //! A static method on the types above would result in,
177 //! StaticStruct(<ast::VariantData of A>, Named(vec![(<ident of x>, <span of x>)]))
179 //! StaticStruct(<ast::VariantData of B>, Unnamed(vec![<span of x>]))
181 //! StaticEnum(<ast::EnumDef of C>,
182 //! vec![(<ident of C0>, <span of C0>, Unnamed(vec![<span of i32>])),
183 //! (<ident of C1>, <span of C1>, Named(vec![(<ident of x>, <span of x>)]))])
186 pub use self::StaticFields::*;
187 pub use self::SubstructureFields::*;
188 use self::StructType::*;
190 use std::cell::RefCell;
191 use std::collections::HashSet;
197 use ast::{EnumDef, Expr, Ident, Generics, VariantData};
200 use attr::AttrMetaMethods;
201 use ext::base::{ExtCtxt, Annotatable};
202 use ext::build::AstBuilder;
203 use codemap::{self, DUMMY_SP};
205 use diagnostic::SpanHandler;
207 use owned_slice::OwnedSlice;
208 use parse::token::{intern, InternedString};
209 use parse::token::special_idents;
212 use self::ty::{LifetimeBounds, Path, Ptr, PtrTy, Self_, Ty};
216 pub struct TraitDef<'a> {
217 /// The span for the current #[derive(Foo)] header.
220 pub attributes: Vec<ast::Attribute>,
222 /// Path of the trait, including any type parameters
225 /// Additional bounds required of any type parameters of the type,
226 /// other than the current trait
227 pub additional_bounds: Vec<Ty<'a>>,
229 /// Any extra lifetimes and/or bounds, e.g. `D: serialize::Decoder`
230 pub generics: LifetimeBounds<'a>,
232 /// Is it an `unsafe` trait?
235 pub methods: Vec<MethodDef<'a>>,
237 pub associated_types: Vec<(ast::Ident, Ty<'a>)>,
241 pub struct MethodDef<'a> {
242 /// name of the method
244 /// List of generics, e.g. `R: rand::Rng`
245 pub generics: LifetimeBounds<'a>,
247 /// Whether there is a self argument (outer Option) i.e. whether
248 /// this is a static function, and whether it is a pointer (inner
250 pub explicit_self: Option<Option<PtrTy<'a>>>,
252 /// Arguments other than the self argument
253 pub args: Vec<Ty<'a>>,
258 pub attributes: Vec<ast::Attribute>,
260 // Is it an `unsafe fn`?
263 pub combine_substructure: RefCell<CombineSubstructureFunc<'a>>,
266 /// All the data about the data structure/method being derived upon.
267 pub struct Substructure<'a> {
269 pub type_ident: Ident,
270 /// ident of the method
271 pub method_ident: Ident,
272 /// dereferenced access to any `Self_` or `Ptr(Self_, _)` arguments
273 pub self_args: &'a [P<Expr>],
274 /// verbatim access to any other arguments
275 pub nonself_args: &'a [P<Expr>],
276 pub fields: &'a SubstructureFields<'a>
279 /// Summary of the relevant parts of a struct/enum field.
280 pub struct FieldInfo<'a> {
282 /// None for tuple structs/normal enum variants, Some for normal
283 /// structs/struct enum variants.
284 pub name: Option<Ident>,
285 /// The expression corresponding to this field of `self`
286 /// (specifically, a reference to it).
288 /// The expressions corresponding to references to this field in
289 /// the other `Self` arguments.
290 pub other: Vec<P<Expr>>,
291 /// The attributes on the field
292 pub attrs: &'a [ast::Attribute],
295 /// Fields for a static method
296 pub enum StaticFields {
297 /// Tuple structs/enum variants like this.
299 /// Normal structs/struct variants.
300 Named(Vec<(Ident, Span)>),
303 /// A summary of the possible sets of fields.
304 pub enum SubstructureFields<'a> {
305 Struct(Vec<FieldInfo<'a>>),
306 /// Matching variants of the enum: variant index, ast::Variant,
307 /// fields: the field name is only non-`None` in the case of a struct
309 EnumMatching(usize, &'a ast::Variant, Vec<FieldInfo<'a>>),
311 /// Non-matching variants of the enum, but with all state hidden from
312 /// the consequent code. The first component holds `Ident`s for all of
313 /// the `Self` arguments; the second component is a slice of all of the
314 /// variants for the enum itself, and the third component is a list of
315 /// `Ident`s bound to the variant index values for each of the actual
316 /// input `Self` arguments.
317 EnumNonMatchingCollapsed(Vec<Ident>, &'a [P<ast::Variant>], &'a [Ident]),
319 /// A static method where `Self` is a struct.
320 StaticStruct(&'a ast::VariantData, StaticFields),
321 /// A static method where `Self` is an enum.
322 StaticEnum(&'a ast::EnumDef, Vec<(Ident, Span, StaticFields)>),
327 /// Combine the values of all the fields together. The last argument is
328 /// all the fields of all the structures.
329 pub type CombineSubstructureFunc<'a> =
330 Box<FnMut(&mut ExtCtxt, Span, &Substructure) -> P<Expr> + 'a>;
332 /// Deal with non-matching enum variants. The tuple is a list of
333 /// identifiers (one for each `Self` argument, which could be any of the
334 /// variants since they have been collapsed together) and the identifiers
335 /// holding the variant index value for each of the `Self` arguments. The
336 /// last argument is all the non-`Self` args of the method being derived.
337 pub type EnumNonMatchCollapsedFunc<'a> =
338 Box<FnMut(&mut ExtCtxt, Span, (&[Ident], &[Ident]), &[P<Expr>]) -> P<Expr> + 'a>;
340 pub fn combine_substructure<'a>(f: CombineSubstructureFunc<'a>)
341 -> RefCell<CombineSubstructureFunc<'a>> {
345 /// This method helps to extract all the type parameters referenced from a
346 /// type. For a type parameter `<T>`, it looks for either a `TyPath` that
347 /// is not global and starts with `T`, or a `TyQPath`.
348 fn find_type_parameters(ty: &ast::Ty, ty_param_names: &[ast::Name]) -> Vec<P<ast::Ty>> {
352 ty_param_names: &'a [ast::Name],
353 types: Vec<P<ast::Ty>>,
356 impl<'a> visit::Visitor<'a> for Visitor<'a> {
357 fn visit_ty(&mut self, ty: &'a ast::Ty) {
359 ast::TyPath(_, ref path) if !path.global => {
360 match path.segments.first() {
362 if self.ty_param_names.contains(&segment.identifier.name) {
363 self.types.push(P(ty.clone()));
372 visit::walk_ty(self, ty)
376 let mut visitor = Visitor {
377 ty_param_names: ty_param_names,
381 visit::Visitor::visit_ty(&mut visitor, ty);
386 impl<'a> TraitDef<'a> {
389 mitem: &ast::MetaItem,
390 item: &'a Annotatable,
391 push: &mut FnMut(Annotatable))
394 Annotatable::Item(ref item) => {
395 let newitem = match item.node {
396 ast::ItemStruct(ref struct_def, ref generics) => {
397 self.expand_struct_def(cx,
402 ast::ItemEnum(ref enum_def, ref generics) => {
403 self.expand_enum_def(cx,
410 cx.span_err(mitem.span,
411 "`derive` may only be applied to structs and enums");
415 // Keep the lint attributes of the previous item to control how the
416 // generated implementations are linted
417 let mut attrs = newitem.attrs.clone();
418 attrs.extend(item.attrs.iter().filter(|a| {
419 match &a.name()[..] {
420 "allow" | "warn" | "deny" | "forbid" | "stable" | "unstable" => true,
424 push(Annotatable::Item(P(ast::Item {
430 cx.span_err(mitem.span, "`derive` may only be applied to structs and enums");
435 /// Given that we are deriving a trait `DerivedTrait` for a type like:
438 /// struct Struct<'a, ..., 'z, A, B: DeclaredTrait, C, ..., Z> where C: WhereTrait {
441 /// b1: <B as DeclaredTrait>::Item,
442 /// c1: <C as WhereTrait>::Item,
443 /// c2: Option<<C as WhereTrait>::Item>,
448 /// create an impl like:
451 /// impl<'a, ..., 'z, A, B: DeclaredTrait, C, ... Z> where
453 /// A: DerivedTrait + B1 + ... + BN,
454 /// B: DerivedTrait + B1 + ... + BN,
455 /// C: DerivedTrait + B1 + ... + BN,
456 /// B::Item: DerivedTrait + B1 + ... + BN,
457 /// <C as WhereTrait>::Item: DerivedTrait + B1 + ... + BN,
464 /// where B1, ..., BN are the bounds given by `bounds_paths`.'. Z is a phantom type, and
465 /// therefore does not get bound by the derived trait.
466 fn create_derived_impl(&self,
470 field_tys: Vec<P<ast::Ty>>,
471 methods: Vec<P<ast::ImplItem>>) -> P<ast::Item> {
472 let trait_path = self.path.to_path(cx, self.span, type_ident, generics);
474 // Transform associated types from `deriving::ty::Ty` into `ast::ImplItem`
475 let associated_types = self.associated_types.iter().map(|&(ident, ref type_def)| {
477 id: ast::DUMMY_NODE_ID,
482 node: ast::TypeImplItem(type_def.to_ty(cx,
490 let Generics { mut lifetimes, ty_params, mut where_clause } =
491 self.generics.to_generics(cx, self.span, type_ident, generics);
492 let mut ty_params = ty_params.into_vec();
494 // Copy the lifetimes
495 lifetimes.extend(generics.lifetimes.iter().cloned());
497 // Create the type parameters.
498 ty_params.extend(generics.ty_params.iter().map(|ty_param| {
499 // I don't think this can be moved out of the loop, since
500 // a TyParamBound requires an ast id
501 let mut bounds: Vec<_> =
502 // extra restrictions on the generics parameters to the type being derived upon
503 self.additional_bounds.iter().map(|p| {
504 cx.typarambound(p.to_path(cx, self.span,
505 type_ident, generics))
508 // require the current trait
509 bounds.push(cx.typarambound(trait_path.clone()));
511 // also add in any bounds from the declaration
512 for declared_bound in ty_param.bounds.iter() {
513 bounds.push((*declared_bound).clone());
516 cx.typaram(self.span,
518 OwnedSlice::from_vec(bounds),
522 // and similarly for where clauses
523 where_clause.predicates.extend(generics.where_clause.predicates.iter().map(|clause| {
525 ast::WherePredicate::BoundPredicate(ref wb) => {
526 ast::WherePredicate::BoundPredicate(ast::WhereBoundPredicate {
528 bound_lifetimes: wb.bound_lifetimes.clone(),
529 bounded_ty: wb.bounded_ty.clone(),
530 bounds: OwnedSlice::from_vec(wb.bounds.iter().cloned().collect())
533 ast::WherePredicate::RegionPredicate(ref rb) => {
534 ast::WherePredicate::RegionPredicate(ast::WhereRegionPredicate {
536 lifetime: rb.lifetime,
537 bounds: rb.bounds.iter().cloned().collect()
540 ast::WherePredicate::EqPredicate(ref we) => {
541 ast::WherePredicate::EqPredicate(ast::WhereEqPredicate {
542 id: ast::DUMMY_NODE_ID,
544 path: we.path.clone(),
551 if !ty_params.is_empty() {
552 let ty_param_names: Vec<ast::Name> = ty_params.iter()
553 .map(|ty_param| ty_param.ident.name)
556 let mut processed_field_types = HashSet::new();
557 for field_ty in field_tys {
558 let tys = find_type_parameters(&*field_ty, &ty_param_names);
561 // if we have already handled this type, skip it
562 if let ast::TyPath(_, ref p) = ty.node {
563 if p.segments.len() == 1
564 && ty_param_names.contains(&p.segments[0].identifier.name)
565 || processed_field_types.contains(&p.segments) {
568 processed_field_types.insert(p.segments.clone());
570 let mut bounds: Vec<_> = self.additional_bounds.iter().map(|p| {
571 cx.typarambound(p.to_path(cx, self.span, type_ident, generics))
574 // require the current trait
575 bounds.push(cx.typarambound(trait_path.clone()));
577 let predicate = ast::WhereBoundPredicate {
579 bound_lifetimes: vec![],
581 bounds: OwnedSlice::from_vec(bounds),
584 let predicate = ast::WherePredicate::BoundPredicate(predicate);
585 where_clause.predicates.push(predicate);
590 let trait_generics = Generics {
591 lifetimes: lifetimes,
592 ty_params: OwnedSlice::from_vec(ty_params),
593 where_clause: where_clause
596 // Create the reference to the trait.
597 let trait_ref = cx.trait_ref(trait_path);
599 // Create the type parameters on the `self` path.
600 let self_ty_params = generics.ty_params.map(|ty_param| {
601 cx.ty_ident(self.span, ty_param.ident)
604 let self_lifetimes: Vec<ast::Lifetime> =
607 .map(|ld| ld.lifetime)
610 // Create the type of `self`.
611 let self_type = cx.ty_path(
612 cx.path_all(self.span, false, vec!( type_ident ), self_lifetimes,
613 self_ty_params.into_vec(), Vec::new()));
615 let attr = cx.attribute(
617 cx.meta_word(self.span,
618 InternedString::new("automatically_derived")));
619 // Just mark it now since we know that it'll end up used downstream
620 attr::mark_used(&attr);
621 let opt_trait_ref = Some(trait_ref);
622 let ident = ast_util::impl_pretty_name(&opt_trait_ref, Some(&*self_type));
623 let unused_qual = cx.attribute(
625 cx.meta_list(self.span,
626 InternedString::new("allow"),
627 vec![cx.meta_word(self.span,
628 InternedString::new("unused_qualifications"))]));
629 let mut a = vec![attr, unused_qual];
630 a.extend(self.attributes.iter().cloned());
632 let unsafety = if self.is_unsafe {
633 ast::Unsafety::Unsafe
635 ast::Unsafety::Normal
642 ast::ItemImpl(unsafety,
643 ast::ImplPolarity::Positive,
647 methods.into_iter().chain(associated_types).collect()))
650 fn expand_struct_def(&self,
652 struct_def: &'a VariantData,
654 generics: &Generics) -> P<ast::Item> {
655 let field_tys: Vec<P<ast::Ty>> = struct_def.fields()
656 .map(|field| field.node.ty.clone())
659 let methods = self.methods.iter().map(|method_def| {
660 let (explicit_self, self_args, nonself_args, tys) =
661 method_def.split_self_nonself_args(
662 cx, self, type_ident, generics);
664 let body = if method_def.is_static() {
665 method_def.expand_static_struct_method_body(
673 method_def.expand_struct_method_body(cx,
681 method_def.create_method(cx,
691 self.create_derived_impl(cx, type_ident, generics, field_tys, methods)
694 fn expand_enum_def(&self,
696 enum_def: &'a EnumDef,
697 type_attrs: &[ast::Attribute],
699 generics: &Generics) -> P<ast::Item> {
700 let mut field_tys = Vec::new();
702 for variant in &enum_def.variants {
703 field_tys.extend(variant.node.data.fields()
704 .map(|field| field.node.ty.clone()));
707 let methods = self.methods.iter().map(|method_def| {
708 let (explicit_self, self_args, nonself_args, tys) =
709 method_def.split_self_nonself_args(cx, self,
710 type_ident, generics);
712 let body = if method_def.is_static() {
713 method_def.expand_static_enum_method_body(
721 method_def.expand_enum_method_body(cx,
730 method_def.create_method(cx,
740 self.create_derived_impl(cx, type_ident, generics, field_tys, methods)
744 fn find_repr_type_name(diagnostic: &SpanHandler,
745 type_attrs: &[ast::Attribute]) -> &'static str {
746 let mut repr_type_name = "i32";
747 for a in type_attrs {
748 for r in &attr::find_repr_attrs(diagnostic, a) {
749 repr_type_name = match *r {
750 attr::ReprAny | attr::ReprPacked | attr::ReprSimd => continue,
751 attr::ReprExtern => "i32",
753 attr::ReprInt(_, attr::SignedInt(ast::TyIs)) => "isize",
754 attr::ReprInt(_, attr::SignedInt(ast::TyI8)) => "i8",
755 attr::ReprInt(_, attr::SignedInt(ast::TyI16)) => "i16",
756 attr::ReprInt(_, attr::SignedInt(ast::TyI32)) => "i32",
757 attr::ReprInt(_, attr::SignedInt(ast::TyI64)) => "i64",
759 attr::ReprInt(_, attr::UnsignedInt(ast::TyUs)) => "usize",
760 attr::ReprInt(_, attr::UnsignedInt(ast::TyU8)) => "u8",
761 attr::ReprInt(_, attr::UnsignedInt(ast::TyU16)) => "u16",
762 attr::ReprInt(_, attr::UnsignedInt(ast::TyU32)) => "u32",
763 attr::ReprInt(_, attr::UnsignedInt(ast::TyU64)) => "u64",
770 impl<'a> MethodDef<'a> {
771 fn call_substructure_method(&self,
775 self_args: &[P<Expr>],
776 nonself_args: &[P<Expr>],
777 fields: &SubstructureFields)
779 let substructure = Substructure {
780 type_ident: type_ident,
781 method_ident: cx.ident_of(self.name),
782 self_args: self_args,
783 nonself_args: nonself_args,
786 let mut f = self.combine_substructure.borrow_mut();
787 let f: &mut CombineSubstructureFunc = &mut *f;
788 f(cx, trait_.span, &substructure)
797 self.ret_ty.to_ty(cx, trait_.span, type_ident, generics)
800 fn is_static(&self) -> bool {
801 self.explicit_self.is_none()
804 fn split_self_nonself_args(&self,
809 -> (ast::ExplicitSelf, Vec<P<Expr>>, Vec<P<Expr>>, Vec<(Ident, P<ast::Ty>)>) {
811 let mut self_args = Vec::new();
812 let mut nonself_args = Vec::new();
813 let mut arg_tys = Vec::new();
814 let mut nonstatic = false;
816 let ast_explicit_self = match self.explicit_self {
817 Some(ref self_ptr) => {
818 let (self_expr, explicit_self) =
819 ty::get_explicit_self(cx, trait_.span, self_ptr);
821 self_args.push(self_expr);
826 None => codemap::respan(trait_.span, ast::SelfStatic),
829 for (i, ty) in self.args.iter().enumerate() {
830 let ast_ty = ty.to_ty(cx, trait_.span, type_ident, generics);
831 let ident = cx.ident_of(&format!("__arg_{}", i));
832 arg_tys.push((ident, ast_ty));
834 let arg_expr = cx.expr_ident(trait_.span, ident);
837 // for static methods, just treat any Self
838 // arguments as a normal arg
839 Self_ if nonstatic => {
840 self_args.push(arg_expr);
842 Ptr(ref ty, _) if **ty == Self_ && nonstatic => {
843 self_args.push(cx.expr_deref(trait_.span, arg_expr))
846 nonself_args.push(arg_expr);
851 (ast_explicit_self, self_args, nonself_args, arg_tys)
854 fn create_method(&self,
860 explicit_self: ast::ExplicitSelf,
861 arg_types: Vec<(Ident, P<ast::Ty>)> ,
862 body: P<Expr>) -> P<ast::ImplItem> {
863 // create the generics that aren't for Self
864 let fn_generics = self.generics.to_generics(cx, trait_.span, type_ident, generics);
866 let self_arg = match explicit_self.node {
867 ast::SelfStatic => None,
868 // creating fresh self id
869 _ => Some(ast::Arg::new_self(trait_.span, ast::MutImmutable, special_idents::self_))
872 let args = arg_types.into_iter().map(|(name, ty)| {
873 cx.arg(trait_.span, name, ty)
875 self_arg.into_iter().chain(args).collect()
878 let ret_type = self.get_ret_ty(cx, trait_, generics, type_ident);
880 let method_ident = cx.ident_of(self.name);
881 let fn_decl = cx.fn_decl(args, ret_type);
882 let body_block = cx.block_expr(body);
884 let unsafety = if self.is_unsafe {
885 ast::Unsafety::Unsafe
887 ast::Unsafety::Normal
890 // Create the method.
892 id: ast::DUMMY_NODE_ID,
893 attrs: self.attributes.clone(),
897 node: ast::MethodImplItem(ast::MethodSig {
898 generics: fn_generics,
900 explicit_self: explicit_self,
902 constness: ast::Constness::NotConst,
909 /// #[derive(PartialEq)]
910 /// struct A { x: i32, y: i32 }
912 /// // equivalent to:
913 /// impl PartialEq for A {
914 /// fn eq(&self, __arg_1: &A) -> bool {
916 /// A {x: ref __self_0_0, y: ref __self_0_1} => {
918 /// A {x: ref __self_1_0, y: ref __self_1_1} => {
919 /// __self_0_0.eq(__self_1_0) && __self_0_1.eq(__self_1_1)
927 fn expand_struct_method_body<'b>(&self,
929 trait_: &TraitDef<'b>,
930 struct_def: &'b VariantData,
932 self_args: &[P<Expr>],
933 nonself_args: &[P<Expr>])
936 let mut raw_fields = Vec::new(); // Vec<[fields of self],
937 // [fields of next Self arg], [etc]>
938 let mut patterns = Vec::new();
939 for i in 0..self_args.len() {
940 let struct_path= cx.path(DUMMY_SP, vec!( type_ident ));
941 let (pat, ident_expr) =
942 trait_.create_struct_pattern(cx,
945 &format!("__self_{}",
949 raw_fields.push(ident_expr);
952 // transpose raw_fields
953 let fields = if !raw_fields.is_empty() {
954 let mut raw_fields = raw_fields.into_iter().map(|v| v.into_iter());
955 let first_field = raw_fields.next().unwrap();
956 let mut other_fields: Vec<vec::IntoIter<_>>
957 = raw_fields.collect();
958 first_field.map(|(span, opt_id, field, attrs)| {
963 other: other_fields.iter_mut().map(|l| {
964 match l.next().unwrap() {
972 cx.span_bug(trait_.span,
973 "no self arguments to non-static method in generic \
977 // body of the inner most destructuring match
978 let mut body = self.call_substructure_method(
986 // make a series of nested matches, to destructure the
987 // structs. This is actually right-to-left, but it shouldn't
989 for (arg_expr, pat) in self_args.iter().zip(patterns) {
990 body = cx.expr_match(trait_.span, arg_expr.clone(),
991 vec!( cx.arm(trait_.span, vec!(pat.clone()), body) ))
996 fn expand_static_struct_method_body(&self,
999 struct_def: &VariantData,
1001 self_args: &[P<Expr>],
1002 nonself_args: &[P<Expr>])
1004 let summary = trait_.summarise_struct(cx, struct_def);
1006 self.call_substructure_method(cx,
1009 self_args, nonself_args,
1010 &StaticStruct(struct_def, summary))
1014 /// #[derive(PartialEq)]
1020 /// // is equivalent to
1022 /// impl PartialEq for A {
1023 /// fn eq(&self, __arg_1: &A) -> ::bool {
1024 /// match (&*self, &*__arg_1) {
1025 /// (&A1, &A1) => true,
1026 /// (&A2(ref __self_0),
1027 /// &A2(ref __arg_1_0)) => (*__self_0).eq(&(*__arg_1_0)),
1029 /// let __self_vi = match *self { A1(..) => 0, A2(..) => 1 };
1030 /// let __arg_1_vi = match *__arg_1 { A1(..) => 0, A2(..) => 1 };
1038 /// (Of course `__self_vi` and `__arg_1_vi` are unused for
1039 /// `PartialEq`, and those subcomputations will hopefully be removed
1040 /// as their results are unused. The point of `__self_vi` and
1041 /// `__arg_1_vi` is for `PartialOrd`; see #15503.)
1042 fn expand_enum_method_body<'b>(&self,
1044 trait_: &TraitDef<'b>,
1045 enum_def: &'b EnumDef,
1046 type_attrs: &[ast::Attribute],
1048 self_args: Vec<P<Expr>>,
1049 nonself_args: &[P<Expr>])
1051 self.build_enum_match_tuple(
1052 cx, trait_, enum_def, type_attrs, type_ident, self_args, nonself_args)
1056 /// Creates a match for a tuple of all `self_args`, where either all
1057 /// variants match, or it falls into a catch-all for when one variant
1060 /// There are N + 1 cases because is a case for each of the N
1061 /// variants where all of the variants match, and one catch-all for
1062 /// when one does not match.
1064 /// As an optimization we generate code which checks whether all variants
1065 /// match first which makes llvm see that C-like enums can be compiled into
1066 /// a simple equality check (for PartialEq).
1068 /// The catch-all handler is provided access the variant index values
1069 /// for each of the self-args, carried in precomputed variables.
1072 /// let __self0_vi = unsafe {
1073 /// std::intrinsics::discriminant_value(&self) } as i32;
1074 /// let __self1_vi = unsafe {
1075 /// std::intrinsics::discriminant_value(&__arg1) } as i32;
1076 /// let __self2_vi = unsafe {
1077 /// std::intrinsics::discriminant_value(&__arg2) } as i32;
1079 /// if __self0_vi == __self1_vi && __self0_vi == __self2_vi && ... {
1081 /// (Variant1, Variant1, ...) => Body1
1082 /// (Variant2, Variant2, ...) => Body2,
1084 /// _ => ::core::intrinsics::unreachable()
1088 /// ... // catch-all remainder can inspect above variant index values.
1091 fn build_enum_match_tuple<'b>(
1094 trait_: &TraitDef<'b>,
1095 enum_def: &'b EnumDef,
1096 type_attrs: &[ast::Attribute],
1098 self_args: Vec<P<Expr>>,
1099 nonself_args: &[P<Expr>]) -> P<Expr> {
1101 let sp = trait_.span;
1102 let variants = &enum_def.variants;
1104 let self_arg_names = self_args.iter().enumerate()
1105 .map(|(arg_count, _self_arg)| {
1107 "__self".to_string()
1109 format!("__arg_{}", arg_count)
1112 .collect::<Vec<String>>();
1114 let self_arg_idents = self_arg_names.iter()
1115 .map(|name|cx.ident_of(&name[..]))
1116 .collect::<Vec<ast::Ident>>();
1118 // The `vi_idents` will be bound, solely in the catch-all, to
1119 // a series of let statements mapping each self_arg to an int
1120 // value corresponding to its discriminant.
1121 let vi_idents: Vec<ast::Ident> = self_arg_names.iter()
1122 .map(|name| { let vi_suffix = format!("{}_vi", &name[..]);
1123 cx.ident_of(&vi_suffix[..]) })
1124 .collect::<Vec<ast::Ident>>();
1126 // Builds, via callback to call_substructure_method, the
1127 // delegated expression that handles the catch-all case,
1128 // using `__variants_tuple` to drive logic if necessary.
1129 let catch_all_substructure = EnumNonMatchingCollapsed(
1130 self_arg_idents, &variants[..], &vi_idents[..]);
1132 // These arms are of the form:
1133 // (Variant1, Variant1, ...) => Body1
1134 // (Variant2, Variant2, ...) => Body2
1136 // where each tuple has length = self_args.len()
1137 let mut match_arms: Vec<ast::Arm> = variants.iter().enumerate()
1138 .map(|(index, variant)| {
1139 let mk_self_pat = |cx: &mut ExtCtxt, self_arg_name: &str| {
1140 let (p, idents) = trait_.create_enum_variant_pattern(cx, type_ident,
1144 (cx.pat(sp, ast::PatRegion(p, ast::MutImmutable)), idents)
1147 // A single arm has form (&VariantK, &VariantK, ...) => BodyK
1148 // (see "Final wrinkle" note below for why.)
1149 let mut subpats = Vec::with_capacity(self_arg_names.len());
1150 let mut self_pats_idents = Vec::with_capacity(self_arg_names.len() - 1);
1151 let first_self_pat_idents = {
1152 let (p, idents) = mk_self_pat(cx, &self_arg_names[0]);
1156 for self_arg_name in &self_arg_names[1..] {
1157 let (p, idents) = mk_self_pat(cx, &self_arg_name[..]);
1159 self_pats_idents.push(idents);
1162 // Here is the pat = `(&VariantK, &VariantK, ...)`
1163 let single_pat = cx.pat_tuple(sp, subpats);
1165 // For the BodyK, we need to delegate to our caller,
1166 // passing it an EnumMatching to indicate which case
1169 // All of the Self args have the same variant in these
1170 // cases. So we transpose the info in self_pats_idents
1171 // to gather the getter expressions together, in the
1172 // form that EnumMatching expects.
1174 // The transposition is driven by walking across the
1175 // arg fields of the variant for the first self pat.
1176 let field_tuples = first_self_pat_idents.into_iter().enumerate()
1177 // For each arg field of self, pull out its getter expr ...
1178 .map(|(field_index, (sp, opt_ident, self_getter_expr, attrs))| {
1179 // ... but FieldInfo also wants getter expr
1180 // for matching other arguments of Self type;
1181 // so walk across the *other* self_pats_idents
1182 // and pull out getter for same field in each
1183 // of them (using `field_index` tracked above).
1184 // That is the heart of the transposition.
1185 let others = self_pats_idents.iter().map(|fields| {
1186 let (_, _opt_ident, ref other_getter_expr, _) =
1187 fields[field_index];
1189 // All Self args have same variant, so
1190 // opt_idents are the same. (Assert
1191 // here to make it self-evident that
1192 // it is okay to ignore `_opt_ident`.)
1193 assert!(opt_ident == _opt_ident);
1195 other_getter_expr.clone()
1196 }).collect::<Vec<P<Expr>>>();
1198 FieldInfo { span: sp,
1200 self_: self_getter_expr,
1204 }).collect::<Vec<FieldInfo>>();
1206 // Now, for some given VariantK, we have built up
1207 // expressions for referencing every field of every
1208 // Self arg, assuming all are instances of VariantK.
1209 // Build up code associated with such a case.
1210 let substructure = EnumMatching(index,
1213 let arm_expr = self.call_substructure_method(
1214 cx, trait_, type_ident, &self_args[..], nonself_args,
1217 cx.arm(sp, vec![single_pat], arm_expr)
1219 // We will usually need the catch-all after matching the
1220 // tuples `(VariantK, VariantK, ...)` for each VariantK of the
1223 // * when there is only one Self arg, the arms above suffice
1224 // (and the deriving we call back into may not be prepared to
1225 // handle EnumNonMatchCollapsed), and,
1227 // * when the enum has only one variant, the single arm that
1228 // is already present always suffices.
1230 // * In either of the two cases above, if we *did* add a
1231 // catch-all `_` match, it would trigger the
1232 // unreachable-pattern error.
1234 if variants.len() > 1 && self_args.len() > 1 {
1235 // Build a series of let statements mapping each self_arg
1236 // to its discriminant value. If this is a C-style enum
1237 // with a specific repr type, then casts the values to
1238 // that type. Otherwise casts to `i32` (the default repr
1241 // i.e. for `enum E<T> { A, B(1), C(T, T) }`, and a deriving
1242 // with three Self args, builds three statements:
1245 // let __self0_vi = unsafe {
1246 // std::intrinsics::discriminant_value(&self) } as i32;
1247 // let __self1_vi = unsafe {
1248 // std::intrinsics::discriminant_value(&__arg1) } as i32;
1249 // let __self2_vi = unsafe {
1250 // std::intrinsics::discriminant_value(&__arg2) } as i32;
1252 let mut index_let_stmts: Vec<P<ast::Stmt>> = Vec::new();
1254 //We also build an expression which checks whether all discriminants are equal
1255 // discriminant_test = __self0_vi == __self1_vi && __self0_vi == __self2_vi && ...
1256 let mut discriminant_test = cx.expr_bool(sp, true);
1258 let target_type_name =
1259 find_repr_type_name(&cx.parse_sess.span_diagnostic, type_attrs);
1261 let mut first_ident = None;
1262 for (&ident, self_arg) in vi_idents.iter().zip(&self_args) {
1263 let path = cx.std_path(&["intrinsics", "discriminant_value"]);
1264 let call = cx.expr_call_global(
1265 sp, path, vec![cx.expr_addr_of(sp, self_arg.clone())]);
1266 let variant_value = cx.expr_block(P(ast::Block {
1269 id: ast::DUMMY_NODE_ID,
1270 rules: ast::UnsafeBlock(ast::CompilerGenerated),
1273 let target_ty = cx.ty_ident(sp, cx.ident_of(target_type_name));
1274 let variant_disr = cx.expr_cast(sp, variant_value, target_ty);
1275 let let_stmt = cx.stmt_let(sp, false, ident, variant_disr);
1276 index_let_stmts.push(let_stmt);
1280 let first_expr = cx.expr_ident(sp, first);
1281 let id = cx.expr_ident(sp, ident);
1282 let test = cx.expr_binary(sp, ast::BiEq, first_expr, id);
1283 discriminant_test = cx.expr_binary(sp, ast::BiAnd, discriminant_test, test)
1286 first_ident = Some(ident);
1291 let arm_expr = self.call_substructure_method(
1292 cx, trait_, type_ident, &self_args[..], nonself_args,
1293 &catch_all_substructure);
1295 //Since we know that all the arguments will match if we reach the match expression we
1296 //add the unreachable intrinsics as the result of the catch all which should help llvm
1298 let path = cx.std_path(&["intrinsics", "unreachable"]);
1299 let call = cx.expr_call_global(
1301 let unreachable = cx.expr_block(P(ast::Block {
1304 id: ast::DUMMY_NODE_ID,
1305 rules: ast::UnsafeBlock(ast::CompilerGenerated),
1307 match_arms.push(cx.arm(sp, vec![cx.pat_wild(sp)], unreachable));
1309 // Final wrinkle: the self_args are expressions that deref
1310 // down to desired l-values, but we cannot actually deref
1311 // them when they are fed as r-values into a tuple
1312 // expression; here add a layer of borrowing, turning
1313 // `(*self, *__arg_0, ...)` into `(&*self, &*__arg_0, ...)`.
1314 let borrowed_self_args = self_args.move_map(|self_arg| cx.expr_addr_of(sp, self_arg));
1315 let match_arg = cx.expr(sp, ast::ExprTup(borrowed_self_args));
1317 //Lastly we create an expression which branches on all discriminants being equal
1318 // if discriminant_test {
1320 // (Variant1, Variant1, ...) => Body1
1321 // (Variant2, Variant2, ...) => Body2,
1323 // _ => ::core::intrinsics::unreachable()
1327 // <delegated expression referring to __self0_vi, et al.>
1329 let all_match = cx.expr_match(sp, match_arg, match_arms);
1330 let arm_expr = cx.expr_if(sp, discriminant_test, all_match, Some(arm_expr));
1332 cx.block_all(sp, index_let_stmts, Some(arm_expr)))
1333 } else if variants.is_empty() {
1334 // As an additional wrinkle, For a zero-variant enum A,
1335 // currently the compiler
1336 // will accept `fn (a: &Self) { match *a { } }`
1337 // but rejects `fn (a: &Self) { match (&*a,) { } }`
1338 // as well as `fn (a: &Self) { match ( *a,) { } }`
1340 // This means that the strategy of building up a tuple of
1341 // all Self arguments fails when Self is a zero variant
1342 // enum: rustc rejects the expanded program, even though
1343 // the actual code tends to be impossible to execute (at
1344 // least safely), according to the type system.
1346 // The most expedient fix for this is to just let the
1347 // code fall through to the catch-all. But even this is
1348 // error-prone, since the catch-all as defined above would
1349 // generate code like this:
1351 // _ => { let __self0 = match *self { };
1352 // let __self1 = match *__arg_0 { };
1353 // <catch-all-expr> }
1355 // Which is yields bindings for variables which type
1356 // inference cannot resolve to unique types.
1358 // One option to the above might be to add explicit type
1359 // annotations. But the *only* reason to go down that path
1360 // would be to try to make the expanded output consistent
1361 // with the case when the number of enum variants >= 1.
1363 // That just isn't worth it. In fact, trying to generate
1364 // sensible code for *any* deriving on a zero-variant enum
1365 // does not make sense. But at the same time, for now, we
1366 // do not want to cause a compile failure just because the
1367 // user happened to attach a deriving to their
1368 // zero-variant enum.
1370 // Instead, just generate a failing expression for the
1371 // zero variant case, skipping matches and also skipping
1372 // delegating back to the end user code entirely.
1374 // (See also #4499 and #12609; note that some of the
1375 // discussions there influence what choice we make here;
1376 // e.g. if we feature-gate `match x { ... }` when x refers
1377 // to an uninhabited type (e.g. a zero-variant enum or a
1378 // type holding such an enum), but do not feature-gate
1379 // zero-variant enums themselves, then attempting to
1380 // derive Debug on such a type could here generate code
1381 // that needs the feature gate enabled.)
1383 cx.expr_unreachable(sp)
1387 // Final wrinkle: the self_args are expressions that deref
1388 // down to desired l-values, but we cannot actually deref
1389 // them when they are fed as r-values into a tuple
1390 // expression; here add a layer of borrowing, turning
1391 // `(*self, *__arg_0, ...)` into `(&*self, &*__arg_0, ...)`.
1392 let borrowed_self_args = self_args.move_map(|self_arg| cx.expr_addr_of(sp, self_arg));
1393 let match_arg = cx.expr(sp, ast::ExprTup(borrowed_self_args));
1394 cx.expr_match(sp, match_arg, match_arms)
1398 fn expand_static_enum_method_body(&self,
1403 self_args: &[P<Expr>],
1404 nonself_args: &[P<Expr>])
1406 let summary = enum_def.variants.iter().map(|v| {
1407 let ident = v.node.name;
1408 let summary = trait_.summarise_struct(cx, &v.node.data);
1409 (ident, v.span, summary)
1411 self.call_substructure_method(cx, trait_, type_ident,
1412 self_args, nonself_args,
1413 &StaticEnum(enum_def, summary))
1417 #[derive(PartialEq)] // dogfooding!
1419 Unknown, Record, Tuple
1422 // general helper methods.
1423 impl<'a> TraitDef<'a> {
1424 fn set_expn_info(&self,
1426 mut to_set: Span) -> Span {
1427 let trait_name = match self.path.path.last() {
1428 None => cx.span_bug(self.span, "trait with empty path in generic `derive`"),
1431 to_set.expn_id = cx.codemap().record_expansion(codemap::ExpnInfo {
1433 callee: codemap::NameAndSpan {
1434 format: codemap::MacroAttribute(intern(&format!("derive({})", trait_name))),
1435 span: Some(self.span),
1436 allow_internal_unstable: false,
1442 fn summarise_struct(&self,
1444 struct_def: &VariantData) -> StaticFields {
1445 let mut named_idents = Vec::new();
1446 let mut just_spans = Vec::new();
1447 for field in struct_def.fields(){
1448 let sp = self.set_expn_info(cx, field.span);
1449 match field.node.kind {
1450 ast::NamedField(ident, _) => named_idents.push((ident, sp)),
1451 ast::UnnamedField(..) => just_spans.push(sp),
1455 match (just_spans.is_empty(), named_idents.is_empty()) {
1456 (false, false) => cx.span_bug(self.span,
1457 "a struct with named and unnamed \
1458 fields in generic `derive`"),
1460 (_, false) => Named(named_idents),
1461 // tuple structs (includes empty structs)
1462 (_, _) => Unnamed(just_spans)
1466 fn create_subpatterns(&self,
1468 field_paths: Vec<ast::SpannedIdent> ,
1469 mutbl: ast::Mutability)
1470 -> Vec<P<ast::Pat>> {
1471 field_paths.iter().map(|path| {
1473 ast::PatIdent(ast::BindByRef(mutbl), (*path).clone(), None))
1477 fn create_struct_pattern(&self,
1479 struct_path: ast::Path,
1480 struct_def: &'a VariantData,
1482 mutbl: ast::Mutability)
1483 -> (P<ast::Pat>, Vec<(Span, Option<Ident>,
1485 &'a [ast::Attribute])>) {
1486 if struct_def.fields().count() == 0 {
1487 return (cx.pat_enum(self.span, struct_path, vec![]), vec![]);
1490 let mut paths = Vec::new();
1491 let mut ident_expr = Vec::new();
1492 let mut struct_type = Unknown;
1494 for (i, struct_field) in struct_def.fields().enumerate() {
1495 let sp = self.set_expn_info(cx, struct_field.span);
1496 let opt_id = match struct_field.node.kind {
1497 ast::NamedField(ident, _) if (struct_type == Unknown ||
1498 struct_type == Record) => {
1499 struct_type = Record;
1502 ast::UnnamedField(..) if (struct_type == Unknown ||
1503 struct_type == Tuple) => {
1504 struct_type = Tuple;
1508 cx.span_bug(sp, "a struct with named and unnamed fields in `derive`");
1511 let ident = cx.ident_of(&format!("{}_{}", prefix, i));
1512 paths.push(codemap::Spanned{span: sp, node: ident});
1514 sp, ast::ExprParen(cx.expr_deref(sp, cx.expr_path(cx.path_ident(sp,ident)))));
1515 ident_expr.push((sp, opt_id, val, &struct_field.node.attrs[..]));
1518 let subpats = self.create_subpatterns(cx, paths, mutbl);
1520 // struct_type is definitely not Unknown, since struct_def.fields
1521 // must be nonempty to reach here
1522 let pattern = if struct_type == Record {
1523 let field_pats = subpats.into_iter().zip(&ident_expr)
1524 .map(|(pat, &(_, id, _, _))| {
1525 // id is guaranteed to be Some
1528 node: ast::FieldPat { ident: id.unwrap(), pat: pat, is_shorthand: false },
1531 cx.pat_struct(self.span, struct_path, field_pats)
1533 cx.pat_enum(self.span, struct_path, subpats)
1536 (pattern, ident_expr)
1539 fn create_enum_variant_pattern(&self,
1541 enum_ident: ast::Ident,
1542 variant: &'a ast::Variant,
1544 mutbl: ast::Mutability)
1545 -> (P<ast::Pat>, Vec<(Span, Option<Ident>, P<Expr>, &'a [ast::Attribute])>) {
1546 let variant_ident = variant.node.name;
1547 let variant_path = cx.path(variant.span, vec![enum_ident, variant_ident]);
1548 self.create_struct_pattern(cx, variant_path, &variant.node.data, prefix, mutbl)
1552 /* helpful premade recipes */
1554 /// Fold the fields. `use_foldl` controls whether this is done
1555 /// left-to-right (`true`) or right-to-left (`false`).
1556 pub fn cs_fold<F>(use_foldl: bool,
1559 mut enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1562 substructure: &Substructure)
1564 F: FnMut(&mut ExtCtxt, Span, P<Expr>, P<Expr>, &[P<Expr>]) -> P<Expr>,
1566 match *substructure.fields {
1567 EnumMatching(_, _, ref all_fields) | Struct(ref all_fields) => {
1569 all_fields.iter().fold(base, |old, field| {
1573 field.self_.clone(),
1577 all_fields.iter().rev().fold(base, |old, field| {
1581 field.self_.clone(),
1586 EnumNonMatchingCollapsed(ref all_args, _, tuple) =>
1587 enum_nonmatch_f(cx, trait_span, (&all_args[..], tuple),
1588 substructure.nonself_args),
1589 StaticEnum(..) | StaticStruct(..) => {
1590 cx.span_bug(trait_span, "static function in `derive`")
1596 /// Call the method that is being derived on all the fields, and then
1597 /// process the collected results. i.e.
1600 /// f(cx, span, vec![self_1.method(__arg_1_1, __arg_2_1),
1601 /// self_2.method(__arg_1_2, __arg_2_2)])
1604 pub fn cs_same_method<F>(f: F,
1605 mut enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1608 substructure: &Substructure)
1610 F: FnOnce(&mut ExtCtxt, Span, Vec<P<Expr>>) -> P<Expr>,
1612 match *substructure.fields {
1613 EnumMatching(_, _, ref all_fields) | Struct(ref all_fields) => {
1614 // call self_n.method(other_1_n, other_2_n, ...)
1615 let called = all_fields.iter().map(|field| {
1616 cx.expr_method_call(field.span,
1617 field.self_.clone(),
1618 substructure.method_ident,
1620 .map(|e| cx.expr_addr_of(field.span, e.clone()))
1624 f(cx, trait_span, called)
1626 EnumNonMatchingCollapsed(ref all_self_args, _, tuple) =>
1627 enum_nonmatch_f(cx, trait_span, (&all_self_args[..], tuple),
1628 substructure.nonself_args),
1629 StaticEnum(..) | StaticStruct(..) => {
1630 cx.span_bug(trait_span, "static function in `derive`")
1635 /// Fold together the results of calling the derived method on all the
1636 /// fields. `use_foldl` controls whether this is done left-to-right
1637 /// (`true`) or right-to-left (`false`).
1639 pub fn cs_same_method_fold<F>(use_foldl: bool,
1642 enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1645 substructure: &Substructure)
1647 F: FnMut(&mut ExtCtxt, Span, P<Expr>, P<Expr>) -> P<Expr>,
1652 vals.into_iter().fold(base.clone(), |old, new| {
1653 f(cx, span, old, new)
1656 vals.into_iter().rev().fold(base.clone(), |old, new| {
1657 f(cx, span, old, new)
1662 cx, trait_span, substructure)
1665 /// Use a given binop to combine the result of calling the derived method
1666 /// on all the fields.
1668 pub fn cs_binop(binop: ast::BinOp_, base: P<Expr>,
1669 enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1670 cx: &mut ExtCtxt, trait_span: Span,
1671 substructure: &Substructure) -> P<Expr> {
1672 cs_same_method_fold(
1673 true, // foldl is good enough
1674 |cx, span, old, new| {
1675 cx.expr_binary(span,
1682 cx, trait_span, substructure)
1685 /// cs_binop with binop == or
1687 pub fn cs_or(enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1688 cx: &mut ExtCtxt, span: Span,
1689 substructure: &Substructure) -> P<Expr> {
1690 cs_binop(ast::BiOr, cx.expr_bool(span, false),
1692 cx, span, substructure)
1695 /// cs_binop with binop == and
1697 pub fn cs_and(enum_nonmatch_f: EnumNonMatchCollapsedFunc,
1698 cx: &mut ExtCtxt, span: Span,
1699 substructure: &Substructure) -> P<Expr> {
1700 cs_binop(ast::BiAnd, cx.expr_bool(span, true),
1702 cx, span, substructure)